Add the round_array function and use it in main3d.

diff --git a/src/MRI.hs b/src/MRI.hs
index 8e2b0acb96303b03339e15ae94d487ce972f36e8..510e4c011191f8cc78c99fac464a04175a8ed9af 100644 (file)
--- a/src/MRI.hs
+++ b/src/MRI.hs
@@ -70,11 +70,15 @@ swap_bytes :: (Shape sh) => (RawData sh) -> (RawData sh)
 swap_bytes arr =
     R.force $ R.map flip16 arr
 
+
 bracket_array :: (Shape sh) => (Values sh) -> (RawData sh)
 bracket_array arr =
-  R.force $ R.map f arr
-  where
-    f = bracket
+  R.force $ R.map bracket arr
+
+
+round_array :: (Shape sh) => (Values sh) -> (RawData sh)
+round_array arr =
+  R.force $ R.map round arr
 
 
 flip_y :: RawData3D -> RawData3D

diff --git a/src/Main.hs b/src/Main.hs
index e577351cc257615a47b60ac455cb2376f2527e5c..0afe2f7cc4a27e19f32823b2474f601a00d265f5 100644 (file)
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -15,7 +15,6 @@ in_file = "./data/mri.bin"
 main :: IO ()
 main = main3d
 
-
 main3d :: IO ()
 main3d = do
   (s:_) <- getArgs
@@ -27,7 +26,7 @@ main3d = do
   let arrMRI = R.reshape mri_shape arr'
   let dbl_data = R.force $ R.map fromIntegral arrMRI
   let output = zoom dbl_data zoom_factor
-  let word16_output = bracket_array output
+  let word16_output = round_array output
   write_word16s out_file word16_output