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 bcbb8ddc0da4f007996dafef1c41beb4d18a518d..b9740b440f6ce3fce2c9587d5dfee53731058f3b 100644 (file)
--- a/gui/com/billkuker/rocketry/motorsim/visual/MultiObjectEditor.java
+++ b/gui/com/billkuker/rocketry/motorsim/visual/MultiObjectEditor.java
@@ -53,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
@@ -83,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
@@ -111,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
@@ -132,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
@@ -171,7 +185,8 @@ 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
@@ -273,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
@@ -283,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