r1d/misc/FieldIo.h

#ifndef R1D_FIELD_IO_H
#define R1D_FIELD_IO_H
 
/*
* PSCF - Polymer Self-Consistent Field Theory
*
* Copyright 2016 - 2022, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
#include <util/containers/DArray.h>     // member
 
namespace Util { class FileMaster; }
 
namespace Pscf {
namespace R1d {
 
   class Domain;
   class Mixture;
   using namespace Util;
 
   class FieldIo 
   {
 
   public:
 
      typedef DArray<double> Field;
 
      FieldIo();
 
      ~FieldIo();
 
      void associate(Domain const & domain,
                     FileMaster const & fileMaster);
 
      void readFields(DArray<Field>& fields, std::istream &in);
 
      void readFields(DArray<Field>& fields, std::string const& filename);
      
      void writeField(Field const& field, std::ostream& out, 
                      bool writeHeader = true) const;
 
      void writeField(Field const&  field, std::string const& filename, 
                      bool writeHeader= true) const;
      
      void writeFields(DArray<Field> const& fields, std::ostream& out, 
                       bool writeHeader= true) const;
 
      void writeFields(DArray<Field> const&  fields, 
                       std::string const& filename, 
                       bool writeHeader= true) const;
 
      void writeBlockCFields(Mixture const& mixture, std::ostream& out) const;
 
      void writeBlockCFields(Mixture const& mixture,
                             std::string const& filename) const;
 
      void writeVertexQ(Mixture const& mixture,
                        int polymerId, int vertexId, 
                        std::ostream& out) const;
 
      void writeVertexQ(Mixture const& mixture,
                        int polymerId, int vertexId, 
                        std::string const& filename) const;
 
      void remesh(DArray<Field> const& fields, int nx, std::ostream& out);
 
      void remesh(DArray<Field> const& fields, int nx,
                  std::string const& filename);
 
      void extend(DArray<Field> const& fields, int m, std::ostream& out);
 
      void extend(DArray<Field> const& fields, int m, 
                  std::string const& filename);
 
   private:
 
      mutable DArray<double> w_;
 
      // Pointers to associated objects.
 
      Domain const * domainPtr_;
 
      FileMaster const * fileMasterPtr_;
 
      // Private accessor functions:
 
      Domain const& domain() const
      {  
         UTIL_ASSERT(domainPtr_);  
         return *domainPtr_; 
      }
 
      FileMaster const & fileMaster() const
      {  
         UTIL_ASSERT(fileMasterPtr_);  
         return *fileMasterPtr_; 
      }
 
   };
 
 
} // namespace R1d
} // namespace Pscf
#endif