[Bf-blender-cvs] SVN commit: /data/svn/bf-blender [53124] trunk/blender/source/blender/ makesrna/intern/rna_nodetree.c: Class method 'get_node_type' for node RNA types.

[Lukas Toenne]

Revision: 53124
          http://projects.blender.org/scm/viewvc.php?view=rev&root=bf-blender&revision=53124
Author:   lukastoenne
Date:     2012-12-18 14:49:32 +0000 (Tue, 18 Dec 2012)
Log Message:
-----------
Class method 'get_node_type' for node RNA types. This can be used to map RNA types and Python classes to the associated node type by means of the node type identifier.

Modified Paths:
--------------
    trunk/blender/source/blender/makesrna/intern/rna_nodetree.c

Modified: trunk/blender/source/blender/makesrna/intern/rna_nodetree.c
===================================================================
--- trunk/blender/source/blender/makesrna/intern/rna_nodetree.c	2012-12-18 14:46:00 UTC (rev 53123)
+++ trunk/blender/source/blender/makesrna/intern/rna_nodetree.c	2012-12-18 14:49:32 UTC (rev 53124)
@@ -271,6 +271,24 @@
 	return BLI_sprintfN("nodes[\"%s\"]", node->name);
 }
 
+/* define a get_type function for each node type */
+#define DEF_NODE_GET_NODE_TYPE(name, enum_name) \
+static const char *rna_##name##_get_node_type() \
+{ \
+	return enum_name; \
+}
+
+#define DefNode(Category, ID, DefFunc, EnumName, StructName, UIName, UIDesc) \
+DEF_NODE_GET_NODE_TYPE(Category##StructName, EnumName)
+
+#include "rna_nodetree_types.h"
+
+/* some nodes not included in rna_nodetree_types.h */
+DEF_NODE_GET_NODE_TYPE(NodeGroup, "GROUP")
+DEF_NODE_GET_NODE_TYPE(NodeFrame, "FRAME")
+DEF_NODE_GET_NODE_TYPE(NodeReroute, "REROUTE")
+
+
 static StructRNA *rna_NodeSocket_refine(PointerRNA *ptr)
 {
 	bNodeSocket *sock = (bNodeSocket *)ptr->data;
@@ -4906,12 +4924,20 @@
 	rna_def_texture_nodetree_api(brna, prop);
 }
 
-static void define_specific_node(BlenderRNA *brna, int id, void (*func)(StructRNA *))
+static void define_specific_node(BlenderRNA *brna, int id, void (*def_func)(StructRNA *), const char *get_node_type_func)
 {
 	StructRNA *srna = def_node(brna, id);
-
-	if (func)
-		func(srna);
+	FunctionRNA *func;
+	PropertyRNA *parm;
+	
+	if (def_func)
+		def_func(srna);
+	
+	func = RNA_def_function(srna, "get_node_type", get_node_type_func);
+	RNA_def_function_ui_description(func, "Get the identifier of the node type");
+	RNA_def_function_flag(func, FUNC_NO_SELF);
+	parm = RNA_def_string(func, "result", "", 0, "Result", "");
+	RNA_def_function_return(func, parm);
 }
 
 void RNA_def_nodetree(BlenderRNA *brna)
@@ -4942,13 +4968,13 @@
 	rna_def_shader_nodetree(brna);
 	rna_def_texture_nodetree(brna);
 	#define DefNode(Category, ID, DefFunc, EnumName, StructName, UIName, UIDesc) \
-		define_specific_node(brna, ID, DefFunc);
+		define_specific_node(brna, ID, DefFunc, "rna_" STRINGIFY_ARG(Category##StructName) "_get_node_type");
 		
 	#include "rna_nodetree_types.h"
 	
-	define_specific_node(brna, NODE_GROUP, def_group);
-	define_specific_node(brna, NODE_FRAME, def_frame);
-	define_specific_node(brna, NODE_REROUTE, 0);
+	define_specific_node(brna, NODE_GROUP, def_group, "rna_NodeGroup_get_node_type");
+	define_specific_node(brna, NODE_FRAME, def_frame, "rna_NodeFrame_get_node_type");
+	define_specific_node(brna, NODE_REROUTE, 0, "rna_NodeReroute_get_node_type");
 	
 	/* special socket types */
 	rna_def_cmp_output_file_slot_file(brna);