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

//

//////////////////////////////////////////////////////////////////////

#include <math.h></math.h>

#include <stdlib.h></stdlib.h>

#include <string.h></string.h>

#include <search.h></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

				cil.ci[cil.nb]=begin;

				cil.nb++;

			}

		} else if ( begin<0 && end>=0 ) {

			if ( cil.nb

				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)

	//		return false;

	return true;

}

void CYOMBInputParam::print(COLOR_INDEX_LIST &cil)

{

	/*	tmsg_info("nb=%d ci=",cil.nb);

	for( int i=0; i

		tmsg_info("i=%d %d",i,(int)cil.ci[i]);*/

}