diff options
Diffstat (limited to 'src/ui.py')
| -rw-r--r-- | src/ui.py | 403 |
1 files changed, 113 insertions, 290 deletions
@@ -9,7 +9,6 @@ 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) @@ -38,213 +37,7 @@ class Pens: radii.setStyle(Qt.DashDotLine) from ui_widgets import * - - - -class ElementEditorWindow(QMainWindow): - windows = list() - - @staticmethod - def open(pol): - win = ElementEditorWindow(pol) - win.show() - ElementEditorWindow.windows.append(win) - - @staticmethod - def close_all(pol=None): - for win in ElementEditorWindow.windows: - if pol is None or pol == win.pol: - win.close() - del(win) - - def __init__(self, pol): - super().__init__(flags=Qt.Dialog) - self.pol = pol - - self.setWindowTitle("Element editor") - root = QVBoxLayout() - root_fucking_random_container = QWidget() - root_fucking_random_container.setLayout(root) - self.setCentralWidget(root_fucking_random_container) - - # Name - box = QHBoxLayout() - root.addLayout(box) - box.addWidget(QLabel("Name")) - self.name = QLineEdit(self.pol.name) - self.name.textChanged.connect(self.change_name) - box.addWidget(self.name) - - # OPTICS - root.addWidget(MeinGroßLabel("Optics")) - - # 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(self.change_type) - root.addWidget(self.type) - - # TRANSMITTANCE - root.addWidget(MeinGroßLabel("Transmittance")) - - box = QHBoxLayout() - root.addLayout(box) - - box.addWidget(QLabel("T")) - self.t1 = QLineEdit("%g" % self.pol.t1) - self.t1.textChanged.connect(self.change_t) - box.addWidget(self.t1) - self.t2 = QLineEdit("%g" % self.pol.t2) - self.t2.textChanged.connect(self.change_t) - box.addWidget(self.t2) - - # ARRANGEMENT - root.addWidget(MeinGroßLabel("Arrangement")) - - # Angle reference - self.ref = QComboBox() - self.populate_ref() - self.ref.currentIndexChanged.connect(update) - root.addWidget(self.ref) - - # Delta angle - self.delta = AngleSlider("Delta") - self.delta.on_change = self.change_delta - root.addLayout(self.delta) - - root.addItem(ExpandingSpacer()) - - # Done - self.done = QPushButton("Done") - self.done.pressed.connect(self.close) - root.addWidget(self.done) - - def change_name(self): - self.pol.name = self.name.text() - self.pol.row.optbox.name.setText(self.pol.name) - - def change_type(self): - self.pol.set_type(self.type.currentData()) - update() - - def change_delta(self): - try: - self.pol.delta = self.delta.angle / 180 * np.pi - update() - except ValueError: - pass - - def change_t(self): - try: - self.pol.t1 = float(self.t1.text()) - self.pol.t2 = float(self.t2.text()) - update() - except ValueError: - pass - - def populate_ref(self): - self.ref.addItem("Absolute", False) - for i, _ in enumerate(system.elements): - if i >= self.pol.rownum - 1: - continue - self.ref.addItem("Relative to #%d" % (i + 1), i) - - if self.pol.ref is not False and i == self.pol.ref: - self.ref.setCurrentIndex(self.pol.ref + 1) - - - -class OptBox(QVBoxLayout): - def __init__(self, row, pol, rownum): - super().__init__() - self.row = row - self.rownum = rownum - - # Name - self.name = MeinGroßLabel(pol.name) - self.name.setContextMenuPolicy(Qt.CustomContextMenu) - self.name.customContextMenuRequested.connect(lambda pos: OptBox.open_menu(self, pos)) - self.addWidget(self.name) - - # Enable checkbox - self.enable = QCheckBox("Enable") - self.enable.setChecked(True) - self.enable.stateChanged.connect(self.change_enable) - self.addWidget(self.enable) - - # Angle - self.angle = AngleSlider() - self.angle.setValue(pol.angle * 180 / np.pi) - self.angle.on_change = lambda: TableRow.angle_change(row) - self.addLayout(self.angle) - self.addItem(ExpandingSpacer()) - - def open_menu(self, pos): - menu = QMenu() - menu.addAction("Edit", lambda: ElementEditorWindow.open(self.row.pol)) - menu.addAction("Insert before", lambda: half_assed_element_creation(self.rownum - 1)) - menu.addAction("Insert after", lambda: half_assed_element_creation(self.rownum)) - menu.addAction("Delete", lambda: half_assed_element_deletion(self.rownum - 1)) - menu.exec_(self.name.mapToGlobal(pos)) - - def change_enable(self): - self.row.pol.enable = self.enable.isChecked() - update() - - -class TableRow: - def __init__(self, pol, grid, rownum): - self.pol = pol - pol.row = self - pol.rownum = rownum - 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.pol.angle = row.angle - update() - - - -def populate_table(): - GUI.input_intensity.setText("%g" % system.input_intensity) - - 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) +from ui_table import * def update(): for i, pol in enumerate(system.elements): @@ -406,96 +199,126 @@ def setup_menubar(win): 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 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.angle += 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)) + +class MainWindow(QMainWindow): + def __init__(self, system_): + super().__init__() + self.statusBar() + self.setWindowTitle("Polarizzazione italiana") + + global system + system = system_ + + ## Needless to say, I'm getting real tired of this whole Layout/Widget + ## clusterfuck in Qt + #root = QVBoxLayout() + #root_fucking_random_container = QWidget() + #root_fucking_random_container.setLayout(root) + #self.setCentralWidget(root_fucking_random_container) + # + ## Top bar (input intensity) + #box = QHBoxLayout() + #root.addLayout(box) + # + #box.addWidget(QLabel("Input intensity")) + #GUI.input_intensity = QLineEdit("1") + #GUI.input_intensity.textChanged.connect(self.change_input_intensity) + #box.addWidget(GUI.input_intensity) + # + #hbox = QHBoxLayout() + #root.addLayout(hbox) + # + #rhs = QVBoxLayout() + # + #Widocques.image = QImage("jones.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.table = QTableWidget() + #populate_table() + # + #hbox.addWidget(GUI.scroll) + #hbox.addLayout(rhs) + + setup_menubar(self) - def I(theta): - op.angle = theta - if op.ref is not False: - op.angle += system.elements[op.ref].angle + root = QVBoxLayout() + self.setCentralWidget(LayoutWrapper(root)) - 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())) + box = QHBoxLayout() + root.addLayout(box) - if which == "min": - f = I - else: - f = lambda theta: -I(theta) + # Top bar + box.addWidget(QLabel("Input intensity")) + GUI.input_intensity = QLineEdit("%g" % system.input_intensity) + GUI.input_intensity.textChanged.connect(self.change_input_intensity) + box.addWidget(GUI.input_intensity) - 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)) + # Splitter + split = QSplitter() + root.addWidget(split) + # Table (LHS) + GUI.table = SystemTable() + split.addWidget(GUI.table) + # Splitter RHS + rhs = QVBoxLayout() + split.addWidget(LayoutWrapper(rhs)) -def change_input_intensity(): - try: - system.input_intensity = float(GUI.input_intensity.text()) - update() - except ValueError: - pass + Widocques.image = QImage("jones.jpg") + GUI.widok = Widocques() + rhs.addWidget(GUI.widok) -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 = QVBoxLayout() - root_fucking_random_container = QWidget() - root_fucking_random_container.setLayout(root) - win.setCentralWidget(root_fucking_random_container) - - # Top bar (input intensity) - box = QHBoxLayout() - root.addLayout(box) - - box.addWidget(QLabel("Input intensity")) - GUI.input_intensity = QLineEdit("1") - GUI.input_intensity.textChanged.connect(change_input_intensity) - box.addWidget(GUI.input_intensity) - - hbox = QHBoxLayout() - root.addLayout(hbox) - - rhs = QVBoxLayout() - - Widocques.image = QImage("jones.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) + GUI.table.populate(system) - hbox.addWidget(GUI.scroll) - hbox.addLayout(rhs) + #update() - setup_menubar(win) + def change_input_intensity(self): + try: + system.input_intensity = float(GUI.input_intensity.text()) + update() + except ValueError: + pass - update() - win.show()
\ No newline at end of file |