// Copyright (C) 2003-2008 Dolphin Project.

// 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, version 2.0.

// 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 2.0 for more details.

// A copy of the GPL 2.0 should have been included with the program.
// If not, see http://www.gnu.org/licenses/

// Official SVN repository and contact information can be found at
// http://code.google.com/p/dolphin-emu/

#ifndef _PIXELSHADERMANAGER_H
#define _PIXELSHADERMANAGER_H

#include <map>
#include <string>

#include "BPMemory.h"
#include "PixelShader.h"

struct FRAGMENTSHADER
{
	FRAGMENTSHADER() : glprogid(0) { }
	GLuint glprogid; // opengl program id
#if defined(_DEBUG) || defined(DEBUGFAST) 
	std::string strprog;
#endif
};

class PixelShaderCache
{
	struct PSCacheEntry
	{
		FRAGMENTSHADER shader;
		int frameCount;
		PSCacheEntry() : frameCount(0) {}
		~PSCacheEntry() {}
		void Destroy() {
			// printf("Destroying ps %i\n", shader.glprogid);
			glDeleteProgramsARB(1, &shader.glprogid);
			shader.glprogid = 0;
		}
	};

	typedef std::map<PIXELSHADERUID, PSCacheEntry> PSCache;

	static PSCache pshaders;

	static PIXELSHADERUID s_curuid; // the current pixel shader uid (progressively changed as memory is written)

public:
	static void Init();
	static void Cleanup();
	static void Shutdown();

	static FRAGMENTSHADER* GetShader();
	static bool CompilePixelShader(FRAGMENTSHADER& ps, const char* pstrprogram);
};

void SetPSConstant4f(int const_number, float f1, float f2, float f3, float f4);
void SetPSConstant4fv(int const_number, const float *f);

// The non-API dependent parts.
class PixelShaderManager
{

	static void SetPSTextureDims(int texid);
public:
	static void Init();
	static void Shutdown();

	static void SetConstants(); // sets pixel shader constants

	// constant management, should be called after memory is committed
	static void SetColorChanged(int type, int index);
	static void SetAlpha(const AlphaFunc& alpha);
	static void SetDestAlpha(const ConstantAlpha& alpha);
	static void SetTexDims(int texmapid, u32 width, u32 height, u32 wraps, u32 wrapt);
	static void SetZTextureBias(u32 bias);
	static void SetIndTexScaleChanged();
	static void SetIndMatrixChanged(int matrixidx);

	static void SetGenModeChanged();
	static void SetTevCombinerChanged(int id);
	static void SetTevKSelChanged(int id);
	static void SetTevOrderChanged(int id);
	static void SetTevIndirectChanged(int id);
	static void SetZTextureOpChanged();
	static void SetTexturesUsed(u32 nonpow2tex);
	static void SetTexDimsChanged(int texmapid);

	static void SetColorMatrix(const float* pmatrix, const float* pfConstAdd);
	static GLuint GetColorMatrixProgram();
};


#endif