Differential D2457 Diff 8102 source/gameengine/Ketsji/BL_Shader.h

Changeset View

Standalone View

source/gameengine/Ketsji/BL_Shader.h

Context not available.
	#define __BL_SHADER_H__	#define __BL_SHADER_H__

	#include "EXP_PyObjectPlus.h"	#include "EXP_PyObjectPlus.h"
	#include "BL_Material.h"	#include "RAS_Shader.h"
	#include "BL_Texture.h"
	#include "MT_Matrix4x4.h"
	#include "MT_Matrix3x3.h"
	#include "MT_Tuple2.h"
	#include "MT_Tuple3.h"
	#include "MT_Tuple4.h"

	/**
	* BL_Sampler
	* Sampler access
	*/
	class BL_Sampler
	{
	public:
	BL_Sampler()
	:
	mLoc(-1)
	{
	}

	int mLoc; // Sampler location	class RAS_MeshSlot;

	#ifdef WITH_CXX_GUARDEDALLOC	class BL_Shader : public PyObjectPlus, public virtual RAS_Shader
	MEM_CXX_CLASS_ALLOC_FUNCS("GE:BL_Sampler")
	#endif
	};

	/**
	* BL_Uniform
	* uniform storage
	*/
	class BL_Uniform
	{	{
	private:
	int mLoc; // Uniform location
	void *mData; // Memory allocated for variable
	bool mDirty; // Caching variable
	int mType; // Enum UniformTypes
	bool mTranspose; // Transpose matrices
	const int mDataLen; // Length of our data
	public:	public:
	BL_Uniform(int data_size);	enum CallbacksType {
	~BL_Uniform();	CALLBACKS_BIND = 0,
		CALLBACKS_OBJECT,
	enum UniformTypes {	CALLBACKS_MAX
	UNI_NONE = 0,
	UNI_INT,
	UNI_FLOAT,
	UNI_INT2,
	UNI_FLOAT2,
	UNI_INT3,
	UNI_FLOAT3,
	UNI_INT4,
	UNI_FLOAT4,
	UNI_MAT3,
	UNI_MAT4,
	UNI_FLOAT_EYE,
	UNI_MAX
	};	};

	bool Apply(class BL_Shader *shader);
	void SetData(int location, int type, bool transpose = false);
	int GetLocation() { return mLoc; }
	void *getData() { return mData; }

	#ifdef WITH_CXX_GUARDEDALLOC
	MEM_CXX_CLASS_ALLOC_FUNCS("GE:BL_Uniform")
	#endif
	};

	/**
	* BL_DefUniform
	* pre defined uniform storage
	*/
	class BL_DefUniform
	{
	public:
	BL_DefUniform()
	:
	mType(0),
	mLoc(0),
	mFlag(0)
	{
	}

	int mType;
	int mLoc;
	unsigned int mFlag;

	#ifdef WITH_CXX_GUARDEDALLOC
	MEM_CXX_CLASS_ALLOC_FUNCS("GE:BL_DefUniform")
	#endif
	};

	/**
	* BL_Shader
	* shader access
	*/
	class BL_Shader : public PyObjectPlus
	{
	Py_Header
	private:	private:
	typedef std::vector<BL_Uniform *> BL_UniformVec;	Py_Header
	typedef std::vector<BL_DefUniform *> BL_UniformVecDef;

	unsigned int mShader; // Shader object
	int mPass; // 1.. unused
	bool mOk; // Valid and ok
	bool mUse;
	//BL_Sampler mSampler[MAXTEX]; // Number of samplers
	int mAttr; // Tangent attribute
	const char *vertProg; // Vertex program string
	const char *fragProg; // Fragment program string
	bool mError;
	bool mDirty;

	// Stored uniform variables
	BL_UniformVec mUniforms;
	BL_UniformVecDef mPreDef;

	// Compiles and links the shader
	bool LinkProgram();

	// search by location
	BL_Uniform *FindUniform(const int location);

	// clears uniform data	#ifdef WITH_PYTHON
	void ClearUniforms();	PyObject *m_callbacks[CALLBACKS_MAX];
		#endif // WITH_PYTHON

	public:	public:
	BL_Shader();	BL_Shader();
	virtual ~BL_Shader();	virtual ~BL_Shader();

	// Unused for now tangent is set as tex coords	#ifdef WITH_PYTHON
	enum AttribTypes {	PyObject *GetCallbacks(CallbacksType type);
	SHD_TANGENT = 1	void SetCallbacks(CallbacksType type, PyObject *callbacks);
	};	#endif // WITH_PYTHON

	enum GenType {	virtual void SetProg(bool enable);
	MODELVIEWMATRIX,
	MODELVIEWMATRIX_TRANSPOSE,
	MODELVIEWMATRIX_INVERSE,
	MODELVIEWMATRIX_INVERSETRANSPOSE,
	MODELMATRIX,
	MODELMATRIX_TRANSPOSE,
	MODELMATRIX_INVERSE,
	MODELMATRIX_INVERSETRANSPOSE,
	VIEWMATRIX,
	VIEWMATRIX_TRANSPOSE,
	VIEWMATRIX_INVERSE,
	VIEWMATRIX_INVERSETRANSPOSE,
	CAM_POS,
	CONSTANT_TIMER
	};

	const char *GetVertPtr();	/** Update the uniform shader for the current rendered mesh slot.
	const char *GetFragPtr();	* The python callbacks are executed in this function and at the end
	void SetVertPtr(char *vert);	* RAS_Shader::Update(rasty, mat) is called.
	void SetFragPtr(char *frag);	*/
	int getNumPass() { return mPass; }	void Update(RAS_IRasterizer rasty, RAS_MeshSlot ms);
	bool GetError() { return mError; }

	//const BL_Sampler *GetSampler(int i);
	void SetSampler(int loc, int unit);
	bool Ok() const;
	unsigned int GetProg();
	void SetProg(bool enable);
	int GetAttribute() { return mAttr; }

	// Apply methods : sets colected uniforms
	void ApplyShader();
	void UnloadShader();

	// Update predefined uniforms each render call
	void Update(const class RAS_MeshSlot &ms, class RAS_IRasterizer *rasty);

	// Set sampler units (copied)
	//void InitializeSampler(int unit, BL_Texture *texture);
	void SetUniformfv(int location, int type, float *param, int size, bool transpose = false);
	void SetUniformiv(int location, int type, int *param, int size, bool transpose = false);
	int GetAttribLocation(const char *name);
	void BindAttribute(const char *attr, int loc);
	int GetUniformLocation(const char *name);
	void SetUniform(int uniform, const MT_Tuple2 &vec);
	void SetUniform(int uniform, const MT_Tuple3 &vec);
	void SetUniform(int uniform, const MT_Tuple4 &vec);
	void SetUniform(int uniform, const MT_Matrix4x4 &vec, bool transpose = false);
	void SetUniform(int uniform, const MT_Matrix3x3 &vec, bool transpose = false);
	void SetUniform(int uniform, const float &val);
	void SetUniform(int uniform, const float *val, int len);
	void SetUniform(int uniform, const int *val, int len);
	void SetUniform(int uniform, const unsigned int &val);
	void SetUniform(int uniform, const int val);

	// Python interface	// Python interface
	#ifdef WITH_PYTHON	#ifdef WITH_PYTHON
	virtual PyObject *py_repr()	virtual PyObject *py_repr()
	{	{
	return PyUnicode_FromFormat("BL_Shader\n\tvertex shader:%s\n\n\tfragment shader%s\n\n", vertProg, fragProg);	return PyUnicode_FromFormat("BL_Shader\n\tvertex shader:%s\n\n\tfragment shader%s\n\n", m_progs[VERTEX_PROGRAM].c_str(), m_progs[FRAGMENT_PROGRAM].c_str());
	}	}

		static PyObject pyattr_get_enabled(void self_v, const KX_PYATTRIBUTE_DEF *attrdef);
		static int pyattr_set_enabled(void self_v, const KX_PYATTRIBUTE_DEF attrdef, PyObject *value);
		static PyObject pyattr_get_callbacks(void self_v, const KX_PYATTRIBUTE_DEF *attrdef);
		static int pyattr_set_callbacks(void self_v, const KX_PYATTRIBUTE_DEF attrdef, PyObject *value);

	// -----------------------------------	// -----------------------------------
	KX_PYMETHOD_DOC(BL_Shader, setSource);	KX_PYMETHOD_DOC(BL_Shader, setSource);
		KX_PYMETHOD_DOC(BL_Shader, setSourceList);
	KX_PYMETHOD_DOC(BL_Shader, delSource);	KX_PYMETHOD_DOC(BL_Shader, delSource);
	KX_PYMETHOD_DOC(BL_Shader, getVertexProg);	KX_PYMETHOD_DOC(BL_Shader, getVertexProg);
	KX_PYMETHOD_DOC(BL_Shader, getFragmentProg);	KX_PYMETHOD_DOC(BL_Shader, getFragmentProg);
Context not available.