///-------------------------------------------------------------------------------------------------
/// 
/// \brief Classes of the Box Features Selection Trainer.
/// \author Thibaut Monseigne (Inria).
/// \version 1.0.
/// \date 14/02/2020.
/// \copyright <a href="https://choosealicense.com/licenses/agpl-3.0/">GNU Affero General Public License v3.0</a>.
/// 
///-------------------------------------------------------------------------------------------------

#include <fstream>
#include <iostream>
#include <string>
#include "tinyxml2.h"
#include <algorithm>

// validation Test take in the algorithm output file, the expected output file (reference) and a tolerance threshold
bool comparison(const char* file1, const char* file2, bool removeSpace)
{
	// Load Files
	tinyxml2::XMLDocument xmlDoc1, xmlDoc2;
	if (xmlDoc1.LoadFile(file1) != 0)	// Check File 1 Exist and Loading
	{
		std::cerr << "Error during loading File." << std::endl;
		return false;
	}
	if (xmlDoc2.LoadFile(file2) != 0)	// Check File 2 Exist and Loading
	{
		std::cerr << "Error during loading Reference." << std::endl;
		return false;
	}

	// Load Roots
	tinyxml2::XMLNode *root1 = xmlDoc1.FirstChild(), *root2 = xmlDoc2.FirstChild();	// Get Root Node
	if (root1 == nullptr)				// Check Root Node Exist
	{
		std::cerr << "File format not compatible." << std::endl;
		return false;
	}
	if (root2 == nullptr)				// Check Root Node Exist
	{
		std::cerr << "Reference format not compatible." << std::endl;
		return false;
	}

	// Count Settings
	size_t n1 = 0, n2 = 0;
	for (tinyxml2::XMLElement* e = root1->FirstChildElement(); e != nullptr; e = e->NextSiblingElement()) { n1++; }
	for (tinyxml2::XMLElement* e = root2->FirstChildElement(); e != nullptr; e = e->NextSiblingElement()) { n2++; }
	if (n1 != n2)
	{
		std::cerr << "File and Reference are not same number of features." << std::endl;
		return false;
	}

	// Load settings
	tinyxml2::XMLElement *e1 = root1->FirstChildElement(), *e2 = root2->FirstChildElement();
	for (size_t i = 0; i < n1; ++i, e1 = e1->NextSiblingElement(), e2 = e2->NextSiblingElement())
	{
		std::string s1(e1->GetText()), s2(e2->GetText());
		std::cout << s1 << std::endl << s2 << std::endl;
		if (removeSpace)
		{
			s1.erase(std::remove_if(s1.begin(), s1.end(), [](const unsigned char x) { return std::isspace(x); }), s1.end());
			s2.erase(std::remove_if(s2.begin(), s2.end(), [](const unsigned char x) { return std::isspace(x); }), s2.end());
		}
		std::cout << s1 << std::endl << s2 << std::endl;

		if (s1 != s2)
		{
			std::cerr << "Setting number " << i << " is different, For File : \n" << s1 << "\nfor Reference : \n" << s2 << std::endl;
			return false;
		}
	}

	return true;
}

int main(const int argc, char* argv[])
{
	if (argc != 4)
	{
		std::cout << "Usage: " << argv[0] << " <file> <reference> <remove space (true or false)>\n";
		return -1;
	}

	bool remove;
	if (strcmp(argv[3], "true") == 0) { remove = true; }
	else if (strcmp(argv[3], "false") == 0) { remove = false; }
	else
	{
		std::cout << "Usage:  <remove space> must be true or false not " << argv[3] << std::endl;
		return -1;
	}

	if (!comparison(argv[1], argv[2], remove))
	{
		std::cout << "Algorithm failed validation test \n" << std::endl;
		return 1;
	}

	std::cout << "Test passed\n" << std::endl;
	return 0;
}