sb-py/python-test/state_wait_temp.py

#!/usr/bin/env python

import uasyncio as asyncio
from poll import set_target_temp, get_current_temp

def draw_graph(lcd, min, val, max):
    if max == min:
        lcd.text("{} -> {} -> {}".format(min, val, max), 0, int(lcd.height / 2) - 5, lcd.white)
        return

    w = lcd.width - 10
    ratio = (val - min) / (max - min)

    #wfull = int(w * ratio)
    #wempty = w - wfull
    #lcd.rect(4, int(lcd.height / 2) - 5, wfull + 1, 50, lcd.green, True)
    #lcd.rect(4 + wfull, int(lcd.height / 2) - 5, wempty + 2, 50, lcd.green, False)

    lcd.pie(lcd.width / 2, lcd.height / 2, w, lcd.red, lcd.green, ratio)

    lcd.text("{}".format(val), int(lcd.width / 2), 125, lcd.white)

class StateWaitTemp:
    def __init__(self, lcd):
        self.lcd = lcd

        self.lock = asyncio.Lock()

    def enter(self, val = None):
        self.value = val
        self.poller = asyncio.create_task(self.poll())
        self.temp = 0.0
        self.min = 0.0
        self.max = 100.0

    def exit(self):
        self.poller.cancel()

        if self.lock.locked():
            self.lock.release()

        return (self.value[0], self.value[1], self.value[2])

    async def poll(self):
        device, workflow, index = self.value
        async with self.lock:
            self.temp = 0.0
            self.min = 0.0
            self.max = workflow["steps"][index][0]

        temp = await get_current_temp(device)
        print("initial temp: {}".format(temp))
        async with self.lock:
            self.temp = temp
            self.min = temp

        print("Setting temperature: {}".format(self.max))
        await set_target_temp(device, self.max)

        while temp < self.max:
            temp = await get_current_temp(device)
            print("now at {}".format(temp))
            async with self.lock:
                self.temp = temp

    async def draw(self):
        device, workflow, index = self.value
        self.lcd.text("Running Workflow - Heat {}".format(workflow["steps"][index][0]), 0, 10, self.lcd.red)

        keys = self.lcd.buttons()

        if keys.once("y"):
            print("user abort")
            return 4 # heat off

        async with self.lock:
            if self.temp == 0.0:
                self.lcd.text("Setting temperature...", 0, int(self.lcd.height / 2) - 5, self.lcd.white)
            else:
                draw_graph(self.lcd, self.min, self.temp, self.max)

            if self.temp >= self.max:
                print("switch, {} >= {}".format(self.temp, self.max))
                return 7 # wait for time

        return -1 # stay in this state