90 lines
3.5 KiB
Python
90 lines
3.5 KiB
Python
# functions.py
|
|
|
|
import yaml
|
|
from PIL import Image, ImageTk
|
|
from tkinter import ttk, filedialog, messagebox
|
|
import tkinter as tk
|
|
import os
|
|
|
|
# Function to load object definitions from YAML
|
|
|
|
# Load a specific parameter set from a YAML file for a group
|
|
def load_yaml(group_name, param_name, folder_parsets):
|
|
yml_file = os.path.join(folder_parsets, f"{group_name}.yml")
|
|
if os.path.exists(yml_file):
|
|
with open(yml_file, 'r') as file:
|
|
data = yaml.safe_load(file)
|
|
return data.get(param_name, {})
|
|
return {}
|
|
|
|
# Save a parameter set to a YAML file for a group
|
|
def save_yaml(group_name, param_name, param_data, folder_parsets):
|
|
yml_file = os.path.join(folder_parsets, f"{group_name}.yml")
|
|
all_data = load_existing_data(yml_file)
|
|
all_data[param_name] = param_data
|
|
with open(yml_file, 'w') as file:
|
|
yaml.dump(all_data, file)
|
|
# messagebox.showinfo("Success", f"Parameter set '{param_name}' saved to {group_name}.yml")
|
|
|
|
def delete_yaml(group_name, param_name, folder_parsets):
|
|
yml_file = os.path.join(folder_parsets, f"{group_name}.yml")
|
|
if os.path.exists(yml_file):
|
|
with open(yml_file, 'r') as file:
|
|
data = yaml.safe_load(file)
|
|
if param_name in data:
|
|
del data[param_name] # Remove the specific parameter set
|
|
with open(yml_file, 'w') as file:
|
|
yaml.safe_dump(data, file)
|
|
|
|
# Load existing data from a YAML file
|
|
def load_existing_data(yml_file):
|
|
if os.path.exists(yml_file):
|
|
with open(yml_file, 'r') as file:
|
|
return yaml.safe_load(file) or {}
|
|
return {}
|
|
|
|
# Fetch existing parameter sets from a group-specific YAML file
|
|
def get_existing_parameters(group_name, folder_parsets):
|
|
yml_file = os.path.join(folder_parsets, f"{group_name}.yml")
|
|
data = load_existing_data(yml_file)
|
|
return list(data.keys())
|
|
|
|
# Load the image for the UI
|
|
def load_image(image_file, sizelimit=200):
|
|
image = Image.open(image_file)
|
|
w, h = image.size
|
|
if w > h:
|
|
new_w, new_h = sizelimit, int(sizelimit * h / w)
|
|
else:
|
|
new_h, new_w = sizelimit, int(sizelimit * w / h)
|
|
return ImageTk.PhotoImage(image.resize((new_w, new_h), Image.Resampling.LANCZOS))
|
|
|
|
# Create a new parameter set and save it
|
|
def create_new_parameter_set(group_name, param_name, input_vars, folder_parsets):
|
|
if not param_name:
|
|
messagebox.showerror("Error", "Parameter set name cannot be empty.")
|
|
return
|
|
param_data = {key: var.get() for key, var in input_vars[group_name].items()}
|
|
save_yaml(group_name, param_name, param_data, folder_parsets)
|
|
|
|
# Update the dropdown after saving new parameter set
|
|
def update_dropdown(param_values, param_name, param_dropdown):
|
|
param_values.insert(-1, param_name)
|
|
param_dropdown['values'] = param_values
|
|
|
|
# Create a popup window for creating a new parameter set
|
|
def create_new_param_popup(root, group_name, input_vars, param_values, param_dropdown, folder_parsets):
|
|
new_window = tk.Toplevel(root)
|
|
new_window.title(f"Create New Parameter Set for {group_name}")
|
|
tk.Label(new_window, text="Parameter Set Name:").pack(pady=10)
|
|
param_name_entry = tk.Entry(new_window)
|
|
param_name_entry.pack(pady=10)
|
|
|
|
def create_param_set():
|
|
param_name = param_name_entry.get()
|
|
create_new_parameter_set(group_name, param_name, input_vars, folder_parsets)
|
|
update_dropdown(param_values, param_name, param_dropdown)
|
|
new_window.destroy()
|
|
|
|
tk.Button(new_window, text="Create", command=create_param_set).pack(pady=10)
|