def load(self):
layout = self.layout()
# Clear
for btn in self._visButtons:
layout.removeWidget(btn)
btn.setVisible(False)
#TODO sip delete?
self._visButtons = []
layers = self._guide.getLayersDepths()
for i, (depth, layer) in enumerate(layers, start=1):
btn = QPushButton(" " * depth + layer.name())
btn.setStyleSheet("Text-align:left")
btn.setMaximumHeight(20)
layout.addWidget(btn, i, 0)
self._visButtons.append(btn)
for j, option in enumerate(OPTIONS, start=1):
btn = QPushButton()
btn.setFixedSize(30, 20)
kwargs = {x: option == x for x in OPTIONS}
visible = layer.isVisible(**kwargs)
self.setButtonColor(btn, visible)
btn.clicked.connect(
partial(self.toggleVisibility, btn, layer, option))
layout.addWidget(btn, i, j)
self._visButtons.append(btn)
def generateButtons(self, parent=None):
""" Generate buttons due to colorDic """
self.colorButtons = []
for color in self.colorList:
button = QPushButton(parent)
button.setObjectName(color[0])
button.setStyleSheet('QPushButton { background-color: %s; }' %
color[1])
button.setFixedSize(self.iconWidth / 2, self.iconHeight / 2)
button.setCheckable(True)
self.colorButtons.append(button)
class MyToolBar(QWidget):
""" ToolBar widget """
# signal
trigger = Signal(int)
def __init__(self, flags, parent=None):
super(MyToolBar, self).__init__(parent)
self.setWindowFlags(Qt.ToolTip)
self.paddingX = 5
self.paddingY = 2
self.iconWidth = self.iconHeight = 28
self.setFixedHeight(self.iconHeight + 2 * self.paddingY)
# self.setFixedWidth(300)
self.rectButton = None
self.ellipseButton = None
self.arrowButton = None
self.lineButton = None
self.freePenButton = None
self.textButton = None
self.undoButton = None
self.cancelButton = None
self.okButton = None
self.saveButton = None
self.button_list = []
self.initWindow(flags)
def initDrawButtons(self, flags):
self.drawButtonGroup = QButtonGroup(self)
# draw action buttons
if flags & constant.RECT:
self.rectButton = QPushButton(self)
self.rectButton.setIcon(QIcon(":/resource/icon/rect.png"))
self.rectButton.setFixedSize(self.iconWidth, self.iconHeight)
self.rectButton.setCheckable(True)
self.drawButtonGroup.addButton(self.rectButton)
self.hlayout.addWidget(self.rectButton)
self.button_list.append(self.rectButton)
if flags & constant.ELLIPSE:
self.ellipseButton = QPushButton(self)
self.ellipseButton.setIcon(QIcon(":/resource/icon/ellipse.png"))
self.ellipseButton.setFixedSize(self.iconWidth, self.iconHeight)
self.ellipseButton.setCheckable(True)
self.drawButtonGroup.addButton(self.ellipseButton)
self.hlayout.addWidget(self.ellipseButton)
self.button_list.append(self.ellipseButton)
if flags & constant.ARROW:
self.arrowButton = QPushButton(self)
self.arrowButton.setIcon(QIcon(":/resource/icon/arrow.png"))
self.arrowButton.setFixedSize(self.iconWidth, self.iconHeight)
self.arrowButton.setCheckable(True)
self.drawButtonGroup.addButton(self.arrowButton)
self.hlayout.addWidget(self.arrowButton)
self.button_list.append(self.arrowButton)
if flags & constant.LINE:
self.lineButton = QPushButton(self)
self.lineButton.setIcon(QIcon(":/resource/icon/line.png"))
self.lineButton.setFixedSize(self.iconWidth, self.iconHeight)
self.lineButton.setCheckable(True)
self.drawButtonGroup.addButton(self.lineButton)
self.hlayout.addWidget(self.lineButton)
self.button_list.append(self.lineButton)
if flags & constant.FREEPEN:
self.freePenButton = QPushButton(self)
self.freePenButton.setIcon(QIcon(":/resource/icon/pen.png"))
self.freePenButton.setFixedSize(self.iconWidth, self.iconHeight)
self.freePenButton.setCheckable(True)
self.drawButtonGroup.addButton(self.freePenButton)
self.hlayout.addWidget(self.freePenButton)
self.button_list.append(self.freePenButton)
if flags & constant.TEXT:
self.textButton = QPushButton(self)
self.textButton.setIcon(QIcon(":/resource/icon/text.png"))
self.textButton.setFixedSize(self.iconWidth, self.iconHeight)
self.textButton.setCheckable(True)
self.drawButtonGroup.addButton(self.textButton)
self.hlayout.addWidget(self.textButton)
self.button_list.append(self.textButton)
self.drawButtonGroup.buttonClicked.connect(self.buttonToggled)
def initOtherButtons(self, flags):
# other action buttons
if len(self.button_list) != 0:
self.separator1 = QFrame(self)
self.separator1.setFrameShape(QFrame.VLine)
self.separator1.setFrameShadow(QFrame.Sunken)
self.hlayout.addWidget(self.separator1)
self.undoButton = QPushButton(self)
self.undoButton.setIcon(QIcon(":/resource/icon/undo.png"))
self.undoButton.setFixedSize(self.iconWidth, self.iconWidth)
self.undoButton.clicked.connect(self.otherButtonsClicked)
self.hlayout.addWidget(self.undoButton)
if flags & constant.SAVE_TO_FILE:
self.saveButton = QPushButton(self)
self.saveButton.setIcon(QIcon(":/resource/icon/save.png"))
self.saveButton.setFixedSize(self.iconWidth, self.iconHeight)
self.saveButton.clicked.connect(self.otherButtonsClicked)
self.hlayout.addWidget(self.saveButton)
self.separator2 = QFrame(self)
self.separator2.setFrameShape(QFrame.VLine)
self.separator2.setFrameShadow(QFrame.Sunken)
self.hlayout.addWidget(self.separator2)
self.cancelButton = QPushButton(self)
self.cancelButton.setIcon(QIcon(":/resource/icon/close.png"))
self.cancelButton.setFixedSize(self.iconWidth, self.iconHeight)
self.cancelButton.clicked.connect(self.otherButtonsClicked)
if flags & constant.CLIPBOARD:
self.okButton = QPushButton(self)
self.okButton.setIcon(QIcon(":/resource/icon/check.png"))
self.okButton.setFixedSize(self.iconWidth, self.iconHeight)
self.okButton.clicked.connect(self.otherButtonsClicked)
self.hlayout.addWidget(self.okButton)
self.hlayout.addWidget(self.cancelButton)
def initWindow(self, flags):
self.hlayout = QHBoxLayout()
self.hlayout.setSpacing(2)
self.hlayout.setContentsMargins(10, 2, 10, 2)
self.setLayout(self.hlayout)
self.initDrawButtons(flags)
self.initOtherButtons(flags)
# slots
def buttonToggled(self, button):
"""
:type button: QPushButton
:param button:
:return:
"""
if button == self.rectButton:
self.trigger.emit(ACTION_RECT)
elif button == self.ellipseButton:
self.trigger.emit(ACTION_ELLIPSE)
elif button == self.arrowButton:
self.trigger.emit(ACTION_ARROW)
elif button == self.lineButton:
self.trigger.emit(ACTION_LINE)
elif button == self.freePenButton:
self.trigger.emit(ACTION_FREEPEN)
elif button == self.textButton:
self.trigger.emit(ACTION_TEXT)
else:
pass
def otherButtonsClicked(self):
if self.sender() == self.undoButton:
self.trigger.emit(ACTION_UNDO)
elif self.sender() == self.cancelButton:
self.trigger.emit(ACTION_CANCEL)
elif self.sender() == self.okButton:
self.trigger.emit(ACTION_SURE)
elif self.sender() == self.saveButton:
self.trigger.emit(ACTION_SAVE)
class RowTableWidget(QFrame):
double_clicked = Signal(object)
def __init__(self,
auto_resize=False,
single_row_select=True,
context_menu_callback=None,
last_column_stretch=True,
has_counters=False,
parent=None
):
QFrame.__init__(self, parent)
self._has_counters = has_counters
self.model = None
self.table_view = RowTableView(auto_resize, single_row_select, context_menu_callback, last_column_stretch)
self.table_view.doubleClicked.connect(self._double_clicked)
if has_counters:
self.counters = Counters()
self.counters.button_clicked.connect(self._counter_clicked)
self.search_bar = QLineEdit()
self.search_bar.setFixedHeight(SEARCHBAR_HEIGHT)
self.search_bar.textChanged.connect(self.set_search_text)
self.search_bar.setToolTip("Search bar")
self.auto_size_button = QPushButton('<>')
self.auto_size_button.setFixedSize(SEARCHBAR_HEIGHT, SEARCHBAR_HEIGHT)
self.auto_size_button.clicked.connect(self._auto_size_clicked)
self.auto_size_button.setToolTip("Auto size")
self.status_label = QLabel(STATUS_LABEL_MESSAGE.format(0, 0))
self.status_label.setFixedWidth(STATUS_LABEL_WIDTH)
self.progress_bar = QProgressBar()
self.progress_bar.setFormat('')
layout = QGridLayout()
layout.addWidget(self.search_bar, 0, 0, 1, 3)
layout.addWidget(self.auto_size_button, 0, 3)
if has_counters:
layout.addWidget(self.counters, 1, 0, 1, 2)
layout.addWidget(self.table_view, 1, 2, 1, 2)
else:
layout.addWidget(self.table_view, 1, 0, 1, 4)
layout.addWidget(self.status_label, 2, 0)
layout.addWidget(self.progress_bar, 2, 1, 1, 3)
layout.setColumnStretch(2, 100)
self.setLayout(layout)
def set_model(self, model):
self.model = model
self.table_view.setModel(model)
model.modelReset.connect(self._set_progress_maximum)
model.modelReset.connect(self._update_status)
if self._has_counters:
model.modelReset.connect(self._update_counters)
self._update_counters()
model.progress_updated.connect(self._update_progress)
self._set_progress_maximum()
self._update_status()
self.progress_bar.setVisible(model.has_background_callback)
@property
def selected_rows(self):
return [self.model.data(index, Qt.UserRole) for index
in self.table_view.selectionModel().selectedIndexes() if index.column() == 0]
@property
def search_text(self):
return self.search_bar.text()
def set_search_text(self, text):
self.search_bar.blockSignals(True)
self.search_bar.setText(text)
self.search_bar.blockSignals(False)
self.model.set_search_text(text)
def _set_progress_maximum(self):
self.progress_bar.setMaximum(self.model.progress_max) # do better ?
def _update_progress(self, value):
self.progress_bar.setValue(value)
def _update_status(self):
self.status_label.setText(STATUS_LABEL_MESSAGE.format(
self.model.rowCount(), self.model.total_row_count
))
def _counter_clicked(self, entry, checked_buttons):
entries = [entry for entry, active in checked_buttons.items() if active]
self.model.set_search_counters(entries)
def _update_counters(self):
self.counters.set(self.model.counters)
def _auto_size_clicked(self):
with Hourglass():
self.table_view.resizeColumnsToContents()
def _double_clicked(self, index):
row = self.model.data(index, Qt.UserRole)
self.double_clicked.emit(row)
def state(self):
header_sizes = list()
header = self.table_view.horizontalHeader()
for section_index in range(header.count()):
header_sizes.append(header.sectionSize(section_index))
return {
'header_sizes': header_sizes,
'search_text': self.search_bar.text()
}
def load_state(self, state):
header = self.table_view.horizontalHeader()
for section_index, size in enumerate(state['header_sizes']):
header.resizeSection(section_index, size)
self.search_bar.setText(state['search_text'])
class PenSetWidget(QWidget):
penSizeTrigger = Signal(int)
penColorTrigger = Signal(str)
fontChangeTrigger = Signal(QFont)
def __init__(self, parent=None):
super(PenSetWidget, self).__init__(parent)
self.paddingX = 5
self.paddingY = 2
self.iconWidth = self.iconHeight = 24
self.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Fixed)
self.setWindowFlags(Qt.ToolTip)
self.initWindows()
self.prevSizeButton = self.penSize1
self.penSize1.setChecked(True)
self.presentColor.setStyleSheet(
'QPushButton { background-color: %s; }' % PENCOLOR)
def generateButtons(self, parent=None):
""" Generate buttons due to colorDic """
self.colorButtons = []
for color in self.colorList:
button = QPushButton(parent)
button.setObjectName(color[0])
button.setStyleSheet('QPushButton { background-color: %s; }' %
color[1])
button.setFixedSize(self.iconWidth / 2, self.iconHeight / 2)
button.setCheckable(True)
self.colorButtons.append(button)
def initWindows(self):
self.mainLayout = QHBoxLayout()
self.setLayout(self.mainLayout)
self.mainLayout.setSpacing(0)
self.mainLayout.setContentsMargins(5, 2, 5, 2)
self.initPenSizeButtons()
self.initFontWidget()
self.initPenColorButtons()
self.separator = QFrame(self)
self.separator.setFrameShape(QFrame.VLine)
self.separator.setFrameShadow(QFrame.Sunken)
self.mainLayout.addWidget(self.penSize)
self.mainLayout.addWidget(self.changeFontButton)
self.mainLayout.addWidget(self.separator)
self.mainLayout.addWidget(self.colorSet)
def initPenSizeButtons(self):
self.penSize = QWidget(self)
self.penSizeLayout = QHBoxLayout()
self.penSize.setLayout(self.penSizeLayout)
# adjust pen size
self.penSize1 = QPushButton(self.penSize)
self.penSize1.setIcon(QIcon(":/resource/icon/pensize1.png"))
self.penSize1.setObjectName('1')
self.penSize1.setFixedSize(self.iconWidth, self.iconHeight)
self.penSize1.setCheckable(True)
self.penSize2 = QPushButton(self.penSize)
self.penSize2.setIcon(QIcon(":/resource/icon/pensize2.png"))
self.penSize2.setObjectName('2')
self.penSize2.setFixedSize(self.iconWidth, self.iconHeight)
self.penSize2.setCheckable(True)
self.penSize3 = QPushButton(self.penSize)
self.penSize3.setIcon(QIcon(":/resource/icon/pensize3.png"))
self.penSize3.setObjectName('3')
self.penSize3.setFixedSize(self.iconWidth, self.iconHeight)
self.penSize3.setCheckable(True)
self.sizeButtonGroup = QButtonGroup(self.penSize)
self.sizeButtonGroup.addButton(self.penSize1)
self.sizeButtonGroup.addButton(self.penSize2)
self.sizeButtonGroup.addButton(self.penSize3)
self.sizeButtonGroup.buttonClicked.connect(self.sizeButtonToggled)
self.penSizeLayout.addWidget(self.penSize1)
self.penSizeLayout.addWidget(self.penSize2)
self.penSizeLayout.addWidget(self.penSize3)
self.penSizeLayout.setSpacing(5)
self.penSizeLayout.setContentsMargins(0, 0, 0, 0)
def initPenColorButtons(self):
self.colorSet = QWidget(self)
self.colorLayout = QHBoxLayout()
self.colorLayout.setSpacing(5)
self.colorLayout.setContentsMargins(5, 0, 5, 0)
self.colorSet.setLayout(self.colorLayout)
self.presentColor = QPushButton(self.colorSet)
self.presentColor.setFixedSize(self.iconWidth, self.iconHeight)
self.presentColor.setEnabled(False)
# adjust pen color
self.colorPick = QWidget(self.colorSet)
self.colorGrid = QGridLayout()
self.colorGrid.setSpacing(0)
self.colorGrid.setContentsMargins(5, 0, 5, 0)
self.colorPick.setLayout(self.colorGrid)
self.colorList = [('white', '#ffffff'), ('red', '#ff0000'),
('green', '#00ff00'), ('blue', '#0000ff'),
('cyan', '#00ffff'), ('magenta', '#ff00ff'),
('yellow', '#ffff00'), ('gray', '#a0a0a4'),
('black', '#000000'), ('darkRed', '#800000'),
('darkGreen', '#008000'), ('darkBlue', '#000080'),
('darkCyan', '#008080'), ('darkMagenta', '#800080'),
('darkYellow', '#808000'), ('darkGray', '#808080')]
self.generateButtons()
self.colorButtonGroup = QButtonGroup(self)
for button in self.colorButtons:
self.colorButtonGroup.addButton(button)
self.colorButtonGroup.buttonClicked.connect(self.colorButtonToggled)
# set the layout
tmp = 0
for x in range(0, 2):
for y in range(0, int(len(self.colorList) / 2)):
self.colorGrid.addWidget(self.colorButtons[tmp], x, y)
tmp += 1
self.colorGrid.setSpacing(0)
self.colorGrid.setContentsMargins(0, 0, 0, 0)
self.colorLayout.addWidget(self.presentColor)
self.colorLayout.addWidget(self.colorPick)
def initFontWidget(self):
self.fontDialog = QFontDialog()
self.changeFontButton = QPushButton(self)
self.fontDialog.setCurrentFont(QFont('Sans serif'))
self.changeFontButton.setText('{0} {1}'.format(
self.fontDialog.currentFont().family(),
self.fontDialog.currentFont().pointSize()))
self.changeFontButton.clicked.connect(self.fontButtonClicked)
def showFontWidget(self):
self.changeFontButton.show()
self.penSize1.hide()
self.penSize2.hide()
self.penSize3.hide()
def showPenWidget(self):
self.changeFontButton.hide()
self.penSize1.show()
self.penSize2.show()
self.penSize3.show()
# slots
def colorButtonToggled(self, button):
self.presentColor.setStyleSheet(
'QPushButton { background-color: %s; }' % button.objectName())
self.penColorTrigger.emit(button.objectName())
def sizeButtonToggled(self, button):
self.penSizeTrigger.emit(int(button.objectName()) * 2)
def fontButtonClicked(self):
ok = True
font = QFontDialog.getFont(self)
if font[1]:
self.changeFontButton.setText('{0} {1}'.format(
font[0].family(), font[0].pointSize()))
self.fontChangeTrigger.emit(font[0])
