Update README.md

README.md

@@ -2,6 +2,8 @@
 > Blender addon for picking rig contollers
 
 Rig_picker is an OpenGl tool for having a 2d interface for the 3d animators allowing them to pick a controller easily.
+The addon is drawing 2d shapes inside a dedicated Node Editor Area using the gpu module.
+You can use multiple pickers for one rig, each picker shapes are in there own collection. 
 
 Video of the previous version : https://vimeo.com/241970235
 
@@ -23,7 +25,12 @@ Video of the previous version : https://vimeo.com/241970235
 
 <!-- INSTALLATION -->
 ## Installation
+For external user, you can clone the repository using:
+```sh
+git clone https://git.autourdeminuit.com/autour_de_minuit/rig_picker.git
+```
 
+For Internal user:
 1. Create your own local directory in 
     ```sh
    /home/<USER>/dev

area.py

@@ -155,6 +155,7 @@ classes = (
 
 
 def register():
+    # Remove the tools inside the picker space
     bpy.types.WM_OT_tool_set_by_id._execute = bpy.types.WM_OT_tool_set_by_id.execute #tool_set_by_id    
     bpy.types.WM_OT_tool_set_by_id.execute = tool_set_by_id
 

core/picker.py

@@ -459,7 +459,7 @@ class Picker:
             if shape.type == 'bone' and shape.hover:
                 shape.assign_bone_event()
         
-        bpy.ops._rigpicker.save_picker(index=self.index)
+        bpy.ops.rigpicker.save_picker(index=self.index)
 
     def press_event(self, mode='SET'):
         for shape in self.shapes:
@@ -712,7 +712,7 @@ class PickerGroup:
 
 def load_picker_data(rig):
     if 'pickers' in rig.data.rig_picker:
-        picker_datas = [p.to_dict() for p in rig.rig_picker['pickers']]
+        picker_datas = [[s.to_dict() for s in p] for p in rig.data.rig_picker['pickers']]
     else:
         picker_datas = []
 
@@ -736,8 +736,11 @@ def get_picker_path(rig, source, start=None):
 
 
 def pack_picker(rig, start=None):
+    if not 'rig_picker' in rig.data:
+        return
+
     pickers = []
-    for picker_source in rig.data.get('rig_picker', {}).get('sources', {}):
+    for picker_source in rig.data['rig_picker'].get('sources', []):
         picker_path = get_picker_path(rig, picker_source['source'], start)
         if not picker_path.exists():
             print(f'{picker_path} not exists')

core/shape.py

@@ -93,7 +93,7 @@ def get_shape_data(ob, matrix=None, depsgraph=None):
 
     for vert in mesh.vertices:
         co = matrix @ (ob.matrix_world @ Vector(vert.co))
-        points.append([round(co[0]), round(co[1])])
+        points.append([round(co[0], 1), round(co[1], 1)])
 
         depths += [co[2]]
     

gizmo.py

@@ -68,6 +68,8 @@ class RP_GT_gizmo(Gizmo):
 
         context.region.tag_redraw()
 
+        #print(location)
+
         return -1
 
     

operators/picker.py

@@ -392,24 +392,19 @@ class RP_OT_reload_picker(Operator):
     #        return
 
     def execute(self, context):
-        #PICKERS.clear()
         if context.object.type == 'ARMATURE':
             rig = context.object
         else:
             collection = get_picker_collection(context.object)
             rig = collection.rig_picker.rig
 
+        #print('Reload', rig)
         load_picker_data(rig)
 
-        '''
         for area in context.screen.areas:
-            #print(area.type, is_picker_space(area.spaces.active))
             if is_picker_space(area.spaces.active):
-
-                print('Tag Redraw Region', area.type)
-                area.regions[0].tag_redraw()
                 area.tag_redraw()
-                '''
+ 
 
         return {"FINISHED"}
 
@@ -551,7 +546,8 @@ class RP_MT_context_menu(Menu):
     def draw(self, context):
         layout = self.layout
         col = layout.column()
-        col.use_property_split = True
+        col.operator_context = 'INVOKE_DEFAULT'
+        #col.use_property_split = True
 
         ob = context.object
         picker = PICKERS.get(ob)
@@ -561,9 +557,25 @@ class RP_MT_context_menu(Menu):
         else:
             bone = context.active_pose_bone
 
+        # Draw Space Switch Operator
+        if getattr(ob.data, 'space_switch'):
+            space_switch = ob.data.space_switch
+
+            data_paths = [f'pose.bones["{bone.name}"]["{k}"]' for k in bone.keys()]
+            space_bone = next((s for s in space_switch.bones if s.data_path in data_paths), None)
+            if space_bone:
+
+                index = list(space_switch.bones).index(space_bone)
+                value = ob.path_resolve(space_bone.data_path)
+                space = next((s.name for s in space_bone.spaces if s.value == value), None)
+
+                op = col.operator("spaceswitch.change_space", text=f'({space})', icon='PINNED')
+                op.index=index
+            col.separator()
+
         if bone:    
             for key in bone.keys():
-                layout.prop(bone,f'["{key}"]', slider=True)
+                col.prop(bone,f'["{key}"]', slider=True)
 
         #layout.operator("rigpicker.show_bone_layer", text="Show Bone Layer", ).type = 'ACTIVE'
         #layout.operator("rigidbody.objects_add", text="B Add Passive").type = 'PASSIVE'

operators/shape.py

@@ -222,7 +222,6 @@ class RP_OT_save_picker(Operator):
 
         bpy.ops.rigpicker.reload_picker()
 
-
         return {'FINISHED'}
 
 classes = (