上传文件至 'v1'

v1/test.cpp

@@ -0,0 +1,126 @@
+#include "WaferRebuild.h"
+#include <fstream>
+#include <iostream>
+#include <regex>
+#include <opencv2/opencv.hpp>
+#include <dirent.h>
+#include <thread>
+
+// 自定义比较函数
+bool numeric_string_compare(const std::string& s1, const std::string& s2) {
+	// 正则表达式匹配所有数字
+	std::regex re("(\\d+)");
+	std::sregex_iterator it1(s1.begin(), s1.end(), re), it2(s2.begin(), s2.end(), re);
+	std::sregex_iterator reg_end;
+
+	while (it1 != reg_end && it2 != reg_end) {
+		int num1 = std::stoi(it1->str());
+		int num2 = std::stoi(it2->str());
+		if (num1 != num2)
+			return num1 < num2;
+		++it1;
+		++it2;
+	}
+
+	// 如果所有数字相等，按字典顺序排序
+	return s1 < s2;
+}
+
+
+int read_files_in_dir(const char* p_dir_name, std::vector<std::string>& file_names) {
+	DIR* p_dir = opendir(p_dir_name);
+	if (p_dir == nullptr) {
+		return -1;
+	}
+
+	struct dirent* p_file = nullptr;
+	while ((p_file = readdir(p_dir)) != nullptr) {
+		if (strcmp(p_file->d_name, ".") != 0 &&
+			strcmp(p_file->d_name, "..") != 0) {
+			//std::string cur_file_name(p_dir_name);
+			//cur_file_name += "/";
+			//cur_file_name += p_file->d_name;
+			std::string cur_file_name(p_file->d_name);
+			file_names.push_back(cur_file_name);
+		}
+	}
+	std::sort(file_names.begin(), file_names.end(), numeric_string_compare);
+	closedir(p_dir);
+	return 0;
+}
+
+int main()
+{
+	WaferRebuild wr;
+	nlohmann::json cfg;
+	// A:/code/win_pratice/PmdPratice/cpp/WaferRebuild/x64/WaferRebuilddll/resource
+	std::ifstream is("A:/code/win_pratice/PmdPratice/cpp/WaferRebuilddll_bak/WaferRebuilddll/resource/cfg.json");
+	is >> cfg;
+	auto a = wr.initParam(cfg);
+
+	std::vector<std::string> file_names;
+	const char* image_path = "A:/code/win_pratice/PmdPratice/cpp/WaferRebuilddll_bak/WaferRebuilddll/resource/img_pats/";
+	if (read_files_in_dir(image_path, file_names) < 0) {
+		std::cout << "read_files_in_dir failed." << std::endl;
+	}
+
+	
+	std::vector< std::shared_ptr<infra::Image>> image_list;
+	for (int f = 0; f < (int)file_names.size(); f++) {
+
+		std::shared_ptr<infra::Image> frame = std::make_shared<infra::Image>(2048 * 2448);
+		cv::Mat colorImage;
+		cv::Mat img = cv::imread(std::string(image_path) + "/" + file_names[f],0);
+		if (img.empty()) continue;
+
+		// cv::cvtColor(img, colorImage, cv::COLOR_GRAY2BGR);
+		// memcpy(frame->data(), colorImage.data, colorImage.rows * colorImage.cols * colorImage.channels());
+		memcpy(frame->data(), img.data, img.rows * img.cols * img.channels());
+		image_list.push_back(frame);
+	}
+	std::shared_ptr<infra::CloudPoint<double>> cloud_points0 = std::make_shared<infra::CloudPoint<double>>();
+	auto result = wr.calcCloud(image_list, cloud_points0, 0);
+
+
+	/*
+	const int numThreads = 4;
+	std::vector<std::thread> threads;
+	// 创建多个 cloud_points 对象
+	std::shared_ptr<infra::CloudPoint<double>> cloud_points0 = std::make_shared<infra::CloudPoint<double>>();
+	std::shared_ptr<infra::CloudPoint<double>> cloud_points1 = std::make_shared<infra::CloudPoint<double>>();
+	std::shared_ptr<infra::CloudPoint<double>> cloud_points2 = std::make_shared<infra::CloudPoint<double>>();
+	std::shared_ptr<infra::CloudPoint<double>> cloud_points3 = std::make_shared<infra::CloudPoint<double>>();
+
+	// 将 cloud_points 放入 vector
+	std::vector<std::shared_ptr<infra::CloudPoint<double>>> cloud_points = { cloud_points0, cloud_points1, cloud_points2, cloud_points3 };
+
+	// 创建多个线程并运行 calcCloud 函数，每个线程传递不同的参数组合
+	for (int i = 0; i < numThreads; ++i) {
+		threads.push_back(std::thread([&wr, &image_list, &cloud_points, i]() {
+			auto result = wr.calcCloud(image_list, cloud_points[i], i);
+			// 可以在这里处理函数返回的结果，如输出或其他逻辑
+			std::cout << "Thread " << i << " finished with result: " << std::get<0>(result) << ", " << std::get<1>(result) << "\n";
+			}));
+	}
+
+	// 等待所有线程完成
+	for (auto& thread : threads) {
+		thread.join();
+	}
+
+	std::map<int32_t, std::shared_ptr<infra::CloudPoint<double>>> cloud_points_list = {
+		{0, cloud_points0},
+		{1, cloud_points1},
+		{2, cloud_points2},
+		{3, cloud_points3}
+	};;
+	std::shared_ptr<infra::CloudPoint<double>> cloud_out = std::make_shared<infra::CloudPoint<double>>();
+	
+	auto m_ret = wr.mergeCloud(cloud_points_list, cloud_out);
+	std::cout << "Merge sucess!" << std::endl;
+	std::cout <<  std::get<0>(m_ret) << ", " << std::get<1>(m_ret) << "\n";
+	*/
+	std::cout << "Success!" << std::endl;
+	system("pause");
+	return 0;
+}