rpc/fts/analyzer/TrajectoryWriter.h

#ifndef RPC_TRAJECTORY_WRITER_H
#define RPC_TRAJECTORY_WRITER_H
 
/*
* PSCF - Polymer Self-Consistent Field
*
* Copyright 2015 - 2025, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
#include "Analyzer.h"
#include <util/global.h>
 
namespace Pscf {
namespace Rpc {
 
   template <int D> class System;
   template <int D> class Simulator;
 
   using namespace Util;

   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 setup();

      virtual void sample(long iStep);

      virtual void output();
 
   protected:
 
      using ParamComposite::read;
      using ParamComposite::setClassName;
      using Analyzer<D>::outputFileName;
      using Analyzer<D>::isAtInterval;

      void writeHeader(std::ofstream& out);

      void writeFrame(std::ofstream& out, long iStep);

      System<D>& system();

      Simulator<D>& simulator();
 
   private:
 
      // Output file stream
      std::ofstream outputFile_;
 
      // Output filename
      std::string filename_;

      long nSample_;

      long isInitialized_;

      Simulator<D>* simulatorPtr_;

      System<D>* systemPtr_;
 
 
   };
 
   // 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_; }
 
   #ifndef RPC_TRAJECTORY_WRITER_TPP
   // Suppress implicit instantiation
   extern template class TrajectoryWriter<1>;
   extern template class TrajectoryWriter<2>;
   extern template class TrajectoryWriter<3>;
   #endif
 
}
}
#endif