plotter/openmc_plotter/overlays.py at b14a92412daa88969aba4099ce2047c24888aa22 · openmc-dev/plotter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
from sys import platform

from PySide6 import QtGui, QtCore
from PySide6.QtWidgets import (QWidget, QTableWidget, QSizePolicy,
                               QPushButton, QTableWidgetItem, QVBoxLayout)

c_key = "⌘" if platform == 'darwin' else "Ctrl"


class ShortcutTableItem(QTableWidgetItem):

    def __init__(self):
        super().__init__()
        # disable selection, editing
        self.setFlags(QtCore.Qt.NoItemFlags)


class ShortcutsOverlay(QWidget):
    shortcuts = {"Color By": [("Cell", "Alt+C"),
                              ("Material", "Alt+M"),
                              ("Temperature", "Alt+T"),
                              ("Density", "Alt+D")],
                  "Views": [("Apply Changes", c_key + "+Enter"),
                            ("Undo", c_key + "+Z"),
                            ("Redo", "Shift+" + c_key+ "+Z"),
                            ("Restore Default Plot", c_key + "+R"),
                            ("Zoom", "Alt+Shift+Z"),
                            ("Zoom", "Shift+Scroll"),
                            ("Toggle Masking", c_key + "+M"),
                            ("Toggle Highlighting", c_key + "+L"),
                            ("Toggle Overlap Coloring", c_key + "+P"),
                            ("Toggle Undefined Coloring", c_key + "+I"),
                            ("Toggle Domain Outlines", c_key + "+U"),
                            ("Set XY Basis", "Alt+X"),
                            ("Set YZ Basis", "Alt+Y"),
                            ("Set XZ Basis", "Alt+Z"),
                            ("Update Plot Origin", "Double-click"),
                            ("Open Context Menu", "Right-click"),
                            ("When zoomed:", ""),
                            ("Vertical Scroll", "Scroll"),
                            ("Horizontal Scroll", "Alt+Scroll")],
                  "Menus": [("Hide/Show Dock", c_key + "+D"),
                            ("Reload Model", "Shift+" + c_key + "+R"),
                            ("Quit", c_key + "+Q"),
                            ("Display Shortcuts", "?")],
                 "Input/Output" : [("Save View", c_key + "+S"),
                                   ("Open View", c_key + "+O"),
                                   ("Save Plot Image", "Shift+" + c_key + "+S")]}

    # colors
    header_color = QtGui.QColor(150, 150, 150, 255)
    fillColor = QtGui.QColor(30, 30, 30, 200)
    framePenColor = QtGui.QColor(255, 255, 255, 120)
    textPenColor = QtGui.QColor(152, 196, 5, 255)

    def __init__(self, parent):
        super().__init__(parent)

        # transparent window fill
        self.layout = QVBoxLayout()
        self.setLayout(self.layout)

        n_rows = max((len(scts) for scts in self.shortcuts.values()))
        n_rows += 1  # plus one for header
        n_cols = len(self.shortcuts.keys()) * 3
        self.tableWidget = QTableWidget(n_rows, n_cols, self)
        self.layout.addWidget(self.tableWidget)

        # set all items to a non-editable cell item
        for i in range(n_rows):
            for j in range(n_cols):
                self.tableWidget.setItem(i, j, ShortcutTableItem())

        self.tableWidget.setShowGrid(False)
        self.tableWidget.setSizePolicy(QSizePolicy.Expanding,
                                       QSizePolicy.Expanding)
        self.tableWidget.verticalHeader().setVisible(False)
        self.tableWidget.horizontalHeader().setVisible(False)
        self.tableWidget.setStyleSheet("background-color: rgba(30, 30, 30, 230);"
                                       "border: 0px;"
                                       "padding: 20px")

        # populate table cells
        self.set_cells()

        self.close_btn = QPushButton(self)
        self.close_btn.setAttribute(QtCore.Qt.WA_TranslucentBackground)
        self.close_btn.setStyleSheet("background-color: rgba(0, 0, 0, 0);"
                                     "border: 0px;"
                                     "color: rgba(150, 150, 150, 255)")
        self.close_btn.setText("X")
        font = QtGui.QFont()
        self.close_btn.setFixedSize(30, 30)
        self.close_btn.clicked.connect(self.hide)

    def set_cells(self):
        # row, col indices
        row_idx = 0
        col_idx = 0

        for menu in self.shortcuts:
            # set menu header
            header_item = self.tableWidget.item(row_idx, col_idx)
            header_item.setForeground(QtGui.QColor(150, 150, 150, 255))
            header_item.setText(menu)
            header_item.setFlags(QtCore.Qt.NoItemFlags)
            row_idx += 1

            for shortcut in self.shortcuts[menu]:
                desc_item = self.tableWidget.item(row_idx, col_idx)
                desc_item.setForeground(self.textPenColor)
                desc_item.setText(shortcut[0])

                key_item = self.tableWidget.item(row_idx, col_idx + 1)
                key_item.setForeground(self.textPenColor)
                key_item.setText(shortcut[1])
                row_idx += 1
            # update for next menu
            row_idx = 0
            col_idx += 3

        self.tableWidget.resizeColumnsToContents()

    def keyPressEvent(self, event):
        # close if visible and esc is pressed
        if self.isVisible() and event.key() == QtCore.Qt.Key_Escape:
            self.close()

    def resizeEvent(self, event):
        overlay_size = self.size()
        btn_size = self.close_btn.size()
        x_pos = int(overlay_size.width() - btn_size.width()) - 5
        self.close_btn.move(x_pos, 5)