rpg/fts/analyzer/TrajectoryWriter.h

#ifndef RPG_TRAJECTORY_WRITER_H
#define RPG_TRAJECTORY_WRITER_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 "Analyzer.h"
#include <util/global.h>
#include <rpg/System.h>
 
namespace Pscf {
namespace Rpg
{
 
   template <int D> class System;
   template <int D> class Simulator;
 
   using namespace Util;
   using namespace Pscf::Prdc;
   using namespace Pscf::Prdc::Cuda;
   
   template <int D>
   class TrajectoryWriter : public Analyzer<D>
   {
   
   public:
   
      TrajectoryWriter(Simulator<D>& simulator, System<D>& system);
   
      virtual ~TrajectoryWriter()
      {} 
   
      virtual void readParameters(std::istream& in);
      
      virtual void save(Serializable::OArchive& ar);
 
      #if 0 
      virtual void loadParameters(Serializable::IArchive& ar);
 
      template <class Archive>
      void serialize(Archive& ar, const unsigned int version);
      #endif
      
      virtual void setup();
  
      virtual void sample(long iStep);
  
      virtual void output();
      
      using ParamComposite::read;
      using ParamComposite::setClassName;
      using Analyzer<D>::outputFileName;
      using Analyzer<D>::isAtInterval;
  
   protected:
      
      // Output file stream
      std::ofstream outputFile_;
      
      // Output filename
      std::string filename_;
 
      long nSample_;
   
      long isInitialized_;
      
      Simulator<D>* simulatorPtr_;
      
      System<D>* systemPtr_;  
 
   protected:
 
      void writeHeader(std::ofstream& out); 
 
      void writeFrame(std::ofstream& out, long iStep);
 
      System<D>& system();
      
      Simulator<D>& simulator();
      
 
   };
 
   #if 0
   /*
   * Serialize to/from an archive. 
   */
   template <class Archive>
   void TrajectoryWriter<D>::serialize(Archive& ar, const unsigned int version)
   {
      Analyzer::serialize(ar, version);
      ar & nSample_;
   }
   #endif
   
   
   // Get the parent system.
   template <int D>
   inline System<D>& TrajectoryWriter<D>::system()
   {  return *systemPtr_; }
   
   //Get parent Simulator object.
   template <int D>
   inline Simulator<D>& TrajectoryWriter<D>::simulator()
   {  return *simulatorPtr_; }
 
   
 
}
}
#endif