Refresh and auto-start loaded startup simulations

2 files changed, 44 insertions, 2 deletions

diff --git a/src/main/java/simulator/VSSimulator.java b/src/main/java/simulator/VSSimulator.java
index 2cf58c3..eac3dec 100644
--- a/src/main/java/simulator/VSSimulator.java
+++ b/src/main/java/simulator/VSSimulator.java
@@ -1553,5 +1553,15 @@ public class VSSimulator extends JPanel implements VSSerializable {
         updateTaskManagerTable();
         updateProcessesComboBox();
         processesComboBox.setSelectedIndex(processesComboBox.getItemCount()-1);
+
+        revalidate();
+        repaint();
+        simulatorVisualization.revalidate();
+        simulatorVisualization.repaint();
+
+        if (simulatorFrame != null) {
+            simulatorFrame.revalidate();
+            simulatorFrame.repaint();
+        }
     }
 }
diff --git a/src/main/java/simulator/VSSimulatorFrame.java b/src/main/java/simulator/VSSimulatorFrame.java
index 818bea9..3f14ca4 100644
--- a/src/main/java/simulator/VSSimulatorFrame.java
+++ b/src/main/java/simulator/VSSimulatorFrame.java
@@ -19,6 +19,7 @@ import javax.swing.JMenuItem;
 import javax.swing.JTabbedPane;
 import javax.swing.JToolBar;
 import javax.swing.KeyStroke;
+import javax.swing.SwingUtilities;
 import javax.swing.event.ChangeEvent;
 import javax.swing.event.ChangeListener;
 
@@ -601,13 +602,44 @@ public class VSSimulatorFrame extends VSFrame {
         VSSerialize serialize = new VSSerialize();
         VSSimulator simulator = serialize.openSimulator(filename, this);
 
-        if (simulator != null)
-            startCurrentSimulator();
+        if (simulator != null) {
+            refreshLoadedSimulator(simulator);
+
+            final VSSimulator loadedSimulator = simulator;
+            if (SwingUtilities.isEventDispatchThread()) {
+                startCurrentSimulator();
+            } else {
+                SwingUtilities.invokeLater(new Runnable() {
+                    public void run() {
+                        if (currentSimulator == loadedSimulator)
+                            startCurrentSimulator();
+                    }
+                });
+            }
+        }
 
         return simulator;
     }
 
     /**
+     * Refreshes the loaded simulator tab after deserialization.
+     *
+     * @param simulator the loaded simulator
+     */
+    private void refreshLoadedSimulator(VSSimulator simulator) {
+        simulator.revalidate();
+        simulator.repaint();
+
+        if (tabbedPane != null) {
+            tabbedPane.revalidate();
+            tabbedPane.repaint();
+        }
+
+        revalidate();
+        repaint();
+    }
+
+    /**
      * Pauses the current simulator
      */
     public void pauseCurrentSimulator() {