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

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

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

revision 854 by gross, Thu Sep 21 05:29:42 2006 UTC revision 1149 by gross, Thu May 17 00:30:15 2007 UTC
# Line 17  Line 17 
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 51  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 260  DataConstant::eigenvalues_and_eigenvecto Line 262  DataConstant::eigenvalues_and_eigenvecto
262    DataArrayView evView=ev->getPointDataView();    DataArrayView evView=ev->getPointDataView();
263    DataArrayView VView=V->getPointDataView();    DataArrayView VView=V->getPointDataView();
264    
265    DataArrayView::eigenvalues_and_eigenvectors(thisView,0,evView,0,VView,tol);    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_NETCDF
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.854  
changed lines
  Added in v.1149

  ViewVC Help
Powered by ViewVC 1.1.26