Add an option to show library classes' graphs in visualization (#947)

Author: CaelmBleidd

utbot-framework-api/src/main/kotlin/org/utbot/framework/UtSettings.kt

@@ -123,6 +123,13 @@ object UtSettings : AbstractSettings(
  */
  val copyVisualizationPathToClipboard get() = useDebugVisualization
 
+ /**
+ * Set the value to true to show library classes' graphs in visualization.
+ *
+ * False by default.
+ */
+ val showLibraryClassesInVisualization by getBooleanProperty(false)
+
  /**
  * Method is paused after this timeout to give an opportunity other methods
  * to work

utbot-framework/src/main/kotlin/org/utbot/engine/selectors/strategies/GraphViz.kt

@@ -11,6 +11,7 @@ import org.utbot.engine.isReturn
 import org.utbot.engine.selectors.PathSelector
 import org.utbot.engine.stmts
 import org.utbot.framework.UtSettings.copyVisualizationPathToClipboard
+import org.utbot.framework.UtSettings.showLibraryClassesInVisualization
 import soot.jimple.Stmt
 import soot.toolkits.graph.ExceptionalUnitGraph
 import java.awt.Toolkit
@@ -103,7 +104,11 @@ class GraphViz(
  graph.allEdges.forEach { edge ->
  val (edgeSrc, edgeDst, _) = edge
 
- if (stmtToSubgraph[edgeSrc] !in libraryGraphs && stmtToSubgraph[edgeDst] !in libraryGraphs) {
+ val srcInLibraryMethod = stmtToSubgraph[edgeSrc] in libraryGraphs
+ val dstInLibraryMethod = stmtToSubgraph[edgeDst] in libraryGraphs
+ val edgeIsRelatedToLibraryMethod = srcInLibraryMethod || dstInLibraryMethod
+
+ if (!edgeIsRelatedToLibraryMethod || showLibraryClassesInVisualization) {
  dotGlobalGraph.addDotEdge(edge)
  }
  }
@@ -143,8 +148,10 @@ class GraphViz(
  }
 
  // Filter library methods
- uncompletedStack.removeIf { it.name in libraryGraphs }
- fullStack.removeIf { it.name in libraryGraphs }
+ if (!showLibraryClassesInVisualization) {
+ uncompletedStack.removeIf { it.name in libraryGraphs }
+ fullStack.removeIf { it.name in libraryGraphs }
+ }
 
  // Update nodes and edges properties
  dotGlobalGraph.updateProperties(executionState)