#include<map>
#include "GpuRgbMemory.hpp"
#include "../FFNvDecoder/cuda_kernels.h"

#include <mutex> 

class GpuRgbMemoryFactory{
public:
    GpuRgbMemoryFactory();
    // 获取内存
    GpuRgbMemory* getMemory(int size, string id, string gpuid) ;

    // 显存置为未使用状态
    void releaseMemory(string id);

private:
    map<string, GpuRgbMemory*> memoryMap;

    std::mutex mtx; 
};