Vis-gui: use dataset from CWD by default

Author: ksmirenko

visualizer-gui/src/main/kotlin/io/github/ksmirenko/kotlin/visualizerGui/model/Model.kt

@@ -17,11 +17,15 @@ object Model {
 
     private var anomalyIterator: Iterator<Anomaly>? = null
     private var curAnomaly: Anomaly? = null
-    private var outFileName: String? = null
+    private var outFile: File? = null
     private var outCsv: CSVPrinter? = null
 
+    fun openDefaultDataset() {
+        openDataset(File("anomalies_grouped"))
+    }
+
     fun openDataset(repoRoot: File) {
-        val dirs = repoRoot.listFiles() ?: return
+        val dirs = repoRoot.listFiles()?.sorted() ?: return
 
         val anomalySequence = buildSequence {
             for (dir in dirs) {
@@ -30,7 +34,7 @@ object Model {
                 }
 
                 val categoryName = dir.name
-                for (file in dir.listFiles()) {
+                for (file in dir.listFiles().sorted()) {
                     if (file.extension != "kt") {
                         continue
                     }
@@ -52,8 +56,8 @@ object Model {
 
         outCsv?.close()
         val timestamp = SimpleDateFormat("MM-dd-HH-mm").format(Date())
-        outFileName = "marked_$timestamp.csv"
-        outCsv = CSVFormat.EXCEL.print(File(outFileName), Charsets.UTF_8)
+        outFile = File("marked_$timestamp.csv")
+        outCsv = CSVFormat.EXCEL.print(outFile, Charsets.UTF_8)
     }
 
     fun processUserResponse(response: UserResponse) {
@@ -117,11 +121,12 @@ object Model {
         outCsv!!.close()
 
         information("Finished dataset!",
-                "You are done! Please send $outFileName to the code anomaly researchers.")
+                "You are done! Please send the file \"${outFile!!.path}\"" +
+                        " (in current working directory) to the code anomaly researchers.")
 
         anomalyIterator = null
         curAnomaly = null
-        outFileName = null
+        outFile = null
         outCsv = null
     }
 

visualizer-gui/src/main/kotlin/io/github/ksmirenko/kotlin/visualizerGui/view/MainWindow.kt

@@ -24,11 +24,13 @@ class MainWindow : View("Code anomaly visualizer GUI") {
         root += buttonRow.root
 
         root.requestFocus()
+
+        Model.openDefaultDataset()
     }
 }
 
 class SelectDatasetButton : View() {
-    override val root = button("SELECT DATASET") {
+    override val root = button("CHANGE DATASET") {
         vboxConstraints {
             marginTopBottom(R.MARGIN_SMALL)
             marginLeftRight(R.MARGIN_SMALL)