#include "gfx_conv_mmx.h"
#include "gfx_conv_c.h"

// Packed
extern "C" void _Convert_YUV422_RGBA32_SSE(void *fromYPtr, void *toPtr,
	int width);
extern "C" void _Convert_YUV422_RGBA32_SSE2(void *fromYPtr, void *toPtr,
	int width);
extern "C" void _Convert_YUV422_RGBA32_SSSE3(void *fromYPtr, void *toPtr,
	int width);

// Planar
extern "C" void _Convert_YUV420P_RGBA32_SSE(void *fromYPtr, void *fromUPtr,
	void *fromVPtr, void *toPtr, int width);
extern "C" void _Convert_YUV420P_RGBA32_SSE2(void *fromYPtr, void *fromUPtr,
	void *fromVPtr, void *toPtr, int width);
extern "C" void _Convert_YUV420P_RGBA32_SSSE3(void *fromYPtr, void *fromUPtr,
	void *fromVPtr, void *toPtr, int width);


// Planar YUV420 means 2 Y lines share a UV line
void
gfx_conv_yuv420p_rgba32_sse(AVFrame *in, AVFrame *out, int width, int height)
{
	// in and out buffers must be aligned to 16 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 16 != 0) {
		gfx_conv_YCbCr420p_RGB32_c(in, out, width, height);
		return;
	}
	
	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *ubase = (uint8 *)in->data[1];
	uint8 *vbase = (uint8 *)in->data[2];
	uint8 *rgbbase = (uint8 *)out->data[0];
	
	int yBaseInc = in->linesize[0];
	int uBaseInc = in->linesize[1];
	int vBaseInc = in->linesize[2];
	int rgbBaseInc = out->linesize[0];
	
	for (int i=0;i<height;i+=2) {
		// First Y row
		_Convert_YUV420P_RGBA32_SSE(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		rgbbase += rgbBaseInc;

		// Second Y row but same u and v row
		_Convert_YUV420P_RGBA32_SSE(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		ubase += uBaseInc;
		vbase += vBaseInc;
		rgbbase += rgbBaseInc;
	}
}

// Planar YUV420 means 2 Y lines share a UV line
void
gfx_conv_yuv420p_rgba32_sse2(AVFrame *in, AVFrame *out, int width, int height)
{	
	// in and out buffers must be aligned to 32 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 32 != 0) {
		gfx_conv_YCbCr420p_RGB32_c(in, out, width, height);
		return;
	}

	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *ubase = (uint8 *)in->data[1];
	uint8 *vbase = (uint8 *)in->data[2];
	uint8 *rgbbase = (uint8 *)out->data[0];
	
	int yBaseInc = in->linesize[0];
	int uBaseInc = in->linesize[1];
	int vBaseInc = in->linesize[2];
	int rgbBaseInc = out->linesize[0];
	
	for (int i=0;i<height;i+=2) {
		// First Y row
		_Convert_YUV420P_RGBA32_SSE2(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		rgbbase += rgbBaseInc;

		// Second Y row but same u and v row
		_Convert_YUV420P_RGBA32_SSE2(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		ubase += uBaseInc;
		vbase += vBaseInc;
		rgbbase += rgbBaseInc;
	}
}

// Planar YUV420 means 2 Y lines share a UV line
void
gfx_conv_yuv420p_rgba32_ssse3(AVFrame *in, AVFrame *out, int width, int height)
{	
	// in and out buffers must be aligned to 32 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 32 != 0) {
		gfx_conv_YCbCr420p_RGB32_c(in, out, width, height);
		return;
	}

	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *ubase = (uint8 *)in->data[1];
	uint8 *vbase = (uint8 *)in->data[2];
	uint8 *rgbbase = (uint8 *)out->data[0];
	
	int yBaseInc = in->linesize[0];
	int uBaseInc = in->linesize[1];
	int vBaseInc = in->linesize[2];
	int rgbBaseInc = out->linesize[0];
	
	for (int i=0;i<height;i+=2) {
		// First Y row
		_Convert_YUV420P_RGBA32_SSSE3(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		rgbbase += rgbBaseInc;

		// Second Y row but same u and v row
		_Convert_YUV420P_RGBA32_SSSE3(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		ubase += uBaseInc;
		vbase += vBaseInc;
		rgbbase += rgbBaseInc;
	}
}

// Planar YUV422 means each Y line has it's own UV line
void
gfx_conv_yuv422p_rgba32_sse(AVFrame *in, AVFrame *out, int width, int height)
{
	// in and out buffers must be aligned to 32 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 32 != 0) {
		gfx_conv_YCbCr422_RGB32_c(in, out, width, height);
		return;
	}

	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *ubase = (uint8 *)in->data[1];
	uint8 *vbase = (uint8 *)in->data[2];
	uint8 *rgbbase = (uint8 *)out->data[0];
	
	int yBaseInc = in->linesize[0];
	int uBaseInc = in->linesize[1];
	int vBaseInc = in->linesize[2];
	int rgbBaseInc = out->linesize[0];
	
	for (int i=0;i<height;i++) {
		_Convert_YUV420P_RGBA32_SSE(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		ubase += uBaseInc;
		vbase += vBaseInc;
		rgbbase += rgbBaseInc;
	}
}

// Planar YUV422 means each Y line has it's own UV line
void
gfx_conv_yuv422p_rgba32_sse2(AVFrame *in, AVFrame *out, int width, int height)
{
	// in and out buffers must be aligned to 32 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 32 != 0) {
		gfx_conv_YCbCr422_RGB32_c(in, out, width, height);
		return;
	}

	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *ubase = (uint8 *)in->data[1];
	uint8 *vbase = (uint8 *)in->data[2];
	uint8 *rgbbase = (uint8 *)out->data[0];
	
	int yBaseInc = in->linesize[0];
	int uBaseInc = in->linesize[1];
	int vBaseInc = in->linesize[2];
	int rgbBaseInc = out->linesize[0];
	
	for (int i=0;i<height;i++) {
		_Convert_YUV420P_RGBA32_SSE2(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		ubase += uBaseInc;
		vbase += vBaseInc;
		rgbbase += rgbBaseInc;
	}
}

// Planar YUV422 means each Y line has it's own UV line
void
gfx_conv_yuv422p_rgba32_ssse3(AVFrame *in, AVFrame *out, int width, int height)
{
	// in and out buffers must be aligned to 32 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 32 != 0) {
		gfx_conv_YCbCr422_RGB32_c(in, out, width, height);
		return;
	}

	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *ubase = (uint8 *)in->data[1];
	uint8 *vbase = (uint8 *)in->data[2];
	uint8 *rgbbase = (uint8 *)out->data[0];
	
	int yBaseInc = in->linesize[0];
	int uBaseInc = in->linesize[1];
	int vBaseInc = in->linesize[2];
	int rgbBaseInc = out->linesize[0];
	
	for (int i=0;i<height;i++) {
		_Convert_YUV420P_RGBA32_SSSE3(ybase, ubase, vbase, rgbbase, width);
		ybase += yBaseInc;
		ubase += uBaseInc;
		vbase += vBaseInc;
		rgbbase += rgbBaseInc;
	}
}

// Packed YUV422 (YUYV)
void
gfx_conv_yuv422_rgba32_sse(AVFrame *in, AVFrame *out, int width, int height)
{
	// in and out buffers must be aligned to 16 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 16 != 0) {
		gfx_conv_YCbCr422_RGB32_c(in, out, width, height);
		return;
	}
		
	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *rgbbase = (uint8 *)out->data[0];

	for (int i = 0; i <= height; i++) {
		_Convert_YUV422_RGBA32_SSE(ybase, rgbbase, width);
		ybase += in->linesize[0];
		rgbbase += out->linesize[0];
	}
}

// Packed YUV422 (YUYV)
void
gfx_conv_yuv422_rgba32_sse2(AVFrame *in, AVFrame *out, int width, int height)
{
	// in and out buffers must be aligned to 32 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 32 != 0) {
		gfx_conv_YCbCr422_RGB32_c(in, out, width, height);
		return;
	}
		
	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *rgbbase = (uint8 *)out->data[0];

	for (int i = 0; i <= height; i++) {
		_Convert_YUV422_RGBA32_SSE2(ybase, rgbbase, width);
		ybase += in->linesize[0];
		rgbbase += out->linesize[0];
	}
}

// Packed YUV422 (YUYV)
void
gfx_conv_yuv422_rgba32_ssse3(AVFrame *in, AVFrame *out, int width, int height)
{
	// in and out buffers must be aligned to 32 bytes,
	// in should be as ffmpeg allocates it
	if ((off_t)out->data[0] % 32 != 0) {
		gfx_conv_YCbCr422_RGB32_c(in, out, width, height);
		return;
	}
		
	uint8 *ybase = (uint8 *)in->data[0];
	uint8 *rgbbase = (uint8 *)out->data[0];

	for (int i = 0; i <= height; i++) {
		_Convert_YUV422_RGBA32_SSSE3(ybase, rgbbase, width);
		ybase += in->linesize[0];
		rgbbase += out->linesize[0];
	}
}