/[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 117 by jgs, Fri Apr 1 05:48:57 2005 UTC trunk/escript/src/DataExpanded.cpp revision 856 by gross, Tue Sep 26 01:00:36 2006 UTC
# Line 1  Line 1 
1  // $Id$  // $Id$
2  /*  /*
3   ******************************************************************************   ************************************************************
4   *                                                                            *   *          Copyright 2006 by ACcESS MNRF                   *
5   *       COPYRIGHT  ACcESS 2004 -  All Rights Reserved                        *   *                                                          *
6   *                                                                            *   *              http://www.access.edu.au                    *
7   * This software is the property of ACcESS. No part of this code              *   *       Primary Business: Queensland, Australia            *
8   * may be copied in any form or by any means without the expressed written    *   *  Licensed under the Open Software License version 3.0    *
9   * consent of ACcESS.  Copying, use or modification of this software          *   *     http://www.opensource.org/licenses/osl-3.0.php       *
10   * by any unauthorised person is illegal unless that person has a software    *   *                                                          *
11   * license agreement with ACcESS.                                             *   ************************************************************
  *                                                                            *  
  ******************************************************************************  
12  */  */
13    
14  #include "escript/Data/DataException.h"  #include "DataExpanded.h"
15  #include "escript/Data/DataExpanded.h"  #include "DataException.h"
16  #include "escript/Data/DataConstant.h"  #include "DataConstant.h"
17  #include "escript/Data/DataTagged.h"  #include "DataTagged.h"
 #include "escript/Data/DataArrayView.h"  
18    
19  #include <boost/python/extract.hpp>  #include <boost/python/extract.hpp>
20    
 #include <iostream>  
   
21  using namespace std;  using namespace std;
22  using namespace boost::python;  using namespace boost::python;
23  using namespace boost;  using namespace boost;
# Line 72  DataExpanded::DataExpanded(const DataTag Line 67  DataExpanded::DataExpanded(const DataTag
67    : DataAbstract(other.getFunctionSpace())    : DataAbstract(other.getFunctionSpace())
68  {  {
69    //    //
70    // initialise the data for this object    // initialise the data array for this object
71    initialise(other.getPointDataView().getShape(),other.getNumSamples(),other.getNumDPPSample());    initialise(other.getPointDataView().getShape(),other.getNumSamples(),other.getNumDPPSample());
72    //    //
73    // for each data point in this object, extract and copy the corresponding data    // for each data point in this object, extract and copy the corresponding data
# Line 80  DataExpanded::DataExpanded(const DataTag Line 75  DataExpanded::DataExpanded(const DataTag
75    int i,j;    int i,j;
76    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();
77    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();
78  #pragma omp parallel for private(i,j) schedule(static)    #pragma omp parallel for private(i,j) schedule(static)
79    for (i=0;i<numRows;i++) {    for (i=0;i<numRows;i++) {
80      for (j=0;j<numCols;j++) {      for (j=0;j<numCols;j++) {
81        try {        try {
82          getPointDataView().copy(getPointOffset(i,j),other.getPointDataView(),other.getPointOffset(i,j));          getPointDataView().copy(getPointOffset(i,j),
83                                    other.getPointDataView(),
84                                    other.getPointOffset(i,j));
85        }        }
86        catch (std::exception& e) {        catch (std::exception& e) {
87          cout << e.what() << endl;          cout << e.what() << endl;
# Line 109  DataExpanded::DataExpanded(const DataExp Line 106  DataExpanded::DataExpanded(const DataExp
106    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();
107    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();
108    int i,j;    int i,j;
109  #pragma omp parallel for private(i,j) schedule(static)    #pragma omp parallel for private(i,j) schedule(static)
110    for (i=0;i<numRows;i++) {    for (i=0;i<numRows;i++) {
111      for (j=0;j<numCols;j++) {      for (j=0;j<numCols;j++) {
112        try {        try {
113          getPointDataView().copySlice(getPointOffset(i,j),other.getPointDataView(),other.getPointOffset(i,j),region_loop_range);          getPointDataView().copySlice(getPointOffset(i,j),
114                                         other.getPointDataView(),
115                                         other.getPointOffset(i,j),
116                                         region_loop_range);
117        }        }
118        catch (std::exception& e) {        catch (std::exception& e) {
119          cout << e.what() << endl;          cout << e.what() << endl;
# Line 137  DataExpanded::DataExpanded(const DataArr Line 137  DataExpanded::DataExpanded(const DataArr
137    copy(value);    copy(value);
138  }  }
139    
140  DataExpanded::~DataExpanded()  DataExpanded::DataExpanded(const FunctionSpace& what,
141                               const DataArrayView::ShapeType &shape,
142                               const DataArrayView::ValueType &data)
143      : DataAbstract(what)
144  {  {
145      //
146      // create the view of the data
147      initialise(shape,what.getNumSamples(),what.getNumDPPSample());
148      //
149      // copy the data in the correct format
150      m_data.getData()=data;
151  }  }
152    
153  void  DataExpanded::~DataExpanded()
 DataExpanded::reshapeDataPoint(const DataArrayView::ShapeType& shape)  
154  {  {
   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());  
   }  
   //  
   // create the new DataBlocks2D data array, and a corresponding DataArrayView  
   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++) {  
       // 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(i,j));  
     }  
   }  
   // swap the new data array for the original  
   m_data.Swap(newData);  
   // set the corresponding DataArrayView  
   DataArrayView temp(m_data.getData(),shape);  
   setPointDataView(temp);  
155  }  }
156    
157  DataAbstract*  DataAbstract*
# Line 200  DataExpanded::setSlice(const DataAbstrac Line 177  DataExpanded::setSlice(const DataAbstrac
177    if (getPointDataView().getRank()!=region.size()) {    if (getPointDataView().getRank()!=region.size()) {
178      throw DataException("Error - Invalid slice region.");      throw DataException("Error - Invalid slice region.");
179    }    }
180    if (tempDataExp->getPointDataView().getRank()>0 and !value->getPointDataView().checkShape(shape)) {    if (tempDataExp->getPointDataView().getRank()>0 && !value->getPointDataView().checkShape(shape)) {
181      throw DataException (value->getPointDataView().createShapeErrorMessage(      throw DataException (value->getPointDataView().createShapeErrorMessage(
182          "Error - Couldn't copy slice due to shape mismatch.",shape));          "Error - Couldn't copy slice due to shape mismatch.",shape));
183    }    }
# Line 209  DataExpanded::setSlice(const DataAbstrac Line 186  DataExpanded::setSlice(const DataAbstrac
186    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();    DataArrayView::ValueType::size_type numRows=m_data.getNumRows();
187    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();    DataArrayView::ValueType::size_type numCols=m_data.getNumCols();
188    int i, j;    int i, j;
189  #pragma omp parallel for private(i,j) schedule(static)    #pragma omp parallel for private(i,j) schedule(static)
190    for (i=0;i<numRows;i++) {    for (i=0;i<numRows;i++) {
191      for (j=0;j<numCols;j++) {      for (j=0;j<numCols;j++) {
192        getPointDataView().copySliceFrom(getPointOffset(i,j),tempDataExp->getPointDataView(),tempDataExp->getPointOffset(i,j),region_loop_range);        getPointDataView().copySliceFrom(getPointOffset(i,j),
193                                           tempDataExp->getPointDataView(),
194                                           tempDataExp->getPointOffset(i,j),
195                                           region_loop_range);
196      }      }
197    }    }
198  }  }
# Line 225  DataExpanded::copy(const DataArrayView& Line 205  DataExpanded::copy(const DataArrayView&
205    int nRows=m_data.getNumRows();    int nRows=m_data.getNumRows();
206    int nCols=m_data.getNumCols();    int nCols=m_data.getNumCols();
207    int i,j;    int i,j;
208  #pragma omp parallel for private(i,j) schedule(static)    #pragma omp parallel for private(i,j) schedule(static)
209    for (i=0;i<nRows;i++) {    for (i=0;i<nRows;i++) {
210      for (j=0;j<nCols;j++) {      for (j=0;j<nCols;j++) {
211        // NOTE: An exception may be thown from this call if        // NOTE: An exception may be thown from this call if
# Line 273  string Line 253  string
253  DataExpanded::toString() const  DataExpanded::toString() const
254  {  {
255    stringstream temp;    stringstream temp;
256      FunctionSpace fs=getFunctionSpace();
257    //    //
258    // create a temporary view as the offset will be changed    // create a temporary view as the offset will be changed
259    DataArrayView tempView(getPointDataView().getData(),getPointDataView().getShape(),getPointDataView().getOffset());    DataArrayView tempView(getPointDataView().getData(),getPointDataView().getShape(),getPointDataView().getOffset());
# Line 280  DataExpanded::toString() const Line 261  DataExpanded::toString() const
261      for (int j=0;j<m_data.getNumCols();j++) {      for (int j=0;j<m_data.getNumCols();j++) {
262        tempView.setOffset(m_data.index(i,j));        tempView.setOffset(m_data.index(i,j));
263        stringstream suffix;        stringstream suffix;
264        suffix << "(" << i << "," << j << ")";        suffix << "( id: " << i << ", ref: " << fs.getReferenceNoFromSampleNo(i) << ", pnt: " << j << ")";
265        temp << tempView.toString(suffix.str());        temp << tempView.toString(suffix.str());
266        if (!(i==(m_data.getNumRows()-1) && j==(m_data.getNumCols()-1))) {        if (!(i==(m_data.getNumRows()-1) && j==(m_data.getNumCols()-1))) {
267          temp << endl;          temp << endl;
# Line 337  DataExpanded::setRefValue(int ref, Line 318  DataExpanded::setRefValue(int ref,
318    
319    for (int n=0; n<numDPPSample; n++) {    for (int n=0; n<numDPPSample; n++) {
320      //      //
321      // Get each data-point in the sample in turn.      // Get *each* data-point in the sample in turn.
322      DataArrayView pointView = getDataPoint(sampleNo, n);      DataArrayView pointView = getDataPoint(sampleNo, n);
323      //      //
324      // Assign the values in the DataArray to this data-point.      // Assign the values in the DataArray to this data-point.
# Line 370  DataExpanded::getRefValue(int ref, Line 351  DataExpanded::getRefValue(int ref,
351    }    }
352    
353    //    //
354    // Get the first data-point associated with this sample number.    // Get the *first* data-point associated with this sample number.
355    DataArrayView pointView = getDataPoint(sampleNo, 0);    DataArrayView pointView = getDataPoint(sampleNo, 0);
356    
357    //    //
# Line 378  DataExpanded::getRefValue(int ref, Line 359  DataExpanded::getRefValue(int ref,
359    value.getView().copy(pointView);    value.getView().copy(pointView);
360  }  }
361    
362    int
363    DataExpanded::archiveData(ofstream& archiveFile,
364                              const DataArrayView::ValueType::size_type noValues) const
365    {
366      return(m_data.archiveData(archiveFile, noValues));
367    }
368    
369    int
370    DataExpanded::extractData(ifstream& archiveFile,
371                              const DataArrayView::ValueType::size_type noValues)
372    {
373      return(m_data.extractData(archiveFile, noValues));
374    }
375    
376    void
377    DataExpanded::copyAll(const boost::python::numeric::array& value) {
378      //
379      // Get the number of samples and data-points per sample.
380      int numSamples = getNumSamples();
381      int numDataPointsPerSample = getNumDPPSample();
382      int dataPointRank = getPointDataView().getRank();
383      ShapeType dataPointShape = getPointDataView().getShape();
384      //
385      // check rank:
386      if (value.getrank()!=dataPointRank+1)
387           throw DataException("Rank of numarray does not match Data object rank");
388      if (value.getshape()[0]!=numSamples*numDataPointsPerSample)
389           throw DataException("leading dimension of numarray is too small");
390      //
391      int dataPoint = 0;
392      for (int sampleNo = 0; sampleNo < numSamples; sampleNo++) {
393        for (int dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
394          DataArrayView dataPointView = getDataPoint(sampleNo, dataPointNo);
395          if (dataPointRank==0) {
396             dataPointView()=extract<double>(value[dataPoint]);
397          } else if (dataPointRank==1) {
398             for (int i=0; i<dataPointShape[0]; i++) {
399                dataPointView(i)=extract<double>(value[dataPoint][i]);
400             }
401          } else if (dataPointRank==2) {
402             for (int i=0; i<dataPointShape[0]; i++) {
403               for (int j=0; j<dataPointShape[1]; j++) {
404                 dataPointView(i,j)=extract<double>(value[dataPoint][i][j]);
405               }
406             }
407           } else if (dataPointRank==3) {
408             for (int i=0; i<dataPointShape[0]; i++) {
409               for (int j=0; j<dataPointShape[1]; j++) {
410                 for (int k=0; k<dataPointShape[2]; k++) {
411                     dataPointView(i,j,k)=extract<double>(value[dataPoint][i][j][k]);
412                 }
413               }
414             }
415           } else if (dataPointRank==4) {
416             for (int i=0; i<dataPointShape[0]; i++) {
417               for (int j=0; j<dataPointShape[1]; j++) {
418                 for (int k=0; k<dataPointShape[2]; k++) {
419                   for (int l=0; l<dataPointShape[3]; l++) {
420                     dataPointView(i,j,k,l)=extract<double>(value[dataPoint][i][j][k][l]);
421                   }
422                 }
423               }
424             }
425          }
426          dataPoint++;
427        }
428      }
429    }
430    void
431    DataExpanded::symmetric(DataAbstract* ev)
432    {
433      int sampleNo,dataPointNo;
434      int numSamples = getNumSamples();
435      int numDataPointsPerSample = getNumDPPSample();
436      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
437      if (temp_ev==0) {
438        throw DataException("Error - DataExpanded::symmetric: casting to DataExpanded failed (propably a programming error).");
439      }
440      DataArrayView& thisView=getPointDataView();
441      DataArrayView& evView=ev->getPointDataView();
442      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
443      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
444        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
445             DataArrayView::symmetric(thisView,getPointOffset(sampleNo,dataPointNo),
446                                        evView,ev->getPointOffset(sampleNo,dataPointNo));
447        }
448      }
449    }
450    void
451    DataExpanded::nonsymmetric(DataAbstract* ev)
452    {
453      int sampleNo,dataPointNo;
454      int numSamples = getNumSamples();
455      int numDataPointsPerSample = getNumDPPSample();
456      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
457      if (temp_ev==0) {
458        throw DataException("Error - DataExpanded::nonsymmetric: casting to DataExpanded failed (propably a programming error).");
459      }
460      DataArrayView& thisView=getPointDataView();
461      DataArrayView& evView=ev->getPointDataView();
462      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
463      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
464        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
465             DataArrayView::nonsymmetric(thisView,getPointOffset(sampleNo,dataPointNo),
466                                        evView,ev->getPointOffset(sampleNo,dataPointNo));
467        }
468      }
469    }
470    void
471    DataExpanded::trace(DataAbstract* ev, int axis_offset)
472    {
473      int sampleNo,dataPointNo;
474      int numSamples = getNumSamples();
475      int numDataPointsPerSample = getNumDPPSample();
476      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
477      if (temp_ev==0) {
478        throw DataException("Error - DataExpanded::trace: casting to DataExpanded failed (propably a programming error).");
479      }
480      DataArrayView& thisView=getPointDataView();
481      DataArrayView& evView=ev->getPointDataView();
482      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
483      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
484        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
485             DataArrayView::trace(thisView,getPointOffset(sampleNo,dataPointNo),
486                                        evView,ev->getPointOffset(sampleNo,dataPointNo),axis_offset);
487        }
488      }
489    }
490    
491    void
492    DataExpanded::transpose(DataAbstract* ev, int axis_offset)
493    {
494      int sampleNo,dataPointNo;
495      int numSamples = getNumSamples();
496      int numDataPointsPerSample = getNumDPPSample();
497      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
498      if (temp_ev==0) {
499        throw DataException("Error - DataExpanded::transpose: casting to DataExpanded failed (propably a programming error).");
500      }
501      DataArrayView& thisView=getPointDataView();
502      DataArrayView& evView=ev->getPointDataView();
503      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
504      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
505        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
506             DataArrayView::transpose(thisView,getPointOffset(sampleNo,dataPointNo),
507                                        evView,ev->getPointOffset(sampleNo,dataPointNo),axis_offset);
508        }
509      }
510    }
511    
512    void
513    DataExpanded::swapaxes(DataAbstract* ev, int axis0, int axis1)
514    {
515      int sampleNo,dataPointNo;
516      int numSamples = getNumSamples();
517      int numDataPointsPerSample = getNumDPPSample();
518      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
519      if (temp_ev==0) {
520        throw DataException("Error - DataExpanded::swapaxes: casting to DataExpanded failed (propably a programming error).");
521      }
522      DataArrayView& thisView=getPointDataView();
523      DataArrayView& evView=ev->getPointDataView();
524      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
525      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
526        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
527             DataArrayView::swapaxes(thisView,getPointOffset(sampleNo,dataPointNo),
528                                        evView,ev->getPointOffset(sampleNo,dataPointNo),axis0,axis1);
529        }
530      }
531    }
532    void
533    DataExpanded::eigenvalues(DataAbstract* ev)
534    {
535      int sampleNo,dataPointNo;
536      int numSamples = getNumSamples();
537      int numDataPointsPerSample = getNumDPPSample();
538      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
539      if (temp_ev==0) {
540        throw DataException("Error - DataExpanded::eigenvalues: casting to DataExpanded failed (propably a programming error).");
541      }
542      DataArrayView& thisView=getPointDataView();
543      DataArrayView& evView=ev->getPointDataView();
544      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
545      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
546        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
547             DataArrayView::eigenvalues(thisView,getPointOffset(sampleNo,dataPointNo),
548                                        evView,ev->getPointOffset(sampleNo,dataPointNo));
549        }
550      }
551    }
552    void
553    DataExpanded::eigenvalues_and_eigenvectors(DataAbstract* ev,DataAbstract* V,const double tol)
554    {
555      int numSamples = getNumSamples();
556      int numDataPointsPerSample = getNumDPPSample();
557      int sampleNo,dataPointNo;
558      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
559      if (temp_ev==0) {
560        throw DataException("Error - DataExpanded::eigenvalues_and_eigenvectors: casting to DataExpanded failed (propably a programming error).");
561      }
562      DataExpanded* temp_V=dynamic_cast<DataExpanded*>(V);
563      if (temp_V==0) {
564        throw DataException("Error - DataExpanded::eigenvalues_and_eigenvectors: casting to DataExpanded failed (propably a programming error).");
565      }
566      DataArrayView& thisView=getPointDataView();
567      DataArrayView& evView=ev->getPointDataView();
568      DataArrayView& VView=V->getPointDataView();
569      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
570      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
571        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
572             DataArrayView::eigenvalues_and_eigenvectors(thisView,getPointOffset(sampleNo,dataPointNo),
573                                                         evView,ev->getPointOffset(sampleNo,dataPointNo),
574                                                         VView,V->getPointOffset(sampleNo,dataPointNo),
575                                                         tol);
576        }
577      }
578    }
579    
580  }  // end of namespace  }  // end of namespace

Legend:
Removed from v.117  
changed lines
  Added in v.856

  ViewVC Help
Powered by ViewVC 1.1.26