/
JumpDetector.py
42 lines (32 loc) · 1002 Bytes
/
JumpDetector.py
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
from collections import deque
WINDOW_SIZE = 30
PIXEL_THRESHOLD = 100
class JumpDetector(object):
def __init__(self):
self.window = deque(maxlen=WINDOW_SIZE)
self.enabled = False
self.jump = False
self.checked = False
def activate(self):
self.enabled = True
def deactivate(self):
self.enabled = False
def reset(self):
self.window = deque(maxlen=WINDOW_SIZE)
self.jump = False
self.checked = False
def addFeet(self, y_left, y_right):
if not self.enabled:
return
self.window.append(max(y_left, y_right))
self.jump = self.enabled and (self.window[0] - self.window[-1] > PIXEL_THRESHOLD)
if not self.jump:
self.checked = False
def failedFeet(self):
# Currently unused
self.window.append(self.window[-1])
def hasJumped(self):
res = (not self.checked) and self.jump
if res:
self.checked = True
return res