datacollection.hpp

Go to the documentation of this file.

// Copyright (c) 2010, Lawrence Livermore National Security, LLC. Produced at
// the Lawrence Livermore National Laboratory. LLNL-CODE-443211. All Rights
// reserved. See file COPYRIGHT for details.
//
// This file is part of the MFEM library. For more information and source code
// availability see http://mfem.org.
//
// MFEM is free software; you can redistribute it and/or modify it under the
// terms of the GNU Lesser General Public License (as published by the Free
// Software Foundation) version 2.1 dated February 1999.

#ifndef MFEM_DATACOLLECTION
#define MFEM_DATACOLLECTION

#include "../config/config.hpp"
#include "gridfunc.hpp"
#include <string>
#include <map>

namespace mfem
{

class DataCollection
{
protected:
   std::string name;

   std::string prefix_path;

   std::map<std::string, GridFunction*> field_map;
   Mesh *mesh;

   int cycle;
   double time;

   bool serial;
   int myid;
   int num_procs;

   int precision;
   int pad_digits;

   static const int precision_default = 6;
   static const int pad_digits_default = 6;

   bool own_data;

   int error;

   DataCollection(const char *collection_name);
   void DeleteData();
   void DeleteAll();

   void SaveOneField(const std::map<std::string,GridFunction*>::iterator &it);

public:
   DataCollection(const char *collection_name, Mesh *_mesh);

   virtual void RegisterField(const char *field_name, GridFunction *gf);
   GridFunction *GetField(const char *field_name);
   bool HasField(const char *name) { return field_map.count(name) == 1; }

   Mesh *GetMesh() { return mesh; }
   virtual void SetMesh(Mesh *new_mesh);

   void SetCycle(int c) { cycle = c; }
   void SetTime(double t) { time = t; }

   int GetCycle() { return cycle; }
   double GetTime() { return time; }
   const char* GetCollectionName() { return name.c_str(); }
   void SetOwnData(bool o) { own_data = o; }

   void SetPrecision(int prec) { precision = prec; }
   void SetPadDigits(int digits) { pad_digits = digits; }
   void SetPrefixPath(const char *prefix);

   virtual void Save();
   virtual void SaveMesh();
   virtual void SaveField(const char *field_name);

   virtual ~DataCollection();

   enum { NO_ERROR = 0, READ_ERROR = 1, WRITE_ERROR = 2 };

   int Error() const { return error; }
   void ResetError(int err = NO_ERROR) { error = err; }
};


class VisItFieldInfo
{
public:
   std::string association;
   int num_components;
   VisItFieldInfo() { association = ""; num_components = 0; }
   VisItFieldInfo(std::string _association, int _num_components)
   { association = _association; num_components = _num_components; }
};

class VisItDataCollection : public DataCollection
{
protected:
   // Additional data needed in the VisIt root file, which describes the mesh
   // and all the fields in the collection
   int spatial_dim, topo_dim;
   int visit_max_levels_of_detail;
   std::map<std::string, VisItFieldInfo> field_info_map;

   std::string GetVisItRootString();
   void ParseVisItRootString(std::string json);

   // Helper functions for Load()
   void LoadVisItRootFile(std::string root_name);
   void LoadMesh();
   void LoadFields();

public:
   VisItDataCollection(const char *collection_name);
   VisItDataCollection(const char *collection_name, Mesh *_mesh);

   virtual void SetMesh(Mesh *new_mesh);

   virtual void RegisterField(const char *field_name, GridFunction *gf);

   void SetMaxLevelsOfDetail(int max_levels_of_detail);

   void DeleteAll();

   virtual void Save();

   void SaveRootFile();

   void Load(int _cycle = 0);

   virtual ~VisItDataCollection() {}
};

}

#endif