niftkLightweightCUDAImage.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

=============================================================================*/


#ifndef niftkLightweightCUDAImage_h
#define niftkLightweightCUDAImage_h

#include "niftkCUDAExports.h"
#include <cuda_runtime_api.h>
#include <QAtomicInt>

namespace niftk
{

// forward-decl
class CUDAManager;

class NIFTKCUDA_EXPORT LightweightCUDAImage
{
  friend class CUDAManager;

public:
  LightweightCUDAImage();

  ~LightweightCUDAImage();

  LightweightCUDAImage(const LightweightCUDAImage& copyme);
  LightweightCUDAImage& operator=(const LightweightCUDAImage& assignme);


  unsigned int GetId() const;

  unsigned int GetWidth() const;
  unsigned int GetHeight() const;
  unsigned int GetBytePitch() const;

  cudaEvent_t GetReadyEvent() const;


private:
  QAtomicInt*     m_RefCount;

  int             m_Device;    // the device this image lives on.
  unsigned int    m_Id;
  cudaEvent_t     m_ReadyEvent;       // signaled when the image is ready for consumption.

  void*           m_DevicePtr;
  std::size_t     m_SizeInBytes;

  unsigned int    m_Width;          // in pixel
  unsigned int    m_Height;         // in (pixel) lines
  unsigned int    m_BytePitch;      // length of a line of pixels in bytes.
  int             m_PixelType;      // FIXME: not fully spec'd!

  // debugging
  cudaStream_t    m_LastUsedByStream;
};

} // end namespace

#endif