From: Michael Orlitzky <michael@orlitzky.com>
Date: Sun, 10 Jul 2011 15:27:27 +0000 (-0400)
Subject: Add a find_containing_cubes function to the Grid module.
X-Git-Tag: 0.0.1~257
X-Git-Url: https://gitweb.michael.orlitzky.com/?a=commitdiff_plain;h=30bc407abc86ffa9a0eb4b26bdb4890f3ea423d1;p=spline3.git

Add a find_containing_cubes function to the Grid module.
---

diff --git a/src/Grid.hs b/src/Grid.hs
index e45a0f4..1a436ac 100644
--- a/src/Grid.hs
+++ b/src/Grid.hs
@@ -6,6 +6,10 @@ where
 
 import Cube (Cube(Cube))
 import FunctionValues
+import Misc (flatten)
+import Point (Point)
+import ThreeDimensional (contains_point)
+
 
 -- | Our problem is defined on a Grid. The grid size is given by the
 --   positive number h. The function values are the values of the
@@ -58,3 +62,13 @@ cube_at g i j k
     | j >= length ((cubes g) !! i) = Nothing
     | k >= length (((cubes g) !! i) !! j) = Nothing
     | otherwise = Just $ (((cubes g) !! i) !! j) !! k
+
+
+-- | Takes a 'Grid', and returns all 'Cube's belonging to it that
+--   contain the given 'Point'.
+find_containing_cubes :: Grid -> Point -> [Cube]
+find_containing_cubes g p =
+    filter contains_our_point all_cubes
+    where
+      all_cubes = flatten $ cubes g
+      contains_our_point = flip contains_point p