git.gag.com Git - sw/motorsim/blob - gui/com/billkuker/rocketry/motorsim/visual/workbench/MotorWorkbench.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.FileDialog;\r
   6 import java.awt.event.ActionEvent;\r
   7 import java.awt.event.ActionListener;\r
   8 import java.beans.PropertyChangeEvent;\r
   9 import java.beans.PropertyChangeListener;\r
  10 import java.io.File;\r
  11 import java.util.HashMap;\r
  12 import java.util.List;\r
  13 import java.util.Vector;\r
  14 \r
  15 import javax.swing.ButtonGroup;\r
  16 import javax.swing.DefaultComboBoxModel;\r
  17 import javax.swing.JFrame;\r
  18 import javax.swing.JMenu;\r
  19 import javax.swing.JMenuBar;\r
  20 import javax.swing.JMenuItem;\r
  21 import javax.swing.JOptionPane;\r
  22 import javax.swing.JPanel;\r
  23 import javax.swing.JRadioButtonMenuItem;\r
  24 import javax.swing.JScrollPane;\r
  25 import javax.swing.JSeparator;\r
  26 import javax.swing.JSplitPane;\r
  27 import javax.swing.JTabbedPane;\r
  28 import javax.swing.JTree;\r
  29 import javax.swing.WindowConstants;\r
  30 import javax.swing.event.TreeSelectionEvent;\r
  31 import javax.swing.event.TreeSelectionListener;\r
  32 import javax.swing.tree.DefaultMutableTreeNode;\r
  33 import javax.swing.tree.TreePath;\r
  34 import javax.swing.tree.TreeSelectionModel;\r
  35 \r
  36 import com.billkuker.rocketry.motorsim.Burn;\r
  37 import com.billkuker.rocketry.motorsim.Fuel;\r
  38 import com.billkuker.rocketry.motorsim.Motor;\r
  39 import com.billkuker.rocketry.motorsim.RocketScience.UnitPreference;\r
  40 import com.billkuker.rocketry.motorsim.fuel.KNDX;\r
  41 import com.billkuker.rocketry.motorsim.fuel.KNER;\r
  42 import com.billkuker.rocketry.motorsim.fuel.KNSB;\r
  43 import com.billkuker.rocketry.motorsim.fuel.KNSU;\r
  44 import com.billkuker.rocketry.motorsim.io.ENGExporter;\r
  45 import com.billkuker.rocketry.motorsim.io.MotorIO;\r
  46 import com.billkuker.rocketry.motorsim.visual.FuelPanel;\r
  47 import com.billkuker.rocketry.motorsim.visual.workbench.WorkbenchTreeModel.FuelEditNode;\r
  48 import com.billkuker.rocketry.motorsim.visual.workbench.WorkbenchTreeModel.FuelNode;\r
  49 \r
  50 public class MotorWorkbench extends JFrame implements TreeSelectionListener {\r
  51         private static final long serialVersionUID = 1L;\r
  52         \r
  53         private List<Class<? extends Fuel>> fuelTypes = new Vector<Class<? extends Fuel>>();\r
  54         {\r
  55                 fuelTypes.add(KNSB.class);\r
  56                 fuelTypes.add(KNSU.class);\r
  57                 fuelTypes.add(KNER.class);\r
  58                 fuelTypes.add(KNDX.class);\r
  59         }\r
  60         \r
  61         private JPanel top;\r
  62         private JSplitPane split;\r
  63         private JTree tree;\r
  64         private JTabbedPane motors;\r
  65         private WorkbenchTreeModel tm;\r
  66         private MultiBurnChart mb;\r
  67         private JFrame allBurns;\r
  68 \r
  69         private HashMap<MotorEditor, File> e2f = new HashMap<MotorEditor, File>();\r
  70         private HashMap<File, MotorEditor> f2e = new HashMap<File, MotorEditor>();\r
  71 \r
  72         private HashMap<Motor, MotorEditor> m2e = new HashMap<Motor, MotorEditor>();\r
  73         \r
  74         private DefaultComboBoxModel fuels = new DefaultComboBoxModel();\r
  75         \r
  76         public MotorWorkbench() {\r
  77                 setTitle("MotorSim 1.0 RC1");\r
  78                 addMenu();\r
  79                 setSize(1024, 768);\r
  80                 top = new JPanel(new BorderLayout());\r
  81                 setContentPane(top);\r
  82                 \r
  83                 mb = new MultiBurnChart();\r
  84                 allBurns = new JFrame();\r
  85                 allBurns.setTitle("All Burns");\r
  86                 allBurns.setSize(800, 600);\r
  87                 setDefaultCloseOperation(WindowConstants.HIDE_ON_CLOSE);\r
  88                 allBurns.add(mb);\r
  89 \r
  90                 motors = new JTabbedPane();\r
  91 \r
  92                 tree = new JTree(tm = new WorkbenchTreeModel());\r
  93                 tree.setCellRenderer(new WorkbenchTreeCellRenderer());\r
  94                 tree.getSelectionModel().setSelectionMode(\r
  95                                 TreeSelectionModel.SINGLE_TREE_SELECTION);\r
  96                 tree.setMinimumSize(new Dimension(200, 100));\r
  97 \r
  98                 // Listen for when the selection changes.\r
  99                 tree.addTreeSelectionListener(this);\r
 100 \r
 101                 split = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT, new JScrollPane(\r
 102                                 tree), motors);\r
 103                 split.setDividerLocation(200);\r
 104                 split.setResizeWeight(0);\r
 105                 split.resetToPreferredSizes();\r
 106                 split.revalidate();\r
 107                 \r
 108                 top.add(split, BorderLayout.CENTER);\r
 109                 \r
 110                 for ( Class<? extends Fuel> f : fuelTypes){\r
 111                         try {\r
 112                                 addFuel(f.newInstance());\r
 113                         } catch (InstantiationException e) {\r
 114                                 // TODO Auto-generated catch block\r
 115                                 e.printStackTrace();\r
 116                         } catch (IllegalAccessException e) {\r
 117                                 // TODO Auto-generated catch block\r
 118                                 e.printStackTrace();\r
 119                         }\r
 120                 }\r
 121 \r
 122                 setDefaultCloseOperation(WindowConstants.EXIT_ON_CLOSE);\r
 123                 setVisible(true);\r
 124 \r
 125         }\r
 126 \r
 127         private void addMenu() {\r
 128 \r
 129                 setJMenuBar(new JMenuBar() {\r
 130                         private static final long serialVersionUID = 1L;\r
 131 \r
 132                         {\r
 133                                 add(new JMenu("File") {\r
 134                                         private static final long serialVersionUID = 1L;\r
 135 \r
 136                                         {\r
 137                                                 add(new JMenuItem("New Motor") {\r
 138                                                         private static final long serialVersionUID = 1L;\r
 139                                                         {\r
 140                                                                 addActionListener(new ActionListener() {\r
 141 \r
 142                                                                         @Override\r
 143                                                                         public void actionPerformed(ActionEvent arg0) {\r
 144                                                                                 addMotor(MotorEditor.defaultMotor(),\r
 145                                                                                                 null);\r
 146                                                                         }\r
 147                                                                 });\r
 148 \r
 149                                                         }\r
 150                                                 });\r
 151                                                 add(new JMenuItem("Open...") {\r
 152                                                         private static final long serialVersionUID = 1L;\r
 153                                                         {\r
 154                                                                 addActionListener(new ActionListener() {\r
 155                                                                         @Override\r
 156                                                                         public void actionPerformed(ActionEvent arg0) {\r
 157 \r
 158                                                                                 final FileDialog fd = new FileDialog(MotorWorkbench.this, "Open Motor", FileDialog.LOAD);\r
 159                                                                                 fd.setVisible(true);\r
 160                                                                                 if ( fd.getFile() != null ) {\r
 161                                                                                         File file = new File(fd.getDirectory() + fd.getFile());\r
 162                                                                                         if (f2e.get(file) != null) {\r
 163                                                                                                 motors.setSelectedComponent(f2e\r
 164                                                                                                                 .get(file));\r
 165                                                                                                 return;\r
 166                                                                                         }\r
 167                                                                                         try {\r
 168                                                                                                 Motor m = MotorIO\r
 169                                                                                                                 .readMotor(file);\r
 170                                                                                                 addFuel(m.getFuel());\r
 171                                                                                                 addMotor(m, file);\r
 172                                                                                                 \r
 173                                                                                         } catch (Exception e) {\r
 174                                                                                                 JOptionPane.showMessageDialog(\r
 175                                                                                                                 MotorWorkbench.this, e\r
 176                                                                                                                                 .getMessage());\r
 177                                                                                         }\r
 178                                                                                 }\r
 179                                                                         }\r
 180                                                                 });\r
 181                                                         }\r
 182                                                 });\r
 183                                                 \r
 184                                                 add(new JSeparator());\r
 185                                                 \r
 186                                                 add(new JMenuItem("Close Motor") {\r
 187                                                         private static final long serialVersionUID = 1L;\r
 188                                                         {\r
 189                                                                 addActionListener(new ActionListener() {\r
 190                                                                         @Override\r
 191                                                                         public void actionPerformed(ActionEvent ev) {\r
 192                                                                                 MotorEditor e = (MotorEditor) motors\r
 193                                                                                                 .getSelectedComponent();\r
 194                                                                                 tm.removeMotor(e.getMotor());\r
 195                                                                                 motors.remove(e);\r
 196                                                                                 f2e.remove(e2f.get(e));\r
 197                                                                                 e2f.remove(e);\r
 198                                                                                 m2e.remove(e.getMotor());\r
 199                                                                                 mb.removeBurn(e.burn);\r
 200                                                                         }\r
 201                                                                 });\r
 202                                                         }\r
 203                                                 });\r
 204                                                 \r
 205                                                 add(new JMenuItem("Save Motor") {\r
 206                                                         private static final long serialVersionUID = 1L;\r
 207                                                         {\r
 208                                                                 addActionListener(new ActionListener() {\r
 209                                                                         @Override\r
 210                                                                         public void actionPerformed(ActionEvent e) {\r
 211                                                                                 MotorEditor m = (MotorEditor) motors\r
 212                                                                                                 .getSelectedComponent();\r
 213                                                                                 File f = e2f.get(m);\r
 214                                                                                 if (f != null)\r
 215                                                                                         save(m.getMotor(), f);\r
 216 \r
 217                                                                         }\r
 218                                                                 });\r
 219                                                         }\r
 220                                                 });\r
 221                                                 add(new JMenuItem("Save Motor As...") {\r
 222                                                         private static final long serialVersionUID = 1L;\r
 223                                                         {\r
 224                                                                 addActionListener(new ActionListener() {\r
 225                                                                         @Override\r
 226                                                                         public void actionPerformed(ActionEvent arg0) {\r
 227 \r
 228                                                                                 final FileDialog fd = new FileDialog(MotorWorkbench.this, "Save Motor As", FileDialog.SAVE);\r
 229                                                                                 fd.setVisible(true);\r
 230                                                                                 if (fd.getFile() != null ) {\r
 231                                                                                         File file = new File(fd.getDirectory() + fd.getFile());\r
 232                                                                                         MotorEditor m = (MotorEditor) motors\r
 233                                                                                                         .getSelectedComponent();\r
 234                                                                                         try {\r
 235                                                                                                 save(m.getMotor(), file);\r
 236                                                                                                 e2f.put(m, file);\r
 237                                                                                                 f2e.put(file, m);\r
 238                                                                                                 motors.setTitleAt(motors\r
 239                                                                                                                 .getSelectedIndex(),\r
 240                                                                                                                 file.getName());\r
 241                                                                                                 // TODO Set tab title\r
 242                                                                                         } catch (Exception e) {\r
 243                                                                                                 JOptionPane.showMessageDialog(\r
 244                                                                                                                 MotorWorkbench.this, e\r
 245                                                                                                                                 .getMessage());\r
 246                                                                                         }\r
 247                                                                                 }\r
 248                                                                         }\r
 249                                                                 });\r
 250                                                         }\r
 251                                                 });\r
 252                                                 \r
 253 \r
 254                                                 add(new JSeparator());\r
 255                                                 add(new JMenuItem("New Fuel") {\r
 256                                                         private static final long serialVersionUID = 1L;\r
 257                                                         {\r
 258                                                                 addActionListener(new ActionListener() {\r
 259                                                                         @Override\r
 260                                                                         public void actionPerformed(ActionEvent arg0) {\r
 261                                                                                 newFuel();\r
 262                                                                         }\r
 263                                                                 });\r
 264 \r
 265                                                         }\r
 266                                                 });\r
 267                                                 add(new JMenuItem("Save Fuel") {\r
 268                                                         private static final long serialVersionUID = 1L;\r
 269                                                 });\r
 270                                                 add(new JSeparator());\r
 271                                                 add(new JMenuItem("Export .ENG"){\r
 272                                                         private static final long serialVersionUID = 1L;\r
 273 \r
 274                                                         {\r
 275                                                                 addActionListener(new ActionListener() {\r
 276                                                                         @Override\r
 277                                                                         public void actionPerformed(ActionEvent arg0) {\r
 278 \r
 279                                                                                 final FileDialog fd = new FileDialog(MotorWorkbench.this, "Export .ENG File", FileDialog.SAVE);\r
 280                                                                                 fd.setFile("motorsim.eng");\r
 281                                                                                 fd.setVisible(true);\r
 282                                                                                 if (fd.getFile() != null ) {\r
 283                                                                                         File file = new File(fd.getDirectory() + fd.getFile());\r
 284                                                                                         Vector<Burn> bb = new Vector<Burn>();\r
 285                                                                                         for( MotorEditor me : m2e.values() )\r
 286                                                                                                 bb.add(me.burn);\r
 287                                                                                         try{\r
 288                                                                                                 ENGExporter.export(bb, file);\r
 289                                                                                         } catch ( Exception e ){\r
 290                                                                                                 e.printStackTrace();\r
 291                                                                                         }\r
 292                                                                                 }\r
 293                                                                         }\r
 294                                                                 });\r
 295                                                         }\r
 296                                                 });\r
 297                                         }\r
 298                                 });\r
 299                                 add(new JMenu("Settings") {\r
 300                                         private static final long serialVersionUID = 1L;\r
 301                                         {\r
 302                                                 ButtonGroup units = new ButtonGroup();\r
 303                                                 JRadioButtonMenuItem sci = new JRadioButtonMenuItem(\r
 304                                                                 "SI");\r
 305                                                 JRadioButtonMenuItem nonsci = new JRadioButtonMenuItem(\r
 306                                                                 "NonSI");\r
 307                                                 units.add(sci);\r
 308                                                 units.add(nonsci);\r
 309                                                 sci.setSelected(UnitPreference.getUnitPreference() == UnitPreference.SI);\r
 310                                                 nonsci.setSelected(UnitPreference.getUnitPreference() == UnitPreference.NONSI);\r
 311                                                 sci.addActionListener(new ActionListener() {\r
 312                                                         public void actionPerformed(ActionEvent arg0) {\r
 313                                                                 UnitPreference\r
 314                                                                                 .setUnitPreference(UnitPreference.SI);\r
 315                                                         }\r
 316                                                 });\r
 317                                                 nonsci.addActionListener(new ActionListener() {\r
 318                                                         public void actionPerformed(ActionEvent arg0) {\r
 319                                                                 UnitPreference\r
 320                                                                                 .setUnitPreference(UnitPreference.NONSI);\r
 321                                                         }\r
 322                                                 });\r
 323                                                 add(sci);\r
 324                                                 add(nonsci);\r
 325                                         }\r
 326                                 });\r
 327                                 add(new JMenu("View") {\r
 328                                         private static final long serialVersionUID = 1L;\r
 329                                         {\r
 330                                                 add(new JMenuItem("Show All Motors Graph") {\r
 331                                                         private static final long serialVersionUID = 1L;\r
 332                                                         {\r
 333                                                                 addActionListener(new ActionListener() {\r
 334                                                                         @Override\r
 335                                                                         public void actionPerformed(ActionEvent arg0) {\r
 336                                                                                 allBurns.setVisible(true);\r
 337                                                                                 allBurns.toFront();\r
 338                                                                         }\r
 339                                                                 });\r
 340                                                         }\r
 341                                                 });\r
 342                                         }\r
 343                                 });\r
 344                         }\r
 345                 });\r
 346         }\r
 347         \r
 348         private void addFuel(Fuel f){\r
 349                 \r
 350                 fuels.addElement(f);\r
 351                 FuelPanel fp = new FuelPanel(f);\r
 352                 FuelNode fn = tm.new FuelNode(fp, f);\r
 353                 tm.getFuels().add(fn);\r
 354                 tm.nodeStructureChanged(tm.getFuels());\r
 355                 motors.addTab(f.getName(), fp);\r
 356         }\r
 357         \r
 358         private void newFuel(){\r
 359                 final SRFuelEditor ed = new SRFuelEditor();\r
 360                 fuels.addElement(ed.getFuel());\r
 361                 final FuelEditNode node = tm.new FuelEditNode(ed);\r
 362                 tm.getFuels().add(node);\r
 363                 tm.nodeStructureChanged(tm.getFuels());\r
 364                 motors.addTab(ed.getFuel().getName(), ed);\r
 365                 ed.getFuel().addPropertyChangeListener(new PropertyChangeListener(){\r
 366                         @Override\r
 367                         public void propertyChange(PropertyChangeEvent evt) {\r
 368                                 if ( evt.getPropertyName().equals("Name")){\r
 369                                         for ( int i = 0; i < motors.getTabCount(); i++ ){\r
 370                                                 if ( motors.getComponent(i) == ed ){\r
 371                                                         motors.setTitleAt(i, ed.getFuel().getName());\r
 372                                                         tm.nodeChanged(node);\r
 373                                                 }\r
 374                                         }\r
 375                                 }\r
 376                         }});\r
 377         }\r
 378 \r
 379         private void save(Motor m, File f) {\r
 380                 try {\r
 381                         MotorIO.writeMotor(m, f);\r
 382                 } catch (Throwable t) {\r
 383                         JOptionPane.showMessageDialog(MotorWorkbench.this, t.getMessage());\r
 384                 }\r
 385         }\r
 386 \r
 387         public void addMotor(Motor m, File f) {\r
 388                 tm.addMotor(m);\r
 389                 MotorEditor e = new MotorEditor(m, fuels);\r
 390                 e.addBurnWatcher(mb);\r
 391                 String title;\r
 392                 if (f == null) {\r
 393                         title = "New Motor";\r
 394                 } else {\r
 395                         title = f.getName();\r
 396                         e2f.put(e, f);\r
 397                         f2e.put(f, e);\r
 398                 }\r
 399                 m2e.put(m, e);\r
 400                 motors.addTab(title, e);\r
 401         }\r
 402 \r
 403         @Override\r
 404         public void valueChanged(TreeSelectionEvent e) {                \r
 405                 if ( e.getPath().getLastPathComponent() instanceof FuelNode ){\r
 406                         FuelNode fen = ((FuelNode)e.getPath().getLastPathComponent());\r
 407                                 motors.setSelectedComponent(fen.getUserObject());                       \r
 408                 }\r
 409                 \r
 410                 Motor m = getMotor(e.getPath());\r
 411                 \r
 412                 if ( m == null )\r
 413                         return;\r
 414 \r
 415                 motors.setSelectedComponent(m2e.get(m));\r
 416 \r
 417                 if (e.getPath().getLastPathComponent() instanceof DefaultMutableTreeNode) {\r
 418                         Object o = ((DefaultMutableTreeNode) e.getPath()\r
 419                                         .getLastPathComponent()).getUserObject();\r
 420                         m2e.get(m).focusOnObject(o);\r
 421                 }\r
 422                 \r
 423 \r
 424         }\r
 425 \r
 426         private Motor getMotor(TreePath p) {\r
 427                 if (p.getLastPathComponent() instanceof WorkbenchTreeModel.MotorNode) {\r
 428                         return ((WorkbenchTreeModel.MotorNode) p.getLastPathComponent())\r
 429                                         .getUserObject();\r
 430                 } else if (p.getPath().length > 1)\r
 431                         return getMotor(p.getParentPath());\r
 432                 return null;\r
 433         }\r
 434 }\r