hongshaorou/openrocket
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Move CSV listener example to extension

Browse Source
This commit is contained in:
JoePfeiffer 2022-12-07 15:10:54 -07:00
parent ba630ab37e
commit c3d2d5fc16
2 changed files with 86 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

133
core/src/net/sf/openrocket/simulation/listeners/example/CSVSaveListener.java → core/src/net/sf/openrocket/simulation/extension/example/CSVSave.java
View File

@ -1,4 +1,4 @@
package net.sf.openrocket.simulation.listeners.example; package net.sf.openrocket.simulation.extension.impl;
import java.io.File; import java.io.File;
import java.io.FileNotFoundException; import java.io.FileNotFoundException;
@ -9,12 +9,14 @@ import net.sf.openrocket.rocketcomponent.FinSet;
import net.sf.openrocket.rocketcomponent.RocketComponent; import net.sf.openrocket.rocketcomponent.RocketComponent;
import net.sf.openrocket.simulation.FlightDataType; import net.sf.openrocket.simulation.FlightDataType;
import net.sf.openrocket.simulation.FlightEvent; import net.sf.openrocket.simulation.FlightEvent;
import net.sf.openrocket.simulation.SimulationConditions;
import net.sf.openrocket.simulation.SimulationStatus; import net.sf.openrocket.simulation.SimulationStatus;
import net.sf.openrocket.simulation.exception.SimulationException; import net.sf.openrocket.simulation.exception.SimulationException;
import net.sf.openrocket.simulation.extension.AbstractSimulationExtension;
import net.sf.openrocket.simulation.listeners.AbstractSimulationListener; import net.sf.openrocket.simulation.listeners.AbstractSimulationListener;
public class CSVSaveListener extends AbstractSimulationListener { public class CSVSave extends AbstractSimulationExtension {
private static enum Types { private static enum Types {
TIME { TIME {
@ -218,78 +220,89 @@ public class CSVSaveListener extends AbstractSimulationListener {
private File file; private File file;
private PrintStream output = null; private PrintStream output = null;
@Override
public String getDescription() {
return "dump a CSV file with a predetermined set of flight variables to a CSV file while running";
}
@Override @Override
public boolean handleFlightEvent(SimulationStatus status, FlightEvent event) throws SimulationException { public void initialize(SimulationConditions conditions) throws SimulationException {
conditions.getSimulationListenerList().add(new CSVSaveListener());
}
private class CSVSaveListener extends AbstractSimulationListener {
if (event.getType() == FlightEvent.Type.LAUNCH) { @Override
int n = 1; public boolean handleFlightEvent(SimulationStatus status, FlightEvent event) throws SimulationException {
if (output != null) { if (event.getType() == FlightEvent.Type.LAUNCH) {
System.err.println("WARNING: Ending simulation logging to CSV file " + int n = 1;
"(SIMULATION_END not encountered).");
if (output != null) {
System.err.println("WARNING: Ending simulation logging to CSV file " +
"(SIMULATION_END not encountered).");
output.close();
output = null;
}
do {
file = new File(String.format(FILENAME_FORMAT, n));
n++;
} while (file.exists());
System.err.println("Opening file " + file + " for CSV output.");
try {
output = new PrintStream(file);
} catch (FileNotFoundException e) {
System.err.println("ERROR OPENING FILE: " + e);
}
final Types[] types = Types.values();
StringBuilder s = new StringBuilder("# " + types[0].toString());
for (int i = 1; i < types.length; i++) {
s.append("," + types[i].toString());
}
output.println(s);
} else if (event.getType() == FlightEvent.Type.SIMULATION_END && output != null) {
System.err.println("Ending simulation logging to CSV file: " + file);
output.close(); output.close();
output = null; output = null;
} else if (event.getType() != FlightEvent.Type.ALTITUDE) {
if (output != null) {
output.println("# Event " + event);
} else {
System.err.println("WARNING: Event " + event + " encountered without open file");
}
} }
do { return true;
file = new File(String.format(FILENAME_FORMAT, n)); }
n++;
} while (file.exists()); @Override
public void postStep(SimulationStatus status) throws SimulationException {
System.err.println("Opening file " + file + " for CSV output.");
try {
output = new PrintStream(file);
} catch (FileNotFoundException e) {
System.err.println("ERROR OPENING FILE: " + e);
}
final Types[] types = Types.values(); final Types[] types = Types.values();
StringBuilder s = new StringBuilder("# " + types[0].toString()); StringBuilder s;
for (int i = 1; i < types.length; i++) {
s.append("," + types[i].toString());
}
output.println(s);
} else if (event.getType() == FlightEvent.Type.SIMULATION_END && output != null) {
System.err.println("Ending simulation logging to CSV file: " + file);
output.close();
output = null;
} else if (event.getType() != FlightEvent.Type.ALTITUDE) {
if (output != null) { if (output != null) {
output.println("# Event " + event);
s = new StringBuilder("" + types[0].getValue(status));
for (int i = 1; i < types.length; i++) {
s.append("," + types[i].getValue(status));
}
output.println(s);
} else { } else {
System.err.println("WARNING: Event " + event + " encountered without open file");
System.err.println("WARNING: stepTaken called with no open file " +
"(t=" + status.getSimulationTime() + ")");
} }
} }
return true;
}
@Override
public void postStep(SimulationStatus status) throws SimulationException {
final Types[] types = Types.values();
StringBuilder s;
if (output != null) {
s = new StringBuilder("" + types[0].getValue(status));
for (int i = 1; i < types.length; i++) {
s.append("," + types[i].getValue(status));
}
output.println(s);
} else {
System.err.println("WARNING: stepTaken called with no open file " +
"(t=" + status.getSimulationTime() + ")");
}
} }
} }

13
core/src/net/sf/openrocket/simulation/extension/example/CSVSaveProvider.java Normal file
View File

@ -0,0 +1,13 @@
package net.sf.openrocket.simulation.extension.impl;
import net.sf.openrocket.plugin.Plugin;
import net.sf.openrocket.simulation.extension.AbstractSimulationExtensionProvider;
@Plugin
public class CSVSaveProvider extends AbstractSimulationExtensionProvider {
public CSVSaveProvider() {
super(CSVSave.class, "Reports", "CSV Save");
}
}