[Bf-blender-cvs] SVN commit: /data/svn/bf-blender [30967] trunk/blender: minor changes to Martni's commit 30961

Campbell Barton ideasman42 at gmail.com
Mon Aug 2 06:20:41 CEST 2010


Revision: 30967
          http://projects.blender.org/plugins/scmsvn/viewcvs.php?view=rev&root=bf-blender&revision=30967
Author:   campbellbarton
Date:     2010-08-02 06:20:41 +0200 (Mon, 02 Aug 2010)

Log Message:
-----------
minor changes to Martni's commit 30961
- removed the immediate option from C/api and now store in python only, when python loads modules it sets it to False.
- unloading a module would clear the entire TypeMap for all modules, only remove the module types that is being unloaded.
- added some checks for bad class registering, report errors rather then crashing.

Modified Paths:
--------------
    trunk/blender/release/scripts/modules/bpy/utils.py
    trunk/blender/release/scripts/modules/bpy_types.py
    trunk/blender/source/blender/python/intern/bpy_interface.c
    trunk/blender/source/blender/python/intern/bpy_rna.c
    trunk/blender/source/blender/python/intern/bpy_rna.h

Modified: trunk/blender/release/scripts/modules/bpy/utils.py
===================================================================
--- trunk/blender/release/scripts/modules/bpy/utils.py	2010-08-02 04:10:16 UTC (rev 30966)
+++ trunk/blender/release/scripts/modules/bpy/utils.py	2010-08-02 04:20:41 UTC (rev 30967)
@@ -102,6 +102,9 @@
     import traceback
     import time
 
+    # must be set back to True on exits
+    _bpy_types._register_immediate = False
+
     t_main = time.time()
 
     loaded_modules = set()
@@ -217,6 +220,8 @@
 
     if _bpy.app.debug:
         print("Python Script Load Time %.4f" % (time.time() - t_main))
+    
+    _bpy_types._register_immediate = True
 
 
 def expandpath(path):

Modified: trunk/blender/release/scripts/modules/bpy_types.py
===================================================================
--- trunk/blender/release/scripts/modules/bpy_types.py	2010-08-02 04:10:16 UTC (rev 30966)
+++ trunk/blender/release/scripts/modules/bpy_types.py	2010-08-02 04:20:41 UTC (rev 30967)
@@ -546,20 +546,24 @@
 # and unregistered on unload
 PropertiesMap = {}
 
+# Using our own loading function we set this to false
+# so when running a script directly in the text editor
+# registers moduals instantly.
+_register_immediate = True
+
 def UnloadModule(module):
-    global TypeMap, PropertiesMap
-    for t in TypeMap.get(module, []):
+    for t in TypeMap.setdefault(module, ()):
         bpy_types.unregister(t)
         
-    TypeMap = {}
+    del TypeMap[module]
 
-    for t in PropertiesMap.get(module, []):
+    for t in PropertiesMap.setdefault(module, ()):
         bpy_types.unregister(t)
         
-    PropertiesMap = {}
-    
+    del PropertiesMap[module]
+
 def LoadModule(module, force=False):
-    for t in TypeMap.get(module, []):
+    for t in TypeMap.get(module, ()):
         bpy_types.register(t)
 
 _bpy.LoadModule = LoadModule
@@ -567,35 +571,33 @@
 
 class RNAMeta(type):
     @classmethod
-    def _immediate(cls):
-        return bpy_types.immediate();
-    
+    def _register_immediate(cls):
+        return _register_immediate
+
     def __new__(cls, name, bases, classdict, **args):
         result = type.__new__(cls, name, bases, classdict)
         if bases and bases[0] != StructRNA:
             module = result.__module__
-            
+
             ClassMap = TypeMap
-            
+
             # Register right away if needed
-            if cls._immediate():
+            if cls._register_immediate():
                 bpy_types.register(result)
                 ClassMap = PropertiesMap 
 
             # first part of packages only
             if "." in module:
                 module = module[:module.index(".")]
-            
-            if not module in ClassMap:
-                ClassMap[module] = []
-                
-            ClassMap[module].append(result)
+
+            ClassMap.setdefault(module, []).append(result)
+
         return result
 
 class RNAMetaRegister(RNAMeta):
     @classmethod
-    def _immediate(cls):
-        return True;
+    def _register_immediate(cls):
+        return True
 
 class OrderedMeta(RNAMeta):
 

Modified: trunk/blender/source/blender/python/intern/bpy_interface.c
===================================================================
--- trunk/blender/source/blender/python/intern/bpy_interface.c	2010-08-02 04:10:16 UTC (rev 30966)
+++ trunk/blender/source/blender/python/intern/bpy_interface.c	2010-08-02 04:20:41 UTC (rev 30967)
@@ -329,8 +329,6 @@
 	if (fn==NULL && text==NULL) {
 		return 0;
 	}
-	
-	bpy_set_immediate_register(1);
 
 	bpy_context_set(C, &gilstate);
 
@@ -397,8 +395,6 @@
 	
 	bpy_context_clear(C, &gilstate);
 
-	bpy_set_immediate_register(0);
-
 	return py_result ? 1:0;
 }
 

Modified: trunk/blender/source/blender/python/intern/bpy_rna.c
===================================================================
--- trunk/blender/source/blender/python/intern/bpy_rna.c	2010-08-02 04:10:16 UTC (rev 30966)
+++ trunk/blender/source/blender/python/intern/bpy_rna.c	2010-08-02 04:20:41 UTC (rev 30967)
@@ -4175,11 +4175,9 @@
 static PyObject *pyrna_basetype_dir(BPy_BaseTypeRNA *self);
 static PyObject *pyrna_basetype_register(PyObject *self, PyObject *py_class);
 static PyObject *pyrna_basetype_unregister(PyObject *self, PyObject *py_class);
-static PyObject *pyrna_register_immediate(PyObject *self);
 
 static struct PyMethodDef pyrna_basetype_methods[] = {
 	{"__dir__", (PyCFunction)pyrna_basetype_dir, METH_NOARGS, ""},
-	{"immediate", (PyCFunction)pyrna_register_immediate, METH_NOARGS, ""},
 	{"register", (PyCFunction)pyrna_basetype_register, METH_O, ""},
 	{"unregister", (PyCFunction)pyrna_basetype_unregister, METH_O, ""},
 	{NULL, NULL, 0, NULL}
@@ -4546,10 +4544,17 @@
 	PyGILState_STATE gilstate;
 
 	bContext *C= BPy_GetContext(); // XXX - NEEDS FIXING, QUITE BAD.
-	bpy_context_set(C, &gilstate);
-
+	
 	py_class= RNA_struct_py_type_get(ptr->type);
 	
+	/* rare case. can happen when registering subclasses */
+	if(py_class==NULL) {
+		fprintf(stderr, "bpy_class_call(): unable to get python class for rna struct '%.200s'\n", RNA_struct_identifier(ptr->type));
+		return -1;
+	}
+	
+	bpy_context_set(C, &gilstate);
+	
 	/* exception, operators store their PyObjects for re-use */
 	if(ptr->data) {
 		if(RNA_struct_is_a(ptr->type, &RNA_Operator)) {
@@ -4738,7 +4743,16 @@
 	prop = RNA_struct_find_property(&ptr, "structs");
 
 	RNA_PROP_BEGIN(&ptr, itemptr, prop) {
-		Py_DECREF(pyrna_struct_Subtype(&itemptr));
+		PyObject *item= pyrna_struct_Subtype(&itemptr);
+		if(item == NULL) {
+			if(PyErr_Occurred()) {
+				PyErr_Print();
+				PyErr_Clear();
+			}
+		}
+		else {
+			Py_DECREF(item);
+		}
 	}
 	RNA_PROP_END;
 
@@ -4771,22 +4785,6 @@
 
 }
 
-static int IMMEDIATE = 0;
-
-void bpy_set_immediate_register(int value)
-{
-	IMMEDIATE = value;
-}
-
-static PyObject *pyrna_register_immediate(PyObject *self)
-{
-	if (IMMEDIATE) {
-		Py_RETURN_TRUE;
-	} else {
-		Py_RETURN_FALSE;
-	}
-}
-
 /* Note! MemLeak XXX
  *
  * There is currently a bug where moving registering a python class does
@@ -4817,6 +4815,14 @@
 	if(srna==NULL)
 		return NULL;
 	
+	/* fails in cases, cant use this check but would like to :| */
+	/*
+	if(RNA_struct_py_type_get(srna)) {
+		PyErr_Format(PyExc_ValueError, "bpy.types.register(...): %.200s's parent class %.200s is alredy registered, this is not allowed.", ((PyTypeObject*)py_class)->tp_name, RNA_struct_identifier(srna));
+		return NULL;
+	}
+	*/
+
 	/* check that we have a register callback for this type */
 	reg= RNA_struct_register(srna);
 

Modified: trunk/blender/source/blender/python/intern/bpy_rna.h
===================================================================
--- trunk/blender/source/blender/python/intern/bpy_rna.h	2010-08-02 04:10:16 UTC (rev 30966)
+++ trunk/blender/source/blender/python/intern/bpy_rna.h	2010-08-02 04:20:41 UTC (rev 30967)
@@ -86,8 +86,6 @@
 
 int pyrna_enum_value_from_id(EnumPropertyItem *item, const char *identifier, int *value, const char *error_prefix);
 
-void bpy_set_immediate_register(int value);
-
 int pyrna_deferred_register_props(struct StructRNA *srna, PyObject *class_dict);
 
 /* called before stopping python */





More information about the Bf-blender-cvs mailing list