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 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()]
