/[escript]/branches/lapack2681/escript/src/DataConstant.cpp
ViewVC logotype

Diff of /branches/lapack2681/escript/src/DataConstant.cpp

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

trunk/esys2/escript/src/Data/DataConstant.cpp revision 123 by jgs, Fri Jul 8 04:08:13 2005 UTC trunk/escript/src/DataConstant.cpp revision 1141 by gross, Tue May 15 04:24:51 2007 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/DataConstant.h"  #include "DataConstant.h"
15  #include "escript/Data/DataException.h"  #include "DataException.h"
16  #include "esysUtils/EsysAssert.h"  #include "esysUtils/EsysAssert.h"
17    
18  #include <iostream>  #include <iostream>
19  #include <boost/python/extract.hpp>  #include <boost/python/extract.hpp>
20    #ifdef USE_NETCDF
21    #include "netcdfcpp.h"
22    #endif
23    
24  using namespace std;  using namespace std;
25    
# Line 53  DataConstant::DataConstant(const DataArr Line 54  DataConstant::DataConstant(const DataArr
54    
55  DataConstant::DataConstant(const DataConstant& other)  DataConstant::DataConstant(const DataConstant& other)
56    : DataAbstract(other.getFunctionSpace())    : DataAbstract(other.getFunctionSpace())
57  {  {  //
   //  
58    // copy the data in the correct format    // copy the data in the correct format
59    m_data=other.m_data;    m_data=other.m_data;
60    //    //
# Line 72  DataConstant::DataConstant(const DataCon Line 72  DataConstant::DataConstant(const DataCon
72    DataArrayView::ShapeType shape(DataArrayView::getResultSliceShape(region));    DataArrayView::ShapeType shape(DataArrayView::getResultSliceShape(region));
73    //    //
74    // allocate space for this new DataConstant's data    // allocate space for this new DataConstant's data
75    m_data.resize(DataArrayView::noValues(shape));    int len = DataArrayView::noValues(shape);
76      m_data.resize(len,0.,len);
77    //    //
78    // create a view of the data with the correct shape    // create a view of the data with the correct shape
79    DataArrayView tempView(m_data,shape);    DataArrayView tempView(m_data,shape);
# Line 153  DataConstant::setSlice(const DataAbstrac Line 154  DataConstant::setSlice(const DataAbstrac
154    if (getPointDataView().getRank()!=region.size()) {    if (getPointDataView().getRank()!=region.size()) {
155      throw DataException("Error - Invalid slice region.");      throw DataException("Error - Invalid slice region.");
156    }    }
157    if (tempDataConst->getPointDataView().getRank()>0 and !value->getPointDataView().checkShape(shape)) {    if (tempDataConst->getPointDataView().getRank()>0 && !value->getPointDataView().checkShape(shape)) {
158      throw DataException (value->getPointDataView().createShapeErrorMessage(      throw DataException (value->getPointDataView().createShapeErrorMessage(
159                  "Error - Couldn't copy slice due to shape mismatch.",shape));                  "Error - Couldn't copy slice due to shape mismatch.",shape));
160    }    }
# Line 161  DataConstant::setSlice(const DataAbstrac Line 162  DataConstant::setSlice(const DataAbstrac
162    getPointDataView().copySliceFrom(tempDataConst->getPointDataView(),region_loop_range);    getPointDataView().copySliceFrom(tempDataConst->getPointDataView(),region_loop_range);
163  }  }
164    
 void  
 DataConstant::reshapeDataPoint(const DataArrayView::ShapeType& shape)  
 {  
   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());  
   }  
   m_data.resize(DataArrayView::noValues(shape),getPointDataView()());  
   DataArrayView newView(m_data,shape);  
   setPointDataView(newView);  
 }  
   
165  int  int
166  DataConstant::archiveData(ofstream& archiveFile,  DataConstant::archiveData(ofstream& archiveFile,
167                            const DataArrayView::ValueType::size_type noValues) const                            const DataArrayView::ValueType::size_type noValues) const
# Line 189  DataConstant::extractData(ifstream& arch Line 176  DataConstant::extractData(ifstream& arch
176    return(m_data.extractData(archiveFile, noValues));    return(m_data.extractData(archiveFile, noValues));
177  }  }
178    
179    void
180    DataConstant::symmetric(DataAbstract* ev)
181    {
182      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
183      if (temp_ev==0) {
184        throw DataException("Error - DataConstant::symmetric: casting to DataConstant failed (propably a programming error).");
185      }
186      DataArrayView& thisView=getPointDataView();
187      DataArrayView& evView=ev->getPointDataView();
188      DataArrayView::symmetric(thisView,0,evView,0);
189    }
190    
191    void
192    DataConstant::nonsymmetric(DataAbstract* ev)
193    {
194      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
195      if (temp_ev==0) {
196        throw DataException("Error - DataConstant::nonsymmetric: casting to DataConstant failed (propably a programming error).");
197      }
198      DataArrayView& thisView=getPointDataView();
199      DataArrayView& evView=ev->getPointDataView();
200      DataArrayView::nonsymmetric(thisView,0,evView,0);
201    }
202    
203    void
204    DataConstant::trace(DataAbstract* ev, int axis_offset)
205    {
206      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
207      if (temp_ev==0) {
208        throw DataException("Error - DataConstant::trace: casting to DataConstant failed (propably a programming error).");
209      }
210      DataArrayView& thisView=getPointDataView();
211      DataArrayView& evView=ev->getPointDataView();
212      DataArrayView::trace(thisView,0,evView,0,axis_offset);
213    }
214    
215    void
216    DataConstant::swapaxes(DataAbstract* ev, int axis0, int axis1)
217    {
218      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
219      if (temp_ev==0) {
220        throw DataException("Error - DataConstant::swapaxes: casting to DataConstant failed (propably a programming error).");
221      }
222      DataArrayView& thisView=getPointDataView();
223      DataArrayView& evView=ev->getPointDataView();
224      DataArrayView::swapaxes(thisView,0,evView,0,axis0,axis1);
225    }
226    
227    void
228    DataConstant::transpose(DataAbstract* ev, int axis_offset)
229    {
230      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
231      if (temp_ev==0) {
232        throw DataException("Error - DataConstant::transpose: casting to DataConstant failed (propably a programming error).");
233      }
234      DataArrayView& thisView=getPointDataView();
235      DataArrayView& evView=ev->getPointDataView();
236      DataArrayView::transpose(thisView,0,evView,0,axis_offset);
237    }
238    
239    void
240    DataConstant::eigenvalues(DataAbstract* ev)
241    {
242      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
243      if (temp_ev==0) {
244        throw DataException("Error - DataConstant::eigenvalues: casting to DataConstant failed (propably a programming error).");
245      }
246      DataArrayView& thisView=getPointDataView();
247      DataArrayView& evView=ev->getPointDataView();
248      DataArrayView::eigenvalues(thisView,0,evView,0);
249    }
250    void
251    DataConstant::eigenvalues_and_eigenvectors(DataAbstract* ev,DataAbstract* V,const double tol)
252    {
253      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
254      if (temp_ev==0) {
255        throw DataException("Error - DataConstant::eigenvalues_and_eigenvectors: casting to DataConstant failed (propably a programming error).");
256      }
257      DataConstant* temp_V=dynamic_cast<DataConstant*>(V);
258      if (temp_V==0) {
259        throw DataException("Error - DataConstant::eigenvalues_and_eigenvectors: casting to DataConstant failed (propably a programming error).");
260      }
261      DataArrayView thisView=getPointDataView();
262      DataArrayView evView=ev->getPointDataView();
263      DataArrayView VView=V->getPointDataView();
264    
265      DataArrayView::eigenvalues_and_eigenvectors(thisView,0,evView,0,VView,0,tol);
266    }
267    
268    void
269    DataConstant::setToZero()
270    {
271        DataArrayView::ValueType::size_type n=m_data.size();
272        for (int i=0; i<n ;++i) m_data[i]=0.;
273    }
274    
275    void
276    DataConstant::dump(const std::string fileName) const
277    {
278       #ifdef PASO_MPI
279       throw DataException("Error - DataConstant:: dump is not implemented for MPI yet.")
280       #endif
281       #ifdef USE_NETCD
282       const NcDim* ncdims[DataArrayView::maxRank];
283       NcVar* var;
284       int rank = getPointDataView().getRank();
285       int type=  getFunctionSpace().getTypeCode();
286       int ndims =0;
287       long dims[DataArrayView::maxRank];
288       const double* d_ptr=&(m_data[0]);
289       DataArrayView::ShapeType shape = getPointDataView().getShape();
290      
291       // netCDF error handler
292       NcError err(NcError::verbose_nonfatal);
293       // Create the file.
294       NcFile dataFile(fileName.c_str(), NcFile::Replace);
295       // check if writing was successful
296       if (!dataFile.is_valid())
297        throw DataException("Error - DataConstant:: opening of netCDF file for output failed.");
298       if (!dataFile.add_att("type_id",0) )
299        throw DataException("Error - DataConstant:: appending data type to netCDF file failed.");
300       if (!dataFile.add_att("rank",rank) )
301        throw DataException("Error - DataConstant:: appending rank attribute to netCDF file failed.");
302       if (!dataFile.add_att("function_space_type",type))
303        throw DataException("Error - DataConstant:: appending function space attribute to netCDF file failed.");
304    
305       if (rank == 0) {
306          if( ! (ncdims[0] = dataFile.add_dim("l", 1)) )
307            throw DataException("Error - DataConstant:: appending ncdimsion 0 to netCDF file failed.");
308          dims[0]=1,
309          ndims=1;
310       } else {
311           ndims=rank;
312           dims[0]=shape[0];
313           if (! (ncdims[0] = dataFile.add_dim("d0",shape[0])) )
314            throw DataException("Error - DataConstant:: appending ncdimsion 0 to netCDF file failed.");
315           if ( rank >1 ) {
316               dims[1]=shape[1];
317               if (! (ncdims[1] = dataFile.add_dim("d1",shape[1])) )
318            throw DataException("Error - DataConstant:: appending ncdimsion 1 to netCDF file failed.");
319           }
320           if ( rank >2 ) {
321               dims[2]=shape[2];
322               if (! (ncdims[2] = dataFile.add_dim("d2", shape[2])) )
323            throw DataException("Error - DataConstant:: appending ncdimsion 2 to netCDF file failed.");
324           }
325           if ( rank >3 ) {
326               dims[3]=shape[3];
327               if (! (ncdims[3] = dataFile.add_dim("d3", shape[3])) )
328            throw DataException("Error - DataConstant:: appending ncdimsion 3 to netCDF file failed.");
329           }
330       }
331      
332       if (! ( var = dataFile.add_var("data", ncDouble, ndims, ncdims)) )
333        throw DataException("Error - DataConstant:: appending variable to netCDF file failed.");
334       if (! (var->put(d_ptr,dims)) )
335             throw DataException("Error - DataConstant:: copy data to netCDF buffer failed.");
336       #else
337       throw DataException("Error - DataConstant:: dump is not configured with netCDF. Please contact your installation manager.");
338       #endif
339    }
340    
341  }  // end of namespace  }  // end of namespace

Legend:
Removed from v.123  
changed lines
  Added in v.1141

  ViewVC Help
Powered by ViewVC 1.1.26