@@ -11,6 +11,7 @@ import org.utbot.engine.isReturn
11
11
import org.utbot.engine.selectors.PathSelector
12
12
import org.utbot.engine.stmts
13
13
import org.utbot.framework.UtSettings.copyVisualizationPathToClipboard
14
+ import org.utbot.framework.UtSettings.showLibraryClassesInVisualization
14
15
import soot.jimple.Stmt
15
16
import soot.toolkits.graph.ExceptionalUnitGraph
16
17
import java.awt.Toolkit
@@ -103,7 +104,10 @@ class GraphViz(
103
104
graph.allEdges.forEach { edge ->
104
105
val (edgeSrc, edgeDst, _) = edge
105
106
106
- if (stmtToSubgraph[edgeSrc] !in libraryGraphs && stmtToSubgraph[edgeDst] !in libraryGraphs) {
107
+ val srcInLibraryMethod = stmtToSubgraph[edgeSrc] in libraryGraphs
108
+ val dstInLibraryMethod = stmtToSubgraph[edgeDst] !in libraryGraphs
109
+
110
+ if (! (srcInLibraryMethod || dstInLibraryMethod) || showLibraryClassesInVisualization) {
107
111
dotGlobalGraph.addDotEdge(edge)
108
112
}
109
113
}
@@ -143,8 +147,10 @@ class GraphViz(
143
147
}
144
148
145
149
// Filter library methods
146
- uncompletedStack.removeIf { it.name in libraryGraphs }
147
- fullStack.removeIf { it.name in libraryGraphs }
150
+ if (! showLibraryClassesInVisualization) {
151
+ uncompletedStack.removeIf { it.name in libraryGraphs }
152
+ fullStack.removeIf { it.name in libraryGraphs }
153
+ }
148
154
149
155
// Update nodes and edges properties
150
156
dotGlobalGraph.updateProperties(executionState)
0 commit comments