MediaInfAiDataFaceRecognition.hpp

//
// Created by tangfahong on 2024/4/16.
//
 
#ifndef EMX_MediaInfAiDataFaceRecognition_HPP
#define EMX_MediaInfAiDataFaceRecognition_HPP
 
#include "MediaInfAiData.hpp"
#include "MediaRegion.hpp"
#include <cstring>
 
namespace Emx {
    class MediaInfAiDataFaceRecognition : public MediaInfAiData {
    public:
        MediaInfAiDataFaceRecognition() : MediaInfAiData(), num(0), array(nullptr) {
            memset(&fullPic, 0, sizeof(MediaInfAiPic));
        }
 
        virtual ~MediaInfAiDataFaceRecognition() { Free(); }
 
    public:
        struct DataMsg {
            MediaInfAiPic pic;     
            uint64_t id;           
            uint8_t quality;       
            uint8_t sharpness;     
            uint8_t age;           
            uint8_t gender;        
            uint8_t skin;          
            uint8_t glasses;       
            uint8_t beard;         
            uint8_t hat;           
            uint8_t hairColor;     
            uint8_t hairStyle;     
            uint8_t emotion;       
            uint8_t liveness;      
            int32_t ipd;           
            uint64_t captureTime;  
            uint64_t startTime;    
            uint64_t endTime;      
            Rect rect;             
            uint32_t isMask;       
            uint32_t maskColor;    
            uint32_t maskType;     
            int32_t libId;         
            int32_t similarity;    
            KeyPoint keyPoints[32];
            int keyPointNums;      
            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 ((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; }
            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++) {
                if (array[i].pic.img && array[i].pic.size > 0) {
                    memcpy(buffer + imgOff, array[i].pic.img, array[i].pic.size);
                }
                array[i].imgOff = imgOff;
                memcpy(buffer + off, array + i, (int) sizeof(DataMsg));
                imgOff += array[i].pic.size;
                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++) {
                    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 + array[i].imgOff, array[i].pic.size);
                    }
                }
            }
        }
 
    public:
        int num;
        DataMsg *array;
        MediaInfAiPic fullPic;//背景图，根据配置决定是否有截图
    };
}// namespace Emx
 
#endif//EMX_MediaInfAiDataFaceRecognition_HPP