/[escript]/trunk/escript/src/DataExpanded.cpp
ViewVC logotype

Diff of /trunk/escript/src/DataExpanded.cpp

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

trunk/esys2/escript/src/Data/DataExpanded.cpp revision 113 by jgs, Mon Feb 28 07:06:33 2005 UTC trunk/escript/src/DataExpanded.cpp revision 2742 by jfenwick, Thu Nov 12 06:03:37 2009 UTC
# Line 1  Line 1 
 // $Id$  
 /*  
  ******************************************************************************  
  *                                                                            *  
  *       COPYRIGHT  ACcESS 2004 -  All Rights Reserved                        *  
  *                                                                            *  
  * This software is the property of ACcESS. No part of this code              *  
  * may be copied in any form or by any means without the expressed written    *  
  * consent of ACcESS.  Copying, use or modification of this software          *  
  * by any unauthorised person is illegal unless that person has a software    *  
  * license agreement with ACcESS.                                             *  
  *                                                                            *  
  ******************************************************************************  
 */  
   
 #include "escript/Data/DataException.h"  
 #include "escript/Data/DataExpanded.h"  
 #include "escript/Data/DataConstant.h"  
 #include "escript/Data/DataTagged.h"  
 #include "escript/Data/DataArrayView.h"  
1    
2  #include <boost/python/extract.hpp>  /*******************************************************
3    *
4    * Copyright (c) 2003-2009 by University of Queensland
5    * Earth Systems Science Computational Center (ESSCC)
6    * http://www.uq.edu.au/esscc
7    *
8    * Primary Business: Queensland, Australia
9    * Licensed under the Open Software License version 3.0
10    * http://www.opensource.org/licenses/osl-3.0.php
11    *
12    *******************************************************/
13    
14    
15    #include "Data.h"
16    #include "DataExpanded.h"
17    #include "DataException.h"
18    #include "DataConstant.h"
19    #include "DataTagged.h"
20    #ifdef USE_NETCDF
21    #include <netcdfcpp.h>
22    #endif
23    #ifdef PASO_MPI
24    #include <mpi.h>
25    #endif
26    
27  #include <iostream>  #include <boost/python/extract.hpp>
28    #include "DataMaths.h"
29    
30  using namespace std;  using namespace std;
31  using namespace boost::python;  using namespace boost::python;
32  using namespace boost;  using namespace boost;
33    using namespace escript::DataTypes;
34    
35    
36    // #define CHECK_FOR_EX_WRITE if (!checkNoSharing()) {throw DataException("Attempt to modify shared object");}
37    
38    #define CHECK_FOR_EX_WRITE if (!checkNoSharing()) {std::ostringstream ss; ss << " Attempt to modify shared object. line " << __LINE__ << " of " << __FILE__; *((int*)0)=17;throw DataException(ss.str());}
39    
40  namespace escript {  namespace escript {
41    
42  DataExpanded::DataExpanded(const boost::python::numeric::array& value,  DataExpanded::DataExpanded(const WrappedArray& value,
43                             const FunctionSpace& what)                             const FunctionSpace& what)
44    : DataAbstract(what)    : parent(what,value.getShape())
45  {  {
   DataArrayView::ShapeType tempShape;  
46    //    //
47    // extract the shape from the python array    // initialise the data array for this object
48    for (int i=0; i<value.getrank(); ++i) {    initialise(what.getNumSamples(),what.getNumDPPSample());
     //cout << extract<int>(value.getshape()[i]) << endl;  
     tempShape.push_back(extract<int>(value.getshape()[i]));  
   }  
   initialise(tempShape,what.getNumSamples(),what.getNumDPPSample());  
49    //    //
50    // copy the given value to every data point    // copy the given value to every data point
51    copy(value);    copy(value);
52  }  }
53    
54  DataExpanded::DataExpanded(const DataExpanded& other)  DataExpanded::DataExpanded(const DataExpanded& other)
55    : DataAbstract(other.getFunctionSpace()),    : parent(other.getFunctionSpace(), other.getShape()),
56    m_data(other.m_data)    m_data(other.m_data)
57  {      {
   //  
   // create the view for the data  
   DataArrayView temp(m_data.getData(),other.getPointDataView().getShape());  
   setPointDataView(temp);  
58  }  }
59    
60  DataExpanded::DataExpanded(const DataConstant& other)  DataExpanded::DataExpanded(const DataConstant& other)
61    : DataAbstract(other.getFunctionSpace())    : parent(other.getFunctionSpace(), other.getShape())
62  {      {
   initialise(other.getPointDataView().getShape(),other.getNumSamples(),other.getNumDPPSample());  
63    //    //
64    // DataConstant only has one value    // initialise the data array for this object
65    copy(other.getPointDataView());    initialise(other.getNumSamples(),other.getNumDPPSample());
66      //
67      // DataConstant only has one value, copy this to every data point
68      copy(other);
69  }  }
70    
71  DataExpanded::DataExpanded(const DataTagged& other)  DataExpanded::DataExpanded(const DataTagged& other)
72    : DataAbstract(other.getFunctionSpace())    : parent(other.getFunctionSpace(), other.getShape())
73  {      {
74    initialise(other.getPointDataView().getShape(),other.getNumSamples(),other.getNumDPPSample());    //
75      // initialise the data array for this object
76      initialise(other.getNumSamples(),other.getNumDPPSample());
77      //
78      // for each data point in this object, extract and copy the corresponding data
79      // value from the given DataTagged object
80    int i,j;    int i,j;
81    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();    DataTypes::ValueType::size_type numRows=m_data.getNumRows();
82    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();    DataTypes::ValueType::size_type numCols=m_data.getNumCols();
83  #pragma omp parallel for private(i,j) schedule(static)    #pragma omp parallel for private(i,j) schedule(static)
84    for (i=0;i<numRows;++i) {    for (i=0;i<numRows;i++) {
85      for (j=0;j<numCols;++j) {      for (j=0;j<numCols;j++) {
86        try {        try {
87          getPointDataView().copy(getPointOffset(i,j), other.getPointDataView(), other.getPointOffset(i,j));             DataTypes::copyPoint(getVectorRW(), getPointOffset(i,j), getNoValues(),
88                                    other.getVectorRO(),
89                                    other.getPointOffset(i,j));
90        }        }
91        catch (std::exception& e) {        catch (std::exception& e) {
92          cout << e.what() << endl;          cout << e.what() << endl;
# Line 86  DataExpanded::DataExpanded(const DataTag Line 96  DataExpanded::DataExpanded(const DataTag
96  }  }
97    
98  DataExpanded::DataExpanded(const DataExpanded& other,  DataExpanded::DataExpanded(const DataExpanded& other,
99                             const DataArrayView::RegionType& region)                             const DataTypes::RegionType& region)
100    : DataAbstract(other.getFunctionSpace())    : parent(other.getFunctionSpace(),DataTypes::getResultSliceShape(region))
101  {  {
102    //    //
103    // get the shape of the slice    // get the shape of the slice
104    DataArrayView::ShapeType shape(DataArrayView::getResultSliceShape(region));  //   DataTypes::ShapeType shape(DataTypes::getResultSliceShape(region));
105    //    //
106    // initialise this Data object to the shape of the slice    // initialise this Data object to the shape of the slice
107    initialise(shape,other.getNumSamples(),other.getNumDPPSample());    initialise(other.getNumSamples(),other.getNumDPPSample());
   //  
   DataArrayView::RegionLoopRangeType region_loop_range=getSliceRegionLoopRange(region);  
108    //    //
109    // copy the data    // copy the data
110    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();    DataTypes::RegionLoopRangeType region_loop_range=DataTypes::getSliceRegionLoopRange(region);
111    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();    DataTypes::ValueType::size_type numRows=m_data.getNumRows();
112      DataTypes::ValueType::size_type numCols=m_data.getNumCols();
113    int i,j;    int i,j;
114  #pragma omp parallel for private(i,j) schedule(static)    #pragma omp parallel for private(i,j) schedule(static)
115    for (i=0;i<numRows;++i) {    for (i=0;i<numRows;i++) {
116      for (j=0;j<numCols;++j) {      for (j=0;j<numCols;j++) {
117        try {        try {
118          getPointDataView().copySlice(getPointOffset(i,j),other.getPointDataView(),other.getPointOffset(i,j),region_loop_range);          DataTypes::copySlice(getVectorRW(),getShape(),getPointOffset(i,j),
119                                         other.getVectorRO(),
120                         other.getShape(),
121                                         other.getPointOffset(i,j),
122                                         region_loop_range);
123        }        }
124        catch (std::exception& e) {        catch (std::exception& e) {
125          cout << e.what() << endl;          cout << e.what() << endl;
# Line 115  DataExpanded::DataExpanded(const DataExp Line 128  DataExpanded::DataExpanded(const DataExp
128    }    }
129  }  }
130    
131  DataExpanded::DataExpanded(const DataArrayView& value,  DataExpanded::DataExpanded(const FunctionSpace& what,
132                             const FunctionSpace& what)                             const DataTypes::ShapeType &shape,
133    : DataAbstract(what)                             const DataTypes::ValueType &data)
134  {    : parent(what,shape)
135    DataArrayView::ShapeType tempShape=value.getShape();  {
136    initialise(tempShape,what.getNumSamples(),what.getNumDPPSample());    EsysAssert(data.size()%getNoValues()==0,
137    copy(value);                   "DataExpanded Constructor - size of supplied data is not a multiple of shape size.");
138    
139      if (data.size()==getNoValues())
140      {
141         ValueType& vec=m_data.getData();
142         //
143         // create the view of the data
144         initialise(what.getNumSamples(),what.getNumDPPSample());
145         // now we copy this value to all elements
146         for (int i=0;i<getLength();)
147         {
148        for (unsigned int j=0;j<getNoValues();++j,++i)
149        {
150            vec[i]=data[j];
151        }
152         }
153      }
154      else
155      {
156         //
157         // copy the data in the correct format
158         m_data.getData()=data;
159      }
160    
161    
162  }  }
163    
164  DataExpanded::~DataExpanded()  DataExpanded::~DataExpanded()
165  {  {
166  }  }
167    
168  void  DataAbstract*
169  DataExpanded::reshapeDataPoint(const DataArrayView::ShapeType& shape)  DataExpanded::deepCopy()
170  {  {
171    //    return new DataExpanded(*this);
   // reshape a rank zero data point  
   if (getPointDataView().getRank()!=0) {  
     stringstream temp;  
     temp << "Error - Can only reshape Data with data points of rank 0. "  
          << "This Data has data points with rank: "  
          << getPointDataView().getRank();  
     throw DataException(temp.str());  
   }  
   DataBlocks2D newData(getNumSamples(),getNumDPPSample(), DataArrayView::noValues(shape));  
   DataArrayView newView(newData.getData(),shape);  
   //  
   // Copy the original data to every value for the new shape  
   int i,j;  
   int nRows=m_data.getNumRows();  
   int nCols=m_data.getNumCols();  
 #pragma omp parallel for private(i,j) schedule(static)  
   for (i=0;i<nRows;++i) {  
     for (j=0;j<nCols;++j) {  
       //  
       // Copy the data into the specified offset  
       // NOTE: An exception may be thown from this call if  
       // DOASSERT is on which of course will play  
       // havoc with the omp threads. Run single threaded  
       // if using DOASSERT.  
       newView.copy(newData.index(i,j),m_data.getData()[m_data.index(i,j)]);  
     }  
   }  
   m_data.Swap(newData);  
   DataArrayView temp(m_data.getData(),shape);  
   setPointDataView(temp);  
172  }  }
173    
174    
175  DataAbstract*  DataAbstract*
176  DataExpanded::getSlice(const DataArrayView::RegionType& region) const  DataExpanded::getSlice(const DataTypes::RegionType& region) const
177  {  {
178    return new DataExpanded(*this,region);    return new DataExpanded(*this,region);
179  }  }
180    
181  void  void
182  DataExpanded::setSlice(const DataAbstract* value,  DataExpanded::setSlice(const DataAbstract* value,
183                         const DataArrayView::RegionType& region)                         const DataTypes::RegionType& region)
184  {  {
185    const DataExpanded* tempDataExp=dynamic_cast<const DataExpanded*>(value);    const DataExpanded* tempDataExp=dynamic_cast<const DataExpanded*>(value);
186    if (tempDataExp==0) {    if (tempDataExp==0) {
187      throw DataException("Programming error - casting to DataExpanded.");      throw DataException("Programming error - casting to DataExpanded.");
188    }    }
189      CHECK_FOR_EX_WRITE
190    //    //
191    DataArrayView::ShapeType shape(DataArrayView::getResultSliceShape(region));    // get shape of slice
192    DataArrayView::RegionLoopRangeType region_loop_range=getSliceRegionLoopRange(region);    DataTypes::ShapeType shape(DataTypes::getResultSliceShape(region));
193      DataTypes::RegionLoopRangeType region_loop_range=DataTypes::getSliceRegionLoopRange(region);
194    //    //
195    // check shape:    // check shape
196    if (getPointDataView().getRank()!=region.size()) {    if (getRank()!=region.size()) {
197      throw DataException("Error - Invalid slice region.");      throw DataException("Error - Invalid slice region.");
198    }    }
199    if (tempDataExp->getPointDataView().getRank()>0 and !value->getPointDataView().checkShape(shape)) {    if (tempDataExp->getRank()>0 && !DataTypes::checkShape(value->getShape(), shape)) {
200      throw DataException (value->getPointDataView().createShapeErrorMessage(      throw DataException (DataTypes::createShapeErrorMessage(
201          "Error - Couldn't copy slice due to shape mismatch.",shape));          "Error - Couldn't copy slice due to shape mismatch.",shape, value->getShape()));
202    }    }
203    //    //
204    // copy the data    // copy the data from the slice into this object
205    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();    DataTypes::ValueType::size_type numRows=m_data.getNumRows();
206    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();    DataTypes::ValueType::size_type numCols=m_data.getNumCols();
207    int i, j;    int i, j;
208  #pragma omp parallel for private(i,j) schedule(static)    ValueType& vec=getVectorRW();
209      const ShapeType& mshape=getShape();
210      const ValueType& tVec=tempDataExp->getVectorRO();
211      const ShapeType& tShape=tempDataExp->getShape();
212      #pragma omp parallel for private(i,j) schedule(static)
213    for (i=0;i<numRows;i++) {    for (i=0;i<numRows;i++) {
214      for (j=0;j<numCols;j++) {      for (j=0;j<numCols;j++) {
215        getPointDataView().copySliceFrom(getPointOffset(i,j),tempDataExp->getPointDataView(),tempDataExp->getPointOffset(i,j),region_loop_range);          DataTypes::copySliceFrom(vec,mshape,getPointOffset(i,j),
216                                           tVec,
217                           tShape,
218                                           tempDataExp->getPointOffset(i,j),
219                                           region_loop_range);
220    
221      }      }
222    }    }
223  }  }
224    
225  void  void
226  DataExpanded::copy(const DataArrayView& value)  DataExpanded::copy(const DataConstant& value)
227  {  {
228      EsysAssert((checkShape(getShape(), value.getShape())),
229                     createShapeErrorMessage("Error - Couldn't copy due to shape mismatch.",value.getShape(),getShape()));
230    
231    //    //
232    // Copy a single value to every data point    // copy a single value to every data point in this object
233    int nRows=m_data.getNumRows();    int nRows=m_data.getNumRows();
234    int nCols=m_data.getNumCols();    int nCols=m_data.getNumCols();
235    int i, j;    int i,j;
236  #pragma omp parallel for private(i,j) schedule(static)    #pragma omp parallel for private(i,j) schedule(static)
237    for (i=0;i<nRows;++i) {    for (i=0;i<nRows;i++) {
238      for (j=0;j<nCols;++j) {      for (j=0;j<nCols;j++) {
239        //        DataTypes::copyPoint(getVectorRW(), getPointOffset(i,j), getNoValues(), value.getVectorRO(), 0);
       // Copy the data into the specified offset  
       // NOTE: An exception may be thown from this call if  
       // DOASSERT is on which of course will play  
       // havoc with the omp threads. Run single threaded  
       // if using DOASSERT.  
       getPointDataView().copy(m_data.index(i,j),value);  
240      }      }
241    }    }
242  }  }
243    
244  void  void
245  DataExpanded::copy(const boost::python::numeric::array& value)  DataExpanded::copy(const WrappedArray& value)
246  {  {
247    //    // check the input shape matches this shape
248    // first convert the numarray into a DataArrayView format    if (!DataTypes::checkShape(getShape(),value.getShape())) {
249    DataArray temp(value);      throw DataException(DataTypes::createShapeErrorMessage(
   //  
   // check the input shape matches this shape, this will throw an exception  
   if (!getPointDataView().checkShape(temp.getView().getShape())) {  
     throw DataException(getPointDataView().createShapeErrorMessage(  
250                          "Error - (DataExpanded) Cannot copy due to shape mismatch.",                          "Error - (DataExpanded) Cannot copy due to shape mismatch.",
251                          temp.getView().getShape()));                          value.getShape(),getShape()));
252    }    }
253    //    getVectorRW().copyFromArray(value, getNumDPPSample()*getNumSamples());
   // now copy over the entire data structure  
   copy(temp.getView());  
254  }  }
255    
256    
257  void  void
258  DataExpanded::initialise(const DataArrayView::ShapeType& shape,  DataExpanded::initialise(int noSamples,
                          int noSamples,  
259                           int noDataPointsPerSample)                           int noDataPointsPerSample)
260  {  {
261      if (noSamples==0)     //retain the default empty object
262      {
263         return;
264      }
265    //    //
266    // resize data to the required size    // resize data array to the required size
267    m_data.resize(noSamples,noDataPointsPerSample,DataArrayView::noValues(shape));    m_data.resize(noSamples,noDataPointsPerSample,getNoValues());
   //  
   // create a point data viewer of the data  
   DataArrayView temp(m_data.getData(),shape);  
   setPointDataView(temp);  
268  }  }
269    
270  string  string
271  DataExpanded::toString() const  DataExpanded::toString() const
272  {  {
273    stringstream temp;    stringstream temp;
274    //    FunctionSpace fs=getFunctionSpace();
275    // create a temporary view as the offset will be changed  
276    DataArrayView tempView(getPointDataView().getData(),getPointDataView().getShape(),getPointDataView().getOffset());    int offset=0;
277    for (int i=0;i<m_data.getNumRows();++i) {    for (int i=0;i<m_data.getNumRows();i++) {
278      for (int j=0;j<m_data.getNumCols();++j) {      for (int j=0;j<m_data.getNumCols();j++) {
279        tempView.setOffset(m_data.index(i,j));        offset=getPointOffset(i,j);
280        stringstream suffix;        stringstream suffix;
281        suffix << "(" << i << "," << j << ")";        suffix << "( id: " << i << ", ref: " << fs.getReferenceIDOfSample(i) << ", pnt: " << j << ")";
282        temp << tempView.toString(suffix.str());        temp << DataTypes::pointToString(getVectorRO(),getShape(),offset,suffix.str());
283        if (!(i==(m_data.getNumRows()-1) && j==(m_data.getNumCols()-1))) {        if (!(i==(m_data.getNumRows()-1) && j==(m_data.getNumCols()-1))) {
284          temp << endl;          temp << endl;
285        }        }
286      }      }
287    }    }
288      string result=temp.str();
289      if (result.empty())
290      {
291        return "(data contains no samples)\n";
292      }
293    return temp.str();    return temp.str();
294  }  }
295    
296  DataArrayView::ValueType::size_type  DataTypes::ValueType::size_type
297  DataExpanded::getPointOffset(int sampleNo,  DataExpanded::getPointOffset(int sampleNo,
298                               int dataPointNo) const                               int dataPointNo) const
299  {  {
300    return m_data.index(sampleNo,dataPointNo);    return m_data.index(sampleNo,dataPointNo);
301  }  }
302    
303  DataArrayView  DataTypes::ValueType::size_type
304  DataExpanded::getDataPoint(int sampleNo,  DataExpanded::getPointOffset(int sampleNo,
305                             int dataPointNo)                               int dataPointNo)
306  {  {
307    DataArrayView temp(m_data.getData(),getPointDataView().getShape(),m_data.index(sampleNo,dataPointNo));    return m_data.index(sampleNo,dataPointNo);
   return temp;  
308  }  }
309    
310  DataArrayView::ValueType::size_type  DataTypes::ValueType::size_type
311  DataExpanded::getLength() const  DataExpanded::getLength() const
312  {  {
313    return m_data.getData().size();    return m_data.size();
314  }  }
315    
316    
317    
318  void  void
319  DataExpanded::setRefValue(int ref,  DataExpanded::copyToDataPoint(const int sampleNo, const int dataPointNo, const double value) {
320                            const DataArray& value)    CHECK_FOR_EX_WRITE
 {  
321    //    //
322    // Get the number of samples and data-points per sample.    // Get the number of samples and data-points per sample.
323    int numSamples = getNumSamples();    int numSamples = getNumSamples();
324    int numDPPSample = getNumDPPSample();    int numDataPointsPerSample = getNumDPPSample();
325      int dataPointRank = getRank();
326      ShapeType dataPointShape = getShape();
327      if (numSamples*numDataPointsPerSample > 0) {
328         //TODO: global error handling
329         if ((sampleNo >= numSamples) || (sampleNo < 0 )) {
330              throw DataException("Error - DataExpanded::copyDataPoint invalid sampleNo.");
331         }
332         if ((dataPointNo >= numDataPointsPerSample) || (dataPointNo < 0)) {
333               throw DataException("Error - DataExpanded::copyDataPoint invalid dataPointNoInSample.");
334         }
335         ValueType::size_type offset = getPointOffset(sampleNo, dataPointNo);
336         ValueType& vec=getVectorRW();
337         if (dataPointRank==0) {
338             vec[offset]=value;
339         } else if (dataPointRank==1) {
340            for (int i=0; i<dataPointShape[0]; i++) {
341                vec[offset+i]=value;
342            }
343         } else if (dataPointRank==2) {
344            for (int i=0; i<dataPointShape[0]; i++) {
345               for (int j=0; j<dataPointShape[1]; j++) {
346                  vec[offset+getRelIndex(dataPointShape,i,j)]=value;
347               }
348            }
349         } else if (dataPointRank==3) {
350            for (int i=0; i<dataPointShape[0]; i++) {
351               for (int j=0; j<dataPointShape[1]; j++) {
352                  for (int k=0; k<dataPointShape[2]; k++) {
353                     vec[offset+getRelIndex(dataPointShape,i,j,k)]=value;
354                  }
355               }
356            }
357         } else if (dataPointRank==4) {
358             for (int i=0; i<dataPointShape[0]; i++) {
359               for (int j=0; j<dataPointShape[1]; j++) {
360                 for (int k=0; k<dataPointShape[2]; k++) {
361                   for (int l=0; l<dataPointShape[3]; l++) {
362                      vec[offset+getRelIndex(dataPointShape,i,j,k,l)]=value;
363                   }
364                 }
365               }
366             }
367         }
368      }
369    }
370    
371    void
372    DataExpanded::copyToDataPoint(const int sampleNo, const int dataPointNo, const WrappedArray& value) {
373      CHECK_FOR_EX_WRITE
374      //
375      // Get the number of samples and data-points per sample.
376      int numSamples = getNumSamples();
377      int numDataPointsPerSample = getNumDPPSample();
378    //    //
379    // Determine the sample number which corresponds to this reference number.    // check rank:
380    int sampleNo = -1;    if (value.getRank()!=getRank())
381    int tempRef = -1;         throw DataException("Rank of value does not match Data object rank");
382    for (int n=0; n<numSamples; n++) {    if (numSamples*numDataPointsPerSample > 0) {
383      tempRef = getFunctionSpace().getReferenceNoFromSampleNo(n);       //TODO: global error handling
384      if (tempRef == ref) {       if ((sampleNo >= numSamples) || (sampleNo < 0 )) {
385        sampleNo = n;            throw DataException("Error - DataExpanded::copyDataPoint invalid sampleNo.");
386        break;       }
387         if ((dataPointNo >= numDataPointsPerSample) || (dataPointNo < 0)) {
388               throw DataException("Error - DataExpanded::copyDataPoint invalid dataPointNoInSample.");
389         }
390         ValueType::size_type offset = getPointOffset(sampleNo, dataPointNo);
391         ValueType& vec=getVectorRW();
392         vec.copyFromArrayToOffset(value,offset,1);
393      }
394    }
395    
396    void
397    DataExpanded::symmetric(DataAbstract* ev)
398    {
399      int sampleNo,dataPointNo;
400      int numSamples = getNumSamples();
401      int numDataPointsPerSample = getNumDPPSample();
402      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
403      if (temp_ev==0) {
404        throw DataException("Error - DataExpanded::symmetric: casting to DataExpanded failed (propably a programming error).");
405      }
406      const ValueType& vec=getVectorRO();
407      const ShapeType& shape=getShape();
408      ValueType& evVec=temp_ev->getVectorRW();
409      const ShapeType& evShape=temp_ev->getShape();
410      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
411      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
412        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
413             DataMaths::symmetric(vec,shape,getPointOffset(sampleNo,dataPointNo),
414                                        evVec,evShape,ev->getPointOffset(sampleNo,dataPointNo));
415      }      }
416    }    }
417    if (sampleNo == -1) {  }
418      throw DataException("DataExpanded::setRefValue error: invalid ref number supplied.");  void
419    DataExpanded::nonsymmetric(DataAbstract* ev)
420    {
421      int sampleNo,dataPointNo;
422      int numSamples = getNumSamples();
423      int numDataPointsPerSample = getNumDPPSample();
424      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
425      if (temp_ev==0) {
426        throw DataException("Error - DataExpanded::nonsymmetric: casting to DataExpanded failed (propably a programming error).");
427      }
428      const ValueType& vec=getVectorRO();
429      const ShapeType& shape=getShape();
430      ValueType& evVec=temp_ev->getVectorRW();
431      const ShapeType& evShape=temp_ev->getShape();
432      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
433      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
434        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
435             DataMaths::nonsymmetric(vec,shape,getPointOffset(sampleNo,dataPointNo),
436                                        evVec,evShape,ev->getPointOffset(sampleNo,dataPointNo));
437        }
438    }    }
439    }
440    void
441    DataExpanded::trace(DataAbstract* ev, int axis_offset)
442    {
443      int sampleNo,dataPointNo;
444      int numSamples = getNumSamples();
445      int numDataPointsPerSample = getNumDPPSample();
446      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
447      if (temp_ev==0) {
448        throw DataException("Error - DataExpanded::trace: casting to DataExpanded failed (propably a programming error).");
449      }
450      const ValueType& vec=getVectorRO();
451      const ShapeType& shape=getShape();
452      ValueType& evVec=temp_ev->getVectorRW();
453      const ShapeType& evShape=temp_ev->getShape();
454      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
455      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
456        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
457             DataMaths::trace(vec,shape,getPointOffset(sampleNo,dataPointNo),
458                                        evVec,evShape,ev->getPointOffset(sampleNo,dataPointNo),axis_offset);
459        }
460      }
461    }
462    
463    for (int n=0; n<numDPPSample; n++) {  void
464      //  DataExpanded::transpose(DataAbstract* ev, int axis_offset)
465      // Get each data-point in the sample in turn.  {
466      DataArrayView pointView = getDataPoint(sampleNo, n);    int sampleNo,dataPointNo;
467      int numSamples = getNumSamples();
468      //    int numDataPointsPerSample = getNumDPPSample();
469      // Assign the values in the DataArray to this data-point.    DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
470      pointView.copy(value.getView());    if (temp_ev==0) {
471        throw DataException("Error - DataExpanded::transpose: casting to DataExpanded failed (propably a programming error).");
472      }
473      const ValueType& vec=getVectorRO();
474      const ShapeType& shape=getShape();
475      ValueType& evVec=temp_ev->getVectorRW();
476      const ShapeType& evShape=temp_ev->getShape();
477      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
478      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
479        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
480             DataMaths::transpose(vec,shape,getPointOffset(sampleNo,dataPointNo),
481                                        evVec,evShape,ev->getPointOffset(sampleNo,dataPointNo),axis_offset);
482        }
483    }    }
484  }  }
485    
486  void  void
487  DataExpanded::getRefValue(int ref,  DataExpanded::swapaxes(DataAbstract* ev, int axis0, int axis1)
488                            DataArray& value)  {
489      int sampleNo,dataPointNo;
490      int numSamples = getNumSamples();
491      int numDataPointsPerSample = getNumDPPSample();
492      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
493      if (temp_ev==0) {
494        throw DataException("Error - DataExpanded::swapaxes: casting to DataExpanded failed (propably a programming error).");
495      }
496      const ValueType& vec=getVectorRO();
497      const ShapeType& shape=getShape();
498      ValueType& evVec=temp_ev->getVectorRW();
499      const ShapeType& evShape=temp_ev->getShape();
500      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
501      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
502        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
503             DataMaths::swapaxes(vec,shape,getPointOffset(sampleNo,dataPointNo),
504                                        evVec,evShape,ev->getPointOffset(sampleNo,dataPointNo),axis0,axis1);
505        }
506      }
507    }
508    void
509    DataExpanded::eigenvalues(DataAbstract* ev)
510    {
511      int sampleNo,dataPointNo;
512      int numSamples = getNumSamples();
513      int numDataPointsPerSample = getNumDPPSample();
514      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
515      if (temp_ev==0) {
516        throw DataException("Error - DataExpanded::eigenvalues: casting to DataExpanded failed (propably a programming error).");
517      }
518      const ValueType& vec=getVectorRO();
519      const ShapeType& shape=getShape();
520      ValueType& evVec=temp_ev->getVectorRW();
521      const ShapeType& evShape=temp_ev->getShape();
522      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
523      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
524        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
525             DataMaths::eigenvalues(vec,shape,getPointOffset(sampleNo,dataPointNo),
526                                        evVec,evShape,ev->getPointOffset(sampleNo,dataPointNo));
527        }
528      }
529    }
530    void
531    DataExpanded::eigenvalues_and_eigenvectors(DataAbstract* ev,DataAbstract* V,const double tol)
532  {  {
   //  
   // Get the number of samples and data-points per sample.  
533    int numSamples = getNumSamples();    int numSamples = getNumSamples();
534    int numDPPSample = getNumDPPSample();    int numDataPointsPerSample = getNumDPPSample();
535      int sampleNo,dataPointNo;
536      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
537      if (temp_ev==0) {
538        throw DataException("Error - DataExpanded::eigenvalues_and_eigenvectors: casting to DataExpanded failed (propably a programming error).");
539      }
540      DataExpanded* temp_V=dynamic_cast<DataExpanded*>(V);
541      if (temp_V==0) {
542        throw DataException("Error - DataExpanded::eigenvalues_and_eigenvectors: casting to DataExpanded failed (propably a programming error).");
543      }
544      const ValueType& vec=getVectorRO();
545      const ShapeType& shape=getShape();
546      ValueType& evVec=temp_ev->getVectorRW();
547      const ShapeType& evShape=temp_ev->getShape();
548      ValueType& VVec=temp_V->getVectorRW();
549      const ShapeType& VShape=temp_V->getShape();
550      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
551      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
552        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
553             DataMaths::eigenvalues_and_eigenvectors(vec,shape,getPointOffset(sampleNo,dataPointNo),
554                                        evVec,evShape,ev->getPointOffset(sampleNo,dataPointNo),
555                                        VVec, VShape,V->getPointOffset(sampleNo,dataPointNo),tol);
556        }
557      }
558    }
559    
560    //  
561    // Determine the sample number which corresponds to this reference number  void
562    int sampleNo = -1;  DataExpanded::matrixInverse(DataAbstract* out) const
563    int tempRef = -1;  {
564    for (int n=0; n<numSamples; n++) {    DataExpanded* temp=dynamic_cast<DataExpanded*>(out);
565      tempRef = getFunctionSpace().getReferenceNoFromSampleNo(n);    if (temp==0)
566      if (tempRef == ref) {    {
567        sampleNo = n;      throw DataException("Error - DataExpanded::matrixInverse: casting to DataExpanded failed (propably a programming error).");
568        break;    }
569    
570      if (getRank()!=2)
571      {
572        throw DataException("Error - DataExpanded::matrixInverse: input must be rank 2.");
573    
574      }
575      int  sampleNo;
576      const int numdpps=getNumDPPSample();
577      const int numSamples = getNumSamples();
578      const ValueType& vec=m_data.getData();
579      int errcode=0;
580      #pragma omp parallel private(sampleNo)
581      {
582         int errorcode=0;
583         LapackInverseHelper h(getShape()[0]);
584         #pragma omp for schedule(static)
585         for (sampleNo = 0; sampleNo < numSamples; sampleNo++)
586         {
587                // not sure I like all those virtual calls to getPointOffset
588            DataTypes::ValueType::size_type offset=getPointOffset(sampleNo,0);
589            int res=DataMaths::matrix_inverse(vec, getShape(), offset, temp->getVectorRW(), temp->getShape(), offset, numdpps, h);
590        if (res>errorcode)
591        {
592            errorcode=res;
593            #pragma omp critical
594            {
595              errcode=errorcode;    // I'm not especially concerned which error gets reported as long as one is
596            }
597        }
598         }
599      }
600      if (errcode)
601      {
602        DataMaths::matrixInverseError(errcode); // throws exceptions
603      }
604    }
605    
606    void
607    DataExpanded::setToZero(){
608    // TODO: Surely there is a more efficient way to do this????
609    // Why is there no memset here? Parallel issues?
610    // A: This ensures that memory is touched by the correct thread.
611      CHECK_FOR_EX_WRITE
612      int numSamples = getNumSamples();
613      int numDataPointsPerSample = getNumDPPSample();
614      DataTypes::ValueType::size_type n = getNoValues();
615      double* p;
616      int  sampleNo,dataPointNo, i;
617      #pragma omp parallel for private(sampleNo,dataPointNo,p,i) schedule(static)
618      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
619        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
620            p=&(m_data[getPointOffset(sampleNo,dataPointNo)]);
621            for (i=0; i<n ;++i) p[i]=0.;
622      }      }
623    }    }
624    if (sampleNo == -1) {  }
625      throw DataException("DataExpanded::getRefValue error: invalid ref number supplied.");  
626    /* Append MPI rank to file name if multiple MPI processes */
627    char *Escript_MPI_appendRankToFileName(const char *fileName, int mpi_size, int mpi_rank) {
628      /* Make plenty of room for the mpi_rank number and terminating '\0' */
629      char *newFileName = (char *)malloc(strlen(fileName)+20);
630      strncpy(newFileName, fileName, strlen(fileName)+1);
631      if (mpi_size>1) sprintf(newFileName+strlen(newFileName), ".%04d", mpi_rank);
632      return(newFileName);
633    }
634    
635    void
636    DataExpanded::dump(const std::string fileName) const
637    {
638       #ifdef USE_NETCDF
639       const int ldims=2+DataTypes::maxRank;
640       const NcDim* ncdims[ldims];
641       NcVar *var, *ids;
642       int rank = getRank();
643       int type=  getFunctionSpace().getTypeCode();
644       int ndims =0;
645       long dims[ldims];
646       const double* d_ptr=&(m_data[0]);
647       const DataTypes::ShapeType& shape = getShape();
648       int mpi_iam=getFunctionSpace().getDomain()->getMPIRank();
649       int mpi_num=getFunctionSpace().getDomain()->getMPISize();
650    #ifdef PASO_MPI
651       MPI_Status status;
652    #endif
653    
654    #ifdef PASO_MPI
655       /* Serialize NetCDF I/O */
656       if (mpi_iam>0) MPI_Recv(&ndims, 0, MPI_INT, mpi_iam-1, 81801, MPI_COMM_WORLD, &status);
657    #endif
658    
659       // netCDF error handler
660       NcError err(NcError::verbose_nonfatal);
661       // Create the file.
662       char *newFileName = Escript_MPI_appendRankToFileName(fileName.c_str(), mpi_num, mpi_iam);
663       NcFile dataFile(newFileName, NcFile::Replace);
664       // check if writing was successful
665       if (!dataFile.is_valid())
666            throw DataException("Error - DataExpanded:: opening of netCDF file for output failed.");
667       if (!dataFile.add_att("type_id",2) )
668            throw DataException("Error - DataExpanded:: appending data type to netCDF file failed.");
669       if (!dataFile.add_att("rank",rank) )
670            throw DataException("Error - DataExpanded:: appending rank attribute to netCDF file failed.");
671       if (!dataFile.add_att("function_space_type",type))
672            throw DataException("Error - DataExpanded:: appending function space attribute to netCDF file failed.");
673       ndims=rank+2;
674       if ( rank >0 ) {
675           dims[0]=shape[0];
676           if (! (ncdims[0] = dataFile.add_dim("d0",shape[0])) )
677                throw DataException("Error - DataExpanded:: appending ncdimension 0 to netCDF file failed.");
678       }
679       if ( rank >1 ) {
680           dims[1]=shape[1];
681           if (! (ncdims[1] = dataFile.add_dim("d1",shape[1])) )
682                throw DataException("Error - DataExpanded:: appending ncdimension 1 to netCDF file failed.");
683       }
684       if ( rank >2 ) {
685           dims[2]=shape[2];
686           if (! (ncdims[2] = dataFile.add_dim("d2", shape[2])) )
687                throw DataException("Error - DataExpanded:: appending ncdimension 2 to netCDF file failed.");
688       }
689       if ( rank >3 ) {
690           dims[3]=shape[3];
691           if (! (ncdims[3] = dataFile.add_dim("d3", shape[3])) )
692                throw DataException("Error - DataExpanded:: appending ncdimension 3 to netCDF file failed.");
693       }
694       dims[rank]=getFunctionSpace().getNumDataPointsPerSample();
695       if (! (ncdims[rank] = dataFile.add_dim("num_data_points_per_sample", dims[rank])) )
696                throw DataException("Error - DataExpanded:: appending num_data_points_per_sample to netCDF file failed.");
697       dims[rank+1]=getFunctionSpace().getNumSamples();
698       if (! (ncdims[rank+1] = dataFile.add_dim("num_samples", dims[rank+1])) )
699                throw DataException("Error - DataExpanded:: appending num_sample to netCDF file failed.");
700    
701       if (getFunctionSpace().getNumSamples()>0)
702       {
703    
704         if (! ( ids = dataFile.add_var("id", ncInt, ncdims[rank+1])) )
705            throw DataException("Error - DataExpanded:: appending reference id to netCDF file failed.");
706         const int* ids_p=getFunctionSpace().borrowSampleReferenceIDs();
707         if (! (ids->put(ids_p,dims[rank+1])) )
708            throw DataException("Error - DataExpanded:: copy reference id  to netCDF buffer failed.");
709         if (! ( var = dataFile.add_var("data", ncDouble, ndims, ncdims)) )
710            throw DataException("Error - DataExpanded:: appending variable to netCDF file failed.");
711         if (! (var->put(d_ptr,dims)) )
712            throw DataException("Error - DataExpanded:: copy data to netCDF buffer failed.");
713       }
714    #ifdef PASO_MPI
715       if (mpi_iam<mpi_num-1) MPI_Send(&ndims, 0, MPI_INT, mpi_iam+1, 81801, MPI_COMM_WORLD);
716    #endif
717       #else
718       throw DataException("Error - DataExpanded:: dump is not configured with netCDF. Please contact your installation manager.");
719       #endif
720    }
721    
722    void  
723    DataExpanded::setTaggedValue(int tagKey,
724               const DataTypes::ShapeType& pointshape,
725                   const DataTypes::ValueType& value,
726               int dataOffset)
727    {
728      CHECK_FOR_EX_WRITE
729      int numSamples = getNumSamples();
730      int numDataPointsPerSample = getNumDPPSample();
731      int sampleNo,dataPointNo, i;
732      DataTypes::ValueType::size_type n = getNoValues();
733      double* p;
734      const double* in=&value[0+dataOffset];
735      
736      if (value.size() != n) {
737        throw DataException("Error - DataExpanded::setTaggedValue: number of input values does not match number of values per data points.");
738      }
739    
740      #pragma omp parallel for private(sampleNo,dataPointNo,p,i) schedule(static)
741      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
742        if (getFunctionSpace().getTagFromSampleNo(sampleNo) == tagKey ) {
743            for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
744                p=&(m_data[getPointOffset(sampleNo,dataPointNo)]);
745                for (i=0; i<n ;++i) p[i]=in[i];
746            }
747        }
748    }    }
749    }
750    
   //  
   // Get the first data-point associated with this sample number.  
   DataArrayView pointView = getDataPoint(sampleNo, 0);  
751    
752    //  void
753    // Load the values from this data-point into the DataArray  DataExpanded::reorderByReferenceIDs(int *reference_ids)
754    value.getView().copy(pointView);  {
755      CHECK_FOR_EX_WRITE
756      int numSamples = getNumSamples();
757      DataTypes::ValueType::size_type n = getNoValues() * getNumDPPSample();
758      int sampleNo, sampleNo2,i;
759      double* p,*p2;
760      register double rtmp;
761      FunctionSpace fs=getFunctionSpace();
762    
763      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
764         const int id_in=reference_ids[sampleNo];
765         const int id=fs.getReferenceIDOfSample(sampleNo);
766         if (id!=id_in) {
767             bool matched=false;
768             for (sampleNo2 = sampleNo+1; sampleNo2 < numSamples; sampleNo2++) {
769                  if (id == reference_ids[sampleNo2]) {
770                     p=&(m_data[getPointOffset(sampleNo,0)]);
771                     p2=&(m_data[getPointOffset(sampleNo2,0)]);
772                     for (i=0; i<n ;i++) {
773                             rtmp=p[i];
774                             p[i]=p2[i];
775                             p2[i]=rtmp;
776                     }
777                     reference_ids[sampleNo]=id;
778                     reference_ids[sampleNo2]=id_in;
779                     matched=true;
780                     break;
781                  }
782             }
783             if (! matched) {
784                throw DataException("Error - DataExpanded::reorderByReferenceIDs: unable to reorder sample data by reference ids");
785             }
786         }
787       }
788    }
789    
790    DataTypes::ValueType&
791    DataExpanded::getVectorRW()
792    {
793        CHECK_FOR_EX_WRITE
794        return m_data.getData();
795  }  }
796    
797    const DataTypes::ValueType&
798    DataExpanded::getVectorRO() const
799    {
800        return m_data.getData();
801    }
802    
803    
804  }  // end of namespace  }  // end of namespace

Legend:
Removed from v.113  
changed lines
  Added in v.2742

  ViewVC Help
Powered by ViewVC 1.1.26