Detection Collision : Broad Phase algorithm

aesee · aesee · commit 8015885c2b85 · 2017-08-24T00:34:28.000+06:00
diff --git a/pygorithm/geometry/__init__.py b/pygorithm/geometry/__init__.py
@@ -0,0 +1,8 @@
+"""
+Collection of special geometry functions
+"""
+from . import collision_detection
+
+__all__ = [
+    'collision_detection.py'
+]
diff --git a/pygorithm/geometry/collision_detection.py b/pygorithm/geometry/collision_detection.py
@@ -0,0 +1,69 @@
+"""
+Author: ALEXEY SARAPULOV
+Created On: 23 August 2017
+"""
+
+# To test if two rectangle intersect, we only have to find out
+# if their projections intersect on all of the coordinate axes
+
+import inspect
+
+class Coord:
+    """Coord
+    Class to initialize Coordinate of one point
+    """
+
+    def __init__(self, x, y):
+        self.x = x
+        self.y = y
+
+
+class Body:
+    """Body
+    Class to initialize Body of Object
+    """
+
+    def __init__(self, coord1, coord2):
+        """
+        :type coord1: object of class Coord
+        :type coord2: object of class Coord
+        """
+        self.min_x = coord1.x
+        self.min_y = coord1.y
+        self.max_x = coord2.x
+        self.max_y = coord2.y
+
+
+def broad_phase(body1, body2):
+    """
+    :type body1: object
+    :type body2: object
+    """
+    d1x = body2.min_x - body1.max_x
+    d1y = body2.min_y - body1.max_y
+    d2x = body1.min_x - body2.max_x
+    d2y = body1.min_y - body2.max_y
+
+    if d1x > 0 or d1y > 0:
+        return False
+
+    if d2x > 0 or d2y > 0:
+        return False
+
+    return True
+
+
+def get_code():
+    """
+    returns the code for the broad phase function
+    """
+    return inspect.getsource(broad_phase)
+
+if __name__ == '__main__':
+    coord1 = Coord(1, 1)
+    coord2 = Coord(6, 8)
+    body1 = Body(coord1, coord2)
+    coord3 = Coord(4, 0)
+    coord4 = Coord(7, 4)
+    body2 = Body(coord3, coord4)
+    print(broad_phase(body1, body2))
diff --git a/pygorithm/geometry/modules.py b/pygorithm/geometry/modules.py
@@ -0,0 +1,14 @@
+"""
+List all modules
+"""
+import pkgutil
+
+def modules():
+    import pygorithm.geometry
+    package = pygorithm.geometry
+    modules = []
+    for importer, modname, ispkg in pkgutil.iter_modules(package.__path__):
+        modules.append(modname)
+    modules.remove('modules')
+    modules.sort()
+    return modules
diff --git a/tests/test_geometry.py b/tests/test_geometry.py
@@ -0,0 +1,32 @@
+import unittest
+
+from pygorithm.geometry import (
+    collision_detection
+                )
+
+
+class TestCollisionDetection(unittest.TestCase):
+    def setUp(self):
+        # first pair of objects
+        self.coord1 = collision_detection.Coord(1, 1)
+        self.coord2 = collision_detection.Coord(6, 8)
+        self.body1 = collision_detection.Body(self.coord1, self.coord2)
+        self.coord3 = collision_detection.Coord(4, 0)
+        self.coord4 = collision_detection.Coord(7, 4)
+        self.body2 = collision_detection.Body(self.coord3, self.coord4)
+        # second pair
+        self.coord1 = collision_detection.Coord(1, 1)
+        self.coord2 = collision_detection.Coord(2, 3)
+        self.body3 = collision_detection.Body(self.coord1, self.coord2)
+        self.coord3 = collision_detection.Coord(4, 3)
+        self.coord4 = collision_detection.Coord(7, 8)
+        self.body4 = collision_detection.Body(self.coord3, self.coord4)
+
+
+class TestBroadPhase(TestCollisionDetection):
+    def test_collision_detect(self):
+        self.assertTrue(collision_detection.broad_phase(self.body1, self.body2))
+        self.assertFalse(collision_detection.broad_phase(self.body3, self.body4))
+
+if __name__ == '__main__':
+    unittest.main()
