Merge pull request #50 from aesee/feature

Collision Detection  : Broad Phase algorithm

Author: OmkarPathak

pygorithm/geometry/__init__.py

@@ -0,0 +1,8 @@
+"""
+Collection of special geometry functions
+"""
+from . import rect_broad_phase
+
+__all__ = [
+    'collision_detection.py'
+]

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

pygorithm/geometry/rect_broad_phase.py

@@ -0,0 +1,61 @@
+"""
+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 SimpleRectangle:
+    """SimpleRectangle
+    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(simpleRect1, simpleRect2):
+    """
+    :type simpleRect1: object
+    :type simpleRect2: object
+    """
+    d1x = simpleRect2.min_x - simpleRect1.max_x
+    d1y = simpleRect2.min_y - simpleRect1.max_y
+    d2x = simpleRect1.min_x - simpleRect2.max_x
+    d2y = simpleRect1.min_y - simpleRect2.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)

tests/test_geometry.py

@@ -0,0 +1,32 @@
+import unittest
+
+from pygorithm.geometry import (
+    rect_broad_phase
+                )
+
+
+class TestCollisionDetection(unittest.TestCase):
+    def setUp(self):
+        # first pair of objects
+        self.coord1 = rect_broad_phase.Coord(1, 1)
+        self.coord2 = rect_broad_phase.Coord(6, 8)
+        self.simpleRect1 = rect_broad_phase.SimpleRectangle(self.coord1, self.coord2)
+        self.coord3 = rect_broad_phase.Coord(4, 0)
+        self.coord4 = rect_broad_phase.Coord(7, 4)
+        self.simpleRect2 = rect_broad_phase.SimpleRectangle(self.coord3, self.coord4)
+        # second pair
+        self.coord1 = rect_broad_phase.Coord(1, 1)
+        self.coord2 = rect_broad_phase.Coord(2, 3)
+        self.simpleRect3 = rect_broad_phase.SimpleRectangle(self.coord1, self.coord2)
+        self.coord3 = rect_broad_phase.Coord(4, 3)
+        self.coord4 = rect_broad_phase.Coord(7, 8)
+        self.simpleRect4 = rect_broad_phase.SimpleRectangle(self.coord3, self.coord4)
+
+
+class TestBroadPhase(TestCollisionDetection):
+    def test_collision_detect(self):
+        self.assertTrue(rect_broad_phase.broad_phase(self.simpleRect1, self.simpleRect2))
+        self.assertFalse(rect_broad_phase.broad_phase(self.simpleRect3, self.simpleRect4))
+
+if __name__ == '__main__':
+    unittest.main()