X-Git-Url: http://gitweb.michael.orlitzky.com/?p=spline3.git;a=blobdiff_plain;f=src%2FMain.hs;h=6848ff86264c3fc461e554cc2067cd6f312db2d4;hp=5e10d6cfdc16146d7d8844de335558253c6ef5b2;hb=0c411010bfcb83bc010d506d30dbac35a0082ae4;hpb=54b1f7c4c6b917b6e507923daeae3b8b6ef8798d

diff --git a/src/Main.hs b/src/Main.hs
index 5e10d6c..6848ff8 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -89,8 +89,11 @@ main3d Args{..} shape = do
   let arr_shaped  = R.reshape shape arr_swapped
   dbl_data <- R.computeUnboxedP $ R.map fromIntegral arr_shaped
   raw_output <- zoom dbl_data zoom_factor
-  word16_output <- R.computeUnboxedP $ round_array raw_output
-  write_word16s output word16_output
+  let word16_output = round_array raw_output
+  -- Switch the bytes order back to what it was. This lets us use the
+  -- same program to view the input/output data.
+  swapped_output <- R.computeUnboxedP $ swap_bytes word16_output
+  write_word16s output swapped_output
 
 
 main2d :: Args -> R.DIM3 -> IO ()