inital commit

CMakeLists.txt

@@ -0,0 +1,22 @@
+cmake_minimum_required(VERSION 3.0)
+
+project(imagetournament)
+
+set(SRC_FILES main.cpp)
+
+set(RESOURCE_LOCATION data)
+
+find_package(OpenCV REQUIRED)
+find_package(PkgConfig REQUIRED)
+pkg_check_modules(POPPLER REQUIRED poppler-cpp)
+
+link_directories(${CMAKE_CURRENT_BINARY_DIR})
+
+add_executable(${PROJECT_NAME} ${SRC_FILES} main.cpp)
+target_link_libraries( ${PROJECT_NAME} pthread ${OpenCV_LIBS})
+target_include_directories(${PROJECT_NAME} PRIVATE  ${OpenCV_INCLUDE_DIRS})
+add_dependencies(${PROJECT_NAME} ${PROJECT_NAME})
+target_compile_options(${PROJECT_NAME} PRIVATE "-std=c++2a" "-Wall" "-O2" "-g" "-fno-strict-aliasing" "-Wfatal-errors" "-Wno-reorder")
+
+set(CMAKE_INSTALL_PREFIX "/usr")
+install(TARGETS ${PROJECT_NAME} RUNTIME DESTINATION bin)

main.cpp

@@ -0,0 +1,139 @@
+#include <iostream>
+#include <filesystem>
+#include <opencv2/highgui.hpp>
+
+static constexpr char OUTPUT_DIR[] = "./out";
+
+static bool extensionMatches(const std::filesystem::path& path)
+{
+	std::string extension = path.extension();
+	if(extension == ".png")
+		return true;
+	if(extension == ".PNG")
+		return true;
+	if(extension == ".jpg")
+		return true;
+	if(extension == ".JPG")
+		return true;
+	if(extension == ".jpeg")
+		return true;
+	return false;
+}
+
+int main(int argc, char** argv)
+{
+	if(argc < 2)
+	{
+		std::cout<<"a directory must be specified\n";
+		return 1;
+	}
+
+	std::cout<<"Using folder "<<argv[1]<<'\n';
+
+	std::vector<std::filesystem::path> images;
+	const std::filesystem::path directoryPath{argv[1]};
+	for(const std::filesystem::directory_entry& dirent : std::filesystem::directory_iterator{directoryPath})
+	{
+		std::string fileName = dirent.path().filename();
+		if(dirent.is_regular_file() && extensionMatches(dirent.path()))
+		{
+			std::cout<<"Found: "<<dirent.path().filename()<<'\n';
+			images.push_back(dirent.path());
+		}
+	}
+
+	if(images.empty())
+	{
+		std::cout<<"No images found\n";
+		return 1;
+	}
+
+	std::filesystem::path outDirPath(OUTPUT_DIR);
+	if(!std::filesystem::is_directory(outDirPath) && !std::filesystem::create_directory(outDirPath))
+	{
+		std::cout<<"Could not create or read "<<OUTPUT_DIR<<'\n';
+		return 1;
+	}
+
+	cv::namedWindow("Viewer", cv::WINDOW_NORMAL);
+	cv::setWindowTitle("Viewer", "Choose");
+
+	std::vector<std::vector<std::filesystem::path>> rounds;
+
+	while(images.size() > 1)
+	{
+		std::cout<<"New round "<<images.size()<<" images remaining\n";
+		rounds.push_back(std::vector<std::filesystem::path>());
+		std::vector<size_t> removeIndexies;
+		for(size_t i = 0; i < images.size()-1; i+=2)
+		{
+			cv::Mat imageA = cv::imread(images[i]);
+			cv::Mat imageB = cv::imread(images[i+1]);
+
+			if(!imageA.data || !imageB.data)
+			{
+				std::cout<<"Images: "<<images[i]<<' '<<images[i+1]<<" failed to read";
+				return 1;
+			}
+			cv::Size2i outSize(imageA.cols+imageB.cols, std::max(imageA.rows, imageB.rows));
+			cv::Mat image = cv::Mat::zeros(outSize, imageA.type());
+
+			cv::Rect roiA(cv::Point(0, 0), imageA.size());
+			cv::Rect roiB(cv::Point(imageA.cols, 0), imageB.size());
+			imageA.copyTo(image(roiA));
+			imageB.copyTo(image(roiB));
+
+			cv::imshow("Viewer", image);
+			while(true)
+			{
+				int keycode = cv::waitKey(0);
+				if(keycode == 49)
+				{
+					removeIndexies.push_back(i+1);
+					break;
+				}
+				else if(keycode == 50)
+				{
+					removeIndexies.push_back(i);
+					break;
+				}
+			}
+		}
+		for(ssize_t i = removeIndexies.size()-1; i >= 0; --i)
+		{
+			rounds.back().push_back(images[removeIndexies[i]]);
+			images.erase(images.begin()+removeIndexies[i]);
+		}
+	}
+
+
+	for(size_t round = 0; round < rounds.size(); ++round)
+	{
+		std::string roundDirName = std::string("round_") + std::to_string(round);
+		if(!std::filesystem::is_directory(outDirPath/roundDirName) && !std::filesystem::create_directory(outDirPath/roundDirName))
+		{
+			std::cout<<"Could not create or read "<<outDirPath/roundDirName<<'\n';
+			return 1;
+		}
+		for(const std::filesystem::path& path : rounds[round])
+		{
+			std::cout<<"Saveing "<<path<<std::endl; //flush
+			std::error_code ec;
+			std::filesystem::create_hard_link(path, outDirPath/roundDirName/(path.filename()), ec);
+			if(ec.value() != 0)
+			{
+				std::cout<<"Could not create hard link for "<<outDirPath/roundDirName/path.filename()<<'\n';
+			}
+		}
+	}
+
+	std::error_code ec;
+	std::filesystem::path winnerPath(outDirPath/(std::string("winner_") + std::string(images[0].filename())));
+	std::filesystem::create_hard_link(images[0], winnerPath, ec);
+	if(ec.value() != 0)
+	{
+		std::cout<<"Could not create hard link for "<<winnerPath<<'\n';
+	}
+
+	return 0;
+}