overlay_new.py

from components import (
    TransparentRectangle,
    DataField,
    SpeedField,
    VoltageField,
    CentrePower,
    DAShboardMessage,
    DASDisconnectMessage,
)
from components.logging_indicator import LoggingIndicator
from overlay import Overlay


class OverlayNew(Overlay):
    def __init__(self, bike=None, bg=None):
        super().__init__(bike, bg=bg)

        # Generate coordinates for each of the data fields in the
        # bottom corners.
        # Note: These coordinates are for the bottom left corner of each field.
        spacing = 20
        row_coords = [
            self.height - (2 * spacing + DataField.height),
            self.height - spacing,
            # top right position for voltage
            (spacing + DataField.height),
        ]
        col_coords = [
            spacing,
            2 * spacing + DataField.width,
            self.width - 2 * (spacing + DataField.width),
            self.width - (spacing + DataField.width),
        ]

        def data_field_coord(x, y):
            """Coordinates of the data field in column x, row y."""
            return col_coords[x], row_coords[y]

        # Dimensions of the bottom TransparentRectangle
        bottom_rect = [
            (0, row_coords[0] - DataField.height - spacing),
            (self.width, self.height),
        ]

        # Dimensions of the top right Transparent rectangle
        top_right_rect = [
            (col_coords[3] + spacing, 0),
            (self.width, DataField.height + 2 * spacing),
        ]
        # Create all overlay components
        self.components = [
            TransparentRectangle(*bottom_rect),
            TransparentRectangle(*top_right_rect),
            DataField(
                "RPM", self.get_data_func("cadence"), data_field_coord(0, 0)
            ),
            DataField(
                "BPM", self.get_data_func("heartRate"), data_field_coord(0, 1)
            ),
            SpeedField(data_field_coord(1, 0)),
            DataField("TIME", self.time_func, data_field_coord(1, 1)),
            DataField(
                "REC KPH",
                self.get_data_func("rec_speed", 1),
                data_field_coord(2, 0),
            ),
            DataField(
                "ZONE KM",
                self.get_data_func("zdist", 2, 0.001),
                data_field_coord(2, 1),
            ),
            DataField(
                "MAX KPH",
                self.get_data_func("max_speed_achieved", 1),
                data_field_coord(3, 0),
            ),
            DataField(
                "DIST KM",
                self.get_data_func("ant_distance", 2, 0.001),
                data_field_coord(3, 1),
            ),
            VoltageField(
                "VOLTAGE",
                self.get_data_func("voltage", 2),
                data_field_coord(3, 2),
            ),
            CentrePower(self.width, self.height),
            DAShboardMessage(),
            DASDisconnectMessage(self.client),
            LoggingIndicator(
                (self.width - spacing, top_right_rect[1][1] + spacing)
            ),
        ]

    def _draw_base_layer(self):
        for component in self.components:
            component.draw_base(self.base_canvas)

    def _update_data_layer(self):
        self.data_canvas.clear()

        for component in self.components:
            component.draw_data(self.data_canvas, self.data)


if __name__ == "__main__":
    args = Overlay.get_overlay_args("An empty, example overlay")
    my_overlay = OverlayNew(args.bike, args.bg)
    my_overlay.connect(ip=args.host)