import asyncio
from tasks import ATask
from ws.message import MsgIo
from components import ATemperatureSensor
from components.pid import TempController

import numpy as np
import scipy.io


class TracerTask(ATask):
	def __init__(self, sensor: ATemperatureSensor, temp_ctrl: TempController, interval, msg_handler: MsgIo):
		ATask.__init__(self, interval)

		self.msg_handler = msg_handler
		msg_handler.set_recv_handler(self.recv)

		self.sensor = sensor
		self.temp_ctrl = temp_ctrl

	async def recv(self, data):
		pass

	async def send(self, data):
		await self.msg_handler.send(data)

	async def on_process(self):
		print("{}: Started with interval {} s".format(self.msg_handler.get_key(), self.interval))

		time = 0
		_time = np.empty(0)

		while True:
			_time = np.append(_time, time)
			time += 1
			scipy.io.savemat('test.mat', {'time': _time})
			await asyncio.sleep(self.interval)