///
/// \author	John Farrier
///
/// \copyright Copyright 2015, 2016, 2017, 2018. 2019 John Farrier
///
/// Licensed under the Apache License, Version 2.0 (the "License");
/// you may not use this file except in compliance with the License.
/// You may obtain a copy of the License at
///
/// http://www.apache.org/licenses/LICENSE-2.0
///
/// Unless required by applicable law or agreed to in writing, software
/// distributed under the License is distributed on an "AS IS" BASIS,
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
/// See the License for the specific language governing permissions and
/// limitations under the License.
///

#include <celero/Archive.h>
#include <celero/Benchmark.h>
#include <celero/Callbacks.h>
#include <celero/Celero.h>
#include <celero/CommandLine.h>
#include <celero/Console.h>
#include <celero/Distribution.h>
#include <celero/Exceptions.h>
#include <celero/Executor.h>
#include <celero/JUnit.h>
#include <celero/Print.h>
#include <celero/ResultTable.h>
#include <celero/TestVector.h>
#include <celero/UserDefinedMeasurement.h>
#include <celero/Utilities.h>
#include <cassert>
#include <cmath>
#include <fstream>
#include <iostream>
#include <set>

using namespace celero;

std::shared_ptr<celero::Benchmark> celero::RegisterTest(const char* groupName, const char* benchmarkName, const uint64_t samples,
														const uint64_t iterations, const uint64_t threads,
														std::shared_ptr<celero::Factory> experimentFactory, const double target)
{
	if(groupName != nullptr && benchmarkName != nullptr)
	{
		auto bm = celero::TestVector::Instance()[groupName];

		if(bm == nullptr)
		{
			bm = std::make_shared<Benchmark>(groupName);
			celero::TestVector::Instance().push_back(bm);
		}

		auto p = std::make_shared<Experiment>(bm);
		p->setIsBaselineCase(false);
		p->setName(benchmarkName);
		p->setSamples(samples);
		p->setIterations(iterations);
		p->setThreads(threads);
		p->setFactory(experimentFactory);
		p->setBaselineTarget(target);

		bm->addExperiment(p);

		return bm;
	}

	return nullptr;
}

std::shared_ptr<celero::Benchmark> celero::RegisterBaseline(const char* groupName, const char* benchmarkName, const uint64_t samples,
															const uint64_t iterations, const uint64_t threads,
															std::shared_ptr<celero::Factory> experimentFactory)
{
	if(groupName != nullptr && benchmarkName != nullptr)
	{
		auto bm = celero::TestVector::Instance()[groupName];

		if(bm == nullptr)
		{
			bm = std::make_shared<Benchmark>(groupName);
			celero::TestVector::Instance().push_back(bm);
		}

		auto p = std::make_shared<Experiment>(bm);
		p->setIsBaselineCase(true);
		p->setName(benchmarkName);
		p->setSamples(samples);
		p->setIterations(iterations);
		p->setThreads(threads);
		p->setFactory(experimentFactory);
		p->setBaselineTarget(1.0);

		bm->setBaseline(p);

		return bm;
	}

	return nullptr;
}

void celero::Run(int argc, char** argv)
{
#ifdef _DEBUG
	std::cout << "Celero is running in Debug.  Results are for debugging only as any measurements made while in Debug are likely not representative "
				 "of non-debug results."
			  << std::endl
			  << std::endl;
#endif

	cmdline::parser args;
	args.add("list", 'l', "Prints a list of all available benchmarks.");
	args.add<std::string>("group", 'g', "Runs a specific group of benchmarks.", false, "");
	args.add<std::string>("outputTable", 't', "Saves a results table to the named file.", false, "");
	args.add<std::string>("junit", 'j', "Saves a JUnit XML-formatted file to the named file.", false, "");
	args.add<std::string>("archive", 'a', "Saves or updates a result archive file.", false, "");
	args.add<uint64_t>("distribution", 'd', "Builds a file to help characterize the distribution of measurements and exits.", false, 0);
	args.add<bool>("catchExceptions", 'e', "Allows Celero to catch exceptions and continue processing following benchmarks.", false, true);
	args.parse_check(argc, argv);

	if(args.exist("list") == true)
	{
		auto& tests = celero::TestVector::Instance();
		std::vector<std::string> testNames;

		for(auto i = size_t(0); i < tests.size(); i++)
		{
			auto bm = celero::TestVector::Instance()[i];
			testNames.push_back(bm->getName());
		}

		std::sort(std::begin(testNames), std::end(testNames));

		std::cout << "Avaliable tests:" << std::endl;
		for(auto i : testNames)
		{
			std::cout << "\t" << i << std::endl;
		}

		return;
	}

	// Initial output
	std::cout << "Celero" << std::endl;

	// Disable dynamic CPU frequency scaling
	celero::timer::CachePerformanceFrequency(false);

	// Shall we build a distribution?
	auto intArgument = args.get<uint64_t>("distribution");
	if(intArgument > 0)
	{
		RunDistribution(intArgument);
	}

	// Has a result output file been specified?
	auto mustCloseFile = false;
	auto argument = args.get<std::string>("outputTable");
	if(argument.empty() == false)
	{
		std::cout << "Writing results to: " << argument << std::endl;
		celero::ResultTable::Instance().setFileName(argument);

		celero::AddExperimentResultCompleteFunction([](std::shared_ptr<celero::ExperimentResult> p) { celero::ResultTable::Instance().add(p); });
		mustCloseFile = true;
	}

	// Has a result output file been specified?
	argument = args.get<std::string>("archive");
	if(argument.empty() == false)
	{
		std::cout << "Archiving results to: " << argument << std::endl;
		celero::Archive::Instance().setFileName(argument);

		celero::AddExperimentResultCompleteFunction([](std::shared_ptr<celero::ExperimentResult> p) { celero::Archive::Instance().add(p); });
	}

	// Has a JUnit output file been specified?
	argument = args.get<std::string>("junit");
	if(argument.empty() == false)
	{
		std::cout << "Writing JUnit results to: " << argument << std::endl;
		celero::JUnit::Instance().setFileName(argument);

		celero::AddExperimentResultCompleteFunction([](std::shared_ptr<celero::ExperimentResult> p) { celero::JUnit::Instance().add(p); });
	}

	// Has a flag to catch exceptions or not been specified?
	if(args.exist("catchExceptions") == true)
	{
		ExceptionSettings::SetCatchExceptions(args.get<bool>("catchExceptions"));
	}

	// Has a run group been specified?
	argument = args.get<std::string>("group");

	// Collect all user-defined fields
	std::set<std::string> userDefinedFields;
	auto collectFromBenchmark = [&](std::shared_ptr<Benchmark> bmark) {
		// Collect from baseline
		auto baselineExperiment = bmark->getBaseline();
		if(baselineExperiment != nullptr)
		{
			auto test = baselineExperiment->getFactory()->Create();
			UserDefinedMeasurementCollector udmCollector(test);
			for(const auto& fieldName : udmCollector.getFields(test))
			{
				userDefinedFields.insert(fieldName);
			}
		}

		// Collect from all experiments
		const auto experimentSize = bmark->getExperimentSize();

		for(size_t i = 0; i < experimentSize; i++)
		{
			auto e = bmark->getExperiment(i);
			assert(e != nullptr);

			auto test = baselineExperiment->getFactory()->Create();
			UserDefinedMeasurementCollector udmCollector(test);
			for(const auto& fieldName : udmCollector.getFields(test))
			{
				userDefinedFields.insert(fieldName);
			}
		}
	};

	if(argument.empty() == false)
	{
		auto bmark = celero::TestVector::Instance()[argument];
		collectFromBenchmark(bmark);
	}
	else
	{
		for(size_t i = 0; i < celero::TestVector::Instance().size(); i++)
		{
			auto bmark = celero::TestVector::Instance()[i];
			collectFromBenchmark(bmark);
		}
	}

	std::vector<std::string> userDefinedFieldsOrder(userDefinedFields.begin(), userDefinedFields.end());

	Printer::get().initialize(userDefinedFieldsOrder);
	Printer::get().TableBanner();

	const auto startTime = celero::timer::GetSystemTime();

	if(argument.empty() == false)
	{
		executor::Run(argument);
	}
	else
	{
		executor::RunAll();
	}

	const auto totalTime = celero::timer::ConvertSystemTime(celero::timer::GetSystemTime() - startTime);

	if(mustCloseFile == true)
	{
		celero::ResultTable::Instance().closeFile();
	}

	// Final output.
	auto hours = std::to_string(static_cast<int>(totalTime) / 3600);
	auto minutes = std::to_string((static_cast<int>(totalTime) % 3600) / 60);
	auto seconds = std::to_string(fmod(totalTime, 60.0));

	if(hours.length() < 2)
	{
		hours = std::string("0") + hours;
	}

	if(minutes.length() < 2)
	{
		minutes = std::string("0") + minutes;
	}

	if(fmod(totalTime, 60.0) < 10.0)
	{
		seconds = std::string("0") + seconds;
	}

	std::cout << "Completed in " << hours << ":" << minutes << ":" << seconds << std::endl;
}