Refactor SimulationHandler to separate class
This commit is contained in:
SiboVG
parent
6084ba6512
commit
6490b08847
@ -0,0 +1,179 @@
|
||||
package net.sf.openrocket.file.rasaero.importt;
|
||||
|
||||
import net.sf.openrocket.aerodynamics.WarningSet;
|
||||
import net.sf.openrocket.document.Simulation;
|
||||
import net.sf.openrocket.file.DocumentLoadingContext;
|
||||
import net.sf.openrocket.file.simplesax.AbstractElementHandler;
|
||||
import net.sf.openrocket.file.simplesax.ElementHandler;
|
||||
import net.sf.openrocket.file.simplesax.PlainTextHandler;
|
||||
import net.sf.openrocket.motor.IgnitionEvent;
|
||||
import net.sf.openrocket.motor.Motor;
|
||||
import net.sf.openrocket.motor.MotorConfiguration;
|
||||
import net.sf.openrocket.motor.ThrustCurveMotor;
|
||||
import net.sf.openrocket.rocketcomponent.FlightConfigurationId;
|
||||
import net.sf.openrocket.rocketcomponent.MotorMount;
|
||||
import net.sf.openrocket.rocketcomponent.Rocket;
|
||||
import net.sf.openrocket.rocketcomponent.RocketComponent;
|
||||
import net.sf.openrocket.rocketcomponent.StageSeparationConfiguration;
|
||||
import net.sf.openrocket.simulation.SimulationOptions;
|
||||
import org.xml.sax.SAXException;
|
||||
|
||||
import java.util.HashMap;
|
||||
|
||||
/**
|
||||
* Handles RASAero simulation elements.
|
||||
* We will only import the motor information from it.
|
||||
*/
|
||||
public class SimulationHandler extends AbstractElementHandler {
|
||||
private final DocumentLoadingContext context;
|
||||
private final Rocket rocket;
|
||||
private final SimulationOptions launchSiteSettings;
|
||||
private final int simulationNr;
|
||||
|
||||
// Motor information
|
||||
private ThrustCurveMotor sustainerEngine;
|
||||
private Double sustainerIgnitionDelay;
|
||||
private ThrustCurveMotor booster1Engine;
|
||||
private Double booster1IgnitionDelay;
|
||||
private Double booster1SeparationDelay;
|
||||
private Boolean includeBooster1;
|
||||
private ThrustCurveMotor booster2Engine;
|
||||
private Double booster2SeparationDelay;
|
||||
private Boolean includeBooster2;
|
||||
|
||||
public SimulationHandler(DocumentLoadingContext context, Rocket rocket, SimulationOptions launchSiteSettings, int simulationNr) {
|
||||
this.context = context;
|
||||
this.rocket = rocket;
|
||||
this.launchSiteSettings = launchSiteSettings;
|
||||
this.simulationNr = simulationNr;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ElementHandler openElement(String element, HashMap<String, String> attributes, WarningSet warnings) throws SAXException {
|
||||
if (RASAeroCommonConstants.SUSTAINER_ENGINE.equals(element) || RASAeroCommonConstants.SUSTAINER_IGNITION_DELAY.equals(element)
|
||||
|| RASAeroCommonConstants.BOOSTER1_ENGINE.equals(element) || RASAeroCommonConstants.BOOSTER1_IGNITION_DELAY.equals(element)
|
||||
|| RASAeroCommonConstants.BOOSTER1_SEPARATION_DELAY.equals(element) || RASAeroCommonConstants.INCLUDE_BOOSTER1.equals(element)
|
||||
|| RASAeroCommonConstants.BOOSTER2_ENGINE.equals(element) || RASAeroCommonConstants.BOOSTER2_SEPARATION_DELAY.equals(element)
|
||||
|| RASAeroCommonConstants.INCLUDE_BOOSTER2.equals(element)) {
|
||||
return PlainTextHandler.INSTANCE;
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void closeElement(String element, HashMap<String, String> attributes, String content, WarningSet warnings) throws SAXException {
|
||||
if (RASAeroCommonConstants.SUSTAINER_ENGINE.equals(element)) {
|
||||
sustainerEngine = RASAeroMotorsLoader.getMotorFromRASAero(content, warnings);
|
||||
} else if (RASAeroCommonConstants.SUSTAINER_IGNITION_DELAY.equals(element)) {
|
||||
sustainerIgnitionDelay = Double.parseDouble(content);
|
||||
} else if (RASAeroCommonConstants.BOOSTER1_ENGINE.equals(element)) {
|
||||
booster1Engine = RASAeroMotorsLoader.getMotorFromRASAero(content, warnings);
|
||||
} else if (RASAeroCommonConstants.BOOSTER1_IGNITION_DELAY.equals(element)) {
|
||||
booster1IgnitionDelay = Double.parseDouble(content);
|
||||
} else if (RASAeroCommonConstants.BOOSTER1_SEPARATION_DELAY.equals(element)) {
|
||||
booster1SeparationDelay = Double.parseDouble(content);
|
||||
} else if (RASAeroCommonConstants.INCLUDE_BOOSTER1.equals(element)) {
|
||||
includeBooster1 = Boolean.parseBoolean(content);
|
||||
} else if (RASAeroCommonConstants.BOOSTER2_ENGINE.equals(element)) {
|
||||
booster2Engine = RASAeroMotorsLoader.getMotorFromRASAero(content, warnings);
|
||||
} else if (RASAeroCommonConstants.BOOSTER2_SEPARATION_DELAY.equals(element)) {
|
||||
booster2SeparationDelay = Double.parseDouble(content);
|
||||
} else if (RASAeroCommonConstants.INCLUDE_BOOSTER2.equals(element)) {
|
||||
includeBooster2 = Boolean.parseBoolean(content);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void endHandler(String element, HashMap<String, String> attributes, String content, WarningSet warnings) throws SAXException {
|
||||
FlightConfigurationId id = new FlightConfigurationId();
|
||||
rocket.createFlightConfiguration(id);
|
||||
// Select if this is the first config
|
||||
if (rocket.getSelectedConfiguration().getId().equals(FlightConfigurationId.DEFAULT_VALUE_FCID)) {
|
||||
rocket.setSelectedConfiguration(id);
|
||||
}
|
||||
|
||||
// Add motors to the rocket
|
||||
addMotorToStage(0, sustainerEngine, sustainerIgnitionDelay, id, warnings);
|
||||
if (includeBooster1) {
|
||||
addMotorToStage(1, booster1Engine, booster1IgnitionDelay, id, warnings);
|
||||
}
|
||||
if (includeBooster2) {
|
||||
addMotorToStage(2, booster2Engine, 0.0, id, warnings);
|
||||
}
|
||||
|
||||
// Set separation settings
|
||||
setSeparationDelay(0, 0.0, id);
|
||||
if (includeBooster1) {
|
||||
setSeparationDelay(1, booster1SeparationDelay, id);
|
||||
}
|
||||
if (includeBooster2) {
|
||||
setSeparationDelay(2, booster2SeparationDelay, id);
|
||||
}
|
||||
|
||||
// Add a new simulation
|
||||
Simulation sim = new Simulation(rocket);
|
||||
sim.setFlightConfigurationId(id);
|
||||
sim.setName("Simulation " + simulationNr);
|
||||
sim.copySimulationOptionsFrom(launchSiteSettings);
|
||||
context.getOpenRocketDocument().addSimulation(sim);
|
||||
}
|
||||
|
||||
private void addMotorToStage(final int stageNr, final Motor motor, final Double ignitionDelay,
|
||||
final FlightConfigurationId id, final WarningSet warnings) {
|
||||
if (motor == null || rocket.getStage(stageNr) == null) {
|
||||
return;
|
||||
}
|
||||
MotorMount mount = getMotorMountForStage(stageNr);
|
||||
if (mount == null) {
|
||||
warnings.add("No motor mount found for stage " + stageNr + ". Ignoring motor.");
|
||||
return;
|
||||
}
|
||||
MotorConfiguration motorConfig = new MotorConfiguration(mount, id);
|
||||
motorConfig.setMotor(motor);
|
||||
|
||||
// RASAero requires apogee deployment, so the sustainer motor should always be plugged
|
||||
if (stageNr == 0) {
|
||||
motorConfig.setEjectionDelay(Motor.PLUGGED_DELAY);
|
||||
}
|
||||
|
||||
|
||||
double delay = ignitionDelay != null ? ignitionDelay : 0.0;
|
||||
motorConfig.setIgnitionDelay(delay);
|
||||
if (stageNr < rocket.getStageCount() - 1) { // Use burnout non-last if multi-staged rocket
|
||||
motorConfig.setIgnitionEvent(IgnitionEvent.BURNOUT);
|
||||
} else {
|
||||
motorConfig.setIgnitionEvent(IgnitionEvent.AUTOMATIC);
|
||||
}
|
||||
mount.setMotorConfig(motorConfig, id);
|
||||
}
|
||||
|
||||
private void setSeparationDelay(final int stageNr, Double separationDelay,
|
||||
final FlightConfigurationId id) {
|
||||
if (rocket.getStage(stageNr) == null) {
|
||||
return;
|
||||
}
|
||||
StageSeparationConfiguration config = rocket.getStage(stageNr).getSeparationConfigurations().get(id);
|
||||
if (separationDelay != null) {
|
||||
config.setSeparationDelay(separationDelay);
|
||||
}
|
||||
config.setSeparationEvent(StageSeparationConfiguration.SeparationEvent.BURNOUT);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the furthest back motor mount in the stage.
|
||||
* @param stage stage number
|
||||
* @return furthest back motor mount of the stage
|
||||
*/
|
||||
private MotorMount getMotorMountForStage(int stage) {
|
||||
MotorMount mount = null;
|
||||
for (RocketComponent component : rocket.getStage(stage)) {
|
||||
if (component instanceof MotorMount) {
|
||||
mount = (MotorMount) component;
|
||||
}
|
||||
}
|
||||
if (mount != null) {
|
||||
mount.setMotorMount(true);
|
||||
}
|
||||
return mount;
|
||||
}
|
||||
}
|
||||
@ -1,20 +1,10 @@
|
||||
package net.sf.openrocket.file.rasaero.importt;
|
||||
|
||||
import net.sf.openrocket.aerodynamics.WarningSet;
|
||||
import net.sf.openrocket.document.Simulation;
|
||||
import net.sf.openrocket.file.DocumentLoadingContext;
|
||||
import net.sf.openrocket.file.simplesax.AbstractElementHandler;
|
||||
import net.sf.openrocket.file.simplesax.ElementHandler;
|
||||
import net.sf.openrocket.file.simplesax.PlainTextHandler;
|
||||
import net.sf.openrocket.motor.IgnitionEvent;
|
||||
import net.sf.openrocket.motor.Motor;
|
||||
import net.sf.openrocket.motor.MotorConfiguration;
|
||||
import net.sf.openrocket.motor.ThrustCurveMotor;
|
||||
import net.sf.openrocket.rocketcomponent.FlightConfigurationId;
|
||||
import net.sf.openrocket.rocketcomponent.MotorMount;
|
||||
import net.sf.openrocket.rocketcomponent.Rocket;
|
||||
import net.sf.openrocket.rocketcomponent.RocketComponent;
|
||||
import net.sf.openrocket.rocketcomponent.StageSeparationConfiguration;
|
||||
import net.sf.openrocket.simulation.SimulationOptions;
|
||||
import org.xml.sax.SAXException;
|
||||
|
||||
@ -52,162 +42,4 @@ public class SimulationListHandler extends AbstractElementHandler {
|
||||
public void endHandler(String element, HashMap<String, String> attributes, String content, WarningSet warnings) throws SAXException {
|
||||
RASAeroMotorsLoader.clearAllMotors();
|
||||
}
|
||||
|
||||
/**
|
||||
* Handles RASAero simulation elements.
|
||||
* We will only import the motor information from it.
|
||||
*/
|
||||
private static class SimulationHandler extends AbstractElementHandler {
|
||||
private final DocumentLoadingContext context;
|
||||
private final Rocket rocket;
|
||||
private final SimulationOptions launchSiteSettings;
|
||||
private final int simulationNr;
|
||||
|
||||
// Motor information
|
||||
private ThrustCurveMotor sustainerEngine;
|
||||
private Double sustainerIgnitionDelay;
|
||||
private ThrustCurveMotor booster1Engine;
|
||||
private Double booster1IgnitionDelay;
|
||||
private Double booster1SeparationDelay;
|
||||
private Boolean includeBooster1;
|
||||
private ThrustCurveMotor booster2Engine;
|
||||
private Double booster2SeparationDelay;
|
||||
private Boolean includeBooster2;
|
||||
|
||||
public SimulationHandler(DocumentLoadingContext context, Rocket rocket, SimulationOptions launchSiteSettings, int simulationNr) {
|
||||
this.context = context;
|
||||
this.rocket = rocket;
|
||||
this.launchSiteSettings = launchSiteSettings;
|
||||
this.simulationNr = simulationNr;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ElementHandler openElement(String element, HashMap<String, String> attributes, WarningSet warnings) throws SAXException {
|
||||
if (RASAeroCommonConstants.SUSTAINER_ENGINE.equals(element) || RASAeroCommonConstants.SUSTAINER_IGNITION_DELAY.equals(element)
|
||||
|| RASAeroCommonConstants.BOOSTER1_ENGINE.equals(element) || RASAeroCommonConstants.BOOSTER1_IGNITION_DELAY.equals(element)
|
||||
|| RASAeroCommonConstants.BOOSTER1_SEPARATION_DELAY.equals(element) || RASAeroCommonConstants.INCLUDE_BOOSTER1.equals(element)
|
||||
|| RASAeroCommonConstants.BOOSTER2_ENGINE.equals(element) || RASAeroCommonConstants.BOOSTER2_SEPARATION_DELAY.equals(element)
|
||||
|| RASAeroCommonConstants.INCLUDE_BOOSTER2.equals(element)) {
|
||||
return PlainTextHandler.INSTANCE;
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void closeElement(String element, HashMap<String, String> attributes, String content, WarningSet warnings) throws SAXException {
|
||||
if (RASAeroCommonConstants.SUSTAINER_ENGINE.equals(element)) {
|
||||
sustainerEngine = RASAeroMotorsLoader.getMotorFromRASAero(content, warnings);
|
||||
} else if (RASAeroCommonConstants.SUSTAINER_IGNITION_DELAY.equals(element)) {
|
||||
sustainerIgnitionDelay = Double.parseDouble(content);
|
||||
} else if (RASAeroCommonConstants.BOOSTER1_ENGINE.equals(element)) {
|
||||
booster1Engine = RASAeroMotorsLoader.getMotorFromRASAero(content, warnings);
|
||||
} else if (RASAeroCommonConstants.BOOSTER1_IGNITION_DELAY.equals(element)) {
|
||||
booster1IgnitionDelay = Double.parseDouble(content);
|
||||
} else if (RASAeroCommonConstants.BOOSTER1_SEPARATION_DELAY.equals(element)) {
|
||||
booster1SeparationDelay = Double.parseDouble(content);
|
||||
} else if (RASAeroCommonConstants.INCLUDE_BOOSTER1.equals(element)) {
|
||||
includeBooster1 = Boolean.parseBoolean(content);
|
||||
} else if (RASAeroCommonConstants.BOOSTER2_ENGINE.equals(element)) {
|
||||
booster2Engine = RASAeroMotorsLoader.getMotorFromRASAero(content, warnings);
|
||||
} else if (RASAeroCommonConstants.BOOSTER2_SEPARATION_DELAY.equals(element)) {
|
||||
booster2SeparationDelay = Double.parseDouble(content);
|
||||
} else if (RASAeroCommonConstants.INCLUDE_BOOSTER2.equals(element)) {
|
||||
includeBooster2 = Boolean.parseBoolean(content);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void endHandler(String element, HashMap<String, String> attributes, String content, WarningSet warnings) throws SAXException {
|
||||
FlightConfigurationId id = new FlightConfigurationId();
|
||||
rocket.createFlightConfiguration(id);
|
||||
// Select if this is the first config
|
||||
if (rocket.getSelectedConfiguration().getId().equals(FlightConfigurationId.DEFAULT_VALUE_FCID)) {
|
||||
rocket.setSelectedConfiguration(id);
|
||||
}
|
||||
|
||||
// Add motors to the rocket
|
||||
addMotorToStage(0, sustainerEngine, sustainerIgnitionDelay, id, warnings);
|
||||
if (includeBooster1) {
|
||||
addMotorToStage(1, booster1Engine, booster1IgnitionDelay, id, warnings);
|
||||
}
|
||||
if (includeBooster2) {
|
||||
addMotorToStage(2, booster2Engine, 0.0, id, warnings);
|
||||
}
|
||||
|
||||
// Set separation settings
|
||||
setSeparationDelay(0, 0.0, id);
|
||||
if (includeBooster1) {
|
||||
setSeparationDelay(1, booster1SeparationDelay, id);
|
||||
}
|
||||
if (includeBooster2) {
|
||||
setSeparationDelay(2, booster2SeparationDelay, id);
|
||||
}
|
||||
|
||||
// Add a new simulation
|
||||
Simulation sim = new Simulation(rocket);
|
||||
sim.setFlightConfigurationId(id);
|
||||
sim.setName("Simulation " + simulationNr);
|
||||
sim.copySimulationOptionsFrom(launchSiteSettings);
|
||||
context.getOpenRocketDocument().addSimulation(sim);
|
||||
}
|
||||
|
||||
private void addMotorToStage(final int stageNr, final Motor motor, final Double ignitionDelay,
|
||||
final FlightConfigurationId id, final WarningSet warnings) {
|
||||
if (motor == null || rocket.getStage(stageNr) == null) {
|
||||
return;
|
||||
}
|
||||
MotorMount mount = getMotorMountForStage(stageNr);
|
||||
if (mount == null) {
|
||||
warnings.add("No motor mount found for stage " + stageNr + ". Ignoring motor.");
|
||||
return;
|
||||
}
|
||||
MotorConfiguration motorConfig = new MotorConfiguration(mount, id);
|
||||
motorConfig.setMotor(motor);
|
||||
|
||||
// RASAero requires apogee deployment, so the sustainer motor should always be plugged
|
||||
if (stageNr == 0) {
|
||||
motorConfig.setEjectionDelay(Motor.PLUGGED_DELAY);
|
||||
}
|
||||
|
||||
|
||||
double delay = ignitionDelay != null ? ignitionDelay : 0.0;
|
||||
motorConfig.setIgnitionDelay(delay);
|
||||
if (stageNr < rocket.getStageCount() - 1) { // Use burnout non-last if multi-staged rocket
|
||||
motorConfig.setIgnitionEvent(IgnitionEvent.BURNOUT);
|
||||
} else {
|
||||
motorConfig.setIgnitionEvent(IgnitionEvent.AUTOMATIC);
|
||||
}
|
||||
mount.setMotorConfig(motorConfig, id);
|
||||
}
|
||||
|
||||
private void setSeparationDelay(final int stageNr, Double separationDelay,
|
||||
final FlightConfigurationId id) {
|
||||
if (rocket.getStage(stageNr) == null) {
|
||||
return;
|
||||
}
|
||||
StageSeparationConfiguration config = rocket.getStage(stageNr).getSeparationConfigurations().get(id);
|
||||
if (separationDelay != null) {
|
||||
config.setSeparationDelay(separationDelay);
|
||||
}
|
||||
config.setSeparationEvent(StageSeparationConfiguration.SeparationEvent.BURNOUT);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the furthest back motor mount in the stage.
|
||||
* @param stage stage number
|
||||
* @return furthest back motor mount of the stage
|
||||
*/
|
||||
private MotorMount getMotorMountForStage(int stage) {
|
||||
MotorMount mount = null;
|
||||
for (RocketComponent component : rocket.getStage(stage)) {
|
||||
if (component instanceof MotorMount) {
|
||||
mount = (MotorMount) component;
|
||||
}
|
||||
}
|
||||
if (mount != null) {
|
||||
mount.setMotorMount(true);
|
||||
}
|
||||
return mount;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user