#include "det_obj_header.h"

#include "../ai_engine_module/HumanFeature.h"
#include "../ai_engine_module/HumanParsing.h"
#include "../ai_engine_module/VehicleColor.h"
#include "../ai_engine_module/VehiclePlate.h"
#include "../ai_engine_module/VPDProcess.h"

#include "../util/vpc_util.h"

#include <map>
#include <vector>

using namespace std;

class FeatureAnalysis
{
public:
    FeatureAnalysis(/* args */);
    ~FeatureAnalysis();

    int init(int dev_id, string model_dir);

    int update_object_info(ObjectInfo& res_obj, vpc_img_info img_info);

private:
    void update_human_info(ObjectInfo& res_obj, vpc_img_info& img_info);
    void update_car_info(ObjectInfo& res_obj, vpc_img_info& img_info);

private:
    HumanFeature m_human_feature;
    HumanParsing m_human_parsing;
    VehicleColor m_vehicle_color;
    VehiclePlate m_vehicle_plate;
    VPDProcess m_vpd;

    int m_dev_id{0};
};