diff --git a/tasks/tracer.py b/tasks/tracer.py index 937492d..4904f6b 100644 --- a/tasks/tracer.py +++ b/tasks/tracer.py @@ -8,11 +8,14 @@ from components import APid import numpy as np import scipy.io +import os class TracerTask(ATask): - def __init__(self, sensor: ATemperatureSensor, heater: AHeater, temp_ctrl: APid, interval, msg_handler: MsgIo): + def __init__(self, sensor: ATemperatureSensor, heater: AHeater, temp_ctrl: APid, interval, msg_handler: MsgIo, path = './logs'): ATask.__init__(self, interval) + if not os.path.exists(path): + os.makedirs(path) self.msg_handler = msg_handler msg_handler.set_recv_handler(self.recv) @@ -20,6 +23,7 @@ class TracerTask(ATask): self.sensor = sensor self.heater = heater self.temp_ctrl = temp_ctrl + self.path = path async def recv(self, data): pass @@ -40,7 +44,7 @@ class TracerTask(ATask): _tc_dtemp_soll = np.empty(0) _tc_dtemp_commanded = np.empty(0) - filename = "brewpi." + time.strftime("%Y%m%d%H%M%S", time.localtime()) + ".mat" + filename = os.path.join(self.directory, '') + "brewpi." + time.strftime("%Y%m%d%H%M%S", time.localtime()) + ".mat" while True: _timestamp = np.append(_timestamp, timestamp) _sensor_temp = np.append(_sensor_temp, self.sensor.temperature())