/**
 * 车号第一步过滤Engine
 * */

#ifndef TRAINSTEP1FILTERENGINE_H
#define TRAINSTEP1FILTERENGINE_H

#include "AppCommon.h"
#include "EngineBase.h"
#include "EngineFactory.h"
#include "json/json.h"

class TrainStep1FilterEngine : public ai_matrix::EngineBase
{
public:
    TrainStep1FilterEngine();
    ~TrainStep1FilterEngine();

    APP_ERROR Init() override;
    APP_ERROR DeInit() override;
    APP_ERROR Process() override;

private:

    std::string strPort0_;
    ai_matrix::BaseConfig baseConfig_;
    ai_matrix::IdentifyConfig identifyConfig_;
    ai_matrix::DataSourceConfig dataSourceConfig_;

    std::map<int, std::vector<CalculateInfo>> mapCalDirection_;

    std::shared_ptr<InferenceResultData> pInferenceResultDataPre_ = nullptr;
    // 第一步无变化帧数
    int iNotChgCount_ = 0;
    // 列车状态标识
    int iTrainStatus_;
    // 保存3帧信息，用于实时判断方向
    std::queue<std::shared_ptr<InferenceResultData>> queInferenceResultData_;
    // 倒车信息记录
    std::stack<TrainBackInfo> stackBackInfo_;
    std::map<int, std::string> mapTargetStr_;
    // 参数初始化(列车结束时需调用)
    void initParam();
    // 读取json文件
    bool readJson(std::string &strFilePath, Json::Value &jvInfo, int i = 1);
    // 获取火车运动状态
    int getTrainStatus();
    // 根据当前帧数据，处理上一帧数据
    void dealProcessDataPre(std::shared_ptr<InferenceResultData> pInferenceResultData);
    // 记录倒车大框信息
    void addBackInfo();
    // 判断是否回到倒车前的位置
    bool isEndDealBackInfo();
    //计算方向新
    void calculateDirection();

    void sendComeTrain(const std::string strTrainDate, const std::string strTrainName, const int iDirection);
};

ENGINE_REGIST(TrainStep1FilterEngine)

#endif