def add_menus(self, menubar):
edit_menu = QtWidgets.QMenu('Edit', menubar)
commonmenus.create_edit_menu_items(
edit_menu, reset_settings_func=self.reset_options)
menubar.addMenu(edit_menu)
help_menu = QtWidgets.QMenu('Help', menubar)
commonmenus.create_help_menu_items(help_menu,
tool_help_func=_open_help)
menubar.addMenu(help_menu)
def __init__(self, parent=None, *args, **kwargs):
super(ScreenSpaceRigBakeLayout, self).__init__(*args, **kwargs)
self.setupUi(self)
self.menubar = QtWidgets.QMenuBar(self)
self.options_menu = self.menubar.addMenu('Freeze')
self.refresh_menu = self.menubar.addMenu('Refresh')
# Freeze options actions
self.animlayer_action = QtWidgets.QAction('Add to AnimLayer', self)
self.animlayer_action.setCheckable(True)
self.options_menu.addAction(self.animlayer_action)
self.options_menu.addSeparator()
self.freeze_action_group = QtWidgets.QActionGroup(self.options_menu)
self.camera_space_action = QtWidgets.QAction('Freeze in Camera Space',
self)
self.camera_space_action.setCheckable(True)
self.options_menu.addAction(self.camera_space_action)
self.camera_space_action.setActionGroup(self.freeze_action_group)
self.world_space_action = QtWidgets.QAction('Freeze in World Space',
self)
self.world_space_action.setCheckable(True)
self.world_space_action.setChecked(True)
self.options_menu.addAction(self.world_space_action)
self.world_space_action.setActionGroup(self.freeze_action_group)
# Refresh action
self.refresh_action = QtWidgets.QAction('Refresh Rigs List', self)
self.refresh_menu.addAction(self.refresh_action)
# Set menubar
self.gridLayout.setMenuBar(self.menubar)
# Help menu
help_menu = QtWidgets.QMenu('Help', self.menubar)
commonmenus.create_help_menu_items(help_menu, tool_help_func=_open_help)
# Add menus
self.menubar.addMenu(self.options_menu)
self.menubar.addMenu(self.refresh_menu)
self.menubar.addMenu(help_menu)
self.create_connections()
self.refresh_rigsList()
self.populate_ui()
def contextMenuEvent(self, event):
LOG.debug('Object TreeView Context Menu Event: %r', event)
menu = QtWidgets.QMenu(self)
label = 'Select Marker / Bundle'
swap_sel_act = QtWidgets.QAction(label, self)
swap_sel_act.triggered.connect(self.selection_swap)
label = 'Select Marker + Bundle'
both_sel_act = QtWidgets.QAction(label, self)
both_sel_act.triggered.connect(self.selection_both_markers_bundles)
label = 'Marker Bundle Rename...'
rename_act = QtWidgets.QAction(label, self)
rename_act.triggered.connect(self.rename_marker_bundles)
menu.addAction(swap_sel_act)
menu.addAction(both_sel_act)
menu.addSeparator()
menu.addAction(rename_act)
menu.exec_(event.globalPos())
return
def addMenuBarContents(self, menubar):
# File Menu
file_menu = QtWidgets.QMenu('File', menubar)
# New Collection
label = 'New Collection'
tooltip = 'Create a new Collection node.'
action = QtWidgets.QAction(label, file_menu)
action.setStatusTip(tooltip)
action.triggered.connect(partial(self.createNewCollectionNodeCB))
file_menu.addAction(action)
# Rename Collection
label = 'Rename Collection...'
tooltip = 'Rename a Collection node.'
action = QtWidgets.QAction(label, file_menu)
action.setStatusTip(tooltip)
action.triggered.connect(partial(self.renameCollectionNodeCB))
file_menu.addAction(action)
file_menu.addSeparator()
# Remove Collection
label = 'Remove Collection...'
tooltip = 'Remove a Collection node.'
action = QtWidgets.QAction(label, file_menu)
action.setStatusTip(tooltip)
action.triggered.connect(partial(self.removeCollectionNodeCB))
file_menu.addAction(action)
file_menu.addSeparator()
# Close Window
label = 'Close Window'
tooltip = 'Close the mmSolver window.'
action = QtWidgets.QAction(label, file_menu)
action.setStatusTip(tooltip)
action.triggered.connect(partial(self.close))
file_menu.addAction(action)
menubar.addMenu(file_menu)
# Edit Menu
edit_menu = QtWidgets.QMenu('Edit', menubar)
edit_menu.setTearOffEnabled(True)
if Qt.IsPySide2 or Qt.IsPyQt5:
edit_menu.addSection('Undo / Redo')
# Undo
label = 'Undo last command (with disabled viewport)'
tooltip = ('Undo the Maya scene state, '
'without updating the viewport or solver UI')
action = QtWidgets.QAction(label, edit_menu)
action.setStatusTip(tooltip)
action.triggered.connect(self.undoTriggeredCB)
edit_menu.addAction(action)
# Redo
label = 'Redo last command (with disabled viewport)'
tooltip = ('Redo the Maya scene state, '
'without updating the viewport or solver UI')
action = QtWidgets.QAction(label, edit_menu)
action.setStatusTip(tooltip)
action.triggered.connect(self.redoTriggeredCB)
edit_menu.addAction(action)
if Qt.IsPySide2 or Qt.IsPyQt5:
edit_menu.addSection('Window Update')
# Auto Update Solver Validation
label = 'Auto-Update Solver Validation'
tooltip = 'Auto-update details of the solver parameter/error numbers.'
value = lib_state.get_auto_update_solver_validation_state()
action = QtWidgets.QAction(label, edit_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.solver_settings.autoUpdateSolverValidationChanged)
edit_menu.addAction(action)
if Qt.IsPySide2 or Qt.IsPyQt5:
edit_menu.addSection('Solver Execution')
# Pre-Solve Force Evaluation
label = 'Pre-Solve Force Evaluation'
tooltip = ('Before starting a solve, '
'update the scene to force an evaluation.')
pre_solve_force_eval = lib_state.get_pre_solve_force_eval_state()
action = QtWidgets.QAction(label, edit_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(pre_solve_force_eval)
action.toggled.connect(type(self).preSolveForceEvalActionToggledCB)
edit_menu.addAction(action)
# Refresh Viewport During Solve
label = 'Refresh Viewport'
tooltip = 'Refresh the viewport while Solving.'
refresh_value = lib_state.get_refresh_viewport_state()
action = QtWidgets.QAction(label, edit_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(refresh_value)
action.toggled.connect(type(self).refreshActionToggledCB)
edit_menu.addAction(action)
# Force DG evaluation.
label = 'Force DG Update'
tooltip = 'Force Maya DG Evaluation while solving.'
force_dg_update_value = lib_state.get_force_dg_update_state()
action = QtWidgets.QAction(label, edit_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(force_dg_update_value)
action.toggled.connect(type(self).forceDgUpdateActionToggledCB)
edit_menu.addAction(action)
menubar.addMenu(edit_menu)
# View Menu
view_menu = QtWidgets.QMenu('View', menubar)
view_menu.setTearOffEnabled(True)
if Qt.IsPySide2 or Qt.IsPyQt5:
view_menu.addSection('Input Object Display')
# Display Object Weight
label = 'Weight Column'
tooltip = 'Display Object weight column'
value = lib_state.get_display_object_weight_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.object_browser.displayWeightColumnChanged)
view_menu.addAction(action)
# Display Object Frame Deviation
label = 'Frame Deviation'
tooltip = 'Display per-frame deviation for each Marker/Camera.'
value = lib_state.get_display_object_frame_deviation_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.object_browser.displayFrameDeviationColumnChanged)
view_menu.addAction(action)
# Display Object Average Deviation
label = 'Average Deviation'
tooltip = 'Display average deviation column'
value = lib_state.get_display_object_average_deviation_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.object_browser.displayAverageDeviationColumnChanged)
view_menu.addAction(action)
# Display Object Maximum Deviation
label = 'Maximum Deviation'
tooltip = 'Display maximum deviation column'
value = lib_state.get_display_object_maximum_deviation_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.object_browser.displayMaximumDeviationColumnChanged)
view_menu.addAction(action)
if Qt.IsPySide2 or Qt.IsPyQt5:
view_menu.addSection('Output Attribute Display')
# Display Attribute State
label = 'Display Attribute State'
tooltip = 'Display Attribute State column'
value = lib_state.get_display_attribute_state_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.attribute_browser.displayStateColumnChanged)
view_menu.addAction(action)
# Display Attribute Smoothness
label = 'Display Attribute Smoothness'
tooltip = 'Display Attribute Smoothness columns'
value = lib_state.get_display_attribute_smoothness_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.attribute_browser.displaySmoothnessColumnChanged)
view_menu.addAction(action)
# Display Attribute Stiffness
label = 'Display Attribute Stiffness'
tooltip = 'Display Attribute Stiffness columns'
value = lib_state.get_display_attribute_stiffness_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.attribute_browser.displayStiffnessColumnChanged)
view_menu.addAction(action)
# Display Attribute Min/Max
label = 'Display Attribute Min/Max'
tooltip = 'Display Attribute Minimum and Maximum columns'
value = lib_state.get_display_attribute_min_max_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
self.subForm.attribute_browser.displayMinMaxColumnChanged)
view_menu.addAction(action)
if Qt.IsPySide2 or Qt.IsPyQt5:
view_menu.addSection('During Solve')
# Display the Image Planes while solving.
#
# TODO: Add other object types to show/hide while solving,
# such as camera, nurbsCurves, nurbsSurfaces, and locators.
label = 'Display Image Planes'
tooltip = 'Display Image Planes while solving.'
value = lib_state.get_display_image_plane_while_solving_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
type(self).displayImagePlaneWhileSolvingActionToggledCB)
view_menu.addAction(action)
# Display the Meshes while solving.
label = 'Display Meshes'
tooltip = 'Display Meshes while solving.'
value = lib_state.get_display_meshes_while_solving_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(value)
action.toggled.connect(
type(self).displayMeshesWhileSolvingActionToggledCB)
view_menu.addAction(action)
# Isolate Objects while solving
label = 'Isolate Objects'
tooltip = 'Isolate visibility of all Markers and Bundles while solving.'
isolate_value = lib_state.get_isolate_object_while_solving_state()
action = QtWidgets.QAction(label, view_menu)
action.setStatusTip(tooltip)
action.setCheckable(True)
action.setChecked(isolate_value)
action.toggled.connect(
type(self).isolateObjectWhileSolvingActionToggledCB)
view_menu.addAction(action)
menubar.addMenu(view_menu)
# Log Menu
# This menu depicts a radio button allowing the user to choose
# how much information is returned to the console (the logging
# level).
log_menu = QtWidgets.QMenu('Log', menubar)
# Errors
label = 'Errors'
tooltip = 'Send Errors to the log.'
logErrorAction = QtWidgets.QAction(label, log_menu)
logErrorAction.setStatusTip(tooltip)
logErrorAction.setCheckable(True)
logErrorAction.triggered.connect(partial(self.logErrorCB))
log_menu.addAction(logErrorAction)
# Warnings
label = 'Warnings'
tooltip = 'Send Warnings to the log.'
logWarningAction = QtWidgets.QAction(label, log_menu)
logWarningAction.setStatusTip(tooltip)
logWarningAction.setCheckable(True)
logWarningAction.triggered.connect(partial(self.logWarningCB))
log_menu.addAction(logWarningAction)
# Information
label = 'Info'
tooltip = 'Send Information to the log.'
logInfoAction = QtWidgets.QAction(label, log_menu)
logInfoAction.setStatusTip(tooltip)
logInfoAction.setCheckable(True)
logInfoAction.triggered.connect(partial(self.logInfoCB))
log_menu.addAction(logInfoAction)
# Verbose
label = 'Verbose'
tooltip = 'Send Verboses to the log.'
logVerboseAction = QtWidgets.QAction(label, log_menu)
logVerboseAction.setStatusTip(tooltip)
logVerboseAction.setCheckable(True)
logVerboseAction.triggered.connect(partial(self.logVerboseCB))
log_menu.addAction(logVerboseAction)
# Debug
label = 'Debug'
tooltip = 'Send Debug messages to the log.'
logDebugAction = QtWidgets.QAction(label, log_menu)
logDebugAction.setStatusTip(tooltip)
logDebugAction.setCheckable(True)
logDebugAction.triggered.connect(partial(self.logDebugCB))
log_menu.addAction(logDebugAction)
# 'Radio' button for logging levels
log_actionGroup = QtWidgets.QActionGroup(log_menu)
log_actionGroup.addAction(logErrorAction)
log_actionGroup.addAction(logWarningAction)
log_actionGroup.addAction(logInfoAction)
log_actionGroup.addAction(logVerboseAction)
log_actionGroup.addAction(logDebugAction)
log_level = lib_state.get_log_level()
if log_level == const.LOG_LEVEL_ERROR:
logErrorAction.setChecked(True)
elif log_level == const.LOG_LEVEL_WARNING:
logWarningAction.setChecked(True)
elif log_level == const.LOG_LEVEL_INFO:
logInfoAction.setChecked(True)
elif log_level == const.LOG_LEVEL_VERBOSE:
logVerboseAction.setChecked(True)
elif log_level == const.LOG_LEVEL_DEBUG:
logDebugAction.setChecked(True)
else:
LOG.warning('Invalid log level given: %r' % log_level)
menubar.addMenu(log_menu)
# Help Menu
help_menu = QtWidgets.QMenu('Help', menubar)
commonmenus.create_help_menu_items(help_menu,
tool_help_func=_open_help)
menubar.addMenu(help_menu)
return
def setupUi(self, parent):
self.transform_icon = QtGui.QIcon(':transform.svg')
# Main layout
self.gridLayout = QtWidgets.QGridLayout(parent)
# Menu
self.menu_bar = QtWidgets.QMenuBar()
self.options_menu = self.menu_bar.addMenu('Freeze')
self.refresh_menu = self.menu_bar.addMenu('Refresh')
# Freeze options actions
self.animlayer_action = QtWidgets.QAction('Add to AnimLayer', self)
self.animlayer_action.setCheckable(True)
self.options_menu.addAction(self.animlayer_action)
self.options_menu.addSeparator()
self.freeze_action_group = QtWidgets.QActionGroup(self)
self.camera_space_action = QtWidgets.QAction('Freeze in Camera Space',
self)
self.camera_space_action.setCheckable(True)
self.options_menu.addAction(self.camera_space_action)
self.camera_space_action.setActionGroup(self.freeze_action_group)
self.world_space_action = QtWidgets.QAction('Freeze in World Space',
self)
self.world_space_action.setCheckable(True)
self.world_space_action.setChecked(True)
self.options_menu.addAction(self.world_space_action)
self.world_space_action.setActionGroup(self.freeze_action_group)
# Refresh action
self.refresh_action = QtWidgets.QAction('Refresh Rigs list', self)
self.refresh_menu.addAction(self.refresh_action)
self.gridLayout.setMenuBar(self.menu_bar)
spacerItem = QtWidgets.QSpacerItem(5, 17,
QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Minimum)
self.gridLayout.addItem(spacerItem, 0, 0, 1, 1)
# Full bake radio button
self.full_bake_rdo_btn = QtWidgets.QRadioButton()
self.gridLayout.addWidget(self.full_bake_rdo_btn, 0, 1, 1, 1)
spacerItem1 = QtWidgets.QSpacerItem(5, 17,
QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Minimum)
self.gridLayout.addItem(spacerItem1, 0, 2, 1, 1)
# Smart bake radio button
self.smart_bake_rdo_btn = QtWidgets.QRadioButton()
self.smart_bake_rdo_btn.setLayoutDirection(QtCore.Qt.LeftToRight)
self.smart_bake_rdo_btn.setAutoFillBackground(False)
self.smart_bake_rdo_btn.setChecked(True)
self.gridLayout.addWidget(self.smart_bake_rdo_btn, 0, 3, 1, 1)
spacerItem2 = QtWidgets.QSpacerItem(5, 10,
QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Minimum)
self.gridLayout.addItem(spacerItem2, 0, 4, 1, 1)
self.horizontalLayout_2 = QtWidgets.QHBoxLayout()
# Rigs label
self.rigs_label = QtWidgets.QLabel()
self.horizontalLayout_2.addWidget(self.rigs_label)
# Rigs list
self.rigs_list = QtWidgets.QTreeWidget()
self.rigs_list.setFocusPolicy(QtCore.Qt.NoFocus)
self.rigs_list.setSelectionMode(
QtWidgets.QAbstractItemView.ExtendedSelection)
self.rigs_list.setSelectionBehavior(
QtWidgets.QAbstractItemView.SelectRows)
self.rigs_list.setIndentation(15)
self.rigs_list.setHeaderHidden(True)
self.horizontalLayout_2.addWidget(self.rigs_list)
self.gridLayout.addLayout(self.horizontalLayout_2, 2, 0, 1, 5)
self.horizontalLayout = QtWidgets.QHBoxLayout()
# Name label
self.name_label = QtWidgets.QLabel()
self.horizontalLayout.addWidget(self.name_label)
# Name line edit
self.name_text = QtWidgets.QLineEdit()
self.name_text.setFocusPolicy(QtCore.Qt.ClickFocus)
self.horizontalLayout.addWidget(self.name_text)
self.gridLayout.addLayout(self.horizontalLayout, 3, 0, 1, 5)
# Widget labels
self.full_bake_rdo_btn.setText('Full bake')
self.smart_bake_rdo_btn.setText('Smart bake')
self.rigs_label.setText('Rigs')
self.name_label.setText('Name')
self.context_menu = QtWidgets.QMenu()
self.create_rig_action = QtWidgets.QAction('Create Screen-Space Rig',
self)
self.context_menu.addAction(self.create_rig_action)
self.create_freeze_rig_action = QtWidgets.QAction(
'Create Freeze Rig', self)
self.context_menu.addAction(self.create_freeze_rig_action)
self.context_menu.addSeparator()
self.match_rig_action = QtWidgets.QAction('Match Screen Z-Depth', self)
self.context_menu.addAction(self.match_rig_action)
self.context_menu.addSeparator()
self.bake_rig_action = QtWidgets.QAction('Bake Rig', self)
self.context_menu.addAction(self.bake_rig_action)
self.delete_rig_action = QtWidgets.QAction('Delete Rig', self)
self.context_menu.addAction(self.delete_rig_action)
self.context_menu.addSeparator()
self.select_action = QtWidgets.QAction('Select in Outliner', self)
self.context_menu.addAction(self.select_action)
return
def contextMenuEvent(self, event):
LOG.debug('Attribute TreeView Context Menu Event: %r', event)
menu = QtWidgets.QMenu(self)
label = 'Edit Details...'
edit_act = QtWidgets.QAction(label, self)
edit_act.triggered.connect(self.set_details_selected_attributes)
label = 'Lock Attributes'
lock_act = QtWidgets.QAction(label, self)
lock_act.triggered.connect(self.lock_selected_attributes)
label = 'Unlock Attributes'
unlock_act = QtWidgets.QAction(label, self)
unlock_act.triggered.connect(self.unlock_selected_attributes)
label = 'Set Keyframe'
set_key_act = QtWidgets.QAction(label, self)
set_key_act.triggered.connect(self.set_keyframe_on_selected_attributes)
label = 'Delete Keyframe'
delete_key_current_frame_act = QtWidgets.QAction(label, self)
delete_key_current_frame_act.triggered.connect(
self.delete_keyframe_current_frame_on_selected_attributes)
label = 'Delete Keyframes (Timeline)'
delete_key_all_frames_act = QtWidgets.QAction(label, self)
delete_key_all_frames_act.triggered.connect(
self.delete_keyframe_all_frames_on_selected_attributes)
label = 'Delete Static Channels'
delete_static_keys_act = QtWidgets.QAction(label, self)
delete_static_keys_act.triggered.connect(
self.delete_static_channel_on_selected_attributes)
label = 'Break Connections'
break_conn_act = QtWidgets.QAction(label, self)
break_conn_act.triggered.connect(
self.break_connections_on_selected_attributes)
label = 'Bake Attributes'
bake_attr_act = QtWidgets.QAction(label, self)
bake_attr_act.triggered.connect(self.bake_selected_attributes)
label = 'Reset Values'
reset_values_act = QtWidgets.QAction(label, self)
reset_values_act.triggered.connect(
self.reset_values_on_selected_attributes)
menu.addAction(edit_act)
menu.addSeparator()
menu.addAction(set_key_act)
menu.addSeparator()
menu.addAction(delete_key_current_frame_act)
menu.addAction(delete_key_all_frames_act)
menu.addAction(delete_static_keys_act)
menu.addSeparator()
menu.addAction(break_conn_act)
menu.addAction(bake_attr_act)
menu.addAction(reset_values_act)
menu.addSeparator()
menu.addAction(lock_act)
menu.addAction(unlock_act)
menu.exec_(event.globalPos())
return
def add_menus(self, menubar):
help_menu = QtWidgets.QMenu('Help', menubar)
commonmenus.create_help_menu_items(help_menu,
tool_help_func=_open_help)
menubar.addMenu(help_menu)