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

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

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

trunk/escript/src/Data/DataTagged.cpp revision 155 by jgs, Wed Nov 9 02:02:19 2005 UTC trunk/escript/src/DataTagged.cpp revision 594 by gross, Tue Mar 14 23:36:08 2006 UTC
# Line 14  Line 14 
14   ******************************************************************************   ******************************************************************************
15  */  */
16    
17  #include "escript/Data/DataTagged.h"  #include "DataTagged.h"
 #include "escript/Data/DataConstant.h"  
 #include "escript/Data/DataExpanded.h"  
 #include "escript/Data/DataException.h"  
18    
19  #include <sstream>  #include "DataConstant.h"
20    #include "DataException.h"
21    
22  using namespace std;  using namespace std;
23    
# Line 28  namespace escript { Line 26  namespace escript {
26  DataTagged::DataTagged()  DataTagged::DataTagged()
27    : DataAbstract(FunctionSpace())    : DataAbstract(FunctionSpace())
28  {  {
29      // default constructor
30    
31    // create a scalar default value    // create a scalar default value
32    m_data.resize(1,0.,1);    m_data.resize(1,0.,1);
33    DataArrayView temp(m_data,DataArrayView::ShapeType());    DataArrayView temp(m_data,DataArrayView::ShapeType());
# Line 40  DataTagged::DataTagged(const TagListType Line 40  DataTagged::DataTagged(const TagListType
40                 const FunctionSpace& what)                 const FunctionSpace& what)
41    : DataAbstract(what)    : DataAbstract(what)
42  {  {
43      // constructor
44    
45    // initialise the array of data values    // initialise the array of data values
46    // the default value is always the first item in the values list    // the default value is always the first item in the values list
47    int len = defaultValue.noValues();    int len = defaultValue.noValues();
# Line 59  DataTagged::DataTagged(const TagListType Line 61  DataTagged::DataTagged(const TagListType
61  DataTagged::DataTagged(const FunctionSpace& what,  DataTagged::DataTagged(const FunctionSpace& what,
62                         const DataArrayView::ShapeType &shape,                         const DataArrayView::ShapeType &shape,
63                         const int tags[],                         const int tags[],
64                         const ValueType &data)                         const ValueType& data)
65    : DataAbstract(what)    : DataAbstract(what)
66  {  {
67      // alternative constructor
68      // not unit_tested tested yet
69    
70    // copy the data    // copy the data
71    m_data=data;    m_data=data;
72    
# Line 80  DataTagged::DataTagged(const DataTagged& Line 85  DataTagged::DataTagged(const DataTagged&
85    m_data(other.m_data),    m_data(other.m_data),
86    m_offsetLookup(other.m_offsetLookup)    m_offsetLookup(other.m_offsetLookup)
87  {  {
88      // copy constructor
89    
90    // create the data view    // create the data view
91    DataArrayView temp(m_data,other.getPointDataView().getShape());    DataArrayView temp(m_data,other.getPointDataView().getShape());
92    setPointDataView(temp);    setPointDataView(temp);
# Line 88  DataTagged::DataTagged(const DataTagged& Line 95  DataTagged::DataTagged(const DataTagged&
95  DataTagged::DataTagged(const DataConstant& other)  DataTagged::DataTagged(const DataConstant& other)
96    : DataAbstract(other.getFunctionSpace())    : DataAbstract(other.getFunctionSpace())
97  {  {
98      // copy constructor
99    
100    // fill the default value with the constant value item from "other"    // fill the default value with the constant value item from "other"
101    const DataArrayView& value=other.getPointDataView();    const DataArrayView& value=other.getPointDataView();
102    int len = value.noValues();    int len = value.noValues();
# Line 101  DataTagged::DataTagged(const DataConstan Line 110  DataTagged::DataTagged(const DataConstan
110    setPointDataView(temp);    setPointDataView(temp);
111  }  }
112    
 DataTagged::DataTagged(const DataTagged& other,  
                const DataArrayView::RegionType& region)  
   : DataAbstract(other.getFunctionSpace())  
 {  
   // get the shape of the slice to copy from other  
   DataArrayView::ShapeType shape(DataArrayView::getResultSliceShape(region));  
   DataArrayView::RegionLoopRangeType region_loop_range=getSliceRegionLoopRange(region);  
   
   // allocate enough space for all values  
   int len = DataArrayView::noValues(shape)*(other.m_offsetLookup.size()+1);  
   m_data.resize(len,0.,len);  
   
   // create the data view  
   DataArrayView temp(m_data,shape);  
   setPointDataView(temp);  
   
   // copy the default value  
   getDefaultValue().copySlice(other.getDefaultValue(),region_loop_range);  
   
   // loop through the tag values copying these  
   DataMapType::const_iterator pos;  
   DataArrayView::ValueType::size_type tagOffset=getPointDataView().noValues();  
   for (pos=other.m_offsetLookup.begin();pos!=other.m_offsetLookup.end();pos++){  
     getPointDataView().copySlice(tagOffset,other.getPointDataView(),pos->second,region_loop_range);  
     m_offsetLookup.insert(DataMapType::value_type(pos->first,tagOffset));  
     tagOffset+=getPointDataView().noValues();  
   }  
 }  
   
113  void  void
114  DataTagged::reshapeDataPoint(const DataArrayView::ShapeType& shape)  DataTagged::reshapeDataPoint(const DataArrayView::ShapeType& shape)
115  {  {
# Line 163  DataTagged::reshapeDataPoint(const DataA Line 143  DataTagged::reshapeDataPoint(const DataA
143  DataAbstract*  DataAbstract*
144  DataTagged::getSlice(const DataArrayView::RegionType& region) const  DataTagged::getSlice(const DataArrayView::RegionType& region) const
145  {  {
146    return new DataTagged(*this,region);    return new DataTagged(*this, region);
147    }
148    
149    DataTagged::DataTagged(const DataTagged& other,
150                   const DataArrayView::RegionType& region)
151      : DataAbstract(other.getFunctionSpace())
152    {
153      // slice constructor
154    
155      // get the shape of the slice to copy from other
156      DataArrayView::ShapeType regionShape(DataArrayView::getResultSliceShape(region));
157      DataArrayView::RegionLoopRangeType regionLoopRange=getSliceRegionLoopRange(region);
158    
159      // allocate enough space in this for all values
160      // (need to add one to allow for the default value)
161      int len = DataArrayView::noValues(regionShape)*(other.m_offsetLookup.size()+1);
162      m_data.resize(len,0.0,len);
163    
164      // create the data view
165      DataArrayView temp(m_data,regionShape);
166      setPointDataView(temp);
167    
168      // copy the default value from other to this
169      getDefaultValue().copySlice(other.getDefaultValue(), regionLoopRange);
170    
171      // loop through the tag values copying these
172      DataMapType::const_iterator pos;
173      DataArrayView::ValueType::size_type tagOffset=getPointDataView().noValues();
174      for (pos=other.m_offsetLookup.begin();pos!=other.m_offsetLookup.end();pos++){
175        getPointDataView().copySlice(tagOffset,other.getPointDataView(),pos->second,regionLoopRange);
176        m_offsetLookup.insert(DataMapType::value_type(pos->first,tagOffset));
177        tagOffset+=getPointDataView().noValues();
178      }
179  }  }
180    
181  void  void
182  DataTagged::setSlice(const DataAbstract* value,  DataTagged::setSlice(const DataAbstract* other,
183                       const DataArrayView::RegionType& region)                       const DataArrayView::RegionType& region)
184  {  {
185    const DataTagged* tempDataTag=dynamic_cast<const DataTagged*>(value);  
186    if (tempDataTag==0) {    // other must be another DataTagged object
187      // Data:setSlice implementation should ensure this
188      const DataTagged* otherTemp=dynamic_cast<const DataTagged*>(other);
189      if (otherTemp==0) {
190      throw DataException("Programming error - casting to DataTagged.");      throw DataException("Programming error - casting to DataTagged.");
191    }    }
192    
193    DataArrayView::ShapeType shape(DataArrayView::getResultSliceShape(region));    // determine shape of the specified region
194    DataArrayView::RegionLoopRangeType region_loop_range=getSliceRegionLoopRange(region);    DataArrayView::ShapeType regionShape(DataArrayView::getResultSliceShape(region));
195    
196      // modify region specification as needed to match rank of this object
197      DataArrayView::RegionLoopRangeType regionLoopRange=getSliceRegionLoopRange(region);
198    
199      // ensure rank/shape of this object is compatible with specified region
200    if (getPointDataView().getRank()!=region.size()) {    if (getPointDataView().getRank()!=region.size()) {
201      throw DataException("Error - Invalid slice region.");      throw DataException("Error - Invalid slice region.");
202    }    }
203    if (tempDataTag->getPointDataView().getRank()>0 && !value->getPointDataView().checkShape(shape)) {    if (otherTemp->getPointDataView().getRank()>0 and !other->getPointDataView().checkShape(regionShape)) {
204      throw DataException (value->getPointDataView().createShapeErrorMessage(      throw DataException (other->getPointDataView().createShapeErrorMessage(
205                  "Error - Couldn't copy slice due to shape mismatch.",shape));                           "Error - Couldn't copy slice due to shape mismatch.",regionShape));
206    }    }
207    
208    getDefaultValue().copySliceFrom(tempDataTag->getDefaultValue(),region_loop_range);    // copy slice from other default value to this default value
209      getDefaultValue().copySliceFrom(otherTemp->getDefaultValue(), regionLoopRange);
210    
211    // loop through the tag values    // loop through tag values in other, adding any which aren't in this, using default value
212    DataMapType::const_iterator pos;    DataMapType::const_iterator pos;
213      for (pos=otherTemp->m_offsetLookup.begin();pos!=otherTemp->m_offsetLookup.end();pos++) {
214        if (!isCurrentTag(pos->first)) {
215          addTaggedValue(pos->first,getDefaultValue());
216        }
217      }
218    
219      // loop through the tag values copying slices from other to this
220    for (pos=m_offsetLookup.begin();pos!=m_offsetLookup.end();pos++) {    for (pos=m_offsetLookup.begin();pos!=m_offsetLookup.end();pos++) {
221      getDataPointByTag(pos->first).copySliceFrom(tempDataTag->getDataPointByTag(pos->first),region_loop_range);      getDataPointByTag(pos->first).copySliceFrom(otherTemp->getDataPointByTag(pos->first), regionLoopRange);
222    }    }
223    
224  }  }
225    
226  int  int
# Line 207  DataTagged::getTagNumber(int dpno) Line 236  DataTagged::getTagNumber(int dpno)
236      throw DataException("DataTagged::getTagNumber error: no data-points associated with this object.");      throw DataException("DataTagged::getTagNumber error: no data-points associated with this object.");
237    }    }
238    
239    if (dpno<0 || dpno>numDataPoints) {    if (dpno<0 || dpno>numDataPoints-1) {
240      throw DataException("DataTagged::getTagNumber error: invalid data-point number supplied.");      throw DataException("DataTagged::getTagNumber error: invalid data-point number supplied.");
241    }    }
242    
# Line 225  DataTagged::getTagNumber(int dpno) Line 254  DataTagged::getTagNumber(int dpno)
254  }  }
255    
256  void  void
257    DataTagged::setTaggedValues(const TagListType& tagKeys,
258                                const ValueListType& values)
259    {
260      addTaggedValues(tagKeys,values);
261    }
262    
263    void
264  DataTagged::setTaggedValue(int tagKey,  DataTagged::setTaggedValue(int tagKey,
265                             const DataArrayView& value)                             const DataArrayView& value)
266  {  {
# Line 246  DataTagged::setTaggedValue(int tagKey, Line 282  DataTagged::setTaggedValue(int tagKey,
282  }  }
283    
284  void  void
285    DataTagged::addTaggedValues(const TagListType& tagKeys,
286                                const ValueListType& values)
287    {
288      if (values.size()==0) {
289        // copy the current default value for each of the tags
290        TagListType::const_iterator iT;
291        for (iT=tagKeys.begin();iT!=tagKeys.end();iT++) {
292          // the point data view for DataTagged points at the default value
293          addTaggedValue(*iT,getPointDataView());
294        }
295      } else if (values.size()==1 && tagKeys.size()>1) {
296        // assume the one given value will be used for all tag values
297        TagListType::const_iterator iT;
298        for (iT=tagKeys.begin();iT!=tagKeys.end();iT++) {
299          addTaggedValue(*iT,values[0]);
300        }
301      } else {
302        if (tagKeys.size()!=values.size()) {
303          stringstream temp;
304          temp << "Error - (addTaggedValue) Number of tags: " << tagKeys.size()
305           << " doesn't match number of values: " << values.size();
306          throw DataException(temp.str());
307        } else {
308          for (int i=0;i<tagKeys.size();i++) {
309            addTaggedValue(tagKeys[i],values[i]);
310          }
311        }
312      }
313    }
314    
315    void
316  DataTagged::addTaggedValue(int tagKey,  DataTagged::addTaggedValue(int tagKey,
317                             const DataArrayView& value)                             const DataArrayView& value)
318  {  {
# Line 276  DataTagged::addTaggedValue(int tagKey, Line 343  DataTagged::addTaggedValue(int tagKey,
343    }    }
344  }  }
345    
 void  
 DataTagged::setTaggedValues(const TagListType& tagKeys,  
                             const ValueListType& values)  
 {  
   addTaggedValues(tagKeys,values);  
 }  
   
 void  
 DataTagged::addTaggedValues(const TagListType& tagKeys,  
                             const ValueListType& values)  
 {  
   if (values.size()==0) {  
     // copy the default value for each of the tags  
     TagListType::const_iterator iT;  
     for (iT=tagKeys.begin();iT!=tagKeys.end();iT++) {  
       // the point data view for DataTagged points at the default value  
       addTaggedValue(*iT,getPointDataView());  
     }  
   } else if (values.size()==1 && tagKeys.size()>1) {  
     // assume the one value will be used for all tag values  
     TagListType::const_iterator iT;  
     for (iT=tagKeys.begin();iT!=tagKeys.end();iT++) {  
       addTaggedValue(*iT,values[0]);  
     }  
   } else {  
     if (tagKeys.size()!=values.size()) {  
       stringstream temp;  
       temp << "Error - (addTaggedValue) Number of tags: " << tagKeys.size()  
        << " doesn't match the number of values: " << values.size();  
       throw DataException(temp.str());  
     } else {  
       for (int i=0;i<tagKeys.size();i++) {  
         addTaggedValue(tagKeys[i],values[i]);  
       }  
     }  
   }  
 }  
   
346  double*  double*
347  DataTagged::getSampleDataByTag(int tag)  DataTagged::getSampleDataByTag(int tag)
348  {  {
# Line 344  DataTagged::toString() const Line 373  DataTagged::toString() const
373    return temp.str();    return temp.str();
374  }  }
375    
376  DataArrayView  DataArrayView::ValueType::size_type
377  DataTagged::getDataPointByTag(int tag) const  DataTagged::getPointOffset(int sampleNo,
378                               int dataPointNo) const
379  {  {
380    DataMapType::const_iterator pos(m_offsetLookup.find(tag));    int tagKey=getFunctionSpace().getTagFromSampleNo(sampleNo);
381      DataMapType::const_iterator pos(m_offsetLookup.find(tagKey));
382    DataArrayView::ValueType::size_type offset=m_defaultValueOffset;    DataArrayView::ValueType::size_type offset=m_defaultValueOffset;
383    if (pos!=m_offsetLookup.end()) {    if (pos!=m_offsetLookup.end()) {
384      offset=pos->second;      offset=pos->second;
385    }    }
386    DataArrayView temp(getPointDataView());    return offset;
   temp.setOffset(offset);  
   return temp;  
387  }  }
388    
389  DataArrayView::ValueType::size_type  DataArrayView
390  DataTagged::getPointOffset(int sampleNo,  DataTagged::getDataPointByTag(int tag) const
                            int dataPointNo) const  
391  {  {
392    int tagKey=getFunctionSpace().getTagFromSampleNo(sampleNo);    DataMapType::const_iterator pos(m_offsetLookup.find(tag));
   DataMapType::const_iterator pos(m_offsetLookup.find(tagKey));  
393    DataArrayView::ValueType::size_type offset=m_defaultValueOffset;    DataArrayView::ValueType::size_type offset=m_defaultValueOffset;
394    if (pos!=m_offsetLookup.end()) {    if (pos!=m_offsetLookup.end()) {
395      offset=pos->second;      offset=pos->second;
396    }    }
397    return offset;    DataArrayView temp(getPointDataView());
398      temp.setOffset(offset);
399      return temp;
400  }  }
401    
402  DataArrayView  DataArrayView
# Line 379  DataTagged::getDataPoint(int sampleNo, Line 408  DataTagged::getDataPoint(int sampleNo,
408    return getDataPointByTag(tagKey);    return getDataPointByTag(tagKey);
409  }  }
410    
 const DataTagged::DataMapType&  
 DataTagged::getTagLookup() const  
 {  
   return m_offsetLookup;  
 }  
   
 DataArrayView::ValueType::size_type  
 DataTagged::getLength() const  
 {  
   return m_data.size();  
 }  
   
411  int  int
412  DataTagged::archiveData(ofstream& archiveFile,  DataTagged::archiveData(ofstream& archiveFile,
413                          const DataArrayView::ValueType::size_type noValues) const                          const DataArrayView::ValueType::size_type noValues) const
# Line 404  DataTagged::extractData(ifstream& archiv Line 421  DataTagged::extractData(ifstream& archiv
421  {  {
422    return(m_data.extractData(archiveFile, noValues));    return(m_data.extractData(archiveFile, noValues));
423  }  }
424    void
425    DataTagged::eigenvalues(DataAbstract* ev)
426    {
427      DataTagged* temp_ev=dynamic_cast<DataTagged*>(ev);
428      if (temp_ev==0) {
429        throw DataException("Error - DataTagged::eigenvalues casting to DataTagged failed (propably a programming error).");
430      }
431      const DataTagged::DataMapType& thisLookup=getTagLookup();
432      DataTagged::DataMapType::const_iterator i;
433      DataTagged::DataMapType::const_iterator thisLookupEnd=thisLookup.end();
434      for (i=thisLookup.begin();i!=thisLookupEnd;i++) {
435          temp_ev->addTaggedValue(i->first,temp_ev->getDefaultValue());
436          DataArrayView thisView=getDataPointByTag(i->first);
437          DataArrayView evView=temp_ev->getDataPointByTag(i->first);
438          DataArrayView::eigenvalues(thisView,0,evView,0);
439      }
440      DataArrayView::eigenvalues(getDefaultValue(),0,temp_ev->getDefaultValue(),0);
441    }
442    void
443    DataTagged::eigenvalues_and_eigenvectors(DataAbstract* ev,DataAbstract* V,const double tol)
444    {
445      DataTagged* temp_ev=dynamic_cast<DataTagged*>(ev);
446      if (temp_ev==0) {
447        throw DataException("Error - DataTagged::eigenvalues_and_eigenvectors casting to DataTagged failed (propably a programming error).");
448      }
449      DataTagged* temp_V=dynamic_cast<DataTagged*>(V);
450      if (temp_V==0) {
451        throw DataException("Error - DataTagged::eigenvalues_and_eigenvectors casting to DataTagged failed (propably a programming error).");
452      }
453      const DataTagged::DataMapType& thisLookup=getTagLookup();
454      DataTagged::DataMapType::const_iterator i;
455      DataTagged::DataMapType::const_iterator thisLookupEnd=thisLookup.end();
456      for (i=thisLookup.begin();i!=thisLookupEnd;i++) {
457          temp_ev->addTaggedValue(i->first,temp_ev->getDefaultValue());
458          temp_V->addTaggedValue(i->first,temp_V->getDefaultValue());
459          DataArrayView thisView=getDataPointByTag(i->first);
460          DataArrayView evView=temp_ev->getDataPointByTag(i->first);
461          DataArrayView VView=temp_V->getDataPointByTag(i->first);
462          DataArrayView::eigenvalues_and_eigenvectors(thisView,0,evView,0,VView,0,tol);
463      }
464      DataArrayView::eigenvalues_and_eigenvectors(getDefaultValue(),0,
465                                                  temp_ev->getDefaultValue(),0,
466                                                  temp_V->getDefaultValue(),0,
467                                                  tol);
468    
469    
470    }
471    
472    
473  }  // end of namespace  }  // end of namespace

Legend:
Removed from v.155  
changed lines
  Added in v.594

  ViewVC Help
Powered by ViewVC 1.1.26