#include <iostream>
#include <string>
#include <signal.h>
#include "deal.h"
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <string.h>
#include "JpgDecoder.h"
#include <filesystem>
#include <algorithm>
#include <sys/time.h>

namespace fs = std::filesystem;
void handler(int sig)
{
	printf("Get handler sig");
	string strcmd = "ps -ef | grep test_JBL_Check| awk '{print $2}' | xargs kill -9 ";
	const char *cmd = strcmd.c_str();
	printf("delete test_JBL_Check success");
	if (-1 == system(cmd))
	{
		std::cout << "error" << std::endl;
	}

	// delete UnderCarriageCoreLogic::GetInstance();
	exit(0);
}

int _sysmkdir(const std::string &dir)
{
	int ret = mkdir(dir.c_str(), S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
	if (ret && errno == EEXIST)
	{
		// printf("dir[%s] already exist.\n", dir.c_str());
	}
	else if (ret)
	{
		printf("create dir[%s] error: %d %s\n", dir.c_str(), ret, strerror(errno));
		return -1;
	}
	else
	{
		printf("create dir[%s] success.\n", dir.c_str());
	}
	return 0;
}
std::string __getParentDir(const std::string &dir)
{
	std::string pdir = dir;
	if (pdir.length() < 1 || (pdir[0] != '/'))
	{
		return "";
	}
	while (pdir.length() > 1 && (pdir[pdir.length() - 1] == '/'))
		pdir = pdir.substr(0, pdir.length() - 1);

	pdir = pdir.substr(0, pdir.find_last_of('/'));
	return pdir;
}
int _sysmkdirs(const std::string &dir)
{
	int ret = 0;
	if (dir.empty())
		return -1;
	std::string pdir;
	if ((ret = _sysmkdir(dir)) == -1)
	{
		pdir = __getParentDir(dir);
		if ((ret = _sysmkdirs(pdir)) == 0)
		{
			ret = _sysmkdirs(dir);
		}
	}

	return ret;
}

int main(int argc, char *argv[])
{

	deal test;

	if (argc > 1 && string(argv[1]) == "-h")
	{
		cout << "******************************************************" << endl;
		cout << "*** 1、./test_JBL_Check:                      默认处理一套图 图片文件夹位于 ../data/img/t1，结果位于/home/aidlux/BOE/testresult" << endl;
		cout << "*** 2、./test_JBL_Check  -s:                  1 的基础上增加 过程存图，包括edge 和字符检测的中间结果存图，图片位于 当前文件夹。" << endl;
		cout << "*** 3、./test_JBL_Check -fjc  filepath:       处理精测套图大图 结果位于/home/aidlux/BOE/ResultImg" << endl;
		cout << "*** 4、./test_JBL_Check -feai  filepath num:  批量测试边缘检测算法，num 至多处理张数，结果：/home/aidlux/BOE/Edge" << endl;
		cout << "*** 5、./test_JBL_Check -falign  filepath num :   批量测试定位算法，num 至多处理张数，结果：/home/aidlux/BOE/Align" << endl;
		cout << "*** 6、./test_JBL_Check -rjson  filepath:      单张复测，filepath xx/xx/3A3K380001B1DK_20240408_152157_Main_0_2_L255" << endl;
		cout << "*** 6、./test_JBL_Check -rjsonall  filepath:      一套图复测，filepath xx/xx/" << endl;
		cout << "*** 7、./test_CellAOI   -decode [dir]:          批量解码 .ytimage → .png 到 /home/aidlux/BOE/CELL_AOI/decode_img" << endl;

		cout << "******************************************************" << endl;
		return 0;
	}

	printf("argc = %d\n", argc);
	for (int i = 0; i < argc; i++)
	{
		printf("argv[%d]=%s\n", i, argv[i]);
	}

	bool bDecodeMode = false;
	std::string decodeInputDir;

	if (argc > 1 && string(argv[1]) != "-h")
	{

		for (int i = 1; i < argc; i++)
		{
			if (string(argv[i]) == "-decode")
			{
				bDecodeMode = true;
				if (i + 1 < argc && argv[i + 1][0] != '-')
				{
					decodeInputDir = string(argv[i + 1]);
					i++;
				}
			}
			else if (string(argv[i]) == "-rs")
			{
				test.runConfig.run_Type = Process_Run_SaveImg;
			}
			else if (string(argv[i]) == "-rd")
			{
				test.runConfig.run_Type = Process_Run_Detect;
			}
			else if (string(argv[i]) == "-rjson")
			{
				if (i + 1 >= argc)
				{
					printf("error Not Path \n");
					return -1;
				}

				test.runConfig.run_Type = Process_Run_ReJson;
				test.runConfig.filePath = string(argv[i + 1]);
			}
			else if (string(argv[i]) == "-f" && i + 1 < argc)
			{
				test.runConfig.det_Type = RUNTYPE_RUN_File;
				test.runConfig.filePath = string(argv[i + 1]);
			}
			else if (string(argv[i]) == "-fe" && i + 1 < argc)
			{
				test.runConfig.det_Type = RUNTYPE_RUN_EDGE_TEST;
				test.runConfig.filePath = string(argv[i + 1]);
			}
			else if (string(argv[i]) == "-feai" && i + 1 < argc)
			{
				test.runConfig.det_Type = RUNTYPE_RUN_EDGE_TEST_AI;
				test.runConfig.filePath = string(argv[i + 1]);
			}
			else if (string(argv[i]) == "-m")
			{
				test.runConfig.det_Type = RUNTYPE_RUN_Merge_Img;
			}
			if (string(argv[i]) == "-s")
			{
				test.runConfig.bSaveProcessImg = true;
			}
			if (string(argv[i]) == "-ud")
			{
				test.runConfig.bdecode = false;
			}
			if (string(argv[i]) == "-fud" && i + 1 < argc)
			{
				test.runConfig.det_Type = RUNTYPE_RUN_File;
				test.runConfig.filePath = string(argv[i + 1]);
				test.runConfig.bdecode = false;
			}
		}
	}

	// ====== 批量解码模式 (-decode) ======
	if (bDecodeMode)
	{
		std::string inputDir = decodeInputDir.empty()
			? std::string("../data/img/t1")
			: decodeInputDir;
		std::string outputDir = "/home/aidlux/BOE/CELL_AOI/decode_img";

		printf("\n========== 批量解码模式 ==========\n");
		printf("  输入目录: %s\n", inputDir.c_str());
		printf("  输出目录: %s\n", outputDir.c_str());
		printf("===================================\n\n");

		// 检查输入目录
		if (!fs::exists(inputDir) || !fs::is_directory(inputDir))
		{
			printf("[decode] 输入目录不存在: %s\n", inputDir.c_str());
			return -1;
		}

		// 创建输出目录
		_sysmkdirs(outputDir);

		// 只解码 .ytimage 原始多JPEG文件
		auto isYtimageFile = [](const std::string &ext) -> bool {
			std::string lower = ext;
			std::transform(lower.begin(), lower.end(), lower.begin(), ::tolower);
			return (lower == ".ytimage");
		};

		// 扫描所有 .ytimage 文件
		std::vector<fs::path> imageFiles;
		try
		{
			for (const auto &entry : fs::recursive_directory_iterator(inputDir))
			{
				if (entry.is_regular_file() && isYtimageFile(entry.path().extension().string()))
					imageFiles.push_back(entry.path());
			}
		}
		catch (const fs::filesystem_error &e)
		{
			printf("[decode] 扫描目录出错: %s\n", e.what());
			return -1;
		}

		if (imageFiles.empty())
		{
			printf("[decode] 未找到 .ytimage 文件: %s\n", inputDir.c_str());
			return 0;
		}

		printf("[decode] 找到 %zu 个 .ytimage 文件，开始批量解码...\n", imageFiles.size());

		// 逐个解码并保存
		int successCount = 0;
		int failCount = 0;
		struct timeval tvTotalStart, tvTotalEnd;
		gettimeofday(&tvTotalStart, nullptr);

		for (size_t i = 0; i < imageFiles.size(); ++i)
		{
			const auto &imgPath = imageFiles[i];
			std::string inputPath = imgPath.string();

			printf("[%zu/%zu] %s ... ", i + 1, imageFiles.size(), imgPath.filename().c_str());
			fflush(stdout);

			struct timeval tv1, tv2;
			gettimeofday(&tv1, nullptr);

			// ★ 调用 JpgDecoder 独立模块解码
			cv::Mat decoded = decodeJpgImage(inputPath);

			gettimeofday(&tv2, nullptr);
			long elapsed = (tv2.tv_sec - tv1.tv_sec) * 1000 + (tv2.tv_usec - tv1.tv_usec) / 1000;

			if (decoded.empty())
			{
				printf("失败 (%ld ms)\n", elapsed);
				failCount++;
				continue;
			}

			// 提取 Product ID（文件路径往上两级目录）
			std::string productId = imgPath.parent_path().parent_path().filename().string();

			// 文件名 stem（去掉 .ytimage 后缀）
			std::string stem = imgPath.stem().string(); // e.g. "CAA"

			// 输出目录：outputDir / ProductID
			fs::path outSubDir = fs::path(outputDir) / productId;
			_sysmkdirs(outSubDir.string());

			// 保存两份：M_ 前缀 + A_ 前缀
			bool ok = true;
			for (const auto &prefix : {"M_", "A_"})
			{
				fs::path outPath = outSubDir / (std::string(prefix) + stem + ".png");
				if (!cv::imwrite(outPath.string(), decoded))
				{
					printf("保存失败: %s\n", outPath.filename().c_str());
					ok = false;
				}
			}

			if (ok)
			{
				printf("OK → %s/M_%s.png + A_%s.png [%dx%d] (%ld ms)\n",
					productId.c_str(), stem.c_str(), stem.c_str(),
					decoded.cols, decoded.rows, elapsed);
				successCount++;
			}
			else
			{
				failCount++;
			}
		}

		gettimeofday(&tvTotalEnd, nullptr);
		long totalElapsed = (tvTotalEnd.tv_sec - tvTotalStart.tv_sec) * 1000 +
							(tvTotalEnd.tv_usec - tvTotalStart.tv_usec) / 1000;

		printf("\n========================================\n");
		printf("[decode] 批量解码完成!\n");
		printf("  输入目录: %s\n", inputDir.c_str());
		printf("  输出目录: %s\n", outputDir.c_str());
		printf("  成功: %d 张\n", successCount);
		printf("  失败: %d 张\n", failCount);
		printf("  总耗时: %ld ms\n", totalElapsed);
		printf("========================================\n");

		return (successCount > 0) ? 0 : -1;
	}

	test.runConfig.print("config");
	// getchar();
	signal(SIGINT, handler);
	test.start();

	while (true)
	{
		usleep(10 * 1000);
	}

	return 0;
}