rpg/fts/trajectory/RGridTrajectoryReader.h

#ifndef RPG_RGRID_TRAJECTORY_READER_H
#define RPG_RGRID_TRAJECTORY_READER_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 "TrajectoryReader.h"
#include <util/global.h>
#include <iostream>
#include <rpg/System.h>
#include <prdc/cuda/RField.h>
#include <util/containers/DArray.h>
 
namespace Pscf {
namespace Rpg
{
 
   template <int D> class System;
   using namespace Util;
 
   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();
      
   protected:
      void allocate();
      
      System<D>& system();
      
      // Pointer to the parent system.
      System<D>* systemPtr_; 
      
      
   private:
 
      //
      IntVec<D> meshDimensions_;
      
      // Trajectory file.
      std::ifstream inputfile_;
      
      // Read Grid Field configuration      
      DArray< RField<D> > wField_;
      
      // Has the variable been allocated?
      bool isAllocated_;
        
 
   }; 
   
   // Get the parent system.
   template <int D>
   inline System<D>& RGridTrajectoryReader<D>::system()
   {  return *systemPtr_; }
 
   #ifndef RPG_RGRID_TRAJECTORY_READER_TPP
   // Suppress implicit instantiation
   extern template class RGridTrajectoryReader<1>;
   extern template class RGridTrajectoryReader<2>;
   extern template class RGridTrajectoryReader<3>;
   #endif
   
}
}
#endif