git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@3170 f3b2605a-c512-4ea7-a41b-209d697bcdaa

lib/atc/OutputManager.h

@@ -0,0 +1,126 @@
+#ifndef OUTPUT_MANAGER_H
+#define OUTPUT_MANAGER_H
+
+#include "ATC_TypeDefs.h"
+
+#include <map>
+#include <string>
+
+// NOTE : number of rows to data type:
+// 1 -> scalar
+// 3 -> vector  x,y,z
+// NOT 6 -> tensor  xx,xy,xz,yy,yz,zz
+// 6 -> tensor  xx,yy,zz,xy,zx,yz
+// 9 -> tensor  xx,xy,xz,yx,yy,yz,zx,zy,zz
+
+using namespace std;
+
+namespace ATC {
+
+  enum OutputType     { ENSIGHT=0, GNUPLOT };
+  enum OutputDataType { POINT=0, MESH };
+  enum OutputDataCardinality { SCALAR_OUTPUT=0, VECTOR_OUTPUT, TENSOR_OUTPUT, 
+    SYM_TENSOR_OUTPUT, LIST_OUTPUT };
+  enum OutputOption   { OUTPUT_VECTOR_COMPONENTS=0, OUTPUT_TENSOR_COMPONENTS};
+
+  class OutputManager{
+
+  public:
+    OutputManager(void);
+    OutputManager(string outputPrefix, OutputType otype = ENSIGHT);
+    ~OutputManager(void);
+
+    /** initialize output */
+    void initialize(string outputPrefix, OutputType otype = ENSIGHT);
+
+    /** set output options */
+    void set_option(OutputOption option, bool value); 
+
+    // Dump text-based field info to disk for later restart
+    void write_restart_file(string fileName, OUTPUT_LIST *data);
+
+    // Read text-based field file written from write_restart_file
+    void read_restart_file(string fileName, OUTPUT_LIST *data);
+
+    /** write initial/reference geometry
+        default is to write point data, 
+        if connectivities are given then mesh data will be output 
+        coordinates : num _total_ points/nodes X num spatial dim 
+        connectivities : num elements X num nodes per element*/
+    void write_geometry(const MATRIX &coordinates,
+                        const Array2D<int> *connectivity=NULL);
+    void write_geometry(OUTPUT_LIST &part_coordinates);
+
+    /** write data from a time step
+        specify node_map to handle periodic soln & data */
+    void write_data(double time, OUTPUT_LIST *data, const int *node_map=NULL);
+
+    void write_data(double time, FIELDS *soln, OUTPUT_LIST *data, 
+      const int *node_map=NULL);
+
+/*
+    void write_data (double time, 
+      map<string, map<string, DenseMatrix<double> * > * > parts_data,
+      const int * node_map = NULL);
+*/
+    /** add a scalar to a text output file */
+    void add_global(const string& name, const double& value) {
+      globalData_[name] = value; }
+
+    /** delete a scalar from the output */
+    void delete_global(const string& name) { globalData_.erase(name); }
+
+    /** reset the stored output scalars */
+    void reset_globals() { globalData_.clear(); writeGlobalsHeader_=true; }
+
+    /** return data type: scalar, vector, tensor, list */
+    int data_type(const DENS_MAT & data) const {
+      return data_type(data.nCols());
+    }
+    int data_type(int cols) const {
+      if      (cols == 1) return SCALAR_OUTPUT;
+      else if (cols == 3) return VECTOR_OUTPUT;
+      else if (cols == 6) return SYM_TENSOR_OUTPUT;
+      else if (cols == 9) return TENSOR_OUTPUT;
+      else                return LIST_OUTPUT;
+    }
+    bool use_component_names(int type) {
+      if ( (type==LIST_OUTPUT) ||     
+       ((type==SYM_TENSOR_OUTPUT || type==TENSOR_OUTPUT) && tensorToComponents_)
+      || (type==VECTOR_OUTPUT && vectorToComponents_) ) 
+        return true;
+      else 
+        return false;
+    }
+
+  private:
+
+    void write_data(string name, const MATRIX *data, const int *node_map=NULL);
+    void write_text(OUTPUT_LIST *data);
+    void write_dictionary(double time, OUTPUT_LIST *data);
+    void write_globals();
+
+    //* status booleans
+    bool initialized_, firstStep_, firstGlobalsWrite_, writeGlobalsHeader_;
+
+    /** number of columns of data */
+    int nDataCols_;
+    /** number of nodes */
+    int number_of_nodes_; // NOTE it would be nicer if node_map came with a size
+    /** data type */
+    int dataType_;
+    /** base name for output files */
+    string outputPrefix_;
+    /** list of output timesteps */
+    vector<double> outputTimes_;
+    /** output type */
+    int outputType_;
+    /** output tensor as its components */
+    bool tensorToComponents_;
+    /** output vector as its components */
+    bool vectorToComponents_;
+    /** global variables */
+    map<string,double> globalData_;
+  };
+}
+#endif