Work on keeping the unfolding skeleton when switching between states

1 files changed, 6 insertions, 1 deletions

diff --git a/Source/ModelViewer/BaseProvider.cs b/Source/ModelViewer/BaseProvider.cs
index a4c61589..766768e2 100644
--- a/Source/ModelViewer/BaseProvider.cs
+++ b/Source/ModelViewer/BaseProvider.cs
@@ -15,13 +15,18 @@ namespace Microsoft.Boogie.ModelViewer.Base
       return true;

     }

 

-    public IEnumerable<IDisplayNode> GetStates(Model m)

+    IEnumerable<IDisplayNode> GetStateNodes(Model m)

     {

       yield return GenericNodes.Functions(m);

       yield return GenericNodes.Constants(m);

       foreach (var s in m.States)

         yield return new StateNode(s);

     }

+

+    public IEnumerable<IState> GetStates(Model m)

+    {

+      yield return new TopState("TOP", GetStateNodes(m));

+    }

   }

 

   public class StateNode : DisplayNode