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 Cube (tetrahedron)
+import Grid (cube_at, make_grid, zoom)
+import PolynomialArray (make_polynomial_array)
+import Tetrahedron (polynomial)
+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 polynomials = make_polynomial_array (255,255,0,23)
+ [ ((i,j,k,tet), polynomial t) | i <- [0..255],
+ j <- [0..255],
+ k <- [0],
+ tet <- [0..23],
+ let c = cube_at g i j k,
+ let t = tetrahedron c tet ]
+
+ let output = zoom g polynomials (8,8,1)
+ write_values_1d output out_file