cxGPUImageBuffer.h

Go to the documentation of this file.

/*=========================================================================
This file is part of CustusX, an Image Guided Therapy Application.
                 
Copyright (c) SINTEF Department of Medical Technology.
All rights reserved.
                 
CustusX is released under a BSD 3-Clause license.
                 
See Lisence.txt (https://github.com/SINTEFMedtek/CustusX/blob/master/License.txt) for details.
=========================================================================*/


#ifndef CXGPUIMAGEBUFFER_H_
#define CXGPUIMAGEBUFFER_H_

#include "cxResourceExport.h"
#include "cxPrecompiledHeader.h"

#include <boost/shared_ptr.hpp>
#include <boost/weak_ptr.hpp>
#include <map>
#include <list>
#include "vtkForwardDeclarations.h"

namespace cx
{

class cxResource_EXPORT GPUImageDataBuffer
{
public:
        virtual ~GPUImageDataBuffer() {}
        virtual void allocate(int textureUnitIndex) = 0;
//      /** Assuming imageData is changed,
//        * send vtkImageData to GPU.
//        */
//      virtual void updateTexture() = 0;
        virtual void bind(int textureUnitIndex) = 0;
        virtual int getMemorySize() = 0;

        virtual unsigned int getTextureUid() const = 0;
        //virtual vtkImageDataPtr getVtkImageData() = 0;
};

typedef boost::shared_ptr<GPUImageDataBuffer> GPUImageDataBufferPtr;
typedef boost::weak_ptr<GPUImageDataBuffer> GPUImageDataBufferWeakPtr;

cxResource_EXPORT GPUImageDataBufferPtr createGPUImageDataBuffer(vtkImageDataPtr volume);

class cxResource_EXPORT GPUImageLutBuffer
{
public:
        virtual ~GPUImageLutBuffer() {}
        virtual void allocate() = 0;
//      /** If data is changed, send data to GPU.
//        */
//      virtual void updateTexture() = 0;
        virtual void bind(int textureUnitIndex) = 0;
        virtual int getLutSize() const = 0;
};

typedef boost::shared_ptr<GPUImageLutBuffer> GPUImageLutBufferPtr;
typedef boost::weak_ptr<GPUImageLutBuffer> GPUImageLutBufferWeakPtr;

cxResource_EXPORT GPUImageDataBufferPtr createGPUImageDataBuffer(vtkImageDataPtr volume);
cxResource_EXPORT GPUImageLutBufferPtr createGPUImageLutBuffer(vtkUnsignedCharArrayPtr lut);

class cxResource_EXPORT GPUImageBufferRepository
{
public:
        static GPUImageBufferRepository* getInstance();
        static void shutdown();

        GPUImageDataBufferPtr getGPUImageDataBuffer(vtkImageDataPtr volume);
        GPUImageLutBufferPtr getGPUImageLutBuffer(vtkUnsignedCharArrayPtr lut);
        int getMemoryUsage(int *textures);
        void clear();
private:
        GPUImageBufferRepository();
        ~GPUImageBufferRepository();
        class GPUImageBufferRepositoryInternal* mInternal;
        static GPUImageBufferRepository* mInstance;
        void tearDown();
};

// Not allowed to export static functions from dll on Windows,
// so we have to use this funciton instead of GPUImageBufferRepository::getInstance
cxResource_EXPORT GPUImageBufferRepository* getGPUImageBufferRepository();

}

#endif /*CXGPUIMAGEBUFFER_H_*/