/**********************************************************************/
/*   ____      _ _____ _   _       ____  __  __        ______     __  */
/*  / ___|    | |_   _| | | |     |  _ \|  \/  |      / ___\ \   / /  */
/*  \___ \ _  | | | | | | | |_____| |_) | |\/| |_____| |    \ \ / /   */
/*   ___) | |_| | | | | |_| |_____|  _ <| |  | |_____| |___  \ V /    */
/*  |____/ \___/  |_|  \___/      |_| \_\_|  |_|      \____|  \_/     */
/*                                                                    */
/**********************************************************************/

#include <iostream>
#include <thread>
#include <opencv2/core/core.hpp>
#include <serial/serial.h>
#include <camera/camera_wrapper.h>
#include <camera/video_wrapper.h>
#include <camera/wrapper_head.h>
#include <energy/energy.h>
#include <armor_finder/armor_finder.h>
#include <options/options.h>
#include <additions/additions.h>
#include <config/setconfig.h>

#define DO_NOT_CNT_TIME

#include <log.h>

using namespace cv;
using namespace std;

mcu_data mcuData = {    // 单片机端回传结构体
        0,              // 当前云台yaw角
        0,              // 当前云台pitch角
        BIG_ENERGY_STATE,    // 当前状态，自瞄-大符-小符
        0,              // 云台角度标记位
        1,              // 是否启用数字识别
        ENEMY_RED,      // 敌方颜色
        0,              // 能量机关x轴补偿量
        0,              // 能量机关y轴补偿量
};

WrapperHead *video_gimbal = nullptr;    // 云台摄像头视频源
WrapperHead *video_chassis = nullptr;   // 底盘摄像头视频源

Serial serial(115200);                  // 串口对象
uint8_t last_state = INIT_STATE;     // 上次状态，用于初始化
// 自瞄主程序对象
ArmorFinder armorFinder(mcuData.enemy_color, serial, PROJECT_DIR"/tools/para/", mcuData.use_classifier);
// 能量机关主程序对象
Energy energy(serial, mcuData.enemy_color);

int box_distance = 0;

int main(int argc, char *argv[]) {
    process_options(argc, argv);            // 处理命令行参数
    thread receive(uartReceive, &serial);   // 开启串口接收线程

    int from_camera = 1;                    // 根据条件选择视频源
    if (!run_with_camera) {
        cout << "Input 1 for camera, 0 for video files" << endl;
        cin >> from_camera;
    }

    while (true) {
        // 打开视频源
        if (from_camera) {
            video_gimbal = new CameraWrapper(ARMOR_CAMERA_GAIN, 0/*, "armor"*/);
            video_chassis = new CameraWrapper(ENERGY_CAMERA_GAIN, 0/*, "energy"*/);
        } else {
            video_gimbal = new VideoWrapper("/home/sun/项目/energy_video/gimbal255.avi");
            video_chassis = new VideoWrapper("/home/sun/项目/energy_video/gimbal255.avi");
        }
        if (video_gimbal->init()) {
            LOGM("video_gimbal source initialization successfully.");
        } else {
            LOGW("video_gimbal source unavailable!");
        }
        if (video_chassis->init()) {
            LOGM("video_chassis source initialization successfully.");
        } else {
            LOGW("video_chassis source unavailable!");
        }

        // 跳过前10帧噪声图像。
        Mat gimbal_src, chassis_src;
        for (int i = 0; i < 10; i++) {
            if (video_gimbal) {
                video_gimbal->read(gimbal_src);
            }
            if (video_chassis) {
                video_chassis->read(chassis_src);
            }
        }
        bool ok = true;
        cout << "start running" << endl;
        do {
//            CNT_TIME("Total", {
            if (mcuData.state == BIG_ENERGY_STATE) {//大能量机关模式
                if (last_state != BIG_ENERGY_STATE) {//若上一帧不是大能量机关模式，即刚往完成切换，则需要初始化
                    destroyAllWindows();
                    if (from_camera) {
                        delete video_gimbal;
                        video_gimbal = new CameraWrapper(ENERGY_CAMERA_GAIN, 0/*, "armor"*/);
                        if (video_gimbal->init()) {
                            LOGM("video_gimbal source initialization successfully.");
                        } else {
                            LOGW("video_gimbal source unavailable!");
                        }
                    }
                    checkReconnect(video_chassis->read(chassis_src));
                    energy.setBigEnergyInit();
                }
                ok = checkReconnect(video_gimbal->read(gimbal_src));
                video_chassis->read(chassis_src);
#ifdef GIMBAL_FLIP_MODE
                flip(gimbal_src, gimbal_src, GIMBAL_FLIP_MODE);
#endif
#ifdef CHASSIS_FLIP_MODE
                flip(chassis_src, chassis_src, CHASSIS_FLIP_MODE);
#endif
                if (!from_camera) extract(gimbal_src, chassis_src);
                if (save_video) saveVideos(gimbal_src, chassis_src);//保存视频
                if (show_origin) showOrigin(gimbal_src, chassis_src);//显示原始图像
                energy.runBig(gimbal_src, chassis_src);
//                energy.runBig(gimbal_src);
                last_state = mcuData.state;//更新上一帧状态
            } else if (mcuData.state == SMALL_ENERGY_STATE) {
                if (mcuData.state != SMALL_ENERGY_STATE) {
                    destroyAllWindows();
                    if (from_camera) {
                        delete video_gimbal;
                        video_gimbal = new CameraWrapper(ENERGY_CAMERA_GAIN, 0/*, "armor"*/);
                        if (video_gimbal->init()) {
                            LOGM("video_gimbal source initialization successfully.");
                        } else {
                            LOGW("video_gimbal source unavailable!");
                        }
                    }
                    energy.setSmallEnergyInit();
                }
                ok = checkReconnect(video_gimbal->read(gimbal_src));
#ifdef GIMBAL_FLIP_MODE
                flip(gimbal_src, gimbal_src, GIMBAL_FLIP_MODE);
#endif
                if (!from_camera) extract(gimbal_src);
                if (save_video) saveVideos(gimbal_src);//保存视频
                if (show_origin) showOrigin(gimbal_src);//显示原始图像
                energy.runSmall(gimbal_src);
                last_state = mcuData.state;//更新上一帧状态
            } else {                                         // 自瞄模式
                if (last_state != ARMOR_STATE) {
                    destroyAllWindows();
                    if (from_camera) {
                        delete video_gimbal;
                        video_gimbal = new CameraWrapper(ARMOR_CAMERA_GAIN, 0/*, "armor"*/);
                        if (video_gimbal->init()) {
                            LOGM("video_gimbal source initialization successfully.");
                        } else {
                            LOGW("video_gimbal source unavailable!");
                        }
                    }
                }
                last_state = mcuData.state;
                ok = checkReconnect(video_gimbal->read(gimbal_src));
#ifdef GIMBAL_FLIP_MODE
                flip(gimbal_src, gimbal_src, GIMBAL_FLIP_MODE);
#endif
                if (!from_camera) extract(gimbal_src);
                if (save_video) saveVideos(gimbal_src);
                if (show_origin) showOrigin(gimbal_src);
                CNT_TIME("Armor Time", {
                    armorFinder.run(gimbal_src);
                });
            }
//                cv::waitKey(0);
//            });
        } while (ok);
        delete video_gimbal;
        video_gimbal = nullptr;
        delete video_chassis;
        video_chassis = nullptr;
        cout << "Program fails. Restarting" << endl;
    }
    return 0;
}

