class VariablesTool(DockTool):
"""docstring for Variables tool."""
def __init__(self):
super(VariablesTool, self).__init__()
self.setMinimumSize(QtCore.QSize(200, 50))
self.content = QWidget()
self.content.setObjectName("VariablesToolContent")
self.verticalLayout = QVBoxLayout(self.content)
self.verticalLayout.setSpacing(0)
self.verticalLayout.setContentsMargins(0, 0, 0, 0)
self.verticalLayout.setObjectName("verticalLayout")
self.setWidget(self.content)
@staticmethod
def getIcon():
return QtGui.QIcon(":variable.png")
@staticmethod
def isSingleton():
return True
def onShow(self):
super(VariablesTool, self).onShow()
varsWidget = VariablesWidget(self.pyFlowInstance)
self.pyFlowInstance.fileBeenLoaded.connect(varsWidget.actualize)
self.verticalLayout.addWidget(varsWidget)
@staticmethod
def toolTip():
return "Variables editing/creation"
@staticmethod
def name():
return str("Variables")
def __init__(self, data, parent=None):
QWidget.__init__(self, parent)
self.data = data
vbox = QVBoxLayout()
self.setLayout(vbox)
title = QLabel('Indexed Strings')
vbox.addWidget(title)
self.frame = QFrame()
vbox.addWidget(self.frame)
self.vbox = QVBoxLayout()
self.frame.setLayout(self.vbox)
self.frame.setFrameShape(QFrame.Panel)
self.frame.setFrameShadow(QFrame.Sunken)
self.table = QTableWidget()
self.table.horizontalHeader().hide()
self.vbox.addWidget(self.table)
self.table.hide()
self.noStringsLabel = QLabel('<i>No indexed strings</i>')
self.vbox.addWidget(self.noStringsLabel)
self.widgets = []
self.populate()
self.data.attributeAdded.connect(self.attributeAddedSlot)
self.data.dataReset.connect(self.dataResetSlot)
self.data.dirtied.connect(self.dataDirtiedSlot)
def __init__(self, parent=None):
super().__init__(parent)
self.m_icon = QLabel()
self.m_title = QLabel()
self.m_message = QLabel()
self.notification = None
self.setWindowFlags(Qt.ToolTip)
rootLayout = QHBoxLayout(self)
rootLayout.addWidget(self.m_icon)
bodyLayout = QVBoxLayout()
rootLayout.addLayout(bodyLayout)
titleLayout = QHBoxLayout()
bodyLayout.addLayout(titleLayout)
titleLayout.addWidget(self.m_title)
titleLayout.addItem(QSpacerItem(0, 0, QSizePolicy.Expanding))
close = QPushButton(self.tr("Close"))
titleLayout.addWidget(close)
close.clicked.connect(self.onClosed)
bodyLayout.addWidget(self.m_message)
self.adjustSize()
def main():
import sys
app = QApplication(sys.argv)
layout = QVBoxLayout()
infos = QSerialPortInfo.availablePorts()
for info in infos:
s = (
f"Port: {info.portName()}",
f"Location: {info.systemLocation()}",
f"Description: {info.description()}",
f"Manufacturer: {info.manufacturer()}",
f"Serial number: {info.serialNumber()}",
"Vendor Identifier: " + f"{info.vendorIdentifier():x}"
if info.hasVendorIdentifier()
else "",
"Product Identifier: " + f"{info.productIdentifier():x}"
if info.hasProductIdentifier()
else "",
)
label = QLabel("\n".join(s))
layout.addWidget(label)
workPage = QWidget()
workPage.setLayout(layout)
area = QScrollArea()
area.setWindowTitle("Info about all available serial ports.")
area.setWidget(workPage)
area.show()
sys.exit(app.exec_())
class HistoryTool(DockTool):
"""docstring for History tool."""
def __init__(self):
super(HistoryTool, self).__init__()
self.setMinimumSize(QtCore.QSize(200, 50))
self.content = QWidget()
self.content.setObjectName("historyToolContent")
self.verticalLayout = QVBoxLayout(self.content)
self.verticalLayout.setSpacing(0)
self.verticalLayout.setContentsMargins(0, 0, 0, 0)
self.verticalLayout.setObjectName("verticalLayout")
self.undoStackView = QUndoView(self)
self.undoStackView.setObjectName("undoStackView")
self.verticalLayout.addWidget(self.undoStackView)
self.setWidget(self.content)
@staticmethod
def getIcon():
return QtGui.QIcon(RESOURCES_DIR + "/history.png")
def onShow(self):
super(HistoryTool, self).onShow()
self.undoStackView.setStack(self.canvas.undoStack)
@staticmethod
def defaultDockArea():
return QtCore.Qt.LeftDockWidgetArea
@staticmethod
def toolTip():
return "Undo stack"
@staticmethod
def name():
return str("History")
def __init__(self, page, parent=None):
super(HelpForm, self).__init__(parent)
self.setAttribute(Qt.WA_DeleteOnClose)
self.setAttribute(Qt.WA_GroupLeader)
backAction = QAction(QIcon(":/back.png"), "&Back", self)
backAction.setShortcut(QKeySequence.Back)
homeAction = QAction(QIcon(":/home.png"), "&Home", self)
homeAction.setShortcut("Home")
self.pageLabel = QLabel()
toolBar = QToolBar()
toolBar.addAction(backAction)
toolBar.addAction(homeAction)
toolBar.addWidget(self.pageLabel)
self.textBrowser = QTextBrowser()
layout = QVBoxLayout()
layout.addWidget(toolBar)
layout.addWidget(self.textBrowser, 1)
self.setLayout(layout)
backAction.triggered.connect(self.tbackward)
homeAction.triggered.connect(self.thome)
self.textBrowser.sourceChanged.connect(self.updatePageTitle)
self.textBrowser.setSearchPaths([":/help"])
self.textBrowser.setSource(QUrl(page))
self.resize(400, 600)
self.setWindowTitle("{0} Help".format(QApplication.applicationName()))
class VariablesTool(DockTool):
"""docstring for Variables tool."""
def __init__(self):
super(VariablesTool, self).__init__()
self.setMinimumSize(QtCore.QSize(200, 50))
self.content = QWidget()
self.content.setObjectName("VariablesToolContent")
self.verticalLayout = QVBoxLayout(self.content)
self.verticalLayout.setSpacing(0)
self.verticalLayout.setContentsMargins(0, 0, 0, 0)
self.verticalLayout.setObjectName("verticalLayout")
self.setWidget(self.content)
@staticmethod
def getIcon():
return QtGui.QIcon(RESOURCES_DIR + "/variable.png")
@staticmethod
def isSingleton():
return True
def onShow(self):
super(VariablesTool, self).onShow()
self.verticalLayout.addWidget(VariablesWidget(self.canvas))
@staticmethod
def toolTip():
return "Variables editing/creation"
@staticmethod
def name():
return str("Variables")
def __init__(self, data, parent=None):
QWidget.__init__(self, parent)
self.data = data
vbox = QVBoxLayout()
self.setLayout(vbox)
title = QLabel('Indexed Strings')
vbox.addWidget(title)
self.frame = QFrame()
vbox.addWidget(self.frame)
self.vbox = QVBoxLayout()
self.frame.setLayout(self.vbox)
self.frame.setFrameShape(QFrame.Panel)
self.frame.setFrameShadow(QFrame.Sunken)
self.table = QTableWidget()
self.table.horizontalHeader().hide()
self.vbox.addWidget(self.table)
self.table.hide()
self.noStringsLabel = QLabel('<i>No indexed strings</i>')
self.vbox.addWidget(self.noStringsLabel)
self.widgets = []
self.populate()
self.data.attributeAdded.connect(self.attributeAddedSlot)
self.data.dataReset.connect(self.dataResetSlot)
self.data.dirtied.connect(self.dataDirtiedSlot)
class NodesBox(QWidget):
"""doc string for NodesBox"""
def __init__(self, parent, canvas=None):
super(NodesBox, self).__init__(parent)
self.canvasRef = weakref.ref(canvas)
self.verticalLayout = QVBoxLayout(self)
self.verticalLayout.setObjectName("verticalLayout")
self.verticalLayout.setContentsMargins(4, 4, 4, 4)
self.lineEdit = NodeBoxLineEdit(self)
self.lineEdit.setObjectName("lineEdit")
self.verticalLayout.addWidget(self.lineEdit)
self.treeWidget = NodeBoxTreeWidget(self)
self.treeWidget.setObjectName("treeWidget")
self.treeWidget.headerItem().setText(0, "1")
self.verticalLayout.addWidget(self.treeWidget)
self.lineEdit.textChanged.connect(self.leTextChanged)
self.treeWidget.refresh()
def sizeHint(self):
return QtCore.QSize(400, 250)
def expandCategory(self):
for i in self.treeWidget.categoryPaths:
self.treeWidget.setItemExpanded(self.treeWidget.categoryPaths[i],
True)
def leTextChanged(self):
if self.lineEdit.text() == '':
self.lineEdit.setPlaceholderText("enter node name..")
self.treeWidget.refresh()
return
self.treeWidget.refresh(None, self.lineEdit.text())
self.expandCategory()
def __init__(self):
super().__init__()
if separate_colorbars:
if rescale_colorbars:
self.vmins = tuple(np.min(u[0]) for u in U)
self.vmaxs = tuple(np.max(u[0]) for u in U)
else:
self.vmins = tuple(np.min(u) for u in U)
self.vmaxs = tuple(np.max(u) for u in U)
else:
if rescale_colorbars:
self.vmins = (min(np.min(u[0]) for u in U),) * len(U)
self.vmaxs = (max(np.max(u[0]) for u in U),) * len(U)
else:
self.vmins = (min(np.min(u) for u in U),) * len(U)
self.vmaxs = (max(np.max(u) for u in U),) * len(U)
layout = QHBoxLayout()
plot_layout = QGridLayout()
self.colorbarwidgets = [cbar_widget(self, vmin=vmin, vmax=vmax) if cbar_widget else None
for vmin, vmax in zip(self.vmins, self.vmaxs)]
plots = [widget(self, grid, vmin=vmin, vmax=vmax, bounding_box=bounding_box, codim=codim)
for vmin, vmax in zip(self.vmins, self.vmaxs)]
if legend:
for i, plot, colorbar, l in zip(range(len(plots)), plots, self.colorbarwidgets, legend):
subplot_layout = QVBoxLayout()
caption = QLabel(l)
caption.setAlignment(Qt.AlignHCenter)
subplot_layout.addWidget(caption)
if not separate_colorbars or backend == 'matplotlib':
subplot_layout.addWidget(plot)
else:
hlayout = QHBoxLayout()
hlayout.addWidget(plot)
if colorbar:
hlayout.addWidget(colorbar)
subplot_layout.addLayout(hlayout)
plot_layout.addLayout(subplot_layout, int(i/columns), (i % columns), 1, 1)
else:
for i, plot, colorbar in zip(range(len(plots)), plots, self.colorbarwidgets):
if not separate_colorbars or backend == 'matplotlib':
plot_layout.addWidget(plot, int(i/columns), (i % columns), 1, 1)
else:
hlayout = QHBoxLayout()
hlayout.addWidget(plot)
if colorbar:
hlayout.addWidget(colorbar)
plot_layout.addLayout(hlayout, int(i/columns), (i % columns), 1, 1)
layout.addLayout(plot_layout)
if not separate_colorbars:
layout.addWidget(self.colorbarwidgets[0])
for w in self.colorbarwidgets[1:]:
w.setVisible(False)
self.setLayout(layout)
self.plots = plots
def __init__(self):
super().__init__()
if separate_colorbars:
if rescale_colorbars:
self.vmins = tuple(np.min(u[0]) for u in U)
self.vmaxs = tuple(np.max(u[0]) for u in U)
else:
self.vmins = tuple(np.min(u) for u in U)
self.vmaxs = tuple(np.max(u) for u in U)
else:
if rescale_colorbars:
self.vmins = (min(np.min(u[0]) for u in U),) * len(U)
self.vmaxs = (max(np.max(u[0]) for u in U),) * len(U)
else:
self.vmins = (min(np.min(u) for u in U),) * len(U)
self.vmaxs = (max(np.max(u) for u in U),) * len(U)
layout = QHBoxLayout()
plot_layout = QGridLayout()
self.colorbarwidgets = [cbar_widget(self, vmin=vmin, vmax=vmax) if cbar_widget else None
for vmin, vmax in zip(self.vmins, self.vmaxs)]
plots = [widget(self, grid, vmin=vmin, vmax=vmax, bounding_box=bounding_box, codim=codim)
for vmin, vmax in zip(self.vmins, self.vmaxs)]
if legend:
for i, plot, colorbar, l in zip(range(len(plots)), plots, self.colorbarwidgets, legend):
subplot_layout = QVBoxLayout()
caption = QLabel(l)
caption.setAlignment(Qt.AlignHCenter)
subplot_layout.addWidget(caption)
if not separate_colorbars or backend == 'matplotlib':
subplot_layout.addWidget(plot)
else:
hlayout = QHBoxLayout()
hlayout.addWidget(plot)
if colorbar:
hlayout.addWidget(colorbar)
subplot_layout.addLayout(hlayout)
plot_layout.addLayout(subplot_layout, int(i/columns), (i % columns), 1, 1)
else:
for i, plot, colorbar in zip(range(len(plots)), plots, self.colorbarwidgets):
if not separate_colorbars or backend == 'matplotlib':
plot_layout.addWidget(plot, int(i/columns), (i % columns), 1, 1)
else:
hlayout = QHBoxLayout()
hlayout.addWidget(plot)
if colorbar:
hlayout.addWidget(colorbar)
plot_layout.addLayout(hlayout, int(i/columns), (i % columns), 1, 1)
layout.addLayout(plot_layout)
if not separate_colorbars:
layout.addWidget(self.colorbarwidgets[0])
for w in self.colorbarwidgets[1:]:
w.setVisible(False)
self.setLayout(layout)
self.plots = plots
def showSystemSettings(system):
from brigks.gui.systemSettingsWidget import SystemSettingsWidget
widget = SystemSettingsWidget(system)
dialog = QDialog()
layout = QVBoxLayout()
layout.addWidget(widget)
dialog.setLayout(layout)
if not dialog.exec_():
return
def DDTree():
tree = TreeWidget()
dialog = QDialog()
layout = QVBoxLayout()
layout.addWidget(tree)
dialog.setLayout(layout)
if not dialog.exec_():
return
def __init__(self, parent: QWidget = None):
super().__init__(parent)
self.setWindowFlags(self.windowFlags()
& ~Qt.WindowContextHelpButtonHint)
self.server = QLocalServer(self)
if not self.server.listen("fortune"):
QMessageBox.critical(
self,
self.tr("Local Fortune Server"),
self.tr("Unable to start the server: %s." %
(self.server.errorString())),
)
QTimer.singleShot(0, self.close)
return
statusLabel = QLabel()
statusLabel.setWordWrap(True)
statusLabel.setText(
self.
tr("The server is running.\nRun the Local Fortune Client example now."
))
self.fortunes = (
self.tr(
"You've been leading a dog's life. Stay off the furniture."),
self.tr("You've got to think about tomorrow."),
self.tr("You will be surprised by a loud noise."),
self.tr("You will feel hungry again in another hour."),
self.tr("You might have mail."),
self.tr("You cannot kill time without injuring eternity."),
self.tr(
"Computers are not intelligent. They only think they are."),
)
quitButton = QPushButton(self.tr("Quit"))
quitButton.setAutoDefault(False)
quitButton.clicked.connect(self.close)
self.server.newConnection.connect(self.sendFortune)
buttonLayout = QHBoxLayout()
buttonLayout.addStretch(1)
buttonLayout.addWidget(quitButton)
buttonLayout.addStretch(1)
mainLayout = QVBoxLayout(self)
mainLayout.addWidget(statusLabel)
mainLayout.addLayout(buttonLayout)
self.setWindowTitle(QGuiApplication.applicationDisplayName())
def get_horizontal_separator():
v_div_w = QWidget()
v_div_l = QVBoxLayout()
v_div_l.setAlignment(Qt.AlignLeft)
v_div_l.setContentsMargins(0, 0, 0, 0)
v_div_l.setSpacing(0)
v_div_w.setLayout(v_div_l)
v_div = QFrame()
v_div.setMinimumHeight(30)
v_div.setFrameShape(QFrame.VLine)
v_div.setFrameShadow(QFrame.Sunken)
v_div_l.addWidget(v_div)
return v_div_w
def _build_ui(self):
self._tree.setRootIsDecorated(False)
self._tree.setSelectionMode(QTreeWidget.ExtendedSelection)
lyt_main = QVBoxLayout()
if self._allow_create:
lyt_top = QHBoxLayout()
lyt_top.setContentsMargins(0, 0, 0, 0)
lyt_top.setSpacing(0)
lyt_top.addWidget(self._btn_add)
lyt_top.addStretch()
lyt_main.addLayout(lyt_top)
lyt_main.addWidget(self._tree)
self.setLayout(lyt_main)
class TextInput(QWidget):
# used when input text
inputChanged = Signal()
okPressed = Signal()
cancelPressed = Signal()
def __init__(self, parent=None):
super(TextInput, self).__init__(parent)
self.setWindowFlags(Qt.Dialog | Qt.FramelessWindowHint)
self.mainLayout = QVBoxLayout()
self.textArea = QTextEdit(self)
self.buttonArea = QWidget(self)
self.buttonLayout = QHBoxLayout()
self.cancelButton = QPushButton('Cancel', self)
self.okButton = QPushButton('Ok', self)
self.buttonLayout.addWidget(self.cancelButton)
self.buttonLayout.addWidget(self.okButton)
self.buttonArea.setLayout(self.buttonLayout)
self.mainLayout.addWidget(self.textArea)
self.mainLayout.addWidget(self.buttonArea)
self.setLayout(self.mainLayout)
self.textArea.textChanged.connect(self.textChanged_)
self.okButton.clicked.connect(self.okButtonClicked)
self.cancelButton.clicked.connect(self.cancelPressed)
def getText(self):
return self.textArea.toPlainText()
def getFocus(self):
self.setFocus()
self.textArea.setFocus()
def clearText(self):
self.textArea.clear()
# slots
def textChanged_(self):
self.inputChanged.emit()
def cancelButtonClicked(self):
self.cancelPressed.emit()
def okButtonClicked(self):
self.okPressed.emit()
def get_column_layout(*widgets):
"""
Returns a QVBoxLayout with all given widgets added to it
:param widgets: list<QWidget>
:return: QVBoxLayout
"""
layout = QVBoxLayout()
for w in widgets:
if isinstance(w, QWidget):
layout.addWidget(w)
elif isinstance(w, QLayout):
layout.addLayout(w)
return layout
def _setup_login_dialog(dialog, account_input, password_input):
dialog.setWindowTitle('登录CGTeamWork')
account_input.setPlaceholderText('CGTeamwork账号名')
password_input.setPlaceholderText('密码')
password_input.setEchoMode(QLineEdit.Password)
ok_button = QPushButton('登录')
ok_button.setDefault(True)
ok_button.clicked.connect(dialog.accept)
layout = QVBoxLayout(dialog)
layout.addWidget(QLabel('帐号'))
layout.addWidget(account_input)
layout.addWidget(QLabel('密码'))
layout.addWidget(password_input)
layout.addWidget(ok_button)
dialog.setLayout(layout)
def _build_ui(self):
self._lyt_grid.setContentsMargins(0, 0, 0, 0)
self._lyt_grid.setSpacing(10)
self._lyt_grid.setAlignment(Qt.AlignTop | Qt.AlignLeft)
scroll_area = QScrollArea()
scroll_area.setFrameStyle(0)
scroll_area.setWidget(QWidget())
scroll_area.widget().setLayout(self._lyt_grid)
scroll_area.setWidgetResizable(True)
lyt_main = QVBoxLayout()
lyt_main.setContentsMargins(0, 0, 0, 0)
lyt_main.setSpacing(0)
lyt_main.addWidget(scroll_area)
lyt_main.addWidget(self._toolbar)
self.setLayout(lyt_main)
def __init__(self, id_, items, parent=None):
super().__init__(parent)
itemLabel = QLabel(self.tr("Item: "))
descriptionLabel = QLabel(self.tr("Description: "))
imageFileLabel = QLabel(self.tr("Image file: "))
self.createButtons()
self.itemText = QLabel()
self.descriptionEditor = QTextEdit()
self.imageFileEditor = QComboBox()
self.imageFileEditor.setModel(items.relationModel(1))
self.imageFileEditor.setModelColumn(
items.relationModel(1).fieldIndex("file"))
self.mapper = QDataWidgetMapper(self)
self.mapper.setModel(items)
self.mapper.setSubmitPolicy(QDataWidgetMapper.ManualSubmit)
self.mapper.setItemDelegate(QSqlRelationalDelegate(self.mapper))
self.mapper.addMapping(self.imageFileEditor, 1)
self.mapper.addMapping(self.itemText, 2, b"text")
self.mapper.addMapping(self.descriptionEditor, 3)
self.mapper.setCurrentIndex(id_)
self.descriptionEditor.textChanged.connect(self.enableButtons)
self.imageFileEditor.currentIndexChanged.connect(self.enableButtons)
formLayout = QFormLayout()
formLayout.addRow(itemLabel, self.itemText)
formLayout.addRow(imageFileLabel, self.imageFileEditor)
formLayout.addRow(descriptionLabel, self.descriptionEditor)
layout = QVBoxLayout()
layout.addLayout(formLayout)
layout.addWidget(self.buttonBox)
self.setLayout(layout)
self.itemId = id_
self.displayedImage = self.imageFileEditor.currentText()
self.setWindowFlags(Qt.Window)
self.enableButtons(False)
self.setWindowTitle(self.itemText.text())
def _setup_login_dialog(dialog, account_input, password_input):
# type: (Any, Any, Any) -> None
dialog.setWindowTitle("登录CGTeamWork")
account_input.setPlaceholderText("CGTeamwork账号名")
password_input.setPlaceholderText("密码")
password_input.setEchoMode(QLineEdit.Password)
ok_button = QPushButton("登录")
ok_button.setDefault(True)
ok_button.clicked.connect(dialog.accept)
layout = QVBoxLayout(dialog)
layout.addWidget(QLabel("帐号"))
layout.addWidget(account_input)
layout.addWidget(QLabel("密码"))
layout.addWidget(password_input)
layout.addWidget(ok_button)
dialog.setLayout(layout)
def __init__(self, name='modelPanelWidget', **kwargs):
super(ModelPanelWidget, self).__init__(**kwargs)
unique_name = name + str(id(self))
self.setObjectName(unique_name + 'Widget')
main_layout = QVBoxLayout(self)
main_layout.setContentsMargins(0, 0, 0, 0)
main_layout.setObjectName(unique_name + 'Layout')
self.setLayout(main_layout)
maya.cmds.setParent(main_layout.objectName())
pane_layout_name = maya.cmds.paneLayout()
self._model_panel = maya.cmds.modelPanel(unique_name, label="ModelPanel", menuBarVisible=False)
pane_layout_widget = gui.to_qt_object(pane_layout_name)
main_layout.addWidget(pane_layout_widget)
self.set_model_panel_options()
self.hide_bar_layout()
self.hide_menu_bar()
class MatplotlibWidget:
def __init__(self, parent):
self.figure = Figure()
self.canvas = FigureCanvas(self.figure)
self.canvas.setParent(parent)
self.mpl_toolbar = NavigationToolbar(self.canvas, parent)
self.axes = self.figure.add_subplot(111)
self.grid_layout = QVBoxLayout()
self.grid_layout.addWidget(self.canvas)
self.grid_layout.addWidget(self.mpl_toolbar)
parent.setLayout(self.grid_layout)
self.clear()
def clear(self, *, keep_limits=False):
"""
Make sure the figure is in a nice state
"""
if keep_limits:
# slightly hacky way to clear axes, but prevents axis limits being reset when
# we redraw
for artist in self.axes.lines + self.axes.collections:
artist.remove()
self.axes.set_prop_cycle(None)
return
self.axes.clear()
self.figure.clear()
self.axes = self.figure.add_subplot(111)
self.axes.grid(True)
# Reset to some hardcoded default values
self.figure.subplots_adjust(left=0.125,
right=0.9,
top=0.9,
bottom=0.1,
wspace=0.2,
hspace=0.2)
self.axes.set_xlabel("R", fontsize=16)
self.axes.set_ylabel("Z", rotation="horizontal", fontsize=16)
self.canvas.draw()
class TextEditDialog(QDialog):
def __init__(self, font, textColor, parent=None):
super(TextEditDialog, self).__init__(parent)
self.setWindowFlags(QtCore.Qt.Window | QtCore.Qt.FramelessWindowHint)
self.resize(QtCore.QSize(400, 300))
self.layout = QVBoxLayout(self)
self.layout.setContentsMargins(2, 2, 2, 2)
self.te = TextEditingField()
self.te.accepted.connect(self.onAccept)
self._font = QtGui.QFont(font)
self.te.setTextColor(textColor)
self.layout.addWidget(self.te)
self.buttons = QDialogButtonBox(
QDialogButtonBox.Ok | QDialogButtonBox.Cancel, self)
self.buttons.accepted.connect(self.onAccept)
self.buttons.rejected.connect(self.onReject)
self.layout.addWidget(self.buttons)
self._result = None
def zoomIn(self, factor):
self.te.zoomIn(factor)
def setHtml(self, html):
self.te.setHtml(html)
self.te.selectAll()
self.te.setFontPointSize(20)
cursor = self.te.textCursor()
cursor.clearSelection()
cursor.movePosition(QtGui.QTextCursor.End)
self.te.setTextCursor(cursor)
def onReject(self):
self._result = "", False
self.reject()
def getResult(self):
return self._result
def onAccept(self):
self.te.selectAll()
self.te.setFontPointSize(self._font.pointSize())
self._result = self.te.toHtml(), True
self.accept()
def __init__(self,
device_info: QAudioDeviceInfo,
parent: QWidget = None) -> None:
super().__init__(parent)
self.m_chart = QChart()
self.m_series = QLineSeries()
chart_view = QChartView(self.m_chart)
chart_view.setMinimumSize(800, 600)
self.m_chart.addSeries(self.m_series)
axisX = QValueAxis()
axisX.setRange(0, XYSeriesIODevice.sampleCount)
axisX.setLabelFormat("%g")
axisX.setTitleText("Samples")
axisY = QValueAxis()
axisY.setRange(-1, 1)
axisY.setTitleText("Audio level")
self.m_chart.addAxis(axisX, Qt.AlignBottom)
self.m_series.attachAxis(axisX)
self.m_chart.addAxis(axisY, Qt.AlignLeft)
self.m_series.attachAxis(axisY)
self.m_chart.legend().hide()
self.m_chart.setTitle(
f"Data from the microphone ({device_info.deviceName()})")
mainLayout = QVBoxLayout(self)
mainLayout.addWidget(chart_view)
formatAudio = QAudioFormat()
formatAudio.setSampleRate(8000)
formatAudio.setChannelCount(1)
formatAudio.setSampleSize(8)
formatAudio.setCodec("audio/pcm")
formatAudio.setByteOrder(QAudioFormat.LittleEndian)
formatAudio.setSampleType(QAudioFormat.UnSignedInt)
self.m_audioInput = QAudioInput(device_info, formatAudio, self)
self.m_device = XYSeriesIODevice(self.m_series, self)
self.m_device.open(QIODevice.WriteOnly)
self.m_audioInput.start(self.m_device)
class TextEditDialog(QDialog):
def __init__(self, font, textColor, parent=None):
super(TextEditDialog, self).__init__(parent)
self.setWindowFlags(QtCore.Qt.Window | QtCore.Qt.FramelessWindowHint)
self.resize(QtCore.QSize(400, 300))
self.layout = QVBoxLayout(self)
self.layout.setContentsMargins(2, 2, 2, 2)
self.te = QTextEdit()
self._font = font
self._font = QtGui.QFont(font)
self.te.setFont(self._font)
self.te.setTextColor(textColor)
self.layout.addWidget(self.te)
self.buttons = QDialogButtonBox(
QDialogButtonBox.Ok | QDialogButtonBox.Cancel, self)
self.buttons.accepted.connect(self.onAccept)
self.buttons.rejected.connect(self.onReject)
self.layout.addWidget(self.buttons)
self._result = None
self.te.textChanged.connect(
lambda color=textColor: self.te.setTextColor(color))
def zoomIn(self, factor):
self.te.zoomIn(factor)
def setHtml(self, html):
self.te.setHtml(html)
def onReject(self):
self._result = "", False
self.reject()
def getResult(self):
return self._result
def onAccept(self):
self.te.selectAll()
# self._font.setPointSize(self.initialPointSize)
self.te.setFont(self._font)
self._result = self.te.toHtml(), True
self.accept()
class Counters(QScrollArea):
button_clicked = Signal(object, object)
SPACING = 2
def __init__(self, parent=None):
QScrollArea.__init__(self, parent)
self.setWidgetResizable(True)
self.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
self._layout = QVBoxLayout()
self._layout.setContentsMargins(0, 0, 0, 0)
self._layout.setSpacing(Counters.SPACING)
widget = QWidget()
widget.setLayout(self._layout)
self.setWidget(widget)
self.setFixedWidth(200)
self.checked_buttons = dict()
def set(self, counters):
previous_buttons = self.checked_buttons.copy()
self.clear()
for column, counter in counters.items():
for entry, value in counter.items():
checked = previous_buttons.get(entry, False)
self.checked_buttons[entry] = checked
button = QPushButton('{} {}'.format(entry, value))
button.setProperty(PROPERTY_SIDE_STROKED, checked)
button.clicked.connect(self._clicked)
self._layout.addWidget(button)
self._layout.addWidget(QLabel())
self._layout.addWidget(QWidget())
self._layout.setStretch(self._layout.count() - 1, 100)
def clear(self):
clear_layout(self._layout)
self.checked_buttons = dict()
def _clicked(self):
button = self.sender()
checked = not button.property(PROPERTY_SIDE_STROKED)
entry = button.text().split()[0]
self.checked_buttons[entry] = checked
button.setProperty(PROPERTY_SIDE_STROKED, checked)
self.button_clicked.emit(entry, self.checked_buttons)
def __init__(self, parent: QWidget = None) -> None:
super().__init__(parent)
self.tcpServer = QTcpServer()
self.tcpClient = QTcpSocket()
self.tcpServerConnection: QTcpSocket = None
self.bytesToWrite = 0
self.bytesWritten = 0
self.bytesReceived = 0
self.clientProgressBar = QProgressBar()
self.clientStatusLabel = QLabel(self.tr("Client ready"))
self.serverProgressBar = QProgressBar()
self.serverStatusLabel = QLabel(self.tr("Server ready"))
self.startButton = QPushButton(self.tr("&Start"))
self.quitButton = QPushButton(self.tr("&Quit"))
self.buttonBox = QDialogButtonBox()
self.buttonBox.addButton(self.startButton, QDialogButtonBox.ActionRole)
self.buttonBox.addButton(self.quitButton, QDialogButtonBox.RejectRole)
self.startButton.clicked.connect(self.start)
self.quitButton.clicked.connect(self.close)
self.tcpServer.newConnection.connect(self.acceptConnection)
self.tcpClient.connected.connect(self.startTransfer)
self.tcpClient.bytesWritten.connect(self.updateClientProgress)
self.tcpClient.error.connect(self.displayError)
mainLayout = QVBoxLayout(self)
mainLayout.addWidget(self.clientProgressBar)
mainLayout.addWidget(self.clientStatusLabel)
mainLayout.addWidget(self.serverProgressBar)
mainLayout.addWidget(self.serverStatusLabel)
mainLayout.addStretch(1)
mainLayout.addSpacing(10)
mainLayout.addWidget(self.buttonBox)
def __init__(self, U, plot, length=1, title=None):
super().__init__()
layout = QVBoxLayout()
if title:
title = QLabel('<b>' + title + '</b>')
title.setAlignment(Qt.AlignHCenter)
layout.addWidget(title)
layout.addWidget(plot)
plot.set(U, 0)
if length > 1:
hlayout = QHBoxLayout()
self.slider = QSlider(Qt.Horizontal)
self.slider.setMinimum(0)
self.slider.setMaximum(length - 1)
self.slider.setTickPosition(QSlider.TicksBelow)
hlayout.addWidget(self.slider)
lcd = QLCDNumber(m.ceil(m.log10(length)))
lcd.setDecMode()
lcd.setSegmentStyle(QLCDNumber.Flat)
hlayout.addWidget(lcd)
layout.addLayout(hlayout)
hlayout = QHBoxLayout()
toolbar = QToolBar()
self.a_play = QAction(self.style().standardIcon(QStyle.SP_MediaPlay), 'Play', self)
self.a_play.setCheckable(True)
self.a_rewind = QAction(self.style().standardIcon(QStyle.SP_MediaSeekBackward), 'Rewind', self)
self.a_toend = QAction(self.style().standardIcon(QStyle.SP_MediaSeekForward), 'End', self)
self.a_step_backward = QAction(self.style().standardIcon(QStyle.SP_MediaSkipBackward),
'Step Back', self)
self.a_step_forward = QAction(self.style().standardIcon(QStyle.SP_MediaSkipForward), 'Step', self)
self.a_loop = QAction(self.style().standardIcon(QStyle.SP_BrowserReload), 'Loop', self)
self.a_loop.setCheckable(True)
toolbar.addAction(self.a_play)
toolbar.addAction(self.a_rewind)
toolbar.addAction(self.a_toend)
toolbar.addAction(self.a_step_backward)
toolbar.addAction(self.a_step_forward)
toolbar.addAction(self.a_loop)
if hasattr(self, 'save'):
self.a_save = QAction(self.style().standardIcon(QStyle.SP_DialogSaveButton), 'Save', self)
toolbar.addAction(self.a_save)
self.a_save.triggered.connect(self.save)
hlayout.addWidget(toolbar)
self.speed = QSlider(Qt.Horizontal)
self.speed.setMinimum(0)
self.speed.setMaximum(100)
hlayout.addWidget(QLabel('Speed:'))
hlayout.addWidget(self.speed)
layout.addLayout(hlayout)
self.timer = QTimer()
self.timer.timeout.connect(self.update_solution)
self.slider.valueChanged.connect(self.slider_changed)
self.slider.valueChanged.connect(lcd.display)
self.speed.valueChanged.connect(self.speed_changed)
self.a_play.toggled.connect(self.toggle_play)
self.a_rewind.triggered.connect(self.rewind)
self.a_toend.triggered.connect(self.to_end)
self.a_step_forward.triggered.connect(self.step_forward)
self.a_step_backward.triggered.connect(self.step_backward)
self.speed.setValue(50)
elif hasattr(self, 'save'):
hlayout = QHBoxLayout()
toolbar = QToolBar()
self.a_save = QAction(self.style().standardIcon(QStyle.SP_DialogSaveButton), 'Save', self)
toolbar.addAction(self.a_save)
hlayout.addWidget(toolbar)
layout.addLayout(hlayout)
self.a_save.triggered.connect(self.save)
self.setLayout(layout)
self.plot = plot
self.U = U
self.length = length
class IndexedStringsWidget(QWidget):
""" Holds the list of indexed string attributes """
def __init__(self, data, parent=None):
QWidget.__init__(self, parent)
self.data = data
vbox = QVBoxLayout()
self.setLayout(vbox)
title = QLabel('Indexed Strings')
vbox.addWidget(title)
self.frame = QFrame()
vbox.addWidget(self.frame)
self.vbox = QVBoxLayout()
self.frame.setLayout(self.vbox)
self.frame.setFrameShape(QFrame.Panel)
self.frame.setFrameShadow(QFrame.Sunken)
self.table = QTableWidget()
self.table.horizontalHeader().hide()
self.vbox.addWidget(self.table)
self.table.hide()
self.noStringsLabel = QLabel('<i>No indexed strings</i>')
self.vbox.addWidget(self.noStringsLabel)
self.widgets = []
self.populate()
self.data.attributeAdded.connect(self.attributeAddedSlot)
self.data.dataReset.connect(self.dataResetSlot)
self.data.dirtied.connect(self.dataDirtiedSlot)
def dataDirtiedSlot(self, dirty):
""" SLOT when the particle data is dirtied or cleaned."""
if not dirty:
for widget in self.widgets:
widget.drawBorder(False)
def dataResetSlot(self):
""" SLOT when particle data is reconstructed """
self.populate()
def attributeAddedSlot(self, name): #pylint:disable=W0613
""" SLOT when an attribute is added to the particle set """
attr = self.data.attributeInfo(name)
if attr.type == partio.INDEXEDSTR:
self.populate()
def populate(self):
""" Populates the table of indexed strings """
self.widgets = []
# If no widgets, just drop that in
attrs = []
for anum in range(self.data.numAttributes()):
attr = self.data.attributeInfo(anum)
if attr.type == partio.INDEXEDSTR:
attrs.append(attr)
if not attrs:
self.table.hide()
self.noStringsLabel.show()
return
self.table.show()
self.noStringsLabel.hide()
self.table.setColumnCount(1)
self.table.setRowCount(len(attrs))
for row, attr in enumerate(attrs):
item = QTableWidgetItem(attr.name)
self.table.setVerticalHeaderItem(row, item)
strings = self.data.indexedStrs(attr)
table = QTableWidget()
table.setColumnCount(1)
table.setRowCount(len(strings))
table.horizontalHeader().hide()
table.setVerticalHeaderLabels([str(i) for i in range(len(strings))])
for i, string in enumerate(strings):
widget = QLabel(string)
table.setCellWidget(i, 0, widget)
self.widgets.append(widget)
self.table.setCellWidget(row, 0, table)
self.table.horizontalHeader().setStretchLastSection(False)
self.table.setTabKeyNavigation(True)
self.table.horizontalHeader().setSectionsMovable(False)
self.table.horizontalHeader().resizeSections(QHeaderView.ResizeToContents)
self.table.verticalHeader().resizeSections(QHeaderView.ResizeToContents)
def __init__(self, parent=None):
QMainWindow.__init__(self, parent)
self.data = ParticleData()
toolbar = self.addToolBar("Test")
openButton = QPushButton("")
openButton.setFlat(True)
openButton.setIconSize( QSize(32, 32) )
openButton.setIcon(QIcon("/jobs2/soft/icons/dlight/open.png"))
openButton.setToolTip( "Open File" )
toolbar.addWidget(openButton)
openButton.clicked.connect(self.openSlot)
QShortcut( QKeySequence(Qt.CTRL + Qt.Key_O), self, self.openSlot )
saveButton = QPushButton("")
saveButton.setFlat(True)
saveButton.setIconSize( QSize(32, 32) )
saveButton.setIcon(QIcon("/jobs2/soft/icons/dlight/file_save.png"))
saveButton.setToolTip( "Save File" )
toolbar.addWidget(saveButton)
saveButton.clicked.connect(self.saveSlot)
QShortcut( QKeySequence(Qt.CTRL + Qt.Key_S), self, self.saveSlot )
saveDeltaButton = QPushButton("")
saveDeltaButton.setFlat(True)
saveDeltaButton.setIconSize( QSize(32, 32) )
saveDeltaButton.setIcon(QIcon("/jobs2/soft/icons/dlight/file_save_as.png"))
saveDeltaButton.setToolTip( "Save File As Delta" )
toolbar.addWidget(saveDeltaButton)
saveDeltaButton.clicked.connect(self.saveDeltaSlot)
QShortcut( QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_S), self, self.saveDeltaSlot )
addParticleButton = QPushButton("Particle")
addParticleButton.setFlat(True)
addParticleButton.setIconSize( QSize(32, 32) )
addParticleButton.setIcon(QIcon("/jobs2/soft/icons/shared/plus.png"))
addParticleButton.setToolTip( "Add Particle" )
toolbar.addWidget(addParticleButton)
addParticleButton.clicked.connect(self.addParticleSlot)
addAttributeButton = QPushButton("Attribute")
addAttributeButton.setFlat(True)
addAttributeButton.setIconSize( QSize(32, 32) )
addAttributeButton.setIcon(QIcon("/jobs2/soft/icons/shared/plus.png"))
addAttributeButton.setToolTip( "Add Attribute" )
toolbar.addWidget(addAttributeButton)
addAttributeButton.clicked.connect(self.addAttributeSlot)
splitter = QSplitter(self)
self.setCentralWidget(splitter)
particleTable = ParticleTableWidget(self.data, self)
splitter.addWidget(particleTable)
right = QWidget(self)
splitter.addWidget(right)
vbox = QVBoxLayout(right)
right.setLayout(vbox)
fixedAttrWidget = FixedAttributesWidget(self.data, self)
vbox.addWidget(fixedAttrWidget)
indexedStrings = IndexedStringsWidget(self.data, self)
vbox.addWidget(indexedStrings)
vbox.addStretch()
# TODD: SCROLLABLE AREAS FOR EVERYTHING
self.data.dirtied.connect(self.dataDirtiedSlot)
# Configure ctrl-w to close the window
QShortcut( QKeySequence(Qt.CTRL + Qt.Key_W), self, self.close )
def __init__(self, parent=None):
QMainWindow.__init__(self, parent)
self.data = ParticleData()
toolbar = self.addToolBar("Test")
openButton = QPushButton("")
openButton.setFlat(True)
openButton.setIconSize(QSize(32, 32))
openButton.setIcon(QIcon("/jobs2/soft/icons/dlight/open.png"))
openButton.setToolTip("Open File")
toolbar.addWidget(openButton)
openButton.clicked.connect(self.openSlot)
QShortcut(QKeySequence(Qt.CTRL + Qt.Key_O), self, self.openSlot)
saveButton = QPushButton("")
saveButton.setFlat(True)
saveButton.setIconSize(QSize(32, 32))
saveButton.setIcon(QIcon("/jobs2/soft/icons/dlight/file_save.png"))
saveButton.setToolTip("Save File")
toolbar.addWidget(saveButton)
saveButton.clicked.connect(self.saveSlot)
QShortcut(QKeySequence(Qt.CTRL + Qt.Key_S), self, self.saveSlot)
saveDeltaButton = QPushButton("")
saveDeltaButton.setFlat(True)
saveDeltaButton.setIconSize(QSize(32, 32))
saveDeltaButton.setIcon(
QIcon("/jobs2/soft/icons/dlight/file_save_as.png"))
saveDeltaButton.setToolTip("Save File As Delta")
toolbar.addWidget(saveDeltaButton)
saveDeltaButton.clicked.connect(self.saveDeltaSlot)
QShortcut(QKeySequence(Qt.CTRL + Qt.SHIFT + Qt.Key_S), self,
self.saveDeltaSlot)
addParticleButton = QPushButton("Particle")
addParticleButton.setFlat(True)
addParticleButton.setIconSize(QSize(32, 32))
addParticleButton.setIcon(QIcon("/jobs2/soft/icons/shared/plus.png"))
addParticleButton.setToolTip("Add Particle")
toolbar.addWidget(addParticleButton)
addParticleButton.clicked.connect(self.addParticleSlot)
addAttributeButton = QPushButton("Attribute")
addAttributeButton.setFlat(True)
addAttributeButton.setIconSize(QSize(32, 32))
addAttributeButton.setIcon(QIcon("/jobs2/soft/icons/shared/plus.png"))
addAttributeButton.setToolTip("Add Attribute")
toolbar.addWidget(addAttributeButton)
addAttributeButton.clicked.connect(self.addAttributeSlot)
splitter = QSplitter(self)
self.setCentralWidget(splitter)
particleTable = ParticleTableWidget(self.data, self)
splitter.addWidget(particleTable)
right = QWidget(self)
splitter.addWidget(right)
vbox = QVBoxLayout(right)
right.setLayout(vbox)
fixedAttrWidget = FixedAttributesWidget(self.data, self)
vbox.addWidget(fixedAttrWidget)
indexedStrings = IndexedStringsWidget(self.data, self)
vbox.addWidget(indexedStrings)
vbox.addStretch()
# TODD: SCROLLABLE AREAS FOR EVERYTHING
self.data.dirtied.connect(self.dataDirtiedSlot)
# Configure ctrl-w to close the window
QShortcut(QKeySequence(Qt.CTRL + Qt.Key_W), self, self.close)
class FixedAttributesWidget(QWidget):
""" A widget for viewing/editing fixed attributes (non-varying) """
def __init__(self, data, parent=None):
QWidget.__init__(self, parent)
self.data = data
vbox = QVBoxLayout()
self.setLayout(vbox)
title = QLabel('Fixed Attributes')
vbox.addWidget(title)
self.frame = QFrame()
vbox.addWidget(self.frame)
self.vbox = QVBoxLayout()
self.frame.setLayout(self.vbox)
self.frame.setFrameShape(QFrame.Panel)
self.frame.setFrameShadow(QFrame.Sunken)
self.table = QTableWidget()
self.table.horizontalHeader().hide()
self.vbox.addWidget(self.table)
self.table.hide()
self.noAttrLabel = QLabel('<i>No fixed attributes</i>')
self.vbox.addWidget(self.noAttrLabel)
self.widgets = []
self.populate()
self.data.fixedAttributeAdded.connect(self.fixedAttributeAddedSlot)
self.data.dataReset.connect(self.dataResetSlot)
self.data.dirtied.connect(self.dataDirtiedSlot)
def dataDirtiedSlot(self, dirty):
""" SLOT when the particle data is dirtied or cleaned."""
if not dirty:
for widget in self.widgets:
widget.drawBorder(False)
def dataResetSlot(self):
""" SLOT when particle data is reconstructed """
self.populate()
def fixedAttributeAddedSlot(self, name): #pylint:disable=W0613
""" SLOT when a fixed attribute is added to the particle set """
self.populate()
def populate(self):
""" Populates the table of fixed attributes """
self.widgets = []
# If no widgets, just drop that in
numAttrs = self.data.numFixedAttributes()
if not numAttrs:
self.table.hide()
self.noAttrLabel.show()
return
self.table.show()
self.noAttrLabel.hide()
self.table.setColumnCount(1)
self.table.setRowCount(numAttrs)
self.attrs = getAttrs(self.data.numFixedAttributes, self.data.fixedAttributeInfo, True)
for row, (_, attr) in enumerate(self.attrs):
item = QTableWidgetItem(attr.name)
tooltip = '<p><tt> Name: {}<br> Type: {}<br>Count: {}</tt></p>'.\
format(attr.name, partio.TypeName(attr.type), attr.count)
item.setToolTip(tooltip)
self.table.setVerticalHeaderItem(row, item)
value = self.data.getFixed(attr)
widget = getWidget(value, self.data, attr)
self.table.setCellWidget(row, 0, widget)
self.widgets.append(widget)
self.table.horizontalHeader().setStretchLastSection(False)
self.table.setTabKeyNavigation(True)
self.table.horizontalHeader().setSectionsMovable(False)
self.table.horizontalHeader().resizeSections(QHeaderView.ResizeToContents)
self.table.verticalHeader().resizeSections(QHeaderView.ResizeToContents)
class IndexedStringsWidget(QWidget):
""" Holds the list of indexed string attributes """
def __init__(self, data, parent=None):
QWidget.__init__(self, parent)
self.data = data
vbox = QVBoxLayout()
self.setLayout(vbox)
title = QLabel('Indexed Strings')
vbox.addWidget(title)
self.frame = QFrame()
vbox.addWidget(self.frame)
self.vbox = QVBoxLayout()
self.frame.setLayout(self.vbox)
self.frame.setFrameShape(QFrame.Panel)
self.frame.setFrameShadow(QFrame.Sunken)
self.table = QTableWidget()
self.table.horizontalHeader().hide()
self.vbox.addWidget(self.table)
self.table.hide()
self.noStringsLabel = QLabel('<i>No indexed strings</i>')
self.vbox.addWidget(self.noStringsLabel)
self.widgets = []
self.populate()
self.data.attributeAdded.connect(self.attributeAddedSlot)
self.data.dataReset.connect(self.dataResetSlot)
self.data.dirtied.connect(self.dataDirtiedSlot)
def dataDirtiedSlot(self, dirty):
""" SLOT when the particle data is dirtied or cleaned."""
if not dirty:
for widget in self.widgets:
widget.drawBorder(False)
def dataResetSlot(self):
""" SLOT when particle data is reconstructed """
self.populate()
def attributeAddedSlot(self, name): #pylint:disable=W0613
""" SLOT when an attribute is added to the particle set """
attr = self.data.attributeInfo(name)
if attr.type == partio.INDEXEDSTR:
self.populate()
def populate(self):
""" Populates the table of indexed strings """
self.widgets = []
# If no widgets, just drop that in
attrs = []
for anum in range(self.data.numAttributes()):
attr = self.data.attributeInfo(anum)
if attr.type == partio.INDEXEDSTR:
attrs.append(attr)
if not attrs:
self.table.hide()
self.noStringsLabel.show()
return
self.table.show()
self.noStringsLabel.hide()
self.table.setColumnCount(1)
self.table.setRowCount(len(attrs))
for row, attr in enumerate(attrs):
item = QTableWidgetItem(attr.name)
self.table.setVerticalHeaderItem(row, item)
strings = self.data.indexedStrs(attr)
table = QTableWidget()
table.setColumnCount(1)
table.setRowCount(len(strings))
table.horizontalHeader().hide()
table.setVerticalHeaderLabels(
[str(i) for i in range(len(strings))])
for i, string in enumerate(strings):
widget = QLabel(string)
table.setCellWidget(i, 0, widget)
self.widgets.append(widget)
self.table.setCellWidget(row, 0, table)
self.table.horizontalHeader().setStretchLastSection(False)
self.table.setTabKeyNavigation(True)
self.table.horizontalHeader().setSectionsMovable(False)
self.table.horizontalHeader().resizeSections(
QHeaderView.ResizeToContents)
self.table.verticalHeader().resizeSections(
QHeaderView.ResizeToContents)
class FixedAttributesWidget(QWidget):
""" A widget for viewing/editing fixed attributes (non-varying) """
def __init__(self, data, parent=None):
QWidget.__init__(self, parent)
self.data = data
vbox = QVBoxLayout()
self.setLayout(vbox)
title = QLabel('Fixed Attributes')
vbox.addWidget(title)
self.frame = QFrame()
vbox.addWidget(self.frame)
self.vbox = QVBoxLayout()
self.frame.setLayout(self.vbox)
self.frame.setFrameShape(QFrame.Panel)
self.frame.setFrameShadow(QFrame.Sunken)
self.table = QTableWidget()
self.table.horizontalHeader().hide()
self.vbox.addWidget(self.table)
self.table.hide()
self.noAttrLabel = QLabel('<i>No fixed attributes</i>')
self.vbox.addWidget(self.noAttrLabel)
self.widgets = []
self.populate()
self.data.fixedAttributeAdded.connect(self.fixedAttributeAddedSlot)
self.data.dataReset.connect(self.dataResetSlot)
self.data.dirtied.connect(self.dataDirtiedSlot)
def dataDirtiedSlot(self, dirty):
""" SLOT when the particle data is dirtied or cleaned."""
if not dirty:
for widget in self.widgets:
widget.drawBorder(False)
def dataResetSlot(self):
""" SLOT when particle data is reconstructed """
self.populate()
def fixedAttributeAddedSlot(self, name): #pylint:disable=W0613
""" SLOT when a fixed attribute is added to the particle set """
self.populate()
def populate(self):
""" Populates the table of fixed attributes """
self.widgets = []
# If no widgets, just drop that in
numAttrs = self.data.numFixedAttributes()
if not numAttrs:
self.table.hide()
self.noAttrLabel.show()
return
self.table.show()
self.noAttrLabel.hide()
self.table.setColumnCount(1)
self.table.setRowCount(numAttrs)
self.attrs = getAttrs(self.data.numFixedAttributes,
self.data.fixedAttributeInfo, True)
for row, (_, attr) in enumerate(self.attrs):
item = QTableWidgetItem(attr.name)
tooltip = '<p><tt> Name: {}<br> Type: {}<br>Count: {}</tt></p>'.\
format(attr.name, partio.TypeName(attr.type), attr.count)
item.setToolTip(tooltip)
self.table.setVerticalHeaderItem(row, item)
value = self.data.getFixed(attr)
widget = getWidget(value, self.data, attr)
self.table.setCellWidget(row, 0, widget)
self.widgets.append(widget)
self.table.horizontalHeader().setStretchLastSection(False)
self.table.setTabKeyNavigation(True)
self.table.horizontalHeader().setSectionsMovable(False)
self.table.horizontalHeader().resizeSections(
QHeaderView.ResizeToContents)
self.table.verticalHeader().resizeSections(
QHeaderView.ResizeToContents)
