Fix error when attempting actions on "extra" tabs

[sw/motorsim] / gui / com / billkuker / rocketry / motorsim / visual / MultiObjectEditor.java

diff --git a/gui/com/billkuker/rocketry/motorsim/visual/MultiObjectEditor.java b/gui/com/billkuker/rocketry/motorsim/visual/MultiObjectEditor.java
index 9fcabe81ba837409fd1fb44e285bb2077b73d996..b9740b440f6ce3fce2c9587d5dfee53731058f3b 100644 (file)
--- a/gui/com/billkuker/rocketry/motorsim/visual/MultiObjectEditor.java
+++ b/gui/com/billkuker/rocketry/motorsim/visual/MultiObjectEditor.java
@@ -13,9 +13,11 @@ import java.util.Map;
 import java.util.Set;\r
 import java.util.Vector;\r
 \r
+import javax.swing.JComponent;\r
 import javax.swing.JMenu;\r
 import javax.swing.JMenuItem;\r
 import javax.swing.JOptionPane;\r
+import javax.swing.JSeparator;\r
 import javax.swing.JTabbedPane;\r
 \r
 import org.apache.log4j.Logger;\r
@@ -32,7 +34,7 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
                public abstract String getName();\r
        }\r
        \r
-       private final Frame frame;\r
+       protected final Frame frame;\r
        \r
        private final String noun;\r
        \r
@@ -51,6 +53,14 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
                this.noun = " " + noun.trim();\r
        }\r
        \r
+       protected boolean has(OBJECT o){\r
+               return objectToEditor.containsKey(o);\r
+       }\r
+       \r
+       protected void objectAdded(OBJECT o, EDITOR e){}\r
+       \r
+       protected void objectRemoved(OBJECT o, EDITOR e){}\r
+       \r
        protected final void addCreator(ObjectCreator c){\r
                creators.add(c);\r
        }\r
@@ -71,9 +81,9 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
                dirty.remove(o);\r
        }\r
        \r
-       public final JMenu getMenu(){\r
+       public JMenu getMenu(){\r
                JMenu ret = new JMenu("File");\r
-               for ( JMenuItem i : getMenuItems() )\r
+               for ( JComponent i : getMenuItems() )\r
                        ret.add(i);\r
                return ret;\r
        }\r
@@ -81,18 +91,19 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
        private void menuNew(ObjectCreator c){\r
                add(c.newObject());\r
        }\r
-       \r
+\r
        @SuppressWarnings("unchecked")\r
-       public EDITOR getSelectedEditor(){\r
-               try {\r
-                       return (EDITOR)super.getSelectedComponent();\r
-               } catch ( ClassCastException e ){\r
-                       return null;\r
-               }\r
+       public EDITOR getSelectedEditor() {\r
+               EDITOR e = (EDITOR) super.getSelectedComponent();\r
+               if (editorToObject.containsKey(e))\r
+                       return e;\r
+               return null;\r
        }\r
-       \r
+\r
        private void close(){\r
                EDITOR e = getSelectedEditor();\r
+               if ( e == null )\r
+                       return;\r
                OBJECT o = editorToObject.get(e);\r
                File f = editorToFile.get(e);\r
                \r
@@ -109,11 +120,14 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
                editorToObject.remove(e);\r
                fileToEditor.remove(f);\r
                editorToFile.remove(e);\r
+               objectRemoved(o, e);\r
                remove(e);\r
        }\r
        \r
        private void saveDialog(){\r
                EDITOR e = getSelectedEditor();\r
+               if ( e == null )\r
+                       return;\r
                if ( !editorToFile.containsKey(e) ){\r
                        log.debug("Editor has no file, saving as...");\r
                        saveAsDialog();\r
@@ -130,6 +144,8 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
        }\r
        private void saveAsDialog(){\r
                EDITOR e = getSelectedEditor();\r
+               if ( e == null )\r
+                       return;\r
                final FileDialog fd = new FileDialog(frame, "Save" + noun + " As", FileDialog.SAVE);\r
                fd.setVisible(true);\r
                if (fd.getFile() != null ) {\r
@@ -169,15 +185,16 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
                                editorToObject.put(e, o);\r
                                fileToEditor.put(file, e);\r
                                editorToFile.put(e, file);\r
-                               addTab(file.getName(), e);                                                                      \r
+                               addTab(file.getName(), e);      \r
+                               objectAdded(o, e);\r
                        } catch (Exception e) {\r
                                errorDialog(e);\r
                        }\r
                }\r
        }\r
        \r
-       public final List<JMenuItem> getMenuItems(){\r
-               List<JMenuItem> ret = new Vector<JMenuItem>();\r
+       public final List<JComponent> getMenuItems(){\r
+               List<JComponent> ret = new Vector<JComponent>();\r
                if ( creators.size() == 1 ){\r
                        final ObjectCreator c = creators.get(0);\r
                        ret.add(new JMenuItem("New " + c.getName()){\r
@@ -237,6 +254,7 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
                                });\r
                        }\r
                });\r
+               ret.add(new JSeparator());\r
                ret.add(new JMenuItem("Save" + noun){\r
                        private static final long serialVersionUID = 1L;\r
                        {\r
@@ -270,6 +288,17 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
                editorToObject.put(e, o);\r
                addTab("new", e);\r
                dirty(o);\r
+               objectAdded(o, e);\r
+       }\r
+       \r
+       protected final void add(final OBJECT o, final File f){\r
+               EDITOR e = createEditor(o);\r
+               objectToEditor.put(o, e);\r
+               editorToObject.put(e, o);\r
+               fileToEditor.put(f, e);\r
+               editorToFile.put(e, f);\r
+               addTab(f.getName(), e);\r
+               objectAdded(o, e);\r
        }\r
        \r
        public final void load(final File f) throws IOException{\r
@@ -280,6 +309,7 @@ public abstract class MultiObjectEditor<OBJECT, EDITOR extends Component> extend
                fileToEditor.put(f, e);\r
                editorToFile.put(e, f);\r
                addTab(f.getName(), e);\r
+               objectAdded(o, e);\r
        }\r
        \r
        public abstract EDITOR createEditor(final OBJECT o);\r