# target_simulator/gui/waypoint_editor_window.py """ A modal dialog window for creating or editing a single Waypoint. """ import tkinter as tk from tkinter import ttk, messagebox from typing import Optional from target_simulator.core.models import Waypoint, ManeuverType, TurnDirection, KNOTS_TO_FPS, FPS_TO_KNOTS class WaypointEditorWindow(tk.Toplevel): """A dialog for entering and editing data for a single waypoint.""" def __init__(self, master, is_first_waypoint: bool = False, waypoint_to_edit: Optional[Waypoint] = None): super().__init__(master) self.transient(master) self.grab_set() self.title("Edit Waypoint" if waypoint_to_edit else "Add Waypoint") if is_first_waypoint: self.title("Set Initial Position") self.resizable(False, False) self.is_first_waypoint = is_first_waypoint self.result_waypoint: Optional[Waypoint] = None self._setup_variables() self._create_widgets() if waypoint_to_edit: self._load_waypoint_data(waypoint_to_edit) self._on_maneuver_type_change() self._center_window() self.protocol("WM_DELETE_WINDOW", self._on_cancel) self.wait_window(self) def _setup_variables(self): """Initializes all tk Variables.""" self.maneuver_type_var = tk.StringVar() # --- Fly to Point --- self.ftp_duration_var = tk.DoubleVar(value=10.0) self.ftp_range_var = tk.DoubleVar(value=20.0) self.ftp_az_var = tk.DoubleVar(value=0.0) self.ftp_alt_var = tk.DoubleVar(value=10000.0) self.ftp_vel_var = tk.DoubleVar(value=300.0) self.ftp_hdg_var = tk.DoubleVar(value=180.0) # --- Fly for Duration --- self.ffd_duration_var = tk.DoubleVar(value=10.0) self.ffd_vel_var = tk.DoubleVar(value=300.0) self.ffd_hdg_var = tk.DoubleVar(value=90.0) self.ffd_alt_var = tk.DoubleVar(value=10000.0) # --- Dynamic Maneuver --- self.dm_duration_var = tk.DoubleVar(value=10.0) self.dm_lon_accel_g_var = tk.DoubleVar(value=0.0) self.dm_lat_accel_g_var = tk.DoubleVar(value=0.0) self.dm_ver_accel_g_var = tk.DoubleVar(value=0.0) self.dm_turn_dir_var = tk.StringVar(value=TurnDirection.RIGHT.value) def _load_waypoint_data(self, wp: Waypoint): """Populates the fields with data from an existing waypoint.""" self.maneuver_type_var.set(wp.maneuver_type.value) if wp.maneuver_type == ManeuverType.FLY_TO_POINT: self.ftp_duration_var.set(wp.duration_s or 10.0) self.ftp_range_var.set(wp.target_range_nm or 0.0) self.ftp_az_var.set(wp.target_azimuth_deg or 0.0) self.ftp_alt_var.set(wp.target_altitude_ft or 10000.0) self.ftp_vel_var.set((wp.target_velocity_fps or 0.0) * FPS_TO_KNOTS) self.ftp_hdg_var.set(wp.target_heading_deg or 0.0) elif wp.maneuver_type == ManeuverType.FLY_FOR_DURATION: self.ffd_duration_var.set(wp.duration_s or 10.0) self.ffd_vel_var.set((wp.target_velocity_fps or 0.0) * FPS_TO_KNOTS) self.ffd_hdg_var.set(wp.target_heading_deg or 0.0) self.ffd_alt_var.set(wp.target_altitude_ft or 10000.0) elif wp.maneuver_type == ManeuverType.DYNAMIC_MANEUVER: self.dm_duration_var.set(wp.duration_s or 10.0) self.dm_lon_accel_g_var.set(wp.longitudinal_acceleration_g or 0.0) self.dm_lat_accel_g_var.set(wp.lateral_acceleration_g or 0.0) self.dm_ver_accel_g_var.set(wp.vertical_acceleration_g or 0.0) if wp.turn_direction: self.dm_turn_dir_var.set(wp.turn_direction.value) def _center_window(self): self.update_idletasks() parent = self.master x = parent.winfo_x() + (parent.winfo_width() // 2) - (self.winfo_width() // 2) y = parent.winfo_y() + (parent.winfo_height() // 2) - (self.winfo_height() // 2) self.geometry(f"+{x}+{y}") def _create_widgets(self): main_frame = ttk.Frame(self, padding=10) main_frame.pack(fill=tk.BOTH, expand=True) type_frame = ttk.Frame(main_frame) type_frame.pack(fill=tk.X, pady=(0, 10)) ttk.Label(type_frame, text="Maneuver Type:").pack(side=tk.LEFT) self.maneuver_combo = ttk.Combobox(type_frame, textvariable=self.maneuver_type_var, values=[m.value for m in ManeuverType], state="readonly") self.maneuver_combo.pack(side=tk.LEFT, expand=True, fill=tk.X, padx=5) self.maneuver_combo.bind("<>", self._on_maneuver_type_change) self.fly_to_point_frame = self._create_fly_to_point_frame(main_frame) self.fly_for_duration_frame = self._create_fly_for_duration_frame(main_frame) self.dynamic_maneuver_frame = self._create_dynamic_maneuver_frame(main_frame) button_frame = ttk.Frame(main_frame) button_frame.pack(fill=tk.X, pady=(15, 0), side=tk.BOTTOM) ttk.Button(button_frame, text="OK", command=self._on_ok).pack(side=tk.RIGHT, padx=5) ttk.Button(button_frame, text="Cancel", command=self._on_cancel).pack(side=tk.RIGHT) if self.is_first_waypoint: self.maneuver_type_var.set(ManeuverType.FLY_TO_POINT.value) self.maneuver_combo.config(state=tk.DISABLED) else: self.maneuver_type_var.set(ManeuverType.FLY_FOR_DURATION.value) def _create_fly_to_point_frame(self, parent) -> ttk.Frame: frame = ttk.LabelFrame(parent, text="Position and State", padding=10) frame.columnconfigure(1, weight=1) ttk.Label(frame, text="Duration to Point (s):").grid(row=0, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0.1, to=3600 * 24, textvariable=self.ftp_duration_var, width=10).grid(row=0, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Target Range (NM):").grid(row=1, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0, to=1000, textvariable=self.ftp_range_var, width=10).grid(row=1, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Target Azimuth (°):").grid(row=2, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=-180, to=180, textvariable=self.ftp_az_var, width=10).grid(row=2, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Target Altitude (ft):").grid(row=3, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=-1000, to=80000, textvariable=self.ftp_alt_var, width=10).grid(row=3, column=1, sticky=tk.EW, pady=2) if self.is_first_waypoint: ttk.Label(frame, text="Initial Velocity (kn):").grid(row=4, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0, to=2000, textvariable=self.ftp_vel_var, width=10).grid(row=4, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Initial Heading (°):").grid(row=5, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0, to=359.9, format="%.1f", increment=0.1, textvariable=self.ftp_hdg_var, width=10).grid(row=5, column=1, sticky=tk.EW, pady=2) return frame def _create_fly_for_duration_frame(self, parent) -> ttk.Frame: frame = ttk.LabelFrame(parent, text="Constant State Flight", padding=10) frame.columnconfigure(1, weight=1) ttk.Label(frame, text="Duration (s):").grid(row=0, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0.1, to=3600*24, textvariable=self.ffd_duration_var, width=10).grid(row=0, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Constant Velocity (kn):").grid(row=1, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0, to=2000, textvariable=self.ffd_vel_var, width=10).grid(row=1, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Constant Heading (°):").grid(row=2, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0, to=359.9, format="%.1f", increment=0.1, textvariable=self.ffd_hdg_var, width=10).grid(row=2, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Constant Altitude (ft):").grid(row=3, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=-1000, to=80000, textvariable=self.ffd_alt_var, width=10).grid(row=3, column=1, sticky=tk.EW, pady=2) return frame def _create_dynamic_maneuver_frame(self, parent) -> ttk.Frame: frame = ttk.LabelFrame(parent, text="Applied Accelerations", padding=10) frame.columnconfigure(1, weight=1) ttk.Label(frame, text="Duration (s):").grid(row=0, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0.1, to=3600 * 24, textvariable=self.dm_duration_var, width=10).grid(row=0, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Longitudinal Accel (g):").grid(row=1, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=-5, to=10, format="%.2f", increment=0.1, textvariable=self.dm_lon_accel_g_var, width=10).grid(row=1, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Lateral Accel (g):").grid(row=2, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=0, to=9, format="%.2f", increment=0.1, textvariable=self.dm_lat_accel_g_var, width=10).grid(row=2, column=1, sticky=tk.EW, pady=2) ttk.Label(frame, text="Turn Direction:").grid(row=3, column=0, sticky=tk.W, pady=2) turn_dir_combo = ttk.Combobox(frame, textvariable=self.dm_turn_dir_var, values=[d.value for d in TurnDirection], state="readonly", width=8) turn_dir_combo.grid(row=3, column=1, sticky=tk.W, pady=2) ttk.Label(frame, text="Vertical Accel (g):").grid(row=4, column=0, sticky=tk.W, pady=2) ttk.Spinbox(frame, from_=-5, to=5, format="%.2f", increment=0.1, textvariable=self.dm_ver_accel_g_var, width=10).grid(row=4, column=1, sticky=tk.EW, pady=2) return frame def _on_maneuver_type_change(self, event=None): try: m_type = ManeuverType(self.maneuver_type_var.get()) except ValueError: return self.fly_to_point_frame.pack_forget() self.fly_for_duration_frame.pack_forget() self.dynamic_maneuver_frame.pack_forget() if m_type == ManeuverType.FLY_TO_POINT: self.fly_to_point_frame.pack(fill=tk.BOTH, expand=True) elif m_type == ManeuverType.FLY_FOR_DURATION: self.fly_for_duration_frame.pack(fill=tk.BOTH, expand=True) elif m_type == ManeuverType.DYNAMIC_MANEUVER: self.dynamic_maneuver_frame.pack(fill=tk.BOTH, expand=True) def _on_ok(self): try: m_type = ManeuverType(self.maneuver_type_var.get()) wp = Waypoint(maneuver_type=m_type) if m_type == ManeuverType.FLY_TO_POINT: wp.duration_s = self.ftp_duration_var.get() wp.target_range_nm = self.ftp_range_var.get() wp.target_azimuth_deg = self.ftp_az_var.get() wp.target_altitude_ft = self.ftp_alt_var.get() if self.is_first_waypoint: wp.target_velocity_fps = self.ftp_vel_var.get() * KNOTS_TO_FPS wp.target_heading_deg = self.ftp_hdg_var.get() elif m_type == ManeuverType.FLY_FOR_DURATION: wp.duration_s = self.ffd_duration_var.get() wp.target_velocity_fps = self.ffd_vel_var.get() * KNOTS_TO_FPS wp.target_heading_deg = self.ffd_hdg_var.get() wp.target_altitude_ft = self.ffd_alt_var.get() elif m_type == ManeuverType.DYNAMIC_MANEUVER: wp.duration_s = self.dm_duration_var.get() wp.longitudinal_acceleration_g = self.dm_lon_accel_g_var.get() wp.lateral_acceleration_g = self.dm_lat_accel_g_var.get() wp.vertical_acceleration_g = self.dm_ver_accel_g_var.get() wp.turn_direction = TurnDirection(self.dm_turn_dir_var.get()) self.result_waypoint = wp self.destroy() except (ValueError, tk.TclError) as e: messagebox.showerror("Invalid Input", f"Please enter valid numbers.\nError: {e}", parent=self) def _on_cancel(self): self.result_waypoint = None self.destroy()