def __selectColourSlot(self, button):
"""
Private slot to select a color.
@param button reference to the button been pressed
@type QPushButton
"""
colorKey = button.property("colorKey")
hasAlpha = button.property("hasAlpha")
colDlg = QColorDialog(self)
if hasAlpha:
colDlg.setOptions(QColorDialog.ShowAlphaChannel)
# Set current colour last to avoid conflicts with alpha channel
colDlg.setCurrentColor(self.__coloursDict[colorKey][0])
colDlg.currentColorChanged.connect(
lambda col: self.colourChanged.emit(colorKey, col))
colDlg.exec_()
if colDlg.result() == colDlg.Accepted:
colour = colDlg.selectedColor()
size = button.iconSize()
pm = QPixmap(size.width(), size.height())
pm.fill(colour)
button.setIcon(QIcon(pm))
self.__coloursDict[colorKey][0] = colour
# Update colour selection
self.colourChanged.emit(colorKey, self.__coloursDict[colorKey][0])
def locs():
qcd = QColorDialog(parent=vis.get_main_window())
qcd.setWindowFlag(Qt.WindowStaysOnTopHint, True)
qcd.setCurrentColor(QColor.fromRgbF(*vis.get_grid_coordinates_color()))
qcd.exec()
if qcd.result() == 1:
vis.set_grid_coordinates_color((qcd.selectedColor().getRgbF()))
def bg():
qcd = QColorDialog(parent=vis.get_main_window())
qcd.setWindowFlag(Qt.WindowStaysOnTopHint, True)
qcd.setCurrentColor(QColor.fromRgbF(*vis.get_background_color()))
qcd.exec()
if qcd.result() == 1:
vis.set_background_color((qcd.selectedColor().getRgbF()[:3]))
def brd():
qcd = QColorDialog(parent=vis.get_main_window())
qcd.setOption(QColorDialog.ShowAlphaChannel)
qcd.setWindowFlag(Qt.WindowStaysOnTopHint, True)
qcd.setCurrentColor(QColor.fromRgbF(*vis.get_grid_border_color()))
qcd.exec()
if qcd.result() == 1:
vis.set_grid_border_color((qcd.selectedColor().getRgbF()))
def cp():
cd = QColorDialog(parent=vis.get_main_window())
cd.setWindowFlag(Qt.WindowStaysOnTopHint, True)
cd.setOptions(QColorDialog.ShowAlphaChannel)
cd.setCurrentColor(QColor.fromRgbF(*vis.get_added_matter_color()))
cd.exec()
if cd.result() == 1:
vis.set_added_matter_color((cd.selectedColor().getRgbF()))
def _on_pick_tree(self, event):
"""a matplotlib callback function for when a tree is clicked on"""
if event.artist != self._treepoints:
# print "you clicked on something other than a node"
return True
ind = event.ind[0]
self.tree_selected = self._tree_list[ind]
print("tree clicked on", self.tree_selected)
# launch a color selector dialog and color
# all subtrees by the selected color
color_dialog = QColorDialog(parent=self)
color_dialog.exec_()
if color_dialog.result():
color = color_dialog.selectedColor()
rgba = color.getRgbF() # red green blue alpha
print("color", rgba)
rgb = rgba[:3]
for tree in self.tree_selected.get_all_trees():
tree.data["colour"] = rgb
self.redraw_disconnectivity_graph()
