diff options
Diffstat (limited to 'src/ui.py')
| -rw-r--r-- | src/ui.py | 421 |
1 files changed, 421 insertions, 0 deletions
diff --git a/src/ui.py b/src/ui.py new file mode 100644 index 0000000..7803d3d --- /dev/null +++ b/src/ui.py @@ -0,0 +1,421 @@ +import numpy as np, scipy.optimize +import traceback +from PyQt5.QtWidgets import * +from PyQt5.QtGui import * +from PyQt5.QtCore import * + +import phys, file + +# global GUI-related shit +class GUI: + table = None + table_rows = list() + monospace = QFont("monospace") + bigfont = QFont("sans-serif", pointSize=10, weight=1000) + +class Pens: + axes = QPen(Qt.gray) + + ellipse = QPen(Qt.black) + ellipse.setWidth(2) + + axis_linear = QPen(QColor(201, 141, 0)) + axis_linear.setWidth(2) + axis_linear.setStyle(Qt.DashDotLine) + + axis_fast = QPen(QColor(51, 87, 123)) + axis_fast.setWidth(2) + axis_fast.setStyle(Qt.DashDotLine) + + axis_slow = QPen(QColor(55, 123, 51)) + axis_slow.setWidth(2) + axis_slow.setStyle(Qt.DashDotLine) + + alpha = QPen(Qt.red) + theta = QPen(Qt.blue) + + radii = QPen(Qt.black) + radii.setStyle(Qt.DashDotLine) + +from ui_widgets import * + + + +class OptBox(QVBoxLayout): + def __init__(self, row, pol, rownum): + super().__init__() + + # Name + self.name = MeinGroßLabel("Element %d" % rownum) + self.addWidget(self.name) + + # Type / Enable box + self.hbox = QHBoxLayout() + self.addLayout(self.hbox) + + # Type combo + self.type = QComboBox() + self.type.addItem("Linear", "linear",) + self.type.addItem("Quarterwave plate", "quarterwave") + self.type.setCurrentIndex(0 if pol.type == "linear" else 1) + self.type.currentIndexChanged.connect(lambda : TableRow.type_change(row)) + self.hbox.addWidget(self.type) + + # Enable checkbox + self.enable = QCheckBox("Enable") + self.enable.setChecked(True) + self.enable.stateChanged.connect(update) + self.hbox.addWidget(self.enable) + + # Delta angle + self.delta = AngleSlider("Delta") + self.delta.setValue(pol.delta * 180 / np.pi) + self.delta.on_change = lambda: TableRow.angle_change(row) + self.addLayout(self.delta) + + # Angle reference + self.ref = QComboBox() + self.ref.addItem("Absolute", False) + for i, _ in enumerate(system.elements): + if i >= rownum - 1: + continue + self.ref.addItem("Relative to element #%d" % (i + 1), i) + + if row.ref is not False and i == row.ref: + self.ref.setCurrentIndex(row.ref + 1) + + self.ref.currentIndexChanged.connect(update) + self.addWidget(self.ref) + + # Angle + self.angle = AngleSlider("Angle") + self.angle.setValue(pol.angle * 180 / np.pi) + self.angle.on_change = lambda: TableRow.angle_change(row) + self.addLayout(self.angle) + self.addItem(ExpandingSpacer()) + + # Add + self.add_above = QPushButton("Add above") + self.add_above.clicked.connect(lambda: half_assed_element_creation(rownum - 1)) + self.add_below = QPushButton("Add below") + self.add_below.clicked.connect(lambda: half_assed_element_creation(rownum)) + self.delete = QPushButton("Delete") + self.delete.clicked.connect(lambda: half_assed_element_deletion(rownum - 1)) + + self.buttons = QHBoxLayout() + self.buttons.addWidget(self.add_above) + self.buttons.addWidget(self.add_below) + self.buttons.addWidget(self.delete) + self.addLayout(self.buttons) + + + +class TableRow: + def __init__(self, pol, grid, rownum): + self.pol = pol + self.grid = grid + self.rownum = rownum + self.angle = pol.angle + self.delta = pol.delta + self.ref = pol.ref + + self.optbox = OptBox(self, pol, rownum) + grid.addLayout(self.optbox, rownum, 1) + + self.ellipse = EllipseWidget(pol) + grid.addWidget(self.ellipse, rownum, 2) + + self.info = QVBoxLayout() + self.info_angle = MeinLabel() + self.info_angle.setAlignment(Qt.AlignLeft) + self.info.addWidget(self.info_angle) + self.info.addWidget(QLabel("Polarization state")) + self.info_jones = MeinLabel() + self.info_jones.setAlignment(Qt.AlignLeft) + self.info.addWidget(self.info_jones) + self.info.addItem(ExpandingSpacer()) + grid.addLayout(self.info, rownum, 3) + + + @staticmethod + def angle_change(row): + row.angle = row.optbox.angle.angle / 180 * np.pi + row.delta = row.optbox.delta.angle / 180 * np.pi + update() + + @staticmethod + def type_change(row): + row.pol.set_type(row.optbox.type.currentData()) + update() + + +def populate_table(): + GUI.table = QGridLayout() + GUI.table.setColumnStretch(1, 1) + GUI.table.setColumnStretch(2, 2) + GUI.table.setColumnStretch(3, 5) + + while GUI.opt_operand.count() > 0: # FIXME + GUI.opt_operand.removeItem(0) + + GUI.table_rows = list() + for i, pol in enumerate(system.elements): + GUI.table_rows.append(TableRow(pol, GUI.table, i + 1)) + GUI.opt_operand.addItem("Element #%d" % (i + 1), i) + +def update(): + system.ignore = [False] * len(system.elements) + for i, pol in enumerate(system.elements): + row = GUI.table_rows[i] + + pol.angle = row.angle + pol.delta = row.delta + pol.ref = row.optbox.ref.currentData() + if pol.ref is not False: + pol.angle += system.elements[pol.ref].angle + + if not row.optbox.enable.isChecked(): + system.ignore[i] = True + + system.recalculate() + + I = 1 + for i, pol in enumerate(system.elements): + row = GUI.table_rows[i] + + # update all the diagrams + row.ellipse.state = system.states[i] + row.ellipse.ellipse = system.ellipses[i] + row.ellipse.is_used = not system.ignore[i] + row.ellipse.repaint() + + text = "%s (%f°)\nat %f°" \ + % (pol.type, pol.delta * 180 / np.pi, + (pol.angle + pol.delta) * 180 / np.pi) + row.info_angle.setText(text) + + state = system.states[i] + ellipse = system.ellipses[i] + + if state is None: + I = 1 + row.info_jones.setText("Unpolarized") + else: + I = np.abs(state[0] * state[0].conjugate() \ + + state[1] * state[1].conjugate()) + + if state is None: + continue + + A, B = state + text = "I = %f\n" % I + text += "%f ∠ %+f°\n%f ∠ %+f°\n" % \ + (np.abs(A), 180 / np.pi * np.angle(A), \ + np.abs(B), 180 / np.pi * np.angle(B)) + text += "α = %f°\nφ = %+f°" % \ + (180 / np.pi * ellipse.alpha, 180 / np.pi * ellipse.theta) + text = text.replace("-", "- ").replace("+", "+ ") + row.info_jones.setText(text) + + GUI.widok.intensity = I + GUI.widok.repaint() + + #if GUI.auto_optimize.isChecked(): + # optimize() + + +#TODO later +#class AddElementWindow(QMainWindow): +# dialog = None +# +# def __init__(self): +# super().__init__() +# +# chuj = QLabel("adolf") +# self.setCentralWidget(chuj) +# self.show() +# +# +# def __del__(self): +# AddElementWindow.dialog = None +# +# def open(): +# if not AddElementWindow.dialog: +# AddElementWindow.dialog = AddElementWindow() +def half_assed_element_creation(index=None): + if index is None: + system.elements.append(phys.Polarizer("linear")) + else: + system.elements.insert(index, phys.Polarizer("linear")) + + populate_table() + GUI.table_frame = QFrame() + GUI.table_frame.setLayout(GUI.table) + GUI.scroll.setWidget(GUI.table_frame) + update() + +def half_assed_clear(): + system.elements = list() + populate_table() + GUI.table_frame = QFrame() + GUI.table_frame.setLayout(GUI.table) + GUI.scroll.setWidget(GUI.table_frame) + update() + +def half_assed_element_deletion(index): + del(system.elements[index]) + + populate_table() + GUI.table_frame = QFrame() + GUI.table_frame.setLayout(GUI.table) + GUI.scroll.setWidget(GUI.table_frame) + update() + + +last_save_path = None +file_filter = "Wery Omportant Zejta (*.woz)" + +def do_save(win, reuse_old): + global system, last_save_path + + if reuse_old and last_save_path: + path = last_save_path + else: + path, _ = QFileDialog.getSaveFileName(win, filter=file_filter) + if path == "": + return + + try: + file.save_system(path, system) + except: + traceback.print_exc() + + last_save_path = path + +def do_open(win): + global system + + path, _ = QFileDialog.getOpenFileName(win, filter=file_filter) + if path == "": + return + + try: + system = file.open_system(path) + except: + traceback.print_exc() + + populate_table() + GUI.table_frame = QFrame() + GUI.table_frame.setLayout(GUI.table) + GUI.scroll.setWidget(GUI.table_frame) + update() + +def setup_menubar(win): + menu = win.menuBar() + + # File + open = QAction("&Open a system", win) + open.setShortcut("Ctrl+O") + open.triggered.connect(lambda: do_open(win)) + save = QAction("&Save a system", win) + save.setShortcut("Ctrl+S") + save.triggered.connect(lambda: do_save(win, True)) + save_as = QAction("&Save a system as...", win) + save_as.triggered.connect(lambda: do_save(win, False)) + close = QAction("Exit", win) + close.setShortcut("Ctrl+Q") + close.triggered.connect(exit) + + file = menu.addMenu("&File") + file.addAction(open) + file.addAction(save) + file.addAction(save_as) + file.addAction(close) + + # System + add = QAction("&Add a new element", win) + add.setShortcut("Ctrl+N") + add.triggered.connect(half_assed_element_creation) + clear = QAction("&Remove all elements", win) + clear.triggered.connect(half_assed_clear) + + system = menu.addMenu("&System") + system.addAction(add) + system.addAction(clear) + + win.statusBar() + +# FIXME: refactor +def optimize(which): + if len(system.elements) == 0: + return + + op_idx = GUI.opt_operand.currentData() + op = system.elements[op_idx] + + def I(theta): + op.angle = theta + if op.ref is not False: + op.ref += system.elements[op.ref].angle + + state = None + for i, pol in enumerate(system.elements): + if i >= len(system.ignore) or not system.ignore[i]: + state = pol.mul(state) + if state is None: + return 1 + else: + return float(np.abs(state[0] * state[0].conjugate() \ + + state[1] * state[1].conjugate())) + + if which == "min": + f = I + else: + f = lambda theta: -I(theta) + + opt = scipy.optimize.minimize_scalar(f, bounds=[0, np.pi], method="bounded") + GUI.table_rows[op_idx].optbox.angle.edit.setText("%g" % round(opt.x * 180 / np.pi, 3)) + + +def setup(win, system_): + global system + system = system_ + + # Needless to say, I'm getting real tired of this whole Layout/Widget + # clusterfuck in Qt + root = QHBoxLayout() + root_fucking_random_container = QWidget() + root_fucking_random_container.setLayout(root) + win.setCentralWidget(root_fucking_random_container) + + rhs = QVBoxLayout() + + Widocques.image = QImage("/home/enneract/lab/test/c.jpg") + GUI.widok = Widocques() + rhs.addWidget(GUI.widok) + + optbox = QHBoxLayout() + rhs.addLayout(optbox) + + GUI.opt_operand = QComboBox() + optbox.addWidget(GUI.opt_operand) + button = QPushButton("Find a minimum") + button.pressed.connect(lambda: optimize("min")) + optbox.addWidget(button) + button = QPushButton("Find a maximum") + button.pressed.connect(lambda: optimize("max")) + optbox.addWidget(button) + + GUI.scroll = QScrollArea() + populate_table() + GUI.table_frame = QFrame() + GUI.table_frame.setLayout(GUI.table) + GUI.scroll.setWidget(GUI.table_frame) + + root.addWidget(GUI.scroll) + root.addLayout(rhs) + + setup_menubar(win) + + update() + win.show()
\ No newline at end of file |