Hu Chunming / FFNvDecoder

#ifndef __DEVICE_RGB_MEMORY_H__
#define __DEVICE_RGB_MEMORY_H__

#include<string>

#include "utiltools.hpp"

using namespace std;

class DeviceRgbMemory{

public:
     DeviceRgbMemory(int _channel, int _width, int _height, string _id, string _dev_id, bool _key_frame, bool _isused){
        channel = _channel;
        width = _width;
        height = _height;
        data_size = channel * width * height;
        isused = _isused;
        id = _id;
        device_id = _dev_id;
        key_frame = _key_frame;
        timestamp = UtilTools::get_cur_time_ms();
    }

    virtual ~DeviceRgbMemory(){}
    
    int getSize() {
        return data_size;
    }
    
    bool isIsused() {
        return isused;
    }

    void setIsused(bool _isused) {
        isused = _isused;
        // 更新时间戳
        timestamp = UtilTools::get_cur_time_ms();
    }

    string getId() {
        return id;
    }

    string getDeviceId() {
        return device_id;
    }

    unsigned char* getMem(){
        return pHwRgb;
    }

    long long getTimesstamp(){
        return timestamp;
    }

    int getWidth(){
        return width;
    }

    int getHeight(){
        return height;
    }

    int getChannel(){
        return channel;
    }

    bool isKeyFrame(){
        return key_frame;
    }

public:
    int data_size;
    bool isused;
    string id;
    string device_id;
    unsigned char * pHwRgb{nullptr};
    long long timestamp;
    int width{0};
    int height{0};
    int channel{3};
    bool key_frame;
};

#endif