gpuRenderer.hpp

Go to the documentation of this file.

#ifndef OPENPOSE_CORE_GPU_RENDERER_HPP
#define OPENPOSE_CORE_GPU_RENDERER_HPP

#include <atomic>
#include <tuple>
#include <openpose/core/common.hpp>
#include <openpose/core/renderer.hpp>

namespace op
{
    class OP_API GpuRenderer : public Renderer
    {
    public:
        explicit GpuRenderer(const float renderThreshold, const float alphaKeypoint,
                             const float alphaHeatMap, const bool blendOriginalFrame = true,
                             const unsigned int elementToRender = 0u, const unsigned int numberElementsToRender = 0u);

        ~GpuRenderer();

        std::tuple<std::shared_ptr<float*>, std::shared_ptr<bool>, std::shared_ptr<std::atomic<unsigned int>>,
                   std::shared_ptr<std::atomic<unsigned long long>>, std::shared_ptr<const unsigned int>>
                   getSharedParameters();

        void setSharedParametersAndIfLast(const std::tuple<std::shared_ptr<float*>, std::shared_ptr<bool>,
                                                           std::shared_ptr<std::atomic<unsigned int>>,
                                                           std::shared_ptr<std::atomic<unsigned long long>>,
                                                           std::shared_ptr<const unsigned int>>& tuple,
                                          const bool isLast);

    protected:
        std::shared_ptr<float*> spGpuMemory;

        void cpuToGpuMemoryIfNotCopiedYet(const float* const cpuMemory, const unsigned long long memoryVolume);

        void gpuToCpuMemoryIfLastRenderer(float* cpuMemory, const unsigned long long memoryVolume);

    private:
        std::shared_ptr<std::atomic<unsigned long long>> spVolume;
        bool mIsFirstRenderer;
        bool mIsLastRenderer;
        std::shared_ptr<bool> spGpuMemoryAllocated;

        DELETE_COPY(GpuRenderer);
    };
}

#endif // OPENPOSE_CORE_GPU_RENDERER_HPP