Hu Chunming / src

  #include<string>
  
  #include "../helpers/cuda_helper.h"

  #include "../helpers/timer.hpp"

  
  using namespace std;
  
  class GpuRgbMemory{
  
  public:

       GpuRgbMemory(int _channel, int _width, int _height, string _id, string _gpuid, bool _isused){

          channel = _channel;
          width = _width;
          height = _height;
          size = channel * width * height;
          isused = _isused;
          id = _id;
          gpuid = _gpuid;
          timestamp = helpers::timer::get_timestamp<std::chrono::milliseconds>();
  
          cudaSetDevice(atoi(gpuid.c_str()));
          CHECK(cudaMalloc((void **)&pHwRgb, size * sizeof(unsigned char)));
      }
  
      ~GpuRgbMemory(){
          if (pHwRgb) {
              cudaSetDevice(atoi(gpuid.c_str()));
              CHECK(cudaFree(pHwRgb));
              pHwRgb = nullptr;
          }
      }
      
      int getSize() {
          return size;
      }
      
      bool isIsused() {
          return isused;
      }
  
      void setIsused(bool _isused) {
          isused = _isused;
          // 更新时间戳

          timestamp = helpers::timer::get_timestamp<std::chrono::milliseconds>();

      }
  
      string getId() {
          return id;
      }
  
      string getGpuId() {
          return gpuid;
      }
  
      unsigned char* getMem(){
          return pHwRgb;
      }
  
      long long getTimesstamp(){
          return timestamp;
      }
  
      int getWidth(){
          return width;
      }
  
      int getHeight(){
          return height;
      }
  
      int getChannel(){
          return channel;
      }
  
  private:
      int size;
      bool isused;
      string id;
      string gpuid;
      unsigned char * pHwRgb{nullptr};
      long long timestamp;
      int width{0};
      int height{0};
      int channel{3};
  };