#include "tscannerutil.h"

#include <cstring>

#define BUFBYTE(X, Y, BUF, BYTEWRAP, BITOFFS) \
	(((UCHAR *)(BUF))[(((X) + (BITOFFS)) >> 3) + (Y) * (BYTEWRAP)])

#define GET_BIT(X, Y, BUF, BYTEWRAP, BITOFFS) \
	((BUFBYTE(X, Y, BUF, BYTEWRAP, BITOFFS) >> (7 - (((X) + (BITOFFS)) & 7))) & (UCHAR)1)

namespace TScannerUtil
{

struct EP {
	unsigned char r, g, b;
};

//-----------------------------------------------------------------------------

void copyRGBBufferToTRaster32(unsigned char *rgbBuffer,
							  int rgbLx, int rgbLy,
							  const TRaster32P &rout, bool internal)
{
	if (internal) {
		TPixelRGBM32 *dst = rout->pixels();
		EP *src = (EP *)(rgbBuffer + (rgbLx * rgbLy - 1) * 3);
		const int w = rout->getWrap();

		for (int x = 0; x < rout->getLx(); ++x) {
			dst = rout->pixels() + x;
			for (int y = 0; y < rout->getLy(); ++y) {
				dst->r = src->r;
				dst->g = src->g;
				dst->b = src->b;
				dst->m = 0xff;
				dst += w;
				--src;
			}
		}
	} else {
		unsigned char *index = rgbBuffer;
		unsigned char *end = index + (rgbLx * rgbLy * 3);
		TPixel32 *outPix = rout->pixels();
		while (index < end) {
			outPix->r = (int)*index;
			index++;
			outPix->g = (int)*index;
			index++;
			outPix->b = (int)*index;
			index++;
			outPix->m = 255;
			outPix++;
		}
		rout->yMirror();
	}
}

//-----------------------------------------------------------------------------

void copyRGBBufferToTRasterGR8(unsigned char *rgbBuffer, int rgbLx, int rgbLy, int rgbWrap,
							   const TRasterGR8P &rout)
{
	TPixelGR8 *dst = rout->pixels();
	EP *src = (EP *)(rgbBuffer + (rgbLx * rgbLy - 1) * 3);
	const int w = rout->getWrap();

	for (int x = 0; x < rout->getLx(); ++x) {
		dst = rout->pixels() + x;
		for (int y = 0; y < rout->getLy(); ++y) {
			*dst = TPixelGR8::from(TPixelRGBM32(src->r, src->g, src->b));
			dst += w;
			--src;
		}
	}
}

//-----------------------------------------------------------------------------

void copyGR8BufferToTRasterGR8(unsigned char *gr8Buffer,
							   int rgbLx, int rgbLy,
							   const TRasterGR8P &rout, bool internal)
{
	if (internal) {
		unsigned char *dst = rout->getRawData();
		unsigned char *src = (gr8Buffer + rgbLx * rgbLy - 1);
		const int w = rout->getWrap();

		for (int x = 0; x < rout->getLx(); ++x) {
			dst = rout->getRawData() + x;
			for (int y = 0; y < rout->getLy(); ++y) {
				*dst = (*src);
				dst += w;
				--src;
			}
		}
	} else {
		memcpy(rout->getRawData(), gr8Buffer, rgbLx * rgbLy);
		rout->yMirror();
	}
}

//-----------------------------------------------------------------------------

void copyGR8BufferToTRasterBW(unsigned char *gr8Buffer,
							  int rgbLx, int rgbLy,
							  const TRasterGR8P &rout, bool internal, float thres)
{
	if (internal) {
		unsigned char *dst = rout->getRawData();
		unsigned char *src = (gr8Buffer + rgbLx * rgbLy - 1);
		const int w = rout->getWrap();

		for (int x = 0; x < rout->getLx(); ++x) {
			dst = rout->getRawData() + x;
			for (int y = 0; y < rout->getLy(); ++y) {
				if (*src < thres)
					*dst = 0;
				else
					*dst = 255;
				dst += w;
				--src;
			}
		}
	} else {
		memcpy(rout->getRawData(), gr8Buffer, rgbLx * rgbLy);
		rout->yMirror();
	}
}

//-----------------------------------------------------------------------------

void copyBWBufferToTRasterGR8(const unsigned char *buffer,
							  int rgbLx, int rgbLy,
							  const TRasterGR8P &rout, bool isBW, bool internal)
{
	if (0)
		assert(0);
	else {
		int i = 0;
		UCHAR *pix = rout->getRawData();
		const unsigned char *byte;
		while (i < rgbLx * rgbLy) {
			int bytePos = i / 8;
			int bitPos = i % 8;
			byte = buffer + bytePos;
			bool bit = (*byte) >> (7 - bitPos);
			if (isBW)
				*pix = (bit ? 255 : 0);
			else
				*pix = (bit ? 0 : 255);

			pix++;
			i++;
		}
		rout->yMirror();
	}
}

//-----------------------------------------------------------------------------

void copy90BWBufferToRasGR8(unsigned char *bwBuffer, int bwLx, int bwLy, int bwWrap, bool isBW,
							TRasterGR8P &rout, int mirror, int ninety)
{
	int x1 = 0;
	int y1 = 0;
	int x2 = bwLx - 1;
	int y2 = bwLy - 1;
	int newx = 0;
	int newy = 0;

	UCHAR *bufin, *bufout, /* *bytein,*/ *byteout;
	int bytewrapin, bitoffsin, wrapout;
	int value_for_0, value_for_1;
	int u, v, lu, lv, su, sv, u00, v00, u0, v0, dudy, dvdy, dudx, dvdx;
	int x, y, lx, ly;
	int u1, v1, u2, v2;

	mirror &= 1;
	ninety &= 3;

	if (!ninety && !mirror) {
		assert(0);
		return;
	}

	if (isBW) {
		value_for_0 = 0;
		value_for_1 = 255;
	} else {
		value_for_0 = 255;
		value_for_1 = 0;
	}

	u1 = x1;
	v1 = y1;
	u2 = x2;
	v2 = y2;

	su = u2 - u1;
	sv = v2 - v1;
	lu = u2 - u1 + 1;
	lv = v2 - v1 + 1;

	if (ninety & 1) {
		lx = lv;
		ly = lu;
	} else {
		lx = lu;
		ly = lv;
	}

	bufin = bwBuffer;
	bytewrapin = (bwWrap + 7) >> 3;
	bitoffsin = 0; //rin->bit_offs;
	bufout = rout->getRawData();
	wrapout = rout->getWrap();

	dudx = 0;
	dudy = 0;
	dvdx = 0;
	dvdy = 0;

	switch ((mirror << 2) + ninety) {
	case (0 << 2) + 0:
		u00 = u1;
		v00 = v1;
		dudx = 1;
		dvdy = 1;
		break;
	case (0 << 2) + 1:
		u00 = u1;
		v00 = v1 + sv;
		dudy = 1;
		dvdx = -1;
		break;
	case (0 << 2) + 2:
		u00 = u1 + su;
		v00 = v1 + sv;
		dudx = -1;
		dvdy = -1;
		break;
	case (0 << 2) + 3:
		u00 = u1 + su;
		v00 = v1;
		dudy = -1;
		dvdx = 1;
		break;
	case (1 << 2) + 0:
		u00 = u1 + su;
		v00 = v1;
		dudx = -1;
		dvdy = 1;
		break;
	case (1 << 2) + 1:
		u00 = u1 + su;
		v00 = v1 + sv;
		dudy = -1;
		dvdx = -1;
		break;
	case (1 << 2) + 2:
		u00 = u1;
		v00 = v1 + sv;
		dudx = 1;
		dvdy = -1;
		break;
	case (1 << 2) + 3:
		u00 = u1;
		v00 = v1;
		dudy = 1;
		dvdx = 1;
		break;
	default:
		abort();
		u00 = v00 = dudy = dvdx = 0;
	}

	if (dudx)
		for (u0 = u00, v0 = v00, y = newy; y < newy + ly; v0 += dvdy, y++) {
			u = u0;
			v = v0;
			byteout = bufout + newx + y * wrapout;
			for (x = newx; x < newx + lx; u += dudx, x++) {
				if (GET_BIT(u, v, bufin, bytewrapin, bitoffsin))
					*byteout++ = value_for_1;
				else
					*byteout++ = value_for_0;
			}
		}
	else
		for (u0 = u00, v0 = v00, y = newy; y < newy + ly; u0 += dudy, y++) {
			u = u0;
			v = v0;
			byteout = bufout + newx + y * wrapout;
			for (x = newx; x < newx + lx; v += dvdx, x++) {
				if (GET_BIT(u, v, bufin, bytewrapin, bitoffsin))
					*byteout++ = value_for_1;
				else
					*byteout++ = value_for_0;
			}
		}
}
};