/[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 854 by gross, Thu Sep 21 05:29:42 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 337  DataExpanded::setRefValue(int ref, Line 317  DataExpanded::setRefValue(int ref,
317    
318    for (int n=0; n<numDPPSample; n++) {    for (int n=0; n<numDPPSample; n++) {
319      //      //
320      // Get each data-point in the sample in turn.      // Get *each* data-point in the sample in turn.
321      DataArrayView pointView = getDataPoint(sampleNo, n);      DataArrayView pointView = getDataPoint(sampleNo, n);
322      //      //
323      // 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 350  DataExpanded::getRefValue(int ref,
350    }    }
351    
352    //    //
353    // Get the first data-point associated with this sample number.    // Get the *first* data-point associated with this sample number.
354    DataArrayView pointView = getDataPoint(sampleNo, 0);    DataArrayView pointView = getDataPoint(sampleNo, 0);
355    
356    //    //
# Line 378  DataExpanded::getRefValue(int ref, Line 358  DataExpanded::getRefValue(int ref,
358    value.getView().copy(pointView);    value.getView().copy(pointView);
359  }  }
360    
361    int
362    DataExpanded::archiveData(ofstream& archiveFile,
363                              const DataArrayView::ValueType::size_type noValues) const
364    {
365      return(m_data.archiveData(archiveFile, noValues));
366    }
367    
368    int
369    DataExpanded::extractData(ifstream& archiveFile,
370                              const DataArrayView::ValueType::size_type noValues)
371    {
372      return(m_data.extractData(archiveFile, noValues));
373    }
374    
375    void
376    DataExpanded::copyAll(const boost::python::numeric::array& value) {
377      //
378      // Get the number of samples and data-points per sample.
379      int numSamples = getNumSamples();
380      int numDataPointsPerSample = getNumDPPSample();
381      int dataPointRank = getPointDataView().getRank();
382      ShapeType dataPointShape = getPointDataView().getShape();
383      //
384      // check rank:
385      if (value.getrank()!=dataPointRank+1)
386           throw DataException("Rank of numarray does not match Data object rank");
387      if (value.getshape()[0]!=numSamples*numDataPointsPerSample)
388           throw DataException("leading dimension of numarray is too small");
389      //
390      int dataPoint = 0;
391      for (int sampleNo = 0; sampleNo < numSamples; sampleNo++) {
392        for (int dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
393          DataArrayView dataPointView = getDataPoint(sampleNo, dataPointNo);
394          if (dataPointRank==0) {
395             dataPointView()=extract<double>(value[dataPoint]);
396          } else if (dataPointRank==1) {
397             for (int i=0; i<dataPointShape[0]; i++) {
398                dataPointView(i)=extract<double>(value[dataPoint][i]);
399             }
400          } else if (dataPointRank==2) {
401             for (int i=0; i<dataPointShape[0]; i++) {
402               for (int j=0; j<dataPointShape[1]; j++) {
403                 dataPointView(i,j)=extract<double>(value[dataPoint][i][j]);
404               }
405             }
406           } else if (dataPointRank==3) {
407             for (int i=0; i<dataPointShape[0]; i++) {
408               for (int j=0; j<dataPointShape[1]; j++) {
409                 for (int k=0; k<dataPointShape[2]; k++) {
410                     dataPointView(i,j,k)=extract<double>(value[dataPoint][i][j][k]);
411                 }
412               }
413             }
414           } else if (dataPointRank==4) {
415             for (int i=0; i<dataPointShape[0]; i++) {
416               for (int j=0; j<dataPointShape[1]; j++) {
417                 for (int k=0; k<dataPointShape[2]; k++) {
418                   for (int l=0; l<dataPointShape[3]; l++) {
419                     dataPointView(i,j,k,l)=extract<double>(value[dataPoint][i][j][k][l]);
420                   }
421                 }
422               }
423             }
424          }
425          dataPoint++;
426        }
427      }
428    }
429    void
430    DataExpanded::symmetric(DataAbstract* ev)
431    {
432      int sampleNo,dataPointNo;
433      int numSamples = getNumSamples();
434      int numDataPointsPerSample = getNumDPPSample();
435      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
436      if (temp_ev==0) {
437        throw DataException("Error - DataExpanded::symmetric: casting to DataExpanded failed (propably a programming error).");
438      }
439      DataArrayView& thisView=getPointDataView();
440      DataArrayView& evView=ev->getPointDataView();
441      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
442      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
443        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
444             DataArrayView::symmetric(thisView,getPointOffset(sampleNo,dataPointNo),
445                                        evView,ev->getPointOffset(sampleNo,dataPointNo));
446        }
447      }
448    }
449    void
450    DataExpanded::nonsymmetric(DataAbstract* ev)
451    {
452      int sampleNo,dataPointNo;
453      int numSamples = getNumSamples();
454      int numDataPointsPerSample = getNumDPPSample();
455      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
456      if (temp_ev==0) {
457        throw DataException("Error - DataExpanded::nonsymmetric: casting to DataExpanded failed (propably a programming error).");
458      }
459      DataArrayView& thisView=getPointDataView();
460      DataArrayView& evView=ev->getPointDataView();
461      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
462      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
463        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
464             DataArrayView::nonsymmetric(thisView,getPointOffset(sampleNo,dataPointNo),
465                                        evView,ev->getPointOffset(sampleNo,dataPointNo));
466        }
467      }
468    }
469    void
470    DataExpanded::trace(DataAbstract* ev, int axis_offset)
471    {
472      int sampleNo,dataPointNo;
473      int numSamples = getNumSamples();
474      int numDataPointsPerSample = getNumDPPSample();
475      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
476      if (temp_ev==0) {
477        throw DataException("Error - DataExpanded::trace: casting to DataExpanded failed (propably a programming error).");
478      }
479      DataArrayView& thisView=getPointDataView();
480      DataArrayView& evView=ev->getPointDataView();
481      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
482      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
483        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
484             DataArrayView::trace(thisView,getPointOffset(sampleNo,dataPointNo),
485                                        evView,ev->getPointOffset(sampleNo,dataPointNo),axis_offset);
486        }
487      }
488    }
489    
490    void
491    DataExpanded::transpose(DataAbstract* ev, int axis_offset)
492    {
493      int sampleNo,dataPointNo;
494      int numSamples = getNumSamples();
495      int numDataPointsPerSample = getNumDPPSample();
496      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
497      if (temp_ev==0) {
498        throw DataException("Error - DataExpanded::transpose: casting to DataExpanded failed (propably a programming error).");
499      }
500      DataArrayView& thisView=getPointDataView();
501      DataArrayView& evView=ev->getPointDataView();
502      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
503      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
504        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
505             DataArrayView::transpose(thisView,getPointOffset(sampleNo,dataPointNo),
506                                        evView,ev->getPointOffset(sampleNo,dataPointNo),axis_offset);
507        }
508      }
509    }
510    
511    void
512    DataExpanded::swapaxes(DataAbstract* ev, int axis0, int axis1)
513    {
514      int sampleNo,dataPointNo;
515      int numSamples = getNumSamples();
516      int numDataPointsPerSample = getNumDPPSample();
517      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
518      if (temp_ev==0) {
519        throw DataException("Error - DataExpanded::swapaxes: casting to DataExpanded failed (propably a programming error).");
520      }
521      DataArrayView& thisView=getPointDataView();
522      DataArrayView& evView=ev->getPointDataView();
523      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
524      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
525        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
526             DataArrayView::swapaxes(thisView,getPointOffset(sampleNo,dataPointNo),
527                                        evView,ev->getPointOffset(sampleNo,dataPointNo),axis0,axis1);
528        }
529      }
530    }
531    void
532    DataExpanded::eigenvalues(DataAbstract* ev)
533    {
534      int sampleNo,dataPointNo;
535      int numSamples = getNumSamples();
536      int numDataPointsPerSample = getNumDPPSample();
537      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
538      if (temp_ev==0) {
539        throw DataException("Error - DataExpanded::eigenvalues: casting to DataExpanded failed (propably a programming error).");
540      }
541      DataArrayView& thisView=getPointDataView();
542      DataArrayView& evView=ev->getPointDataView();
543      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
544      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
545        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
546             DataArrayView::eigenvalues(thisView,getPointOffset(sampleNo,dataPointNo),
547                                        evView,ev->getPointOffset(sampleNo,dataPointNo));
548        }
549      }
550    }
551    void
552    DataExpanded::eigenvalues_and_eigenvectors(DataAbstract* ev,DataAbstract* V,const double tol)
553    {
554      int numSamples = getNumSamples();
555      int numDataPointsPerSample = getNumDPPSample();
556      int sampleNo,dataPointNo;
557      DataExpanded* temp_ev=dynamic_cast<DataExpanded*>(ev);
558      if (temp_ev==0) {
559        throw DataException("Error - DataExpanded::eigenvalues_and_eigenvectors: casting to DataExpanded failed (propably a programming error).");
560      }
561      DataExpanded* temp_V=dynamic_cast<DataExpanded*>(V);
562      if (temp_V==0) {
563        throw DataException("Error - DataExpanded::eigenvalues_and_eigenvectors: casting to DataExpanded failed (propably a programming error).");
564      }
565      DataArrayView& thisView=getPointDataView();
566      DataArrayView& evView=ev->getPointDataView();
567      DataArrayView& VView=V->getPointDataView();
568      #pragma omp parallel for private(sampleNo,dataPointNo) schedule(static)
569      for (sampleNo = 0; sampleNo < numSamples; sampleNo++) {
570        for (dataPointNo = 0; dataPointNo < numDataPointsPerSample; dataPointNo++) {
571             DataArrayView::eigenvalues_and_eigenvectors(thisView,getPointOffset(sampleNo,dataPointNo),
572                                                         evView,ev->getPointOffset(sampleNo,dataPointNo),
573                                                         VView,V->getPointOffset(sampleNo,dataPointNo),
574                                                         tol);
575        }
576      }
577    }
578    
579  }  // end of namespace  }  // end of namespace

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

  ViewVC Help
Powered by ViewVC 1.1.26