X-Git-Url: http://gitweb.michael.orlitzky.com/?a=blobdiff_plain;f=src%2FGrid.hs;h=fcdeb5e6246c9b8525424e9486938886e7aa929e;hb=1f45b6c0880c08809163d18f5f7b63d773ddcbe7;hp=d31ffabdf2aa122cadacf0cc9d85513f4989033a;hpb=aed1b3cc557b67aca7d8b3259f44715078db94ae;p=spline3.git diff --git a/src/Grid.hs b/src/Grid.hs index d31ffab..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 @@ -365,18 +363,18 @@ test_trilinear9x9x9_reproduced = -- -- Example from before the fix: -- --- > b0 (tetrahedron c 15) p --- -3.4694469519536365e-18 +-- b1 (tetrahedron c 20) (0, 17.5, 0.5) +-- -0.0 -- test_tetrahedra_collision_sensitivity :: Assertion test_tetrahedra_collision_sensitivity = assertTrue "tetrahedron collision tests isn't too sensitive" $ - contains_point t15 p + contains_point t20 p where g = make_grid 1 naturals_1d - cube = cube_at g 0 17 1 - p = (0, 16.75, 0.5) :: Point - t15 = tetrahedron cube 15 + cube = cube_at g 0 18 0 + p = (0, 17.5, 0.5) :: Point + t20 = tetrahedron cube 20 prop_cube_indices_never_go_out_of_bounds :: Grid -> Gen Bool