/**
 * ParamTable.cc
 *
 *  Created on: 10 oct. 2014
 *  Author: AKKA
 */
#include "ParamTable.hh"

#include "Parameter.hh"
#include "ParamData.hh"
#include "ParamInfo.hh"
#include "ParamMgr.hh"

using namespace log4cxx;

namespace AMDA {
namespace Info {

LoggerPtr ParamTable::_logger(Logger::getLogger("AMDA-Kernel.ParamTable"));

ParamTable::ParamTable(const char *paramId) : _paramId(paramId),
		_tableName(""), _tableUnits(""), _variable("false")
{
}

ParamTable::~ParamTable()
{
}

std::string ParamTable::getTableParamKeyForInfo() {
	return "";
}

/*
 * @brief Get param id
 */
std::string ParamTable::getParamId(void)
{
	return _paramId;
}

/*
 * @brief Set name of the table
 */
void ParamTable::setName(std::string name)
{
	_tableName = name;
}

/*
 * @brief Get name of the table
 */
std::string ParamTable::getName(void)
{
	if (_tableName.empty() && _variable) {
		std::string paramKeyForInfo = getTableParamKeyForInfo();
		if (!paramKeyForInfo.empty()) {
			std::string tableParam = getTableParamByKey(paramKeyForInfo);
			AMDA::Info::ParamInfoSPtr paramTableInfo = AMDA::Info::ParamMgr::getInstance()->getParamInfoFromId(tableParam);
			if (paramTableInfo != nullptr)
				return paramTableInfo->getName();
		}
	}

	return _tableName;
}

/*
 * @brief Set units
 */
void ParamTable::setUnits(std::string units)
{
	_tableUnits = units;
}

/*
 * @brief Get units of the table
 */
std::string ParamTable::getUnits(void)
{
	if (_tableName.empty() && _variable) {
		std::string paramKeyForInfo = getTableParamKeyForInfo();
		if (!paramKeyForInfo.empty()) {
			std::string tableParam = getTableParamByKey(paramKeyForInfo);
			AMDA::Info::ParamInfoSPtr paramTableInfo = AMDA::Info::ParamMgr::getInstance()->getParamInfoFromId(tableParam);
			if (paramTableInfo != nullptr)
				return paramTableInfo->getUnits();
		}
	}

	return _tableUnits;
}

/*
 * @brief Set isVariable attribute
 */
void ParamTable::setIsVariable(bool isVariable)
{
	_variable = isVariable;
}

/*
 * @brief Get if it's a variable table
 */
bool ParamTable::isVariable(void)
{
	return _variable;
}

void ParamTable::addTableParam(std::string key, std::string name)
{
	_tableParams[key] = name;
}

std::string ParamTable::getTableParamByKey(std::string key)
{
	return _tableParams[key];
}

std::map<std::string, std::string>& ParamTable::getTableParams()
{
	return _tableParams;
}

std::vector<double> ParamTable::getConstantTableParamValuesByKey(ParameterManager *parameterManager, std::string key)
{
	std::vector<double> paramValues;
	ParameterSPtr tmpParam = parameterManager->getParameter(_paramId);
	if (tmpParam == nullptr)
		return paramValues;
	AMDA::Parameters::Parameter::InfoList lInfoList = tmpParam->getInfoList();
	if (_tableParams.find(key) == _tableParams.end())
		return paramValues;
	std::string tableParamName = _tableParams[key];
	if (lInfoList.find(tableParamName) == lInfoList.end())
		return paramValues;
	paramValues = *lInfoList[tableParamName].get();
	return paramValues;
}

std::vector<double> ParamTable::getVariableTableParamValuesByKey(std::map<std::string, std::vector<double>>* paramsTableData, std::string key)
{
	std::vector<double> paramValues;
	if (paramsTableData == NULL)
		return paramValues;
	if (paramsTableData->find(key) == paramsTableData->end())
		return paramValues;
	return (*paramsTableData)[key];
}

std::vector<double> ParamTable::getTableParamValuesByKey(ParameterManager *parameterManager, std::map<std::string, std::vector<double>>* paramsTableData, std::string key)
{
	if (_variable)
		return getVariableTableParamValuesByKey(paramsTableData, key);
	else
		return getConstantTableParamValuesByKey(parameterManager, key);
}

std::string ParamBoundsTable::_boundsParamKey = "TABLEPARAM_BOUNDS";

ParamBoundsTable::ParamBoundsTable(const char *paramId) :
		ParamTable(paramId)
{
}

std::string ParamBoundsTable::getTableParamKeyForInfo() {
	return ParamBoundsTable::_boundsParamKey;
}

/*
 * @brief Get size of the table
 */
int ParamBoundsTable::getSize(ParameterManager *parameterManager)
{
	if (!_variable)
	{
		std::vector<double> boundsValues = getConstantTableParamValuesByKey(parameterManager, ParamBoundsTable::_boundsParamKey);
		return boundsValues.size() - 1;
	}
	else
	{
		return 0;
	}
}

/*
 * @brief Get a bounds for a specified index
 */
t_TableBound ParamBoundsTable::getBound(ParameterManager *parameterManager, unsigned int index, std::map<std::string, std::vector<double>>* paramsTableData)
{
	t_TableBound bound;
	bound.index = index;

	std::vector<double> boundsValues = getTableParamValuesByKey(parameterManager, paramsTableData, ParamBoundsTable::_boundsParamKey);

	if (boundsValues.empty() || (index >= boundsValues.size() - 1))
	{
		LOG4CXX_ERROR(_logger, "Index " << index << " outside of table definition => Cannot get real bound" );
		bound.min = index;
		bound.max = index+1;
		return bound;
	}
	bound.min = std::min(boundsValues[index], boundsValues[index+1]);
	bound.max = std::max(boundsValues[index], boundsValues[index+1]);

	return bound;
}

std::string ParamCenterTable::_centersParamKey = "TABLEPARAM_CENTERS";

ParamCenterTable::ParamCenterTable(const char *paramId, double size) :
		ParamTable(paramId), _size(size)
{
}

std::string ParamCenterTable::getTableParamKeyForInfo() {
	return ParamCenterTable::_centersParamKey;
}

/*
 * @brief Get size of the table
 */
int ParamCenterTable::getSize(ParameterManager *parameterManager)
{
	if (!_variable)
	{
		std::vector<double> centersValues = getConstantTableParamValuesByKey(parameterManager, ParamCenterTable::_centersParamKey);
		return centersValues.size();
	}
	else
	{
		return 0;
	}
}

/*
 * @brief Get a bound for a specified index
 */
t_TableBound ParamCenterTable::getBound(ParameterManager *parameterManager, unsigned int index, std::map<std::string, std::vector<double>>* paramsTableData)
{
	t_TableBound bound;
	bound.index = index;

	std::vector<double> centersValues = getTableParamValuesByKey(parameterManager, paramsTableData, ParamCenterTable::_centersParamKey);

	if (index >= centersValues.size())
	{
		LOG4CXX_ERROR(_logger, "Index " << index << " outside of table definition => Cannot get real bound" );
		bound.min = index;
		bound.max = index+1;
		return bound;
	}

	if (!std::isnan(centersValues[index]))
	{
		bound.min = centersValues[index] - _size/2.;
		bound.max = centersValues[index] + _size/2.;
	}
	else
	{
		bound.min = NAN;
		bound.max = NAN;
	}

	return bound;
}

std::string ParamCenterWidthTable::_centersParamKey = "TABLEPARAM_CENTERS";
std::string ParamCenterWidthTable::_widthsParamKey = "TABLEPARAM_WIDTHS";

ParamCenterWidthTable::ParamCenterWidthTable(const char *paramId) :
		ParamTable(paramId)
{
}

std::string ParamCenterWidthTable::getTableParamKeyForInfo() {
	return ParamCenterWidthTable::_centersParamKey;
}

/*
 * @brief Get size of the table
 */
int ParamCenterWidthTable::getSize(ParameterManager *parameterManager)
{
	if (!_variable)
	{
		std::vector<double> centersValues = getConstantTableParamValuesByKey(parameterManager, ParamCenterWidthTable::_centersParamKey);
		return centersValues.size();
	}
	else
	{
		return 0;
	}
}

/*
 * @brief Get a bound for a specified index
 */
t_TableBound ParamCenterWidthTable::getBound(ParameterManager *parameterManager, unsigned int index, std::map<std::string, std::vector<double>>* paramsTableData)
{
	t_TableBound bound;
	bound.index = index;

	std::vector<double> centersValues = getTableParamValuesByKey(parameterManager, paramsTableData, ParamCenterWidthTable::_centersParamKey);
	std::vector<double> widthsValues = getTableParamValuesByKey(parameterManager, paramsTableData, ParamCenterWidthTable::_widthsParamKey);

	if ((index >= centersValues.size()) || (index >= widthsValues.size()))
	{
		LOG4CXX_ERROR(_logger, "Index " << index << " outside of table definition => Cannot get real bound" );
		bound.min = index;
		bound.max = index+1;
		return bound;
	}

	if (!std::isnan(centersValues[index]) && !std::isnan(widthsValues[index]))
	{
		bound.min = centersValues[index] - widthsValues[index]/2.;
		bound.max = centersValues[index] + widthsValues[index]/2.;
	}
	else
	{
		bound.min = NAN;
		bound.max = NAN;
	}

	return bound;
}

std::string ParamCenterAutoTable::_centersParamKey = "TABLEPARAM_CENTERS";

ParamCenterAutoTable::ParamCenterAutoTable(const char *paramId, bool log) :
		ParamTable(paramId), _log(log)
{
}

std::string ParamCenterAutoTable::getTableParamKeyForInfo() {
	return ParamCenterAutoTable::_centersParamKey;
}

/*
 * @brief Get size of the table
 */
int ParamCenterAutoTable::getSize(ParameterManager *parameterManager)
{
	if (!_variable)
	{
		std::vector<double> centersValues = getConstantTableParamValuesByKey(parameterManager, ParamCenterAutoTable::_centersParamKey);
		return centersValues.size();
	}
	else
	{
		return 0;
	}
}

/*
 * @brief Get a bound for a specified index
 */
t_TableBound ParamCenterAutoTable::getBound(ParameterManager *parameterManager, unsigned int index, std::map<std::string, std::vector<double>>* paramsTableData)
{
	t_TableBound bound;
	bound.index = index;

	std::vector<double> centersValues = getTableParamValuesByKey(parameterManager, paramsTableData, ParamCenterAutoTable::_centersParamKey);

	if (index >= centersValues.size())
	{
		LOG4CXX_ERROR(_logger, "Index " << index << " outside of table definition => Cannot get real bound" );
		bound.min = index;
		bound.max = index+1;
		return bound;
	}

	//Compute bounds

	if (centersValues.size() <= 1)
	{
		LOG4CXX_ERROR(_logger, "Table dimension too small to compute bound" );
		bound.min = index;
		bound.max = index+1;
		return bound;
	}

	double minus = 0.;
	double plus  = 0.;

	if (index == 0)
	{
		if (!std::isnan(centersValues[1]) && !std::isnan(centersValues[0]))
		{
			if (_log)
				plus  = (log10(centersValues[1]) - log10(centersValues[0]))/2.;
			else
				plus  = (centersValues[1] - centersValues[0])/2.;
		}
		else
			plus = NAN;
		minus = plus;
	}
	else if (index == centersValues.size() - 1)
	{
		if (!std::isnan(centersValues[centersValues.size() - 1]) && !std::isnan(centersValues[centersValues.size() - 2]))
		{
			if (_log)
				minus = (log10(centersValues[centersValues.size() - 1]) - log10(centersValues[centersValues.size() - 2]))/2.;
			else
				minus = (centersValues[centersValues.size() - 1] - centersValues[centersValues.size() - 2])/2.;
		}
		else
			minus = NAN;
		plus  = minus;

	}
	else
	{
		if (_log)
		{
			if (!std::isnan(centersValues[index]) && !std::isnan(centersValues[index-1]) && (centersValues[index] > 0) && (centersValues[index-1] > 0))
				minus = (log10(centersValues[index]) - log10(centersValues[index-1]))/2.;
			else
				minus = NAN;
			if (!std::isnan(centersValues[index+1]) && !std::isnan(centersValues[index]) && (centersValues[index+1] > 0) && (centersValues[index] > 0))
				plus  = (log10(centersValues[index+1]) - log10(centersValues[index]))/2.;
			else
				plus = NAN;
		}
		else
		{
			if (!std::isnan(centersValues[index]) && !std::isnan(centersValues[index-1]))
				minus = (centersValues[index] - centersValues[index-1])/2.;
			else
				minus = NAN;
			if (!std::isnan(centersValues[index+1]) && !std::isnan(centersValues[index]))
				plus  = (centersValues[index+1] - centersValues[index])/2.;
			else
				plus = NAN;
		}
	}

	if (_log)
	{
		if (!std::isnan(minus))
			bound.min = exp(log10(centersValues[index]) - minus);
		else
			bound.min = NAN;
		if (!std::isnan(plus))
			bound.max = exp(log10(centersValues[index]) + plus);
		else
			bound.max = NAN;
	}
	else
	{
		if (!std::isnan(minus))
			bound.min = centersValues[index] - minus;
		else
			bound.min = NAN;
		if (!std::isnan(plus))
			bound.max = centersValues[index] + plus;
		else
			bound.max = NAN;
	}

	if (bound.min > bound.max)
	{
		double temp = bound.max;
		bound.max = bound.min;
		bound.min = temp;
	}

	return bound;
}

std::string ParamMinMaxTable::_minParamKey = "TABLEPARAM_MIN";
std::string ParamMinMaxTable::_maxParamKey = "TABLEPARAM_MAX";

ParamMinMaxTable::ParamMinMaxTable(const char *paramId) : ParamTable(paramId)
{
}

/*
 * @brief Get size of the table
 */
int ParamMinMaxTable::getSize(ParameterManager *parameterManager)
{
	if (!_variable)
	{
		std::vector<double> minValues = getConstantTableParamValuesByKey(parameterManager, ParamMinMaxTable::_minParamKey);
		return minValues.size();
	}
	else
	{
		return 0;
	}
}

/*
 * @brief Get bound for a specified index
 */
t_TableBound ParamMinMaxTable::getBound(ParameterManager *parameterManager, unsigned int index, std::map<std::string, std::vector<double>>* paramsTableData)
{
	t_TableBound bound;
	bound.index = index;

	std::vector<double> minValues = getTableParamValuesByKey(parameterManager, paramsTableData, ParamMinMaxTable::_minParamKey);
	std::vector<double> maxValues = getTableParamValuesByKey(parameterManager, paramsTableData, ParamMinMaxTable::_maxParamKey);

	if ((index >= minValues.size()) || (index >= maxValues.size()))
	{
		LOG4CXX_ERROR(_logger, "Index " << index << " outside of table definition => Cannot get real bound" );
		bound.min = index;
		bound.max = index+1;
		return bound;
	}

	bound.min = std::min(minValues[index], maxValues[index]);
	bound.max = std::max(minValues[index], maxValues[index]);

	return bound;
}

} /* namespace Info */
} /* namespace AMDA */