where
import Data.Array.Repa (
- Array,
- DIM1,
DIM3,
- Shape,
Z(..),
(:.)(..),
- index,
- reshape,
- toList
)
-import Data.Array.Repa.IO.Vector (readVectorFromTextFile)
+import System.Environment (getArgs)
---import Grid(make_grid, zoom)
+import Grid (make_grid, zoom)
+import Values (read_values_3d, write_values_1d)
+
+mri_shape :: DIM3
+mri_shape = (Z :. 256 :. 256 :. 1)
-read_mri_data :: IO (Array DIM1 Int)
-read_mri_data = readVectorFromTextFile "./data/mridata.txt"
-mri_shape :: DIM3
-mri_shape = (Z :. 256 :. 256 :. 109)
main :: IO ()
main = do
- mridata <- read_mri_data
- let mridata2 = reshape mri_shape mridata
- let tmp = Data.Array.Repa.toList mridata2
- print tmp
--- let g = make_grid 1 (Data.Array.Repa.toList mridata2)
--- let output = zoom g 2
--- print "Hello, world."
+ args <- getArgs
+ let color = head args
+ let in_file = "./data/MRbrain.40." ++ color
+ let out_file = "MRbrain.40." ++ color ++ ".out"
+ mridata <- read_values_3d mri_shape in_file
+
+ let g = make_grid 1 mridata
+ let output = zoom g (4,4,1)
+ write_values_1d output out_file