def setTable(self, data, sortData=(True, True), indexColCheckable=None): name_row, name_column = [], [] self.blockSignals(True) self.setColumnCount(max(map(len, data.values()))) self.setRowCount(len(data.keys())) # - Populate for row, value in enumerate(sorted(data.keys()) if sortData[0] else data.keys()): name_row.append(value) for col, key in enumerate(sorted(data[value].keys()) if sortData[1] else data[value].keys()): name_column.append(key) rowData = data[value][key] if isinstance(rowData, basestring): newitem = QtGui.QTableWidgetItem(str(rowData)) else: newitem = QtGui.QTableWidgetItem() newitem.setData(QtCore.Qt.EditRole, rowData) # - Make the columnt checkable if indexColCheckable is not None and col in indexColCheckable: newitem.setFlags(QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsEnabled) newitem.setCheckState(QtCore.Qt.Unchecked) self.setItem(row, col, newitem) self.setHorizontalHeaderLabels(name_column) self.setVerticalHeaderLabels(name_row) self.blockSignals(False)
def setTable(self, data, data_check=[], reset=False):
name_row, name_column = [], []
self.blockSignals(True)
self.setColumnCount(max(map(len, data.values())))
self.setRowCount(len(data.keys()))
# - Populate
for n, layer in enumerate(sorted(data.keys())):
name_row.append(layer)
for m, key in enumerate(data[layer].keys()):
# -- Build name column
name_column.append(key)
# -- Add first data column
newitem = QtGui.QTableWidgetItem(str(data[layer][key])) if m == 0 else QtGui.QTableWidgetItem()
# -- Selectively colorize missing data
if m == 0 and len(data_check) and data[layer][key] not in data_check: newitem.setBackground(QtGui.QColor('red'))
# -- Build Checkbox columns
if m > 0: newitem.setFlags(QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsEnabled)
if m > 0: newitem.setCheckState(QtCore.Qt.Unchecked if not data[layer][key] else QtCore.Qt.Checked)
self.setItem(n, m, newitem)
self.setHorizontalHeaderLabels(name_column)
self.setVerticalHeaderLabels(name_row)
self.blockSignals(False)
def fillTableFromData(self):
self.fillingTable = True
self.tableWidget.clear()
self.tableWidget.setColumnCount(self.materialData.nComponents + 1)
labels = [self.inputVariableComboBox.currentText]
if self.materialData.nComponents == 1:
labels.append(self.materialData.name)
else:
labels.extend(self.materialData.componentNames)
self.tableWidget.setHorizontalHeaderLabels(labels)
tableData = self.materialData.tableData.data
if tableData is not None:
self.tableWidget.setRowCount(tableData.shape[1])
it = numpy.nditer(tableData, flags=['multi_index'])
while not it.finished:
col = it.multi_index[0]
row = 0 if len(it.multi_index) == 1 else it.multi_index[1]
self.tableWidget.setItem(row, col, QtGui.QTableWidgetItem(str(it[0])))
it.iternext()
self.fillingTable = False
def setTableHeaders(self, table, columnNames):
columnNum = len(columnNames)
table.setColumnCount(columnNum)
for i in xrange(columnNum):
twi = QtGui.QTableWidgetItem(columnNames[i])
table.setHorizontalHeaderItem(i, twi)
return
def _makeTableWidgetItem(lbl, checkable, editable):
item = QtGui.QTableWidgetItem(lbl)
f = Qt.Qt.ItemIsEnabled | Qt.Qt.ItemIsSelectable
if editable:
f = f | Qt.Qt.ItemIsEditable
if checkable:
f = f | Qt.Qt.ItemIsUserCheckable
item.setFlags(f)
return item
def addTableRow(self, table, columnTexts, isSlack):
rowIndex = 0
table.insertRow(rowIndex)
for i in xrange(len(columnTexts)):
twi = QtGui.QTableWidgetItem(columnTexts[i])
twi.setFlags(twi.flags() & (~Qt.Qt.ItemIsEditable))
self.curFont.setItalic(isSlack)
twi.setFont(self.curFont)
table.setItem(rowIndex, i, twi)
self.resizeTable(table)
return
def addTableRow(self, table, columnTexts, atEnd=False):
rowIndex = 0
if atEnd:
rowIndex = table.rowCount
table.insertRow(rowIndex)
for i in xrange(len(columnTexts)):
twi = QtGui.QTableWidgetItem(columnTexts[i])
twi.setFlags(twi.flags() & (~Qt.Qt.ItemIsEditable))
self.twiHash.setdefault(table, []).append(twi)
table.setItem(rowIndex, i, twi)
self.resizeTable(table)
return
def refresh(self):
defaultParams = self.plugin.client.problem.getAvailable(
'defaultparameter')
params = self.plugin.client.problem.getAvailable('parameter')
self.tableWidget.setRowCount(len(defaultParams))
for i, p in enumerate(defaultParams):
pdoc = self.plugin.client.problem.getParameterDoc(p)
pval = self.plugin.client.problem.getParameter(p)
if self.tableWidget.item(i, _NAME) is None:
# initialize the row
self.tableWidget.setItem(i, _NAME, QtGui.QTableWidgetItem(p))
#self.tableWidget.setItem(i, _DOCUMENTATION, QtGui.QTableWidgetItem (pdoc))
self.tableWidget.setItem(
i, _VALUE, QtGui.QTableWidgetItem(str(pval.value())))
self.tableWidget.setItem(
i, _TYPE, QtGui.QTableWidgetItem(str(pval.typecode())))
else:
self.tableWidget.item(i, _NAME).setText(p)
#self.tableWidget.item(i, _DOCUMENTATION).text = pdoc
self.tableWidget.item(i, _VALUE).setText(str(pval.value()))
self.tableWidget.item(i, _TYPE).setText(str(pval.typecode()))
self.tableWidget.item(i, _NAME).setToolTip(pdoc)
def setTable(self, data, reset=False):
name_row, name_column = [], []
self.blockSignals(True)
self.setColumnCount(max(map(len, data.values())))
self.setRowCount(len(data.keys()))
# - Populate
for n, layer in enumerate(sorted(data.keys())):
name_row.append(layer)
for m, key in enumerate(data[layer].keys()):
name_column.append(key)
newitem = QtGui.QTableWidgetItem(str(data[layer][key]))
self.setItem(n, m, newitem)
self.setHorizontalHeaderLabels(name_column)
self.setVerticalHeaderLabels(name_row)
self.blockSignals(False)
def setupUi(self):
#set Main Window Title.
self.setWindowTitle("xg Texture Exporter")
self.setObjectName("xgTextureExportGUI")
self.setEnabled(True)
self.resize(545, 702)
##===================================== CentralLayout ============================
self.centralwidget = QtGui.QWidget()
self.centralwidget.setObjectName("centralwidget")
##===================================== MasterLayout ============================
self.master_GridLayout = QtGui.QGridLayout(self.centralwidget)
self.master_GridLayout.setObjectName("master_GridLayout")
##===================================== TopLayout ============================
self.top_GridLayout = QtGui.QGridLayout()
self.top_GridLayout.setObjectName("top_GridLayout")
##=======exportPathLine======##
self.exportPathLineEdit = QtGui.QLineEdit(self.centralwidget)
self.exportPathLineEdit.setMinimumSize(QtCore.QSize(0, 30))
self.exportPathLineEdit.setObjectName("exportPathLineEdit")
self.top_GridLayout.addWidget(self.exportPathLineEdit, 2, 0, 1, 1)
##=======FolderLable=======##
self.outputFolderLabel = QtGui.QLabel("Output Folder",
self.centralwidget)
setBold(self.outputFolderLabel)
self.outputFolderLabel.setObjectName("outputFolderLabel")
self.top_GridLayout.addWidget(self.outputFolderLabel, 0, 0, 1, 1)
##=======BrowseButton=======##
self.browseButton = xgPushButton("browseButton", "Browse", 0,
"Choose texture output directory.")
self.browseButton.setMinimumSize(QtCore.QSize(0, 30))
self.top_GridLayout.addWidget(self.browseButton, 2, 1, 1, 1)
self.master_GridLayout.addLayout(self.top_GridLayout, 0, 0, 1, 1)
##===================================== MidLayout ============================
self.mid_HBoxLayout = QtGui.QHBoxLayout()
self.mid_HBoxLayout.setObjectName("mid_HBoxLayout")
self.midLeft_GridLayout = QtGui.QGridLayout()
self.midLeft_GridLayout.setObjectName("midLeft_GridLayout")
##=======channelsLable=======##
self.channels_Label = QtGui.QLabel("Channels", self.centralwidget)
setBold(self.channels_Label)
self.channels_Label.setObjectName("channels_Label")
self.midLeft_GridLayout.addWidget(self.channels_Label, 0, 0, 1, 1)
##=======ChannelButtons=======##
self.removeChannel_Button = xgPushButton(
"removeChannel_Button", "-", 0,
"remove selected channels from export list.")
self.removeChannel_Button.setMinimumSize(QtCore.QSize(0, 45))
self.addChannel_Button = xgPushButton(
"addChannel_Button", "+", 0,
"add selected channels to export list.")
self.addChannel_Button.setMinimumSize(QtCore.QSize(0, 45))
self.midLeft_GridLayout.addWidget(self.addChannel_Button, 2, 0, 1, 1)
self.midLeft_GridLayout.addWidget(self.removeChannel_Button, 2, 1, 1,
1)
##=======ChannelList=======##
self.channelsList_ListWidget = ChannelsToExportList()
self.channelsList_ListWidget.isSortingEnabled()
self.channelsList_ListWidget.setSortingEnabled(False)
self.channelsList_ListWidget.setObjectName("channelsList_ListWidget")
QtGui.QListWidgetItem(self.channelsList_ListWidget)
self.midLeft_GridLayout.addWidget(self.channelsList_ListWidget, 1, 0,
1, 2)
self.mid_HBoxLayout.addLayout(self.midLeft_GridLayout)
self.options_GroupBox = QtGui.QGroupBox("Options", self.centralwidget)
self.options_GroupBox.setObjectName("options_GroupBox")
self.outputFormat_Label = QtGui.QLabel("Output Format :",
self.options_GroupBox)
self.outputFormat_Label.setGeometry(QtCore.QRect(20, 40, 121, 21))
self.outputFormat_Label.setObjectName("outputFormat_Label")
self.resolution_Label = QtGui.QLabel("Resolution:",
self.options_GroupBox)
self.resolution_Label.setGeometry(QtCore.QRect(20, 70, 121, 21))
self.resolution_Label.setObjectName("resolution_Label")
self.processTextures_Label = QtGui.QLabel("process textures:",
self.options_GroupBox)
self.processTextures_Label.setGeometry(QtCore.QRect(20, 130, 121, 21))
self.processTextures_Label.setObjectName("processTextures_Label")
##=======Options=======##
self.outFormat_ComboBox = QtGui.QComboBox(self.options_GroupBox)
self.outFormat_ComboBox.setToolTip("define output texture format.")
self.outFormat_ComboBox.setGeometry(QtCore.QRect(130, 40, 81, 25))
self.outFormat_ComboBox.setEditable(False)
self.outFormat_ComboBox.setObjectName("outFormat_ComboBox")
self.outFormat_ComboBox.addItem("exr")
self.outFormat_ComboBox.addItem("tif")
self.outFormat_ComboBox.setCurrentIndex(1)
self.resolution_ComboBox = QtGui.QComboBox(self.options_GroupBox)
self.resolution_ComboBox.setToolTip(
"define output texture resolution.")
self.resolution_ComboBox.setGeometry(QtCore.QRect(100, 70, 111, 25))
self.resolution_ComboBox.setObjectName("resolution_ComboBox")
self.resolution_ComboBox.addItem("channel res")
self.resolution_ComboBox.addItem("full (8K)")
self.resolution_ComboBox.addItem("heigh (4K)")
self.resolution_ComboBox.addItem("mid (2K)")
self.resolution_ComboBox.addItem("low (1K)")
self.clearExpChan_CheckBox = QtGui.QCheckBox("Clear export channels",
self.options_GroupBox)
self.clearExpChan_CheckBox.setGeometry(QtCore.QRect(20, 100, 181, 23))
self.clearExpChan_CheckBox.setChecked(True)
self.clearExpChan_CheckBox.setObjectName("clearExpChan_CheckBox")
self.clearExpChan_CheckBox.setToolTip(
"delete the flattened channels after export.")
self.processTextures_ComboBox = QtGui.QComboBox(self.options_GroupBox)
self.processTextures_ComboBox.setToolTip(
"define textures processing method.")
self.processTextures_ComboBox.setGeometry(
QtCore.QRect(135, 130, 105, 25))
self.processTextures_ComboBox.setObjectName("processTextures_ComboBox")
self.processTextures_ComboBox.addItem("None")
self.processTextures_ComboBox.addItem("Local process")
#self.processTextures_ComboBox.addItem("Farm process")
#self.texturePublish_CheckBox = QtGui.QCheckBox("Publish farm Textures", self.options_GroupBox)
#self.texturePublish_CheckBox.setToolTip("process textures on the farm via texturePublish. \n (convert only will not publish.)")
#self.texturePublish_CheckBox.setGeometry(QtCore.QRect(20, 160, 181, 23))
#self.texturePublish_CheckBox.setCheckable(False)
#self.texturePublish_CheckBox.setObjectName("texturePublish_CheckBox")
'''
self.linear_CheckBox = QtGui.QCheckBox("Local process textures.", self.options_GroupBox)
self.linear_CheckBox.setToolTip("convert textures to Mipmap exr localy.")
self.linear_CheckBox.setGeometry(QtCore.QRect(20, 130, 181, 23))
#self.linear_CheckBox.setChecked(True)
self.linear_CheckBox.setObjectName("linear_CheckBox")
self.publish_CheckBox = QtGui.QCheckBox("Publish After Export", self.options_GroupBox)
self.publish_CheckBox.setGeometry(QtCore.QRect(20, 190, 181, 23))
self.publish_CheckBox.setCheckable(False)
self.publish_CheckBox.setObjectName("publish_CheckBox")
'''
self.mid_HBoxLayout.addWidget(self.options_GroupBox)
self.master_GridLayout.addLayout(self.mid_HBoxLayout, 1, 0, 1, 1)
self.bottom_VBoxLayout = QtGui.QVBoxLayout()
self.bottom_VBoxLayout.setObjectName("bottom_VBoxLayout")
self.exportChannels_Label = QtGui.QLabel("Channels For Export",
self.centralwidget)
self.exportChannels_Label.setObjectName("exportChannels_Label")
setBold(self.exportChannels_Label)
self.bottom_VBoxLayout.addWidget(self.exportChannels_Label)
##======table=======##
self.exportChannelsList_tableWidget = QtGui.QTableWidget(
self.centralwidget)
self.exportChannelsList_tableWidget.setWordWrap(True)
self.exportChannelsList_tableWidget.setCornerButtonEnabled(True)
#self.exportChannelsList_tableWidget.setRowCount(2)
self.exportChannelsList_tableWidget.setObjectName(
"exportChannelsList_tableWidget")
self.exportChannelsList_tableWidget.setColumnCount(5)
self.exportChannelsList_tableWidget.setRowCount(0)
self.exportChannelsList_tableWidget.setSelectionBehavior(
QtGui.QAbstractItemView.SelectRows)
self.exportChannelsList_tableWidget.setSelectionMode(
QtGui.QAbstractItemView.SingleSelection)
self.exportChannelsList_tableWidget.horizontalHeader().setVisible(True)
self.exportChannelsList_tableWidget.horizontalHeader(
).setCascadingSectionResizes(False)
self.exportChannelsList_tableWidget.horizontalHeader(
).setMinimumSectionSize(25)
self.exportChannelsList_tableWidget.horizontalHeader(
).setSortIndicatorShown(False)
#self.exportChannelsList_tableWidget.horizontalHeader().setStretchLastSection(True)
self.exportChannelsList_tableWidget.verticalHeader(
).setCascadingSectionResizes(False)
self.exportChannelsList_tableWidget.verticalHeader(
).setDefaultSectionSize(28)
self.exportChannelsList_tableWidget.verticalHeader(
).setMinimumSectionSize(10)
self.itemLine0 = QtGui.QLineEdit("channel")
self.item0 = QtGui.QTableWidgetItem(self.itemLine0.text)
self.exportChannelsList_tableWidget.setHorizontalHeaderItem(
0, self.item0)
self.itemLine1 = QtGui.QLineEdit("type")
self.item1 = QtGui.QTableWidgetItem(self.itemLine1.text)
self.exportChannelsList_tableWidget.setHorizontalHeaderItem(
1, self.item1)
self.itemLine2 = QtGui.QLineEdit("version")
self.item2 = QtGui.QTableWidgetItem(self.itemLine2.text)
self.exportChannelsList_tableWidget.setHorizontalHeaderItem(
2, self.item2)
self.itemLine3 = QtGui.QLineEdit("non_color")
self.item3 = QtGui.QTableWidgetItem(self.itemLine3.text)
self.exportChannelsList_tableWidget.setHorizontalHeaderItem(
3, self.item3)
self.itemLine4 = QtGui.QLineEdit("variation")
self.item4 = QtGui.QTableWidgetItem(self.itemLine4.text)
self.exportChannelsList_tableWidget.setHorizontalHeaderItem(
4, self.item4)
self.exportChannelsList_tableWidget.horizontalHeader(
).setCascadingSectionResizes(False)
self.bottom_VBoxLayout.addWidget(self.exportChannelsList_tableWidget)
self.exportButton_HBoxLayout = QtGui.QHBoxLayout()
self.exportButton_HBoxLayout.setObjectName("exportButton_HBoxLayout")
self.cancel_Button = xgPushButton("cancel_Button", "Cancel", 1)
self.cancel_Button.setMinimumSize(QtCore.QSize(0, 45))
self.exportButton_HBoxLayout.addWidget(self.cancel_Button)
self.exportPatch_Button = xgPushButton("exportPatch_Button",
"Export Selected Patches", 1)
self.exportPatch_Button.setMinimumSize(QtCore.QSize(200, 45))
self.exportButton_HBoxLayout.addWidget(self.exportPatch_Button)
self.export_Button = xgPushButton("export_Button", "Export", 0)
self.export_Button.setMinimumSize(QtCore.QSize(200, 45))
self.exportButton_HBoxLayout.addWidget(self.export_Button)
self.bottom_VBoxLayout.addLayout(self.exportButton_HBoxLayout)
self.master_GridLayout.addLayout(self.bottom_VBoxLayout, 2, 0, 1, 1)
self.setLayout(self.master_GridLayout)
def fillRowWithZeroes(self, row):
for col in xrange(self.tableWidget.columnCount):
self.tableWidget.setItem(row, col, QtGui.QTableWidgetItem('0'))
