From c7b7975482bb33e7fb352c5759920748b6d3769b Mon Sep 17 00:00:00 2001
From: jussi
Date: Wed, 17 Jul 2024 00:06:43 +0300
Subject: DrawGridEx.

---
 src/lua_core.c |  1 +
 src/models.c   | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+)

(limited to 'src')

diff --git a/src/lua_core.c b/src/lua_core.c
index 0955f77..2d5ddcc 100644
--- a/src/lua_core.c
+++ b/src/lua_core.c
@@ -1726,6 +1726,7 @@ void luaRegister() {
 	assingGlobalFunction( "DrawQuad3DTexture", lmodelDrawQuad3DTexture );
 	assingGlobalFunction( "DrawRay", lmodelsDrawRay );
 	assingGlobalFunction( "DrawGrid", lmodelsDrawGrid );
+	assingGlobalFunction( "DrawGridEx", lmodelsDrawGridEx );
 		/* Model management functions. */
 	assingGlobalFunction( "LoadModel", lmodelsLoadModel );
 	assingGlobalFunction( "LoadModelFromMesh", lmodelsLoadModelFromMesh );
diff --git a/src/models.c b/src/models.c
index 61b9c2b..652a6c0 100644
--- a/src/models.c
+++ b/src/models.c
@@ -496,6 +496,58 @@ int lmodelsDrawGrid( lua_State* L ) {
 	return 0;
 }
 
+/*
+> RL.DrawGridEx( Vector2 slices, Vector2 spacing, Matrix transform, Color color, Vector2|nil divider, Color|nil dividerColor )
+
+Draw a grid with extended parameters. Optionally you can define divider with different color for every n slices
+*/
+int lmodelsDrawGridEx( lua_State* L ) {
+	Vector2 slices = uluaGetVector2( L, 1 );
+	Vector2 spacing = uluaGetVector2( L, 2 );
+	Matrix transform = uluaGetMatrix( L, 3 );
+	Color color = uluaGetColor( L, 4 );
+
+	Vector2 divider = { 0, 0 };
+	Color divColor = WHITE;
+
+	if ( !lua_isnil( L, 5 ) && !lua_isnone( L, 5 ) ) {
+		divider = uluaGetVector2( L, 5 );
+	}
+	if ( !lua_isnil( L, 6 ) && !lua_isnone( L, 6 ) ) {
+		divColor = uluaGetColor( L, 6 );
+	}
+
+	rlPushMatrix();
+		rlMultMatrixf( MatrixToFloat( transform ) );
+
+		rlBegin( RL_LINES );
+			for ( int x = 0; x < (int)slices.x + 1; x++ ) {
+				if ( 0 < x && x < slices.x && 0 < divider.x && x % (int)divider.x == 0 ) {
+					rlColor4ub( divColor.r, divColor.g, divColor.b, divColor.a );
+				}
+				else {
+					rlColor4ub( color.r, color.g, color.b, color.a );
+				}
+
+				rlVertex3f( spacing.x * x, 0, 0 );
+				rlVertex3f( spacing.x * x, 0, slices.y * spacing.y );
+			}
+			for ( int y = 0; y < (int)slices.y + 1; y++ ) {
+				if ( 0 < y && y < slices.y && 0 < divider.y && y % (int)divider.y == 0 ) {
+					rlColor4ub( divColor.r, divColor.g, divColor.b, divColor.a );
+				}
+				else {
+					rlColor4ub( color.r, color.g, color.b, color.a );
+				}
+				rlVertex3f( 0, 0, spacing.y * y );
+				rlVertex3f( slices.x * spacing.x, 0, spacing.y * y );
+			}
+		rlEnd();
+	rlPopMatrix();
+
+	return 0;
+}
+
 /*
 ## Models - Model management functions
 */
-- 
cgit v1.2.3