MediaInfAiDataClassify.hpp

//
// Created by tangfahong on 2024/07/20.
//
 
#ifndef EMX_MediaInfAiDataClassify_HPP
#define EMX_MediaInfAiDataClassify_HPP
 
#include "MediaInfAiData.hpp"
#include "MediaRegion.hpp"
#include <cstring>
 
namespace Emx {
    class MediaInfAiDataClassify : public MediaInfAiData {
    public:
        MediaInfAiDataClassify() : MediaInfAiData(), num(0), array(nullptr) {
            memset(&fullPic, 0, sizeof(MediaInfAiPic));
        }
 
        virtual ~MediaInfAiDataClassify() { Free(); }
 
    public:
        enum TypeE {
            None = 0,           
            Face = 1,           
            Person = 2,         
            HumanHead = 3,      
            Bicycle = 4,        
            Car = 5,            
            Motorcycle = 6,     
            Bus = 7,            
            Truck = 8,          
            Train = 9,          
            Bird = 10,          
            Cat = 11,           
            Dog = 12,           
            Horse = 13,         
            Cow = 14,           
            Sheep = 15,         
            Pig = 16,           
            Rabbit = 17,        
            Duck = 18,          
            Goose = 19,         
            Chicken = 20,       
            Fish = 21,          
            Elephant = 22,      
            Lion = 23,          
            Tiger = 24,         
            Monkey = 25,        
            Panda = 26,         
            Giraffe = 27,       
            Snake = 28,         
            Tree = 29,          
            Flower = 30,        
            Grass = 31,         
            Mushroom = 32,      
            Cactus = 33,        
            Boat = 34,          
            Submarine = 35,     
            Airplane = 36,      
            Helicopter = 37,    
            Bed = 38,           
            Sofa = 39,          
            Table = 40,         
            Chair = 41,         
            Wardrobe = 42,      
            Television = 43,    
            Refrigerator = 44,  
            WashingMachine = 45,
            MicrowaveOven = 46, 
            Computer = 47,      
            MobilePhone = 48,   
            House = 49,         
            Villa = 50,         
            School = 51,        
            Hospital = 52,      
            Library = 53,       
            ShoppingMall = 54,  
            Stadium = 55,       
            Eye = 56,           
            Nose = 57,          
            Mouth = 58,         
            Hand = 59,          
            Foot = 60,          
            Apple = 61,         
            Banana = 62,        
            Orange = 63,        
            Grape = 64,         
            Water = 65,         
            Milk = 66,          
            Coffee = 67,        
            Tea = 68,           
            Lake = 69,          
            Ocean = 70,         
            Pond = 71,          
            River = 72,         
            Stream = 73,        
            Flame = 74,         
            Smoke = 75,         
            End
        };
 
        struct DataMsg {
            TypeE type;
            MediaInfAiPic pic;
            uint64_t id;      
            Rect rect;        
            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++) {
                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;
                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;
                    }
                }
            }
        }
 
    public:
        int num;
        DataMsg *array;
        MediaInfAiPic fullPic;//背景图，根据配置决定是否有截图
    };
}// namespace Emx
 
#endif//EMX_MediaInfAiDataClassify_HPP