Start reworking the layout using a QTableWidget

1 files changed, 113 insertions, 290 deletions

diff --git a/src/ui.py b/src/ui.py
index f54134f..fb0932a 100644
--- a/src/ui.py
+++ b/src/ui.py
@@ -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