version 1.0.0

[debian/openrocket] / src / net / sf / openrocket / utils / GraphicalMotorSelector.java

diff --git a/src/net/sf/openrocket/utils/GraphicalMotorSelector.java b/src/net/sf/openrocket/utils/GraphicalMotorSelector.java
new file mode 100644 (file)
index 0000000..3e3d352
--- /dev/null
+++ b/src/net/sf/openrocket/utils/GraphicalMotorSelector.java
@@ -0,0 +1,143 @@
+package net.sf.openrocket.utils;
+
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+
+import net.sf.openrocket.file.GeneralMotorLoader;
+import net.sf.openrocket.motor.Motor;
+import net.sf.openrocket.util.Pair;
+
+public class GraphicalMotorSelector {
+       
+       public static void main(String[] args) throws IOException {
+
+               if (args.length == 0) {
+                       System.err.println("MotorPlot <files>");
+                       System.exit(1);
+               }
+               
+               // Load files
+               Map<String, List<Pair<String, Motor>>> map = 
+                       new LinkedHashMap<String, List<Pair<String, Motor>>>(); 
+               
+               GeneralMotorLoader loader = new GeneralMotorLoader();
+               for (String file: args) {
+
+                       for (Motor m: loader.load(new FileInputStream(file), file)) {
+                               System.out.println("Loaded " + m + " from file "+file);
+                               
+                               Pair<String, Motor> pair = new Pair<String, Motor>(file, m);
+                               String key = m.getManufacturer() + ":" + m.getDesignation();
+                               
+                               List<Pair<String, Motor>> list = map.get(key);
+                               if (list == null) {
+                                       list = new ArrayList<Pair<String, Motor>>();
+                                       map.put(key, list);
+                               }
+                               
+                               list.add(pair);
+                       }
+               }
+               
+               
+               // Go through different motors
+               int count = 0;
+               for (String key: map.keySet()) {
+                       count++;
+                       List<Pair<String, Motor>> list = map.get(key);
+                       
+                       
+                       // Select best one of identical motors
+                       List<String> filenames = new ArrayList<String>();
+                       List<Motor> motors = new ArrayList<Motor>();
+                       for (Pair<String, Motor> pair: list) {
+                               String file = pair.getU();
+                               Motor m = pair.getV();
+                               
+                               int index = indexOf(motors, m);
+                               if (index >= 0) {
+                                       // Replace previous if this has more delays, a known type or longer comment
+                                       Motor m2 = motors.get(index);
+                                       if (m.getStandardDelays().length > m2.getStandardDelays().length ||
+                                                       (m2.getMotorType() == Motor.Type.UNKNOWN &&
+                                                       m.getMotorType() != Motor.Type.UNKNOWN) ||
+                                                       (m.getDescription().trim().length() > 
+                                                       m2.getDescription().trim().length())) {
+                                               
+                                               filenames.set(index, file);
+                                               motors.set(index, m);
+                                               
+                                       }
+                               } else {
+                                       filenames.add(file);
+                                       motors.add(m);
+                               }
+                       }
+                       
+                       if (filenames.size() == 0) {
+                               
+                               System.out.println("ERROR selecting from " + list);
+                               System.exit(1);
+                               
+                       } else if (filenames.size() == 1) {
+                               
+                               select(filenames.get(0), list, false);
+                               
+                       } else {
+                               
+                               System.out.println("Choosing from " + filenames + 
+                                               " (" + count + "/" + map.size() + ")");
+                               MotorPlot plot = new MotorPlot(filenames, motors);
+                               plot.setVisible(true);
+                               plot.dispose();
+                               int n = plot.getSelected();
+                               if (n < 0) {
+                                       System.out.println("NONE SELECTED from " + filenames);
+                               } else {
+                                       select(filenames.get(n), list, true);
+                               }
+
+                       }
+                       
+               }
+
+       }
+       
+       private static void select(String selected, List<Pair<String, Motor>> list, boolean manual) {
+               System.out.print("SELECT " + selected + " ");
+               if (manual) {
+                       System.out.println("(manual)");
+               } else if (list.size() == 1) {
+                       System.out.println("(only)");
+               } else {
+                       System.out.println("(identical)");
+               }
+               
+               boolean started = false;
+               for (Pair<String, Motor> pair: list) {
+                       String file = pair.getU();
+                       if (!file.equals(selected)) {
+                               System.out.println("IGNORE " + file);
+                       }
+               }
+       }
+
+       
+       private static int indexOf(List<Motor> motors, Motor motor) {
+               for (int i=0; i<motors.size(); i++) {
+                       Motor m = motors.get(i);
+                       if (m.similar(motor)) {
+                               if (m.getStandardDelays().length == 0 || motor.getStandardDelays().length == 0 ||
+                                               Arrays.equals(m.getStandardDelays(), motor.getStandardDelays())) {
+                                       return i;
+                               }
+                       }
+               }
+               return -1;
+       }
+}