/*
 *	Copyright (C) 2007-2009 Gabest
 *	http://www.gabest.org
 *
 *  This Program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2, or (at your option)
 *  any later version.
 *
 *  This Program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with GNU Make; see the file COPYING.  If not, write to
 *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
 *  http://www.gnu.org/copyleft/gpl.html
 *
 */

// TODO: x64

#include "stdafx.h"
#include "GSSetupPrimCodeGenerator.h"

using namespace Xbyak;

GSSetupPrimCodeGenerator::GSSetupPrimCodeGenerator(void* param, uint64 key, void* code, size_t maxsize)
	: GSCodeGenerator(code, maxsize)
	, m_local(*(GSScanlineLocalData*)param)
{
	#if _M_AMD64
	#error TODO
	#endif

	m_sel.key = key;

	m_en.z = m_sel.zb ? 1 : 0;
	m_en.f = m_sel.fb && m_sel.fge ? 1 : 0;
	m_en.t = m_sel.fb && m_sel.tfx != TFX_NONE ? 1 : 0;
	m_en.c = m_sel.fb && !(m_sel.tfx == TFX_DECAL && m_sel.tcc) ? 1 : 0;

	Generate();
}

void GSSetupPrimCodeGenerator::Generate()
{
	if((m_en.z || m_en.f) && !m_sel.sprite || m_en.t || m_en.c && m_sel.iip)
	{
		for(int i = 0; i < 5; i++)
		{
			if(m_cpu.has(util::Cpu::tAVX))
			{
				vmovaps(Xmm(3 + i), ptr[&m_shift[i]]);
			}
			else
			{
				movaps(Xmm(3 + i), ptr[&m_shift[i]]);
			}
		}
	}

	Depth();

	Texture();

	Color();

	ret();
}

void GSSetupPrimCodeGenerator::Depth()
{
	if(!m_en.z && !m_en.f)
	{
		return;
	}

	if(m_cpu.has(util::Cpu::tAVX))
	{
		if(!m_sel.sprite)
		{
			// GSVector4 p = dscan.p;

			vmovaps(xmm0, ptr[edx + 16]);

			if(m_en.f)
			{
				// GSVector4 df = p.wwww();

				vshufps(xmm1, xmm0, xmm0, _MM_SHUFFLE(3, 3, 3, 3));

				// m_local.d4.f = GSVector4i(df * 4.0f).xxzzlh();

				vmulps(xmm2, xmm1, xmm3);
				vcvttps2dq(xmm2, xmm2);
				vpshuflw(xmm2, xmm2, _MM_SHUFFLE(2, 2, 0, 0));
				vpshufhw(xmm2, xmm2, _MM_SHUFFLE(2, 2, 0, 0));
				vmovdqa(ptr[&m_local.d4.f], xmm2);

				for(int i = 0; i < 4; i++)
				{
					// m_local.d[i].f = GSVector4i(df * m_shift[i]).xxzzlh();

					vmulps(xmm2, xmm1, Xmm(4 + i));
					vcvttps2dq(xmm2, xmm2);
					vpshuflw(xmm2, xmm2, _MM_SHUFFLE(2, 2, 0, 0));
					vpshufhw(xmm2, xmm2, _MM_SHUFFLE(2, 2, 0, 0));
					vmovdqa(ptr[&m_local.d[i].f], xmm2);
				}
			}

			if(m_en.z)
			{
				// GSVector4 dz = p.zzzz();

				vshufps(xmm0, xmm0, _MM_SHUFFLE(2, 2, 2, 2));

				// m_local.d4.z = dz * 4.0f;

				vmulps(xmm1, xmm0, xmm3);
				vmovdqa(ptr[&m_local.d4.z], xmm1);

				for(int i = 0; i < 4; i++)
				{
					// m_local.d[i].z = dz * m_shift[i];

					vmulps(xmm1, xmm0, Xmm(4 + i));
					vmovdqa(ptr[&m_local.d[i].z], xmm1);
				}
			}
		}
		else
		{
			// GSVector4 p = vertices[0].p;

			vmovaps(xmm0, ptr[ecx + 16]);

			if(m_en.f)
			{
				// m_local.p.f = GSVector4i(p).zzzzh().zzzz();

				vcvttps2dq(xmm1, xmm0);
				vpshufhw(xmm1, xmm1, _MM_SHUFFLE(2, 2, 2, 2));
				vpshufd(xmm1, xmm1, _MM_SHUFFLE(2, 2, 2, 2));
				vmovdqa(ptr[&m_local.p.f], xmm1);
			}

			if(m_en.z)
			{
				// GSVector4 z = p.zzzz();

				vshufps(xmm0, xmm0, _MM_SHUFFLE(2, 2, 2, 2));

				if(m_sel.zoverflow)
				{
					// m_local.p.z = (GSVector4i(z * 0.5f) << 1) | (GSVector4i(z) & GSVector4i::x00000001());

					static const float half = 0.5f;

					vbroadcastss(xmm1, dword[&half]);
					vmulps(xmm1, xmm0);
					vcvttps2dq(xmm1, xmm1);
					vpslld(xmm1, 1);

					vcvttps2dq(xmm0, xmm0);
					vpcmpeqd(xmm2, xmm2);
					vpsrld(xmm2, 31);
					vpand(xmm0, xmm2);

					vpor(xmm0, xmm1);
				}
				else
				{
					// m_local.p.z = GSVector4i(z);

					vcvttps2dq(xmm0, xmm0);
				}

				vmovdqa(ptr[&m_local.p.z], xmm0);
			}
		}
	}
	else
	{
		if(!m_sel.sprite)
		{
			// GSVector4 p = dscan.p;

			movaps(xmm0, ptr[edx + 16]);

			if(m_en.f)
			{
				// GSVector4 df = p.wwww();

				movaps(xmm1, xmm0);
				shufps(xmm1, xmm1, _MM_SHUFFLE(3, 3, 3, 3));

				// m_local.d4.f = GSVector4i(df * 4.0f).xxzzlh();

				movaps(xmm2, xmm1);
				mulps(xmm2, xmm3);
				cvttps2dq(xmm2, xmm2);
				pshuflw(xmm2, xmm2, _MM_SHUFFLE(2, 2, 0, 0));
				pshufhw(xmm2, xmm2, _MM_SHUFFLE(2, 2, 0, 0));
				movdqa(ptr[&m_local.d4.f], xmm2);

				for(int i = 0; i < 4; i++)
				{
					// m_local.d[i].f = GSVector4i(df * m_shift[i]).xxzzlh();

					movaps(xmm2, xmm1);
					mulps(xmm2, Xmm(4 + i));
					cvttps2dq(xmm2, xmm2);
					pshuflw(xmm2, xmm2, _MM_SHUFFLE(2, 2, 0, 0));
					pshufhw(xmm2, xmm2, _MM_SHUFFLE(2, 2, 0, 0));
					movdqa(ptr[&m_local.d[i].f], xmm2);
				}
			}

			if(m_en.z)
			{
				// GSVector4 dz = p.zzzz();

				shufps(xmm0, xmm0, _MM_SHUFFLE(2, 2, 2, 2));

				// m_local.d4.z = dz * 4.0f;

				movaps(xmm1, xmm0);
				mulps(xmm1, xmm3);
				movdqa(ptr[&m_local.d4.z], xmm1);

				for(int i = 0; i < 4; i++)
				{
					// m_local.d[i].z = dz * m_shift[i];

					movaps(xmm1, xmm0);
					mulps(xmm1, Xmm(4 + i));
					movdqa(ptr[&m_local.d[i].z], xmm1);
				}
			}
		}
		else
		{
			// GSVector4 p = vertices[0].p;

			movaps(xmm0, ptr[ecx + 16]);

			if(m_en.f)
			{
				// m_local.p.f = GSVector4i(p).zzzzh().zzzz();

				cvttps2dq(xmm1, xmm0);
				pshufhw(xmm1, xmm1, _MM_SHUFFLE(2, 2, 2, 2));
				pshufd(xmm1, xmm1, _MM_SHUFFLE(2, 2, 2, 2));
				movdqa(ptr[&m_local.p.f], xmm1);
			}

			if(m_en.z)
			{
				// GSVector4 z = p.zzzz();

				shufps(xmm0, xmm0, _MM_SHUFFLE(2, 2, 2, 2));

				if(m_sel.zoverflow)
				{
					// m_local.p.z = (GSVector4i(z * 0.5f) << 1) | (GSVector4i(z) & GSVector4i::x00000001());

					static const float half = 0.5f;

					movss(xmm1, dword[&half]);
					shufps(xmm1, xmm1, _MM_SHUFFLE(0, 0, 0, 0));
					mulps(xmm1, xmm0);
					cvttps2dq(xmm1, xmm1);
					pslld(xmm1, 1);

					cvttps2dq(xmm0, xmm0);
					pcmpeqd(xmm2, xmm2);
					psrld(xmm2, 31);
					pand(xmm0, xmm2);

					por(xmm0, xmm1);
				}
				else
				{
					// m_local.p.z = GSVector4i(z);

					cvttps2dq(xmm0, xmm0);
				}

				movdqa(ptr[&m_local.p.z], xmm0);
			}
		}
	}
}

void GSSetupPrimCodeGenerator::Texture()
{
	if(!m_en.t)
	{
		return;
	}

	if(m_cpu.has(util::Cpu::tAVX))
	{
		// GSVector4 t = dscan.t;

		vmovaps(xmm0, ptr[edx + 32]);

		vmulps(xmm1, xmm0, xmm3);

		if(m_sel.fst)
		{
			// m_local.d4.st = GSVector4i(t * 4.0f);

			vcvttps2dq(xmm1, xmm1);
			vmovdqa(ptr[&m_local.d4.st], xmm1);
		}
		else
		{
			// m_local.d4.stq = t * 4.0f;

			vmovaps(ptr[&m_local.d4.stq], xmm1);
		}

		for(int j = 0, k = m_sel.fst ? 2 : 3; j < k; j++)
		{
			// GSVector4 ds = t.xxxx();
			// GSVector4 dt = t.yyyy();
			// GSVector4 dq = t.zzzz();

			vshufps(xmm1, xmm0, xmm0, (uint8)_MM_SHUFFLE(j, j, j, j));

			for(int i = 0; i < 4; i++)
			{
				// GSVector4 v = ds/dt * m_shift[i];

				vmulps(xmm2, xmm1, Xmm(4 + i));

				if(m_sel.fst)
				{
					// m_local.d[i].si/ti = GSVector4i(v);

					vcvttps2dq(xmm2, xmm2);

					switch(j)
					{
					case 0: vmovdqa(ptr[&m_local.d[i].si], xmm2); break;
					case 1: vmovdqa(ptr[&m_local.d[i].ti], xmm2); break;
					}
				}
				else
				{
					// m_local.d[i].s/t/q = v;

					switch(j)
					{
					case 0: vmovaps(ptr[&m_local.d[i].s], xmm2); break;
					case 1: vmovaps(ptr[&m_local.d[i].t], xmm2); break;
					case 2: vmovaps(ptr[&m_local.d[i].q], xmm2); break;
					}
				}
			}
		}
	}
	else
	{
		// GSVector4 t = dscan.t;

		movaps(xmm0, ptr[edx + 32]);

		movaps(xmm1, xmm0);
		mulps(xmm1, xmm3);

		if(m_sel.fst)
		{
			// m_local.d4.st = GSVector4i(t * 4.0f);

			cvttps2dq(xmm1, xmm1);
			movdqa(ptr[&m_local.d4.st], xmm1);
		}
		else
		{
			// m_local.d4.stq = t * 4.0f;

			movaps(ptr[&m_local.d4.stq], xmm1);
		}

		for(int j = 0, k = m_sel.fst ? 2 : 3; j < k; j++)
		{
			// GSVector4 ds = t.xxxx();
			// GSVector4 dt = t.yyyy();
			// GSVector4 dq = t.zzzz();

			movaps(xmm1, xmm0);
			shufps(xmm1, xmm1, (uint8)_MM_SHUFFLE(j, j, j, j));

			for(int i = 0; i < 4; i++)
			{
				// GSVector4 v = ds/dt * m_shift[i];

				movaps(xmm2, xmm1);
				mulps(xmm2, Xmm(4 + i));

				if(m_sel.fst)
				{
					// m_local.d[i].si/ti = GSVector4i(v);

					cvttps2dq(xmm2, xmm2);

					switch(j)
					{
					case 0: movdqa(ptr[&m_local.d[i].si], xmm2); break;
					case 1: movdqa(ptr[&m_local.d[i].ti], xmm2); break;
					}
				}
				else
				{
					// m_local.d[i].s/t/q = v;

					switch(j)
					{
					case 0: movaps(ptr[&m_local.d[i].s], xmm2); break;
					case 1: movaps(ptr[&m_local.d[i].t], xmm2); break;
					case 2: movaps(ptr[&m_local.d[i].q], xmm2); break;
					}
				}
			}
		}
	}
}

void GSSetupPrimCodeGenerator::Color()
{
	if(!m_en.c)
	{
		return;
	}

	if(m_cpu.has(util::Cpu::tAVX))
	{
		if(m_sel.iip)
		{
			// GSVector4 c = dscan.c;

			vmovaps(xmm0, ptr[edx]);

			// m_local.d4.c = GSVector4i(c * 4.0f).xzyw().ps32();

			vmulps(xmm1, xmm0, xmm3);
			vcvttps2dq(xmm1, xmm1);
			vpshufd(xmm1, xmm1, _MM_SHUFFLE(3, 1, 2, 0));
			vpackssdw(xmm1, xmm1);
			vmovdqa(ptr[&m_local.d4.c], xmm1);

			// xmm3 is not needed anymore

			// GSVector4 dr = c.xxxx();
			// GSVector4 db = c.zzzz();

			vshufps(xmm2, xmm0, xmm0, _MM_SHUFFLE(0, 0, 0, 0));
			vshufps(xmm3, xmm0, xmm0, _MM_SHUFFLE(2, 2, 2, 2));

			for(int i = 0; i < 4; i++)
			{
				// GSVector4i r = GSVector4i(dr * m_shift[i]).ps32();

				vmulps(xmm0, xmm2, Xmm(4 + i));
				vcvttps2dq(xmm0, xmm0);
				vpackssdw(xmm0, xmm0);

				// GSVector4i b = GSVector4i(db * m_shift[i]).ps32();

				vmulps(xmm1, xmm3, Xmm(4 + i));
				vcvttps2dq(xmm1, xmm1);
				vpackssdw(xmm1, xmm1);

				// m_local.d[i].rb = r.upl16(b);

				vpunpcklwd(xmm0, xmm1);
				vmovdqa(ptr[&m_local.d[i].rb], xmm0);
			}

			// GSVector4 c = dscan.c;

			vmovaps(xmm0, ptr[edx]); // not enough regs, have to reload it

			// GSVector4 dg = c.yyyy();
			// GSVector4 da = c.wwww();

			vshufps(xmm2, xmm0, xmm0, _MM_SHUFFLE(1, 1, 1, 1));
			vshufps(xmm3, xmm0, xmm0, _MM_SHUFFLE(3, 3, 3, 3));

			for(int i = 0; i < 4; i++)
			{
				// GSVector4i g = GSVector4i(dg * m_shift[i]).ps32();

				vmulps(xmm0, xmm2, Xmm(4 + i));
				vcvttps2dq(xmm0, xmm0);
				vpackssdw(xmm0, xmm0);

				// GSVector4i a = GSVector4i(da * m_shift[i]).ps32();

				vmulps(xmm1, xmm3, Xmm(4 + i));
				vcvttps2dq(xmm1, xmm1);
				vpackssdw(xmm1, xmm1);

				// m_local.d[i].ga = g.upl16(a);

				vpunpcklwd(xmm0, xmm1);
				vmovdqa(ptr[&m_local.d[i].ga], xmm0);
			}
		}
		else
		{
			// GSVector4i c = GSVector4i(vertices[0].c);

			vcvttps2dq(xmm0, ptr[ecx]);

			// c = c.upl16(c.zwxy());

			vpshufd(xmm1, xmm0, _MM_SHUFFLE(1, 0, 3, 2));
			vpunpcklwd(xmm0, xmm1);

			// if(!tme) c = c.srl16(7);

			if(m_sel.tfx == TFX_NONE)
			{
				vpsrlw(xmm0, 7);
			}

			// m_local.c.rb = c.xxxx();
			// m_local.c.ga = c.zzzz();

			vpshufd(xmm1, xmm0, _MM_SHUFFLE(0, 0, 0, 0));
			vpshufd(xmm2, xmm0, _MM_SHUFFLE(2, 2, 2, 2));

			vmovdqa(ptr[&m_local.c.rb], xmm1);
			vmovdqa(ptr[&m_local.c.ga], xmm2);
		}
	}
	else
	{
		if(m_sel.iip)
		{
			// GSVector4 c = dscan.c;

			movaps(xmm0, ptr[edx]);
			movaps(xmm1, xmm0);

			// m_local.d4.c = GSVector4i(c * 4.0f).xzyw().ps32();

			movaps(xmm2, xmm0);
			mulps(xmm2, xmm3);
			cvttps2dq(xmm2, xmm2);
			pshufd(xmm2, xmm2, _MM_SHUFFLE(3, 1, 2, 0));
			packssdw(xmm2, xmm2);
			movdqa(ptr[&m_local.d4.c], xmm2);

			// xmm3 is not needed anymore

			// GSVector4 dr = c.xxxx();
			// GSVector4 db = c.zzzz();

			shufps(xmm0, xmm0, _MM_SHUFFLE(0, 0, 0, 0));
			shufps(xmm1, xmm1, _MM_SHUFFLE(2, 2, 2, 2));

			for(int i = 0; i < 4; i++)
			{
				// GSVector4i r = GSVector4i(dr * m_shift[i]).ps32();

				movaps(xmm2, xmm0);
				mulps(xmm2, Xmm(4 + i));
				cvttps2dq(xmm2, xmm2);
				packssdw(xmm2, xmm2);

				// GSVector4i b = GSVector4i(db * m_shift[i]).ps32();

				movaps(xmm3, xmm1);
				mulps(xmm3, Xmm(4 + i));
				cvttps2dq(xmm3, xmm3);
				packssdw(xmm3, xmm3);

				// m_local.d[i].rb = r.upl16(b);

				punpcklwd(xmm2, xmm3);
				movdqa(ptr[&m_local.d[i].rb], xmm2);
			}

			// GSVector4 c = dscan.c;

			movaps(xmm0, ptr[edx]); // not enough regs, have to reload it
			movaps(xmm1, xmm0);

			// GSVector4 dg = c.yyyy();
			// GSVector4 da = c.wwww();

			shufps(xmm0, xmm0, _MM_SHUFFLE(1, 1, 1, 1));
			shufps(xmm1, xmm1, _MM_SHUFFLE(3, 3, 3, 3));

			for(int i = 0; i < 4; i++)
			{
				// GSVector4i g = GSVector4i(dg * m_shift[i]).ps32();

				movaps(xmm2, xmm0);
				mulps(xmm2, Xmm(4 + i));
				cvttps2dq(xmm2, xmm2);
				packssdw(xmm2, xmm2);

				// GSVector4i a = GSVector4i(da * m_shift[i]).ps32();

				movaps(xmm3, xmm1);
				mulps(xmm3, Xmm(4 + i));
				cvttps2dq(xmm3, xmm3);
				packssdw(xmm3, xmm3);

				// m_local.d[i].ga = g.upl16(a);

				punpcklwd(xmm2, xmm3);
				movdqa(ptr[&m_local.d[i].ga], xmm2);
			}
		}
		else
		{
			// GSVector4i c = GSVector4i(vertices[0].c);

			movaps(xmm0, ptr[ecx]);
			cvttps2dq(xmm0, xmm0);

			// c = c.upl16(c.zwxy());

			pshufd(xmm1, xmm0, _MM_SHUFFLE(1, 0, 3, 2));
			punpcklwd(xmm0, xmm1);

			// if(!tme) c = c.srl16(7);

			if(m_sel.tfx == TFX_NONE)
			{
				psrlw(xmm0, 7);
			}

			// m_local.c.rb = c.xxxx();
			// m_local.c.ga = c.zzzz();

			pshufd(xmm1, xmm0, _MM_SHUFFLE(0, 0, 0, 0));
			pshufd(xmm2, xmm0, _MM_SHUFFLE(2, 2, 2, 2));

			movdqa(ptr[&m_local.c.rb], xmm1);
			movdqa(ptr[&m_local.c.ga], xmm2);
		}
	}
}

const GSVector4 GSSetupPrimCodeGenerator::m_shift[5] =
{
	GSVector4(4.0f, 4.0f, 4.0f, 4.0f),
	GSVector4(0.0f, 1.0f, 2.0f, 3.0f),
	GSVector4(-1.0f, 0.0f, 1.0f, 2.0f),
	GSVector4(-2.0f, -1.0f, 0.0f, 1.0f),
	GSVector4(-3.0f, -2.0f, -1.0f, 0.0f),
};