# -*- coding: utf-8 -*- from random import randint, randrange, choice from time import time from operator import itemgetter import RPi.GPIO as GPIO from pygame import Surface, PixelArray, Rect from pygame.draw import aalines, polygon from pygame.font import Font from pygame.mixer import Sound from pygame.locals import * from lib.pgfw.pgfw.Game import Game from lib.pgfw.pgfw.GameChild import GameChild from lib.pgfw.pgfw.Sprite import Sprite from lib.pgfw.pgfw.extension import render_box class ElectricSieve(Game): # The GPIO pins corresponding to the buttons and LED indicators PIN_BUTTON_UP = 17 PIN_BUTTON_DOWN = 27 PIN_LED_UP = 22 PIN_LED_DOWN = 23 def __init__(self): self.initialize_gpio() Game.__init__(self) self.background = Surface(self.display.screen.get_size()) self.background.fill((255, 80, 190)) self.title.activate() def initialize_gpio(self): """ Set pin numbering mode to GPIO, initialize all buttons to input pullup. """ # Use GPIO numbering GPIO.setmode(GPIO.BCM) # Set all button pins to pullup for pin in self.PIN_BUTTON_UP, self.PIN_BUTTON_DOWN: GPIO.setup(pin, GPIO.IN, pull_up_down=GPIO.PUD_UP) def set_children(self): Game.set_children(self) self.title = Title(self) self.sieve = Sieve(self) self.triangles = Triangles(self) self.acid = Acid(self) self.static = Static(self) def update(self): self.title.update() if self.triangles.active: self.display.screen.blit(self.background, (0, 0)) self.triangles.update() self.sieve.update() self.static.update() class Title(GameChild): def __init__(self, parent): GameChild.__init__(self, parent) self.display_surface = self.get_display_surface() self.delegate = self.parent.delegate bg_color = (255, 222, 173) self.background = surface = Surface(self.display_surface.get_size()) tile = Surface((2, 2)) tile.fill(bg_color) tile.set_at((0, 1), (220, 119, 41)) tile.set_at((1, 0), (220, 119, 41)) for y in range(0, surface.get_height(), 2): for x in range(0, surface.get_width(), 2): surface.blit(tile, (x, y)) # font = Font(self.get_resource("display", "title-font-path"), 20) # font.set_italic(True) # font.set_bold(True) # self.captions = captions = Sprite(self), Sprite(self) # colors = (0, 68, 170), (255, 255, 255), (128, 128, 128), \ # (220, 119, 41), (255, 80, 80), (0, 90, 110) # texts = ["", ""] # for ii, text in \ # enumerate(self.get_configuration("display", # "caption").upper().split()): # texts[ii] += "•" * (5 if ii else 3) # for ch in text: # texts[ii] += ch + " " # texts[ii] = texts[ii].strip() + "•" * (5 if ii else 3) # for _ in range(25): # color = choice(colors) # captions[0].add_frame(font.render(texts[0], True, color, (220, 208, 255))) # captions[1].add_frame(font.render(texts[1], True, color, (220, 208, 255))) # cx = self.display_surface.get_rect().centerx # captions[0].location.center = cx, 301 # captions[1].location.center = cx, 398 self.scoreboard = Scoreboard(self) self.music = Sound(self.get_resource("audio", "title")) self.advance = Sound(self.get_resource("audio", "title-advance")) self.subscribe(self.respond) def respond(self, event): if self.active: if self.delegate.compare(event, "advance"): self.deactivate() self.parent.triangles.activate() self.parent.sieve.activate() self.parent.static.activate() self.advance.play() def activate(self): self.active = True self.music.play(-1) def deactivate(self): self.active = False self.music.fadeout(500) def update(self): if self.active: self.display_surface.blit(self.background, (0, 0)) # for caption in self.captions: # caption.update() self.scoreboard.update() class Strip(Sprite): LEFT, RIGHT = range(2) def __init__(self, parent): Sprite.__init__(self, parent) GPIO.add_event_detect(ElectricSieve.PIN_BUTTON_UP, GPIO.BOTH, self.respond) GPIO.add_event_detect(ElectricSieve.PIN_BUTTON_DOWN, GPIO.BOTH, self.respond) self.deactivate() self.display_surface = self.get_display_surface() self.delegate = self.get_game().delegate self.hshifts = Shift(self, 1, "shift-2"), Shift(self, -1, "shift-2") self.add_frames() self.subscribe(self.respond) def deactivate(self): self.active = False def reset(self): for shift in self.hshifts: shift.reset() def add_frames(self): pass def respond(self, event): if type(event) == int: pressed = "pressed" if GPIO.input(event) == GPIO.LOW else "released" print(f"pin {event} {pressed}") if self.active: move = False if type(event) == int: if event == ElectricSieve.PIN_BUTTON_UP: direction = self.LEFT move = True elif event == ElectricSieve.PIN_BUTTON_DOWN: direction = self.RIGHT move = True active = GPIO.input(event) == GPIO.LOW else: compare = self.delegate.compare if compare(event, "left") or compare(event, "left", True): direction = self.LEFT move = True elif compare(event, "right") or compare(event, "right", True): direction = self.RIGHT move = True active = not event.cancel if move: self.hshifts[direction].active = active def activate(self): self.active = True def update(self): if self.active: for shift in self.hshifts: shift.update() if shift.time: self.move(shift.get_change()) Sprite.update(self) class Shift(GameChild): def __init__(self, parent, direction, nodeset): GameChild.__init__(self, parent) self.direction = direction self.reset() self.timer = self.get_game().time_filter self.nodeset = self.get_game().interpolator.get_nodeset(nodeset) def reset(self): self.active = False self.time = 0 def update(self): least, greatest = self.nodeset[0].x, self.nodeset[-1].x if self.active and self.time < greatest: self.time = min(self.time + self.timer.get_last_frame_duration(), greatest) elif not self.active and self.time > least: self.time = max(self.time - self.timer.get_last_frame_duration(), least) def get_change(self): return self.nodeset.get_y(self.time) * self.direction class Scoreboard(GameChild): BACKGROUND = 255, 255, 255 FOREGROUND = 27, 27, 27 NEW = 27, 27, 27 SPACING = 70 MARGIN = 30 BLINK_INTERVAL = 400 PADDING = 6 BORDER = 1 SCORE_COUNT = 9 def __init__(self, parent): GameChild.__init__(self, parent) ds = self.display_surface = self.get_display_surface() self.scores_path = self.get_resource("score", "path") self.most_recent_score = None self.load() def load(self): self.sprites = sprites = [] font_path = self.get_resource("display", "scoreboard-font-path") sizes = [24] * self.SCORE_COUNT blink = False for ii, score in enumerate(self.get_scores()[:len(sizes)]): font = Font(font_path, sizes[ii]) sprites.append((Sprite(self, self.BLINK_INTERVAL), Sprite(self, self.BLINK_INTERVAL))) score_text = str(score[1]) color = self.BACKGROUND if (self.most_recent_score and not blink and score[1:] == \ self.most_recent_score) else self.FOREGROUND score_plate = font.render(score_text, False, color, self.BACKGROUND) rect = score_plate.get_rect() surface = Surface(rect.inflate((2, 2)).size) surface.fill(self.FOREGROUND) rect.center = surface.get_rect().center surface.blit(score_plate, rect) sprites[ii][1].add_frame(render_box(font, score_text, False, color, self.BACKGROUND, self.FOREGROUND, self.BORDER, self.PADDING)) sprites[ii][0].add_frame(render_box(font, score[2], False, color, self.BACKGROUND, self.FOREGROUND, self.BORDER, self.PADDING)) if self.most_recent_score and not blink and score[1:] == self.most_recent_score: sprites[ii][1].add_frame(render_box(font, score_text, False, self.NEW, self.BACKGROUND, self.FOREGROUND, self.BORDER, self.PADDING)) sprites[ii][0].add_frame(render_box(font, score[2], False, self.NEW, self.BACKGROUND, self.FOREGROUND, self.BORDER, self.PADDING)) blink = True sprites[ii][0].location.left = self.MARGIN sprites[ii][1].location.right = self.get_display_surface().get_rect().right - self.MARGIN y = self.get_display_surface().get_rect().centery + self.SPACING * (ii - len(sizes) / 2) for sprite in sprites[ii]: sprite.location.centery = y def get_scores(self): scores = [] for line in open(self.scores_path, "r"): fields = line.split() scores.append((float(fields[0]), int(fields[1]), fields[2])) scores = sorted(scores, key=itemgetter(0)) return sorted(scores, key=itemgetter(1), reverse=True) def write(self, initials): score = int(round(self.get_game().triangles.score)) fields = str(time()), str(score), initials open(self.scores_path, "a").write(fields[0] + " " + fields[1] + " " + fields[2] + "\n") self.most_recent_score = score, initials self.load() def update(self): for pair in self.sprites: for sprite in pair: sprite.update() class Sieve(Strip): UP, DOWN = range(2) def __init__(self, parent): Strip.__init__(self, parent) self.delegate = self.get_game().delegate self.electric = Electric(self) self.add_location(offset=(self.location.w, 0)) def add_frames(self): bar_locations = [] self.bar_rects = bar_rects = [] x = 0 sh = 30 nodeset = self.get_game().interpolator.get_nodeset("scale") self.bar_w = bar_w = 3 self.gaps = gaps = [] while x < nodeset[-1].x: bar_locations.append(x) bar_rects.append(Rect(x, 0, bar_w, sh)) gaps.append(nodeset.get_y(x, natural=True)) x += gaps[-1] surface = Surface((x, sh)) transparent_color = (255, 0, 255) surface.fill(transparent_color) surface.set_colorkey(transparent_color) frames = surface, surface.copy() colors = (0, 255, 0), (153, 0, 204) for x in bar_locations: bar_rects.append(Rect(x + surface.get_width(), 0, bar_w, sh)) for ii, frame in enumerate(frames): frame.fill(colors[ii], (x, 0, bar_w, sh)) frame.fill(colors[ii - 1], (x + 1, 1, 1, sh - 2)) for frame in frames: self.add_frame(frame) def reset(self): Strip.reset(self) self.location.centerx = self.display_surface.get_rect().centerx self.locations[1].centerx = self.location.centerx + self.location.w def update(self): if self.active: if self.location.right < 0: self.move(self.location.w) if self.locations[1].left > self.display_surface.get_width(): self.move(-self.location.w) for location in self.locations: location.bottom = self.parent.acid.get_top() self.electric.location.centery = self.location.centery + 13 self.electric.update() for rect in self.bar_rects: rect.centery = self.location.centery Strip.update(self) class Electric(Sprite): def __init__(self, parent): Sprite.__init__(self, parent) self.display_surface = self.get_display_surface() self.add_frames() def add_frames(self): surface = Surface((self.display_surface.get_width(), self.parent.location.h - 10)) frames = surface, surface.copy() colors = (255, 255, 0), (100, 89, 213) pixel_arrays = PixelArray(frames[0]), PixelArray(frames[1]) for x in range(len(pixel_arrays[0])): for y in range( len(pixel_arrays[0][0])): pixel_arrays[0][x][y] = colors[(y + x) % 2] pixel_arrays[1][x][y] = colors[(y + x + 1) % 2] for pixels in pixel_arrays: del pixels for frame in frames: self.add_frame(frame) class Triangles(GameChild, list): def __init__(self, parent): GameChild.__init__(self, parent) self.music = Sound(self.get_resource("audio", "triangles")) self.deactivate() self.display_surface = self.get_display_surface() self.delegate = self.get_game().delegate self.booster = Shift(self, 1, "boost") self.hit = Sound(self.get_resource("audio", "hit")) self.miss = Sound(self.get_resource("audio", "miss")) self.reset() self.subscribe(self.respond) def deactivate(self): self.active = False self.music.fadeout(500) def reset(self): list.__init__(self, []) self.streak = 0 self.score = 0 self.booster.reset() def populate(self): if not self: self.append(Triangle(self)) self[-1].location.bottom = 0 self.set_next_gap() while self[-1].location.top > -self.display_surface.get_height(): self.append(Triangle(self)) self[-1].location.bottom = self[-2].location.top - self.next_gap self.set_next_gap() def set_next_gap(self): self.next_gap = randint(500, 800) def respond(self, event): if self.active: compare = self.delegate.compare if compare(event, "down") or compare(event, "down", True): self.booster.active = not event.cancel def get_boost(self): return self.booster.get_change() def activate(self): self.active = True self.music.play(-1, 0, 500) def update(self): if self.active: self.populate() self.booster.update() if self[0].location.collidelist(self.parent.sieve.locations) != -1: sieve = self.parent.sieve if self[0].location.colliderect(sieve.electric.location): self.parent.acid.increase() self.streak += 1 self.score += self.streak ** .8 + \ self.parent.acid.get_volume() * 5 + \ self[0].count self.remove(self[0]) self.hit.play() else: for br in sieve.bar_rects: for tr in self[0].collision_rects: if tr.move((self[0].location.left, 0)).colliderect(br.move((sieve.location.left, 0))): self.remove(self[0]) self.parent.static.increase() self.streak = 0 self.miss.play() break for triangle in self: triangle.update() class Triangle(Sprite): def __init__(self, parent): Sprite.__init__(self, parent) mark = randint(112, 328) sieve = self.parent.parent.sieve gaps = sieve.gaps start = randrange(0, len(gaps)) widths = [gaps[start]] while sum(widths) < mark: widths.append(gaps[(start + len(widths)) % len(gaps)]) surface = Surface((sum(widths), 20)) surface.set_colorkey((0, 0, 0)) x = 0 height = surface.get_height() margin = 26 self.collision_rects = collision_rects = [] for width in widths: x += sieve.bar_w points = (x + margin / 2, height - 2), \ (x + width - margin / 2 - 1, height - 2), \ (x + width / 2.0, 1) polygon(surface, (60, 255, 220), points) collision_rects.append(Rect(points[0], (width - margin - 1, 1))) x += width - sieve.bar_w self.add_frame(surface) self.location.centerx = self.get_display_surface().get_rect().centerx self.count = len(widths) def update(self): self.move(dy=9.5 * self.get_game().acid.get_volume() + 3.8 + \ self.parent.get_boost()) for rect in self.collision_rects: rect.bottom = self.location.bottom Sprite.update(self) class Acid(GameChild): def __init__(self, parent): GameChild.__init__(self, parent) self.display_surface = self.get_display_surface() self.level_r = 80, 320 self.nodeset = self.get_game().interpolator.get_nodeset("volume") self.reset() def reset(self): self.substance = 0 def get_top(self): return self.display_surface.get_height() - self.get_level() def get_level(self): return self.get_volume() * (self.level_r[1] - self.level_r[0]) + \ self.level_r[0] def get_volume(self): return self.nodeset.get_y(self.substance) def increase(self): self.substance += 1 class Static(Sprite): def __init__(self, parent): Sprite.__init__(self, parent, 120) self.noise = Sound(self.get_resource("audio", "noise")) self.end = Sound(self.get_resource("audio", "end")) self.deactivate() self.delegate = self.get_game().delegate self.increaser = Shift(self, 1, "intensity") self.total = Total(self) self.initials = Initials(self) self.reset() self.add_frames() self.subscribe(self.respond) def deactivate(self): self.active = False self.end.fadeout(500) def reset(self): self.complete = False self.intensity = 0 self.noise.set_volume(0) self.increaser.reset() def add_frames(self): surface = Surface(self.get_display_surface().get_size()) frames = surface, surface.copy(), surface.copy(), surface.copy() tiles = [] for _ in range(32): tiles.append(Surface((16, 16))) pixel_arrays = [] for tile in tiles: pixel_arrays.append(PixelArray(tile)) colors = (0, 0, 0), (64, 64, 64), (128, 128, 128), (196, 196, 196), \ (255, 255, 255) for x in range(len(pixel_arrays[0])): for y in range(len(pixel_arrays[0][0])): for pixels in pixel_arrays: pixels[x][y] = choice(colors) for pixels in pixel_arrays: del pixels del pixel_arrays for frame in frames: for y in range(0, frame.get_height(), tiles[0].get_height()): for x in range(0, frame.get_width(), tiles[0].get_width()): frame.blit(choice(tiles), (x, y)) self.add_frame(frame) def respond(self, event): if self.active and self.complete and not self.initials.active: if self.delegate.compare(event, "advance"): if self.get_game().triangles.score > \ self.get_game().title.scoreboard.get_scores()[Scoreboard.SCORE_COUNT - 1][1]: self.total.deactivate() self.initials.activate() else: self.finish(wipe=True) def finish(self, text="---", wipe=False): if wipe: self.parent.title.scoreboard.most_recent_score = None self.parent.title.scoreboard.write(text) self.total.deactivate() self.deactivate() self.reset() self.parent.acid.reset() self.parent.triangles.reset() self.parent.sieve.reset() self.parent.title.activate() def increase(self): self.intensity += self.increaser.get_change() if self.intensity > 1: self.intensity = 1 self.increaser.time += 12000 if self.increaser.time >= self.increaser.nodeset[-1].x + 5000: self.increaser.time = self.increaser.nodeset[-1].x + 5000 def activate(self): self.active = True self.noise.play(-1) def update(self): if self.active: if not self.complete and self.intensity >= .65: self.complete = True self.parent.sieve.deactivate() self.parent.triangles.deactivate() self.set_alpha(255) self.noise.fadeout(6000) self.end.play(-1, 0, 4000) self.total.load() elif not self.complete: self.set_alpha(int(self.intensity * 255)) if self.intensity > 0: self.intensity *= .998 self.increaser.update() self.noise.set_volume(self.intensity) Sprite.update(self) self.total.update() self.initials.update() class Initials(GameChild): LETTER_SIZE = 24 FOREGROUND = 27, 27, 27 BACKGROUND = 255, 255, 255 PADDING = 10 ARROW_MARGIN = 40 ARROW_HEIGHT = 10 def __init__(self, parent): GameChild.__init__(self, parent) self.reset() self.deactivate() self.font = Font(self.get_resource("display", "initials-font"), self.LETTER_SIZE) self.subscribe(self.respond) def reset(self): self.text = "---" self.index = 0 def deactivate(self): self.active = False def respond(self, event): if self.active: compare = self.get_game().delegate.compare if compare(event, "right") or compare(event, "advance"): self.index += 1 if self.index == len(self.text): self.index = 0 if compare(event, "advance"): self.deactivate() self.parent.finish(self.text) elif compare(event, "left"): self.index -= 1 if self.index == -1: self.index = len(self.text) - 1 elif compare(event, "up") or compare(event, "down"): if compare(event, "up"): increment = 1 elif compare(event, "down"): increment = -1 letter = self.text[self.index] if letter == '-': letter = 'A' if increment == 1 else 'Z' else: letter = chr(ord(letter) + increment) if ord(letter) == 91 or ord(letter) == 64: letter = '-' replacement = "" for ii in range(len(self.text)): if ii == self.index: replacement += letter else: replacement += self.text[ii] self.text = replacement def activate(self): self.active = True def update(self): if self.active: ds = self.get_display_surface() for ii, letter in enumerate(self.text): box = render_box(self.font, letter, False, self.FOREGROUND, self.BACKGROUND, self.FOREGROUND, padding=self.PADDING) rect = box.get_rect() rect.centery = ds.get_rect().centery rect.centerx = ii * ds.get_width() / 3 + ds.get_width() / 6 ds.blit(box, rect) if ii == self.index: margin = self.ARROW_MARGIN polygon(ds, (0, 255, 0), ((rect.left, rect.top - margin), (rect.right, rect.top - margin), (rect.centerx, rect.top - margin - self.ARROW_HEIGHT))) polygon(ds, (0, 255, 0), ((rect.left, rect.bottom + margin), (rect.right, rect.bottom + margin), (rect.centerx, rect.bottom + margin + self.ARROW_HEIGHT))) class Total(Sprite): def __init__(self, parent): Sprite.__init__(self, parent, 68) self.deactivate() self.font = Font(self.get_resource("display", "score-font-path"), 72) # self.font.set_italic(True) def deactivate(self): self.active = False def load(self): self.clear_frames() score = "" for ch in str(int(round(self.get_game().triangles.score))): score += ch + " " colors = (255, 255, 180), (180, 255, 255), (255, 180, 255), \ (255, 220, 160), (160, 255, 220), (220, 160, 255) template = Surface((self.display_surface.get_width(), 100)) transparent_color = (255, 0, 255) template.fill(transparent_color) template.set_colorkey(transparent_color) tr = template.get_rect() template.fill((255, 0, 0), (0, 20, tr.w, 1)) template.fill((255, 128, 128), (0, 21, tr.w, 1)) for y in range(22, 78, 2): template.fill((255, 255, 255), (0, y, tr.w, 1)) template.fill((255, 128, 128), (0, 78, tr.w, 1)) template.fill((255, 0, 0), (0, 79, tr.w, 1)) for _ in range(20): # surface = template.copy() surface = Surface(template.get_size(), SRCALPHA) # polygon(surface, choice(colors), ((tr.centerx - 7, 19), # (tr.centerx, 0), # (tr.centerx + 7, 19))) text = self.font.render(score, True, choice(colors)) rect = text.get_rect() rect.center = tr.centerx, tr.centery + 2 surface.blit(text, rect) # polygon(surface, choice(colors), ((tr.centerx - 7, 80), # (tr.centerx, tr.h - 1), # (tr.centerx + 7, 80))) self.add_frame(surface) self.location.center = self.display_surface.get_rect().center self.active = True def update(self): if self.active: Sprite.update(self)