From 02a34bf79286cd63edcc62e8610fe045feb1dd68 Mon Sep 17 00:00:00 2001 From: Michael Orlitzky Date: Fri, 9 Sep 2011 10:22:53 -0400 Subject: [PATCH] Make Main a little more amenable to changes. --- src/Main.hs | 42 +++++++++++++++++++++++++++++++----------- 1 file changed, 31 insertions(+), 11 deletions(-) diff --git a/src/Main.hs b/src/Main.hs index f21856f..5d2a090 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -1,7 +1,7 @@ module Main where -import qualified Data.Array.Repa as R (map, reshape) +import qualified Data.Array.Repa as R (map, force, reshape) import qualified Data.Array.Repa.IO.BMP as R (writeComponentsToBMP) import System.Environment (getArgs) @@ -9,26 +9,46 @@ import Grid (make_grid, zoom) import MRI import Values (drop_z, zoom_shape) -zoom_factor :: (Int, Int, Int) -zoom_factor = (1,2,2) +main :: IO () +main = main2d +in_file :: FilePath +in_file = "./data/mri.bin" -main :: IO () -main = do --- args <- getArgs --- let color = head args - let in_file = "./data/mri.bin" +main3d :: IO () +main3d = do + (s:_) <- getArgs + let scale = read s :: Int + let zoom_factor = (scale, scale, scale) + let out_file = "output.bin" + arr <- read_word16s in_file + let arr' = swap_bytes arr +-- let arrInv = flip_x $ flip_y arr' + let arrMRI = R.reshape mri_shape arr' + let dbl_data = R.force $ R.map fromIntegral arrMRI + let g = make_grid 1 dbl_data + let output = zoom g zoom_factor + let word16_output = bracket_array output + write_word16s out_file word16_output + +main2d :: IO () +main2d = do + (s:_) <- getArgs + let scale = read s :: Int + let zoom_factor = (1, scale, scale) let out_file = "output.bmp" arr <- read_word16s in_file - let arrBrack = bracket_array arr - let arrInv = flip_x $ flip_y arrBrack + let arr' = swap_bytes arr + let arrInv = flip_x $ flip_y arr' let arrSlice = z_slice 50 arrInv let arrSlice' = R.reshape mri_slice3d arrSlice let dbl_data = R.map fromIntegral arrSlice' let g = make_grid 1 dbl_data let output = zoom g zoom_factor + let arrBrack = bracket_array output let mri_slice2d = drop_z $ zoom_shape zoom_factor mri_slice3d - let colors = values_to_colors $ R.reshape mri_slice2d output + let colors = values_to_colors $ R.reshape mri_slice2d + $ R.map fromIntegral arrBrack let routput = R.map (\(red, _, _) -> red) colors let goutput = R.map (\(_, green, _) -> green) colors let boutput = R.map (\(_, _, blue) -> blue) colors -- 2.44.2