object_collection_manager/qcd_init.py

   1 # SPDX-FileCopyrightText: 2011 Ryan Inch
   2 #
   3 # SPDX-License-Identifier: GPL-2.0-or-later
   4
   5 if "bpy" in locals():
   6     import importlib
   7
   8     importlib.reload(internals)
   9     importlib.reload(qcd_move_widget)
  10     importlib.reload(qcd_operators)
  11     importlib.reload(ui)
  12     importlib.reload(preferences)
  13
  14 else:
  15     from . import internals
  16     from . import qcd_move_widget
  17     from . import qcd_operators
  18     from . import ui
  19     from . import preferences
  20
  21 import os
  22 import bpy
  23 import bpy.utils.previews
  24 from bpy.app.handlers import persistent
  25
  26
  27 addon_qcd_keymaps = []
  28 addon_qcd_view_hotkey_keymaps = []
  29 addon_qcd_view_edit_mode_hotkey_keymaps = []
  30 qcd_registered = False
  31
  32
  33 qcd_classes = (
  34     qcd_move_widget.QCDMoveWidget,
  35     qcd_operators.EnableAllQCDSlotsMeta,
  36     qcd_operators.EnableAllQCDSlots,
  37     qcd_operators.EnableAllQCDSlotsIsolated,
  38     qcd_operators.IsolateSelectedObjectsCollections,
  39     qcd_operators.DisableSelectedObjectsCollections,
  40     qcd_operators.DisableAllNonQCDSlots,
  41     qcd_operators.DisableAllCollections,
  42     qcd_operators.SelectAllQCDObjects,
  43     qcd_operators.DiscardQCDHistory,
  44     qcd_operators.MoveToQCDSlot,
  45     qcd_operators.ViewQCDSlot,
  46     qcd_operators.ViewMoveQCDSlot,
  47     qcd_operators.UnassignedQCDSlot,
  48     qcd_operators.CreateAllQCDSlots,
  49     qcd_operators.RenumerateQCDSlots,
  50     ui.EnableAllQCDSlotsMenu,
  51     )
  52
  53
  54 @persistent
  55 def save_internal_data(dummy):
  56     cm = bpy.context.scene.collection_manager
  57
  58     cm.qcd_slots_blend_data = internals.qcd_slots.get_data_for_blend()
  59
  60
  61 @persistent
  62 def load_internal_data(dummy):
  63     cm = bpy.context.scene.collection_manager
  64     data = cm.qcd_slots_blend_data
  65
  66     if not data:
  67         return
  68
  69     internals.qcd_slots.load_blend_data(data)
  70
  71
  72 @persistent
  73 def load_pre_handler(dummy):
  74     internals.qcd_collection_state.clear()
  75     for key in list(internals.qcd_history.keys()):
  76         del internals.qcd_history[key]
  77
  78
  79 def register_qcd():
  80     global qcd_registered
  81     if qcd_registered:
  82         # guard to handle default value updates (mouse hover + backspace)
  83         return
  84
  85     for cls in qcd_classes:
  86         bpy.utils.register_class(cls)
  87
  88     pcoll = bpy.utils.previews.new()
  89     icons_dir = os.path.join(os.path.dirname(__file__), "icons")
  90     pcoll.load("active_icon_base", os.path.join(icons_dir, "minus.png"), 'IMAGE', True)
  91     pcoll.load("active_icon_text", os.path.join(icons_dir, "minus.png"), 'IMAGE', True)
  92     pcoll.load("active_icon_text_sel", os.path.join(icons_dir, "minus.png"), 'IMAGE', True)
  93     ui.preview_collections["icons"] = pcoll
  94
  95     wm = bpy.context.window_manager
  96     if wm.keyconfigs.addon: # not present when started with --background
  97         km = wm.keyconfigs.addon.keymaps.new(name='Object Mode')
  98         kmi = km.keymap_items.new('view3d.qcd_move_widget', 'V', 'PRESS')
  99         addon_qcd_keymaps.append((km, kmi))
 100
 101     bpy.app.handlers.save_pre.append(save_internal_data)
 102     bpy.app.handlers.load_post.append(load_internal_data)
 103     bpy.app.handlers.load_pre.append(load_pre_handler)
 104
 105     prefs = bpy.context.preferences.addons[__package__].preferences
 106
 107     if prefs.enable_qcd_view_hotkeys:
 108         register_qcd_view_hotkeys()
 109
 110     if prefs.enable_qcd_view_edit_mode_hotkeys:
 111         register_qcd_view_edit_mode_hotkeys()
 112
 113     if prefs.enable_qcd_3dview_header_widget:
 114         register_qcd_3dview_header_widget()
 115
 116     qcd_registered = True
 117
 118
 119 def register_qcd_view_hotkeys():
 120     if addon_qcd_view_hotkey_keymaps:
 121         # guard to handle default value updates (mouse hover + backspace)
 122         return
 123
 124     wm = bpy.context.window_manager
 125     if wm.keyconfigs.addon: # not present when started with --background
 126         # create qcd hotkeys
 127         qcd_hotkeys = [
 128             ["ONE", False, "1"],
 129             ["TWO", False, "2"],
 130             ["THREE", False, "3"],
 131             ["FOUR", False, "4"],
 132             ["FIVE", False, "5"],
 133             ["SIX", False, "6"],
 134             ["SEVEN", False, "7"],
 135             ["EIGHT", False, "8"],
 136             ["NINE", False, "9"],
 137             ["ZERO", False, "10"],
 138             ["ONE", True, "11"],
 139             ["TWO", True, "12"],
 140             ["THREE", True, "13"],
 141             ["FOUR", True, "14"],
 142             ["FIVE", True, "15"],
 143             ["SIX", True, "16"],
 144             ["SEVEN", True, "17"],
 145             ["EIGHT", True, "18"],
 146             ["NINE", True, "19"],
 147             ["ZERO", True, "20"],
 148         ]
 149
 150         for mode in ['Object Mode', 'Pose', 'Weight Paint']:
 151             for key in qcd_hotkeys:
 152                 km = wm.keyconfigs.addon.keymaps.new(name=mode)
 153                 kmi = km.keymap_items.new('view3d.view_qcd_slot', key[0], 'PRESS', alt=key[1])
 154                 kmi.properties.slot = key[2]
 155                 kmi.properties.toggle = False
 156                 addon_qcd_view_hotkey_keymaps.append((km, kmi))
 157
 158                 km = wm.keyconfigs.addon.keymaps.new(name=mode)
 159                 kmi = km.keymap_items.new('view3d.view_qcd_slot', key[0], 'PRESS',shift=True,  alt=key[1])
 160                 kmi.properties.slot = key[2]
 161                 kmi.properties.toggle = True
 162                 addon_qcd_view_hotkey_keymaps.append((km, kmi))
 163
 164             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 165             kmi = km.keymap_items.new('view3d.enable_all_qcd_slots', 'PLUS', 'PRESS', shift=True)
 166             addon_qcd_view_hotkey_keymaps.append((km, kmi))
 167
 168             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 169             kmi = km.keymap_items.new('view3d.isolate_selected_objects_collections', 'EQUAL', 'PRESS')
 170             addon_qcd_view_hotkey_keymaps.append((km, kmi))
 171
 172             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 173             kmi = km.keymap_items.new('view3d.disable_selected_objects_collections', 'MINUS', 'PRESS')
 174             addon_qcd_view_hotkey_keymaps.append((km, kmi))
 175
 176             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 177             kmi = km.keymap_items.new('view3d.disable_all_non_qcd_slots', 'PLUS', 'PRESS', shift=True, alt=True)
 178             addon_qcd_view_hotkey_keymaps.append((km, kmi))
 179
 180             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 181             kmi = km.keymap_items.new('view3d.disable_all_collections', 'EQUAL', 'PRESS', alt=True, ctrl=True)
 182             addon_qcd_view_hotkey_keymaps.append((km, kmi))
 183
 184             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 185             kmi = km.keymap_items.new('view3d.select_all_qcd_objects', 'PLUS', 'PRESS', shift=True, ctrl=True)
 186             addon_qcd_view_hotkey_keymaps.append((km, kmi))
 187
 188
 189             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 190             kmi = km.keymap_items.new('view3d.discard_qcd_history', 'EQUAL', 'PRESS', alt=True)
 191             addon_qcd_view_hotkey_keymaps.append((km, kmi))
 192
 193
 194 def register_qcd_view_edit_mode_hotkeys():
 195     if addon_qcd_view_edit_mode_hotkey_keymaps:
 196         # guard to handle default value updates (mouse hover + backspace)
 197         return
 198
 199     wm = bpy.context.window_manager
 200     if wm.keyconfigs.addon: # not present when started with --background
 201         # create qcd hotkeys
 202         qcd_hotkeys = [
 203             ["ONE", False, "1"],
 204             ["TWO", False, "2"],
 205             ["THREE", False, "3"],
 206             ["FOUR", False, "4"],
 207             ["FIVE", False, "5"],
 208             ["SIX", False, "6"],
 209             ["SEVEN", False, "7"],
 210             ["EIGHT", False, "8"],
 211             ["NINE", False, "9"],
 212             ["ZERO", False, "10"],
 213             ["ONE", True, "11"],
 214             ["TWO", True, "12"],
 215             ["THREE", True, "13"],
 216             ["FOUR", True, "14"],
 217             ["FIVE", True, "15"],
 218             ["SIX", True, "16"],
 219             ["SEVEN", True, "17"],
 220             ["EIGHT", True, "18"],
 221             ["NINE", True, "19"],
 222             ["ZERO", True, "20"],
 223         ]
 224
 225         for mode in ["Mesh", "Curve", "Armature", "Metaball", "Lattice", "Grease Pencil Stroke Edit Mode"]:
 226             for key in qcd_hotkeys:
 227                 km = wm.keyconfigs.addon.keymaps.new(name=mode)
 228                 kmi = km.keymap_items.new('view3d.view_qcd_slot', key[0], 'PRESS', alt=key[1])
 229                 kmi.properties.slot = key[2]
 230                 kmi.properties.toggle = False
 231                 addon_qcd_view_edit_mode_hotkey_keymaps.append((km, kmi))
 232
 233                 km = wm.keyconfigs.addon.keymaps.new(name=mode)
 234                 kmi = km.keymap_items.new('view3d.view_qcd_slot', key[0], 'PRESS',shift=True,  alt=key[1])
 235                 kmi.properties.slot = key[2]
 236                 kmi.properties.toggle = True
 237                 addon_qcd_view_edit_mode_hotkey_keymaps.append((km, kmi))
 238
 239             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 240             kmi = km.keymap_items.new('view3d.enable_all_qcd_slots', 'PLUS', 'PRESS', shift=True)
 241             addon_qcd_view_edit_mode_hotkey_keymaps.append((km, kmi))
 242
 243             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 244             kmi = km.keymap_items.new('view3d.isolate_selected_objects_collections', 'EQUAL', 'PRESS')
 245             addon_qcd_view_edit_mode_hotkey_keymaps.append((km, kmi))
 246
 247             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 248             kmi = km.keymap_items.new('view3d.disable_all_non_qcd_slots', 'PLUS', 'PRESS', shift=True, alt=True)
 249             addon_qcd_view_edit_mode_hotkey_keymaps.append((km, kmi))
 250
 251             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 252             kmi = km.keymap_items.new('view3d.disable_all_collections', 'EQUAL', 'PRESS', alt=True, ctrl=True)
 253             addon_qcd_view_edit_mode_hotkey_keymaps.append((km, kmi))
 254
 255             km = wm.keyconfigs.addon.keymaps.new(name=mode)
 256             kmi = km.keymap_items.new('view3d.discard_qcd_history', 'EQUAL', 'PRESS', alt=True)
 257             addon_qcd_view_edit_mode_hotkey_keymaps.append((km, kmi))
 258
 259
 260         km = wm.keyconfigs.addon.keymaps.new(name="Mesh")
 261         kmi = km.keymap_items.new('wm.call_menu', 'ACCENT_GRAVE', 'PRESS')
 262         kmi.properties.name = "VIEW3D_MT_edit_mesh_select_mode"
 263         addon_qcd_view_edit_mode_hotkey_keymaps.append((km, kmi))
 264
 265
 266 def register_qcd_3dview_header_widget():
 267     # unregister first to guard against default value updates (mouse hover + backspace)
 268     # if the widget isn't registered it will just do nothing
 269     unregister_qcd_3dview_header_widget()
 270
 271     # add the widget to the header, and an update function to the top bar to get view layer changes
 272     bpy.types.VIEW3D_HT_header.append(ui.view3d_header_qcd_slots)
 273     bpy.types.TOPBAR_HT_upper_bar.append(ui.view_layer_update)
 274
 275
 276
 277 def unregister_qcd():
 278     global qcd_registered
 279     if not qcd_registered:
 280         # guard to handle default value updates (mouse hover + backspace)
 281         return
 282
 283     unregister_qcd_3dview_header_widget()
 284
 285     for cls in qcd_classes:
 286         bpy.utils.unregister_class(cls)
 287
 288     bpy.app.handlers.save_pre.remove(save_internal_data)
 289     bpy.app.handlers.load_post.remove(load_internal_data)
 290     bpy.app.handlers.load_pre.remove(load_pre_handler)
 291
 292     for pcoll in ui.preview_collections.values():
 293         bpy.utils.previews.remove(pcoll)
 294     ui.preview_collections.clear()
 295     ui.last_icon_theme_text = None
 296     ui.last_icon_theme_text_sel = None
 297
 298     # remove keymaps when qcd is deactivated
 299     for km, kmi in addon_qcd_keymaps:
 300         km.keymap_items.remove(kmi)
 301     addon_qcd_keymaps.clear()
 302
 303
 304     unregister_qcd_view_hotkeys()
 305
 306     unregister_qcd_view_edit_mode_hotkeys()
 307
 308     qcd_registered = False
 309
 310
 311 def unregister_qcd_view_hotkeys():
 312     # remove keymaps when qcd view hotkeys are deactivated
 313     for km, kmi in addon_qcd_view_hotkey_keymaps:
 314         km.keymap_items.remove(kmi)
 315     addon_qcd_view_hotkey_keymaps.clear()
 316
 317
 318 def unregister_qcd_view_edit_mode_hotkeys():
 319     # remove keymaps when qcd view hotkeys are deactivated
 320     for km, kmi in addon_qcd_view_edit_mode_hotkey_keymaps:
 321         km.keymap_items.remove(kmi)
 322     addon_qcd_view_edit_mode_hotkey_keymaps.clear()
 323
 324
 325 def unregister_qcd_3dview_header_widget():
 326     bpy.types.VIEW3D_HT_header.remove(ui.view3d_header_qcd_slots)
 327     bpy.types.TOPBAR_HT_upper_bar.remove(ui.view_layer_update)