rpc/fts/trajectory/RGridTrajectoryReader.h

#ifndef RPC_RGRID_TRAJECTORY_READER_H
#define RPC_RGRID_TRAJECTORY_READER_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 "TrajectoryReader.h"              // base class
 
#include <prdc/cpu/RField.h>               // member
#include <pscf/math/IntVec.h>              // member
#include <util/containers/DArray.h>        // member
 
#include <string>
#include <iostream>
 
namespace Pscf {
namespace Rpc {
 
   template <int D> class System;
 
   using namespace Util;
   using namespace Pscf::Prdc::Cpu; 

   template <int D>
   class RGridTrajectoryReader : public TrajectoryReader<D>
   {
 
   public:

      RGridTrajectoryReader<D>(System<D>& system);

      virtual ~RGridTrajectoryReader(){};

      void open(std::string filename);

      bool readFrame();

      void close();
 
      void readHeader();
      
      void readFieldsRGrid(std::istream &in, DArray<RField<D> >& fields);
 
   protected:

      void allocate();
 
      using TrajectoryReader<D>::system;
 
   private:
 
      // Field configuration  
      DArray< RField<D> > wField_;
      
      // Dimensions of computational mesh (# of points in each direction)
      IntVec<D> meshDimensions_;
  
      // Trajectory file input stream
      std::ifstream inputfile_;
      
      // Has wField_ been allocated?
      bool isAllocated_;
 
   }; 
 
   #ifndef RPC_RGRID_TRAJECTORY_READER_TPP
   // Suppress implicit instantiation
   extern template class RGridTrajectoryReader<1>;
   extern template class RGridTrajectoryReader<2>;
   extern template class RGridTrajectoryReader<3>;
   #endif
 
}
}
#endif