import com.google.inject.AbstractModule;
import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.Module;
import com.google.inject.util.Modules;
import info.openrocket.core.document.Simulation;
import info.openrocket.core.l10n.DebugTranslator;
import info.openrocket.core.l10n.Translator;
import info.openrocket.core.models.wind.PinkNoiseWindModel;
import info.openrocket.core.motor.Manufacturer;
import info.openrocket.core.motor.Motor;
import info.openrocket.core.motor.MotorConfiguration;
import info.openrocket.core.motor.ThrustCurveMotor;
import info.openrocket.core.plugin.PluginModule;
import info.openrocket.core.rocketcomponent.FlightConfiguration;
import info.openrocket.core.rocketcomponent.InnerTube;
import info.openrocket.core.rocketcomponent.Rocket;
import info.openrocket.core.rocketcomponent.RocketComponent;
import info.openrocket.core.simulation.FlightData;
import info.openrocket.core.simulation.SimulationOptions;
import info.openrocket.core.startup.Application;
import info.openrocket.core.startup.CoreModule;
import info.openrocket.core.util.Coordinate;
import info.openrocket.core.util.TestRockets;

/**
 * Flies OpenRocket's own Estes Alpha III fixture on its A8 and D21 test
 * motors with zero wind and prints both apogees plus the component masses,
 * so the Rust flight-parity tests can pin them. See README.md in this
 * directory.
 */
public class Apogee {

    private static Motor motor(String designation, double thrustPeak, double m0, double m1, double m2) {
        return new ThrustCurveMotor.Builder()
                .setManufacturer(Manufacturer.getManufacturer("Test"))
                .setDesignation(designation)
                .setDescription("").setCaseInfo("SU 18.0x70.0")
                .setMotorType(Motor.Type.SINGLE)
                .setStandardDelays(new double[] {})
                .setDiameter(0.018).setLength(0.070)
                .setTimePoints(new double[] { 0, 1, 2 })
                .setThrustPoints(new double[] { 0, thrustPeak, 0 })
                .setCGPoints(new Coordinate[] {
                        new Coordinate(0.035, 0, 0, m0),
                        new Coordinate(0.035, 0, 0, m1),
                        new Coordinate(0.035, 0, 0, m2) })
                .setDigest("test")
                .build();
    }

    private static Rocket preparedRocket() {
        Rocket rocket = TestRockets.makeEstesAlphaIII();
        java.util.List<RocketComponent> toRemove = new java.util.ArrayList<>();
        for (RocketComponent c : rocket) {
            if (c instanceof info.openrocket.core.rocketcomponent.TrapezoidFinSet) {
                // Our Barrowman modeler assumes streamlined fins; OpenRocket's
                // default SQUARE section adds ~(t/MAC)·A_fins/A_ref ≈ 1.0 of
                // pressure drag, which would compare nothing like-for-like.
                ((info.openrocket.core.rocketcomponent.TrapezoidFinSet) c)
                        .setCrossSection(info.openrocket.core.rocketcomponent.FinSet.CrossSection.ROUNDED);
            }
            if (c instanceof info.openrocket.core.rocketcomponent.ExternalComponent) {
                // SMOOTH = ~20 µm paint in OpenRocket's finish table,
                // matching our modeler's default roughness.
                ((info.openrocket.core.rocketcomponent.ExternalComponent) c)
                        .setFinish(info.openrocket.core.rocketcomponent.ExternalComponent.Finish.SMOOTH);
            }
            if (c instanceof info.openrocket.core.rocketcomponent.LaunchLug) {
                // Our geometry modeler has no launch lug; remove it from the
                // fixture so both sides model the same airframe.
                toRemove.add(c);
            }
        }
        for (RocketComponent c : toRemove) {
            c.getParent().removeChild(c);
        }
        return rocket;
    }

    private static double fly(Rocket rocket, Motor motor) throws Exception {
        InnerTube inner = null;
        for (RocketComponent c : rocket) {
            if (c instanceof InnerTube) {
                inner = (InnerTube) c;
            }
        }
        MotorConfiguration mc = new MotorConfiguration(inner, TestRockets.TEST_FCID_0);
        mc.setMotor(motor);
        mc.setEjectionDelay(1000.0); // chute never fires: apogee-only comparison
        inner.setMotorConfig(mc, TestRockets.TEST_FCID_0);

        Simulation sim = new Simulation(rocket);
        sim.setFlightConfigurationId(TestRockets.TEST_FCID_0);
        SimulationOptions opts = sim.getOptions();
        opts.setISAAtmosphere(true);
        opts.setTimeStep(0.05);
        PinkNoiseWindModel wind = opts.getAverageWindModel();
        wind.setAverage(0.0);
        wind.setStandardDeviation(0.0);
        sim.simulate();
        FlightData data = sim.getSimulatedData();
        return data.getMaxAltitude();
    }

    public static void main(String[] args) throws Exception {
        // Bootstrap the same way the unit tests do, minus JUnit.
        Module debugTranslator = new AbstractModule() {
            @Override
            protected void configure() {
                bind(Translator.class).toInstance(new DebugTranslator(null));
            }
        };
        Injector injector = Guice.createInjector(
                Modules.override(new CoreModule()).with(debugTranslator), new PluginModule());
        Application.setInjector(injector);

        Rocket rocket = preparedRocket();
        for (RocketComponent c : rocket) {
            double m = c.getComponentMass();
            if (m > 0.0) {
                System.out.printf("MASS %-24s %8.5f cg=%.4f%n",
                        c.getComponentName(), m, c.getComponentCG().x);
            }
        }

        // A8: 9 N-s triangle, 3.3 g propellant, 13.1 g casing. Subsonic
        // (Mach < 0.45) — pins the base drag model.
        double a8 = fly(preparedRocket(), motor("A8", 9.0, 0.0164, 0.0145, 0.0131));
        System.out.printf("APOGEE_A8 %.4f%n", a8);

        // D21: 32 N-s triangle, 9.6 g propellant, 15.4 g casing. Reaches
        // Mach 0.85 — pins the transonic pressure rise.
        double d21 = fly(preparedRocket(), motor("D21", 32.0, 0.025, 0.020, 0.0154));
        System.out.printf("APOGEE_D21 %.4f%n", d21);
    }
}