def addChannel(self): '''Adds Channel Items''' parent = self.addObject() resolution = '%sk' % str(sceneData('res'))[0] bitDepth = '%sbit' % sceneData('depth') channel = sceneData('chan') chan_items = [] ## Check for existing try: for index in range(parent.childCount()): child = parent.child(index) if child.text(0) == channel: chanItem = child return chanItem except: pass ## Create and Build Channel items chanItem = QtGui.QTreeWidgetItem([channel, '', bitDepth, resolution]) chanItem.setTextAlignment(1, 0x0004) chanItem.setTextAlignment(2, 0x0004) chanItem.setIcon(0, QtGui.QIcon('%s/Channel.png' % icon_path)) chanItem.setData(0, 32, 'CHANNEL') chan_items.append(chanItem) parent.insertChildren(0, chan_items) return chanItem
def addObject(self): '''Adds Object Items''' geo = sceneData('geo') geo_items = [] ## Check for existing try: objectItem = self.exportList.findItems(geo, 0, 0)[0] return objectItem ## Create and Build Object items except: objectItem = QtGui.QTreeWidgetItem([geo, '', '', '']) objectItem.setIcon(0, QtGui.QIcon('%s/Objects.png' % icon_path)) objectItem.setData(0, 32, 'OBJECT') geo_items.append(objectItem) self.exportList.insertTopLevelItems(0, geo_items) objectItem.setExpanded(True) return objectItem
def addToObjectModel(self, obj, parentObj=None): assert obj._tree is None parentItem = self._getItemForObject(parentObj) objName = obj.getProperty('Name') item = QtGui.QTreeWidgetItem(parentItem, [objName]) item.setIcon(0, Icons.getIcon(obj.getProperty('Icon'))) obj._tree = self self._objects[item] = obj self.updateVisIcon(obj) if parentItem is None: tree = self.getTreeWidget() tree.addTopLevelItem(item) tree.expandItem(item)
def addToObjectModel(self, obj, parentObj=None): assert obj._tree is None parentItem = self._getItemForObject(parentObj) objName = obj.getProperty("Name") item = QtGui.QTreeWidgetItem(parentItem, [objName]) item.setIcon(0, Icons.getIcon(obj.getProperty("Icon"))) obj._tree = self self._objects[item] = obj self.updateVisIcon(obj) if parentItem is None: tree = self.getTreeWidget() tree.addTopLevelItem(item) tree.expandItem(item) self.callbacks.process(self.OBJECT_ADDED, self, obj)
def addToObjectModel(self, obj, parentObj=None): assert obj._tree is None parentItem = self._getItemForObject(parentObj) if parentObj else None objName = obj.getProperty('Name') item = QtGui.QTreeWidgetItem(parentItem, [objName]) item.setIcon(0, Icons.getIcon(obj.getProperty('Icon'))) obj._tree = self self._objectToItem[obj] = item self._itemToObject[item] = obj self._itemToName[item] = objName self._nameToItems[objName].add(item) self.updateVisIcon(obj) if parentItem is None: tree = self.getTreeWidget() tree.addTopLevelItem(item) tree.expandItem(item) self.callbacks.process(self.OBJECT_ADDED, self, obj)
def addUDIM(self): '''Adds UDIM to list. Note: chain trigger, this function builds everything, Nothing will be built if no UDIM selected. ''' selected_udim = sceneData('udim') ## Exit if no UDIM selected if not selected_udim: return parent = self.addChannel() exists_udim = [] add_udim = [] udim_items = [] ## Check for existing try: for index in range(parent.childCount()): child = parent.child(index) exists_udim.append(child.text(0)) for patch in selected_udim: if patch not in exists_udim: add_udim.append(patch) except: pass ## Create and Build UDIM items for patch in add_udim: udimItem = QtGui.QTreeWidgetItem([patch, '', '', '']) udimItem.setIcon(0, QtGui.QIcon('%s/Plus.png' % icon_path)) udimItem.setData(0, 32, 'UDIM') udim_items.append(udimItem) parent.insertChildren(0, udim_items) self.udimCount() ## Resize self.resize()