DeviceMemory.h

#ifndef PSCF_DEVICE_MEMORY_H
#define PSCF_DEVICE_MEMORY_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/misc/ReferenceCounter.h>
 
// Forward declaration
namespace Util { class CountedReference; }
 
namespace Pscf {
 
   using namespace Util;

   class DeviceMemory
   {
 
   public:

      DeviceMemory();

      DeviceMemory(int capacity);

      virtual ~DeviceMemory();

      void allocate(int capacity);

      void deallocate();

      void resize(int capacity);

      void addReference(CountedReference& reference);

      void* cArray() const;

      int capacity() const;

      bool isAllocated() const;
 
   private:

      void* dataPtr_;

      int capacity_;

      ReferenceCounter refCounter_;
 
   };
 
   /*
   * Return true if the array has been allocated, false otherwise.
   */
   inline
   bool DeviceMemory::isAllocated() const
   {  return (bool)dataPtr_; }
 
   /*
   * Return allocated capacity.
   */
   inline 
   int DeviceMemory::capacity() const
   {  return capacity_; }
 
} // namespace Pscf
#endif