X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FMain.hs;h=f21856f41e0879189c10ebe84aeb550ee4688bb5;hb=f3d720f107f74b07a5f9f96e6a7c9c7a9022aeb0;hp=a2d6b1ffb2411eb37ca02c1a2fc0fb719236c711;hpb=ec2c118022bc46a191ebc323f6be17d4f017fcfb;p=spline3.git

diff --git a/src/Main.hs b/src/Main.hs
index a2d6b1f..f21856f 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -1,55 +1,35 @@
 module Main
 where
 
---import Cube
---import Face
---import Grid
---import Misc (flatten)
---import Point
---import RealFunction
---import Tetrahedron
---import ThreeDimensional
+import qualified Data.Array.Repa as R (map, reshape)
+import qualified Data.Array.Repa.IO.BMP as R (writeComponentsToBMP)
+import System.Environment (getArgs)
 
+import Grid (make_grid, zoom)
+import MRI
+import Values (drop_z, zoom_shape)
 
+zoom_factor :: (Int, Int, Int)
+zoom_factor = (1,2,2)
 
---find_point_value :: RealFunction Point
---find_point_value p = poly p
---    where
---      g0 = make_grid 1 trilinear
---      the_cubes = flatten (cubes g0)
---      good_cubes = filter ((flip contains_point) p) the_cubes
---       target_cube = head good_cubes
---       good_tets = filter ((flip contains_point) p) (tetrahedrons target_cube)
---       target_tetrahedron = head good_tets
---       poly = polynomial target_tetrahedron
 
 main :: IO ()
 main = do
-  putStrLn "Hello, World."
-  -- print $ find_point_value (0,0,0)
-  -- print $ find_point_value (1,0,0)
-  -- print $ find_point_value (2,0,0)
-  -- print $ find_point_value (0,1,0)
-  -- print $ find_point_value (1,1,0)
-  -- print $ find_point_value (2,1,0)
-  -- print $ find_point_value (0,2,0)
-  -- print $ find_point_value (1,2,0)
-  -- print $ find_point_value (2,2,0)
-  -- print $ find_point_value (0,0,1)
-  -- print $ find_point_value (1,0,1)
-  -- print $ find_point_value (2,0,1)
-  -- print $ find_point_value (0,1,1)
-  -- print $ find_point_value (1,1,1)
-  -- print $ find_point_value (2,1,1)
-  -- print $ find_point_value (0,2,1)
-  -- print $ find_point_value (1,2,1)
-  -- print $ find_point_value (2,2,1)
-  -- print $ find_point_value (0,0,2)
-  -- print $ find_point_value (1,0,2)
-  -- print $ find_point_value (2,0,2)
-  -- print $ find_point_value (0,1,2)
-  -- print $ find_point_value (1,1,2)
-  -- print $ find_point_value (2,1,2)
-  -- print $ find_point_value (0,2,2)
-  -- print $ find_point_value (1,2,2)
-  -- print $ find_point_value (2,2,2)
+--  args <- getArgs
+--  let color = head args
+  let in_file = "./data/mri.bin"
+  let out_file = "output.bmp"
+  arr <- read_word16s in_file
+  let arrBrack = bracket_array arr
+  let arrInv   = flip_x $ flip_y arrBrack
+  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 mri_slice2d = drop_z $ zoom_shape zoom_factor mri_slice3d
+  let colors = values_to_colors $ R.reshape mri_slice2d output
+  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