buddy.py

import time

# Each frame below is the whole crab, 16 wide x 10 tall, drawn as an ASCII
# grid that you can read at a glance and edit by hand. The character inside
# each non-transparent cell picks the pen via _PEN:
#
#   B = body    C = claw (body pen)    L = leg (body pen)
#   E = eye     h = heart / accent
#   .  or space = transparent
#
# Adding a new pose = copy a frame and change cells. That's it.

_PEN = {
    "B": "body",
    "C": "body",
    "L": "body",
    "E": "eye",
    "e": "eye",    # half-height eye — thin slit, used for closed / blinking eyes
    "h": "accent",
}

_FRAME_BASE = [
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBEBBBBBBEBB..",   # eyes centred
    "..BBEBBBBBBEBB..",
    "CCBBBBBBBBBBBBCC",   # claws extend sideways
    "CCBBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",   # 4 legs planted
    "...L.L....L.L...",
]

_FRAME_BLINK = [
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",   # eyelid (top half of the eye area)
    "..BeeBBBBBBeeB..",   # two-cell-wide slit, 1 logical px tall
    "CCBBBBBBBBBBBBCC",
    "CCBBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "...L.L....L.L...",
]

_FRAME_LOOK_LEFT = [
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BEBBBBBBEBBB..",   # both eyes shifted 1 col left
    "..BEBBBBBBEBBB..",
    "CCBBBBBBBBBBBBCC",
    "CCBBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "...L.L....L.L...",
]

_FRAME_LOOK_RIGHT = [
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBBEBBBBBBEB..",   # both eyes shifted 1 col right
    "..BBBEBBBBBBEB..",
    "CCBBBBBBBBBBBBCC",
    "CCBBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "...L.L....L.L...",
]

_FRAME_WIDE = [
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBEEBBBBEEBB..",   # 2-wide pupils, moved inward
    "..BBEEBBBBEEBB..",
    "CCBBBBBBBBBBBBCC",
    "CCBBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "...L.L....L.L...",
]

_FRAME_HEART = [
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBhBBBBBBhBB..",   # heart-accent eyes
    "..BBhBBBBBBhBB..",
    "CCBBBBBBBBBBBBCC",
    "CCBBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "...L.L....L.L...",
]

# Right claw raised above the head on the top-right corner, left claw at its
# usual post. Two frames so the tip bobs in and out for a waving-hello motion.
# Taller (12 rows) than the standard 10-row frame — Buddy.draw anchors the
# last 10 rows at the normal body position and lets the overhang rows extend
# above, which is what reads as "arm above the head" rather than beside it.
_FRAME_HAND_A = [
    "............CC..",
    "............CC..",
    "..BBBBBBBBBBBB..", 
    "..BBBBBBBBBBBB..",
    "..BBEBBBBBBEBB..", 
    "..BBEBBBBBBEBB..",
    "CCBBBBBBBBBBBB..", 
    "CCBBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "...L.L....L.L...",
]

_FRAME_HAND_B = [
    "................",
    "............CC..",
    "............CC..",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBEBBBBBBEBB..",
    "..BBEBBBBBBEBB..",
    "CCBBBBBBBBBBBB..",
    "CCBBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
]

# Dance poses: each frame has opposite arm/leg raised for a jig feel.
# Frame A = right arm up + left legs lifted; frame B = mirror.
_FRAME_CELEBRATE_A = [
    "................",   # right claw raised above the head
    "................",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBEBBBBBBEBB..",
    "..BBEBBBBBBEBBCC",
    "..BBBBBBBBBBBBCC",   # both side claws at normal middle position
    "CCBBBBBBBBBBBB..",
    "CCBBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    ".....L......L...",   # left legs lifted (right pair planted)
]

_FRAME_CELEBRATE_B = [
    "................",   # left claw raised above the head
    "................",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBEBBBBBBEBB..",
    "CCBBEBBBBBBEBB..",
    "CCBBBBBBBBBBBB..",
    "..BBBBBBBBBBBBCC",
    "..BBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "...L......L.....",   # right legs lifted (left pair planted)
]

# Busy scuttle: alternate which pair of legs is planted vs lifted.
_FRAME_WALK_A = [
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBEBBBBBBEBB..",
    "..BBEBBBBBBEBB..",
    "CCBBBBBBBBBBBBCC",
    "CCBBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "...L.L..........",   # right pair lifted
]

_FRAME_WALK_B = [
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "..BBEBBBBBBEBB..",
    "..BBEBBBBBBEBB..",
    "CCBBBBBBBBBBBBCC",
    "CCBBBBBBBBBBBBCC",
    "..BBBBBBBBBBBB..",
    "..BBBBBBBBBBBB..",
    "...L.L....L.L...",
    "..........L.L...",   # left pair lifted
]


_PALETTES = {
    "default":   {"body": (195, 100,  70), "eye": ( 30,  30,  30), "accent": (245, 210,  60)},
    "attention": {"body": (230,  90,  70), "eye": ( 30,  30,  30), "accent": (255,  80,  60)},
    "heart":     {"body": (230, 140, 170), "eye": (180,  50,  90), "accent": (255, 150, 180)},
}


class Buddy:
    def __init__(self, display, top_y=16, px=8):
        self.display = display
        self.top_y = top_y
        self.px = px
        self.sprite_w = 16 * px
        self.sprite_h = 10 * px
        w, _ = display.get_bounds()
        self.x = (w - self.sprite_w) // 2

        self._pens = {
            name: {k: display.create_pen(*v) for k, v in rgb.items()}
            for name, rgb in _PALETTES.items()
        }

        # Whole-sprite animation offset (pixels). State handlers set these
        # for hops, shakes and nudges that don't warrant a new frame.
        self.sprite_dx = 0
        self.sprite_dy = 0

    def _pick(self, state):
        """Pick (frame, palette) and set sprite_dx/dy for this tick."""
        self.sprite_dx = 0
        self.sprite_dy = 0
        t = time.ticks_ms()

        if state == "sleep":
            return _FRAME_BLINK, "default"

        if state == "idle":
            # 6 s cycle: blink, hold, glance left, hold, glance right, hold.
            phase = t % 6000
            if phase < 140:
                return _FRAME_BLINK, "default"
            if 1500 <= phase < 1800:
                return _FRAME_LOOK_LEFT, "default"
            if 3000 <= phase < 3300:
                return _FRAME_LOOK_RIGHT, "default"
            return _FRAME_BASE, "default"

        if state == "busy":
            return (_FRAME_WALK_A if (t // 110) & 1 else _FRAME_WALK_B), "default"

        if state == "attention":
            self.sprite_dx = 1 if (t // 110) & 1 else -1
            return _FRAME_WIDE, "attention"

        if state == "celebrate":
            return (_FRAME_CELEBRATE_A if (t // 280) & 1 else _FRAME_CELEBRATE_B), "default"

        if state == "heart":
            return _FRAME_HEART, "heart"

        if state == "hand":
            return (_FRAME_HAND_A if (t // 320) & 1 else _FRAME_HAND_B), "default"

        return _FRAME_BASE, "default"

    def draw(self, state):
        frame, palette_key = self._pick(state)
        pens = self._pens[palette_key]
        d = self.display
        px = self.px
        # Frames taller than the 10-row baseline extend upward: the bottom
        # 10 rows always sit at top_y, and any prepended rows stick up above.
        overhang = max(0, len(frame) - 10)
        ox = self.x + self.sprite_dx
        oy = self.top_y + self.sprite_dy - overhang * px
        slit_h = 4
        slit_dy = (px - slit_h) // 2
        body_pen = pens["body"]
        for r, line in enumerate(frame):
            y = oy + r * px
            for c, ch in enumerate(line):
                pen_key = _PEN.get(ch)
                if pen_key is None:
                    continue
                if ch == "e":
                    # Fill the cell with body first so the non-slit area isn't
                    # the screen background showing through.
                    d.set_pen(body_pen)
                    d.rectangle(ox + c * px, y, px, px)
                    d.set_pen(pens[pen_key])
                    d.rectangle(ox + c * px, y + slit_dy, px, slit_h)
                else:
                    d.set_pen(pens[pen_key])
                    d.rectangle(ox + c * px, y, px, px)