MediaInfAiDataVehicleRecognition.hpp

//
// Created by tangfahong on 2024/4/10.
//
 
#ifndef EMX_MediaInfAiDataVehicleRecognition_HPP
#define EMX_MediaInfAiDataVehicleRecognition_HPP
 
#include "MediaInfAiData.hpp"
#include "MediaRegion.hpp"
#include <cstring>
 
namespace Emx {
    class MediaInfAiDataVehicleRecognition : public MediaInfAiData {
    public:
        MediaInfAiDataVehicleRecognition() : MediaInfAiData(), num(0), array(nullptr) {
            memset(&fullPic, 0, sizeof(MediaInfAiPic));
        }
 
        virtual ~MediaInfAiDataVehicleRecognition() { Free(); }
 
    public:
        struct DataMsg {
            MediaInfAiPic pic;     
            MediaInfAiPic platePic;
            uint64_t id;           
            uint64_t plateId;      
            Rect rect;             
            Rect plateRect;        
            /*目标类型*/
            enum class TypeE : uint32_t {
                LightCar,  
                HeavyCar,  
                Sedan,     
                SUV,       
                Van,       
                Truck,     
                Motorcycle,
                SportsCar, 
                Pickup     
            } type;
            uint32_t color;
            uint32_t isRun;
            char brand[16];
            /*车牌颜色*/
            enum class PlateColorE : uint32_t {
                Blue,  
                Yellow,
                White, 
                Black, 
                Green  
            } plateColor;
            uint8_t score;               
            uint8_t attrScore;           
            uint8_t plateScore;          
            uint8_t plateLine;           
            uint8_t plateNumber[2][32];  
            float plateAngleH;           
            float plateAngleV;           
            float plateArea;             
            float plateKeyPointsDistance;
 
            int imgOff;
        };
 
        void Alloc(int n) {
            Free();
            array = (DataMsg *) EP_MALLOC(n * (int) sizeof(DataMsg));
            if (array) { memset((void *) array, 0, n * sizeof(DataMsg)); }
            memset(&fullPic, 0, sizeof(MediaInfAiPic));
        }
 
        void Free() {
            if (array) {
                for (int i = 0; i < num; i++) {
                    if ((array[i].pic.size > 0) && array[i].pic.img) {
                        free(array[i].pic.img);
                        array[i].pic.img = nullptr;
                        array[i].pic.size = 0;
                    }
                    if ((array[i].platePic.size > 0) && array[i].platePic.img) {
                        free(array[i].platePic.img);
                        array[i].platePic.img = nullptr;
                        array[i].platePic.size = 0;
                    }
                }
                if ((fullPic.size > 0) && fullPic.img) {
                    free(fullPic.img);
                    fullPic.img = nullptr;
                }
                free(array);
                array = nullptr;
            }
        }
 
        int GetSerializeSize() override {
            int totalImgSize = 0;
            totalImgSize += sizeof(MediaInfAiPic);
            totalImgSize += fullPic.size;
            for (int i = 0; i < num; i++) {
                totalImgSize += array[i].pic.size;
                totalImgSize += array[i].platePic.size;
            }
            return (int) sizeof(num) + num * (int) sizeof(DataMsg) + totalImgSize;
        }
 
        void Serialize(uint8_t *buffer) override {
            int off = 0;
            memcpy(buffer + off, &num, sizeof(num));
            off += (int) sizeof(num);
            memcpy(buffer + off, &fullPic, sizeof(MediaInfAiPic));
            int fullImgOff = off + (int) sizeof(MediaInfAiPic);
            if (fullPic.img && fullPic.size > 0) { memcpy(buffer + fullImgOff, fullPic.img, fullPic.size); }
            off += (fullPic.size + sizeof(MediaInfAiPic));
 
            int imgOff = off + num * (int) sizeof(DataMsg);
            for (int i = 0; i < num; i++) {
                array[i].imgOff = imgOff;
                if (array[i].pic.img && array[i].pic.size > 0) {
                    memcpy(buffer + imgOff, array[i].pic.img, array[i].pic.size);
                }
                imgOff += array[i].pic.size;
                if (array[i].platePic.img && array[i].platePic.size > 0) {
                    memcpy(buffer + imgOff, array[i].platePic.img, array[i].platePic.size);
                }
                imgOff += array[i].platePic.size;
                memcpy(buffer + off, array + i, (int) sizeof(DataMsg));
                off += (int) sizeof(DataMsg);
            }
        }
 
        void Parse(const uint8_t *buffer, int size) override {
            int off = 0;
            memcpy(&num, buffer + off, sizeof(num));
            off += sizeof(num);
            Alloc(num);
            memcpy(&fullPic, buffer + off, (int) sizeof(MediaInfAiPic));
            off += (int) sizeof(MediaInfAiPic);
            if (fullPic.size > 0) {
                fullPic.img = (uint8_t *) EP_MALLOC(fullPic.size);
                if (fullPic.img) { memcpy(fullPic.img, buffer + off, fullPic.size); }
            }
            off += fullPic.size;
            if (num * (int) sizeof(DataMsg) + off <= size) {
                memcpy(array, buffer + off, num * (int) sizeof(DataMsg));
                for (int i = 0; i < num; i++) {
                    int imgOff = array[i].imgOff;
                    if (array[i].pic.size) {
                        array[i].pic.img = (uint8_t *) EP_MALLOC(array[i].pic.size);
                        if (array[i].pic.img) memcpy(array[i].pic.img, buffer + imgOff, array[i].pic.size);
                        imgOff += array[i].pic.size;
                    }
                    if (array[i].platePic.size) {
                        array[i].platePic.img = (uint8_t *) EP_MALLOC(array[i].platePic.size);
                        if (array[i].platePic.img)
                            memcpy(array[i].platePic.img, buffer + imgOff, array[i].platePic.size);
                    }
                }
            }
        }
 
    public:
        int num;
        DataMsg *array;
        MediaInfAiPic fullPic;//背景图，根据配置决定是否有截图
    };
}// namespace Emx
 
#endif//EMX_MediaInfAiDataVehicleRecognition_HPP