X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FGrid.hs;h=fcdeb5e6246c9b8525424e9486938886e7aa929e;hb=769784e81d7fb70a131cd0523ff41e1cb56f4768;hp=893fe5c360d3f39c73a99ebab26d1320fd798815;hpb=89a8f8f3728c6e8730a37304708312b67bc23274;p=spline3.git diff --git a/src/Grid.hs b/src/Grid.hs index 893fe5c..fcdeb5e 100644 --- a/src/Grid.hs +++ b/src/Grid.hs @@ -27,9 +27,8 @@ import Cube (Cube(Cube), import Examples import FunctionValues import Point (Point) -import PolynomialArray (PolynomialArray) import ScaleFactor -import Tetrahedron (Tetrahedron, c, number, polynomial, v0, v1, v2, v3) +import Tetrahedron (Tetrahedron, c, polynomial, v0, v1, v2, v3) import ThreeDimensional import Values (Values3D, dims, empty3d, zoom_shape) @@ -133,15 +132,15 @@ find_containing_cube g p = {-# INLINE zoom_lookup #-} -zoom_lookup :: Grid -> PolynomialArray -> ScaleFactor -> a -> (R.DIM3 -> Double) -zoom_lookup g polynomials scale_factor _ = - zoom_result g polynomials scale_factor +zoom_lookup :: Grid -> ScaleFactor -> a -> (R.DIM3 -> Double) +zoom_lookup g scale_factor _ = + zoom_result g scale_factor {-# INLINE zoom_result #-} -zoom_result :: Grid -> PolynomialArray -> ScaleFactor -> R.DIM3 -> Double -zoom_result g polynomials (sfx, sfy, sfz) (R.Z R.:. m R.:. n R.:. o) = - (polynomials ! (i, j, k, (number t))) p +zoom_result :: Grid -> ScaleFactor -> R.DIM3 -> Double +zoom_result g (sfx, sfy, sfz) (R.Z R.:. m R.:. n R.:. o) = + f p where offset = (h g)/2 m' = (fromIntegral m) / (fromIntegral sfx) - offset @@ -150,15 +149,14 @@ zoom_result g polynomials (sfx, sfy, sfz) (R.Z R.:. m R.:. n R.:. o) = p = (m', n', o') :: Point cube = find_containing_cube g p -- Figure out i,j,k without importing those functions. - Cube _ i j k _ _ = cube t = find_containing_tetrahedron cube p - + f = polynomial t -zoom :: Grid -> PolynomialArray -> ScaleFactor -> Values3D -zoom g polynomials scale_factor +zoom :: Grid -> ScaleFactor -> Values3D +zoom g scale_factor | xsize == 0 || ysize == 0 || zsize == 0 = empty3d | otherwise = - R.force $ R.traverse arr transExtent (zoom_lookup g polynomials scale_factor) + R.force $ R.traverse arr transExtent (zoom_lookup g scale_factor) where arr = function_values g (xsize, ysize, zsize) = dims arr