Only simulatorstages are allowed to connect to an agenda stage.

src/gui/simulator/StageDialog.java

@@ -27,13 +27,14 @@ public class StageDialog extends JDialog {
 
     private JPanel panel(){
         JPanel panel = new JPanel();
-
         //TODO dialog
         panel.setLayout(new BoxLayout(panel, BoxLayout.Y_AXIS));
-        panel.add(new JLabel("Stage: "));
-        this.comboBox = this.comboBox();
-        panel.add(this.comboBox);
-        panel.add(this.buttons());
+        if(this.drawEngine instanceof SimulatorStage){
+            panel.add(new JLabel("Stage: "));
+            this.comboBox = this.comboBox();
+            panel.add(this.comboBox);
+            panel.add(this.buttons());
+        }
         return panel;
     }
 
@@ -45,22 +46,17 @@ public class StageDialog extends JDialog {
         save.addActionListener(new ActionListener() {
             @Override
             public void actionPerformed(ActionEvent e) {
-                //TODO save
-                if(drawEngine instanceof SimulatorStage){
-                    SimulatorStage stage = (SimulatorStage) drawEngine;
-
-                    //if stage isn't set
-                    if(stage.getStage() == null){
-                        stage.setStage((Stage) comboBox.getSelectedItem());
-                        stage.getStage().setInUse(true);
-                    }else{
-                        stage.getStage().setInUse(false);
-                        stage.setStage((Stage) comboBox.getSelectedItem());
-                        stage.getStage().setInUse(true);
-                    }
-
-                    dispose();
+                SimulatorStage stage = (SimulatorStage) drawEngine;
+                //if stage isn't set
+                if(stage.getStage() == null){
+                    stage.setStage((Stage) comboBox.getSelectedItem());
+                    stage.getStage().setInUse(true);
+                }else{
+                    stage.getStage().setInUse(false);
+                    stage.setStage((Stage) comboBox.getSelectedItem());
+                    stage.getStage().setInUse(true);
                 }
+                dispose();
             }
         });
         panel.add(save);