/*****************************************************************************
The Dark Mod GPL Source Code

This file is part of the The Dark Mod Source Code, originally based
on the Doom 3 GPL Source Code as published in 2011.

The Dark Mod Source Code 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 3 of the License,
or (at your option) any later version. For details, see LICENSE.TXT.

Project: The Dark Mod (http://www.thedarkmod.com/)

******************************************************************************/

#ifndef __GLSL_PROGRAM_H__
#define __GLSL_PROGRAM_H__

#include <typeindex>

extern idCVar r_debugGLSL;

class GLSLUniformGroup;

class GLSLProgram {
public:
	using Generator = std::function<void( GLSLProgram* )>;

	GLSLProgram(const char *name, const Generator &generator);
	~GLSLProgram();

	void SetGenerator(const Generator &generator) { this->generator = generator; }

	void Regenerate();

	void Init();
	void Destroy();

	void AttachVertexShader( const char *sourceFile, const idHashMapDict &defines = {} );
	void AttachGeometryShader( const char *sourceFile, const idHashMapDict &defines = {} );
	void AttachFragmentShader( const char *sourceFile, const idHashMapDict &defines = {} );

	void BindAttribLocation( unsigned int location, const char *attribName );
	void BindUniformBlockLocation( unsigned int location, const char *blockName );

	bool Link();
	bool Validate();

	void LoadFromFiles( const char *vertexFile, const char *fragmentFile, const idHashMapDict &defines = {} );

	void Activate();
	static void Deactivate();

	int GetUniformLocation( const char *uniformName ) const;

	template<typename Group>
	Group *GetUniformGroup() {
		GLSLUniformGroup *& group = FindUniformGroup(typeid(Group));
		if( group == nullptr ) {
			Activate();
			group = new Group( this );
		}
		return static_cast<Group*>(group);
	}

	const idStr &GetName() const { return name; }

	static GLSLProgram *GetCurrentProgram() { return currentProgram; }

	GLSLProgram( const GLSLProgram &other ) = delete;
	GLSLProgram & operator=( const GLSLProgram &other ) = delete;
	GLSLProgram( GLSLProgram &&other ) = delete;
	GLSLProgram & operator=( GLSLProgram &&other ) = delete;

private:
	static GLSLProgram *currentProgram;

	idStr name;
	GLuint program;
	Generator generator;

	struct ActiveUniformGroup {
		std::type_index type;
		GLSLUniformGroup *group;
	};
	std::vector<ActiveUniformGroup> uniformGroups;

	GLSLUniformGroup *&FindUniformGroup( const std::type_index &type );
	void LoadAndAttachShader( GLint shaderType, const char *sourceFile, const idHashMapDict &defines );
	GLuint CompileShader( GLint shaderType, const char *sourceFile, const idHashMapDict &defines );
	void SetDefaultUniformBlockBindings();
};

#endif