core.h

//
// Created by yaione on 2/26/2022.
//
 
#ifndef RMCV_CORE_H
#define RMCV_CORE_H
 
#include "utils.h"
 
namespace rm {
    enum CampType {
        CAMP_RED = 0, CAMP_BLUE = 1, CAMP_GUIDELIGHT = 2, CAMP_NEUTRAL = -1
    };
 
    enum AimMode {
        AIM_COMBAT = 0, AIM_BUFF = 1, AIM_SNIPE = 2
    };
 
    typedef std::vector<cv::Point> Contour;
 
    class LightBlob {
    public:
        float angle = 0;                      
        rm::CampType camp = rm::CAMP_NEUTRAL; 
        cv::Point2f center;                   
        cv::Point2f vertices[4];              
        cv::Size2f size;                      
 
        LightBlob() = default;
 
        explicit LightBlob(cv::RotatedRect box, rm::CampType camp = rm::CAMP_NEUTRAL);
    };
 
    class Armour {
    public:
        float rank = 0;                          
        rm::CampType camp = rm::CAMP_NEUTRAL;    
        cv::Point2f icon[4];                     
        cv::Point2f vertices[4];                 
 
        Armour() = default;
 
        explicit Armour(std::vector<rm::LightBlob> lightBlobs, float rank = 0, rm::CampType camp = rm::CAMP_NEUTRAL);
    };
 
    class Package {
    public:
        rm::CampType camp = rm::CAMP_NEUTRAL;  
        rm::AimMode mode = rm::AIM_COMBAT;     
        unsigned char speed = 0;               
        float pitch = 0;                       
        cv::Mat frame;
        cv::Mat binary;
        std::vector<rm::Armour> armours;
 
        Package(rm::CampType camp, rm::AimMode mode, unsigned char speed, float pitch, const cv::Mat &inputFrame,
                const cv::Mat &inputBinary);
 
        explicit Package(const std::shared_ptr<rm::Package> &input);
    };
 
}
 
#endif //RMCV_CORE_H