#ifndef COMBINER_H
#define COMBINER_H

#include <map>
#include <memory>

#include "GLideN64.h"
#include "GraphicsDrawer.h"
#include "gDP.h"
#include "CombinerKey.h"
#include "Graphics/CombinerProgram.h"
#include "Graphics/ShaderProgram.h"

/*
* G_SETCOMBINE: color combine modes
*/
/* Color combiner constants: */
#define G_CCMUX_COMBINED	0
#define G_CCMUX_TEXEL0		1
#define G_CCMUX_TEXEL1		2
#define G_CCMUX_PRIMITIVE	3
#define G_CCMUX_SHADE		4
#define G_CCMUX_ENVIRONMENT	5
#define G_CCMUX_CENTER		6
#define G_CCMUX_SCALE		6
#define G_CCMUX_COMBINED_ALPHA	7
#define G_CCMUX_TEXEL0_ALPHA	8
#define G_CCMUX_TEXEL1_ALPHA	9
#define G_CCMUX_PRIMITIVE_ALPHA	10
#define G_CCMUX_SHADE_ALPHA	11
#define G_CCMUX_ENV_ALPHA	12
#define G_CCMUX_LOD_FRACTION	13
#define G_CCMUX_PRIM_LOD_FRAC	14
#define G_CCMUX_NOISE		7
#define G_CCMUX_K4		7
#define G_CCMUX_K5		15
#define G_CCMUX_1		6
#define G_CCMUX_0		31

/* Alpha combiner constants: */
#define G_ACMUX_COMBINED	0
#define G_ACMUX_TEXEL0		1
#define G_ACMUX_TEXEL1		2
#define G_ACMUX_PRIMITIVE	3
#define G_ACMUX_SHADE		4
#define G_ACMUX_ENVIRONMENT	5
#define G_ACMUX_LOD_FRACTION	0
#define G_ACMUX_PRIM_LOD_FRAC	6
#define G_ACMUX_1		6
#define G_ACMUX_0		7

#define EncodeCombineMode( a0, b0, c0, d0, Aa0, Ab0, Ac0, Ad0,	\
	a1, b1, c1, d1,	Aa1, Ab1, Ac1, Ad1 ) \
	static_cast<u64>((static_cast<u64>(_SHIFTL( G_CCMUX_##a0, 20, 4 ) | _SHIFTL( G_CCMUX_##c0, 15, 5 ) | \
	_SHIFTL( G_ACMUX_##Aa0, 12, 3 ) | _SHIFTL( G_ACMUX_##Ac0, 9, 3 ) | \
	_SHIFTL( G_CCMUX_##a1, 5, 4 ) | _SHIFTL( G_CCMUX_##c1, 0, 5 )) << 32) | \
	static_cast<u64>(_SHIFTL( G_CCMUX_##b0, 28, 4 ) | _SHIFTL( G_CCMUX_##d0, 15, 3 ) | \
	_SHIFTL( G_ACMUX_##Ab0, 12, 3 ) | _SHIFTL( G_ACMUX_##Ad0, 9, 3 ) | \
	_SHIFTL( G_CCMUX_##b1, 24, 4 ) | _SHIFTL( G_ACMUX_##Aa1, 21, 3 ) | \
	_SHIFTL( G_ACMUX_##Ac1, 18, 3 ) | _SHIFTL( G_CCMUX_##d1, 6, 3 ) | \
	_SHIFTL( G_ACMUX_##Ab1, 3, 3 ) | _SHIFTL( G_ACMUX_##Ad1, 0, 3 )))

// Internal combiner commands
#define LOAD		0
#define SUB			1
#define MUL			2
#define ADD			3
#define INTER		4

// Internal generalized combiner inputs
#define G_GCI_COMBINED			0
#define G_GCI_TEXEL0			1
#define G_GCI_TEXEL1			2
#define G_GCI_PRIMITIVE			3
#define G_GCI_SHADE				4
#define G_GCI_ENVIRONMENT		5
#define G_GCI_CENTER			6
#define G_GCI_SCALE				7
#define G_GCI_COMBINED_ALPHA	8
#define G_GCI_TEXEL0_ALPHA		9
#define G_GCI_TEXEL1_ALPHA		10
#define G_GCI_PRIMITIVE_ALPHA	11
#define G_GCI_SHADE_ALPHA		12
#define G_GCI_ENV_ALPHA			13
#define G_GCI_LOD_FRACTION		14
#define G_GCI_PRIM_LOD_FRAC		15
#define G_GCI_NOISE				16
#define G_GCI_K4				17
#define G_GCI_K5				18
#define G_GCI_ONE				19
#define G_GCI_ZERO				20
#define G_GCI_HALF				21
#define G_GCI_HW_LIGHT			22
#define G_GCI_HW_LIGHT			22
#define G_GCI_LAST				23

struct CombinerOp
{
	u32 op = LOAD;
	u32 param1 = G_GCI_LAST;
	u32 param2 = G_GCI_LAST;
	u32 param3 = G_GCI_LAST;
};

struct CombinerStage
{
	u32 numOps;
	CombinerOp op[6];
};

struct Combiner
{
	u32 numStages;
	CombinerStage stage[2];
};

struct CombineCycle
{
	u32 sa, sb, m, a;
};

class CombinerInfo
{
public:
	void init();
	void destroy();
	void update();
	void setCombine(u64 _mux);
	void updateParameters();

	void setDepthFogCombiner();
	graphics::ShaderProgram * getTexrectUpscaleCopyProgram();
	graphics::ShaderProgram * getTexrectColorAndDepthUpscaleCopyProgram();
	graphics::ShaderProgram * getTexrectDownscaleCopyProgram();
	graphics::ShaderProgram * getTexrectColorAndDepthDownscaleCopyProgram();

	graphics::CombinerProgram * getCurrent() const { return m_pCurrent; }
	bool isChanged() const {return m_bChanged;}
	bool isShaderCacheSupported() const;

	static CombinerInfo & get();

	void setPolygonMode(DrawingState _drawingState);
	bool isRectMode() const { return m_rectMode; }

private:
	CombinerInfo()
		: m_bChanged(false)
		, m_rectMode(true)
		, m_shadersLoaded(0)
		, m_configOptionsBitSet(0)
		, m_pCurrent(nullptr) {}
	CombinerInfo(const CombinerInfo &) = delete;

	void _saveShadersStorage() const;
	bool _loadShadersStorage();

	bool m_bChanged;
	bool m_rectMode;
	u32 m_shadersLoaded;
	u32 m_configOptionsBitSet;

	graphics::CombinerProgram * m_pCurrent;
	graphics::Combiners m_combiners;

	std::unique_ptr<graphics::ShaderProgram> m_shadowmapProgram;
	std::unique_ptr<graphics::ShaderProgram> m_texrectUpscaleCopyProgram;
	std::unique_ptr<graphics::ShaderProgram> m_texrectColorAndDepthUpscaleCopyProgram;
	std::unique_ptr<graphics::ShaderProgram> m_texrectDownscaleCopyProgram;
	std::unique_ptr<graphics::ShaderProgram> m_texrectColorAndDepthDownscaleCopyProgram;
};

inline
graphics::CombinerProgram * currentCombiner() {
	return CombinerInfo::get().getCurrent();
}

void Combiner_Init();
void Combiner_Destroy();
graphics::CombinerProgram * Combiner_Compile(CombinerKey key);

#endif