src/Main.hs

   1 module Main
   2 where
   3
   4 import Data.Array.Repa (
   5   DIM3,
   6   Z(..),
   7   (:.)(..),
   8   )
   9
  10 import System.Environment (getArgs)
  11
  12 import Grid (make_grid, zoom)
  13 import Values (read_values_3d, write_values_1d)
  14
  15 mri_shape :: DIM3
  16 mri_shape = (Z :. 256 :. 256 :. 1)
  17
  18
  19
  20
  21 main :: IO ()
  22 main = do
  23   args <- getArgs
  24   let color = head args
  25   let in_file  = "./data/MRbrain.40." ++ color
  26   let out_file = "MRbrain.40." ++ color ++ ".out"
  27   mridata <- read_values_3d mri_shape in_file
  28
  29   let g = make_grid 1 mridata
  30   let output = zoom g (4,4,1)
  31   write_values_1d output out_file