def createTreeView(self): """ Set up the tree view. """ self.treeView = object_treeview.ObjectTreeView() self.ui.treeViewLayout.addWidget(self.treeView) root = object_nodes.ObjectNode('root') self.model = object_nodes.ObjectModel(root, font=self.font) self.filterModel = QtCore.QSortFilterProxyModel() self.filterModel.setSourceModel(self.model) self.filterModel.setDynamicSortFilter(False) self.header = QtWidgets.QHeaderView(QtCore.Qt.Horizontal, parent=self.treeView) Qt.QtCompat.QHeaderView.setSectionResizeMode( self.header, QtWidgets.QHeaderView.ResizeToContents) self.treeView.setHeader(self.header) self.treeView.setModel(self.filterModel) self.treeView.setSortingEnabled(True) self.treeView.sortByColumn(0, QtCore.Qt.AscendingOrder) self.treeView.setSelectionMode( QtWidgets.QAbstractItemView.MultiSelection) self.selModel = self.treeView.selectionModel() self.selModel.selectionChanged.connect(self.selectionChanged) # Always hide the UUID Column - it's used for selection of # ModelIndexes with Maya node UUIDs only. hidden = True column = self.model.getColumnIndexFromColumnName( const.OBJECT_COLUMN_NAME_UUID) self.treeView.setColumnHidden(column, hidden) return
def markersToUINodes(mkr_list): """ Convert a list of markers into a hierarchy to show the user. """ root = object_nodes.ObjectNode('root') cam_nodes_store = {} for mkr in mkr_list: mkr_name = mkr.get_node() mkr_name = mkr_name.rpartition('|')[-1] cam = mkr.get_camera() bnd = mkr.get_bundle() # Get camera cam_shp_node = cam.get_shape_node() cam_name = cam.get_shape_node() cam_name = cam_name.rpartition('|')[-1] cam_node = None if cam_shp_node not in cam_nodes_store: data = { 'marker': mkr, 'camera': cam, } cam_node = object_nodes.CameraNode(cam_name, data=data, parent=root) cam_nodes_store[cam_shp_node] = cam_node else: cam_node = cam_nodes_store[cam_shp_node] assert cam_node is not None # The marker. data = { 'marker': mkr, 'camera': cam, } mkr_node = object_nodes.MarkerNode(mkr_name, data=data, parent=cam_node) # Get Bundle under marker. if bnd is None: continue bnd_name = bnd.get_node() bnd_name = bnd_name.rpartition('|')[-1] data = { 'marker': mkr, 'bundle': bnd, 'camera': cam, } assert mkr_node is not None bnd_node = object_nodes.BundleNode(bnd_name, data=data, parent=mkr_node) return root
def createTreeView(self): """ Set up the tree view. """ root = object_nodes.ObjectNode('root') self.model = object_nodes.ObjectModel(root, font=self.font) self.filterModel = QtCore.QSortFilterProxyModel() self.filterModel.setSourceModel(self.model) self.filterModel.setDynamicSortFilter(False) self.header = QtWidgets.QHeaderView(QtCore.Qt.Horizontal, parent=self.treeView) Qt.QtCompat.QHeaderView.setSectionResizeMode( self.header, QtWidgets.QHeaderView.ResizeToContents) self.treeView.setHeader(self.header) self.treeView.setModel(self.filterModel) self.treeView.setSortingEnabled(True) self.treeView.sortByColumn(0, QtCore.Qt.AscendingOrder) self.treeView.setSelectionMode( QtWidgets.QAbstractItemView.MultiSelection) self.selModel = self.treeView.selectionModel() self.selModel.selectionChanged.connect(self.selectionChanged) return
def __init__(self, parent=None, *args, **kwargs): super(SolverLayout, self).__init__(*args, **kwargs) self.setupUi(self) # Store the parent window class, so we can set the applyBtn enabled # state. self._parentObject = parent # Hide the Solve Info line until it's set up. GitHub Issue #56 self.solveInfoLine_lineEdit.setVisible(False) # Collection Combo Box. self.collectionName_model = uimodels.StringDataListModel() self.collectionName_comboBox.setModel(self.collectionName_model) self.collectionName_comboBox.currentIndexChanged.connect( self.collectionIndexChanged ) # Collection Select self.collectionSelect_pushButton.clicked.connect( self.collectionSelectClicked ) # Object Nodes # TODO: Perhaps we should remove a tree view and research how # we can embed a Maya Outliner inside our layout with a filter # to only show markers (and their parents), this would provide # a lot of default functionality from Maya. root = object_nodes.ObjectNode('root') self.object_model = object_nodes.ObjectModel(root, font=self.font) self.object_filterModel = QtCore.QSortFilterProxyModel() self.object_filterModel.setSourceModel(self.object_model) self.object_filterModel.setDynamicSortFilter(False) self.object_treeView.setModel(self.object_filterModel) self.object_treeView.setSortingEnabled(True) self.object_treeView.sortByColumn(0, QtCore.Qt.AscendingOrder) self.object_treeView.setSelectionMode(QtWidgets.QAbstractItemView.MultiSelection) self.object_selModel = self.object_treeView.selectionModel() self.object_selModel.selectionChanged.connect( self.objectNodeSelectionChanged ) self.objectToggleCamera_toolButton.clicked.connect( self.objectToggleCameraClicked, ) self.objectToggleMarker_toolButton.clicked.connect( self.objectToggleMarkerClicked, ) self.objectToggleBundle_toolButton.clicked.connect( self.objectToggleBundleClicked, ) # Object Add and Remove buttons self.objectAdd_toolButton.clicked.connect( self.objectAddClicked ) self.objectRemove_toolButton.clicked.connect( self.objectRemoveClicked ) # Attr Nodes root = attr_nodes.PlugNode('root') self.attribute_model = attr_nodes.AttrModel(root, font=self.font) self.attribute_filterModel = QtCore.QSortFilterProxyModel() self.attribute_filterModel.setSourceModel(self.attribute_model) self.attribute_filterModel.setDynamicSortFilter(False) self.attribute_treeView.setModel(self.attribute_filterModel) self.attribute_treeView.setSortingEnabled(True) self.attribute_treeView.sortByColumn(0, QtCore.Qt.AscendingOrder) self.attribute_treeView.setSelectionMode(QtWidgets.QAbstractItemView.MultiSelection) self.attribute_selModel = self.attribute_treeView.selectionModel() self.attribute_selModel.selectionChanged.connect( self.attrNodeSelectionChanged ) # Attr Add and Remove buttons self.attributeAdd_toolButton.clicked.connect( self.attrAddClicked ) self.attributeRemove_toolButton.clicked.connect( self.attrRemoveClicked ) # Solver Nodes self.solver_model = solver_nodes.SolverModel(font=self.font) self.solver_filterModel = QtCore.QSortFilterProxyModel() self.solver_filterModel.setSourceModel(self.solver_model) self.solver_filterModel.setDynamicSortFilter(False) self.solver_tableView.setModel(self.solver_filterModel) self.solver_tableView.setSortingEnabled(False) self.solver_selModel = self.solver_tableView.selectionModel() # Set up custom widgets for viewing and editing the columns. self.solver_attrFilterDelegate = solver_nodes.AttributeComboBoxDelegate() self.solver_tableView.setItemDelegateForColumn( 2, self.solver_attrFilterDelegate, ) self.solver_strategyDelegate = solver_nodes.StrategyComboBoxDelegate() self.solver_tableView.setItemDelegateForColumn( 3, self.solver_strategyDelegate, ) # Solver Add and Remove buttons self.solverAdd_toolButton.clicked.connect( self.solverAddClicked ) self.solverRemove_toolButton.clicked.connect( self.solverRemoveClicked ) # TODO: Write functions for Move Up/Down buttons. Until then, # we'll hide the buttons to avoid confusion. self.solverMoveUp_toolButton.setVisible(False) self.solverMoveDown_toolButton.setVisible(False) # self.solverMoveUp_toolButton.clicked.connect( # self.solverMoveUpClicked # ) # self.solverMoveDown_toolButton.clicked.connect( # self.solverMoveDownClicked # ) # Override Current Frame self.overrideCurrentFrame_checkBox.stateChanged.connect( self.overrideCurrentFrameChanged ) # Populate the UI with data. self.updateDynamicWindowTitle() self.updateCollectionModel() self.updateObjectToggleButtons() self.updateObjectModel() self.updateAttributeModel() self.updateSolverModel() self.updateSolveValidState() return
def markersToUINodes(mkr_list, show_cam, show_mkr, show_bnd): """ Convert a list of markers into a hierarchy to show the user. :param mkr_list: List of Marker objects to convert into UI nodes. :type mkr_list: [Marker, ..] :param show_cam: Should we show cameras? :type show_cam: bool :param show_mkr: Should we show markers? :type show_mkr: bool :param show_bnd: Should we show bundles? :type show_bnd: bool :return: A list of UI MarkerNode objects. :rtype: [MarkerNode, ..] """ assert isinstance(show_cam, bool) assert isinstance(show_mkr, bool) assert isinstance(show_bnd, bool) root = object_nodes.ObjectNode('root') cam_nodes_store = {} for mkr in mkr_list: mkr_name = mkr.get_node() mkr_name = mkr_name.rpartition('|')[-1] cam = mkr.get_camera() bnd = mkr.get_bundle() # Get camera if show_cam is False: cam_node = root else: cam_tfm_node = cam.get_transform_node() cam_name = cam.get_transform_node() cam_name = cam_name.rpartition('|')[-1] cam_node = None if cam_tfm_node not in cam_nodes_store: cam_uuid = cam.get_transform_uid() data = { 'uuid': cam_uuid, 'marker': mkr, 'camera': cam, } cam_node = object_nodes.CameraNode(cam_name, data=data, parent=root) cam_nodes_store[cam_tfm_node] = cam_node else: cam_node = cam_nodes_store[cam_tfm_node] assert cam_node is not None # The marker. mkr_node = cam_node if show_mkr is True: mkr_uuid = mkr.get_node_uid() data = { 'uuid': mkr_uuid, 'marker': mkr, 'camera': cam, } mkr_node = object_nodes.MarkerNode(mkr_name, data=data, parent=cam_node) # Get Bundle under marker. if bnd is None: continue if show_bnd is False: continue bnd_name = bnd.get_node() bnd_name = bnd_name.rpartition('|')[-1] bnd_uuid = bnd.get_node_uid() data = { 'marker': mkr, 'bundle': bnd, 'camera': cam, 'uuid': bnd_uuid, } assert mkr_node is not None bnd_node = object_nodes.BundleNode(bnd_name, data=data, parent=mkr_node) return root
def __init__(self, parent=None, *args, **kwargs): super(SolverLayout, self).__init__(*args, **kwargs) self.setupUi(self) # Store the parent window class, so we can set the applyBtn enabled # state. self._parentObject = parent # Collection Combo Box. self.collectionName_model = uimodels.StringDataListModel() self.collectionName_comboBox.setModel(self.collectionName_model) self.collectionName_comboBox.currentIndexChanged.connect( self.collectionIndexChanged) # Collection Select self.collectionSelect_pushButton.clicked.connect( self.collectionSelectClicked) # Object Nodes root = object_nodes.ObjectNode('root') self.object_model = object_nodes.ObjectModel(root, font=self.font) self.object_filterModel = QtCore.QSortFilterProxyModel() self.object_filterModel.setSourceModel(self.object_model) self.object_filterModel.setDynamicSortFilter(False) self.object_header = QtWidgets.QHeaderView(QtCore.Qt.Horizontal, parent=self.object_treeView) self.object_header.setSectionResizeMode( QtWidgets.QHeaderView.ResizeToContents) self.object_treeView.setHeader(self.object_header) self.object_treeView.setModel(self.object_filterModel) self.object_treeView.setSortingEnabled(True) self.object_treeView.sortByColumn(0, QtCore.Qt.AscendingOrder) self.object_treeView.setSelectionMode( QtWidgets.QAbstractItemView.MultiSelection) self.object_selModel = self.object_treeView.selectionModel() self.object_selModel.selectionChanged.connect( self.objectNodeSelectionChanged) self.objectToggleCamera_toolButton.clicked.connect( self.objectToggleCameraClicked, ) self.objectToggleMarker_toolButton.clicked.connect( self.objectToggleMarkerClicked, ) self.objectToggleBundle_toolButton.clicked.connect( self.objectToggleBundleClicked, ) # Object Add and Remove buttons self.objectAdd_toolButton.clicked.connect(self.objectAddClicked) self.objectRemove_toolButton.clicked.connect(self.objectRemoveClicked) # Attr Nodes root = attr_nodes.PlugNode('root') self.attribute_model = attr_nodes.AttrModel(root, font=self.font) self.attribute_filterModel = QtCore.QSortFilterProxyModel() self.attribute_filterModel.setSourceModel(self.attribute_model) self.attribute_filterModel.setDynamicSortFilter(False) self.attribute_header = QtWidgets.QHeaderView( QtCore.Qt.Horizontal, parent=self.attribute_treeView) self.attribute_header.setSectionResizeMode( QtWidgets.QHeaderView.ResizeToContents) self.attribute_treeView.setHeader(self.attribute_header) self.attribute_treeView.setModel(self.attribute_filterModel) self.attribute_treeView.setSortingEnabled(True) self.attribute_treeView.sortByColumn(0, QtCore.Qt.AscendingOrder) self.attribute_treeView.setSelectionMode( QtWidgets.QAbstractItemView.MultiSelection) self.attribute_selModel = self.attribute_treeView.selectionModel() self.attribute_selModel.selectionChanged.connect( self.attrNodeSelectionChanged) self.attributeToggleAnimated_toolButton.clicked.connect( self.attributeToggleAnimatedClicked, ) self.attributeToggleStatic_toolButton.clicked.connect( self.attributeToggleStaticClicked, ) self.attributeToggleLocked_toolButton.clicked.connect( self.attributeToggleLockedClicked, ) # Attr Add and Remove buttons self.attributeAdd_toolButton.clicked.connect(self.attrAddClicked) self.attributeRemove_toolButton.clicked.connect(self.attrRemoveClicked) # Solver Nodes self.solver_model = solver_nodes.SolverModel(font=self.font) self.solver_filterModel = QtCore.QSortFilterProxyModel() self.solver_filterModel.setSourceModel(self.solver_model) self.solver_filterModel.setDynamicSortFilter(False) self.solver_tableView.setModel(self.solver_filterModel) self.solver_tableView.setSortingEnabled(False) self.solver_selModel = self.solver_tableView.selectionModel() # Set up custom widgets for viewing and editing the columns. self.solver_attrFilterDelegate = solver_nodes.AttributeComboBoxDelegate( ) attr_idx = self.solver_model.getColumnIndexFromColumnName( const.SOLVER_COLUMN_NAME_ATTRIBUTES) strategy_idx = self.solver_model.getColumnIndexFromColumnName( const.SOLVER_COLUMN_NAME_STRATEGY) self.solver_tableView.setItemDelegateForColumn( attr_idx, self.solver_attrFilterDelegate, ) self.solver_strategyDelegate = solver_nodes.StrategyComboBoxDelegate() self.solver_tableView.setItemDelegateForColumn( strategy_idx, self.solver_strategyDelegate, ) # Solver Add and Remove buttons self.solverAdd_toolButton.clicked.connect(self.solverAddClicked) self.solverRemove_toolButton.clicked.connect(self.solverRemoveClicked) # TODO: Write functions for Move Up/Down buttons. Until then, # we'll hide the buttons to avoid confusion. self.solverMoveUp_toolButton.setVisible(False) self.solverMoveDown_toolButton.setVisible(False) # self.solverMoveUp_toolButton.clicked.connect( # self.solverMoveUpClicked # ) # self.solverMoveDown_toolButton.clicked.connect( # self.solverMoveDownClicked # ) # Override Current Frame self.overrideCurrentFrame_checkBox.stateChanged.connect( self.overrideCurrentFrameChanged) # Populate the UI with data. self.updateDynamicWindowTitle() self.updateCollectionModel() self.updateObjectToggleButtons() self.updateObjectColumnVisibility() self.updateObjectModel() self.updateAttributeToggleButtons() self.updateAttributeColumnVisibility() self.updateAttributeModel() self.updateSolverModel() self.updateSolveValidState() return