class ColorChoiceWidget(ColorWidget): def __init__(self): super(ColorChoiceWidget, self).__init__() self.buttons = [] self.buttonGroup = QButtonGroup() self.buttonGroup.setExclusive(True) self.buttonGroup.buttonClicked.connect(self.selectColor) def setColors(self, colors): layout = self.buttonWidget.layout() id = 0 for color in colors: button = ColorButton() button.setColor(color) button.setCheckable(True) layout.addWidget(button) self.buttonGroup.addButton(button) self.buttonGroup.setId(button, id) id += 1 layout.addStretch(10) def setColor(self, color): self.color = color for button in self.buttonGroup.buttons(): diffs = map(lambda x, y: abs(x - y), button.color, color) if sum(diffs) < 0.0001: button.setChecked(True) break def selectColor(self, button): self.color = button.color self.valueChanged.emit(self.color)
class PickerTypeDialog(QDialog): """ PickerTypeDialog """ def __init__(self, parent): super(PickerTypeDialog, self).__init__(parent) self.pickerType = None self.pickerTypes = [(SurfaceType, "Surface picker"), (TwoStepType, "Two step picker")] self.radioButtons = [] for picker in self.pickerTypes: self.radioButtons.append(QRadioButton(picker[1])) # self.radioButtons[0].setChecked(True) self.buttonGroup = QButtonGroup() ind = 0 for button in self.radioButtons: self.buttonGroup.addButton(button) self.buttonGroup.setId(button, ind) ind += 1 self.nextButton = QPushButton("Choose") self.nextButton.clicked.connect(self.choose) self.cancelButton = QPushButton("Cancel") self.cancelButton.clicked.connect(self.cancel) groupLayout = QVBoxLayout() for radioButton in self.radioButtons: groupLayout.addWidget(radioButton) self.groupBox = QGroupBox("Choose picker type:") self.groupBox.setLayout(groupLayout) self.setModal(True) layout = QGridLayout() layout.setAlignment(Qt.AlignTop) layout.addWidget(self.groupBox, 0, 0, 1, 2) layout.addWidget(self.cancelButton, 1, 0) layout.addWidget(self.nextButton, 1, 1) self.setLayout(layout) @Slot() def choose(self): selectedIndex = self.buttonGroup.checkedId() if selectedIndex >= 0: # load choosen picker type self.pickerType = self.pickerTypes[selectedIndex][0] self.accept() @Slot() def cancel(self): self.reject()
class ChoiceDialog(QDialog): def __init__(self, choices, help_texts=[], parent=None): super(ChoiceDialog, self).__init__(parent) title = "select one from choices" self.setWindowTitle(title) layout = QVBoxLayout(self) self.choiceButtonGroup = QButtonGroup(self) # it is not a visible UI self.choiceButtonGroup.setExclusive(True) if choices and len(choices) >= 1: if len(help_texts) < len(choices): help_texts = choices self.choices = choices for id, choice in enumerate(self.choices): rb = QRadioButton(choice) rb.setToolTip(help_texts[id]) self.choiceButtonGroup.addButton(rb) self.choiceButtonGroup.setId( rb, id) # negative id if not specified layout.addWidget(rb) if id == 0: rb.setChecked(True) self.choiceButtonGroup.buttonClicked.connect(self.choiceChanged) # OK and Cancel buttons buttons = QDialogButtonBox( QDialogButtonBox.Ok | QDialogButtonBox.Cancel, Qt.Horizontal, self) buttons.accepted.connect(self.accept) buttons.rejected.connect(self.reject) layout.addWidget(buttons) self.setLayout(layout) def choice(self): return self.choices[self.choiceButtonGroup.checkedId()] def choiceId(self): return self.choiceButtonGroup.checkedId() def choiceChanged(self): #print(self.choiceButtonGroup.checkedId()) self.currentChoice = self.choices[self.choiceButtonGroup.checkedId()]
class ChoiceWidget(QWidget): def __init__(self, choices, help_texts=[], parent=None): super(ChoiceWidget, self).__init__(parent) title = "select one from choices" self.setWindowTitle(title) layout = QVBoxLayout(self) self.choiceButtonGroup = QButtonGroup(self) # it is not a visible UI self.choiceButtonGroup.setExclusive(True) if choices and len(choices) >= 1: if len(help_texts) < len(choices): help_texts = choices self.choices = choices for id, choice in enumerate(self.choices): rb = QRadioButton(choice) rb.setToolTip(help_texts[id]) self.choiceButtonGroup.addButton(rb) self.choiceButtonGroup.setId( rb, id) # negative id if not specified layout.addWidget(rb) if id == 0: rb.setChecked(True) self.choiceButtonGroup.buttonClicked.connect(self.choiceChanged) self.setLayout(layout) def choice(self): return self.choices[self.choiceButtonGroup.checkedId()] def setChoice(self, choice): for i, ch in enumerate(self.choices): if ch == choice: self.choiceButtonGroup.button(i).setChecked(True) def choiceId(self): return self.choiceButtonGroup.checkedId() def choiceChanged(self): #print(self.choiceButtonGroup.checkedId()) self.currentChoice = self.choices[self.choiceButtonGroup.checkedId()]