start implementation of recipe selection
This commit is contained in:
67
frontend/views/recipe_selection.py
Normal file
67
frontend/views/recipe_selection.py
Normal file
@@ -0,0 +1,67 @@
|
||||
from typing import Tuple
|
||||
|
||||
from .base import View
|
||||
from .button_interface import ButtonInterface
|
||||
|
||||
from PIL import ImageDraw, Image
|
||||
|
||||
class RecipeSelection(View, ButtonInterface):
|
||||
|
||||
recipes = [
|
||||
"Recipe 1",
|
||||
"Recipe 2",
|
||||
]
|
||||
|
||||
def __init__(self, parent, im_size, center):
|
||||
self.selected_index = 0
|
||||
super().__init__(parent, im_size, center)
|
||||
|
||||
def _get_visual_recipes(self):
|
||||
recipes = self.recipes + ['+']
|
||||
if len(self.recipes) < 5:
|
||||
return recipes, 0
|
||||
start = max(0, self.selected_index - 2)
|
||||
end = min(len(recipes), start + 5)
|
||||
|
||||
recipes = recipes[start:end]
|
||||
if len(recipes) < 5:
|
||||
recipes += ['+']
|
||||
|
||||
return recipes, start
|
||||
|
||||
def update_weight(self, weight: float) -> Image.Image:
|
||||
im = self.bkg_im.copy()
|
||||
draw = ImageDraw.Draw(im)
|
||||
|
||||
recipes, start = self._get_visual_recipes()
|
||||
for idx, recipe in enumerate(recipes):
|
||||
if idx + start == self.selected_index:
|
||||
draw.rectangle((35, 8 + idx * 20, 130, 25 + idx * 20), fill='black')
|
||||
draw.text((40, 10 + idx * 20), recipe, fill='white')
|
||||
else:
|
||||
draw.text((40, 10 + idx * 20), recipe, fill='black')
|
||||
|
||||
|
||||
return im
|
||||
|
||||
def left_press(self):
|
||||
self.selected_index = (self.selected_index - 1) % (len(self.recipes) + 1)
|
||||
|
||||
def right_press(self):
|
||||
self.selected_index = (self.selected_index + 1) % (len(self.recipes) + 1)
|
||||
|
||||
|
||||
def has_button(self) -> Tuple[bool, bool, bool, bool]:
|
||||
return True, True, True, True
|
||||
|
||||
def render_left_press(self, draw, x, y):
|
||||
draw.regular_polygon((x, y+2, 5), 3, fill='black')
|
||||
|
||||
def render_left_long_press(self, draw, x, y):
|
||||
pass
|
||||
|
||||
def render_right_press(self, draw, x, y):
|
||||
draw.regular_polygon((x, y+4, 5), 3, fill='black', rotation=180)
|
||||
|
||||
def render_right_long_press(self, draw, x, y):
|
||||
pass
|
||||
Reference in New Issue
Block a user