rpg/field/WFieldContainer.h

#ifndef RPG_W_FIELD_CONTAINER_H
#define RPG_W_FIELD_CONTAINER_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 <prdc/field/WFieldsReal.h>     // base class template
#include <prdc/cpu/RField.h>            // template parameter
#include <rpg/field/FieldIo.h>          // template parameter
 
namespace Pscf {
namespace Rpg {
 
   using namespace Util;
   using namespace Pscf::Prdc;
   using namespace Pscf::Prdc::Cuda;

   template <int D>
   class WFieldContainer 
     : public WFieldsReal<D, Prdc::Cuda::RField<D>, Rpg::FieldIo<D> >
   {
 
   public:

      typedef WFieldsReal<D, RField<D>, FieldIo<D> > Base;
 
      // Inherited public member functions
      using Base::setFieldIo;
      using Base::setNMonomer;
      using Base::allocateRGrid;
      using Base::allocateBasis;
      using Base::allocate;
      using Base::setBasis;
      using Base::setRGrid;
      using Base::readBasis;
      using Base::readRGrid;
      using Base::symmetrize;
      using Base::clear;
      using Base::writeBasis;
      using Base::writeRGrid;
      using Base::basis;
      using Base::rgrid;
      using Base::isAllocatedRGrid;
      using Base::isAllocatedBasis;
      using Base::hasData;
      using Base::isSymmetric;

      void setRGrid(DeviceArray<cudaReal>& fields);
 
   protected:
 
      using Base::meshDimensions;
      using Base::meshSize;
      using Base::nBasis;
      using Base::nMonomer;
      using Base::fieldIo;
 
   private:

      void assignRField(RField<D>& lhs, RField<D> const & rhs) const 
      override;
 
   };
 
   #ifndef RPG_W_FIELD_CONTAINER_TPP
   // Suppress implicit instantiation
   extern template class WFieldContainer<1>;
   extern template class WFieldContainer<2>;
   extern template class WFieldContainer<3>;
   #endif
 
} // namespace Rpg
 
#ifndef RPG_W_FIELD_CONTAINER_TPP
namespace Prdc {
   // Suppress implicit instantiation
   extern template class WFieldsReal<1, RField<1>, Rpg::FieldIo<1> >;
   extern template class WFieldsReal<2, RField<2>, Rpg::FieldIo<2> >;
   extern template class WFieldsReal<3, RField<3>, Rpg::FieldIo<3> >;
} // namespace Prdc
#endif
 
} // namespace Pscf
#endif