Issue 76: preserve column order from the columns option

astrolabsoftware · May 14, 2019 · b299cdd · b299cdd
1 parent 7aa656b
commit b299cdd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/scala/com/astrolabsoftware/sparkfits/FitsHduBintable.scala b/src/main/scala/com/astrolabsoftware/sparkfits/FitsHduBintable.scala
@@ -46,8 +46,12 @@ object FitsHduBintable {
       colNames.values.toList.asInstanceOf[List[String]]
     }
 
-    val colPositions = selectedColNames.map(
-      x => getColumnPos(keyValues, x)).toList.sorted
+    val colPositions = if (selectedColumns != null) {
+      selectedColNames.map(x => getColumnPos(keyValues, x)).toList
+    } else {
+      selectedColNames.map(x => getColumnPos(keyValues, x)).toList.sorted
+    }
+
 
     val rowTypes = getColTypes(keyValues)