Repaired regression bug in Show Trace to State action

Author: rensink

src/main/java/nl/utwente/groove/gui/jgraph/LTSJGraph.java

@@ -470,21 +470,16 @@ private Collection<LTSJCell> getTransitionCells(GraphTransition trans) {
     }
 
     /** Returns the traces from the given set of states to the start state. */
-    public Set<LTSJCell> findTraces(Iterable<GraphState> states) {
-        Set<GraphTransition> simulatorTrace = new HashSet<>();
-        Set<LTSJCell> result = new HashSet<>();
-        LTSJModel model = getNonNullModel();
+    public Set<GraphTransition> findTraces(Iterable<GraphState> states) {
+        Set<GraphTransition> result = new HashSet<>();
         for (GraphState state : states) {
             while (state instanceof GraphNextState) {
-                result.add((LTSJVertex) model.getJCellForNode(state));
                 GraphTransition trans = ((GraphNextState) state).getInTransition();
-                result.add((LTSJEdge) model.getJCellForEdge(trans));
-                simulatorTrace.add(trans);
+                result.add(trans);
                 state = trans.source();
             }
-            result.add((LTSJCell) model.getJCellForNode(state));
         }
-        getSimulatorModel().setTrace(simulatorTrace);
+        getSimulatorModel().setTrace(result);
         return result;
     }
 

src/main/java/nl/utwente/groove/gui/menu/ShowHideMenu.java

@@ -57,6 +57,7 @@
 import nl.utwente.groove.io.GrooveFileChooser;
 import nl.utwente.groove.io.HTMLConverter;
 import nl.utwente.groove.lts.GTS;
+import nl.utwente.groove.lts.GraphTransition;
 import nl.utwente.groove.util.parse.FormatException;
 
 /**
@@ -699,10 +700,19 @@ public void actionPerformed(ActionEvent evt) {
 
         @Override
         protected boolean isInvolved(JCell<@NonNull GTS> jCell) {
-            return this.trace.contains(jCell);
+            boolean result = false;
+            if (jCell instanceof LTSJCell ltsJCell) {
+                for (var t : ltsJCell.getEdges()) {
+                    result = this.trace.contains(t);
+                    if (result) {
+                        break;
+                    }
+                }
+            }
+            return result;
         }
 
-        private Set<LTSJCell> trace;
+        private Set<GraphTransition> trace;
     }
 
     /**