X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FMain.hs;h=4040276b6cd5c657d62588370a6f5a3228c6269e;hb=aed1b3cc557b67aca7d8b3259f44715078db94ae;hp=c8ba34554feb27bb2cd4f0f2d16aa36b0f3bd34a;hpb=518b4b43bfd6c7c1d09841ca3ff5eeae089351ea;p=spline3.git diff --git a/src/Main.hs b/src/Main.hs index c8ba345..4040276 100644 --- a/src/Main.hs +++ b/src/Main.hs @@ -9,12 +9,18 @@ import Data.Array.Repa ( import System.Environment (getArgs) +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) -import Grid (make_grid, zoom) mri_shape :: DIM3 mri_shape = (Z :. 256 :. 256 :. 1) + + + main :: IO () main = do args <- getArgs @@ -22,6 +28,15 @@ main = do 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 (8,8,1) + 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