def setup_ui(self):
super(SettingUI, self).setup_ui()
max_width_label = QLabel(text="最大宽度",
alignment=QtCore.Qt.AlignRight
| QtCore.Qt.AlignVCenter)
max_width_combo = QComboBox()
max_width_combo.addItems(Globals.max_width_arr)
max_width_combo.setCurrentIndex(self.max_width_index)
max_width_combo.currentIndexChanged.connect(self.on_combo_changed)
size_label = QLabel(text="输出字号",
alignment=QtCore.Qt.AlignRight
| QtCore.Qt.AlignVCenter)
size_spin = QSpinBox()
size_spin.setRange(10, 64)
size_spin.setValue(Globals.config.get(Globals.UserData.font_size))
size_spin.valueChanged.connect(self.on_save_font_size)
output_name_label = QLabel(text="输出名称",
alignment=QtCore.Qt.AlignRight
| QtCore.Qt.AlignVCenter)
output_name_edit = QLineEdit(
Globals.config.get(Globals.UserData.font_save_name))
output_name_edit.textEdited.connect(self.on_output_name_edited)
output_label = QLabel(text="输出目录",
alignment=QtCore.Qt.AlignRight
| QtCore.Qt.AlignVCenter)
output_line_edit = QLineEdit(
Globals.config.get(Globals.UserData.output_dir))
output_line_edit.setReadOnly(True)
output_line_edit.cursorPositionChanged.connect(
self.on_output_choose_clicked)
mode_1_radio = QRadioButton("图集模式")
mode_2_radio = QRadioButton("字体模式")
mode_1_radio.setChecked(True)
mode_1_radio.toggled.connect(self.on_mode_1_radio_toggled)
mode_radio_group = QButtonGroup(self)
mode_radio_group.addButton(mode_1_radio, 0)
mode_radio_group.addButton(mode_2_radio, 1)
self.main_layout.setColumnStretch(1, 1)
self.main_layout.setColumnStretch(3, 1)
self.main_layout.addWidget(mode_1_radio, 0, 0, 1, 1)
self.main_layout.addWidget(mode_2_radio, 0, 1, 1, 1)
self.main_layout.addWidget(output_name_label, 1, 0, 1, 1)
self.main_layout.addWidget(output_name_edit, 1, 1, 1, 1)
self.main_layout.addWidget(max_width_label, 1, 2, 1, 1)
self.main_layout.addWidget(max_width_combo, 1, 3, 1, 1)
self.main_layout.addWidget(output_label, 2, 0, 1, 1)
self.main_layout.addWidget(output_line_edit, 2, 1, 1, 1)
self.main_layout.addWidget(size_label, 2, 2, 1, 1)
self.main_layout.addWidget(size_spin, 2, 3, 1, 1)
self.max_width_combo = max_width_combo
self.size_spin = size_spin
self.output_name_edit = output_name_edit
self.output_line_edit = output_line_edit
def create_data_group(self):
"""Creates group for holding specific choice data selection"""
button_group = QButtonGroup(self)
button_group.setExclusive(True)
colecao_radio_button = QRadioButton('Coleção')
self.partidas_radio_button = QRadioButton('Partidas')
colecao_radio_button.setChecked(True)
button_group.addButton(colecao_radio_button)
button_group.addButton(self.partidas_radio_button)
(self.min_date_picker,
min_date_label) = create_date_picker('À Partir de:', self)
(self.max_date_picker,
max_date_label) = create_date_picker('Até:', self)
self.min_date_picker.dateChanged.connect(
self.max_date_picker.setMinimumDate)
colecao_radio_button.toggled.connect(self.min_date_picker.setDisabled)
colecao_radio_button.toggled.connect(self.max_date_picker.setDisabled)
self.map_users_button = QPushButton(
'Ver mapa de usuarios BGG -> Ludopedia', self)
self.map_users_button.setEnabled(False)
self.map_users_button.clicked.connect(self.user_map)
colecao_radio_button.toggled.connect(self.map_users_button.setDisabled)
group_box = QGroupBox('Dados')
grid_layout = QGridLayout(group_box)
grid_layout.addWidget(colecao_radio_button, 1, 1)
grid_layout.addWidget(self.partidas_radio_button, 1, 2)
grid_layout.addWidget(min_date_label, 2, 1)
grid_layout.addWidget(self.min_date_picker, 2, 2)
grid_layout.addWidget(max_date_label, 3, 1)
grid_layout.addWidget(self.max_date_picker, 3, 2)
grid_layout.addWidget(self.map_users_button, 4, 1, 1, 2)
group_box.setLayout(grid_layout)
return group_box
def setup_ui(self):
super(FontUI, self).setup_ui()
sys_radio = QRadioButton("系统字体")
sys_radio.setChecked(True)
font_combo = QComboBox()
for item in self.system_fonts:
font_combo.addItem(item[1])
font_combo.setCurrentIndex(0)
font_combo.currentIndexChanged.connect(self.refresh_font)
custom_radio = QRadioButton("自选字体")
custom_radio.setCheckable(True)
custom_radio.setChecked(False)
custom_radio.toggled.connect(self.on_custom_radio_toggled)
custom_edit = QLineEdit()
custom_edit.setPlaceholderText("自定义字体（*.ttf, *.otf）")
custom_edit.setReadOnly(True)
custom_edit.setEnabled(False)
custom_btn = QPushButton(text="浏览")
custom_btn.setEnabled(False)
custom_btn.clicked.connect(self.on_custom_clicked)
font_group = QButtonGroup(self)
font_group.addButton(sys_radio, 0)
font_group.addButton(custom_radio, 1)
font_group.setExclusive(True)
input_label = QLabel("输出字符")
input_edit = MLineEdit()
import_btn = QPushButton(text="从文本文件导入")
import_btn.clicked.connect(self.on_import_btn_clicked)
self.main_layout.setColumnStretch(1, 1)
self.main_layout.addWidget(sys_radio, 1, 0, 1, 1)
self.main_layout.addWidget(font_combo, 1, 1, 1, 2)
self.main_layout.addWidget(custom_radio, 2, 0, 1, 1)
self.main_layout.addWidget(custom_edit, 2, 1, 1, 1)
self.main_layout.addWidget(custom_btn, 2, 2, 1, 1)
self.main_layout.addWidget(input_label, 3, 0, 1, 1)
self.main_layout.addWidget(import_btn, 3, 2, 1, 1)
self.main_layout.addWidget(input_edit, 4, 0, 1, 3)
self.input_edit = input_edit
self.font_combo = font_combo
self.custom_btn = custom_btn
self.custom_edit = custom_edit
self.refresh_font()
def __init__(self, existing_rtss, parent=None):
QDialog.__init__(self, parent=parent)
if platform.system() == 'Darwin':
self.stylesheet_path = "res/stylesheet.qss"
else:
self.stylesheet_path = "res/stylesheet-win-linux.qss"
stylesheet = open(resource_path(self.stylesheet_path)).read()
self.setStyleSheet(stylesheet)
self.setWindowTitle("Multiple RTSTRUCTs detected!")
self.setMinimumSize(350, 180)
self.icon = QtGui.QIcon()
self.icon.addPixmap(
QtGui.QPixmap(resource_path("res/images/icon.ico")),
QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.setWindowIcon(self.icon)
self.explanation_text = QLabel("Multiple RTSTRUCTs attached to the "
"selected image set have been "
"identified."
"\nPlease select 1 "
"RTSTRUCTs to continue!")
# Create scrolling area widget to contain the content.
self.scroll_area = QScrollArea()
self.scroll_area.setWidgetResizable(True)
self.scroll_area_content = QWidget(self.scroll_area)
self.scroll_area.ensureWidgetVisible(self.scroll_area_content)
# Create layout for checkboxes
self.layout_content = QVBoxLayout(self.scroll_area_content)
self.layout_content.setContentsMargins(5, 5, 5, 5)
self.layout_content.setSpacing(0)
self.layout_content.setAlignment(QtCore.Qt.AlignTop
| QtCore.Qt.AlignTop)
# Add all the attached RTSSs as checkboxes
self.checkbox_group = QButtonGroup()
self.checkbox_group.setExclusive(True)
for i in range(len(existing_rtss)):
checkbox = QCheckBox()
checkbox.rtss = existing_rtss[i]
rtss = dcmread(checkbox.rtss.get_files()[0])
checkbox.setFocusPolicy(QtCore.Qt.NoFocus)
checkbox.setText("Series: %s (%s, %s %s)" % (
checkbox.rtss.series_description,
checkbox.rtss.get_series_type(),
len(rtss.StructureSetROISequence),
"ROIs" if len(rtss.StructureSetROISequence) > 1 else "ROI"
))
self.checkbox_group.addButton(checkbox)
self.layout_content.addWidget(checkbox)
self.checkbox_group.buttonClicked.connect(self.on_checkbox_clicked)
# Create a cancel button
self.cancel_button = QPushButton("Cancel")
self.cancel_button.clicked.connect(self.on_cancel_clicked)
# Create a continue button
self.continue_button = QPushButton("Continue Process")
self.continue_button.setDisabled(True)
self.continue_button.clicked.connect(self.on_continue_clicked)
# Create a widget to contain cancel and continue buttons
self.button_area = QWidget()
self.button_layout = QHBoxLayout()
self.button_layout.addWidget(self.cancel_button)
self.button_layout.addWidget(self.continue_button)
self.button_area.setLayout(self.button_layout)
# Add all components to a vertical layout
self.layout = QVBoxLayout()
self.layout.addWidget(self.explanation_text)
self.layout.addWidget(self.scroll_area)
self.layout.addWidget(self.button_area)
self.setLayout(self.layout)
def __init__(self, ruler, parent=None):
QWidget.__init__(self, parent)
# units (cm/px/inch)
unitsGroup = QButtonGroup(self)
pixels = QRadioButton('Pixels')
pixels.unit = 'px'
centimeters = QRadioButton('Centimeters')
centimeters.unit = 'cm'
inches = QRadioButton('Inches')
inches.unit = 'inch'
selectedUnit = ruler.data.get('units')
if selectedUnit == 'cm':
centimeters.setChecked(True)
elif selectedUnit == 'inch':
inches.setChecked(True)
else:
pixels.setChecked(True)
unitsGroup.addButton(pixels)
unitsGroup.addButton(centimeters)
unitsGroup.addButton(inches)
def changeUnits(radioButton):
nonlocal selectedUnit
newUnit = radioButton.unit
if newUnit != selectedUnit:
ruler.data.update('units', newUnit)
selectedUnit = newUnit
ruler.update()
unitsGroup.buttonClicked.connect(changeUnits)
# orientation (horizontal/vertical)
horizontalOrientation = ruler.data.get('horizontal_orientation')
orientationGroup = QButtonGroup(self)
self.horizontal_string = 'Horizontal'
self.vertical_string = 'Vertical'
horizontal = QRadioButton(self.horizontal_string)
vertical = QRadioButton(self.vertical_string)
if horizontalOrientation:
horizontal.setChecked(True)
self.selected_orientation = self.horizontal_string
else:
vertical.setChecked(True)
self.selected_orientation = self.vertical_string
def changeOrientation(radioButton):
text = radioButton.text()
if self.selected_orientation != text:
self.selected_orientation = text
ruler.rotate()
orientationGroup.addButton(horizontal)
orientationGroup.addButton(vertical)
orientationGroup.buttonClicked.connect(changeOrientation)
# length settings (always above + division lines + current length)
alwaysAbove = QCheckBox('Always Above', self)
alwaysAbove.setChecked(ruler.data.get('always_above'))
def alwaysAboveSetting():
if alwaysAbove.isChecked():
ruler.data.update('always_above', True)
ruler.setWindowFlags(ruler.windowFlags()
| Qt.WindowStaysOnTopHint)
else:
ruler.data.update('always_above', False)
ruler.setWindowFlags(ruler.windowFlags()
& ~Qt.WindowStaysOnTopHint)
ruler.show()
alwaysAbove.clicked.connect(alwaysAboveSetting)
divisionLines = QCheckBox('1/2 1/4 3/4 divisions', self)
divisionLines.setChecked(ruler.data.get('division_lines'))
def divisionLinesSetting():
ruler.data.update('division_lines', divisionLines.isChecked())
ruler.update()
divisionLines.clicked.connect(divisionLinesSetting)
currentLength = QLabel('0px')
currentLength.setAlignment(Qt.AlignCenter)
# setup the layout
layout = QGridLayout()
layout.setSpacing(10)
# set the same width/height for all the columns/rows
columnWidth = 120
rowHeight = 25
layout.setRowMinimumHeight(0, rowHeight)
layout.setRowMinimumHeight(1, rowHeight)
layout.setRowMinimumHeight(2, rowHeight)
layout.setColumnMinimumWidth(0, columnWidth)
layout.setColumnMinimumWidth(1, columnWidth)
layout.setColumnMinimumWidth(2, columnWidth)
# first column
layout.addWidget(pixels, 0, 0)
layout.addWidget(centimeters, 1, 0)
layout.addWidget(inches, 2, 0)
# second column
layout.addWidget(horizontal, 0, 1)
layout.addWidget(vertical, 1, 1)
# third column
layout.addWidget(alwaysAbove, 0, 2)
layout.addWidget(divisionLines, 1, 2)
layout.addWidget(currentLength, 2, 2)
layout.setSizeConstraint(QLayout.SetFixedSize)
self.current_length = currentLength
self.vertical = vertical
self.horizontal = horizontal
self.setLayout(layout)
def setupUi(self, Img):
if not Img.objectName():
Img.setObjectName(u"Img")
Img.resize(620, 559)
self.gridLayout = QGridLayout(Img)
self.gridLayout.setObjectName(u"gridLayout")
self.graphicsView = QGraphicsView(Img)
self.graphicsView.setObjectName(u"graphicsView")
self.gridLayout.addWidget(self.graphicsView, 0, 0, 1, 1)
self.verticalLayout = QVBoxLayout()
self.verticalLayout.setObjectName(u"verticalLayout")
self.line_3 = QFrame(Img)
self.line_3.setObjectName(u"line_3")
self.line_3.setFrameShape(QFrame.VLine)
self.line_3.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line_3)
self.verticalLayout_2 = QVBoxLayout()
self.verticalLayout_2.setObjectName(u"verticalLayout_2")
self.line_5 = QFrame(Img)
self.line_5.setObjectName(u"line_5")
self.line_5.setFrameShape(QFrame.VLine)
self.line_5.setFrameShadow(QFrame.Sunken)
self.verticalLayout_2.addWidget(self.line_5)
self.checkBox = QCheckBox(Img)
self.checkBox.setObjectName(u"checkBox")
self.checkBox.setMaximumSize(QSize(100, 16777215))
self.checkBox.setChecked(True)
self.verticalLayout_2.addWidget(self.checkBox)
self.ttaModel = QCheckBox(Img)
self.ttaModel.setObjectName(u"ttaModel")
self.ttaModel.setMaximumSize(QSize(70, 16777215))
self.verticalLayout_2.addWidget(self.ttaModel)
self.horizontalLayout_3 = QHBoxLayout()
self.horizontalLayout_3.setObjectName(u"horizontalLayout_3")
self.scaleRadio = QRadioButton(Img)
self.buttonGroup_2 = QButtonGroup(Img)
self.buttonGroup_2.setObjectName(u"buttonGroup_2")
self.buttonGroup_2.addButton(self.scaleRadio)
self.scaleRadio.setObjectName(u"scaleRadio")
self.scaleRadio.setMaximumSize(QSize(80, 16777215))
self.scaleRadio.setChecked(True)
self.horizontalLayout_3.addWidget(self.scaleRadio)
self.scaleEdit = QLineEdit(Img)
self.scaleEdit.setObjectName(u"scaleEdit")
self.scaleEdit.setMaximumSize(QSize(160, 16777215))
self.scaleEdit.setAlignment(Qt.AlignCenter)
self.horizontalLayout_3.addWidget(self.scaleEdit)
self.verticalLayout_2.addLayout(self.horizontalLayout_3)
self.horizontalLayout_4 = QHBoxLayout()
self.horizontalLayout_4.setObjectName(u"horizontalLayout_4")
self.heighRadio = QRadioButton(Img)
self.buttonGroup_2.addButton(self.heighRadio)
self.heighRadio.setObjectName(u"heighRadio")
self.heighRadio.setMaximumSize(QSize(80, 16777215))
self.horizontalLayout_4.addWidget(self.heighRadio)
self.widthEdit = QLineEdit(Img)
self.widthEdit.setObjectName(u"widthEdit")
self.widthEdit.setEnabled(False)
self.widthEdit.setMaximumSize(QSize(60, 16777215))
self.widthEdit.setAlignment(Qt.AlignCenter)
self.horizontalLayout_4.addWidget(self.widthEdit)
self.label_2 = QLabel(Img)
self.label_2.setObjectName(u"label_2")
self.label_2.setMaximumSize(QSize(20, 16777215))
self.horizontalLayout_4.addWidget(self.label_2)
self.heighEdit = QLineEdit(Img)
self.heighEdit.setObjectName(u"heighEdit")
self.heighEdit.setEnabled(False)
self.heighEdit.setMaximumSize(QSize(60, 16777215))
self.heighEdit.setAlignment(Qt.AlignCenter)
self.horizontalLayout_4.addWidget(self.heighEdit)
self.verticalLayout_2.addLayout(self.horizontalLayout_4)
self.horizontalLayout_5 = QHBoxLayout()
self.horizontalLayout_5.setObjectName(u"horizontalLayout_5")
self.label_4 = QLabel(Img)
self.label_4.setObjectName(u"label_4")
self.label_4.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_5.addWidget(self.label_4)
self.noiseCombox = QComboBox(Img)
self.noiseCombox.addItem("")
self.noiseCombox.addItem("")
self.noiseCombox.addItem("")
self.noiseCombox.addItem("")
self.noiseCombox.addItem("")
self.noiseCombox.setObjectName(u"noiseCombox")
self.noiseCombox.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_5.addWidget(self.noiseCombox)
self.verticalLayout_2.addLayout(self.horizontalLayout_5)
self.horizontalLayout_6 = QHBoxLayout()
self.horizontalLayout_6.setObjectName(u"horizontalLayout_6")
self.label_5 = QLabel(Img)
self.label_5.setObjectName(u"label_5")
self.label_5.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_6.addWidget(self.label_5)
self.comboBox = QComboBox(Img)
self.comboBox.addItem("")
self.comboBox.addItem("")
self.comboBox.addItem("")
self.comboBox.setObjectName(u"comboBox")
self.comboBox.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_6.addWidget(self.comboBox)
self.verticalLayout_2.addLayout(self.horizontalLayout_6)
self.horizontalLayout_7 = QHBoxLayout()
self.horizontalLayout_7.setObjectName(u"horizontalLayout_7")
self.changeJpg = QPushButton(Img)
self.changeJpg.setObjectName(u"changeJpg")
self.changeJpg.setMaximumSize(QSize(100, 16777215))
self.horizontalLayout_7.addWidget(self.changeJpg)
self.changePng = QPushButton(Img)
self.changePng.setObjectName(u"changePng")
self.changePng.setMaximumSize(QSize(100, 16777215))
self.horizontalLayout_7.addWidget(self.changePng)
self.changeLabel = QLabel(Img)
self.changeLabel.setObjectName(u"changeLabel")
self.changeLabel.setMaximumSize(QSize(100, 16777215))
self.changeLabel.setAlignment(Qt.AlignCenter)
self.horizontalLayout_7.addWidget(self.changeLabel)
self.verticalLayout_2.addLayout(self.horizontalLayout_7)
self.horizontalLayout_11 = QHBoxLayout()
self.horizontalLayout_11.setObjectName(u"horizontalLayout_11")
self.verticalLayout_2.addLayout(self.horizontalLayout_11)
self.verticalLayout.addLayout(self.verticalLayout_2)
self.line = QFrame(Img)
self.line.setObjectName(u"line")
self.line.setFrameShape(QFrame.VLine)
self.line.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line)
self.line_4 = QFrame(Img)
self.line_4.setObjectName(u"line_4")
self.line_4.setFrameShape(QFrame.HLine)
self.line_4.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line_4)
self.verticalLayout_3 = QVBoxLayout()
self.verticalLayout_3.setObjectName(u"verticalLayout_3")
self.horizontalLayout_8 = QHBoxLayout()
self.horizontalLayout_8.setObjectName(u"horizontalLayout_8")
self.label_8 = QLabel(Img)
self.label_8.setObjectName(u"label_8")
self.label_8.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_8.addWidget(self.label_8)
self.resolutionLabel = QLabel(Img)
self.resolutionLabel.setObjectName(u"resolutionLabel")
self.resolutionLabel.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_8.addWidget(self.resolutionLabel)
self.verticalLayout_3.addLayout(self.horizontalLayout_8)
self.horizontalLayout_9 = QHBoxLayout()
self.horizontalLayout_9.setObjectName(u"horizontalLayout_9")
self.label_10 = QLabel(Img)
self.label_10.setObjectName(u"label_10")
self.label_10.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_9.addWidget(self.label_10)
self.sizeLabel = QLabel(Img)
self.sizeLabel.setObjectName(u"sizeLabel")
self.sizeLabel.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_9.addWidget(self.sizeLabel)
self.verticalLayout_3.addLayout(self.horizontalLayout_9)
self.horizontalLayout = QHBoxLayout()
self.horizontalLayout.setObjectName(u"horizontalLayout")
self.label = QLabel(Img)
self.label.setObjectName(u"label")
self.label.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout.addWidget(self.label)
self.gpuName = QLabel(Img)
self.gpuName.setObjectName(u"gpuName")
self.gpuName.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout.addWidget(self.gpuName)
self.verticalLayout_3.addLayout(self.horizontalLayout)
self.horizontalLayout_10 = QHBoxLayout()
self.horizontalLayout_10.setObjectName(u"horizontalLayout_10")
self.label_6 = QLabel(Img)
self.label_6.setObjectName(u"label_6")
self.label_6.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_10.addWidget(self.label_6)
self.tickLabel = QLabel(Img)
self.tickLabel.setObjectName(u"tickLabel")
self.tickLabel.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_10.addWidget(self.tickLabel)
self.verticalLayout_3.addLayout(self.horizontalLayout_10)
self.oepnButton = QPushButton(Img)
self.oepnButton.setObjectName(u"oepnButton")
self.oepnButton.setMaximumSize(QSize(100, 16777215))
self.verticalLayout_3.addWidget(self.oepnButton)
self.horizontalLayout_2 = QHBoxLayout()
self.horizontalLayout_2.setObjectName(u"horizontalLayout_2")
self.verticalLayout_3.addLayout(self.horizontalLayout_2)
self.pushButton_3 = QPushButton(Img)
self.pushButton_3.setObjectName(u"pushButton_3")
self.pushButton_3.setMaximumSize(QSize(100, 16777215))
self.verticalLayout_3.addWidget(self.pushButton_3)
self.pushButton = QPushButton(Img)
self.pushButton.setObjectName(u"pushButton")
self.pushButton.setMaximumSize(QSize(100, 16777215))
self.verticalLayout_3.addWidget(self.pushButton)
self.saveButton = QPushButton(Img)
self.saveButton.setObjectName(u"saveButton")
self.saveButton.setMaximumSize(QSize(100, 16777215))
self.verticalLayout_3.addWidget(self.saveButton)
self.verticalLayout.addLayout(self.verticalLayout_3)
self.line_6 = QFrame(Img)
self.line_6.setObjectName(u"line_6")
self.line_6.setFrameShape(QFrame.HLine)
self.line_6.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line_6)
self.verticalSpacer = QSpacerItem(20, 40, QSizePolicy.Minimum,
QSizePolicy.Expanding)
self.verticalLayout.addItem(self.verticalSpacer)
self.line_2 = QFrame(Img)
self.line_2.setObjectName(u"line_2")
self.line_2.setFrameShape(QFrame.VLine)
self.line_2.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line_2)
self.gridLayout.addLayout(self.verticalLayout, 0, 1, 1, 1)
self.retranslateUi(Img)
self.checkBox.clicked.connect(Img.SwithPicture)
self.saveButton.clicked.connect(Img.SavePicture)
self.heighEdit.textChanged.connect(Img.CheckScaleRadio)
self.pushButton_3.clicked.connect(Img.ReduceScalePic)
self.heighRadio.clicked.connect(Img.CheckScaleRadio)
self.ttaModel.clicked.connect(Img.CheckScaleRadio)
self.oepnButton.clicked.connect(Img.OpenPicture)
self.widthEdit.textChanged.connect(Img.CheckScaleRadio)
self.pushButton.clicked.connect(Img.AddScalePic)
self.scaleEdit.textChanged.connect(Img.CheckScaleRadio)
self.scaleRadio.clicked.connect(Img.CheckScaleRadio)
self.comboBox.currentIndexChanged.connect(Img.ChangeModel)
self.changeJpg.clicked.connect(Img.StartWaifu2x)
self.noiseCombox.currentIndexChanged.connect(Img.CheckScaleRadio)
self.changePng.clicked.connect(Img.StartWaifu2xPng)
self.changeJpg.clicked.connect(Img.StartWaifu2xJPG)
QMetaObject.connectSlotsByName(Img)
class Ui_Img(object):
def setupUi(self, Img):
if not Img.objectName():
Img.setObjectName(u"Img")
Img.resize(620, 559)
self.gridLayout = QGridLayout(Img)
self.gridLayout.setObjectName(u"gridLayout")
self.graphicsView = QGraphicsView(Img)
self.graphicsView.setObjectName(u"graphicsView")
self.gridLayout.addWidget(self.graphicsView, 0, 0, 1, 1)
self.verticalLayout = QVBoxLayout()
self.verticalLayout.setObjectName(u"verticalLayout")
self.line_3 = QFrame(Img)
self.line_3.setObjectName(u"line_3")
self.line_3.setFrameShape(QFrame.VLine)
self.line_3.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line_3)
self.verticalLayout_2 = QVBoxLayout()
self.verticalLayout_2.setObjectName(u"verticalLayout_2")
self.line_5 = QFrame(Img)
self.line_5.setObjectName(u"line_5")
self.line_5.setFrameShape(QFrame.VLine)
self.line_5.setFrameShadow(QFrame.Sunken)
self.verticalLayout_2.addWidget(self.line_5)
self.checkBox = QCheckBox(Img)
self.checkBox.setObjectName(u"checkBox")
self.checkBox.setMaximumSize(QSize(100, 16777215))
self.checkBox.setChecked(True)
self.verticalLayout_2.addWidget(self.checkBox)
self.ttaModel = QCheckBox(Img)
self.ttaModel.setObjectName(u"ttaModel")
self.ttaModel.setMaximumSize(QSize(70, 16777215))
self.verticalLayout_2.addWidget(self.ttaModel)
self.horizontalLayout_3 = QHBoxLayout()
self.horizontalLayout_3.setObjectName(u"horizontalLayout_3")
self.scaleRadio = QRadioButton(Img)
self.buttonGroup_2 = QButtonGroup(Img)
self.buttonGroup_2.setObjectName(u"buttonGroup_2")
self.buttonGroup_2.addButton(self.scaleRadio)
self.scaleRadio.setObjectName(u"scaleRadio")
self.scaleRadio.setMaximumSize(QSize(80, 16777215))
self.scaleRadio.setChecked(True)
self.horizontalLayout_3.addWidget(self.scaleRadio)
self.scaleEdit = QLineEdit(Img)
self.scaleEdit.setObjectName(u"scaleEdit")
self.scaleEdit.setMaximumSize(QSize(160, 16777215))
self.scaleEdit.setAlignment(Qt.AlignCenter)
self.horizontalLayout_3.addWidget(self.scaleEdit)
self.verticalLayout_2.addLayout(self.horizontalLayout_3)
self.horizontalLayout_4 = QHBoxLayout()
self.horizontalLayout_4.setObjectName(u"horizontalLayout_4")
self.heighRadio = QRadioButton(Img)
self.buttonGroup_2.addButton(self.heighRadio)
self.heighRadio.setObjectName(u"heighRadio")
self.heighRadio.setMaximumSize(QSize(80, 16777215))
self.horizontalLayout_4.addWidget(self.heighRadio)
self.widthEdit = QLineEdit(Img)
self.widthEdit.setObjectName(u"widthEdit")
self.widthEdit.setEnabled(False)
self.widthEdit.setMaximumSize(QSize(60, 16777215))
self.widthEdit.setAlignment(Qt.AlignCenter)
self.horizontalLayout_4.addWidget(self.widthEdit)
self.label_2 = QLabel(Img)
self.label_2.setObjectName(u"label_2")
self.label_2.setMaximumSize(QSize(20, 16777215))
self.horizontalLayout_4.addWidget(self.label_2)
self.heighEdit = QLineEdit(Img)
self.heighEdit.setObjectName(u"heighEdit")
self.heighEdit.setEnabled(False)
self.heighEdit.setMaximumSize(QSize(60, 16777215))
self.heighEdit.setAlignment(Qt.AlignCenter)
self.horizontalLayout_4.addWidget(self.heighEdit)
self.verticalLayout_2.addLayout(self.horizontalLayout_4)
self.horizontalLayout_5 = QHBoxLayout()
self.horizontalLayout_5.setObjectName(u"horizontalLayout_5")
self.label_4 = QLabel(Img)
self.label_4.setObjectName(u"label_4")
self.label_4.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_5.addWidget(self.label_4)
self.noiseCombox = QComboBox(Img)
self.noiseCombox.addItem("")
self.noiseCombox.addItem("")
self.noiseCombox.addItem("")
self.noiseCombox.addItem("")
self.noiseCombox.addItem("")
self.noiseCombox.setObjectName(u"noiseCombox")
self.noiseCombox.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_5.addWidget(self.noiseCombox)
self.verticalLayout_2.addLayout(self.horizontalLayout_5)
self.horizontalLayout_6 = QHBoxLayout()
self.horizontalLayout_6.setObjectName(u"horizontalLayout_6")
self.label_5 = QLabel(Img)
self.label_5.setObjectName(u"label_5")
self.label_5.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_6.addWidget(self.label_5)
self.comboBox = QComboBox(Img)
self.comboBox.addItem("")
self.comboBox.addItem("")
self.comboBox.addItem("")
self.comboBox.setObjectName(u"comboBox")
self.comboBox.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_6.addWidget(self.comboBox)
self.verticalLayout_2.addLayout(self.horizontalLayout_6)
self.horizontalLayout_7 = QHBoxLayout()
self.horizontalLayout_7.setObjectName(u"horizontalLayout_7")
self.changeJpg = QPushButton(Img)
self.changeJpg.setObjectName(u"changeJpg")
self.changeJpg.setMaximumSize(QSize(100, 16777215))
self.horizontalLayout_7.addWidget(self.changeJpg)
self.changePng = QPushButton(Img)
self.changePng.setObjectName(u"changePng")
self.changePng.setMaximumSize(QSize(100, 16777215))
self.horizontalLayout_7.addWidget(self.changePng)
self.changeLabel = QLabel(Img)
self.changeLabel.setObjectName(u"changeLabel")
self.changeLabel.setMaximumSize(QSize(100, 16777215))
self.changeLabel.setAlignment(Qt.AlignCenter)
self.horizontalLayout_7.addWidget(self.changeLabel)
self.verticalLayout_2.addLayout(self.horizontalLayout_7)
self.horizontalLayout_11 = QHBoxLayout()
self.horizontalLayout_11.setObjectName(u"horizontalLayout_11")
self.verticalLayout_2.addLayout(self.horizontalLayout_11)
self.verticalLayout.addLayout(self.verticalLayout_2)
self.line = QFrame(Img)
self.line.setObjectName(u"line")
self.line.setFrameShape(QFrame.VLine)
self.line.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line)
self.line_4 = QFrame(Img)
self.line_4.setObjectName(u"line_4")
self.line_4.setFrameShape(QFrame.HLine)
self.line_4.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line_4)
self.verticalLayout_3 = QVBoxLayout()
self.verticalLayout_3.setObjectName(u"verticalLayout_3")
self.horizontalLayout_8 = QHBoxLayout()
self.horizontalLayout_8.setObjectName(u"horizontalLayout_8")
self.label_8 = QLabel(Img)
self.label_8.setObjectName(u"label_8")
self.label_8.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_8.addWidget(self.label_8)
self.resolutionLabel = QLabel(Img)
self.resolutionLabel.setObjectName(u"resolutionLabel")
self.resolutionLabel.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_8.addWidget(self.resolutionLabel)
self.verticalLayout_3.addLayout(self.horizontalLayout_8)
self.horizontalLayout_9 = QHBoxLayout()
self.horizontalLayout_9.setObjectName(u"horizontalLayout_9")
self.label_10 = QLabel(Img)
self.label_10.setObjectName(u"label_10")
self.label_10.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_9.addWidget(self.label_10)
self.sizeLabel = QLabel(Img)
self.sizeLabel.setObjectName(u"sizeLabel")
self.sizeLabel.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_9.addWidget(self.sizeLabel)
self.verticalLayout_3.addLayout(self.horizontalLayout_9)
self.horizontalLayout = QHBoxLayout()
self.horizontalLayout.setObjectName(u"horizontalLayout")
self.label = QLabel(Img)
self.label.setObjectName(u"label")
self.label.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout.addWidget(self.label)
self.gpuName = QLabel(Img)
self.gpuName.setObjectName(u"gpuName")
self.gpuName.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout.addWidget(self.gpuName)
self.verticalLayout_3.addLayout(self.horizontalLayout)
self.horizontalLayout_10 = QHBoxLayout()
self.horizontalLayout_10.setObjectName(u"horizontalLayout_10")
self.label_6 = QLabel(Img)
self.label_6.setObjectName(u"label_6")
self.label_6.setMaximumSize(QSize(60, 16777215))
self.horizontalLayout_10.addWidget(self.label_6)
self.tickLabel = QLabel(Img)
self.tickLabel.setObjectName(u"tickLabel")
self.tickLabel.setMaximumSize(QSize(160, 16777215))
self.horizontalLayout_10.addWidget(self.tickLabel)
self.verticalLayout_3.addLayout(self.horizontalLayout_10)
self.oepnButton = QPushButton(Img)
self.oepnButton.setObjectName(u"oepnButton")
self.oepnButton.setMaximumSize(QSize(100, 16777215))
self.verticalLayout_3.addWidget(self.oepnButton)
self.horizontalLayout_2 = QHBoxLayout()
self.horizontalLayout_2.setObjectName(u"horizontalLayout_2")
self.verticalLayout_3.addLayout(self.horizontalLayout_2)
self.pushButton_3 = QPushButton(Img)
self.pushButton_3.setObjectName(u"pushButton_3")
self.pushButton_3.setMaximumSize(QSize(100, 16777215))
self.verticalLayout_3.addWidget(self.pushButton_3)
self.pushButton = QPushButton(Img)
self.pushButton.setObjectName(u"pushButton")
self.pushButton.setMaximumSize(QSize(100, 16777215))
self.verticalLayout_3.addWidget(self.pushButton)
self.saveButton = QPushButton(Img)
self.saveButton.setObjectName(u"saveButton")
self.saveButton.setMaximumSize(QSize(100, 16777215))
self.verticalLayout_3.addWidget(self.saveButton)
self.verticalLayout.addLayout(self.verticalLayout_3)
self.line_6 = QFrame(Img)
self.line_6.setObjectName(u"line_6")
self.line_6.setFrameShape(QFrame.HLine)
self.line_6.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line_6)
self.verticalSpacer = QSpacerItem(20, 40, QSizePolicy.Minimum,
QSizePolicy.Expanding)
self.verticalLayout.addItem(self.verticalSpacer)
self.line_2 = QFrame(Img)
self.line_2.setObjectName(u"line_2")
self.line_2.setFrameShape(QFrame.VLine)
self.line_2.setFrameShadow(QFrame.Sunken)
self.verticalLayout.addWidget(self.line_2)
self.gridLayout.addLayout(self.verticalLayout, 0, 1, 1, 1)
self.retranslateUi(Img)
self.checkBox.clicked.connect(Img.SwithPicture)
self.saveButton.clicked.connect(Img.SavePicture)
self.heighEdit.textChanged.connect(Img.CheckScaleRadio)
self.pushButton_3.clicked.connect(Img.ReduceScalePic)
self.heighRadio.clicked.connect(Img.CheckScaleRadio)
self.ttaModel.clicked.connect(Img.CheckScaleRadio)
self.oepnButton.clicked.connect(Img.OpenPicture)
self.widthEdit.textChanged.connect(Img.CheckScaleRadio)
self.pushButton.clicked.connect(Img.AddScalePic)
self.scaleEdit.textChanged.connect(Img.CheckScaleRadio)
self.scaleRadio.clicked.connect(Img.CheckScaleRadio)
self.comboBox.currentIndexChanged.connect(Img.ChangeModel)
self.changeJpg.clicked.connect(Img.StartWaifu2x)
self.noiseCombox.currentIndexChanged.connect(Img.CheckScaleRadio)
self.changePng.clicked.connect(Img.StartWaifu2xPng)
self.changeJpg.clicked.connect(Img.StartWaifu2xJPG)
QMetaObject.connectSlotsByName(Img)
# setupUi
def retranslateUi(self, Img):
Img.setWindowTitle(QCoreApplication.translate("Img", u"Form", None))
self.checkBox.setText(
QCoreApplication.translate("Img", u"waifu2x", None))
#if QT_CONFIG(tooltip)
self.ttaModel.setToolTip(
QCoreApplication.translate(
"Img",
u"\u753b\u8d28\u63d0\u5347\uff0c\u8017\u65f6\u589e\u52a0",
None))
#endif // QT_CONFIG(tooltip)
self.ttaModel.setText(
QCoreApplication.translate("Img", u"tta\u6a21\u5f0f", None))
self.scaleRadio.setText(
QCoreApplication.translate("Img", u"\u500d\u6570\u653e\u5927",
None))
self.scaleEdit.setText(QCoreApplication.translate("Img", u"2", None))
self.heighRadio.setText(
QCoreApplication.translate("Img", u"\u56fa\u5b9a\u957f\u5bbd",
None))
self.label_2.setText(QCoreApplication.translate("Img", u"x", None))
self.label_4.setText(
QCoreApplication.translate("Img", u"\u964d\u566a\uff1a", None))
self.noiseCombox.setItemText(
0, QCoreApplication.translate("Img", u"3", None))
self.noiseCombox.setItemText(
1, QCoreApplication.translate("Img", u"2", None))
self.noiseCombox.setItemText(
2, QCoreApplication.translate("Img", u"1", None))
self.noiseCombox.setItemText(
3, QCoreApplication.translate("Img", u"0", None))
self.noiseCombox.setItemText(
4, QCoreApplication.translate("Img", u"-1", None))
self.label_5.setText(
QCoreApplication.translate("Img", u"\u6a21\u578b\uff1a", None))
self.comboBox.setItemText(
0, QCoreApplication.translate("Img", u"cunet", None))
self.comboBox.setItemText(
1, QCoreApplication.translate("Img", u"photo", None))
self.comboBox.setItemText(
2, QCoreApplication.translate("Img", u"anime_style_art_rgb", None))
self.changeJpg.setText(
QCoreApplication.translate("Img", u"\u8f6c\u6362JPG", None))
self.changePng.setText(
QCoreApplication.translate("Img", u"\u8f6c\u6362PNG", None))
self.changeLabel.setText("")
self.label_8.setText(
QCoreApplication.translate("Img", u"\u5206\u8fa8\u7387\uff1a",
None))
self.resolutionLabel.setText(
QCoreApplication.translate("Img", u"TextLabel", None))
self.label_10.setText(
QCoreApplication.translate("Img", u"\u5927 \u5c0f\uff1a", None))
self.sizeLabel.setText(
QCoreApplication.translate("Img", u"TextLabel", None))
self.label.setText(QCoreApplication.translate("Img", u"GPU:", None))
self.gpuName.setText(
QCoreApplication.translate("Img", u"TextLabel", None))
self.label_6.setText(
QCoreApplication.translate("Img", u"\u8017\u65f6\uff1a", None))
self.tickLabel.setText("")
self.oepnButton.setText(
QCoreApplication.translate("Img", u"\u6253\u5f00\u56fe\u7247",
None))
self.pushButton_3.setText(
QCoreApplication.translate("Img", u"\u7f29\u5c0f", None))
self.pushButton.setText(
QCoreApplication.translate("Img", u"\u653e\u5927", None))
self.saveButton.setText(
QCoreApplication.translate("Img", u"\u4fdd\u5b58\u56fe\u7247",
None))
def __init__(self) -> None:
QMainWindow.__init__(self)
self.setSizePolicy(
QSizePolicy(QSizePolicy.Maximum, QSizePolicy.Maximum))
self.setMaximumSize(QSize(1920, 1080))
self.setStyleSheet("padding: 0px; margin: 0px;")
self.setIconSize(QSize(32, 32))
self.setWindowTitle("BossyBot 2000 - Image Tagger")
self.setWindowIcon(self.load_icon(icon))
self.menubar = QMenuBar(self)
self.menubar.setSizePolicy(EXP_MAX)
self.menubar.setMaximumSize(QSize(INFINITE, 30))
self.menu_file = QMenu('File', self.menubar)
self.menu_options = QMenu('Options', self.menubar)
self.menu_help = QMenu('Help', self.menubar)
self.menubar.addAction(self.menu_file.menuAction())
self.menubar.addAction(self.menu_options.menuAction())
self.menubar.addAction(self.menu_help.menuAction())
self.open = QAction('Open', self)
self.menu_file.addAction(self.open)
self.open.triggered.connect(self.open_file)
self.exit_button = QAction('Exit', self)
self.exit_button.triggered.connect(lambda: sys.exit(0),
Qt.QueuedConnection)
self.menu_file.addAction(self.exit_button)
self.setMenuBar(self.menubar)
self.previous_button = QAction(self.load_icon(previous), '<<', self)
self.next_button = QAction(self.load_icon(next_icon), '>>', self)
self.rotate_left_button = QAction(self.load_icon(left), '', self)
self.rotate_right_button = QAction(self.load_icon(right), '', self)
self.play_button = QAction(self.load_icon(play), '', self)
self.play_button.setCheckable(True)
self.delete_button = QAction(self.load_icon(delete), '', self)
self.reload_button = QAction(self.load_icon(reload), '', self)
self.mirror_button = QAction('Mirror', self)
self.actual_size_button = QAction('Actual Size', self)
self.browser_button = QAction('Browser', self)
self.browser_button.setCheckable(True)
self.browser_button.setChecked(True)
self.crop_button = QAction('Crop', self)
self.crop_button.setCheckable(True)
self.toolbuttons = {
self.rotate_left_button: {
'shortcut':
',',
'connect':
lambda: self.pixmap.setRotation(self.pixmap.rotation() - 90)
},
self.rotate_right_button: {
'shortcut':
'.',
'connect':
lambda: self.pixmap.setRotation(self.pixmap.rotation() + 90)
},
self.delete_button: {
'shortcut': 'Del',
'connect': self.delete
},
self.previous_button: {
'shortcut': 'Left',
'connect': self.previous
},
self.play_button: {
'shortcut': 'Space',
'connect': self.play
},
self.next_button: {
'shortcut': 'Right',
'connect': self.next
},
self.reload_button: {
'shortcut': 'F5',
'connect': self.reload
}
}
self.toolbar = QToolBar(self)
self.toolbar.setSizePolicy(EXP_MAX)
self.toolbar.setMaximumSize(QSize(INFINITE, 27))
for _ in (self.browser_button, self.crop_button, self.mirror_button,
self.actual_size_button):
self.toolbar.addAction(_)
self.addToolBar(Qt.TopToolBarArea, self.toolbar)
for button in self.toolbuttons:
button.setShortcut(self.toolbuttons[button]['shortcut'])
button.triggered.connect(self.toolbuttons[button]['connect'])
self.toolbar.addAction(button)
self.centralwidget = QWidget(self)
self.centralwidget.setSizePolicy(EXP_EXP)
self.setCentralWidget(self.centralwidget)
self.grid = QGridLayout(self.centralwidget)
self.media = QGraphicsScene(self)
self.media.setItemIndexMethod(QGraphicsScene.NoIndex)
self.media.setBackgroundBrush(QBrush(Qt.black))
self.view = MyView(self.media, self)
self.view.setSizePolicy(EXP_EXP)
self.media.setSceneRect(0, 0, self.view.width(), self.view.height())
self.grid.addWidget(self.view, 0, 0, 1, 1)
self.frame = QFrame(self.centralwidget)
self.frame.setSizePolicy(
QSizePolicy(QSizePolicy.Fixed, QSizePolicy.MinimumExpanding))
self.frame.setMinimumSize(QSize(325, 500))
self.frame.setStyleSheet(
"QFrame { border: 4px inset #222; border-radius: 10; }")
self.layout_widget = QWidget(self.frame)
self.layout_widget.setGeometry(QRect(0, 400, 321, 91))
self.layout_widget.setContentsMargins(15, 15, 15, 15)
self.grid2 = QGridLayout(self.layout_widget)
self.grid2.setContentsMargins(0, 0, 0, 0)
self.save_button = QPushButton('Yes (Save)', self.layout_widget)
self.save_button.setSizePolicy(FIX_FIX)
self.save_button.setMaximumSize(QSize(120, 26))
self.save_button.setVisible(False)
self.grid2.addWidget(self.save_button, 1, 0, 1, 1)
self.no_save_button = QPushButton('No (Reload)', self.layout_widget)
self.no_save_button.setSizePolicy(FIX_FIX)
self.no_save_button.setMaximumSize(QSize(120, 26))
self.no_save_button.setVisible(False)
self.grid2.addWidget(self.no_save_button, 1, 1, 1, 1)
self.label = QLabel("Current image modified, save it?",
self.layout_widget)
self.label.setSizePolicy(FIX_FIX)
self.label.setMaximumSize(QSize(325, 60))
self.label.setVisible(False)
self.label.setAlignment(Qt.AlignCenter)
self.grid2.addWidget(self.label, 0, 0, 1, 2)
self.layout_widget = QWidget(self.frame)
self.layout_widget.setGeometry(QRect(0, 0, 321, 213))
self.ass = QRadioButton('Ass', self.layout_widget)
self.ass_exposed = QRadioButton('Ass (exposed)', self.layout_widget)
self.ass_reset = QRadioButton(self.frame)
self.ass_group = QButtonGroup(self)
self.breasts = QRadioButton('Breasts', self.layout_widget)
self.breasts_exposed = QRadioButton('Breasts (exposed)',
self.layout_widget)
self.breasts_reset = QRadioButton(self.frame)
self.breasts_group = QButtonGroup(self)
self.pussy = QRadioButton('Pussy', self.layout_widget)
self.pussy_exposed = QRadioButton('Pussy (exposed)',
self.layout_widget)
self.pussy_reset = QRadioButton(self.frame)
self.pussy_group = QButtonGroup(self)
self.fully_clothed = QRadioButton('Fully Clothed', self.layout_widget)
self.fully_nude = QRadioButton('Fully Nude', self.layout_widget)
self.nudity_reset = QRadioButton(self.frame)
self.nudity = QButtonGroup(self)
self.smiling = QRadioButton('Smiling', self.layout_widget)
self.glaring = QRadioButton('Glaring', self.layout_widget)
self.expression_reset = QRadioButton(self.frame)
self.expression = QButtonGroup(self)
self.grid3 = QGridLayout(self.layout_widget)
self.grid3.setVerticalSpacing(15)
self.grid3.setContentsMargins(0, 15, 0, 0)
self.radios = {
self.ass: {
'this': 'ass',
'that': 'ass_exposed',
'group': self.ass_group,
'reset': self.ass_reset,
'grid': (0, 0, 1, 1)
},
self.ass_exposed: {
'this': 'ass_exposed',
'that': 'ass',
'group': self.ass_group,
'reset': self.ass_reset,
'grid': (0, 1, 1, 1)
},
self.breasts: {
'this': 'breasts',
'that': 'breasts_exposed',
'group': self.breasts_group,
'reset': self.breasts_reset,
'grid': (1, 0, 1, 1)
},
self.breasts_exposed: {
'this': 'breasts_exposed',
'that': 'breasts',
'group': self.breasts_group,
'reset': self.breasts_reset,
'grid': (1, 1, 1, 1)
},
self.pussy: {
'this': 'pussy',
'that': 'pussy_exposed',
'group': self.pussy_group,
'reset': self.pussy_reset,
'grid': (2, 0, 1, 1)
},
self.pussy_exposed: {
'this': 'pussy_exposed',
'that': 'pussy',
'group': self.pussy_group,
'reset': self.pussy_reset,
'grid': (2, 1, 1, 1)
},
self.fully_clothed: {
'this': 'fully_clothed',
'that': 'fully_nude',
'group': self.nudity,
'reset': self.nudity_reset,
'grid': (3, 0, 1, 1)
},
self.fully_nude: {
'this': 'fully_nude',
'that': 'fully_clothed',
'group': self.nudity,
'reset': self.nudity_reset,
'grid': (3, 1, 1, 1)
},
self.smiling: {
'this': 'smiling',
'that': 'glaring',
'group': self.expression,
'reset': self.expression_reset,
'grid': (4, 0, 1, 1)
},
self.glaring: {
'this': 'glaring',
'that': 'smiling',
'group': self.expression,
'reset': self.expression_reset,
'grid': (4, 1, 1, 1)
},
}
for radio in self.radios:
radio.setSizePolicy(FIX_FIX)
radio.setMaximumSize(QSize(150, 22))
self.radios[radio]['reset'].setGeometry(QRect(0, 0, 0, 0))
self.grid3.addWidget(radio, *self.radios[radio]['grid'])
if self.radios[radio]['group'] != self.nudity:
radio.toggled.connect(
lambda x=_, y=radio: self.annotate(self.radios[y]['this']))
self.radios[radio]['group'].addButton(radio)
self.radios[radio]['group'].addButton(self.radios[radio]['reset'])
self.save_tags_button = QPushButton('Save Tags', self.layout_widget)
self.save_tags_button.setSizePolicy(FIX_FIX)
self.save_tags_button.setMaximumSize(QSize(120, 26))
self.grid3.addWidget(self.save_tags_button, 5, 1, 1, 1)
self.grid.addWidget(self.frame, 0, 1, 1, 1)
self.browse_bar = QLabel(self.centralwidget)
self.browse_bar.setSizePolicy(EXP_FIX)
self.browse_bar.setMinimumSize(QSize(0, 100))
self.browse_bar.setMaximumSize(QSize(INFINITE, 100))
self.browse_bar.setStyleSheet("background: #000;")
self.browse_bar.setAlignment(Qt.AlignCenter)
self.h_box2 = QHBoxLayout(self.browse_bar)
self.h_box2.setContentsMargins(4, 0, 0, 0)
self.grid.addWidget(self.browse_bar, 1, 0, 1, 2)
hiders = [
self.no_save_button.clicked, self.save_button.clicked,
self.reload_button.triggered
]
for hider in hiders:
hider.connect(self.save_button.hide)
hider.connect(self.no_save_button.hide)
hider.connect(self.label.hide)
showers = [
self.mirror_button.triggered, self.rotate_right_button.triggered,
self.rotate_left_button.triggered
]
for shower in showers:
shower.connect(self.save_button.show)
shower.connect(self.no_save_button.show)
shower.connect(self.label.show)
self.no_save_button.clicked.connect(self.reload)
self.browser_button.toggled.connect(self.browse_bar.setVisible)
self.play_button.toggled.connect(lambda: self.frame.setVisible(
(True, False)[self.frame.isVisible()]))
self.reload_button.triggered.connect(self.reload)
self.mirror_button.triggered.connect(lambda: self.pixmap.setScale(-1))
self.save_button.clicked.connect(self.save_image)
self.play_button.toggled.connect(
lambda: self.browser_button.setChecked(
(True, False)[self.browse_bar.isVisible()]))
self.crop_button.toggled.connect(self.view.reset)
self.actual_size_button.triggered.connect(self.actual_size)
self.browser_button.triggered.connect(self.browser)
self.save_tags_button.clicked.connect(self.save_tags)
self.view.got_rect.connect(self.set_rect)
self.crop_rect = QRect(QPoint(0, 0), QSize(0, 0))
self.dir_now = os.getcwd()
self.files = []
self.index = 0
self.refresh_files()
self.pixmap_is_scaled = False
self.pixmap = QGraphicsPixmapItem()
self.active_tag = ''
self.reset_browser = False
self.txt = PngInfo()
def __init__(
self,
controller: StageControllerShot702,
controller_status: DeviceStatus,
parent: QWidget = None,
):
super().__init__(parent=parent)
self._controller = controller
self._controller_status = controller_status
self._lcd_position = ALabel()
self._int_slider = IntSlider()
self._t_btn_up = create_tool_button(
arrow_type=Qt.UpArrow,
fixed_height=50,
fixed_width=50,
toggled=self.up_stage,
)
self._t_btn_down = create_tool_button(
arrow_type=Qt.DownArrow,
fixed_height=50,
fixed_width=50,
icon_size=QSize(30, 30),
toggled=self.down_stage,
)
self._t_btn_stop = create_tool_button(
fixed_width=50,
fixed_height=50,
icon=create_qicon(IconNames.STOP_WHITE),
icon_size=QSize(70, 70),
toggled=self.stop_stage,
)
self._p_button_fix_zero = create_push_button(clicked=self.fix_zero,
text="Fix Zero")
self._p_button_move_machine_zero = create_push_button(
clicked=self.move_to_machine_zero, text="Move Machine Zero-Point")
self._p_button_set_speed = create_push_button(
clicked=self.set_stage_speed, fixed_width=100, text="Set")
# timer
self.timer_measure_position = create_timer(
parent=self, timeout=self.measure_position)
# setup
self._int_slider.range = 1, 50000
button_group = QButtonGroup(self)
button_group.addButton(self._t_btn_up)
button_group.addButton(self._t_btn_stop)
button_group.addButton(self._t_btn_down)
button_group.setExclusive(True)
# setup layout
v_layout_control = AVBoxLayout()
v_layout_control.addWidget(self._t_btn_up)
v_layout_control.addWidget(self._t_btn_stop)
v_layout_control.addWidget(self._t_btn_down)
v_layout_control.setAlignment(Qt.AlignHCenter)
group_control = QGroupBox("Control")
group_control.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
group_control.setLayout(v_layout_control)
h_layout_position = AHBoxLayout()
h_layout_position.addWidget(add_unit(self._lcd_position, "μm"))
group_position = QGroupBox("Current Position")
group_position.setLayout(h_layout_position)
v_layout_origin = AVBoxLayout()
v_layout_origin.addWidget(self._p_button_fix_zero)
v_layout_origin.addWidget(self._p_button_move_machine_zero)
group_origin = QGroupBox("Origin")
group_origin.setLayout(v_layout_origin)
v_layout_speed = AVBoxLayout()
v_layout_speed.addWidget(add_unit(self._int_slider, "μm/msec"))
v_layout_speed.addWidget(self._p_button_set_speed,
alignment=Qt.AlignRight)
group_speed = QGroupBox("Speed Setting")
group_speed.setLayout(v_layout_speed)
g_layout = QGridLayout()
g_layout.addWidget(group_control, 1, 1, 2, 1)
g_layout.addWidget(group_position, 1, 2)
g_layout.addWidget(group_origin, 2, 2)
g_layout.addWidget(group_speed, 3, 1, 1, 2)
self.setLayout(g_layout)