forked from zietzm/Helmholtz_Test_Bench
Martin Zietz
137 lines
4.3 KiB
Python
137 lines
4.3 KiB
Python
from tkinter import *
|
|
from tkinter import ttk
|
|
import settings as g
|
|
import cage_func as func
|
|
import random as rand
|
|
|
|
NORM_FONT = ()
|
|
SUB_HEADER_FONT = ("Arial", 9, "bold")
|
|
|
|
|
|
class HelmholtzGUI(Tk):
|
|
|
|
def __init__(self):
|
|
Tk.__init__(self)
|
|
|
|
Tk.wm_title(self, "Helmholtz Cage Control")
|
|
Tk.wm_iconbitmap(self, "Helmholtz.ico")
|
|
|
|
self.Menu = TopMenu(self) # displays menu bar at the top
|
|
|
|
mainArea = Frame(self)
|
|
mainArea.pack(side="top", fill="both", expand=False)
|
|
|
|
mainArea.grid_rowconfigure(0, weight=1)
|
|
mainArea.grid_columnconfigure(0, weight=1)
|
|
|
|
self.frames = {} # dictionary for storing all pages
|
|
|
|
for F in (TestFrame, StatusDisplay):
|
|
frame = F(mainArea, self)
|
|
self.frames[F] = frame
|
|
frame.grid(row=0, column=0, sticky="nsew")
|
|
|
|
self.show_frame(StatusDisplay)
|
|
|
|
def show_frame(self, key):
|
|
frame = self.frames[key] # gets correct page from the dictionary
|
|
frame.tkraise() # brings this frame to the front
|
|
|
|
|
|
class TopMenu:
|
|
|
|
def __init__(self, window):
|
|
print("menu called")
|
|
menu = Menu(window)
|
|
window.config(menu=menu)
|
|
|
|
ModeSelector = Menu(menu)
|
|
menu.add_cascade(label="Mode", menu=ModeSelector)
|
|
ModeSelector.add_command(label="Full Manual", command=self.manual_mode)
|
|
|
|
def manual_mode(self):
|
|
print("Switching to manual mode")
|
|
|
|
|
|
class TestFrame(Frame):
|
|
|
|
def __init__(self, parent, controller):
|
|
Frame.__init__(self, parent)
|
|
print("TestFrame called")
|
|
|
|
one = Label(self, text="One", bg="red")
|
|
one.pack(fill=X)
|
|
two = Label(self, text="Two", bg="blue")
|
|
two.pack()
|
|
button = ttk.Button(self, text="Print stuff", command=lambda: print_stuff("Hello"))
|
|
button.pack()
|
|
|
|
|
|
class StatusDisplay(Frame):
|
|
|
|
def __init__(self, parent, controller):
|
|
Frame.__init__(self, parent)
|
|
print("StatusDisplay called") # ToDo: remove
|
|
rowCounter = 0 # keep track of which row we are at in the grid layout
|
|
|
|
col = 0
|
|
for header in ["X-Axis", "Y-Axis", "Z-Axis"]: # create Column headers
|
|
headLabel = Label(self, text=header, font=SUB_HEADER_FONT)
|
|
headLabel.grid(columnspan=2, row=rowCounter, column=col*2, sticky="ew")
|
|
col = col + 1 # move to next column
|
|
rowCounter = rowCounter + 1 # increase row counter to place future stuff below header
|
|
|
|
TextLabels = ["Port:", "Channel:", "Output:"] # define content of row entries
|
|
self.rowNo = len(TextLabels) # get number of label rows
|
|
|
|
self.columnNo = 6 # number of label columns
|
|
# prepare list of lists to contain all labels for row entries in all columns:
|
|
self.Labels = [[] for _ in range(self.columnNo)]
|
|
|
|
self.label_dict = {}
|
|
for name in TextLabels:
|
|
self.label_dict[name] = [StringVar() for _ in range(int(self.columnNo/2))]
|
|
for col in range(int(self.columnNo/2)):
|
|
self.Labels[col*2].append(Label(self, text=name))
|
|
self.Labels[col*2+1].append(Label(self, textvariable=self.label_dict[name][col]))
|
|
|
|
col = 0
|
|
for LabelCol in self.Labels: # place row entries in grid layout for all columns
|
|
for row in range(self.rowNo): # place row entries
|
|
LabelCol[row].grid(row=row+rowCounter, column=col, sticky="w")
|
|
col = col + 1
|
|
rowCounter = rowCounter + self.rowNo # increase row counter to place future stuff below this
|
|
|
|
self.update_labels(controller)
|
|
|
|
def update_labels(self, controller):
|
|
i = 0
|
|
for axis in g.AXES: # ToDo: switch to proper axes when PSU connected
|
|
if axis.device is not None:
|
|
axis.update_values()
|
|
self.label_dict["Port:"][i].set(g.ports[i])
|
|
self.label_dict["Channel:"][i].set(axis.channel)
|
|
self.label_dict["Output:"][i].set(axis.output_active)
|
|
i = i+1
|
|
controller.after(2000, lambda: self.update_labels(controller))
|
|
|
|
|
|
def print_stuff(stuff):
|
|
print(stuff)
|
|
|
|
|
|
def random_no():
|
|
return rand.uniform(0, 20)
|
|
|
|
|
|
class TestValues:
|
|
def __init__(self):
|
|
self.val1 = 0
|
|
self.val2 = 0
|
|
self.val3 = 0
|
|
|
|
def update_values(self):
|
|
self.val1 = rand.uniform(0, 20)
|
|
self.val2 = rand.uniform(0, 20)
|
|
self.val3 = rand.uniform(0, 20)
|