/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * Color Conversion Routines
 *
 * Copyright 2010 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 * Copyright 2016 Armin Novak <armin.novak@thincast.com>
 * Copyright 2016 Thincast Technologies GmbH
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#include <winpr/crt.h>

#include <freerdp/log.h>
#include <freerdp/freerdp.h>
#include <freerdp/primitives.h>

#define TAG FREERDP_TAG("color")

BYTE* freerdp_glyph_convert(UINT32 width, UINT32 height, const BYTE* data)
{
	UINT32 x, y;
	const BYTE* srcp;
	BYTE* dstp;
	BYTE* dstData;
	UINT32 scanline;
	/*
	 * converts a 1-bit-per-pixel glyph to a one-byte-per-pixel glyph:
	 * this approach uses a little more memory, but provides faster
	 * means of accessing individual pixels in blitting operations
	 */
	scanline = (width + 7) / 8;
	dstData = (BYTE*) _aligned_malloc(width * height, 16);

	if (!dstData)
		return NULL;

	ZeroMemory(dstData, width * height);
	dstp = dstData;

	for (y = 0; y < height; y++)
	{
		srcp = data + (y * scanline);

		for (x = 0; x < width; x++)
		{
			if ((*srcp & (0x80 >> (x % 8))) != 0)
				*dstp = 0xFF;

			dstp++;

			if (((x + 1) % 8 == 0) && x != 0)
				srcp++;
		}
	}

	return dstData;
}

BOOL freerdp_image_copy_from_monochrome(BYTE* pDstData, UINT32 DstFormat,
                                        UINT32 nDstStep, UINT32 nXDst, UINT32 nYDst,
                                        UINT32 nWidth, UINT32 nHeight,
                                        const BYTE* pSrcData,
                                        UINT32 backColor, UINT32 foreColor,
                                        const gdiPalette* palette)
{
	UINT32 x, y;
	BOOL vFlip;
	UINT32 monoStep;
	const UINT32 dstBytesPerPixel = GetBytesPerPixel(DstFormat);

	if (!pDstData || !pSrcData || !palette)
		return FALSE;

	if (nDstStep == 0)
		nDstStep = dstBytesPerPixel * nWidth;

	vFlip = FALSE;
	monoStep = (nWidth + 7) / 8;

	for (y = 0; y < nHeight; y++)
	{
		const BYTE* monoBits;
		BYTE* pDstLine = &pDstData[((nYDst + y) * nDstStep)];
		UINT32 monoBit = 0x80;

		if (!vFlip)
			monoBits = &pSrcData[monoStep * y];
		else
			monoBits = &pSrcData[monoStep * (nHeight - y - 1)];

		for (x = 0; x < nWidth; x++)
		{
			BYTE* pDstPixel = &pDstLine[((nXDst + x) * GetBytesPerPixel(DstFormat))];
			BOOL monoPixel = (*monoBits & monoBit) ? TRUE : FALSE;

			if (!(monoBit >>= 1))
			{
				monoBits++;
				monoBit = 0x80;
			}

			if (monoPixel)
				WriteColor(pDstPixel, DstFormat, backColor);
			else
				WriteColor(pDstPixel, DstFormat, foreColor);
		}
	}

	return TRUE;
}

static INLINE UINT32 freerdp_image_inverted_pointer_color(UINT32 x, UINT32 y,
        UINT32 format)
{
#if 1
	/**
	 * Inverted pointer colors (where individual pixels can change their
	 * color to accommodate the background behind them) only seem to be
	 * supported on Windows.
	 * Using a static replacement color for these pixels (e.g. black)
	 * might result in invisible pointers depending on the background.
	 * This function returns either black or white, depending on the
	 * pixel's position.
	 */
	BYTE fill = (x + y) & 1 ? 0x00 : 0xFF;
#else
	BYTE fill = 0x00;
#endif
	return FreeRDPGetColor(format, fill, fill, fill, 0xFF);
}

/**
 * Drawing Monochrome Pointers:
 * http://msdn.microsoft.com/en-us/library/windows/hardware/ff556143/
 *
 * Drawing Color Pointers:
 * http://msdn.microsoft.com/en-us/library/windows/hardware/ff556138/
 */

BOOL freerdp_image_copy_from_pointer_data(
    BYTE* pDstData, UINT32 DstFormat, UINT32 nDstStep,
    UINT32 nXDst, UINT32 nYDst, UINT32 nWidth, UINT32 nHeight,
    const BYTE* xorMask, UINT32 xorMaskLength,
    const BYTE* andMask, UINT32 andMaskLength,
    UINT32 xorBpp, const gdiPalette* palette)
{
	UINT32 x, y;
	BOOL vFlip;
	UINT32 xorStep;
	UINT32 andStep;
	UINT32 xorBit;
	UINT32 andBit;
	UINT32 xorPixel;
	UINT32 andPixel;
	UINT32 dstBitsPerPixel;
	UINT32 dstBytesPerPixel;
	dstBitsPerPixel = GetBitsPerPixel(DstFormat);
	dstBytesPerPixel = GetBytesPerPixel(DstFormat);

	if (nDstStep <= 0)
		nDstStep = dstBytesPerPixel * nWidth;

	vFlip = (xorBpp == 1) ? FALSE : TRUE;
	andStep = (nWidth + 7) / 8;
	andStep += (andStep % 2);

	if (!xorMask || (xorMaskLength == 0))
		return FALSE;

	switch (xorBpp)
	{
		case 1:
			if (!andMask || (andMaskLength == 0))
				return FALSE;

			xorStep = (nWidth + 7) / 8;
			xorStep += (xorStep % 2);

			if (xorStep * nHeight > xorMaskLength)
				return FALSE;

			if (andStep * nHeight > andMaskLength)
				return FALSE;

			for (y = 0; y < nHeight; y++)
			{
				const BYTE* andBits;
				const BYTE* xorBits;
				BYTE* pDstPixel = &pDstData[((nYDst + y) * nDstStep) +
				                                         (nXDst * GetBytesPerPixel(DstFormat))];
				xorBit = andBit = 0x80;

				if (!vFlip)
				{
					xorBits = &xorMask[xorStep * y];
					andBits = &andMask[andStep * y];
				}
				else
				{
					xorBits = &xorMask[xorStep * (nHeight - y - 1)];
					andBits = &andMask[andStep * (nHeight - y - 1)];
				}

				for (x = 0; x < nWidth; x++)
				{
					UINT32 color = 0;
					xorPixel = (*xorBits & xorBit) ? 1 : 0;

					if (!(xorBit >>= 1))
					{
						xorBits++;
						xorBit = 0x80;
					}

					andPixel = (*andBits & andBit) ? 1 : 0;

					if (!(andBit >>= 1))
					{
						andBits++;
						andBit = 0x80;
					}

					if (!andPixel && !xorPixel)
						color = FreeRDPGetColor(DstFormat, 0, 0, 0, 0xFF); /* black */
					else if (!andPixel && xorPixel)
						color = FreeRDPGetColor(DstFormat, 0xFF, 0xFF, 0xFF, 0xFF); /* white */
					else if (andPixel && !xorPixel)
						color = FreeRDPGetColor(DstFormat, 0, 0, 0, 0); /* transparent */
					else if (andPixel && xorPixel)
						color = freerdp_image_inverted_pointer_color(x, y, DstFormat); /* inverted */

					WriteColor(pDstPixel, DstFormat, color);
					pDstPixel += GetBytesPerPixel(DstFormat);
				}
			}

			return TRUE;

		case 8:
		case 16:
		case 24:
		case 32:
			{
				UINT32 xorBytesPerPixel = xorBpp >> 3;
				xorStep = nWidth * xorBytesPerPixel;

				if (xorBpp == 8 && !palette)
				{
					WLog_ERR(TAG, "null palette in conversion from %"PRIu32" bpp to %"PRIu32" bpp",
					         xorBpp, dstBitsPerPixel);
					return FALSE;
				}

				if (xorStep * nHeight > xorMaskLength)
					return FALSE;

				if (andMask)
				{
					if (andStep * nHeight > andMaskLength)
						return FALSE;
				}

				for (y = 0; y < nHeight; y++)
				{
					const BYTE* xorBits;
					const BYTE* andBits = NULL;
					BYTE* pDstPixel = &pDstData[((nYDst + y) * nDstStep) +
					                                         (nXDst * GetBytesPerPixel(DstFormat))];
					andBit = 0x80;

					if (!vFlip)
					{
						if (andMask)
							andBits = &andMask[andStep * y];

						xorBits = &xorMask[xorStep * y];
					}
					else
					{
						if (andMask)
							andBits = &andMask[andStep * (nHeight - y - 1)];

						xorBits = &xorMask[xorStep * (nHeight - y - 1)];
					}

					for (x = 0; x < nWidth; x++)
					{
						UINT32 pixelFormat;
						UINT32 color;

						if (xorBpp == 32)
						{
							pixelFormat = PIXEL_FORMAT_BGRA32;
							xorPixel = ReadColor(xorBits, pixelFormat);
						}
						else if (xorBpp == 16)
						{
							pixelFormat = PIXEL_FORMAT_RGB15;
							xorPixel = ReadColor(xorBits, pixelFormat);
						}
						else if (xorBpp == 8)
						{
							pixelFormat = palette->format;
							xorPixel = palette->palette[xorBits[0]];
						}
						else
						{
							pixelFormat = PIXEL_FORMAT_BGR24;
							xorPixel = ReadColor(xorBits, pixelFormat);
						}

						xorPixel = FreeRDPConvertColor(xorPixel,
						                               pixelFormat,
						                               PIXEL_FORMAT_ARGB32,
						                               palette);
						xorBits += xorBytesPerPixel;
						andPixel = 0;

						if (andMask)
						{
							andPixel = (*andBits & andBit) ? 1 : 0;

							if (!(andBit >>= 1))
							{
								andBits++;
								andBit = 0x80;
							}
						}

						if (andPixel)
						{
							if (xorPixel == 0xFF000000) /* black -> transparent */
								xorPixel = 0x00000000;
							else if (xorPixel == 0xFFFFFFFF) /* white -> inverted */
								xorPixel = freerdp_image_inverted_pointer_color(x, y, PIXEL_FORMAT_ARGB32);
						}

						color = FreeRDPConvertColor(xorPixel, PIXEL_FORMAT_ARGB32,
						                            DstFormat, palette);
						WriteColor(pDstPixel, DstFormat, color);
						pDstPixel += GetBytesPerPixel(DstFormat);
					}
				}

				return TRUE;
			}

		default:
			WLog_ERR(TAG, "failed to convert from %"PRIu32" bpp to %"PRIu32" bpp",
			         xorBpp, dstBitsPerPixel);
			return FALSE;
	}
}

static INLINE BOOL overlapping(const BYTE* pDstData, UINT32 nXDst, UINT32 nYDst,
                               UINT32 nDstStep, UINT32 dstBytesPerPixel,
                               const BYTE* pSrcData, UINT32 nXSrc, UINT32 nYSrc,
                               UINT32 nSrcStep, UINT32 srcBytesPerPixel,
                               UINT32 nWidth, UINT32 nHeight)
{
	const BYTE* pDstStart = &pDstData[nXDst * dstBytesPerPixel + nYDst * nDstStep];
	const BYTE* pDstEnd = pDstStart + nHeight * nDstStep;
	const BYTE* pSrcStart = &pSrcData[nXSrc * srcBytesPerPixel + nYSrc * nSrcStep];
	const BYTE* pSrcEnd = pSrcStart + nHeight * nSrcStep;

	if ((pDstStart >= pSrcStart) && (pDstStart <= pSrcEnd))
		return TRUE;

	if ((pDstEnd >= pSrcStart) && (pDstEnd <= pSrcEnd))
		return TRUE;

	return FALSE;
}

BOOL freerdp_image_copy(BYTE* pDstData, DWORD DstFormat,
                        UINT32 nDstStep, UINT32 nXDst, UINT32 nYDst,
                        UINT32 nWidth, UINT32 nHeight,
                        const BYTE* pSrcData, DWORD SrcFormat,
                        UINT32 nSrcStep, UINT32 nXSrc, UINT32 nYSrc,
                        const gdiPalette* palette, UINT32 flags)
{
	const UINT32 dstByte = GetBytesPerPixel(DstFormat);
	const UINT32 srcByte = GetBytesPerPixel(SrcFormat);
	const UINT32 copyDstWidth = nWidth * dstByte;
	const UINT32 xSrcOffset = nXSrc * srcByte;
	const UINT32 xDstOffset = nXDst * dstByte;
	const BOOL vSrcVFlip = flags & FREERDP_FLIP_VERTICAL;
	UINT32 srcVOffset = 0;
	INT32 srcVMultiplier = 1;
	UINT32 dstVOffset = 0;
	INT32 dstVMultiplier = 1;

	if (!pDstData || !pSrcData)
		return FALSE;

	if (nDstStep == 0)
		nDstStep = nWidth * GetBytesPerPixel(DstFormat);

	if (nSrcStep == 0)
		nSrcStep = nWidth * GetBytesPerPixel(SrcFormat);

	if (vSrcVFlip)
	{
		srcVOffset = (nHeight - 1) * nSrcStep;
		srcVMultiplier = -1;
	}

	if (AreColorFormatsEqualNoAlpha(SrcFormat, DstFormat))
	{
		INT32 y;

		if (overlapping(pDstData, nXDst, nYDst, nDstStep, dstByte,
		                pSrcData, nXSrc, nYSrc, nSrcStep, srcByte,
		                nWidth, nHeight))
		{
			/* Copy down */
			if (nYDst < nYSrc)
			{
				for (y = 0; y < nHeight; y++)
				{
					const BYTE* srcLine = &pSrcData[(y + nYSrc) *
					                                            nSrcStep * srcVMultiplier +
					                                            srcVOffset];
					BYTE* dstLine = &pDstData[(y + nYDst) *
					                                      nDstStep * dstVMultiplier +
					                                      dstVOffset];
					memcpy(&dstLine[xDstOffset],
					       &srcLine[xSrcOffset], copyDstWidth);
				}
			}
			/* Copy up */
			else if (nYDst > nYSrc)
			{
				for (y = nHeight - 1; y >= 0; y--)
				{
					const BYTE* srcLine = &pSrcData[(y + nYSrc) *
					                                            nSrcStep * srcVMultiplier +
					                                            srcVOffset];
					BYTE* dstLine = &pDstData[(y + nYDst) *
					                                      nDstStep * dstVMultiplier +
					                                      dstVOffset];
					memcpy(&dstLine[xDstOffset],
					       &srcLine[xSrcOffset], copyDstWidth);
				}
			}
			/* Copy left */
			else if (nXSrc > nXDst)
			{
				for (y = 0; y < nHeight; y++)
				{
					const BYTE* srcLine = &pSrcData[(y + nYSrc) *
					                                            nSrcStep * srcVMultiplier +
					                                            srcVOffset];
					BYTE* dstLine = &pDstData[(y + nYDst) *
					                                      nDstStep * dstVMultiplier +
					                                      dstVOffset];
					memmove(&dstLine[xDstOffset],
					        &srcLine[xSrcOffset], copyDstWidth);
				}
			}
			/* Copy right */
			else if (nXSrc < nXDst)
			{
				for (y = nHeight - 1; y >= 0; y--)
				{
					const BYTE* srcLine = &pSrcData[(y + nYSrc) *
					                                            nSrcStep * srcVMultiplier +
					                                            srcVOffset];
					BYTE* dstLine = &pDstData[(y + nYDst) *
					                                      nDstStep * dstVMultiplier +
					                                      dstVOffset];
					memmove(&dstLine[xDstOffset],
					        &srcLine[xSrcOffset], copyDstWidth);
				}
			}
			/* Source and destination are equal... */
			else
			{
			}
		}
		else
		{
			for (y = 0; y < nHeight; y++)
			{
				const BYTE* srcLine = &pSrcData[(y + nYSrc) *
				                                            nSrcStep * srcVMultiplier +
				                                            srcVOffset];
				BYTE* dstLine = &pDstData[(y + nYDst) *
				                                      nDstStep * dstVMultiplier +
				                                      dstVOffset];
				memcpy(&dstLine[xDstOffset],
				       &srcLine[xSrcOffset], copyDstWidth);
			}
		}
	}
	else
	{
		UINT32 x, y;

		for (y = 0; y < nHeight; y++)
		{
			const BYTE* srcLine = &pSrcData[(y + nYSrc) *
			                                            nSrcStep * srcVMultiplier +
			                                            srcVOffset];
			BYTE* dstLine = &pDstData[(y + nYDst) *
			                                      nDstStep * dstVMultiplier + dstVOffset];

			for (x = 0; x < nWidth; x++)
			{
				UINT32 dstColor;
				UINT32 color = ReadColor(&srcLine[(x + nXSrc) * srcByte],
				                         SrcFormat);
				dstColor = FreeRDPConvertColor(color, SrcFormat, DstFormat, palette);
				WriteColor(&dstLine[(x + nXDst) * dstByte], DstFormat, dstColor);
			}
		}
	}

	return TRUE;
}

BOOL freerdp_image_fill(BYTE* pDstData, DWORD DstFormat,
                        UINT32 nDstStep, UINT32 nXDst, UINT32 nYDst,
                        UINT32 nWidth, UINT32 nHeight, UINT32 color)
{
	UINT32 x, y;
	const UINT32 bpp = GetBytesPerPixel(DstFormat);
	BYTE* pFirstDstLine = &pDstData[nYDst * nDstStep];
	BYTE* pFirstDstLineXOffset = &pFirstDstLine[nXDst * bpp];

	for (x = 0; x < nWidth; x++)
	{
		BYTE* pDst = &pFirstDstLine[(x + nXDst) * bpp];
		WriteColor(pDst, DstFormat, color);
	}

	for (y = 1; y < nHeight; y++)
	{
		BYTE* pDstLine = &pDstData[(y + nYDst) * nDstStep + nXDst * bpp];
		memcpy(pDstLine, pFirstDstLineXOffset, nWidth * bpp);
	}

	return TRUE;
}