cpu/RField.h

#ifndef PRDC_CPU_R_FIELD_H
#define PRDC_CPU_R_FIELD_H
 
/*
* PSCF Package 
*
* Copyright 2016 - 2022, The Regents of the University of Minnesota
* Distributed under the terms of the GNU General Public License.
*/
 
#include <prdc/cpu/FftwDArray.h>
#include <pscf/math/IntVec.h>
#include <util/global.h>
 
namespace Pscf {
namespace Prdc {
namespace Cpu {
 
   using namespace Util;
   using namespace Pscf;
 
   template <int D>
   class RField : public FftwDArray<double>
   {
 
   public:
 
      typedef double ElementType;
 
      RField();
 
      RField(const RField& other);
 
      virtual ~RField();
 
      RField& operator = (const RField& other);
 
      void allocate(IntVec<D> const & meshDimensions);
 
      virtual void deallocate();
 
      const IntVec<D>& meshDimensions() const;
 
      template <class Archive>
      void serialize(Archive& ar, const unsigned int version);
 
   private:
 
      // Vector containing number of grid points in each direction.
      IntVec<D> meshDimensions_;
 
      using FftwDArray<double>::allocate;
 
   };
 
   /*
   * Return mesh dimensions by constant reference.
   */
   template <int D>
   inline const IntVec<D>& RField<D>::meshDimensions() const
   {  return meshDimensions_; }
 
   /*
   * Serialize a Field to/from an Archive.
   */
   template <int D>
   template <class Archive>
   void RField<D>::serialize(Archive& ar, const unsigned int version)
   {
      FftwDArray<double>::serialize(ar, version);
      ar & meshDimensions_;
   }
 
   #ifndef PRDC_CPU_R_FIELD_TPP
   extern template class RField<1>;
   extern template class RField<2>;
   extern template class RField<3>;
   #endif
 
}
}
}
#endif