From: Michael Orlitzky Date: Wed, 14 Sep 2011 13:31:13 +0000 (-0400) Subject: Add some junk to Main where I am experimenting. X-Git-Tag: 0.0.1~134 X-Git-Url: https://gitweb.michael.orlitzky.com/?a=commitdiff_plain;h=374082b271180b6ffc64d49c334ace155a196d59;p=spline3.git Add some junk to Main where I am experimenting. --- diff --git a/src/Main.hs b/src/Main.hs index 5d2a090..9742a1f 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -1,7 +1,7 @@ module Main where -import qualified Data.Array.Repa as R (map, force, reshape) +import qualified Data.Array.Repa as R import qualified Data.Array.Repa.IO.BMP as R (writeComponentsToBMP) import System.Environment (getArgs) @@ -10,7 +10,30 @@ import MRI import Values (drop_z, zoom_shape) main :: IO () -main = main2d +main = 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 arr' = swap_bytes arr + let arrInv = flip_x $ flip_y arr' + let arrSlice = z_slice3 50 arrInv + 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 + print $ "arrBrack extent:" ++ (show $ R.extent arrBrack) + let arrBrack' = z_slice 1 arrBrack + print $ "arrBrack' extent:" ++ (show $ R.extent arrBrack') + let mri_slice2d = drop_z $ zoom_shape zoom_factor mri_slice3d + 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 + R.writeComponentsToBMP out_file routput goutput boutput + in_file :: FilePath in_file = "./data/mri.bin"