fts/trajectory/RGridTrajectoryReader.h

#ifndef RP_RGRID_TRAJECTORY_READER_H
#define RP_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 <pscf/math/IntVec.h>           // member
#include <util/containers/DArray.h>     // member
#include <fstream>                      // member
#include <string>                       // function parameter
 
namespace Pscf {
namespace Rp {
 
   using namespace Util;

   template <int D, class T>
   class RGridTrajectoryReader : public T::TrajectoryReader
   {
 
   public:
 
      // Protected constructor and destructor (see below).

      void open(std::string filename) override;

      void readHeader() override;

      bool readFrame() override;

      void close() override;
 
   protected:

      RGridTrajectoryReader(typename T::System& system);

      ~RGridTrajectoryReader() = default;
 
      using TrajectoryReaderT = typename T::TrajectoryReader;
      using TrajectoryReaderT::system;
 
   private:
 
      // Field configuration
      DArray< typename T::RField > wField_;
 
      // Dimensions of computational mesh (# of points in each direction)
      IntVec<D> meshDimensions_;
 
      // Trajectory file.
      std::ifstream inputfile_;
 
      // Has wField_ been allocated?
      bool isAllocated_;

      void allocate();
 
   };
 
}
}
#endif