0047fcfbc85d77b0a62315e9cbb734b2fb1b173e
[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                 }\r
303 \r
304                 public CaseEditor() {\r
305                         super(JSplitPane.VERTICAL_SPLIT);\r
306                         setName("General Parameters");\r
307                         this.addComponentListener(this);\r
308                         \r
309                         JPanel parts = new JPanel();\r
310                         parts.setLayout(new BoxLayout(parts, BoxLayout.X_AXIS));\r
311                         setTopComponent(parts);\r
312                         \r
313                         JPanel nameAndFuel = new JPanel();\r
314                         nameAndFuel.setLayout(new BoxLayout(nameAndFuel, BoxLayout.Y_AXIS));\r
315 \r
316                         JLabel l = new JLabel("Name:");\r
317                         l.setAlignmentX(LEFT_ALIGNMENT);\r
318                         nameAndFuel.add(l);\r
319                         nameAndFuel.add(new JTextField(motor.getName()) {\r
320                                 private static final long serialVersionUID = 1L;\r
321                                 {\r
322                                         setAlignmentX(LEFT_ALIGNMENT);\r
323                                         setMinimumSize(new Dimension(200, 20));\r
324                                         setMaximumSize(new Dimension(Short.MAX_VALUE, 20));\r
325                                         final JTextField t = this;\r
326                                         addFocusListener(new FocusListener() {\r
327 \r
328                                                 @Override\r
329                                                 public void focusLost(FocusEvent e) {\r
330                                                         String n = t.getText();\r
331                                                         if (!"".equals(n) && !n.equals(motor.getName())) {\r
332                                                                 motor.setName(n);\r
333                                                         } else {\r
334                                                                 t.setText(motor.getName());\r
335                                                         }\r
336                                                 }\r
337 \r
338                                                 @Override\r
339                                                 public void focusGained(FocusEvent e) {\r
340 \r
341                                                 }\r
342                                         });\r
343 \r
344                                 }\r
345                         });\r
346                         \r
347                         l = new JLabel("Fuel:");\r
348                         l.setAlignmentX(LEFT_ALIGNMENT);\r
349                         nameAndFuel.add(l);\r
350                         \r
351                         nameAndFuel.add( new JComboBox(availableFuels){\r
352                                 private static final long serialVersionUID = 1L;\r
353                                 {\r
354                                         setAlignmentX(LEFT_ALIGNMENT);\r
355                                         this.setSelectedItem(motor.getFuel());\r
356                                         setMinimumSize(new Dimension(200, 20));\r
357                                         setMaximumSize(new Dimension(Short.MAX_VALUE, 20));\r
358                                         addActionListener(new ActionListener(){\r
359                                                 @Override\r
360                                                 public void actionPerformed(ActionEvent e) {\r
361                                                         motor.setFuel((Fuel)getSelectedItem());\r
362                                                         System.out.println("FUEL CHANGED");\r
363                                                 }});\r
364                                 }\r
365                         });\r
366                         \r
367                         l = new JLabel("Casing:");\r
368                         l.setAlignmentX(LEFT_ALIGNMENT);\r
369                         nameAndFuel.add(l);\r
370                         \r
371                         nameAndFuel.add(new ClassChooser<Chamber>(chamberTypes, motor.getChamber()) {\r
372                                 private static final long serialVersionUID = 1L;\r
373                                 {\r
374                                         setAlignmentX(LEFT_ALIGNMENT);\r
375                                         setMinimumSize(new Dimension(200, 20));\r
376                                         setMaximumSize(new Dimension(Short.MAX_VALUE, 20));\r
377                                 }\r
378                                 @Override\r
379                                 protected Chamber classSelected(Class<? extends Chamber> clazz, Chamber c) {\r
380                                         try {\r
381                                                 if ( c != null ){\r
382                                                         motor.setChamber(c);\r
383                                                 } else {\r
384                                                         motor.setChamber(clazz.newInstance());\r
385                                                 }\r
386                                                 return motor.getChamber();\r
387                                         } catch (InstantiationException e) {\r
388                                                 // TODO Auto-generated catch block\r
389                                                 e.printStackTrace();\r
390                                         } catch (IllegalAccessException e) {\r
391                                                 // TODO Auto-generated catch block\r
392                                                 e.printStackTrace();\r
393                                         }\r
394                                         return null;\r
395                                 }\r
396                         });\r
397                         \r
398                         \r
399                         nameAndFuel.add(Box.createVerticalGlue());\r
400                         parts.add(nameAndFuel);\r
401                         \r
402                         casing = new JPanel();\r
403                         casing.setLayout(new BoxLayout(casing, BoxLayout.Y_AXIS));\r
404                         l = new JLabel("Casing:");\r
405                         l.setAlignmentX(LEFT_ALIGNMENT);\r
406                         casing.add(l);\r
407                         parts.add(casing);\r
408                         \r
409                         nozzle = new JPanel();\r
410                         nozzle.setLayout(new BoxLayout(nozzle, BoxLayout.Y_AXIS));\r
411                         l = new JLabel("Nozzle:");\r
412                         l.setAlignmentX(LEFT_ALIGNMENT);\r
413                         nozzle.add(l);\r
414                         parts.add(nozzle);\r
415                         \r
416                         motor.addPropertyChangeListener(new PropertyChangeListener() {\r
417                                 @Override\r
418                                 public void propertyChange(PropertyChangeEvent arg0) {\r
419                                         setup();\r
420                                         setResizeWeight(.5);\r
421                                         setDividerLocation(.5);\r
422                                 }\r
423                         });\r
424                         \r
425                         setup();\r
426                 }\r
427 \r
428                 @Override\r
429                 public void componentHidden(ComponentEvent arg0) {\r
430 \r
431                 }\r
432 \r
433                 @Override\r
434                 public void componentMoved(ComponentEvent arg0) {\r
435 \r
436                 }\r
437 \r
438                 @Override\r
439                 public void componentResized(ComponentEvent arg0) {\r
440                         setResizeWeight(.5);\r
441                         setDividerLocation(.5);\r
442                 }\r
443 \r
444                 @Override\r
445                 public void componentShown(ComponentEvent arg0) {\r
446 \r
447                 }\r
448         }\r
449 \r
450 \r
451 \r
452 \r
453         public Motor getMotor() {\r
454                 return motor;\r
455         }\r
456 \r
457 \r
458         private void setMotor(Motor m) {\r
459                 if (motor != null)\r
460                         motor.removePropertyChangeListener(this);\r
461                 motor = m;\r
462                 motor.addPropertyChangeListener(this);\r
463                 if (grainEditor != null)\r
464                         remove(grainEditor);\r
465                 while (tabs.getTabCount() > 1)\r
466                         tabs.removeTabAt(1);\r
467                 tabs.add(new CaseEditor(), CASING_TAB);\r
468                 tabs.add(new GrainEditor(motor.getGrain()), GRAIN_TAB);\r
469                 tabs.add(bt = new BurnTab(), BURN_TAB);\r
470         }\r
471 \r
472         private static int idx;\r
473         public static Motor defaultMotor() {\r
474                 Motor m = new Motor();\r
475                 m.setName("New Motor " + ++idx);\r
476                 try {\r
477                         m.setFuel(FuelResolver.getFuel(new URI("motorsim:KNDX")));\r
478                 } catch (Exception e) {\r
479                         throw new Error(e);\r
480                 }\r
481 \r
482                 CylindricalChamber c = new CylindricalChamber();\r
483                 c.setLength(Amount.valueOf(420, SI.MILLIMETER));\r
484                 c.setID(Amount.valueOf(70, SI.MILLIMETER));\r
485                 c.setOD(Amount.valueOf(72, SI.MILLIMETER));\r
486                 m.setChamber(c);\r
487 \r
488                 CoredCylindricalGrain g = new CoredCylindricalGrain();\r
489                 try {\r
490                         g.setLength(Amount.valueOf(100, SI.MILLIMETER));\r
491                         g.setOD(Amount.valueOf(62, SI.MILLIMETER));\r
492                         g.setID(Amount.valueOf(20, SI.MILLIMETER));\r
493                 } catch (PropertyVetoException v) {\r
494                         throw new Error(v);\r
495                 }\r
496                 \r
497                 MultiGrain mg = new MultiGrain(g, 4);\r
498                 mg.setSpacing(Amount.valueOf(6, SI.MILLIMETER));\r
499                 m.setGrain(mg);\r
500 \r
501                 ConvergentDivergentNozzle n = new ConvergentDivergentNozzle();\r
502                 n.setThroatDiameter(Amount.valueOf(14.089, SI.MILLIMETER));\r
503                 n.setExitDiameter(Amount.valueOf(44.55, SI.MILLIMETER));\r
504                 n.setEfficiency(.85);\r
505                 m.setNozzle(n);\r
506 \r
507                 return m;\r
508         }\r
509 \r
510         public void focusOnObject(Object o) {\r
511                 if (o instanceof Grain)\r
512                         tabs.setSelectedIndex(GRAIN_TAB);\r
513                 if (o instanceof Chamber || o instanceof Nozzle)\r
514                         tabs.setSelectedIndex(CASING_TAB);\r
515         }\r
516 \r
517         public void addBurnWatcher(BurnWatcher bw) {\r
518                 burnWatchers.add(bw);\r
519         }\r
520 \r
521         @Deprecated\r
522         public void showAsWindow() {\r
523                 JFrame f = new JFrame();\r
524                 f.setSize(1024, 768);\r
525                 f.setContentPane(this);\r
526                 f.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);\r
527                 f.setVisible(true);\r
528         }\r
529 \r
530         public static void main(String args[]) throws Exception {\r
531                 try {\r
532                         UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());\r
533                 } catch (Exception e1) {\r
534                         e1.printStackTrace();\r
535                 }\r
536                 Vector<Fuel> ff = new Vector<Fuel>();\r
537                 ff.add(new KNSU());\r
538                 //new MotorEditor(defaultMotor(), ff).showAsWindow();\r
539         }\r
540 \r
541         public void propertyChange(PropertyChangeEvent evt) {\r
542                 // Dont re-burn for a name change!\r
543                 if (!evt.getPropertyName().equals("Name")){\r
544                         bt.reBurn();\r
545                 } else {\r
546                         for (BurnWatcher bw : burnWatchers)\r
547                                 bw.replace(burn, burn);\r
548                 }\r
549         }\r
550 \r
551 \r
552 }\r