Watch fuel for changes, reburn
[sw/motorsim] / gui / com / billkuker / rocketry / motorsim / visual / workbench / MotorEditor.java
1 package com.billkuker.rocketry.motorsim.visual.workbench;\r
2 \r
3 import java.awt.BorderLayout;\r
4 import java.awt.Color;\r
5 import java.awt.Dimension;\r
6 import java.awt.event.ActionEvent;\r
7 import java.awt.event.ActionListener;\r
8 import java.awt.event.ComponentEvent;\r
9 import java.awt.event.ComponentListener;\r
10 import java.awt.event.FocusEvent;\r
11 import java.awt.event.FocusListener;\r
12 import java.beans.PropertyChangeEvent;\r
13 import java.beans.PropertyChangeListener;\r
14 import java.beans.PropertyVetoException;\r
15 import java.net.URI;\r
16 import java.util.List;\r
17 import java.util.Vector;\r
18 \r
19 import javax.measure.unit.SI;\r
20 import javax.swing.Box;\r
21 import javax.swing.BoxLayout;\r
22 import javax.swing.DefaultComboBoxModel;\r
23 import javax.swing.JComboBox;\r
24 import javax.swing.JFrame;\r
25 import javax.swing.JLabel;\r
26 import javax.swing.JPanel;\r
27 import javax.swing.JSplitPane;\r
28 import javax.swing.JTabbedPane;\r
29 import javax.swing.JTextArea;\r
30 import javax.swing.JTextField;\r
31 import javax.swing.SwingUtilities;\r
32 import javax.swing.UIManager;\r
33 import javax.swing.WindowConstants;\r
34 \r
35 import org.apache.log4j.Logger;\r
36 import org.jscience.physics.amount.Amount;\r
37 \r
38 import com.billkuker.rocketry.motorsim.Burn;\r
39 import com.billkuker.rocketry.motorsim.Chamber;\r
40 import com.billkuker.rocketry.motorsim.ChangeListening;\r
41 import com.billkuker.rocketry.motorsim.Colors;\r
42 import com.billkuker.rocketry.motorsim.ConvergentDivergentNozzle;\r
43 import com.billkuker.rocketry.motorsim.CylindricalChamber;\r
44 import com.billkuker.rocketry.motorsim.Fuel;\r
45 import com.billkuker.rocketry.motorsim.Grain;\r
46 import com.billkuker.rocketry.motorsim.Motor;\r
47 import com.billkuker.rocketry.motorsim.Nozzle;\r
48 import com.billkuker.rocketry.motorsim.cases.Schedule40;\r
49 import com.billkuker.rocketry.motorsim.cases.Schedule80;\r
50 import com.billkuker.rocketry.motorsim.fuel.FuelResolver;\r
51 import com.billkuker.rocketry.motorsim.fuel.KNSU;\r
52 import com.billkuker.rocketry.motorsim.grain.CSlot;\r
53 import com.billkuker.rocketry.motorsim.grain.CoredCylindricalGrain;\r
54 import com.billkuker.rocketry.motorsim.grain.EndBurner;\r
55 import com.billkuker.rocketry.motorsim.grain.Finocyl;\r
56 import com.billkuker.rocketry.motorsim.grain.Moonburner;\r
57 import com.billkuker.rocketry.motorsim.grain.MultiGrain;\r
58 import com.billkuker.rocketry.motorsim.grain.RodAndTubeGrain;\r
59 import com.billkuker.rocketry.motorsim.grain.Star;\r
60 import com.billkuker.rocketry.motorsim.visual.BurnPanel;\r
61 import com.billkuker.rocketry.motorsim.visual.ClassChooser;\r
62 import com.billkuker.rocketry.motorsim.visual.Editor;\r
63 import com.billkuker.rocketry.motorsim.visual.GrainPanel;\r
64 import com.billkuker.rocketry.motorsim.visual.HardwarePanel;\r
65 import com.billkuker.rocketry.motorsim.visual.SummaryPanel;\r
66 \r
67 public class MotorEditor extends JPanel implements PropertyChangeListener, FuelResolver.FuelsChangeListener{\r
68         private static final long serialVersionUID = 1L;\r
69         private static Logger log = Logger.getLogger(MotorEditor.class);\r
70         Motor motor;\r
71         GrainEditor grainEditor;\r
72         BurnTab bt;\r
73         Burn burn;\r
74         SummaryPanel sp;\r
75         JTextArea error;\r
76         JTabbedPane tabs;\r
77 \r
78         private Vector<BurnWatcher> burnWatchers = new Vector<BurnWatcher>();\r
79         private DefaultComboBoxModel availableFuels = new DefaultComboBoxModel();\r
80         \r
81         public MotorEditor(Motor m) {\r
82                 setLayout( new BorderLayout());\r
83                 tabs = new JTabbedPane(JTabbedPane.TOP);\r
84                 add(tabs, BorderLayout.CENTER);\r
85                 setMotor(m);\r
86                 fuelsChanged();\r
87         }\r
88 \r
89         @Override\r
90         public void fuelsChanged() {\r
91                 availableFuels.removeAllElements();\r
92                 availableFuels.addElement(motor.getFuel());\r
93                 for ( Fuel f : FuelResolver.getFuelMap().values() ){\r
94                         availableFuels.addElement(f);\r
95                 }\r
96         }\r
97 \r
98         //private static final int XML_TAB = 0;\r
99         private static final int CASING_TAB = 0;\r
100         private static final int GRAIN_TAB = 1;\r
101         private static final int BURN_TAB = 2;\r
102 \r
103         private List<Class<? extends Grain>> grainTypes = new Vector<Class<? extends Grain>>();\r
104         {\r
105                 grainTypes.add(CoredCylindricalGrain.class);\r
106                 grainTypes.add(Finocyl.class);\r
107                 grainTypes.add(Star.class);\r
108                 grainTypes.add(Moonburner.class);\r
109                 grainTypes.add(RodAndTubeGrain.class);\r
110                 grainTypes.add(CSlot.class);\r
111                 grainTypes.add(EndBurner.class);\r
112         }\r
113         \r
114         private List<Class<? extends Chamber>> chamberTypes = new Vector<Class<? extends Chamber>>();\r
115         {\r
116                 chamberTypes.add(CylindricalChamber.class);\r
117                 chamberTypes.add(Schedule40.class);\r
118                 chamberTypes.add(Schedule80.class);\r
119         }\r
120 \r
121         private class BurnTab extends JPanel {\r
122                 private static final long serialVersionUID = 1L;\r
123                 private Thread currentThread;\r
124                 \r
125                 public BurnTab() {\r
126                         setLayout(new BorderLayout());\r
127                         setName("Simulation Results");\r
128                         reBurn();\r
129                 }\r
130                 \r
131                 private class BurnCanceled extends RuntimeException{\r
132                         private static final long serialVersionUID = 1L;\r
133                 };\r
134 \r
135                 public void reBurn() {\r
136                         removeAll();\r
137                         if ( error != null ){\r
138                                 MotorEditor.this.remove(error);\r
139                                 error = null;\r
140                         }\r
141                         if ( sp != null ){\r
142                                 MotorEditor.this.remove(sp);\r
143                                 sp = null;\r
144                         }\r
145                         currentThread = new Thread() {\r
146                                 public void run() {\r
147                                         final Thread me = this;\r
148                                         try {                                           \r
149                                                 final Burn b = new Burn(motor);\r
150                                                 b.addBurnProgressListener(\r
151                                                                 new Burn.BurnProgressListener() {\r
152                                                                         @Override\r
153                                                                         public void burnComplete(){};\r
154                                                                         @Override\r
155                                                                         public void setProgress(float f) {\r
156                                                                                 if ( currentThread != me ){\r
157                                                                                         throw new BurnCanceled();\r
158                                                                                 }\r
159                                                                         }\r
160                                                                 });\r
161 \r
162                                                 MotorEditor.this.add(sp = new SummaryPanel(b), BorderLayout.NORTH);\r
163                                                 revalidate();\r
164                                                 b.burn();\r
165 \r
166                                                 final BurnPanel bp = new BurnPanel(b);\r
167                                                 SwingUtilities.invokeLater(new Thread() {\r
168                                                         public void run() {\r
169                                                                 add(bp, BorderLayout.CENTER);\r
170                                                                 for (BurnWatcher bw : burnWatchers)\r
171                                                                         bw.replace(burn, b);\r
172                                                                 burn = b;\r
173                                                                 revalidate();\r
174                                                         }\r
175                                                 });\r
176                                         } catch (BurnCanceled c){\r
177                                                 log.info("Burn Canceled!");\r
178                                         } catch (final Exception e) {\r
179                                                 SwingUtilities.invokeLater(new Thread() {\r
180                                                         public void run() {\r
181                                                                 if ( sp != null )\r
182                                                                         MotorEditor.this.remove(sp);\r
183                                                                 error = new JTextArea(e.getMessage());\r
184                                                                 error.setBackground(Colors.RED);\r
185                                                                 error.setForeground(Color.WHITE);\r
186                                                                 error.setEditable(false);\r
187                                                                 MotorEditor.this.add(error, BorderLayout.NORTH);\r
188                                                                 revalidate();\r
189                                                         }\r
190                                                 });\r
191                                         }\r
192                                 }\r
193                         };\r
194                         currentThread.start();\r
195                 }\r
196         }\r
197 \r
198         private class GrainEditor extends JSplitPane {\r
199                 private static final long serialVersionUID = 1L;\r
200 \r
201                 public GrainEditor(final Grain g) {\r
202                         super(JSplitPane.HORIZONTAL_SPLIT);\r
203                         setName("Grain Geometry");\r
204                         setRightComponent(new GrainPanel(g));\r
205                         if (g instanceof Grain.Composite) {\r
206                                 final JPanel p = new JPanel();\r
207                                 p.setLayout(new BoxLayout(p, BoxLayout.Y_AXIS));\r
208                                 \r
209                                 Editor grainEditor = new Editor(g);\r
210                                 grainEditor.setAlignmentX(LEFT_ALIGNMENT);\r
211                                 p.add(grainEditor);\r
212                                 \r
213                                 for (Grain gg : ((Grain.Composite) g).getGrains()) {\r
214                                         final int grainEditorIndex = p.getComponentCount() + 2;\r
215                                         \r
216                                         JLabel l = new JLabel("Grain Type:");\r
217                                         l.setAlignmentX(LEFT_ALIGNMENT);\r
218                                         p.add(l);\r
219                                         \r
220                                         p.add(new ClassChooser<Grain>(grainTypes, gg) {\r
221                                                 private static final long serialVersionUID = 1L;\r
222                                                 {setAlignmentX(LEFT_ALIGNMENT);}\r
223                                                 @Override\r
224                                                 protected Grain classSelected(\r
225                                                                 Class<? extends Grain> clazz, Grain ng) {\r
226                                                         if ( ng == null ){\r
227                                                                 try {\r
228                                                                         ng = clazz.newInstance();\r
229                                                                 } catch (InstantiationException e) {\r
230                                                                         e.printStackTrace();\r
231                                                                 } catch (IllegalAccessException e) {\r
232                                                                         e.printStackTrace();\r
233                                                                 }\r
234                                                         }\r
235                                                         if (g instanceof MultiGrain) {\r
236                                                                 ((MultiGrain) g).setGrain(ng);\r
237                                                                 p.remove(grainEditorIndex);\r
238                                                                 p.add(new Editor(ng), grainEditorIndex);\r
239                                                                 p.remove(0);\r
240                                                                 p.add(new Editor(g), 0);\r
241                                                         }\r
242                                                         return ng;\r
243 \r
244                                                 }\r
245                                         });\r
246                                         \r
247                                         Editor ggEditor = new Editor(gg);\r
248                                         ggEditor.setAlignmentX(LEFT_ALIGNMENT);\r
249                                         p.add(ggEditor);\r
250                                         \r
251                                         if (gg instanceof ChangeListening.Subject) {\r
252                                                 ((ChangeListening.Subject) gg)\r
253                                                                 .addPropertyChangeListener(MotorEditor.this);\r
254                                         }\r
255                                 }\r
256                                 setLeftComponent(p);\r
257                         } else {\r
258                                 setLeftComponent(new Editor(g));\r
259                         }\r
260                         // setDividerLocation(.25);\r
261                         // setResizeWeight(.25);\r
262                         if (g instanceof ChangeListening.Subject) {\r
263                                 ((ChangeListening.Subject) g)\r
264                                                 .addPropertyChangeListener(MotorEditor.this);\r
265                         }\r
266                 }\r
267         }\r
268 \r
269         private class CaseEditor extends JSplitPane implements ComponentListener {\r
270                 private static final long serialVersionUID = 1L;\r
271                 \r
272                 private HardwarePanel hp;\r
273                 private JPanel casing;\r
274                 private JPanel nozzle;\r
275                 private Editor casingEditor;\r
276                 private Editor nozzleEditor;\r
277                 \r
278                 private void setup() {\r
279                         if (casingEditor != null)\r
280                                 casing.remove(casingEditor);\r
281                         casingEditor = new Editor(motor.getChamber());\r
282                         casingEditor.setAlignmentX(LEFT_ALIGNMENT);\r
283                         casing.add(casingEditor);\r
284                         \r
285                         if (nozzleEditor != null)\r
286                                 nozzle.remove(nozzleEditor);\r
287                         nozzleEditor = new Editor(motor.getNozzle());\r
288                         nozzleEditor.setAlignmentX(LEFT_ALIGNMENT);\r
289                         nozzle.add(nozzleEditor);\r
290                         \r
291                         if (hp != null)\r
292                                 remove(hp);\r
293                         setBottomComponent(hp = new HardwarePanel(motor));\r
294                         if (motor.getNozzle() instanceof ChangeListening.Subject) {\r
295                                 ((ChangeListening.Subject) motor.getNozzle())\r
296                                                 .addPropertyChangeListener(MotorEditor.this);\r
297                         }\r
298                         if (motor.getChamber() instanceof ChangeListening.Subject) {\r
299                                 ((ChangeListening.Subject) motor.getChamber())\r
300                                                 .addPropertyChangeListener(MotorEditor.this);\r
301                         }\r
302                         if (motor.getFuel() instanceof ChangeListening.Subject ){\r
303                                 ((ChangeListening.Subject) motor.getFuel())\r
304                                                 .addPropertyChangeListener(MotorEditor.this);\r
305                         }\r
306                 }\r
307 \r
308                 public CaseEditor() {\r
309                         super(JSplitPane.VERTICAL_SPLIT);\r
310                         setName("General Parameters");\r
311                         this.addComponentListener(this);\r
312                         \r
313                         JPanel parts = new JPanel();\r
314                         parts.setLayout(new BoxLayout(parts, BoxLayout.X_AXIS));\r
315                         setTopComponent(parts);\r
316                         \r
317                         JPanel nameAndFuel = new JPanel();\r
318                         nameAndFuel.setLayout(new BoxLayout(nameAndFuel, BoxLayout.Y_AXIS));\r
319 \r
320                         JLabel l = new JLabel("Name:");\r
321                         l.setAlignmentX(LEFT_ALIGNMENT);\r
322                         nameAndFuel.add(l);\r
323                         nameAndFuel.add(new JTextField(motor.getName()) {\r
324                                 private static final long serialVersionUID = 1L;\r
325                                 {\r
326                                         setAlignmentX(LEFT_ALIGNMENT);\r
327                                         setMinimumSize(new Dimension(200, 20));\r
328                                         setMaximumSize(new Dimension(Short.MAX_VALUE, 20));\r
329                                         final JTextField t = this;\r
330                                         addFocusListener(new FocusListener() {\r
331 \r
332                                                 @Override\r
333                                                 public void focusLost(FocusEvent e) {\r
334                                                         String n = t.getText();\r
335                                                         if (!"".equals(n) && !n.equals(motor.getName())) {\r
336                                                                 motor.setName(n);\r
337                                                         } else {\r
338                                                                 t.setText(motor.getName());\r
339                                                         }\r
340                                                 }\r
341 \r
342                                                 @Override\r
343                                                 public void focusGained(FocusEvent e) {\r
344 \r
345                                                 }\r
346                                         });\r
347 \r
348                                 }\r
349                         });\r
350                         \r
351                         l = new JLabel("Fuel:");\r
352                         l.setAlignmentX(LEFT_ALIGNMENT);\r
353                         nameAndFuel.add(l);\r
354                         \r
355                         nameAndFuel.add( new JComboBox(availableFuels){\r
356                                 private static final long serialVersionUID = 1L;\r
357                                 {\r
358                                         setAlignmentX(LEFT_ALIGNMENT);\r
359                                         this.setSelectedItem(motor.getFuel());\r
360                                         setMinimumSize(new Dimension(200, 20));\r
361                                         setMaximumSize(new Dimension(Short.MAX_VALUE, 20));\r
362                                         addActionListener(new ActionListener(){\r
363                                                 @Override\r
364                                                 public void actionPerformed(ActionEvent e) {\r
365                                                         motor.setFuel((Fuel)getSelectedItem());\r
366                                                         System.out.println("FUEL CHANGED");\r
367                                                 }});\r
368                                 }\r
369                         });\r
370                         \r
371                         l = new JLabel("Casing:");\r
372                         l.setAlignmentX(LEFT_ALIGNMENT);\r
373                         nameAndFuel.add(l);\r
374                         \r
375                         nameAndFuel.add(new ClassChooser<Chamber>(chamberTypes, motor.getChamber()) {\r
376                                 private static final long serialVersionUID = 1L;\r
377                                 {\r
378                                         setAlignmentX(LEFT_ALIGNMENT);\r
379                                         setMinimumSize(new Dimension(200, 20));\r
380                                         setMaximumSize(new Dimension(Short.MAX_VALUE, 20));\r
381                                 }\r
382                                 @Override\r
383                                 protected Chamber classSelected(Class<? extends Chamber> clazz, Chamber c) {\r
384                                         try {\r
385                                                 if ( c != null ){\r
386                                                         motor.setChamber(c);\r
387                                                 } else {\r
388                                                         motor.setChamber(clazz.newInstance());\r
389                                                 }\r
390                                                 return motor.getChamber();\r
391                                         } catch (InstantiationException e) {\r
392                                                 // TODO Auto-generated catch block\r
393                                                 e.printStackTrace();\r
394                                         } catch (IllegalAccessException e) {\r
395                                                 // TODO Auto-generated catch block\r
396                                                 e.printStackTrace();\r
397                                         }\r
398                                         return null;\r
399                                 }\r
400                         });\r
401                         \r
402                         \r
403                         nameAndFuel.add(Box.createVerticalGlue());\r
404                         parts.add(nameAndFuel);\r
405                         \r
406                         casing = new JPanel();\r
407                         casing.setLayout(new BoxLayout(casing, BoxLayout.Y_AXIS));\r
408                         l = new JLabel("Casing:");\r
409                         l.setAlignmentX(LEFT_ALIGNMENT);\r
410                         casing.add(l);\r
411                         parts.add(casing);\r
412                         \r
413                         nozzle = new JPanel();\r
414                         nozzle.setLayout(new BoxLayout(nozzle, BoxLayout.Y_AXIS));\r
415                         l = new JLabel("Nozzle:");\r
416                         l.setAlignmentX(LEFT_ALIGNMENT);\r
417                         nozzle.add(l);\r
418                         parts.add(nozzle);\r
419                         \r
420                         motor.addPropertyChangeListener(new PropertyChangeListener() {\r
421                                 @Override\r
422                                 public void propertyChange(PropertyChangeEvent arg0) {\r
423                                         setup();\r
424                                         setResizeWeight(.5);\r
425                                         setDividerLocation(.5);\r
426                                 }\r
427                         });\r
428                         \r
429                         setup();\r
430                 }\r
431 \r
432                 @Override\r
433                 public void componentHidden(ComponentEvent arg0) {\r
434 \r
435                 }\r
436 \r
437                 @Override\r
438                 public void componentMoved(ComponentEvent arg0) {\r
439 \r
440                 }\r
441 \r
442                 @Override\r
443                 public void componentResized(ComponentEvent arg0) {\r
444                         setResizeWeight(.5);\r
445                         setDividerLocation(.5);\r
446                 }\r
447 \r
448                 @Override\r
449                 public void componentShown(ComponentEvent arg0) {\r
450 \r
451                 }\r
452         }\r
453 \r
454 \r
455 \r
456 \r
457         public Motor getMotor() {\r
458                 return motor;\r
459         }\r
460 \r
461 \r
462         private void setMotor(Motor m) {\r
463                 if (motor != null)\r
464                         motor.removePropertyChangeListener(this);\r
465                 motor = m;\r
466                 motor.addPropertyChangeListener(this);\r
467                 if (grainEditor != null)\r
468                         remove(grainEditor);\r
469                 while (tabs.getTabCount() > 1)\r
470                         tabs.removeTabAt(1);\r
471                 tabs.add(new CaseEditor(), CASING_TAB);\r
472                 tabs.add(new GrainEditor(motor.getGrain()), GRAIN_TAB);\r
473                 tabs.add(bt = new BurnTab(), BURN_TAB);\r
474         }\r
475 \r
476         private static int idx;\r
477         public static Motor defaultMotor() {\r
478                 Motor m = new Motor();\r
479                 m.setName("New Motor " + ++idx);\r
480                 try {\r
481                         m.setFuel(FuelResolver.getFuel(new URI("motorsim:KNDX")));\r
482                 } catch (Exception e) {\r
483                         throw new Error(e);\r
484                 }\r
485 \r
486                 CylindricalChamber c = new CylindricalChamber();\r
487                 c.setLength(Amount.valueOf(420, SI.MILLIMETER));\r
488                 c.setID(Amount.valueOf(70, SI.MILLIMETER));\r
489                 c.setOD(Amount.valueOf(72, SI.MILLIMETER));\r
490                 m.setChamber(c);\r
491 \r
492                 CoredCylindricalGrain g = new CoredCylindricalGrain();\r
493                 try {\r
494                         g.setLength(Amount.valueOf(100, SI.MILLIMETER));\r
495                         g.setOD(Amount.valueOf(62, SI.MILLIMETER));\r
496                         g.setID(Amount.valueOf(20, SI.MILLIMETER));\r
497                 } catch (PropertyVetoException v) {\r
498                         throw new Error(v);\r
499                 }\r
500                 \r
501                 MultiGrain mg = new MultiGrain(g, 4);\r
502                 mg.setSpacing(Amount.valueOf(6, SI.MILLIMETER));\r
503                 m.setGrain(mg);\r
504 \r
505                 ConvergentDivergentNozzle n = new ConvergentDivergentNozzle();\r
506                 n.setThroatDiameter(Amount.valueOf(14.089, SI.MILLIMETER));\r
507                 n.setExitDiameter(Amount.valueOf(44.55, SI.MILLIMETER));\r
508                 n.setEfficiency(.85);\r
509                 m.setNozzle(n);\r
510 \r
511                 return m;\r
512         }\r
513 \r
514         public void focusOnObject(Object o) {\r
515                 if (o instanceof Grain)\r
516                         tabs.setSelectedIndex(GRAIN_TAB);\r
517                 if (o instanceof Chamber || o instanceof Nozzle)\r
518                         tabs.setSelectedIndex(CASING_TAB);\r
519         }\r
520 \r
521         public void addBurnWatcher(BurnWatcher bw) {\r
522                 burnWatchers.add(bw);\r
523         }\r
524 \r
525         @Deprecated\r
526         public void showAsWindow() {\r
527                 JFrame f = new JFrame();\r
528                 f.setSize(1024, 768);\r
529                 f.setContentPane(this);\r
530                 f.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);\r
531                 f.setVisible(true);\r
532         }\r
533 \r
534         public static void main(String args[]) throws Exception {\r
535                 try {\r
536                         UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());\r
537                 } catch (Exception e1) {\r
538                         e1.printStackTrace();\r
539                 }\r
540                 Vector<Fuel> ff = new Vector<Fuel>();\r
541                 ff.add(new KNSU());\r
542                 //new MotorEditor(defaultMotor(), ff).showAsWindow();\r
543         }\r
544 \r
545         public void propertyChange(PropertyChangeEvent evt) {\r
546                 // Dont re-burn for a name change!\r
547                 if (!evt.getPropertyName().equals("Name")){\r
548                         bt.reBurn();\r
549                 } else {\r
550                         for (BurnWatcher bw : burnWatchers)\r
551                                 bw.replace(burn, burn);\r
552                 }\r
553         }\r
554 \r
555 \r
556 }\r