X-Git-Url: https://git.gag.com/?a=blobdiff_plain;f=src%2Fnet%2Fsf%2Fopenrocket%2Futils%2FGraphicalMotorSelector.java;fp=src%2Fnet%2Fsf%2Fopenrocket%2Futils%2FGraphicalMotorSelector.java;h=3e3d3520276dd539aedbc06405cf3a3b3713cf72;hb=bf76931b27b597f7bc1dd5a0e7e2460ac065415b;hp=0000000000000000000000000000000000000000;hpb=def27b03bd9ebe6628421ec61112ebe4f45f043d;p=debian%2Fopenrocket

diff --git a/src/net/sf/openrocket/utils/GraphicalMotorSelector.java b/src/net/sf/openrocket/utils/GraphicalMotorSelector.java
new file mode 100644
index 00000000..3e3d3520
--- /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;
+	}
+}