git.gag.com Git - sw/motorsim/blob - 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.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.HashMap;\r
  14 import java.util.List;\r
  15 import java.util.Map;\r
  16 import java.util.Vector;\r
  17 \r
  18 import javax.measure.quantity.Length;\r
  19 import javax.measure.unit.SI;\r
  20 import javax.swing.Box;\r
  21 import javax.swing.BoxLayout;\r
  22 import javax.swing.ComboBoxModel;\r
  23 import javax.swing.JButton;\r
  24 import javax.swing.JComboBox;\r
  25 import javax.swing.JFrame;\r
  26 import javax.swing.JLabel;\r
  27 import javax.swing.JPanel;\r
  28 import javax.swing.JProgressBar;\r
  29 import javax.swing.JSplitPane;\r
  30 import javax.swing.JTabbedPane;\r
  31 import javax.swing.JTextArea;\r
  32 import javax.swing.JTextField;\r
  33 import javax.swing.SwingUtilities;\r
  34 import javax.swing.UIManager;\r
  35 import javax.swing.WindowConstants;\r
  36 \r
  37 import org.apache.log4j.Logger;\r
  38 import org.fife.ui.rsyntaxtextarea.RSyntaxTextArea;\r
  39 import org.fife.ui.rsyntaxtextarea.SyntaxConstants;\r
  40 import org.jscience.physics.amount.Amount;\r
  41 \r
  42 import com.billkuker.rocketry.motorsim.Burn;\r
  43 import com.billkuker.rocketry.motorsim.Chamber;\r
  44 import com.billkuker.rocketry.motorsim.ChangeListening;\r
  45 import com.billkuker.rocketry.motorsim.ConvergentDivergentNozzle;\r
  46 import com.billkuker.rocketry.motorsim.CylindricalChamber;\r
  47 import com.billkuker.rocketry.motorsim.Fuel;\r
  48 import com.billkuker.rocketry.motorsim.Grain;\r
  49 import com.billkuker.rocketry.motorsim.Motor;\r
  50 import com.billkuker.rocketry.motorsim.Nozzle;\r
  51 import com.billkuker.rocketry.motorsim.RocketScience;\r
  52 import com.billkuker.rocketry.motorsim.fuel.KNSU;\r
  53 import com.billkuker.rocketry.motorsim.grain.CSlot;\r
  54 import com.billkuker.rocketry.motorsim.grain.CoredCylindricalGrain;\r
  55 import com.billkuker.rocketry.motorsim.grain.EndBurner;\r
  56 import com.billkuker.rocketry.motorsim.grain.Finocyl;\r
  57 import com.billkuker.rocketry.motorsim.grain.Moonburner;\r
  58 import com.billkuker.rocketry.motorsim.grain.MultiGrain;\r
  59 import com.billkuker.rocketry.motorsim.grain.RodAndTubeGrain;\r
  60 import com.billkuker.rocketry.motorsim.grain.Star;\r
  61 import com.billkuker.rocketry.motorsim.io.MotorIO;\r
  62 import com.billkuker.rocketry.motorsim.visual.BurnPanel;\r
  63 import com.billkuker.rocketry.motorsim.visual.Editor;\r
  64 import com.billkuker.rocketry.motorsim.visual.GrainPanel;\r
  65 import com.billkuker.rocketry.motorsim.visual.HardwarePanel;\r
  66 \r
  67 public class MotorEditor extends JTabbedPane implements PropertyChangeListener {\r
  68         private static final long serialVersionUID = 1L;\r
  69         private static Logger log = Logger.getLogger(MotorEditor.class);\r
  70         RSyntaxTextArea text = new RSyntaxTextArea();\r
  71         Motor motor;\r
  72         GrainEditor grainEditor;\r
  73         BurnTab bt;\r
  74         Burn burn;\r
  75 \r
  76         private Vector<BurnWatcher> burnWatchers = new Vector<BurnWatcher>();\r
  77         private ComboBoxModel availableFuels;\r
  78 \r
  79         //private static final int XML_TAB = 0;\r
  80         private static final int CASING_TAB = 0;\r
  81         private static final int GRAIN_TAB = 1;\r
  82         private static final int BURN_TAB = 2;\r
  83 \r
  84         private List<Class<? extends Grain>> grainTypes = new Vector<Class<? extends Grain>>();\r
  85         {\r
  86                 grainTypes.add(CoredCylindricalGrain.class);\r
  87                 grainTypes.add(Finocyl.class);\r
  88                 grainTypes.add(Star.class);\r
  89                 grainTypes.add(Moonburner.class);\r
  90                 grainTypes.add(RodAndTubeGrain.class);\r
  91                 grainTypes.add(CSlot.class);\r
  92                 grainTypes.add(EndBurner.class);\r
  93         }\r
  94 \r
  95         private abstract class Chooser<T> extends JPanel {\r
  96                 private static final long serialVersionUID = 1L;\r
  97                 private List<Class<? extends T>> types;\r
  98                 private Map<Class<? extends T>, T> old = new HashMap<Class<? extends T>, T>();\r
  99 \r
 100                 @SuppressWarnings("unchecked")\r
 101                 public Chooser(T initial, List<Class<? extends T>> ts) {\r
 102                         types = ts;\r
 103                         if ( initial != null )\r
 104                                 old.put((Class<? extends T>)initial.getClass(), initial);\r
 105                         setLayout(new BoxLayout(this, BoxLayout.Y_AXIS));\r
 106                         for (final Class<? extends T> c : types) {\r
 107                                 JButton b = new JButton(c.getSimpleName());\r
 108                                 add(b);\r
 109                                 b.addActionListener(new ActionListener() {\r
 110                                         public void actionPerformed(ActionEvent e) {\r
 111                                                 try {\r
 112                                                         T val = old.get(c);\r
 113                                                         if ( val == null ){\r
 114                                                                 System.err.println("CREATED NEW =========================");\r
 115                                                                 val = c.newInstance();\r
 116                                                                 old.put(c, val);\r
 117                                                         }\r
 118                                                         choiceMade(val);\r
 119                                                 } catch (InstantiationException e1) {\r
 120                                                         e1.printStackTrace();\r
 121                                                 } catch (IllegalAccessException e1) {\r
 122                                                         e1.printStackTrace();\r
 123                                                 }\r
 124                                         }\r
 125                                 });\r
 126                         }\r
 127                 }\r
 128 \r
 129                 protected abstract void choiceMade(T o);\r
 130         }\r
 131 \r
 132         private class BurnTab extends JPanel {\r
 133                 private static final long serialVersionUID = 1L;\r
 134                 private Thread currentThread;\r
 135                 \r
 136                 public BurnTab() {\r
 137                         setLayout(new BorderLayout());\r
 138                         setName("Simulation Results");\r
 139                         reBurn();\r
 140                 }\r
 141                 \r
 142                 private class BurnCanceled extends RuntimeException{\r
 143                         private static final long serialVersionUID = 1L;\r
 144                 };\r
 145 \r
 146                 public void reBurn() {\r
 147                         removeAll();\r
 148                         currentThread = new Thread() {\r
 149                                 public void run() {\r
 150                                         final Thread me = this;\r
 151                                         final JProgressBar bar = new JProgressBar(0, 100);\r
 152                                         add(bar, BorderLayout.NORTH);\r
 153                                         final JLabel progress = new JLabel();\r
 154                                         add(progress, BorderLayout.CENTER);\r
 155                                         try {\r
 156                                                 final Burn b = new Burn(motor,\r
 157                                                                 new Burn.BurnProgressListener() {\r
 158                                                                         @Override\r
 159                                                                         public void setProgress(float f) {\r
 160                                                                                 int pct = (int)(f*100);\r
 161                                                                                 bar.setValue(pct);\r
 162                                                                                 Amount<Length> web = motor.getGrain().webThickness();\r
 163                                                                                 Amount<Length> remaining = web.times(1.0 - f);\r
 164                                                                                 \r
 165                                                                                 progress.setText("Progress: " + pct + "% (" + RocketScience.ammountToRoundedString(remaining) + " web thickness remaining)");\r
 166                                                                                 if ( currentThread != me ){\r
 167                                                                                         throw new BurnCanceled();\r
 168                                                                                 }\r
 169                                                                         }\r
 170                                                                 });\r
 171 \r
 172                                                 final BurnPanel bp = new BurnPanel(b);\r
 173                                                 SwingUtilities.invokeLater(new Thread() {\r
 174                                                         public void run() {\r
 175                                                                 remove(bar);\r
 176                                                                 add(bp, BorderLayout.CENTER);\r
 177 \r
 178                                                                 for (BurnWatcher bw : burnWatchers)\r
 179                                                                         bw.replace(burn, b);\r
 180                                                                 burn = b;\r
 181 \r
 182                                                                 revalidate();\r
 183                                                         }\r
 184                                                 });\r
 185                                         } catch (BurnCanceled c){\r
 186                                                 log.info("Burn Canceled!");\r
 187                                         } catch (Exception e) {\r
 188                                                 remove(bar);\r
 189                                                 JTextArea t = new JTextArea(e.getMessage());\r
 190                                                 t.setEditable(false);\r
 191                                                 add(t);\r
 192                                         }\r
 193                                 }\r
 194                         };\r
 195                         currentThread.start();\r
 196                 }\r
 197         }\r
 198 \r
 199         private class GrainEditor extends JSplitPane {\r
 200                 private static final long serialVersionUID = 1L;\r
 201 \r
 202                 public GrainEditor(final Grain g) {\r
 203                         super(JSplitPane.HORIZONTAL_SPLIT);\r
 204                         setName("Grain Geometry");\r
 205                         setRightComponent(new GrainPanel(g));\r
 206                         if (g instanceof Grain.Composite) {\r
 207                                 final JPanel p = new JPanel();\r
 208                                 p.setLayout(new BoxLayout(p, BoxLayout.Y_AXIS));\r
 209                                 p.add(new Editor(g));\r
 210                                 for (Grain gg : ((Grain.Composite) g).getGrains()) {\r
 211                                         final int grainEditorIndex = p.getComponentCount() + 1;\r
 212                                         p.add(new Chooser<Grain>(gg, grainTypes) {\r
 213                                                 private static final long serialVersionUID = 1L;\r
 214 \r
 215                                                 @Override\r
 216                                                 protected void choiceMade(Grain ng) {\r
 217                                                         if (g instanceof MultiGrain) {\r
 218                                                                 ((MultiGrain) g).setGrain(ng);\r
 219                                                                 p.remove(grainEditorIndex);\r
 220                                                                 p.add(new Editor(ng), grainEditorIndex);\r
 221                                                                 p.remove(0);\r
 222                                                                 p.add(new Editor(g), 0);\r
 223                                                         }\r
 224                                                 }\r
 225                                         });\r
 226                                         p.add(new Editor(gg));\r
 227                                         if (gg instanceof ChangeListening.Subject) {\r
 228                                                 ((ChangeListening.Subject) gg)\r
 229                                                                 .addPropertyChangeListener(MotorEditor.this);\r
 230                                         }\r
 231                                 }\r
 232                                 setLeftComponent(p);\r
 233                         } else {\r
 234                                 setLeftComponent(new Editor(g));\r
 235                         }\r
 236                         // setDividerLocation(.25);\r
 237                         // setResizeWeight(.25);\r
 238                         if (g instanceof ChangeListening.Subject) {\r
 239                                 ((ChangeListening.Subject) g)\r
 240                                                 .addPropertyChangeListener(MotorEditor.this);\r
 241                         }\r
 242                 }\r
 243         }\r
 244 \r
 245         private class CaseEditor extends JSplitPane {\r
 246                 private static final long serialVersionUID = 1L;\r
 247 \r
 248                 public CaseEditor(Nozzle n, Chamber c) {\r
 249                         super(JSplitPane.VERTICAL_SPLIT);\r
 250                         setName("General Parameters");\r
 251                         \r
 252                         JPanel parts = new JPanel();\r
 253                         parts.setLayout(new BoxLayout(parts, BoxLayout.X_AXIS));\r
 254                         setTopComponent(parts);\r
 255                         setBottomComponent(new HardwarePanel(n, c));\r
 256                         \r
 257                         JPanel nameAndFuel = new JPanel();\r
 258                         nameAndFuel.setLayout(new BoxLayout(nameAndFuel, BoxLayout.Y_AXIS));\r
 259 \r
 260                         nameAndFuel.add(new JLabel("Name:"));\r
 261                         nameAndFuel.add(new JTextField(motor.getName()) {\r
 262                                 private static final long serialVersionUID = 1L;\r
 263                                 {\r
 264                                         setMinimumSize(new Dimension(200, 20));\r
 265                                         setMaximumSize(new Dimension(Short.MAX_VALUE, 20));\r
 266                                         final JTextField t = this;\r
 267                                         addFocusListener(new FocusListener() {\r
 268 \r
 269                                                 @Override\r
 270                                                 public void focusLost(FocusEvent e) {\r
 271                                                         String n = t.getText();\r
 272                                                         if (!"".equals(n) && !n.equals(motor.getName())) {\r
 273                                                                 motor.setName(n);\r
 274                                                         } else {\r
 275                                                                 t.setText(motor.getName());\r
 276                                                         }\r
 277                                                 }\r
 278 \r
 279                                                 @Override\r
 280                                                 public void focusGained(FocusEvent e) {\r
 281 \r
 282                                                 }\r
 283                                         });\r
 284 \r
 285                                 }\r
 286                         });\r
 287                         nameAndFuel.add(new JLabel("Fuel:"));\r
 288                         nameAndFuel.add( new JComboBox(availableFuels){\r
 289                                 {\r
 290                                         this.setSelectedItem(motor.getFuel());\r
 291                                 }\r
 292                                 private static final long serialVersionUID = 1L;\r
 293                                 {\r
 294                                         setMinimumSize(new Dimension(200, 20));\r
 295                                         setMaximumSize(new Dimension(Short.MAX_VALUE, 20));\r
 296                                         addActionListener(new ActionListener(){\r
 297                                                 @Override\r
 298                                                 public void actionPerformed(ActionEvent e) {\r
 299                                                         motor.setFuel((Fuel)getSelectedItem());\r
 300                                                         System.out.println("FUEL CHANGED");\r
 301                                                 }});\r
 302                                 }\r
 303                         });\r
 304                         nameAndFuel.add(Box.createVerticalGlue());\r
 305                         parts.add(nameAndFuel);\r
 306                         \r
 307                         JPanel casing = new JPanel();\r
 308                         casing.setLayout(new BoxLayout(casing, BoxLayout.Y_AXIS));\r
 309                         casing.add(new JLabel("Casing:"));\r
 310                         casing.add(new Editor(c));\r
 311                         parts.add(casing);\r
 312                         \r
 313                         JPanel nozzle = new JPanel();\r
 314                         nozzle.setLayout(new BoxLayout(nozzle, BoxLayout.Y_AXIS));\r
 315                         nozzle.add(new JLabel("Nozzle:"));\r
 316                         nozzle.add(new Editor(n));\r
 317                         parts.add(nozzle);\r
 318 \r
 319                         if (n instanceof ChangeListening.Subject) {\r
 320                                 ((ChangeListening.Subject) n)\r
 321                                                 .addPropertyChangeListener(MotorEditor.this);\r
 322                         }\r
 323                         if (c instanceof ChangeListening.Subject) {\r
 324                                 ((ChangeListening.Subject) c)\r
 325                                                 .addPropertyChangeListener(MotorEditor.this);\r
 326                         }\r
 327                 }\r
 328         }\r
 329 \r
 330         {\r
 331                 text.setSyntaxEditingStyle(SyntaxConstants.SYNTAX_STYLE_XML);\r
 332 \r
 333         }\r
 334 \r
 335         public MotorEditor(Motor m, ComboBoxModel fuels) {\r
 336                 super(JTabbedPane.BOTTOM);\r
 337                 this.availableFuels = fuels;\r
 338                 text.setName("XML");\r
 339                 text.setEditable(false);\r
 340                 //add(text, XML_TAB);\r
 341                 setMotor(m, true);\r
 342         }\r
 343 \r
 344         public Motor getMotor() {\r
 345                 return motor;\r
 346         }\r
 347 \r
 348         private void reText() {\r
 349                 try {\r
 350                         text.setText(MotorIO.writeMotor(motor));\r
 351                 } catch (IOException e) {\r
 352                         throw new Error(e);\r
 353                 }\r
 354         }\r
 355 \r
 356         private void setMotor(Motor m, boolean retext) {\r
 357                 if (motor != null)\r
 358                         motor.removePropertyChangeListener(this);\r
 359                 motor = m;\r
 360                 motor.addPropertyChangeListener(this);\r
 361                 if (retext)\r
 362                         reText();\r
 363                 if (grainEditor != null)\r
 364                         remove(grainEditor);\r
 365                 while (getTabCount() > 1)\r
 366                         removeTabAt(1);\r
 367                 add(new CaseEditor(motor.getNozzle(), motor.getChamber()), CASING_TAB);\r
 368                 add(new GrainEditor(motor.getGrain()), GRAIN_TAB);\r
 369                 add(bt = new BurnTab(), BURN_TAB);\r
 370         }\r
 371 \r
 372         public static Motor defaultMotor() {\r
 373                 Motor m = new Motor();\r
 374                 m.setName("Example Motor");\r
 375                 m.setFuel(new KNSU());\r
 376 \r
 377                 CylindricalChamber c = new CylindricalChamber();\r
 378                 c.setLength(Amount.valueOf(200, SI.MILLIMETER));\r
 379                 c.setID(Amount.valueOf(30, SI.MILLIMETER));\r
 380                 m.setChamber(c);\r
 381 \r
 382                 CoredCylindricalGrain g = new CoredCylindricalGrain();\r
 383                 try {\r
 384                         g.setLength(Amount.valueOf(70, SI.MILLIMETER));\r
 385                         g.setOD(Amount.valueOf(30, SI.MILLIMETER));\r
 386                         g.setID(Amount.valueOf(10, SI.MILLIMETER));\r
 387                 } catch (PropertyVetoException v) {\r
 388                         throw new Error(v);\r
 389                 }\r
 390 \r
 391                 m.setGrain(new MultiGrain(g, 2));\r
 392 \r
 393                 ConvergentDivergentNozzle n = new ConvergentDivergentNozzle();\r
 394                 n.setThroatDiameter(Amount.valueOf(7.962, SI.MILLIMETER));\r
 395                 n.setExitDiameter(Amount.valueOf(13.79, SI.MILLIMETER));\r
 396                 n.setEfficiency(.85);\r
 397                 m.setNozzle(n);\r
 398 \r
 399                 return m;\r
 400         }\r
 401 \r
 402         public void focusOnObject(Object o) {\r
 403                 if (o instanceof Grain)\r
 404                         setSelectedIndex(GRAIN_TAB);\r
 405                 if (o instanceof Chamber || o instanceof Nozzle)\r
 406                         setSelectedIndex(CASING_TAB);\r
 407         }\r
 408 \r
 409         public void addBurnWatcher(BurnWatcher bw) {\r
 410                 burnWatchers.add(bw);\r
 411         }\r
 412 \r
 413         @Deprecated\r
 414         public void showAsWindow() {\r
 415                 JFrame f = new JFrame();\r
 416                 f.setSize(1024, 768);\r
 417                 f.setContentPane(this);\r
 418                 f.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);\r
 419                 f.setVisible(true);\r
 420         }\r
 421 \r
 422         public static void main(String args[]) throws Exception {\r
 423                 try {\r
 424                         UIManager.setLookAndFeel(UIManager.getSystemLookAndFeelClassName());\r
 425                 } catch (Exception e1) {\r
 426                         e1.printStackTrace();\r
 427                 }\r
 428                 Vector<Fuel> ff = new Vector<Fuel>();\r
 429                 ff.add(new KNSU());\r
 430                 //new MotorEditor(defaultMotor(), ff).showAsWindow();\r
 431         }\r
 432 \r
 433         public void propertyChange(PropertyChangeEvent evt) {\r
 434                 reText();\r
 435                 // Dont re-burn for a name change!\r
 436                 if (!evt.getPropertyName().equals("Name")){\r
 437                         bt.reBurn();\r
 438                 } else {\r
 439                         for (BurnWatcher bw : burnWatchers)\r
 440                                 bw.replace(burn, burn);\r
 441                 }\r
 442         }\r
 443 \r
 444 }\r