Skip to content
Snippets Groups Projects
ui_pie_menus_official.py 15 KiB
Newer Older
  • Learn to ignore specific revisions
  • # ##### BEGIN GPL LICENSE BLOCK #####
    #
    #  This program is free software; you can redistribute it and/or
    #  modify it under the terms of the GNU General Public License
    #  as published by the Free Software Foundation; either version 2
    #  of the License, or (at your option) any later version.
    #
    #  This program is distributed in the hope that it will be useful,
    #  but WITHOUT ANY WARRANTY; without even the implied warranty of
    #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    #  GNU General Public License for more details.
    #
    #  You should have received a copy of the GNU General Public License
    #  along with this program; if not, write to the Free Software Foundation,
    #  Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
    #
    # ##### END GPL LICENSE BLOCK #####
    
    # <pep8 compliant>
    
    bl_info = {
        "name": "Pie Menus Official",
    
        "author": "Antony Riakiotakis, Sebastian Koenig",
        "version": (1, 0, 2),
    
        "blender": (2, 71, 4),
    
        "description": "Enable official Pie Menus in Blender",
    
        "category": "User Interface",
    
        "wiki_url": "http://wiki.blender.org/index.php/Dev:Ref/Release_Notes/2.72/UI/Pie_Menus"
    
    from bpy.types import Menu, Operator
    from bpy.props import EnumProperty
    
    
    
    class VIEW3D_PIE_object_mode(Menu):
        bl_label = "Mode"
    
        def draw(self, context):
            layout = self.layout
    
            pie = layout.menu_pie()
            pie.operator_enum("OBJECT_OT_mode_set", "mode")
    
    
    class VIEW3D_PIE_view_more(Menu):
        bl_label = "More"
    
        def draw(self, context):
            layout = self.layout
    
            pie = layout.menu_pie()
            pie.operator("VIEW3D_OT_view_persportho", text="Persp/Ortho", icon='RESTRICT_VIEW_OFF')
            pie.operator("VIEW3D_OT_camera_to_view")
            pie.operator("VIEW3D_OT_view_selected")
            pie.operator("VIEW3D_OT_view_all")
    
            pie.operator("VIEW3D_OT_localview")
    
            pie.operator("SCREEN_OT_region_quadview")
    
    
    class VIEW3D_PIE_view(Menu):
        bl_label = "View"
    
        def draw(self, context):
            layout = self.layout
    
            pie = layout.menu_pie()
            pie.operator_enum("VIEW3D_OT_viewnumpad", "type")
    
            pie.operator("wm.call_menu_pie", text="More", icon='PLUS').name = "VIEW3D_PIE_view_more"
    
    
    
    class VIEW3D_PIE_shade(Menu):
        bl_label = "Shade"
    
        def draw(self, context):
            layout = self.layout
    
            pie = layout.menu_pie()
            pie.prop(context.space_data, "viewport_shade", expand=True)
    
            if context.active_object:
                if(context.mode == 'EDIT_MESH'):
                    pie.operator("MESH_OT_faces_shade_smooth")
                    pie.operator("MESH_OT_faces_shade_flat")
                else:
                    pie.operator("OBJECT_OT_shade_smooth")
                    pie.operator("OBJECT_OT_shade_flat")
    
    
    
    class VIEW3D_manipulator_set(Operator):
        bl_label = "Set Manipulator"
        bl_idname = "view3d.manipulator_set"
    
        type = EnumProperty(
                name="Type",
                items=(('TRANSLATE', "Translate", "Use the manipulator for movement transformations"),
                       ('ROTATE', "Rotate", "Use the manipulator for rotation transformations"),
                       ('SCALE', "Scale", "Use the manipulator for scale transformations"),
    
            # show manipulator if user selects an option
    
            context.space_data.show_manipulator = True
    
            context.space_data.transform_manipulators = {self.type}
    
            return {'FINISHED'}
    
    
    
    class VIEW3D_PIE_manipulator(Menu):
        bl_label = "Manipulator"
    
        def draw(self, context):
            layout = self.layout
    
            pie = layout.menu_pie()
    
            pie.operator("view3d.manipulator_set", icon='MAN_TRANS', text="Translate").type = 'TRANSLATE'
            pie.operator("view3d.manipulator_set", icon='MAN_ROT', text="Rotate").type = 'ROTATE'
            pie.operator("view3d.manipulator_set", icon='MAN_SCALE', text="Scale").type = 'SCALE'
    
            pie.prop(context.space_data, "show_manipulator")
    
    
    class VIEW3D_PIE_pivot(Menu):
        bl_label = "Pivot"
    
        def draw(self, context):
            layout = self.layout
    
            pie = layout.menu_pie()
            pie.prop(context.space_data, "pivot_point", expand=True)
            if context.active_object.mode == 'OBJECT':
                pie.prop(context.space_data, "use_pivot_point_align", text="Center Points")
    
    
    class VIEW3D_PIE_snap(Menu):
        bl_label = "Snapping"
    
        def draw(self, context):
            layout = self.layout
    
            toolsettings = context.tool_settings
            pie = layout.menu_pie()
            pie.prop(toolsettings, "snap_element", expand=True)
            pie.prop(toolsettings, "use_snap")
    
    
    
    Campbell Barton's avatar
    Campbell Barton committed
    class CLIP_PIE_refine_pie(Menu):
        # Refinement Options
        bl_label = "Refine Intrinsics"
    
        def draw(self, context):
            clip = context.space_data.clip
            settings = clip.tracking.settings
    
            layout = self.layout
            pie = layout.menu_pie()
    
            pie.prop(settings, "refine_intrinsics", expand=True)
    
    
    class CLIP_PIE_geometry_reconstruction(Menu):
        # Geometry Reconstruction
        bl_label = "Reconstruction"
    
        def draw(self, context):
            layout = self.layout
            pie = layout.menu_pie()
    
            pie.operator("clip.bundles_to_mesh", icon='MESH_DATA')
            pie.operator("clip.track_to_empty", icon='EMPTY_DATA')
    
    
    class CLIP_PIE_proxy_pie(Menu):
        # Proxy Controls
        bl_label = "Proxy Size"
    
        def draw(self, context):
            space = context.space_data
    
            layout = self.layout
            pie = layout.menu_pie()
    
            pie.prop(space.clip, "use_proxy", text="Use Proxy")
            pie.prop(space.clip_user, "proxy_render_size", expand=True)
    
    
    class CLIP_PIE_display_pie(Menu):
        # Display Options
        bl_label = "Marker Display"
    
        def draw(self, context):
            space = context.space_data
    
            layout = self.layout
            pie = layout.menu_pie()
    
            pie.prop(space, "show_names", text="Show Track Info", icon='WORDWRAP_ON')
            pie.prop(space, "show_disabled", text="Show Disabled Tracks", icon='VISIBLE_IPO_ON')
            pie.prop(space, "show_marker_search", text="Display Search Area", icon='VIEWZOOM')
            pie.prop(space, "show_marker_pattern", text="Display Pattern Area", icon='BORDERMOVE')
    
    
    class CLIP_PIE_marker_pie(Menu):
        # Settings for the individual markers
        bl_label = "Marker Settings"
    
        def draw(self, context):
            clip = context.space_data.clip
            track_active = clip.tracking.tracks.active
    
            layout = self.layout
            pie = layout.menu_pie()
    
            prop = pie.operator("wm.context_set_enum", text="Loc", icon='OUTLINER_DATA_EMPTY')
            prop.data_path = "space_data.clip.tracking.tracks.active.motion_model"
            prop.value = "Loc"
            prop = pie.operator("wm.context_set_enum", text="Affine", icon='OUTLINER_DATA_LATTICE')
            prop.data_path = "space_data.clip.tracking.tracks.active.motion_model"
            prop.value = "Affine"
    
            pie.operator("clip.track_settings_to_track", icon='COPYDOWN')
            pie.operator("clip.track_settings_as_default", icon='SETTINGS')
    
            if track_active:
                pie.prop(track_active, "use_normalization", text="Normalization")
                pie.prop(track_active, "use_brute", text="Use Brute Force")
                pie.prop(track_active, "use_blue_channel", text="Blue Channel")
    
                if track_active.pattern_match == "PREV_FRAME":
                    prop = pie.operator("wm.context_set_enum", text="Match Previous", icon='KEYINGSET')
                    prop.data_path = "space_data.clip.tracking.tracks.active.pattern_match"
                    prop.value = 'KEYFRAME'
                else:
                    prop = pie.operator("wm.context_set_enum", text="Match Keyframe", icon='KEY_HLT')
    
                    prop.data_path = "space_data.clip.tracking.tracks.active.pattern_match"
    
    Campbell Barton's avatar
    Campbell Barton committed
                    prop.value = 'PREV_FRAME'
    
    
    class CLIP_PIE_tracking_pie(Menu):
        # Tracking Operators
        bl_label = "Tracking"
    
        def draw(self, context):
            layout = self.layout
            pie = layout.menu_pie()
    
            prop = pie.operator("clip.track_markers", icon='PLAY_REVERSE')
            prop.backwards = True
            prop.sequence = True
            prop = pie.operator("clip.track_markers", icon='PLAY')
            prop.backwards = False
            prop.sequence = True
    
            pie.operator("clip.disable_markers", icon='RESTRICT_VIEW_ON')
            pie.operator("clip.detect_features", icon='ZOOM_SELECTED')
    
            pie.operator("clip.clear_track_path", icon='BACK').action = 'UPTO'
            pie.operator("clip.clear_track_path", icon='FORWARD').action = 'REMAINED'
    
            pie.operator("clip.refine_markers", icon='LOOP_BACK').backwards = True
            pie.operator("clip.refine_markers", icon='LOOP_FORWARDS').backwards = False
    
    
    class CLIP_PIE_clipsetup_pie(Menu):
        # Setup the clip display options
        bl_label = "Clip and Display Setup"
    
        def draw(self, context):
            space = context.space_data
    
            layout = self.layout
            pie = layout.menu_pie()
    
            pie.operator("clip.reload", text="Reload Footage", icon='FILE_REFRESH')
            pie.operator("clip.prefetch", text="Prefetch Footage", icon='LOOP_FORWARDS')
    
            pie.prop(space, "use_mute_footage", text="Mute Footage", icon='MUTE_IPO_ON')
            pie.prop(space.clip_user, "use_render_undistorted", text="Render Undistorted")
            pie.operator("clip.set_scene_frames", text="Set Scene Frames", icon='SCENE_DATA')
            pie.operator("wm.call_menu_pie", text="Marker Display", icon='PLUS').name = "CLIP_PIE_display_pie"
            pie.operator("clip.set_active_clip", icon='CLIP')
            pie.operator("wm.call_menu_pie", text="Proxy", icon='PLUS').name = "CLIP_PIE_proxy_pie"
    
    
    class CLIP_PIE_solver_pie(Menu):
        # Operators to solve the scene
        bl_label = "Solving"
    
        def draw(self, context):
            clip = context.space_data.clip
            settings = clip.tracking.settings
    
            layout = self.layout
            pie = layout.menu_pie()
    
            pie.operator("clip.create_plane_track", icon='MESH_PLANE')
            pie.operator("clip.solve_camera", text="Solve Camera", icon='OUTLINER_OB_CAMERA')
    
            pie.operator("wm.call_menu_pie", text="Refinement", icon='CAMERA_DATA').name = "CLIP_PIE_refine_pie"
            pie.prop(settings, "use_tripod_solver", text="Tripod Solver")
    
            pie.operator("clip.set_solver_keyframe", text="Set Keyframe A", icon='KEY_HLT').keyframe = 'KEYFRAME_A'
            pie.operator("clip.set_solver_keyframe", text="Set Keyframe B", icon='KEY_HLT').keyframe = 'KEYFRAME_B'
    
            prop = pie.operator("clip.clean_tracks", icon='STICKY_UVS_DISABLE')
            pie.operator("clip.filter_tracks", icon='FILTER')
            prop.frames = 15
            prop.error = 2
    
    
    class CLIP_PIE_reconstruction_pie(Menu):
        # Scene Reconstruction
        bl_label = "Reconstruction"
    
        def draw(self, context):
            layout = self.layout
            pie = layout.menu_pie()
    
            pie.operator("clip.set_viewport_background", text="Set Viewport Background", icon='SCENE_DATA')
            pie.operator("clip.setup_tracking_scene", text="Setup Tracking Scene", icon='SCENE_DATA')
    
            pie.operator("clip.set_plane", text="Setup Floor", icon='MESH_PLANE')
            pie.operator("clip.set_origin", text="Set Origin", icon='MANIPUL')
    
            pie.operator("clip.set_axis", text="Set X Axis", icon='AXIS_FRONT').axis = 'X'
            pie.operator("clip.set_axis", text="Set Y Axis", icon='AXIS_SIDE').axis = 'Y'
    
            pie.operator("clip.set_scale", text="Set Scale", icon='ARROW_LEFTRIGHT')
            pie.operator("wm.call_menu_pie", text="Reconstruction", icon='MESH_DATA').name = "CLIP_PIE_geometry_reconstruction"
    
    
    class CLIP_PIE_timecontrol_pie(Menu):
        # Time Controls
        bl_label = "Time Control"
    
        def draw(self, context):
            layout = self.layout
            pie = layout.menu_pie()
    
            pie.operator("screen.frame_jump", text="Jump to Startframe", icon='TRIA_LEFT').end = False
            pie.operator("screen.frame_jump", text="Jump to Endframe", icon='TRIA_RIGHT').end = True
    
            pie.operator("clip.frame_jump", text="Start of Track", icon='REW').position = 'PATHSTART'
            pie.operator("clip.frame_jump", text="End of Track", icon='FF').position = 'PATHEND'
    
            pie.operator("screen.animation_play", text="Playback Backwards", icon='PLAY_REVERSE').reverse = True
            pie.operator("screen.animation_play", text="Playback Forwards", icon='PLAY').reverse = False
    
            pie.operator("screen.frame_offset", text="Previous Frame", icon='TRIA_LEFT').delta = -1
            pie.operator("screen.frame_offset", text="Next Frame", icon='TRIA_RIGHT').delta = 1
    
    
    
    addon_keymaps = []
    
    
    classes = (
        VIEW3D_manipulator_set,
    
        VIEW3D_PIE_object_mode,
        VIEW3D_PIE_view,
        VIEW3D_PIE_view_more,
        VIEW3D_PIE_shade,
        VIEW3D_PIE_manipulator,
        VIEW3D_PIE_pivot,
        VIEW3D_PIE_snap,
    
    Campbell Barton's avatar
    Campbell Barton committed
    
        CLIP_PIE_geometry_reconstruction,
        CLIP_PIE_tracking_pie,
        CLIP_PIE_display_pie,
        CLIP_PIE_proxy_pie,
        CLIP_PIE_marker_pie,
        CLIP_PIE_solver_pie,
        CLIP_PIE_refine_pie,
        CLIP_PIE_reconstruction_pie,
        CLIP_PIE_clipsetup_pie,
        CLIP_PIE_timecontrol_pie,
    
    
    def register():
        for cls in classes:
            bpy.utils.register_class(cls)
    
    
        wm = bpy.context.window_manager
    
    
        if wm.keyconfigs.addon:
            km = wm.keyconfigs.addon.keymaps.new(name='Object Non-modal')
            kmi = km.keymap_items.new('wm.call_menu_pie', 'TAB', 'PRESS')
            kmi.properties.name = 'VIEW3D_PIE_object_mode'
            kmi = km.keymap_items.new('wm.call_menu_pie', 'Z', 'PRESS')
            kmi.properties.name = 'VIEW3D_PIE_shade'
            kmi = km.keymap_items.new('wm.call_menu_pie', 'Q', 'PRESS')
            kmi.properties.name = 'VIEW3D_PIE_view'
            kmi = km.keymap_items.new('wm.call_menu_pie', 'SPACE', 'PRESS', ctrl=True)
            kmi.properties.name = 'VIEW3D_PIE_manipulator'
            kmi = km.keymap_items.new('wm.call_menu_pie', 'PERIOD', 'PRESS')
            kmi.properties.name = 'VIEW3D_PIE_pivot'
    
            kmi = km.keymap_items.new('wm.call_menu_pie', 'TAB', 'PRESS', ctrl=True, shift=True)
    
            kmi.properties.name = 'VIEW3D_PIE_snap'
    
    
    Campbell Barton's avatar
    Campbell Barton committed
            km = wm.keyconfigs.addon.keymaps.new(name="Clip", space_type='CLIP_EDITOR')
            kmi = km.keymap_items.new("wm.call_menu_pie", 'Q', 'PRESS')
            kmi.properties.name = "CLIP_PIE_marker_pie"
            kmi = km.keymap_items.new("wm.call_menu_pie", 'W', 'PRESS')
            kmi.properties.name = "CLIP_PIE_clipsetup_pie"
            kmi = km.keymap_items.new("wm.call_menu_pie", 'E', 'PRESS')
            kmi.properties.name = "CLIP_PIE_tracking_pie"
            kmi = km.keymap_items.new("wm.call_menu_pie", 'S', 'PRESS', shift=True)
            kmi.properties.name = "CLIP_PIE_solver_pie"
            kmi = km.keymap_items.new("wm.call_menu_pie", 'W', 'PRESS', shift=True)
            kmi.properties.name = "CLIP_PIE_reconstruction_pie"
    
            km = wm.keyconfigs.addon.keymaps.new(name="Frames")
            kmi = km.keymap_items.new("wm.call_menu_pie", 'A', 'PRESS', oskey=True)
            kmi.properties.name = "CLIP_PIE_timecontrol_pie"
    
    
    
    
    def unregister():
    
        for cls in classes:
            bpy.utils.unregister_class(cls)
    
    
        wm = bpy.context.window_manager
    
    
        if wm.keyconfigs.addon:
            for km in addon_keymaps:
                for kmi in km.keymap_items:
                    km.keymap_items.remove(kmi)
    
                wm.keyconfigs.addon.keymaps.remove(km)
    
        addon_keymaps.clear()