rpg/fts/brownian/BdStep.h

#ifndef RPG_BD_STEP_H
#define RPG_BD_STEP_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 <util/param/ParamComposite.h>
#include <pscf/cuda/CudaRandom.h>
 
namespace Pscf {
namespace Rpg {
 
   using namespace Util;
 
   template <int D> class System;
   template <int D> class BdSimulator;

   template <int D>
   class BdStep : public ParamComposite
   {
 
   public:

      BdStep(BdSimulator<D>& simulator);

      virtual ~BdStep();

      virtual void readParameters(std::istream &in);

      virtual void setup();

      virtual bool step() = 0;
      
      virtual bool needsCc()
      {  return false; }
      
      virtual bool needsDc()
      { return true; }
      
      virtual void outputTimers(std::ostream& out);
      
      virtual void clearTimers();
 
      // Accessor Functions

      virtual void output();
 
   protected:

      System<D>& system();

      BdSimulator<D>& simulator();

      CudaRandom& cudaRandom();
 
   private:

      BdSimulator<D>* simulatorPtr_;

      System<D>* systemPtr_;

      CudaRandom  *cudaRandomPtr_;
 
   };
 
   // Protected inline methods
 
   /*
   * Get parent System object.
   */
   template <int D>
   inline System<D>& BdStep<D>::system()
   {  return *systemPtr_; }
 
   /*
   * Get parent BdSimulator object.
   */
   template <int D>
   inline BdSimulator<D>& BdStep<D>::simulator()
   {  return *simulatorPtr_; }
 
   /*
   * Get Random number generator.
   */
   template <int D>
   inline CudaRandom& BdStep<D>::cudaRandom()
   {  return *cudaRandomPtr_; }
 
   // Explicit instantiation declarations
   extern template class BdStep<1>;
   extern template class BdStep<2>;
   extern template class BdStep<3>;
 
}
}
#endif