/[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 122 by jgs, Thu Jun 9 05:38:05 2005 UTC trunk/escript/src/DataConstant.cpp revision 1118 by gross, Tue Apr 24 08:55:04 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 72  DataConstant::DataConstant(const DataCon Line 73  DataConstant::DataConstant(const DataCon
73    DataArrayView::ShapeType shape(DataArrayView::getResultSliceShape(region));    DataArrayView::ShapeType shape(DataArrayView::getResultSliceShape(region));
74    //    //
75    // allocate space for this new DataConstant's data    // allocate space for this new DataConstant's data
76    m_data.resize(DataArrayView::noValues(shape));    int len = DataArrayView::noValues(shape);
77      m_data.resize(len,0.,len);
78    //    //
79    // create a view of the data with the correct shape    // create a view of the data with the correct shape
80    DataArrayView tempView(m_data,shape);    DataArrayView tempView(m_data,shape);
# Line 153  DataConstant::setSlice(const DataAbstrac Line 155  DataConstant::setSlice(const DataAbstrac
155    if (getPointDataView().getRank()!=region.size()) {    if (getPointDataView().getRank()!=region.size()) {
156      throw DataException("Error - Invalid slice region.");      throw DataException("Error - Invalid slice region.");
157    }    }
158    if (tempDataConst->getPointDataView().getRank()>0 and !value->getPointDataView().checkShape(shape)) {    if (tempDataConst->getPointDataView().getRank()>0 && !value->getPointDataView().checkShape(shape)) {
159      throw DataException (value->getPointDataView().createShapeErrorMessage(      throw DataException (value->getPointDataView().createShapeErrorMessage(
160                  "Error - Couldn't copy slice due to shape mismatch.",shape));                  "Error - Couldn't copy slice due to shape mismatch.",shape));
161    }    }
# Line 161  DataConstant::setSlice(const DataAbstrac Line 163  DataConstant::setSlice(const DataAbstrac
163    getPointDataView().copySliceFrom(tempDataConst->getPointDataView(),region_loop_range);    getPointDataView().copySliceFrom(tempDataConst->getPointDataView(),region_loop_range);
164  }  }
165    
166    int
167    DataConstant::archiveData(ofstream& archiveFile,
168                              const DataArrayView::ValueType::size_type noValues) const
169    {
170      return(m_data.archiveData(archiveFile, noValues));
171    }
172    
173    int
174    DataConstant::extractData(ifstream& archiveFile,
175                              const DataArrayView::ValueType::size_type noValues)
176    {
177      return(m_data.extractData(archiveFile, noValues));
178    }
179    
180    void
181    DataConstant::symmetric(DataAbstract* ev)
182    {
183      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
184      if (temp_ev==0) {
185        throw DataException("Error - DataConstant::symmetric: casting to DataConstant failed (propably a programming error).");
186      }
187      DataArrayView& thisView=getPointDataView();
188      DataArrayView& evView=ev->getPointDataView();
189      DataArrayView::symmetric(thisView,0,evView,0);
190    }
191    
192    void
193    DataConstant::nonsymmetric(DataAbstract* ev)
194    {
195      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
196      if (temp_ev==0) {
197        throw DataException("Error - DataConstant::nonsymmetric: casting to DataConstant failed (propably a programming error).");
198      }
199      DataArrayView& thisView=getPointDataView();
200      DataArrayView& evView=ev->getPointDataView();
201      DataArrayView::nonsymmetric(thisView,0,evView,0);
202    }
203    
204    void
205    DataConstant::trace(DataAbstract* ev, int axis_offset)
206    {
207      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
208      if (temp_ev==0) {
209        throw DataException("Error - DataConstant::trace: casting to DataConstant failed (propably a programming error).");
210      }
211      DataArrayView& thisView=getPointDataView();
212      DataArrayView& evView=ev->getPointDataView();
213      DataArrayView::trace(thisView,0,evView,0,axis_offset);
214    }
215    
216    void
217    DataConstant::swapaxes(DataAbstract* ev, int axis0, int axis1)
218    {
219      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
220      if (temp_ev==0) {
221        throw DataException("Error - DataConstant::swapaxes: casting to DataConstant failed (propably a programming error).");
222      }
223      DataArrayView& thisView=getPointDataView();
224      DataArrayView& evView=ev->getPointDataView();
225      DataArrayView::swapaxes(thisView,0,evView,0,axis0,axis1);
226    }
227    
228    void
229    DataConstant::transpose(DataAbstract* ev, int axis_offset)
230    {
231      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
232      if (temp_ev==0) {
233        throw DataException("Error - DataConstant::transpose: casting to DataConstant failed (propably a programming error).");
234      }
235      DataArrayView& thisView=getPointDataView();
236      DataArrayView& evView=ev->getPointDataView();
237      DataArrayView::transpose(thisView,0,evView,0,axis_offset);
238    }
239    
240    void
241    DataConstant::eigenvalues(DataAbstract* ev)
242    {
243      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
244      if (temp_ev==0) {
245        throw DataException("Error - DataConstant::eigenvalues: casting to DataConstant failed (propably a programming error).");
246      }
247      DataArrayView& thisView=getPointDataView();
248      DataArrayView& evView=ev->getPointDataView();
249      DataArrayView::eigenvalues(thisView,0,evView,0);
250    }
251    void
252    DataConstant::eigenvalues_and_eigenvectors(DataAbstract* ev,DataAbstract* V,const double tol)
253    {
254      DataConstant* temp_ev=dynamic_cast<DataConstant*>(ev);
255      if (temp_ev==0) {
256        throw DataException("Error - DataConstant::eigenvalues_and_eigenvectors: casting to DataConstant failed (propably a programming error).");
257      }
258      DataConstant* temp_V=dynamic_cast<DataConstant*>(V);
259      if (temp_V==0) {
260        throw DataException("Error - DataConstant::eigenvalues_and_eigenvectors: casting to DataConstant failed (propably a programming error).");
261      }
262      DataArrayView thisView=getPointDataView();
263      DataArrayView evView=ev->getPointDataView();
264      DataArrayView VView=V->getPointDataView();
265    
266      DataArrayView::eigenvalues_and_eigenvectors(thisView,0,evView,0,VView,tol);
267    }
268    
269    void
270    DataConstant::setToZero()
271    {
272        DataArrayView::ValueType::size_type n=m_data.size();
273        for (int i=0; i<n ;++i) m_data[i]=0.;
274    }
275    
276  void  void
277  DataConstant::reshapeDataPoint(const DataArrayView::ShapeType& shape)  DataConstant::dump(const std::string fileName) const
278  {  {
279    if (getPointDataView().getRank()!=0) {     #ifdef PASO_MPI
280      stringstream temp;     throw DataException("Error - DataConstant:: dump is not implemented for MPI yet.")
281      temp << "Error - Can only reshape Data with data points of rank 0. "     #endif
282           << "This Data has data points with rank: " << getPointDataView().getRank();     #ifdef USE_NETCDF
283      throw DataException(temp.str());     const NcDim* ncdims[DataArrayView::maxRank];
284    }     NcVar* var;
285    m_data.resize(DataArrayView::noValues(shape),getPointDataView()());     int rank = getPointDataView().getRank();
286    DataArrayView newView(m_data,shape);     int type=  getFunctionSpace().getTypeCode();
287    setPointDataView(newView);     int ndims =0;
288       long dims[DataArrayView::maxRank];
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","constant") )
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       if (! ( var = dataFile.add_var("data", ncDouble, ndims, ncdims)) )
332        throw DataException("Error - DataConstant:: appending variable to netCDF file failed.");
333       if (! (var->put(&m_data[0],dims)) )
334        throw DataException("Error - DataConstant:: copy data to netCDF buffer failed.");
335       #else
336       throw DataException("Error - DataConstant:: dump is not configured with netCDF. Please contact your installation manager.");
337       #endif
338  }  }
339    
340  }  // end of namespace  }  // end of namespace

Legend:
Removed from v.122  
changed lines
  Added in v.1118

  ViewVC Help
Powered by ViewVC 1.1.26