affa101169
Currently steps from the last six minutes do not appear in the data. We can fix this by updating the data before creating the iterator. Signed-off-by: Daniel Thompson <daniel@redfelineninja.org.uk>
164 lines
4.2 KiB
Python
164 lines
4.2 KiB
Python
# SPDX-License-Identifier: LGPL-3.0-or-later
|
|
# Copyright (C) 2020 Daniel Thompson
|
|
|
|
"""Step logger
|
|
~~~~~~~~~~~~~~
|
|
|
|
Capture and record data from the step counter
|
|
"""
|
|
|
|
import array
|
|
import os
|
|
import time
|
|
import wasp
|
|
|
|
from micropython import const
|
|
|
|
TICK_PERIOD = const(6 * 60)
|
|
DUMP_LENGTH = const(5)
|
|
DUMP_PERIOD = const(DUMP_LENGTH * TICK_PERIOD)
|
|
|
|
class StepIterator:
|
|
def __init__(self, fname, data=None):
|
|
self._fname = fname
|
|
self._f = None
|
|
self._d = data
|
|
|
|
def __del__(self):
|
|
self.close()
|
|
|
|
def __iter__(self):
|
|
self.close()
|
|
self._f = open(self._fname, 'rb')
|
|
self._c = 0
|
|
return self
|
|
|
|
def __next__(self):
|
|
self._c += 1
|
|
if self._c > (24*60*60) // TICK_PERIOD:
|
|
raise StopIteration
|
|
|
|
if self._f:
|
|
spl = self._f.read(2)
|
|
if spl:
|
|
return spl[0] + (spl[1] << 8)
|
|
self.close()
|
|
self._i = 0
|
|
|
|
if self._d and self._i < len(self._d):
|
|
i = self._i
|
|
self._i += 1
|
|
return self._d[i]
|
|
|
|
return 0
|
|
|
|
def close(self):
|
|
if self._f:
|
|
self._f.close()
|
|
self._f = None
|
|
|
|
class StepLogger:
|
|
def __init__(self):
|
|
self._data = array.array('H', (0,) * DUMP_LENGTH)
|
|
self._steps = wasp.watch.accel.steps
|
|
|
|
try:
|
|
os.mkdir('logs')
|
|
except:
|
|
pass
|
|
|
|
# Queue a tick
|
|
self._t = int(wasp.watch.rtc.time()) // TICK_PERIOD * TICK_PERIOD
|
|
wasp.system.set_alarm(self._t + TICK_PERIOD, self._tick)
|
|
|
|
def _tick(self):
|
|
"""Capture the current step count in N minute intervals.
|
|
|
|
The samples are queued in a small RAM buffer in order to reduce
|
|
the number of flash access. The data is written out every few hours
|
|
in a binary format ready to be reloaded and graphed when it is
|
|
needed.
|
|
"""
|
|
t = self._t
|
|
|
|
# Work out where we are in the dump period
|
|
i = t % DUMP_PERIOD // TICK_PERIOD
|
|
|
|
# Get the current step count and record it
|
|
steps = wasp.watch.accel.steps
|
|
self._data[i] = steps - self._steps
|
|
self._steps = steps
|
|
|
|
# Queue the next tick
|
|
wasp.system.set_alarm(t + TICK_PERIOD, self._tick)
|
|
self._t += TICK_PERIOD
|
|
|
|
if i < (DUMP_LENGTH-1):
|
|
return
|
|
|
|
# Record the data in the flash
|
|
walltime = time.localtime(t)
|
|
yyyy = walltime[0]
|
|
mm = walltime[1]
|
|
dd = walltime[2]
|
|
|
|
# Find when (in seconds) "today" started
|
|
then = int(time.mktime((yyyy, mm, dd, 0, 0, 0, 0, 0, 0)))
|
|
elapsed = t - then
|
|
|
|
# Work out how dumps we expect to find in today's dumpfile
|
|
dump_num = elapsed // DUMP_PERIOD
|
|
|
|
# Update the log data
|
|
try:
|
|
os.mkdir('logs/' + str(yyyy))
|
|
except:
|
|
pass
|
|
fname = 'logs/{}/{:02d}-{:02d}.steps'.format(yyyy, mm, dd)
|
|
offset = dump_num * DUMP_LENGTH * 2
|
|
try:
|
|
sz = os.stat(fname)[6]
|
|
except:
|
|
sz = 0
|
|
f = open(fname, 'ab')
|
|
# This is a portable (between real Python and MicroPython) way to
|
|
# grow the file to the right size.
|
|
f.seek(min(sz, offset))
|
|
for _ in range(sz, offset, 2):
|
|
f.write(b'\x00\x00')
|
|
f.write(self._data)
|
|
f.close()
|
|
|
|
# Wipe the data
|
|
data = self._data
|
|
for i in range(DUMP_LENGTH):
|
|
data[i] = 0
|
|
|
|
def data(self, t):
|
|
try:
|
|
yyyy = t[0]
|
|
except:
|
|
t = time.localtime(t)
|
|
yyyy = t[0]
|
|
mm = t[1]
|
|
dd = t[2]
|
|
|
|
fname = 'logs/{}/{:02d}-{:02d}.steps'.format(yyyy, mm, dd)
|
|
try:
|
|
os.stat(fname)
|
|
except:
|
|
return None
|
|
|
|
# Record the data in the flash
|
|
now = time.localtime(self._t)
|
|
if now[:3] == t[:3]:
|
|
latest = self._data
|
|
|
|
# Work out where we are in the dump period and update
|
|
# with the latest counts
|
|
i = self._t % DUMP_PERIOD // TICK_PERIOD
|
|
latest[i] = wasp.watch.accel.steps - self._steps
|
|
else:
|
|
latest = None
|
|
|
|
return StepIterator(fname, latest)
|