#!/usr/bin/env python3
"""Rebuild the idealized AEPS-to-spacecraft example from NASA's paired test row."""
import csv, math
from pathlib import Path
HERE = Path(__file__).resolve().parent
with (HERE / "aeps-ppe-test-data.csv").open(newline="", encoding="utf-8") as f:
row = next(r for r in csv.DictReader(f) if r["point_id"] == "8")
THRUST_N = float(row["thrust_mN"]) / 1000
ISP_S = float(row["specific_impulse_s"])
VE = ISP_S * 9.80665
MDOT = THRUST_N / VE
M0 = 1000.0
DAYS = (0, 1, 3, 7, 14, 21, 30)
with (HERE / "aeps-measured-mission-example.csv").open("w", newline="", encoding="utf-8") as f:
w = csv.writer(f)
w.writerow(["elapsed_days", "ideal_delta_v_m_s", "propellant_used_kg", "spacecraft_mass_kg", "assumed_constant_thrust_mN", "assumed_constant_specific_impulse_s", "assumed_discharge_power_kW", "initial_mass_kg", "evidence_status"])
for day in DAYS:
used = MDOT * day * 86400
mass = M0 - used
dv = VE * math.log(M0 / mass)
w.writerow([day, round(dv, 3), round(used, 4), round(mass, 4), row["thrust_mN"], ISP_S, float(row["discharge_power_W"]) / 1000, M0, "calculated from NASA lab test point; hypothetical 1000 kg spacecraft; idealized"])
points=[]
for day in range(31):
dv=VE*math.log(M0/(M0-MDOT*day*86400))
points.append((90+day*25, 330-dv*.115, day, dv))
poly=" ".join(f"{x},{y:.2f}" for x,y,_,_ in points)
marks=[]
for x,y,day,dv in points:
if day in (0,1,7,14,21,30):
marks.append(f'{dv:.0f}')
svg=f''''''
(HERE / "aeps-measured-mission-example.svg").write_text(svg, encoding="utf-8")
mobile_points = []
for day in range(31):
dv = VE * math.log(M0 / (M0 - MDOT * day * 86400))
mobile_points.append((55 + day * (310 / 30), 235 - dv * .09, day, dv))
mobile_poly = " ".join(f"{x:.1f},{y:.2f}" for x, y, _, _ in mobile_points)
mobile_marks = "".join(f'' for x, y, day, _ in mobile_points if day in (0,10,20,30))
mobile_svg = f''''''
(HERE / "aeps-measured-mission-example-mobile.svg").write_text(mobile_svg, encoding="utf-8")
print("Rebuilt AEPS idealized mission CSV and SVG from NASA test row 8.")