#pragma once
#include "GLShaderParam.h"
#include "Emu/GS/RSXFragmentProgram.h"

struct GLFragmentDecompilerThread : public ThreadBase
{
	union OPDEST
	{
		u32 HEX;

		struct
		{
			u32 end					: 1;
			u32 dest_reg			: 6;
			u32 fp16				: 1;
			u32 set_cond			: 1;
			u32 mask_x				: 1;
			u32 mask_y				: 1;
			u32 mask_z				: 1;
			u32 mask_w				: 1;
			u32 src_attr_reg_num	: 4;
			u32 tex_num				: 4;
			u32 exp_tex				: 1;
			u32 prec 				: 2;
			u32 opcode 				: 6;
			u32 no_dest 			: 1;
			u32 saturate 			: 1;
		};
	} dst;

	union SRC0
	{
		u32 HEX;

		struct
		{
			u32 reg_type			: 2;
			u32 tmp_reg_index		: 6;
			u32 fp16				: 1;
			u32 swizzle_x			: 2;
			u32 swizzle_y			: 2;
			u32 swizzle_z			: 2;
			u32 swizzle_w			: 2;
			u32 neg					: 1;
			u32 exec_if_lt			: 1;
			u32 exec_if_eq			: 1;
			u32 exec_if_gr			: 1;
			u32 cond_swizzle_x		: 2;
			u32 cond_swizzle_y		: 2;
			u32 cond_swizzle_z		: 2;
			u32 cond_swizzle_w		: 2;
			u32 abs					: 1;
			u32 cond_mod_reg_index	: 1;
			u32 cond_reg_index		: 1;
		};
	} src0;

	union SRC1
	{
		u32 HEX;

		struct
		{
			u32 reg_type			: 2;
			u32 tmp_reg_index		: 6;
			u32 fp16				: 1;
			u32 swizzle_x			: 2;
			u32 swizzle_y			: 2;
			u32 swizzle_z			: 2;
			u32 swizzle_w			: 2;
			u32 neg					: 1;
			u32 abs					: 1;
			u32 input_mod_src0		: 3;
			u32						: 6;
			u32 scale				: 3;
			u32 opcode_is_branch	: 1;
		};
	} src1;

	union SRC2
	{
		u32 HEX;

		struct
		{
			u32 reg_type		: 2;
			u32 tmp_reg_index	: 6;
			u32 fp16			: 1;
			u32 swizzle_x		: 2;
			u32 swizzle_y		: 2;
			u32 swizzle_z		: 2;
			u32 swizzle_w		: 2;
			u32 neg				: 1;
			u32 abs				: 1;
			u32 addr_reg		: 11;
			u32 use_index_reg	: 1;
			u32 perspective_corr: 1;
		};
	} src2;

	std::string main;
	std::string& m_shader;
	GLParamArray& m_parr;
	u32 m_addr;
	u32& m_size;
	u32 m_const_index;
	u32 m_offset;
	u32 m_location;
	u32 m_ctrl;

	GLFragmentDecompilerThread(std::string& shader, GLParamArray& parr, u32 addr, u32& size, u32 ctrl)
		: ThreadBase("Fragment Shader Decompiler Thread")
		, m_shader(shader)
		, m_parr(parr)
		, m_addr(addr)
		, m_size(size) 
		, m_const_index(0)
		, m_location(0)
		, m_ctrl(ctrl)
	{
		m_size = 0;
	}

	std::string GetMask();

	void AddCode(std::string code, bool append_mask = true);
	std::string AddReg(u32 index, int fp16);
	bool HasReg(u32 index, int fp16);
	std::string AddCond(int fp16);
	std::string AddConst();
	std::string AddTex();

	template<typename T> std::string GetSRC(T src);
	std::string BuildCode();

	virtual void Task();

	u32 GetData(const u32 d) const { return d << 16 | d >> 16; }
};

struct GLShaderProgram
{
	GLShaderProgram();
	~GLShaderProgram();

	GLFragmentDecompilerThread* m_decompiler_thread;

	GLParamArray parr;

	std::string shader;

	u32 id;
	
	void Wait()
	{
		if(m_decompiler_thread && m_decompiler_thread->IsAlive())
		{
			m_decompiler_thread->Join();
		}
	}
	void Decompile(RSXShaderProgram& prog);
	void Compile();

	void Delete();
};