From 1896b092007bdc8091373f7e63e586723becd82f Mon Sep 17 00:00:00 2001
From: dgelessus <dgelessus@users.noreply.github.com>
Date: Thu, 12 Jul 2018 12:45:19 +0200
Subject: [PATCH] Provide source code to error highlighter in :find
---
src/main/java/de/prob2/jupyter/commands/FindCommand.java | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/main/java/de/prob2/jupyter/commands/FindCommand.java b/src/main/java/de/prob2/jupyter/commands/FindCommand.java
index 2abd00a..cd5b7db 100644
--- a/src/main/java/de/prob2/jupyter/commands/FindCommand.java
+++ b/src/main/java/de/prob2/jupyter/commands/FindCommand.java
@@ -40,8 +40,10 @@ public final class FindCommand implements Command {
@Override
public @NotNull DisplayData run(final @NotNull String argString) {
final Trace trace = this.animationSelector.getCurrentTrace();
- final IEvalElement pred = trace.getModel().parseFormula(argString, FormulaExpand.EXPAND);
- final Trace newTrace = trace.getStateSpace().getTraceToState(pred);
+ final Trace newTrace = CommandUtils.withSourceCode(argString, () -> {
+ final IEvalElement pred = trace.getModel().parseFormula(argString, FormulaExpand.EXPAND);
+ return trace.getStateSpace().getTraceToState(pred);
+ });
this.animationSelector.changeCurrentAnimation(newTrace);
return new DisplayData("Found state: " + newTrace.getCurrentState().getId());
}
--
GitLab