#ifdef WIN32
#include "Windows.h"
#endif
#include <vector>
 
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <sys/types.h>
#include <sys/timeb.h>
#include <string.h>
#include <math.h>
 
#include "Params.h"
#include "PatternMapParam.h"
#include "STColSelPic.h"
#include "BlurMatrix.h"
#include "SDirection.h"
#include "SDef.h"
#include "CallCircle.h"
#include "CallParam.h"
#include "EraseContour.h"
#include "PatternPosition.h"
#include "Pattern.h"
#include "SError.h"
//#include "tmsg.h"
#include "patternmap.h"
 
using namespace std;
 
#ifdef __cplusplus
 
extern "C" {
#endif
 
//#define P(d) tmsg_warning(" - %d -\n",d)
#define COPY_RASTER(iras, oras, border) tP.copy_raster(iras, oras, border, border, iras->lx - border - 1, iras->ly - border - 1, 0, 0)
 
#define DIRACCURACY 12
 
// ----- PatternMapping for UCHAR pixels (range 0-255) --------------------------
void patternmapUC(const RASTER *iras, RASTER *oras, CPatternMapParam &pmP,
				  const int border, RASTER *imgContour) // throw (SMemAllocError,SWriteRasterError,SFileReadError)
{
	try {
		SRECT rect = {border, border, iras->lx - border - 1, iras->ly - border - 1};
		SPOINT p = {0, 0};
 
		CSTColSelPic<UC_PIXEL> ipUC, ipOri;
		ipUC.read(iras);
		ipUC.initSel();
		ipOri = ipUC;
		// Selection of pixels using CM
		CCIL emptyCIL;
		int nbContourPixel = 0;
 
		if (pmP.m_isKeepContour) {
			CEraseContour ec(ipUC);
			nbContourPixel = ec.makeSelection(pmP.m_ink);
			//			nbContourPixel=ipUC.makeSelectionCMAP(pmP.m_ink,emptyCIL);
		} else {
			CEraseContour ec(ipUC);
			nbContourPixel = ec.doIt(pmP.m_ink);
		}
		if (nbContourPixel > 0) {
			if (!pmP.m_isRandomDir) {
				CSDirection dir(ipUC.m_lX, ipUC.m_lY, ipUC.m_sel, DIRACCURACY);
				dir.doDir();
				dir.getResult(ipUC.m_sel);
			}
			CPatternPosition pPos(ipUC, nbContourPixel, pmP.m_density, pmP.m_minDist, pmP.m_maxDist);
			// Reads the pattern
			CPattern pat(imgContour);
 
			for (vector<SPOINT>::iterator pp = pPos.m_pos.begin();
				 pp != pPos.m_pos.end();
				 pp++) {
 
				// Calculates the rotation angle
				double angle = 0.0;
				if (pmP.m_isRandomDir)
					angle = pmP.m_minDirAngle + (double)(rand() % 1001) * 0.001 * (pmP.m_maxDirAngle - pmP.m_minDirAngle);
				else {
					if (pp->x >= 0 && pp->y >= 0 && pp->x < ipUC.m_lX && pp->y < ipUC.m_lY) {
						UCHAR *sel = ipUC.m_sel + pp->y * ipUC.m_lX + pp->x;
						if (*sel > (UCHAR)0)
							angle = (double)(*sel) - 50.0;
					}
					angle += pmP.m_minDirAngle + (double)(rand() % 1001) * 0.001 * (pmP.m_maxDirAngle - pmP.m_minDirAngle);
				}
 
				// Calculates the scale
				double scale = 1.0;
				scale = pmP.m_minScale + (double)(rand() % 1001) * 0.001 * (pmP.m_maxScale - pmP.m_minScale);
 
				// Mapping of Pattern
				pat.mapIt(ipUC, ipOri, pp->x, pp->y, scale, angle, pmP.m_isUseInkColor, pmP.m_isIncludeAlpha);
			}
		}
 
		//		ipUC.showSelection();
 
		ipUC.write(oras, rect, p);
	} catch (SMemAllocError) {
		//SMemAllocError();
	} catch (SWriteRasterError) {
		//	SWriteRasterError();
	} catch (SFileReadError) {
		//	SFileReadError();
	}
}
 
// ----- PatternMapping for USHORT pixels (range 0-255) --------------------------
void patternmapUS(const RASTER *iras, RASTER *oras, CPatternMapParam &pmP,
				  const int border, RASTER *imgContour) // throw (SMemAllocError,SWriteRasterError,SFileReadError)
{
	try {
		SRECT rect = {border, border, iras->lx - border - 1, iras->ly - border - 1};
		SPOINT p = {0, 0};
 
		CSTColSelPic<US_PIXEL> ipUS, ipOri;
		ipUS.read(iras);
		ipUS.initSel();
		ipOri = ipUS;
		// Selection of pixels using CM
		CCIL emptyCIL;
		int nbContourPixel = 0;
 
		if (pmP.m_isKeepContour) {
			CEraseContour ec(ipUS);
			nbContourPixel = ec.makeSelection(pmP.m_ink);
			// a 255-nek utananezni !!!
			//			nbContourPixel=ipUS.makeSelectionCMAP(pmP.m_ink,emptyCIL);
		} else {
			CEraseContour ec(ipUS);
			nbContourPixel = ec.doIt(pmP.m_ink);
		}
		if (nbContourPixel > 0) {
			if (!pmP.m_isRandomDir) {
				CSDirection dir(ipUS.m_lX, ipUS.m_lY, ipUS.m_sel, DIRACCURACY);
				dir.doDir();
				dir.getResult(ipUS.m_sel);
			}
			CPatternPosition pPos(ipUS, nbContourPixel, pmP.m_density, pmP.m_minDist, pmP.m_maxDist);
			// Reads the pattern
			CPattern pat(imgContour);
 
			for (vector<SPOINT>::iterator pp = pPos.m_pos.begin();
				 pp != pPos.m_pos.end();
				 pp++) {
 
				// Calculates the rotation angle
				double angle = 0.0;
				if (pmP.m_isRandomDir)
					angle = pmP.m_minDirAngle + (double)(rand() % 1001) * 0.001 * (pmP.m_maxDirAngle - pmP.m_minDirAngle);
				else {
					if (pp->x >= 0 && pp->y >= 0 && pp->x < ipUS.m_lX && pp->y < ipUS.m_lY) {
						UCHAR *sel = ipUS.m_sel + pp->y * ipUS.m_lX + pp->x;
						if (*sel > (UCHAR)0)
							angle = (double)(*sel) - 50.0;
					}
					angle += pmP.m_minDirAngle + (double)(rand() % 1001) * 0.001 * (pmP.m_maxDirAngle - pmP.m_minDirAngle);
				}
 
				// Calculates the scale
				double scale = 1.0;
				scale = pmP.m_minScale + (double)(rand() % 1001) * 0.001 * (pmP.m_maxScale - pmP.m_minScale);
 
				// Mapping of Pattern
				pat.mapIt(ipUS, ipOri, pp->x, pp->y, scale, angle, pmP.m_isUseInkColor, pmP.m_isIncludeAlpha);
			}
		}
		//		ipUC.showSelection();
		ipUS.write(oras, rect, p);
	} catch (SMemAllocError) {
		//SMemAllocError();
	} catch (SWriteRasterError) {
		//	SWriteRasterError();
	} catch (SFileReadError) {
		//	SFileReadError();
	}
}
 
int patternmap(const RASTER *iras, RASTER *oras, const int border,
			   int argc, const char *argv[], const int shrink, RASTER *imgContour)
{
	// The input raster must be RAS_CM16 or RAS_CM24!!!!
	CSTPic<UC_PIXEL> tP;
	if (iras->type != RAS_CM32) {
		COPY_RASTER(iras, oras, border);
		return 0;
	}
	try {
		CPatternMapParam pmP(argc, argv, shrink);
		if (!pmP.isOK()) {
			COPY_RASTER(iras, oras, border);
			return 0;
		}
		srand(10);
		if (oras->type == RAS_RGBM) {
			patternmapUC(iras, oras, pmP, border, imgContour);
		} else if (oras->type == RAS_RGBM64) {
			patternmapUS(iras, oras, pmP, border, imgContour);
		} else {
			COPY_RASTER(iras, oras, border);
			return 0;
		}
	} catch (SMemAllocError e) {
		e.debug_print();
		COPY_RASTER(iras, oras, border);
		return 0;
	} catch (SWriteRasterError e) {
		e.debug_print();
		COPY_RASTER(iras, oras, border);
		return 0;
	} catch (SFileReadError e) {
		e.debug_print();
		COPY_RASTER(iras, oras, border);
		return 0;
	}
 
	return 1;
}
 
#ifdef __cplusplus
}
#endif