fix: require predictable iteration order for extractors in CSVExporter,

fixes #1002
AlchemistSimulator · Dec 28, 2021 · 2d505c1 · 2d505c1
1 parent d9e30c1
commit 2d505c1
Showing 1 changed file with 22 additions and 4 deletions.
diff --git a/alchemist-loading/src/main/kotlin/it/unibo/alchemist/loader/export/exporters/CSVExporter.kt b/alchemist-loading/src/main/kotlin/it/unibo/alchemist/loader/export/exporters/CSVExporter.kt
@@ -21,6 +21,7 @@ import java.io.PrintStream
 import java.text.SimpleDateFormat
 import java.util.Date
 import java.util.Locale
+import java.util.SortedMap
 import java.util.TimeZone
 import kotlin.io.path.absolutePathString
 import kotlin.io.path.createTempDirectory
@@ -83,10 +84,27 @@ class CSVExporter<T, P : Position<P>> @JvmOverloads constructor(
 
     override fun exportData(environment: Environment<T, P>, reaction: Reaction<T>?, time: Time, step: Long) {
         with(outputPrintStream) {
-            dataExtractors.forEach {
-                it.extractData(environment, reaction, time, step).values.forEach { value ->
-                    print(value)
-                    print(' ')
+            dataExtractors.forEach { extractor ->
+                val data = extractor.extractData(environment, reaction, time, step)
+                val names = extractor.columnNames
+                when {
+                    data.size <= 1 -> data.values.forEach { print("$it ") }
+                    // Labels and keys match
+                    data.size == names.size && data.keys.containsAll(names) -> names.forEach {
+                        print(requireNotNull(data[it]) { "Bug in ${this::class.simpleName}" })
+                        print(' ')
+                    }
+                    // If the labels do not match keys, require predictable iteration order
+                    else -> {
+                        require(data is LinkedHashMap || data is SortedMap) {
+                            """
+                            Extractor "${extractor::class.simpleName}" is likely bugged:
+                            1. the set of labels $names does not match the keys ${data.keys}, but iteration may fail as
+                            2. it returned a map with non-predictable iteration order of type ${data::class.simpleName}"
+                            """.trimIndent()
+                        }
+                        data.values.forEach { print("$it ") }
+                    }
                 }
             }
             println()