system_property_chart.py

   1 # ##### BEGIN GPL LICENSE BLOCK #####
   2 #
   3 #  This program is free software; you can redistribute it and/or
   4 #  modify it under the terms of the GNU General Public License
   5 #  as published by the Free Software Foundation; either version 2
   6 #  of the License, or (at your option) any later version.
   7 #
   8 #  This program is distributed in the hope that it will be useful,
   9 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 #  GNU General Public License for more details.
  12 #
  13 #  You should have received a copy of the GNU General Public License
  14 #  along with this program; if not, write to the Free Software Foundation,
  15 #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  16 #
  17 # ##### END GPL LICENSE BLOCK #####
  18
  19 # <pep8 compliant>
  20
  21 bl_info = {
  22     "name": "Property Chart",
  23     "author": "Campbell Barton (ideasman42)",
  24     "version": (0, 1, 1),
  25     "blender": (2, 80, 0),
  26     "location": "View3D > Sidebar > Item Tab",
  27     "description": ("Edit arbitrary selected properties for "
  28                     "objects/sequence strips of the same type"),
  29     "warning": "",
  30     "wiki_url": "http://wiki.blender.org/index.php/Extensions:2.6/Py/"
  31                 "Scripts/System/Object Property Chart",
  32     "category": "System",
  33 }
  34
  35 """List properties of selected objects"""
  36
  37 import bpy
  38 from bl_operators.presets import AddPresetBase
  39 from bpy.types import (
  40     Menu,
  41     Operator,
  42     Panel,
  43 )
  44
  45
  46 class AddPresetProperties(AddPresetBase, Operator):
  47     """Add an properties preset"""
  48     bl_idname = "scene.properties_preset_add"
  49     bl_label = "Add Properties Preset"
  50     preset_menu = "SCENE_MT_properties_presets"
  51
  52     preset_defines = [
  53         "scene = bpy.context.scene",
  54     ]
  55
  56     def pre_cb(self, context):
  57         space_type = context.space_data.type
  58         if space_type == 'VIEW_3D':
  59             self.preset_subdir = "system_property_chart_view3d"
  60             self.preset_values = ["scene.view3d_edit_props"]
  61         else:
  62             self.preset_subdir = "system_property_chart_sequencer"
  63             self.preset_values = ["scene.sequencer_edit_props"]
  64
  65
  66 class SCENE_MT_properties_presets(Menu):
  67     bl_label = "Properties Presets"
  68     preset_operator = "script.execute_preset"
  69
  70     def draw(self, context):
  71         space_type = context.space_data.type
  72
  73         if space_type == 'VIEW_3D':
  74             self.preset_subdir = "system_property_chart_view3d"
  75         else:
  76             self.preset_subdir = "system_property_chart_sequencer"
  77
  78         Menu.draw_preset(self, context)
  79
  80
  81 def _property_chart_data_get(self, context):
  82     # eg. context.active_object
  83     obj = eval("context.%s" % self.context_data_path_active)
  84
  85     if obj is None:
  86         return None, None
  87
  88     # eg. context.selected_objects[:]
  89     selected_objects = eval("context.%s" % self.context_data_path_selected)[:]
  90
  91     if not selected_objects:
  92         return None, None
  93
  94     return obj, selected_objects
  95
  96
  97 def _property_chart_draw(self, context):
  98     """
  99     This function can run for different types.
 100     """
 101     obj, selected_objects = _property_chart_data_get(self, context)
 102
 103     if not obj:
 104         return
 105
 106     # active first
 107     try:
 108         active_index = selected_objects.index(obj)
 109     except ValueError:
 110         active_index = -1
 111
 112     if active_index > 0:  # not the first already
 113         selected_objects[0], selected_objects[active_index] = selected_objects[active_index], selected_objects[0]
 114
 115     id_storage = context.scene
 116
 117     strings = getattr(id_storage, self._PROP_STORAGE_ID)
 118
 119     # Collected all props, now display them all
 120     layout = self.layout
 121
 122     if strings:
 123
 124         def obj_prop_get(obj, attr_string):
 125             """return a pair (rna_base, "rna_property") to give to the rna UI property function"""
 126             attrs = attr_string.split(".")
 127             val_new = obj
 128             for i, attr in enumerate(attrs):
 129                 val_old = val_new
 130                 val_new = getattr(val_old, attr, Ellipsis)
 131
 132                 if val_new == Ellipsis:
 133                     return None, None
 134             return val_old, attrs[-1]
 135
 136         strings = strings.split()
 137
 138         prop_all = []
 139
 140         for obj in selected_objects:
 141             prop_pairs = []
 142             prop_found = False
 143             for attr_string in strings:
 144                 prop_pairs.append(obj_prop_get(obj, attr_string))
 145                 if prop_found is False and prop_pairs[-1] != (None, None):
 146                     prop_found = True
 147
 148             if prop_found:
 149                 prop_all.append((obj, prop_pairs))
 150
 151         row = layout.row(align=True)
 152
 153         col = row.column(align=True)
 154         col.label(text="name")
 155         for obj, prop_pairs in prop_all:
 156             col.prop(obj, "name", text="")
 157
 158         for i in range(len(strings)):
 159             col = row.column(align=True)
 160
 161             # name and copy button
 162             rowsub = col.row(align=False)
 163             rowsub.label(text=strings[i].rsplit(".", 1)[-1])
 164             props = rowsub.operator("wm.chart_copy", text="", icon='PASTEDOWN', emboss=False)
 165             props.data_path_active = self.context_data_path_active
 166             props.data_path_selected = self.context_data_path_selected
 167             props.data_path = strings[i]
 168
 169             for obj, prop_pairs in prop_all:
 170                 data, attr = prop_pairs[i]
 171                 # row is needed for vector buttons
 172                 if data:
 173                     col.row().prop(data, attr, text="")  # , emboss=obj==active_object
 174                 else:
 175                     col.label(text="<missing>")
 176
 177     # Presets for properties
 178     col = layout.column()
 179     col.label(text="Properties")
 180     row = col.row(align=True)
 181     row.menu("SCENE_MT_properties_presets", text=bpy.types.SCENE_MT_properties_presets.bl_label)
 182     row.operator("scene.properties_preset_add", text="", icon='ADD')
 183     row.operator("scene.properties_preset_add", text="", icon='REMOVE').remove_active = True
 184     # edit the display props
 185     col.prop(id_storage, self._PROP_STORAGE_ID, text="")
 186
 187
 188 class View3DEditProps(Panel):
 189     bl_idname = "SYSPROP_CHART_PT_view3d"
 190     bl_space_type = 'VIEW_3D'
 191     bl_region_type = 'UI'
 192     bl_category = "Item"
 193     bl_label = "Property Chart"
 194     bl_context = "objectmode"
 195
 196     _PROP_STORAGE_ID = "view3d_edit_props"
 197     _PROP_STORAGE_ID_DESCR = "Properties of objects in the context"
 198     _PROP_STORAGE_DEFAULT = "data data.use_auto_smooth"
 199
 200     # _property_chart_draw needs these
 201     context_data_path_active = "active_object"
 202     context_data_path_selected = "selected_objects"
 203
 204     draw = _property_chart_draw
 205
 206
 207 class SequencerEditProps(Panel):
 208     bl_idname = "SYSPROP_CHART_PT_sequencer"
 209     bl_space_type = 'SEQUENCE_EDITOR'
 210     bl_region_type = 'UI'
 211     bl_category = "Item"
 212     bl_label = "Property Chart"
 213
 214     _PROP_STORAGE_ID = "sequencer_edit_props"
 215     _PROP_STORAGE_ID_DESCR = "Properties of sequencer strips in the context"
 216     _PROP_STORAGE_DEFAULT = "blend_type blend_alpha"
 217
 218     # _property_chart_draw needs these
 219     context_data_path_active = "scene.sequence_editor.active_strip"
 220     context_data_path_selected = "selected_sequences"
 221
 222     draw = _property_chart_draw
 223
 224     @classmethod
 225     def poll(cls, context):
 226         return context.scene.sequence_editor is not None
 227
 228 # Operator to copy properties
 229
 230
 231 def _property_chart_copy(self, context):
 232     obj, selected_objects = _property_chart_data_get(self, context)
 233
 234     if not obj:
 235         return
 236
 237     data_path = self.data_path
 238
 239     # quick & nasty method!
 240     for obj_iter in selected_objects:
 241         if obj != obj_iter:
 242             try:
 243                 exec("obj_iter.%s = obj.%s" % (data_path, data_path))
 244             except:
 245                 # just in case we need to know what went wrong!
 246                 import traceback
 247                 traceback.print_exc()
 248
 249 from bpy.props import StringProperty
 250
 251
 252 class CopyPropertyChart(Operator):
 253     "Open a path in a file browser"
 254     bl_idname = "wm.chart_copy"
 255     bl_label = "Copy properties from active to selected"
 256
 257     data_path_active: StringProperty()
 258     data_path_selected: StringProperty()
 259     data_path: StringProperty()
 260
 261     def execute(self, context):
 262         # so attributes are found for '_property_chart_data_get()'
 263         self.context_data_path_active = self.data_path_active
 264         self.context_data_path_selected = self.data_path_selected
 265
 266         _property_chart_copy(self, context)
 267
 268         return {'FINISHED'}
 269
 270 # List The Classes #
 271
 272 classes = (
 273     AddPresetProperties,
 274     SCENE_MT_properties_presets,
 275     View3DEditProps,
 276     SequencerEditProps,
 277     CopyPropertyChart
 278 )
 279
 280 def register():
 281     for cls in classes:
 282         bpy.utils.register_class(cls)
 283
 284     Scene = bpy.types.Scene
 285
 286     for cls in View3DEditProps, SequencerEditProps:
 287         setattr(
 288             Scene,
 289             cls._PROP_STORAGE_ID,
 290             StringProperty(
 291                 name="Properties",
 292                 description=cls._PROP_STORAGE_ID_DESCR + " (separated by spaces)",
 293                 default=cls._PROP_STORAGE_DEFAULT, maxlen=1024,
 294             ),
 295         )
 296
 297
 298 def unregister():
 299     for cls in classes:
 300         bpy.utils.unregister_class(cls)
 301
 302     Scene = bpy.types.Scene
 303
 304     for cls in View3DEditProps, SequencerEditProps:
 305         delattr(Scene,
 306                 cls._PROP_STORAGE_ID,
 307                 )
 308
 309
 310 if __name__ == "__main__":
 311     register()