Allow adding debug annotations to OpenGL objects, as well as defining GPU "zones" to achieve functionality very similar to Tracy, but on the GPU.

rts/Lua/LuaConstPlatform.cpp

@@ -109,6 +109,7 @@ bool LuaConstPlatform::PushEntries(lua_State* L)
 	LuaPushNamedString(L, "hwConfig", Platform::GetHardwareStr());
 	LuaPushNamedNumber(L, "cpuLogicalCores", Threading::GetLogicalCpuCores());
 	LuaPushNamedNumber(L, "cpuPhysicalCores", Threading::GetPhysicalCpuCores());
+	LuaPushNamedString(L, "cpuBrand", Threading::GetCPUBrand());
 
 	LuaPushNamedString(L, "sysInfoHash", Platform::GetSysInfoHash());
 	LuaPushNamedString(L, "macAddrHash", Platform::GetMacAddrHash());

rts/Lua/LuaOpenGL.cpp

@@ -471,6 +471,11 @@ bool LuaOpenGL::PushEntries(lua_State* L)
 	REGISTER_LUA_CFUNC(GetWaterRendering);
 	REGISTER_LUA_CFUNC(GetMapRendering);
 
+
+	REGISTER_LUA_CFUNC(ObjectLabel);
+	REGISTER_LUA_CFUNC(PushDebugGroup);
+	REGISTER_LUA_CFUNC(PopDebugGroup);
+
 	if (canUseShaders)
 		LuaShaders::PushEntries(L);
 
@@ -5661,5 +5666,69 @@ int LuaOpenGL::GetMapRendering(lua_State* L)
 	return 0;
 }
 
+
+/**
+ * @function gl.ObjectLabel labels an object for use with debugging tools
+ * @param objectTypeIdentifier GLenum Specifies the type of object being labeled.
+ * @param objectID GLuint Specifies the name or ID of the object to label.
+ * @param label string A null-terminated string containing the label to be assigned to the object.
+ * @treturn nil
+ */
+int LuaOpenGL::ObjectLabel(lua_State* L) {
+	GLenum identifier = (GLenum)luaL_checkinteger(L, 1);
+	if (identifier != GL_BUFFER && identifier != GL_SHADER && identifier != GL_PROGRAM &&
+		identifier != GL_VERTEX_ARRAY && identifier != GL_QUERY && identifier != GL_PROGRAM_PIPELINE &&
+		identifier != GL_TRANSFORM_FEEDBACK && identifier != GL_TEXTURE && identifier != GL_RENDERBUFFER &&
+		identifier != GL_FRAMEBUFFER) {
+		return 0;
+	}
+    GLuint objectID = (GLuint)luaL_checkinteger(L, 2);
+    const char* label = luaL_checkstring(L, 3);
+
+    // Ensure that identifier is a valid GLenum
+    // The length should be passed as -1 because the strings are null terminated
+    // Check that the GLuint name refers to a valid openGL object
+    // Check that the label is a valid string
+    glObjectLabel(identifier, objectID, -1, label);
+
+    return 0;
+}
+
+
+// https://registry.khronos.org/OpenGL-Refpages/gl4/html/glPushDebugGroup.xhtml
+/**
+ * @function gl.PushDebugGroup pushes a debug marker for nVidia nSight 2024.04, does not seem to work when FBO's are raw bound
+ * @param id GLuint A numeric identifier for the group.
+ * @param message string A human-readable string describing the debug group.
+ * @param source boolean true for GL_DEBUG_SOURCE_APPLICATION, false for GL_DEBUG_SOURCE_THIRD_PARTY. default false
+ * @treturn nil
+ */
+int LuaOpenGL::PushDebugGroup(lua_State* L) {
+	GLuint id = (GLuint)luaL_checkinteger(L, 1);
+	const char* message = luaL_checkstring(L, 2);
+	bool source = luaL_optboolean(L, 3, false);
+
+	GLint maxLength;
+	glGetIntegerv(GL_MAX_DEBUG_MESSAGE_LENGTH, &maxLength);
+
+	if (strlen(message) >= maxLength) {
+		luaL_error(L, "Message length exceeds GL_MAX_DEBUG_MESSAGE_LENGTH");
+		return 0;
+	}
+
+	glPushDebugGroup((source ? GL_DEBUG_SOURCE_APPLICATION: GL_DEBUG_SOURCE_THIRD_PARTY) , id, -1, message);
+
+	return 0;
+}
+
+/**
+ * @function gl.PopDebugGroup
+ * @treturn nil
+ */
+int LuaOpenGL::PopDebugGroup(lua_State* L) {
+    glPopDebugGroup();
+    return 0;
+}
+
 /******************************************************************************/
 /******************************************************************************/

rts/Lua/LuaOpenGL.h

@@ -353,6 +353,10 @@ class LuaOpenGL {
 		static int GetSun(lua_State* L);
 		static int GetWaterRendering(lua_State* L);
 		static int GetMapRendering(lua_State* L);
+
+		static int ObjectLabel(lua_State* L);
+		static int PushDebugGroup(lua_State* L);
+		static int PopDebugGroup(lua_State* L);
 };
 
 inline void LuaOpenGL::InitMatrixState(lua_State* L, const char* fn) {

rts/Lua/LuaShaders.cpp

@@ -555,6 +555,7 @@ GLint LuaShaders::GetUniformLocation(LuaShaders::Program* p, const char* name)
  * @function gl.CreateShader ( table shaderParams )
  * @tparam table shaderParams
  * @treturn number shaderID
+ * @treturn number glProgID
  *
  *    ({[ vertex   = "glsl code" ,]
  *      [ tcs      = "glsl code" ,]
@@ -738,7 +739,8 @@ int LuaShaders::CreateShader(lua_State* L)
 
 	// note: index, not raw ID
 	lua_pushnumber(L, shaders.AddProgram(p));
-	return 1;
+	lua_pushnumber(L, (int) prog);
+	return 2;
 }
 
 

rts/Lua/LuaVBO.cpp

@@ -64,7 +64,8 @@ bool LuaVBOs::PushEntries(lua_State* L)
 		"UnbindBufferRange", &LuaVBOImpl::UnbindBufferRange,
 
 		"DumpDefinition", &LuaVBOImpl::DumpDefinition,
-		"GetBufferSize", &LuaVBOImpl::GetBufferSize
+		"GetBufferSize", &LuaVBOImpl::GetBufferSize,
+		"GetID", &LuaVBOImpl::GetID
 	);
 
 	gl.set("VBO", sol::lua_nil); // don't want this to be accessible directly without gl.GetVBO

rts/Lua/LuaVBOImpl.cpp

@@ -1445,6 +1445,17 @@ void LuaVBOImpl::DumpDefinition()
 	LOG("%s", ss.str().c_str());
 }
 
+/*** Gets the OpenGL Buffer ID
+ * 
+ * @function VBO:GetID
+ * @treturn number buffer ID
+ */
+uint32_t LuaVBOImpl::GetID()
+{
+	VBOExistenceCheck(vbo, __func__);
+	return vbo->GetId();
+}
+
 void LuaVBOImpl::AllocGLBuffer(size_t byteSize)
 {
 	if (defTarget == GL_UNIFORM_BUFFER && bufferSizeInBytes > UBO_SAFE_SIZE_BYTES) {

rts/Lua/LuaVBOImpl.h

@@ -52,6 +52,8 @@ class LuaVBOImpl {
 	int UnbindBufferRange(const GLuint index, const sol::optional<int> elemOffsetOpt, const sol::optional<int> elemCountOpt, const sol::optional<GLenum> targetOpt);
 
 	void DumpDefinition();
+	uint32_t GetID();
+
 public:
 	static bool Supported(GLenum target);
 private:

rts/System/Platform/CpuID.cpp

@@ -150,6 +150,7 @@ namespace springproc {
 
 		for (int group = 0; group < system.num_cpu_types; ++group) {
 			cpu_id_t cpu_id = system.cpu_types[group];
+			cpuid_brand = std::string(cpu_id.brand_str);
 			switch(cpu_id.purpose) {
 			case PURPOSE_GENERAL:
 			case PURPOSE_PERFORMANCE:

rts/System/Platform/CpuID.h

@@ -10,6 +10,7 @@
 #endif
 
 #include <cstdint>
+#include <string>
 
 namespace springproc {
 	_noinline void ExecCPUID(unsigned int* a, unsigned int* b, unsigned int* c, unsigned int* d);
@@ -44,6 +45,7 @@ namespace springproc {
 		    cores. */
 		int GetNumPhysicalCores() const { return numPhysicalCores; }
 		int GetNumLogicalCores() const { return numLogicalCores; }
+		std::string GetCPUBrand() const { return cpuid_brand; }
 
 		/** Total number of physical processor dies in the system. */
 		int GetTotalNumPackages() const { return totalNumPackages; }
@@ -71,6 +73,8 @@ namespace springproc {
 		uint64_t affinityMaskOfPackages[MAX_PROCESSORS];
 		uint64_t availableProceesorAffinityMask;
 
+		std::string cpuid_brand;
+
 		////////////////////////
 		// Intel specific fields
 

rts/System/Platform/Threading.cpp

@@ -213,6 +213,10 @@ namespace Threading {
 		return springproc::CPUID::GetInstance().GetNumPhysicalCores();
 	}
 
+	std::string GetCPUBrand() {
+		return springproc::CPUID::GetInstance().GetCPUBrand();
+	}
+
 	bool HasHyperThreading() { return (GetLogicalCpuCores() > GetPhysicalCpuCores()); }
 
 

rts/System/Platform/Threading.h

@@ -124,6 +124,8 @@ namespace Threading {
 	int GetLogicalCpuCores();  /// physical + hyperthreading
 	bool HasHyperThreading();
 
+	std::string GetCPUBrand();
+
 	/**
 	 * Inform the OS kernel that we are a cpu-intensive task
 	 */

rts/lib/headlessStubs/glstub.c

@@ -605,6 +605,10 @@ GLAPI void APIENTRY glDepthRange(GLdouble nearVal, GLdouble farVal) {}
 GLAPI void APIENTRY  glBeginConditionalRender(GLuint id, GLenum mode) {}
 GLAPI void APIENTRY  glEndConditionalRender(void) {}
 
+GLAPI void APIENTRY  glObjectLabel(GLenum identifier, GLuint name, GLsizei length, const GLchar *label) {}
+GLAPI void APIENTRY  glPushDebugGroup(GLenum source, GLuint id, GLsizei length, const GLchar *message) {}
+GLAPI void APIENTRY  glPopDebugGroup() {}
+
 #ifdef __cplusplus
 } // extern "C"
 #endif