#ifndef _PDF_READER_FUNCTION_H
#define _PDF_READER_FUNCTION_H

#include "Object.h"

namespace PdfReader
{
	class Dict;
	class Stream;
	struct PSObject;
	class PSStack;

	//------------------------------------------------------------------------
	// Function
	//------------------------------------------------------------------------

#define funcMaxInputs        32
#define funcMaxOutputs       32
#define sampledFuncMaxInputs 16

	class Function
	{
	public:

		Function();

		virtual ~Function();

		static Function *Parse(Object *pFuncObject);

		// Заполняем Domain и Range
		bool Initialize(Dict *pDict);

		virtual Function *Copy() = 0;

		// Типы функций
		//   -1 : identity
		//    0 : sampled
		//    2 : exponential
		//    3 : stitching
		//    4 : PostScript
		virtual int GetType() = 0;

		int GetInputSize()
		{
			return m_nInputDim;
		}
		int GetOutputSize()
		{
			return m_nOutputDim;
		}

		double GetDomainMin(int nIndex)
		{
			return m_arrDomain[nIndex][0];
		}
		double GetDomainMax(int nIndex)
		{
			return m_arrDomain[nIndex][1];
		}
		double GetRangeMin(int nIndex)
		{
			return m_arrRange[nIndex][0];
		}
		double GetRangeMax(int nIndex)
		{
			return m_arrRange[nIndex][1];
		}
		bool   HasRange()
		{
			return m_bHasRange;
		}

		// Собственно, работа самой функции. pInput - набор входящих параметров, pOutput - выходящие значения.
		virtual void Transform(double *pInput, double *pOutput) = 0;

		virtual bool IsValid() = 0;

	protected:


		int    m_nInputDim;                   // Function: R^(InputDim) -> R^(OutputDim)
		int    m_nOutputDim;                  // 
		double m_arrDomain[funcMaxInputs][2]; // Минимум и максимум для каждой входящей переменной
		double m_arrRange[funcMaxOutputs][2]; // Минимум и максимум для каждого выходящего параметра
		bool   m_bHasRange;                   // Domain должен быть задан всегда, а Range может отсутствовать
	};

	//------------------------------------------------------------------------
	// IdentityFunction
	//------------------------------------------------------------------------

	class IdentityFunction : public Function
	{
	public:

		IdentityFunction();
		virtual ~IdentityFunction();
		virtual Function *Copy()
		{
			return new IdentityFunction();
		}
		virtual int GetType()
		{
			return -1;
		}
		virtual void Transform(double *pInput, double *pOutput);
		virtual bool IsValid()
		{
			return true;
		}

	private:
	};

	//------------------------------------------------------------------------
	// SampledFunction (Type0 Function)
	//------------------------------------------------------------------------

	// На самом деле это аппроксимационная функция. Здесь должны быть заданы 
	// точки и значения функции в данных точках, и тогда данная функция 
	// воспринимается как аппромаксионная функция по заданным значениям, причем
	// аппрокимация зависит от поля Order, 1 - линейная интерполяция, 
	// 3 - интерполяция.

	// Пока реализована только линейная интерполяция.
	class SampledFunction : public Function
	{
	public:

		SampledFunction(Object *pFuncObject, Dict *pDict);
		virtual ~SampledFunction();
		virtual Function *Copy()
		{
			return new SampledFunction(this);
		}
		virtual int GetType()
		{
			return 0;
		}
		virtual void Transform(double *pInput, double *pOutput);
		virtual bool IsValid()
		{
			return m_bValid;
		}


		int     GetSampleSize(int nIndex)
		{
			return m_arrSize[nIndex];
		}
		double  GetEncodeMin(int nIndex)
		{
			return m_arrEncoder[nIndex][0];
		}
		double  GetEncodeMax(int nIndex)
		{
			return m_arrEncoder[nIndex][1];
		}
		double  GetDecodeMin(int nIndex)
		{
			return m_arrDecoder[nIndex][0];
		}
		double  GetDecodeMax(int nIndex)
		{
			return m_arrDecoder[nIndex][1];
		}
		double *GetSamples()
		{
			return m_pSamples;
		}

	private:

		SampledFunction(SampledFunction *pFunc);

	private:

		int     m_arrSize[funcMaxInputs];        // Количество различных значений, которые может принимать данный параметр
		double  m_arrEncoder[funcMaxInputs][2];  // Минимум и максимум для Domain encoder
		double  m_arrDecoder[funcMaxOutputs][2]; // Минимум и максимум для Range decoder
		double  m_arrInputMult[funcMaxInputs];   // 
		int     m_arrIndexMult[funcMaxInputs];	 // 
		double *m_pSamples;                      // Собственно сами значения
		int     m_nSamplesCount;                 // Количество заданных значений
		double *m_pBuffer;                       // Буфер для функции преобразования

		bool    m_bValid;
	};

	//------------------------------------------------------------------------
	// ExponentialFunction
	//------------------------------------------------------------------------

	class ExponentialFunction : public Function
	{
	public:

		ExponentialFunction(Object *pFuncObject, Dict *pDict);
		virtual ~ExponentialFunction();
		virtual Function *Copy()
		{
			return new ExponentialFunction(this);
		}
		virtual int GetType()
		{
			return 2;
		}
		virtual void Transform(double *pInput, double *pOutput);
		virtual bool IsValid()
		{
			return m_bValid;
		}

		double *GetC0()
		{
			return m_arrC0;
		}
		double *GetC1()
		{
			return m_arrC1;
		}
		double  GetN()
		{
			return m_dN;
		}

	private:

		ExponentialFunction(ExponentialFunction *pFunc);

	private:

		double m_arrC0[funcMaxOutputs];
		double m_arrC1[funcMaxOutputs];
		double m_dN;

		bool   m_bValid;
	};

	//------------------------------------------------------------------------
	// StitchingFunction
	//------------------------------------------------------------------------

	// Кусочная функция. В массиве m_arrBounds задаются точки деления исходного
	// отрезка. Domain0 < Bounds0 < Bounds1 < ... < Bounds(K - 2) < Domain1.
	// i-ая функция задана на полуинтервале [ Bounds(i - 1), Bounds(i) ),
	// где Bounds(-1) = Domain0, Bound(k-1) = Domain1 (услованая запись, в массиве
	// m_arrBounds эти значения не входят).
	class StitchingFunction : public Function
	{
	public:

		StitchingFunction(Object *pFuncObject, Dict *pDict);
		virtual ~StitchingFunction();
		virtual Function *Copy()
		{
			return new StitchingFunction(this);
		}
		virtual int GetType()
		{
			return 3;
		}
		virtual void Transform(double *pInput, double *pOutput);
		virtual bool IsValid()
		{
			return m_bValid;
		}

		int GetFunctionsCount()
		{
			return m_nCount;
		}
		Function *GetFunction(int nIndex)
		{
			return m_ppFuncs[nIndex];
		}
		double *GetBounds()
		{
			return m_arrBounds;
		}
		double *GetEncode()
		{
			return m_arrEncode;
		}
		double *GetScale()
		{
			return m_arrScale;
		}

	private:

		StitchingFunction(StitchingFunction *pFunc);

	private:

		int        m_nCount;    // количество функций
		Function **m_ppFuncs;   // сами функции
		double    *m_arrBounds; // Границы областей определения функций
		double    *m_arrEncode; // Границы множств значений функций
		double    *m_arrScale;

		bool       m_bValid;
	};

	//------------------------------------------------------------------------
	// PostScriptFunction
	//------------------------------------------------------------------------

	class PostScriptFunction : public Function
	{
	public:

		PostScriptFunction(Object *pFuncObject, Dict *pDict);
		virtual ~PostScriptFunction();
		virtual Function *Copy()
		{
			return new PostScriptFunction(this);
		}
		virtual int GetType()
		{
			return 4;
		}
		virtual void Transform(double *pInput, double *pOutput);
		virtual bool IsValid()
		{
			return m_bValid;
		}
		StringExt *GetCodeString()
		{
			return m_seCodeString;
		}

	private:

		PostScriptFunction(PostScriptFunction *pFunc);
		bool ParseCode(Stream *pStream, int *pnCodePos);
		StringExt *GetToken(Stream *pStream);
		void ResizeCode(int nNewSize);
		void Exec(PSStack *pStack, int nCodePos);

	private:

		StringExt *m_seCodeString;
		PSObject  *m_pCode;
		int        m_nCodeSize;

		bool       m_bValid;
	};
}

#endif // _PDF_READER_FUNCTION_H