// YOMBInputParam.cpp: implementation of the CYOMBInputParam class.
//
//////////////////////////////////////////////////////////////////////

#include <math.h>
#include <stdlib.h>
#include <string.h>
#include <search.h>
#include "YOMBInputParam.h"
//#include "tmsg.h"
#ifdef _WIN32
#pragma warning(disable : 4996)
#endif
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

bool CYOMBInputParam::isRange(const char *s) const
{
	for (int i = 0; i < (int)strlen(s); i++)
		if (s[i] == '-')
			return true;
	return false;
}

int CYOMBInputParam::getRangeBegin(const char *s) const
{
	char ss[100];

	strcpy(ss, s);
	for (int i = 0; i < (int)strlen(ss); i++)
		if (ss[i] == '-') {
			ss[i] = '\0';
			break;
		}
	if (strlen(ss) == 0)
		return -1;
	return atoi(ss);
}

int CYOMBInputParam::getRangeEnd(const char *s) const
{
	char ss[100];
	int i = 0;
	for (i = strlen(s) - 1; i >= 0; i--)
		if (s[i] == '-')
			break;
	strcpy(ss, &s[i + 1]);
	if (strlen(ss) == 0)
		return -1;
	return atoi(ss);
}

void CYOMBInputParam::strToColorIndex(const char *s, COLOR_INDEX_LIST &cil,
									  const int maxIndex)
{
	/* If s=="-1", all of the indices are put into the ColorIndexList
*/
	if (strcmp(s, "-1") == 0) {
		for (int i = 0; i <= maxIndex; i++)
			if (cil.nb < MAXNBCIL) {
				cil.ci[cil.nb] = (unsigned short)i;
				cil.nb++;
			} else
				return;
		return;
	}

	if (isRange(s)) {
		int begin = getRangeBegin(s);
		int end = getRangeEnd(s);
		if (begin >= 0 && end >= 0) {
			begin = min(begin, maxIndex);
			end = min(end, maxIndex);
			for (int i = min(begin, end); i <= max(begin, end) && cil.nb < MAXNBCIL; i++)
				cil.ci[cil.nb++] = (unsigned short)i;
		}
		// If there is no 'begin' or 'end' of range, it is considered
		// 'bad parameter'. For example: '-3' or '5-'.
		/*		else if ( begin>=0 && end<0 ) {
			if ( cil.nb<MAXNBCIL ) {
				cil.ci[cil.nb]=begin;
				cil.nb++;
			}
		} else if ( begin<0 && end>=0 ) {
			if ( cil.nb<MAXNBCIL ) {
				cil.ci[cil.nb]=end;
				cil.nb++;
			}
		}
*/
	} else {
		if (cil.nb < MAXNBCIL) {
			int q = atoi(s);
			if (q >= 0 && q <= maxIndex)
				cil.ci[cil.nb++] = (unsigned short)q;
		}
	}
}

int ushortCompare(const void *a, const void *b)
{
	unsigned short *aa;
	unsigned short *bb;
	aa = (unsigned short *)a;
	bb = (unsigned short *)b;
	if (*aa == *bb)
		return 0;
	if (*aa < *bb)
		return -1;
	return 1;
}

void CYOMBInputParam::makeColorIndexList(const char *s, COLOR_INDEX_LIST &cil,
										 const int maxIndex)
{
	int ln = strlen(s);
	int i, j;
	char ss[100];
	COLOR_INDEX_LIST tmpCil;

	cil.nb = tmpCil.nb = 0;
	for (i = 0; i < ln; i++) {
		if (s[i] != ',') {
			strcpy(ss, &s[i]);
			for (j = 0; j < (int)strlen(ss); j++)
				if (ss[j] == ',') {
					ss[j] = '\0';
					break;
				}
			i += (strlen(ss) - 1);
			strToColorIndex(ss, tmpCil, maxIndex);
		}
	}
	qsort(tmpCil.ci, tmpCil.nb, sizeof(unsigned short), ushortCompare);

	if (tmpCil.nb > 0) {
		cil.ci[0] = tmpCil.ci[0];
		cil.nb = 1;
		for (i = j = 1; i < tmpCil.nb; i++)
			if (tmpCil.ci[i] != tmpCil.ci[i - 1])
				cil.ci[cil.nb++] = tmpCil.ci[i];
	} else
		cil.nb = 0;
}

// Constructor of RGBM input parameters.
// It supposes 30 parameters:
//		0,1,2,3,4		isColor4, color4 (RGBM)
//		5,6,7,8,9		isColor3, color3 (RGBM)
//		10,11,12,13,14		isColor2, color2 (RGBM)
//		15,16,17,18,19		isColor1, color1 (RGBM)
//		20,21,22,23,24		isColor0, color0 (RGBM)
//		25. sensibility of selection (dAB)
//		26. nbSamples (Quality). The number of sampling pixels.
//		27. dSamples (Amount). The distance of sampling.
//		28. isStopAtContour (Y/N) Stops the sampling at contour line.
//		29. isRandomSampling (Y/N) Random or equal sampling.
CYOMBInputParam::CYOMBInputParam(const int argc, const char *argv[],
								 const int shrink)
{
	m_nbColor = m_nbSample = 0;
	m_dSample = 0.0;
	m_scale = shrink > 0 ? 1.0 / (double)shrink : 1.0;
	m_isEconf = false;
	m_isShowSelection = false;
	if (argc == 30) {
		m_isCM = false;
		m_isRandomSampling = argv[29][0] == '0' ? false : true;
		m_isStopAtContour = argv[28][0] == '0' ? false : true;
		// Amount - dSample
		m_dSample = atof(argv[27]) * m_scale;
		// Quality - nbSamples
		m_nbSample = (int)atof(argv[26]);
		if (shrink > 1) {
			//			if ( m_nbSample>2 ) {
			//				m_nbSample= (int)((double)m_nbSample*sqrt(m_scale)+0.5);
			//				m_nbSample= m_nbSample<3 ? 3 : m_nbSample;
			//			}
		} else
			m_nbSample = m_nbSample < 1 ? 1 : m_nbSample;
		if ((int)(m_dSample * m_dSample * 2.5) < m_nbSample)
			m_nbSample = (int)(m_dSample * m_dSample * 2.5);

		// Sensibility - dist A
		m_dA = 3.0 * atof(argv[25]) / 50.0;
		m_dA = m_dA <= 0.001 ? 0.001 : m_dA;
		// Sensibility - dist AB
		m_dAB = atof(argv[25]) / 50.0;
		m_dAB = m_dAB <= 0.001 ? 0.001 : m_dAB;

		int j = 24;
		m_nbColor = 0;
		for (int i = 0; i < 5; i++) {
			if (atoi(argv[j--]) > 0) {
				m_color[m_nbColor][0] = atoi(argv[j--]);
				m_color[m_nbColor][1] = atoi(argv[j--]);
				m_color[m_nbColor][2] = atoi(argv[j--]);
				m_color[m_nbColor][3] = atoi(argv[j--]);
				m_nbColor++;
			} else
				j -= 4;
		}
	}
}

// Constructor of RGBM input parameters.
// It supposes 6 parameters:
//		0. Paint index list
//		1. Ink index list
//		2. nbSamples (Quality). The number of sampling pixels.
//		3. dSamples (Amount). The distance of sampling.
//		4. isStopAtContour (Y/N) Stops the sampling at contour line.
//		5. isRandomSampling (Y/N) Random or equal sampling.
// isCM16 - informs about the type of colormap CM16/CM24.
CYOMBInputParam::CYOMBInputParam(const int argc, const char *argv[],
								 const int shrink, const bool isCM16)
{
	int nbInk = isCM16 ? 31 : 4095;
	int nbPaint = isCM16 ? 127 : 4095;

	m_nbColor = m_nbSample = 0;
	m_dSample = 0.0;
	m_scale = shrink > 0 ? 1.0 / (double)shrink : 1.0;
	m_isEconf = false;
	m_isShowSelection = false;
	m_dA = m_dAB = 0.01;
	if (argc == 6) {
		m_isCM = true;
		m_isRandomSampling = argv[5][0] == '0' ? false : true;
		m_isStopAtContour = argv[4][0] == '0' ? false : true;
		// Amount - dSample
		m_dSample = atof(argv[3]) * m_scale;
		// Quality - nbSamples
		m_nbSample = (int)atof(argv[2]);
		if (shrink > 1) {
			if (m_nbSample > 2) {
				m_nbSample = (int)((double)m_nbSample * sqrt(m_scale) + 0.5);
				m_nbSample = m_nbSample < 2 ? 2 : m_nbSample;
			}
		} else
			m_nbSample = m_nbSample < 1 ? 1 : m_nbSample;
		if ((int)(m_dSample * m_dSample * 2.5) < m_nbSample)
			m_nbSample = (int)(m_dSample * m_dSample * 2.5);

		// Preparing of INK and PAINT index lists
		makeColorIndexList(&argv[1][0], m_ink, nbInk);
		makeColorIndexList(&argv[0][0], m_paint, nbPaint);
	}
}

bool CYOMBInputParam::isOK()
{
	if (!m_isCM && m_nbColor <= 1)
		return false;
	if (m_isCM && m_paint.nb <= 1)
		return false;
	if (m_dSample < 1.1)
		return false;
	if (m_nbSample <= 1)
		return false;
	//	if ( (int)(m_dSample*m_dSample*3.1)<m_nbSample)
	//		return false;

	return true;
}

void CYOMBInputParam::print(COLOR_INDEX_LIST &cil)
{
	/*	tmsg_info("nb=%d ci=",cil.nb);
	for( int i=0; i<cil.nb; i++ )
		tmsg_info("i=%d %d",i,(int)cil.ci[i]);*/
}