From bf4cb88d2852d1f13b29d45e6e49af093ae0f45b Mon Sep 17 00:00:00 2001
From: uvos <devnull@uvos.xyz>
Date: Fri, 18 Jun 2021 16:51:34 +0200
Subject: [PATCH] allow arbitary calcurve points, change calcurve format a bit

---
 src/main.cpp | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/src/main.cpp b/src/main.cpp
index 8db9a30..4b6af77 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -275,10 +275,19 @@ int perfromOperation(int operation, char** fileNames, const Config& config)
 	}
 	else if(operation == CREATE_CURVE)
 	{
-		cv::Mat curve = cv::Mat::zeros(2, 10, CV_32FC1);
+		std::cout<<"how many coordinate pares are required?\n> ";
+		int num = 0;
+		std::cin>>num;
+		if(std::cin.fail())
+		{
+			std::cin.clear();
+			std::cout<<"invalid number";
+			return -1;
+		}
+		cv::Mat curve = cv::Mat::zeros(2, num, CV_32FC1);
 		float* keys   = curve.ptr<float>(0);
 		float* values = curve.ptr<float>(1);
-		std::cout<<"Please type 10 coordinate pairs\n";
+		std::cout<<"Please type "<<num<<" coordinate pairs\n";
 		for(int i = 0; i < curve.cols; ++i)
 		{
 			std::cout<<i<<"> ";
@@ -309,7 +318,7 @@ int perfromOperation(int operation, char** fileNames, const Config& config)
 			std::cout<<keys[i]<<' '<<values[i]<<'\n';
 		
 		cv::FileStorage fs(!config.output.empty() ? config.output : "curve.mat", cv::FileStorage::WRITE);
-		fs<<"curve"<<curve;
+		fs<<"cal"<<curve;
 		fs.release();
 		
 		std::cout<<"Curve saved to "<<(!config.output.empty() ? config.output : "curve.mat")<<'\n';