diff options
Diffstat (limited to 'main.py.example')
| -rw-r--r-- | main.py.example | 46 |
1 files changed, 45 insertions, 1 deletions
diff --git a/main.py.example b/main.py.example index b468d5d..562d4b6 100644 --- a/main.py.example +++ b/main.py.example @@ -108,6 +108,46 @@ def notify(scope, **kwargs): urequests.post(NOTIFICATION_URL, data=data, headers={'Content-Type': 'application/json'}) +def classify(trace, mean): + runs = [] + cur_sign, cur_run = 1, 0 + for x in trace[len(trace)//2:]: + x -= mean + if x > 0 and cur_sign < 0: + runs.append(cur_run) + cur_sign = 1 + cur_run = 0 + elif x < 0 and cur_sign >= 0: + runs.append(cur_run) + cur_sign = -1 + cur_run = 0 + else: + cur_run += 1 + + run_means = [] + for start in range(0, len(runs), 8): + k = 32 + run_means.append(sum(runs[start:start+k])/k) + + bin_low, bin_high = 0, 0 + for e in run_means: + if 0.05 < e < 0.25: + bin_low += 1 + elif 0.30 < e < 0.50: + bin_high += 1 + total = len(run_means) + + bin_low /= total + bin_high /= total + + if 0.3 < bin_low < 0.7 and 0.3 < bin_high < 0.7: + return 'downstairs' + elif bin_low < 0.3 and bin_high > 0.7: + return 'upstairs' + else: + return 'none' + + def format_exc(limit=None, chain=True): return "".join(repr(i) for i in sys.exc_info()) @@ -124,11 +164,15 @@ def loop(): now = unix_time() if (now - last_notification) > NOTIFICATION_COOLDOWN and rms > RMS_THRESHOLD: old_capture = capture + classification = classify(capture, mean) rms = 0 while rms == 0: time.sleep(0.1) rms = 0 - notify('default', rms=rms, capture=[old_capture, capture]) + if classification in ('downstairs', 'upstairs'): + notify('default', classification=classification, rms=rms, capture=[old_capture, capture]) + else: + notify('info', info_msg='Unclassified capture', rms=rms, capture=[old_capture, capture]) last_notification = now if (now - last_heartbeat) > HEARTBEAT_INTERVAL: |