- [[[f p | i <- [0..scaled_zsize],
- let i' = scale_dimension i,
- let j' = scale_dimension j,
- let k' = scale_dimension k,
- let p = (i', j', k') :: Point,
- let c = (find_containing_cubes g p) !! 0,
- let t = (find_containing_tetrahedra c p) !! 0,
- let f = polynomial t]
- | j <- [0..scaled_ysize]]
- | k <- [0..scaled_xsize]]
- where
- scale_dimension :: Int -> Double
- scale_dimension x = (fromIntegral x) / (fromIntegral scale_factor)
-
- fvs = function_values g
- (xsize, ysize, zsize) = dims fvs
- scaled_xsize = xsize * scale_factor
- scaled_ysize = ysize * scale_factor
- scaled_zsize = zsize * scale_factor
-
+ R.traverse arr transExtent (\_ -> newlookup)
+ where
+ fvs = function_values g
+ (xsize, ysize, zsize) = dims fvs
+ arr = fvs
+ transExtent = zoom_shape scale_factor
+ newlookup :: R.DIM3 -> Double
+ newlookup (R.Z R.:. i R.:. j R.:. k) =
+ f p
+ where
+ i' = fromIntegral i
+ j' = fromIntegral j
+ k' = fromIntegral k
+ p = (i', j', k') :: Point
+ c = head (find_containing_cubes g p)
+ t = head (find_containing_tetrahedra c p)
+ f = polynomial t