Allow multiple types of FuelEditors

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

diff --git a/gui/com/billkuker/rocketry/motorsim/visual/workbench/WorkbenchTreeModel.java b/gui/com/billkuker/rocketry/motorsim/visual/workbench/WorkbenchTreeModel.java
index aa942c6e5445cd5ae60b38dc994105d1b08f4010..1db767f1d87238f5c17c4733bf82e255882696d9 100644 (file)
--- a/gui/com/billkuker/rocketry/motorsim/visual/workbench/WorkbenchTreeModel.java
+++ b/gui/com/billkuker/rocketry/motorsim/visual/workbench/WorkbenchTreeModel.java
@@ -61,7 +61,7 @@ public class WorkbenchTreeModel extends DefaultTreeModel {
        public class FuelEditNode extends FuelNode {
                private static final long serialVersionUID = 1L;
 
-               public FuelEditNode(SRFuelEditor sr){
+               public FuelEditNode(AbstractFuelEditor sr){
                        super(sr, sr.getFuel());
                        sr.getFuel().addPropertyChangeListener(new PropertyChangeListener(){
 
@@ -72,8 +72,8 @@ public class WorkbenchTreeModel extends DefaultTreeModel {
                }
                
                @Override
-               public SRFuelEditor getUserObject(){
-                       return (SRFuelEditor)super.getUserObject();
+               public AbstractFuelEditor getUserObject(){
+                       return (AbstractFuelEditor)super.getUserObject();
                }
 
        }