Hu Chunming / FFNvDecoder

#include<string>

#include "../interface/DeviceRgbMemory.hpp"
#include "cuda_kernels.h"
#include "define.hpp"
#include "common_header.h"

using namespace std;

class GpuRgbMemory : public DeviceRgbMemory{

public:
     GpuRgbMemory(int _channel, int _width, int _height, string _id, string _gpuid, bool _key_frame, bool _isused)
     :DeviceRgbMemory(_channel, _width, _height, _id, _gpuid, _key_frame, _isused){
        gpuid = _gpuid;
        cudaSetDevice(atoi(gpuid.c_str()));
        CHECK_CUDA(cudaMalloc((void **)&pHwRgb, data_size * sizeof(unsigned char)));
    }

    ~GpuRgbMemory(){
        if (pHwRgb) {
            cudaSetDevice(atoi(gpuid.c_str()));
            CHECK_CUDA(cudaFree(pHwRgb));
            pHwRgb = nullptr;
        }
    }

    string getGpuId() {
        return gpuid;
    }

private:
    string gpuid;
};