where
import Data.Array.Repa (
- Array,
- DIM1,
+ DIM2,
DIM3,
- Shape,
Z(..),
(:.)(..),
- index,
+ slice,
reshape,
- toList
+ Any(..),
+ All(..)
)
+import qualified Data.Array.Repa as R (map)
-import Data.Array.Repa.IO.Vector (readVectorFromTextFile)
+import Data.Array.Repa.IO.BMP (writeComponentsToBMP)
+import Data.Word
+import System.Environment (getArgs)
---import Grid(make_grid, zoom)
+import Grid (make_grid, zoom)
+import MRI
+import Values (read_values_3d, write_values_1d)
+mri_shape2d :: DIM2
+mri_shape2d = (Z :. 256*2 :. 256*2)
-read_mri_data :: IO (Array DIM1 Int)
-read_mri_data = readVectorFromTextFile "./data/mridata.txt"
-
-mri_shape :: DIM3
-mri_shape = (Z :. 256 :. 256 :. 109)
+mri_shape3d :: DIM3
+mri_shape3d = (Z :. 256 :. 256 :. 1)
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.50.red.out"
+ arr <- read_word16s
+ let arrBrack = bracket_array arr
+ let arrInv = flip_x $ flip_y arrBrack
+ let arrSlice = slice arrInv (Any :. (50 :: Int) :. All :. All)
+ let arrColor = raw_data_to_color arrSlice
+
+ let arrColor' = reshape mri_shape3d arrColor
+ let rdata = red_dbl_data arrColor'
+ let gdata = green_dbl_data arrColor'
+ let bdata = blue_dbl_data arrColor'
+
+-- mridata <- read_values_3d mri_shape in_file
+
+ let gr = make_grid 1 rdata
+ let routput = zoom gr (2,2,1)
+
+ let gg = make_grid 1 gdata
+ let goutput = zoom gg (2,2,1)
+
+ let gb = make_grid 1 bdata
+ let boutput = zoom gb (2,2,1)
+
+ let routput' = R.map double_to_word8 (reshape mri_shape2d routput)
+ let goutput' = R.map double_to_word8 (reshape mri_shape2d goutput)
+ let boutput' = R.map double_to_word8 (reshape mri_shape2d boutput)
+
+-- write_values_1d output out_file
+ writeComponentsToBMP out_file routput' goutput' boutput'
+ where
+ double_to_word8 :: Double -> Word8
+ double_to_word8 x =
+ if x > 255 then
+ 255 :: Word8
+ else
+ if x < 0 then
+ 0 :: Word8
+ else
+ fromIntegral $ truncate x