/* UMANS: Unified Microscopic Agent Navigation Simulator
** Copyright (C) 2018-2020  Inria Rennes Bretagne Atlantique - Rainbow - Julien Pettré
**
** This program is free software: you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation, either version 3 of the License, or
** (at your option) any later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program. If not, see <https://www.gnu.org/licenses/>.
**
** Contact: crowd_group@inria.fr
** Website: https://project.inria.fr/crowdscience/
** See the file AUTHORS.md for a list of all contributors.
*/

#include <core/crowdSimulator.h>

#include <vector>
#include <iostream>
#include <fstream>

using namespace std;

void printBasicInfo()
{
	std::cout
		<< "-----------------------------------------------------" << std::endl
		<< "UMANS: Unified Microscopic Agent Navigation Simulator" << std::endl
		<< "Copyright(C) 2018 - 2020 Inria Rennes Bretagne Atlantique - Rainbow - Julien Pettre" << std::endl
		<< "-----------------------------------------------------" << std::endl << std::endl;
}

void printUsageInfo(const std::string& programName)
{
	std::cout
		<< "Usage: " << programName << " -i [-o] [-t]" << std::endl
		<< "  -i (or -input)     = An XML file describing the simulation scenario to run." << std::endl
		<< "                       For help on creating scenario files, please see the UMANS documentation." << std::endl
		<< "  -o (or -output)    = (optional) Name of a folder to which the simulation output will be written." << std::endl
		<< "                       The program will write a CSV file for each agent's trajectory." << std::endl
		<< "                       If you omit this, the program will run faster, but no results will be saved." << std::endl
		<< "  -t (or -nrThreads) = (optional, default=1) The number of parallel threads to use." << std::endl << std::endl;
}

int main( int argc, char * argv[] )
{
	printBasicInfo();
	
	std::string configFile = "", outputFolder = "";
	int nrThreads = 1;

	// parse the arguments one by one
	for (int i = 1; i + 1 < argc; i += 2)
	{
		std::string paramName(argv[i]);
		std::string paramValue(argv[i + 1]);

		if (paramName[0] != '-')
		{
			std::cerr << "Input error: " << paramName << " is not a valid parameter name." << std::endl;
			printUsageInfo(argv[0]);
			return 0;
		}

		if (paramName == "-i" || paramName == "-input")
			configFile = paramValue;
		else if (paramName == "-o" || paramName == "-output")
			outputFolder = paramValue;
		else if (paramName == "-t" || paramName == "-nrThreads")
			nrThreads = atoi(paramValue.c_str());
	}

	// input file is mandatory
	if (configFile == "")
	{
		std::cerr << "Input error: Please specify an input scenario." << std::endl;
		printUsageInfo(argv[0]);
		return 0;
	}

	// number of threads must be at least 1
	if (nrThreads < 1)
	{
		std::cerr << "Input error: Please specify a positive number of threads." << std::endl;
		printUsageInfo(argv[0]);
		return 0;
	}

	// output folder may be empty; if so, show a warning
	if (outputFolder == "")
	{
		std::cout << "Input warning: No output folder specified." << std::endl
			<< "The program will not write any simulation results to CSV files." << std::endl;
	}
	
	// run the simulation
	CrowdSimulator* cs = CrowdSimulator::FromConfigFile(configFile, true);
	if (cs != nullptr)
	{
		cs->GetWorld()->SetNumberOfThreads(nrThreads);
		if (outputFolder != "")
			cs->StartCSVOutput(outputFolder, false); // false = don't save any files until the simulation ends
		cs->RunSimulationUntilEnd();
	}

	std::cout << "Simulation done" << std::endl;

	delete cs;

	std::cout << "---------------" << std::endl;

}