1 package com.billkuker.rocketry.motorsim.visual.workbench;
\r
3 import java.awt.BorderLayout;
\r
4 import java.awt.Dimension;
\r
5 import java.awt.event.ActionEvent;
\r
6 import java.awt.event.ActionListener;
\r
7 import java.awt.event.FocusEvent;
\r
8 import java.awt.event.FocusListener;
\r
9 import java.beans.PropertyChangeEvent;
\r
10 import java.beans.PropertyChangeListener;
\r
11 import java.beans.PropertyVetoException;
\r
12 import java.io.IOException;
\r
13 import java.util.Collection;
\r
14 import java.util.HashMap;
\r
15 import java.util.List;
\r
16 import java.util.Map;
\r
17 import java.util.Vector;
\r
19 import javax.measure.quantity.Length;
\r
20 import javax.measure.unit.SI;
\r
21 import javax.swing.Box;
\r
22 import javax.swing.BoxLayout;
\r
23 import javax.swing.DefaultComboBoxModel;
\r
24 import javax.swing.JButton;
\r
25 import javax.swing.JComboBox;
\r
26 import javax.swing.JFrame;
\r
27 import javax.swing.JLabel;
\r
28 import javax.swing.JPanel;
\r
29 import javax.swing.JProgressBar;
\r
30 import javax.swing.JSplitPane;
\r
31 import javax.swing.JTabbedPane;
\r
32 import javax.swing.JTextArea;
\r
33 import javax.swing.JTextField;
\r
34 import javax.swing.SwingUtilities;
\r
35 import javax.swing.UIManager;
\r
36 import javax.swing.WindowConstants;
\r
38 import org.apache.log4j.Logger;
\r
39 import org.fife.ui.rsyntaxtextarea.RSyntaxTextArea;
\r
40 import org.fife.ui.rsyntaxtextarea.SyntaxConstants;
\r
41 import org.jscience.physics.amount.Amount;
\r
43 import com.billkuker.rocketry.motorsim.Burn;
\r
44 import com.billkuker.rocketry.motorsim.Chamber;
\r
45 import com.billkuker.rocketry.motorsim.ChangeListening;
\r
46 import com.billkuker.rocketry.motorsim.ConvergentDivergentNozzle;
\r
47 import com.billkuker.rocketry.motorsim.CylindricalChamber;
\r
48 import com.billkuker.rocketry.motorsim.Fuel;
\r
49 import com.billkuker.rocketry.motorsim.Grain;
\r
50 import com.billkuker.rocketry.motorsim.Motor;
\r
51 import com.billkuker.rocketry.motorsim.Nozzle;
\r
52 import com.billkuker.rocketry.motorsim.RocketScience;
\r
53 import com.billkuker.rocketry.motorsim.fuel.KNSU;
\r
54 import com.billkuker.rocketry.motorsim.grain.CSlot;
\r
55 import com.billkuker.rocketry.motorsim.grain.CoredCylindricalGrain;
\r
56 import com.billkuker.rocketry.motorsim.grain.EndBurner;
\r
57 import com.billkuker.rocketry.motorsim.grain.Finocyl;
\r
58 import com.billkuker.rocketry.motorsim.grain.Moonburner;
\r
59 import com.billkuker.rocketry.motorsim.grain.MultiGrain;
\r
60 import com.billkuker.rocketry.motorsim.grain.RodAndTubeGrain;
\r
61 import com.billkuker.rocketry.motorsim.grain.Star;
\r
62 import com.billkuker.rocketry.motorsim.io.MotorIO;
\r
63 import com.billkuker.rocketry.motorsim.visual.BurnPanel;
\r
64 import com.billkuker.rocketry.motorsim.visual.Editor;
\r
65 import com.billkuker.rocketry.motorsim.visual.GrainPanel;
\r
66 import com.billkuker.rocketry.motorsim.visual.HardwarePanel;
\r
68 public class MotorEditor extends JTabbedPane implements PropertyChangeListener {
\r
69 private static final long serialVersionUID = 1L;
\r
70 private static Logger log = Logger.getLogger(MotorEditor.class);
\r
71 RSyntaxTextArea text = new RSyntaxTextArea();
\r
73 GrainEditor grainEditor;
\r
77 private Vector<BurnWatcher> burnWatchers = new Vector<BurnWatcher>();
\r
78 private DefaultComboBoxModel availableFuels = new DefaultComboBoxModel();
\r
80 public void addFuel(Fuel f){
\r
81 availableFuels.addElement(f);
\r
84 //private static final int XML_TAB = 0;
\r
85 private static final int CASING_TAB = 0;
\r
86 private static final int GRAIN_TAB = 1;
\r
87 private static final int BURN_TAB = 2;
\r
89 private List<Class<? extends Grain>> grainTypes = new Vector<Class<? extends Grain>>();
\r
91 grainTypes.add(CoredCylindricalGrain.class);
\r
92 grainTypes.add(Finocyl.class);
\r
93 grainTypes.add(Star.class);
\r
94 grainTypes.add(Moonburner.class);
\r
95 grainTypes.add(RodAndTubeGrain.class);
\r
96 grainTypes.add(CSlot.class);
\r
97 grainTypes.add(EndBurner.class);
\r
100 private abstract class Chooser<T> extends JPanel {
\r
101 private static final long serialVersionUID = 1L;
\r
102 private List<Class<? extends T>> types;
\r
103 private Map<Class<? extends T>, T> old = new HashMap<Class<? extends T>, T>();
\r
105 @SuppressWarnings("unchecked")
\r
106 public Chooser(T initial, List<Class<? extends T>> ts) {
\r
108 if ( initial != null )
\r
109 old.put((Class<? extends T>)initial.getClass(), initial);
\r
110 setLayout(new BoxLayout(this, BoxLayout.Y_AXIS));
\r
111 for (final Class<? extends T> c : types) {
\r
112 JButton b = new JButton(c.getSimpleName());
\r
114 b.addActionListener(new ActionListener() {
\r
115 public void actionPerformed(ActionEvent e) {
\r
117 T val = old.get(c);
\r
118 if ( val == null ){
\r
119 System.err.println("CREATED NEW =========================");
\r
120 val = c.newInstance();
\r
124 } catch (InstantiationException e1) {
\r
125 e1.printStackTrace();
\r
126 } catch (IllegalAccessException e1) {
\r
127 e1.printStackTrace();
\r
134 protected abstract void choiceMade(T o);
\r
137 private class BurnTab extends JPanel {
\r
138 private static final long serialVersionUID = 1L;
\r
139 private Thread currentThread;
\r
142 setLayout(new BorderLayout());
\r
143 setName("Simulation Results");
\r
147 private class BurnCanceled extends RuntimeException{
\r
148 private static final long serialVersionUID = 1L;
\r
151 public void reBurn() {
\r
153 currentThread = new Thread() {
\r
154 public void run() {
\r
155 final Thread me = this;
\r
156 final JProgressBar bar = new JProgressBar(0, 100);
\r
157 add(bar, BorderLayout.NORTH);
\r
158 final JLabel progress = new JLabel();
\r
159 add(progress, BorderLayout.CENTER);
\r
161 final Burn b = new Burn(motor,
\r
162 new Burn.BurnProgressListener() {
\r
164 public void setProgress(float f) {
\r
165 int pct = (int)(f*100);
\r
167 Amount<Length> web = motor.getGrain().webThickness();
\r
168 Amount<Length> remaining = web.times(1.0 - f);
\r
170 progress.setText("Progress: " + pct + "% (" + RocketScience.ammountToRoundedString(remaining) + " web thickness remaining)");
\r
171 if ( currentThread != me ){
\r
172 throw new BurnCanceled();
\r
177 final BurnPanel bp = new BurnPanel(b);
\r
178 SwingUtilities.invokeLater(new Thread() {
\r
179 public void run() {
\r
181 add(bp, BorderLayout.CENTER);
\r
183 for (BurnWatcher bw : burnWatchers)
\r
184 bw.replace(burn, b);
\r
190 } catch (BurnCanceled c){
\r
191 log.info("Burn Canceled!");
\r
192 } catch (Exception e) {
\r
194 JTextArea t = new JTextArea(e.getMessage());
\r
195 t.setEditable(false);
\r
200 currentThread.start();
\r
204 private class GrainEditor extends JSplitPane {
\r
205 private static final long serialVersionUID = 1L;
\r
207 public GrainEditor(final Grain g) {
\r
208 super(JSplitPane.HORIZONTAL_SPLIT);
\r
209 setName("Grain Geometry");
\r
210 setRightComponent(new GrainPanel(g));
\r
211 if (g instanceof Grain.Composite) {
\r
212 final JPanel p = new JPanel();
\r
213 p.setLayout(new BoxLayout(p, BoxLayout.Y_AXIS));
\r
214 p.add(new Editor(g));
\r
215 for (Grain gg : ((Grain.Composite) g).getGrains()) {
\r
216 final int grainEditorIndex = p.getComponentCount() + 1;
\r
217 p.add(new Chooser<Grain>(gg, grainTypes) {
\r
218 private static final long serialVersionUID = 1L;
\r
221 protected void choiceMade(Grain ng) {
\r
222 if (g instanceof MultiGrain) {
\r
223 ((MultiGrain) g).setGrain(ng);
\r
224 p.remove(grainEditorIndex);
\r
225 p.add(new Editor(ng), grainEditorIndex);
\r
227 p.add(new Editor(g), 0);
\r
231 p.add(new Editor(gg));
\r
232 if (gg instanceof ChangeListening.Subject) {
\r
233 ((ChangeListening.Subject) gg)
\r
234 .addPropertyChangeListener(MotorEditor.this);
\r
237 setLeftComponent(p);
\r
239 setLeftComponent(new Editor(g));
\r
241 // setDividerLocation(.25);
\r
242 // setResizeWeight(.25);
\r
243 if (g instanceof ChangeListening.Subject) {
\r
244 ((ChangeListening.Subject) g)
\r
245 .addPropertyChangeListener(MotorEditor.this);
\r
250 private class CaseEditor extends JSplitPane {
\r
251 private static final long serialVersionUID = 1L;
\r
253 public CaseEditor(Nozzle n, Chamber c) {
\r
254 super(JSplitPane.VERTICAL_SPLIT);
\r
255 setName("General Parameters");
\r
257 JPanel parts = new JPanel();
\r
258 parts.setLayout(new BoxLayout(parts, BoxLayout.X_AXIS));
\r
259 setTopComponent(parts);
\r
260 setBottomComponent(new HardwarePanel(n, c));
\r
262 JPanel nameAndFuel = new JPanel();
\r
263 nameAndFuel.setLayout(new BoxLayout(nameAndFuel, BoxLayout.Y_AXIS));
\r
265 nameAndFuel.add(new JLabel("Name:"));
\r
266 nameAndFuel.add(new JTextField(motor.getName()) {
\r
267 private static final long serialVersionUID = 1L;
\r
269 setMinimumSize(new Dimension(200, 20));
\r
270 setMaximumSize(new Dimension(Short.MAX_VALUE, 20));
\r
271 final JTextField t = this;
\r
272 addFocusListener(new FocusListener() {
\r
275 public void focusLost(FocusEvent e) {
\r
276 String n = t.getText();
\r
277 if (!"".equals(n) && !n.equals(motor.getName())) {
\r
280 t.setText(motor.getName());
\r
285 public void focusGained(FocusEvent e) {
\r
292 nameAndFuel.add(new JLabel("Fuel:"));
\r
293 nameAndFuel.add( new JComboBox(availableFuels){
\r
295 this.setSelectedItem(motor.getFuel());
\r
297 private static final long serialVersionUID = 1L;
\r
299 setMinimumSize(new Dimension(200, 20));
\r
300 setMaximumSize(new Dimension(Short.MAX_VALUE, 20));
\r
301 addActionListener(new ActionListener(){
\r
303 public void actionPerformed(ActionEvent e) {
\r
304 motor.setFuel((Fuel)getSelectedItem());
\r
305 System.out.println("FUEL CHANGED");
\r
309 nameAndFuel.add(Box.createVerticalGlue());
\r
310 parts.add(nameAndFuel);
\r
312 JPanel casing = new JPanel();
\r
313 casing.setLayout(new BoxLayout(casing, BoxLayout.Y_AXIS));
\r
314 casing.add(new JLabel("Casing:"));
\r
315 casing.add(new Editor(c));
\r
318 JPanel nozzle = new JPanel();
\r
319 nozzle.setLayout(new BoxLayout(nozzle, BoxLayout.Y_AXIS));
\r
320 nozzle.add(new JLabel("Nozzle:"));
\r
321 nozzle.add(new Editor(n));
\r
324 if (n instanceof ChangeListening.Subject) {
\r
325 ((ChangeListening.Subject) n)
\r
326 .addPropertyChangeListener(MotorEditor.this);
\r
328 if (c instanceof ChangeListening.Subject) {
\r
329 ((ChangeListening.Subject) c)
\r
330 .addPropertyChangeListener(MotorEditor.this);
\r
336 text.setSyntaxEditingStyle(SyntaxConstants.SYNTAX_STYLE_XML);
\r
340 public MotorEditor(Motor m, Collection<Fuel> fuels) {
\r
341 super(JTabbedPane.BOTTOM);
\r
342 for ( Fuel f : fuels )
\r
344 text.setName("XML");
\r
345 text.setEditable(false);
\r
346 //add(text, XML_TAB);
\r
350 public Motor getMotor() {
\r
354 private void reText() {
\r
356 text.setText(MotorIO.writeMotor(motor));
\r
357 } catch (IOException e) {
\r
358 throw new Error(e);
\r
362 private void setMotor(Motor m, boolean retext) {
\r
364 motor.removePropertyChangeListener(this);
\r
366 motor.addPropertyChangeListener(this);
\r
369 if (grainEditor != null)
\r
370 remove(grainEditor);
\r
371 while (getTabCount() > 1)
\r
373 add(new CaseEditor(motor.getNozzle(), motor.getChamber()), CASING_TAB);
\r
374 add(new GrainEditor(motor.getGrain()), GRAIN_TAB);
\r
375 add(bt = new BurnTab(), BURN_TAB);
\r
378 public static Motor defaultMotor() {
\r
379 Motor m = new Motor();
\r
380 m.setName("Example Motor");
\r
381 m.setFuel(new KNSU());
\r
383 CylindricalChamber c = new CylindricalChamber();
\r
384 c.setLength(Amount.valueOf(200, SI.MILLIMETER));
\r
385 c.setID(Amount.valueOf(30, SI.MILLIMETER));
\r
388 CoredCylindricalGrain g = new CoredCylindricalGrain();
\r
390 g.setLength(Amount.valueOf(70, SI.MILLIMETER));
\r
391 g.setOD(Amount.valueOf(30, SI.MILLIMETER));
\r
392 g.setID(Amount.valueOf(10, SI.MILLIMETER));
\r
393 } catch (PropertyVetoException v) {
\r
394 throw new Error(v);
\r
397 m.setGrain(new MultiGrain(g, 2));
\r
399 ConvergentDivergentNozzle n = new ConvergentDivergentNozzle();
\r
400 n.setThroatDiameter(Amount.valueOf(7.962, SI.MILLIMETER));
\r
401 n.setExitDiameter(Amount.valueOf(13.79, SI.MILLIMETER));
\r
402 n.setEfficiency(.85);
\r
408 public void focusOnObject(Object o) {
\r
409 if (o instanceof Grain)
\r
410 setSelectedIndex(GRAIN_TAB);
\r
411 if (o instanceof Chamber || o instanceof Nozzle)
\r
412 setSelectedIndex(CASING_TAB);
\r
415 public void addBurnWatcher(BurnWatcher bw) {
\r
416 burnWatchers.add(bw);
\r
420 public void showAsWindow() {
\r
421 JFrame f = new JFrame();
\r
422 f.setSize(1024, 768);
\r
423 f.setContentPane(this);
\r
424 f.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);
\r
425 f.setVisible(true);
\r
428 public static void main(String args[]) throws Exception {
\r
430 UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());
\r
431 } catch (Exception e1) {
\r
432 e1.printStackTrace();
\r
434 Vector<Fuel> ff = new Vector<Fuel>();
\r
435 ff.add(new KNSU());
\r
436 //new MotorEditor(defaultMotor(), ff).showAsWindow();
\r
439 public void propertyChange(PropertyChangeEvent evt) {
\r
441 // Dont re-burn for a name change!
\r
442 if (!evt.getPropertyName().equals("Name")){
\r
445 for (BurnWatcher bw : burnWatchers)
\r
446 bw.replace(burn, burn);
\r