Added ServerEntity.lookup_bone()

Author: Ayuto

addons/source-python/packages/source-python/studio/constants.py

@@ -16,6 +16,7 @@
 # Source.Python Imports
 #   Studio
 from _studio._constants import INVALID_ATTACHMENT_INDEX
+from _studio._constants import INVALID_BONE_INDEX
 from _studio._constants import ModelType
 from _studio._constants import STUDIO_NONE
 from _studio._constants import STUDIO_RENDER
@@ -35,6 +36,7 @@
 # >> ALL DECLARATION
 # =============================================================================
 __all__ = ('INVALID_ATTACHMENT_INDEX',
+           'INVALID_BONE_INDEX',
            'ModelType',
            'StudioFlags',
            )

src/core/modules/entities/entities.cpp

@@ -107,4 +107,20 @@ int ServerEntityExt::lookup_attachment(IServerEntity* pEntity, const char* name)
 	}
 
 	return INVALID_ATTACHMENT_INDEX;
+}
+
+int ServerEntityExt::lookup_bone(IServerEntity* pEntity, const char* name)
+{
+	studiohdr_t* header = get_model_header(pEntity);
+	if (!header)
+		return INVALID_BONE_INDEX;
+
+	for (int i=0; i < header->numbones; ++i)
+	{
+		mstudiobone_t* bone = header->pBone(i);
+		if (V_strcmp(bone->pszName(), name) == 0)
+			return i;
+	}
+
+	return INVALID_BONE_INDEX;
 }

src/core/modules/entities/entities.h

@@ -52,7 +52,8 @@ class ServerEntityExt
 	static void SetModelName(IServerEntity *pServerEntity, const char *szModelName);
 	static MDLHandle_t get_model_handle(IServerEntity *pServerEntity);
 	static studiohdr_t* get_model_header(IServerEntity *pServerEntity);
-	static int ServerEntityExt::lookup_attachment(IServerEntity* pEntity, const char* name);
+	static int lookup_attachment(IServerEntity* pEntity, const char* name);
+	static int lookup_bone(IServerEntity* pEntity, const char* name);
 };
 
 

src/core/modules/entities/entities_wrap.cpp

@@ -232,6 +232,14 @@ void export_server_entity(scope _entities)
 			":rtype: int"
 		)
 
+		.def("lookup_bone",
+			&ServerEntityExt::lookup_bone,
+			"Return the bone index matching the given name.\n\n"
+			":param str name:\n"
+			"	The name of the bone.\n"
+			":rtype: int"
+		)
+
 		ADD_MEM_TOOLS(IServerEntity)
 	;
 }

src/core/modules/studio/studio.h

@@ -47,6 +47,7 @@ extern IMDLCache *modelcache;
 // DEFINES
 //-----------------------------------------------------------------------------
 #define INVALID_ATTACHMENT_INDEX -1
+#define INVALID_BONE_INDEX -1
 
 
 //-----------------------------------------------------------------------------

src/core/modules/studio/studio_constants_wrap.cpp

@@ -47,6 +47,7 @@ DECLARE_SP_SUBMODULE(_studio, _constants)
 	export_studio_flags(_constants);
 
 	_constants.attr("INVALID_ATTACHMENT_INDEX") = INVALID_ATTACHMENT_INDEX;
+	_constants.attr("INVALID_BONE_INDEX") = INVALID_BONE_INDEX;
 }