DS2-Project/SpatialHashing.py at master · Rayyan98/DS2-Project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from Collision_detection import Collision_Detection
from MyScreen import MyScreen

class Collision_Detection_Spatial_hashing(Collision_Detection):
    def __init__(self):
        self.Xboxes = MyScreen.width//100
        self.Yboxes = MyScreen.height//100

    def Hash(self, x,y):
        X = x//100
        Y = y//100

        h = X + Y * self.Xboxes
        return h


    def BroadPhase(self,rects):
        self.lst = [set() for i in range(self.Xboxes * self.Yboxes)]
        for r in rects:
            self.lst[self.Hash(r.x, r.y)].add(r)
            try:
                self.lst[self.Hash(r.x + r.w, r.y)].add(r)
            except:
                self.lst[self.Hash(r.x + r.w - 1, r.y)].add(r)
            try:
                self.lst[self.Hash(r.x, r.y + r.h)].add(r)
            except:
                self.lst[self.Hash(r.x, r.y + r.h - 1)].add(r)
            try:
                self.lst[self.Hash(r.x + r.w, r.y + r.h)].add(r)
            except:
                try:
                    self.lst[self.Hash(r.x + r.w - 1, r.y + r.h)].add(r)
                except:
                    try:
                        self.lst[self.Hash(r.x + r.w , r.y + r.h - 1)].add(r)
                    except:
                        self.lst[self.Hash(r.x + r.w - 1, r.y + r.h - 1)].add(r)


    def CheckCollisions(self, rects):
        self.BroadPhase(rects)
        S = set()
        for i in self.lst:
            while len(i) != 0:
                p = i.pop()
                for j in i:
                    if self.takra(p,j) == True:
                        S.add(frozenset([p,j]))
        return S