Shaders management functions.

4 files changed, 172 insertions, 18 deletions

diff --git a/src/core.c b/src/core.c
index 92eb5bd..bd1ffe0 100644
--- a/src/core.c
+++ b/src/core.c
@@ -553,7 +553,6 @@ int lcoreLoadBuffer( lua_State *L ) {
 	int type = luaL_checkinteger( L, 2 );
 
 	Buffer buffer = { 0 };
-	// Buffer *buffer = lua_newuserdata( L, sizeof( Buffer ) );
 	int len = uluaGetTableLenIndex( L, 1 );
 
 	switch ( type ) {
diff --git a/src/lua_core.c b/src/lua_core.c
index 10f48b8..7c447f6 100644
--- a/src/lua_core.c
+++ b/src/lua_core.c
@@ -2285,6 +2285,17 @@ void luaRegister() {
 	assingGlobalFunction( "rlFramebufferAttach", lrlglFramebufferAttach );
 	assingGlobalFunction( "rlFramebufferComplete", lrlglFramebufferComplete );
 	assingGlobalFunction( "rlUnloadFramebuffer", lrlglUnloadFramebuffer );
+		/* Shaders management */
+	assingGlobalFunction( "rlLoadShaderCode", lrlglLoadShaderCode );
+	assingGlobalFunction( "rlCompileShader", lrlglCompileShader );
+	assingGlobalFunction( "rlLoadShaderProgram", lrlglLoadShaderProgram );
+	assingGlobalFunction( "rlUnloadShaderProgram", lrlglUnloadShaderProgram );
+	assingGlobalFunction( "rlGetLocationUniform", lrlglGetLocationUniform );
+	assingGlobalFunction( "rlGetLocationAttrib", lrlglGetLocationAttrib );
+	assingGlobalFunction( "rlSetUniform", lrlglSetUniform );
+	assingGlobalFunction( "rlSetUniformMatrix", lrlglSetUniformMatrix );
+	assingGlobalFunction( "rlSetUniformSampler", lrlglSetUniformSampler );
+	assingGlobalFunction( "rlSetShader", lrlglSetShader );
 		/* Matrix state management. */
 	assingGlobalFunction( "rlGetMatrixModelview", lrlglGetMatrixModelview );
 	assingGlobalFunction( "rlGetMatrixProjection", lrlglGetMatrixProjection );
diff --git a/src/rlgl.c b/src/rlgl.c
index 8fba3a1..4a13c81 100644
--- a/src/rlgl.c
+++ b/src/rlgl.c
@@ -928,7 +928,7 @@ Load a vertex buffer attribute
 - Success return int
 */
 int lrlglLoadVertexBuffer( lua_State *L ) {
-	Buffer *buffer = luaL_checkudata( L, 1, "Buffer" );
+	Buffer *buffer = uluaGetBuffer( L, 1 );
 	bool dynamic = uluaGetBoolean( L, 2 );
 
 	lua_pushinteger( L, rlLoadVertexBuffer( buffer->data, buffer->size, dynamic ) );
@@ -944,7 +944,7 @@ Load a new attributes element buffer
 - Success return int
 */
 int lrlglLoadVertexBufferElement( lua_State *L ) {
-	Buffer *buffer = luaL_checkudata( L, 1, "Buffer" );
+	Buffer *buffer = uluaGetBuffer( L, 1 );
 	bool dynamic = uluaGetBoolean( L, 2 );
 
 	lua_pushinteger( L, rlLoadVertexBufferElement( buffer->data, buffer->size, dynamic ) );
@@ -959,7 +959,7 @@ Update GPU buffer with new data
 */
 int lrlglUpdateVertexBuffer( lua_State *L ) {
 	int bufferId = luaL_checkinteger( L, 1 );
-	Buffer *buffer = luaL_checkudata( L, 2, "Buffer" );
+	Buffer *buffer = uluaGetBuffer( L, 2 );
 	int offset = luaL_checkinteger( L, 3 );
 
 	rlUpdateVertexBuffer( bufferId, buffer->data, buffer->size, offset );
@@ -974,7 +974,7 @@ Update vertex buffer elements with new data
 */
 int lrlglUpdateVertexBufferElements( lua_State *L ) {
 	int bufferId = luaL_checkinteger( L, 1 );
-	Buffer *buffer = luaL_checkudata( L, 2, "Buffer" );
+	Buffer *buffer = uluaGetBuffer( L, 2 );
 	int offset = luaL_checkinteger( L, 3 );
 
 	rlUpdateVertexBufferElements( bufferId, buffer->data, buffer->size, offset );
@@ -1083,7 +1083,7 @@ Draw vertex array elements
 int lrlglDrawVertexArrayElements( lua_State *L ) {
 	int offset = luaL_checkinteger( L, 1 );
 	int count = luaL_checkinteger( L, 2 );
-	Buffer *buffer = luaL_checkudata( L, 3, "Buffer" );
+	Buffer *buffer = uluaGetBuffer( L, 3 );
 
 	rlDrawVertexArrayElements( offset, count, buffer->data );
 
@@ -1113,7 +1113,7 @@ Draw vertex array elements instanced
 int lrlglDrawVertexArrayElementsInstanced( lua_State *L ) {
 	int offset = luaL_checkinteger( L, 1 );
 	int count = luaL_checkinteger( L, 2 );
-	Buffer *buffer = luaL_checkudata( L, 3, "Buffer" );
+	Buffer *buffer = uluaGetBuffer( L, 3 );
 	int instances = luaL_checkinteger( L, 4 );
 
 	rlDrawVertexArrayElementsInstanced( offset, count, buffer->data, instances );
@@ -1238,23 +1238,160 @@ int lrlglUnloadFramebuffer( lua_State *L ) {
 */
 
 /*
-> success = RL.rlLoadShaderCode( string vsCode, string fsCode )
+> shaderId = RL.rlLoadShaderCode( string vsCode, string fsCode )
 
 Load shader from code strings
 
-- Failure return nil
 - Success return int
 */
-// int lrlglUnloadFramebuffer( lua_State *L ) {
-// 	if ( !lua_isstring( L, 1 ) || !lua_isstring( L, 2 ) ) {
-// 		TraceLog( state->logLevelInvalid, "%s", "Bad call of function. RL.rlLoadShaderCode( string vsCode, string fsCode )" );
-// 		lua_pushnil( L );
-// 		return 1;
-// 	}
-// 	lua_pushinteger( L, rlLoadShaderCode( luaL_checkstring( L, 1 ), luaL_checkstring( L, 2 ) ) );
+int lrlglLoadShaderCode( lua_State *L ) {
+	lua_pushinteger( L, rlLoadShaderCode( luaL_checkstring( L, 1 ), luaL_checkstring( L, 2 ) ) );
 
-// 	return 1;
-// }
+	return 1;
+}
+
+/*
+> shaderId = RL.rlCompileShader( string shaderCode, int type )
+
+Compile custom shader and return shader id (type: RL_VERTEX_SHADER, RL_FRAGMENT_SHADER, RL_COMPUTE_SHADER)
+
+- Success return int
+*/
+int lrlglCompileShader( lua_State *L ) {
+	int type = luaL_checkinteger( L, 2 );
+
+	lua_pushinteger( L, rlCompileShader( luaL_checkstring( L, 1 ), type ) );
+
+	return 1;
+}
+
+/*
+> shaderProgramId = RL.rlLoadShaderProgram( int vShaderId, int fShaderId )
+
+Load custom shader program
+
+- Success return int
+*/
+int lrlglLoadShaderProgram( lua_State *L ) {
+	unsigned int vShaderId = (unsigned int)luaL_checkinteger( L, 1 );
+	unsigned int fShaderId = (unsigned int)luaL_checkinteger( L, 2 );
+
+	lua_pushinteger( L, rlLoadShaderProgram( vShaderId, fShaderId ) );
+
+	return 1;
+}
+
+/*
+> RL.rlUnloadShaderProgram( int id )
+
+Unload shader program
+*/
+int lrlglUnloadShaderProgram( lua_State *L ) {
+	unsigned int id = (unsigned int)luaL_checkinteger( L, 1 );
+
+	rlUnloadShaderProgram( id );
+
+	return 0;
+}
+
+/*
+> location = RL.rlGetLocationUniform( int shaderId, string uniformName )
+
+Get shader location uniform
+
+- Success return int
+*/
+int lrlglGetLocationUniform( lua_State *L ) {
+	unsigned int shaderId = (unsigned int)luaL_checkinteger( L, 1 );
+
+	lua_pushinteger( L, rlGetLocationUniform( shaderId, luaL_checkstring( L, 2 ) ) );
+
+	return 1;
+}
+
+/*
+> location = RL.rlGetLocationAttrib( int shaderId, string attribName )
+
+Get shader location attribute
+
+- Success return int
+*/
+int lrlglGetLocationAttrib( lua_State *L ) {
+	unsigned int shaderId = (unsigned int)luaL_checkinteger( L, 1 );
+
+	lua_pushinteger( L, rlGetLocationAttrib( shaderId, luaL_checkstring( L, 2 ) ) );
+
+	return 1;
+}
+
+/*
+> RL.rlSetUniform( int locIndex, Buffer value, int uniformType, int count )
+
+Set shader value uniform
+*/
+int lrlglSetUniform( lua_State *L ) {
+	int locIndex = luaL_checkinteger( L, 1 );
+	Buffer *value = uluaGetBuffer( L, 2 );
+	int uniformType = luaL_checkinteger( L, 3 );
+	int count = luaL_checkinteger( L, 4 );
+
+	rlSetUniform( locIndex, value->data, uniformType, count );
+
+	return 0;
+}
+
+/*
+> RL.rlSetUniformMatrix( int locIndex, Matrix mat )
+
+Set shader value matrix
+*/
+int lrlglSetUniformMatrix( lua_State *L ) {
+	int locIndex = luaL_checkinteger( L, 1 );
+	Matrix mat = uluaGetMatrixIndex( L, 2 );
+
+	rlSetUniformMatrix( locIndex, mat );
+
+	return 0;
+}
+
+/*
+> RL.rlSetUniformSampler( int locIndex, int textureId )
+
+Set shader value sampler
+*/
+int lrlglSetUniformSampler( lua_State *L ) {
+	int locIndex = luaL_checkinteger( L, 1 );
+	unsigned int textureId = (unsigned int)luaL_checkinteger( L, 2 );
+
+	rlSetUniformSampler( locIndex, textureId );
+
+	return 0;
+}
+
+/*
+> RL.rlSetShader( int id, int{} locs )
+
+Set shader currently active (id and locations)
+*/
+int lrlglSetShader( lua_State *L ) {
+	unsigned int id = (unsigned int)luaL_checkinteger( L, 1 );
+
+	int t = 2, i = 0;
+    lua_pushnil( L );
+
+	while ( lua_next( L, t ) != 0 ) {
+        if ( lua_isnumber( L, -1 ) ) {
+			if ( i < RL_MAX_SHADER_LOCATIONS ) {
+				state->RLGLcurrentShaderLocs[i] = lua_tointeger( L, -1 );
+			}
+        }
+        i++;
+        lua_pop( L, 1 );
+    }
+	rlSetShader( id, state->RLGLcurrentShaderLocs );
+
+	return 0;
+}
 
 /*
 ## RLGL - Matrix state management
diff --git a/src/state.c b/src/state.c
index b96b557..228cbce 100644
--- a/src/state.c
+++ b/src/state.c
@@ -36,6 +36,12 @@ bool stateInit( int argn, const char **argc, const char *exePath ) {
 	}
 	state->defaultFont = GetFontDefault();
 	state->defaultMaterial = LoadMaterialDefault();
+	state->RLGLcurrentShaderLocs = malloc( RL_MAX_SHADER_LOCATIONS * sizeof( int ) );
+	int *defaultShaderLocs = rlGetShaderLocsDefault();
+
+	for ( int i = 0; i < RL_MAX_SHADER_LOCATIONS; i++ ) {
+		state->RLGLcurrentShaderLocs[i] = defaultShaderLocs[i];
+	}
 
 	return state->run;
 }
@@ -57,5 +63,6 @@ void stateFree() {
 		CloseWindow();
 	}
 	free( state->exePath );
+	free( state->RLGLcurrentShaderLocs );
 	free( state );
 }