Fixed benchmark printing (#877)

Fraunhofer-AISEC · Sep 1, 2022 · bbc1ec5 · bbc1ec5
1 parent 015454c
commit bbc1ec5
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 9 deletions.
diff --git a/cpg-core/src/main/java/de/fraunhofer/aisec/cpg/TranslationResult.java b/cpg-core/src/main/java/de/fraunhofer/aisec/cpg/TranslationResult.java
@@ -61,7 +61,7 @@ public class TranslationResult extends Node implements StatisticsHolder {
    */
   private final Set<Node> additionalNodes = new HashSet<>();
 
-  private final List<MeasurementHolder> benchmarks = new ArrayList<>();
+  private final Set<MeasurementHolder> benchmarks = new LinkedHashSet<>();
 
   public TranslationResult(TranslationManager translationManager) {
     this.translationManager = translationManager;
@@ -171,7 +171,7 @@ public void addBenchmark(@NotNull MeasurementHolder b) {
   }
 
   @NotNull
-  public List<MeasurementHolder> getBenchmarks() {
+  public Set<MeasurementHolder> getBenchmarks() {
     return benchmarks;
   }
 

diff --git a/cpg-core/src/main/java/de/fraunhofer/aisec/cpg/helpers/MeasurementHolder.kt b/cpg-core/src/main/java/de/fraunhofer/aisec/cpg/helpers/MeasurementHolder.kt
@@ -55,24 +55,30 @@ class BenchmarkResults(val entries: List<List<Any>>) {
 /** Interface definition to hold different statistics about the translation process. */
 interface StatisticsHolder {
     val translatedFiles: List<String>
-    val benchmarks: List<MeasurementHolder>
+    val benchmarks: Set<MeasurementHolder>
     val config: TranslationConfiguration
 
     fun addBenchmark(b: MeasurementHolder)
 
     val benchmarkResults: BenchmarkResults
         get() {
-            return BenchmarkResults(
-                listOf(
+            val results =
+                mutableListOf(
                     listOf("Translation config", config),
                     listOf("Number of files translated", translatedFiles.size),
                     listOf(
                         "Translated file(s)",
                         translatedFiles.map { relativeOrAbsolute(Path.of(it), config.topLevel) }
                     ),
-                    *benchmarks.map { it.benchmarkedValues }.toTypedArray()
                 )
-            )
+
+            benchmarks.forEach {
+                it.measurements.forEach { measurement ->
+                    results += listOf(measurement.key, measurement.value)
+                }
+            }
+
+            return BenchmarkResults(results)
         }
 }
 
@@ -212,7 +218,7 @@ constructor(
     ): Any? {
         if (measurementKey == null || measurementValue == null) return null
 
-        measurements["Measured $measurementKey"] = measurementValue
+        measurements["Measurement: $measurementKey"] = measurementValue
         logDebugMsg("$caller $measurementKey: result is $measurementValue")
 
         // update our holder, if we have any

diff --git a/cpg-core/src/main/java/de/fraunhofer/aisec/cpg/passes/StatisticsCollectionPass.kt b/cpg-core/src/main/java/de/fraunhofer/aisec/cpg/passes/StatisticsCollectionPass.kt
@@ -56,7 +56,7 @@ class StatisticsCollectionPass : Pass() {
 
         val nodeMeasurement =
             MeasurementHolder(this.javaClass, "Measuring Nodes", false, translationResult)
-        nodeMeasurement.addMeasurement("Graph nodes", nodes.toString())
+        nodeMeasurement.addMeasurement("Total graph nodes", nodes.toString())
         nodeMeasurement.addMeasurement("Problem nodes", problemNodes.toString())
     }