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()
