Select Git revision
Forked from
card10 / firmware
Source project has a limited visibility.
-
rahix authoredSigned-off-by:
Rahix <rahix@rahix.de>rahix authoredSigned-off-by:Rahix <rahix@rahix.de>
select.py 6.50 KiB
from st3m.ui.view import BaseView, ViewManager, ViewTransitionSwipeLeft
from st3m.ui.interactions import ScrollController
import media
import math
import os, stat
import flower
import difficulty
import songinfo
import utils
def discover_songs(path: str):
path = path.rstrip("/")
try:
l = os.listdir(path)
except Exception as e:
print(f"Could not discover songs in {path}: {e}")
l = []
songs = []
to_process = []
for d in l:
dirpath = path + "/" + d
st = os.stat(dirpath)
if not stat.S_ISDIR(st[0]):
continue
inipath = dirpath + "/song.ini"
try:
st = os.stat(inipath)
if not stat.S_ISREG(st[0]):
continue
except Exception:
continue
try:
s = songinfo.SongInfo(dirpath)
except Exception as e:
print(f"Failed to read the song from {inipath}: {e}")
continue
inipath = dirpath + "/diffs.pet"
try:
st = os.stat(inipath)
if not stat.S_ISREG(st[0]):
to_process.append(s)
except Exception:
to_process.append(s)
songs.append(s)
return songs, to_process
class SelectView(BaseView):
def __init__(self, app):
super().__init__()
self.app = app
self.flower = flower.Flower(0.001)
self._sc = ScrollController()
self.songs, self.to_process = discover_songs("/sd/PetalHero")
self.total_process = len(self.to_process)
self.processing_now = None
self._sc.set_item_count(len(self.songs))
self._scroll_pos = 0
self.process_delay = 250
self.pos = -1
def draw(self, ctx: Context) -> None:
utils.background(ctx)
ctx.save()
ctx.scale(1.9, 1.9)
ctx.translate(-52, 0)
ctx.rgba(0.1, 0.4, 0.3, 0.42)
self.flower.draw(ctx)
ctx.restore()
if self.processing_now:
self.processing_now.getDifficulties()
self.processing_now.saveDifficulties()
self.processing_now = None
if self.to_process:
if self.process_delay <= 0:
self.processing_now = self.to_process.pop()
utils.fire_gradient(ctx)
ctx.font = "Camp Font 1"
ctx.font_size = 18
ctx.text_align = ctx.CENTER
ctx.text_baseline = ctx.MIDDLE
ctx.move_to (0, -10)
ctx.text("PROCESSING NEW SONGS")
ctx.gray(0.8)
ctx.rectangle(
-120.0,
3.0,
240.0 * (self.total_process - len(self.to_process)) / self.total_process,
10.0,
).fill()
return
ctx.save()
ctx.gray(1.0)
ctx.rectangle(
-120.0,
-15.0,
240.0,
30.0,
).fill()
ctx.translate(0, -30 * self._sc.current_position())
offset = 0
ctx.font = "Camp Font 3"
ctx.text_align = ctx.CENTER
ctx.text_baseline = ctx.MIDDLE
ctx.move_to(0, 0)
if not self.songs:
ctx.gray(0.0)
ctx.font_size = 24
ctx.text("No songs found!")
for idx, song in enumerate(self.songs):
target = idx == self._sc.target_position()
if target:
ctx.gray(0.0)
else:
ctx.gray(1.0)
distance = self._sc.current_position() - idx
if abs(distance) <= 3:
xpos = 0.0
ctx.font_size = 24 - abs(distance) * 3
if target and (width := ctx.text_width(song.name)) > 220:
xpos = math.sin(self._scroll_pos) * (width - 220) / 2
ctx.move_to(xpos, offset + distance * abs(distance) * 2)
ctx.global_alpha = max(0.0, 1.0 - abs(distance) / 2.5)
ctx.text(song.name)
ctx.global_alpha = 1.0
offset += 30
ctx.restore()
ctx.rgba(1.0, 1.0, 1.0, 0.05)
ctx.rectangle(-120, -120, 240, 55)
ctx.fill()
ctx.rectangle(-120, 65, 240, 55)
ctx.fill()
utils.fire_gradient(ctx)
ctx.font = "Camp Font 1"
ctx.font_size = 25
ctx.text_align = ctx.CENTER
ctx.text_baseline = ctx.MIDDLE
ctx.move_to (0, -80)
ctx.text("SELECT SONG")
ctx.font_size = 15
ctx.move_to(0, 78)
ctx.text("Put songs into")
ctx.move_to(0, 94)
ctx.font_size = 15
ctx.gray(0.75)
ctx.text("/sd/PetalHero")
def think(self, ins: InputState, delta_ms: int) -> None:
super().think(ins, delta_ms)
self._sc.think(ins, delta_ms)
if self.process_delay > 0:
self.process_delay -= delta_ms
if not self.to_process and not self.processing_now:
media.think(delta_ms)
self.flower.think(delta_ms)
self._scroll_pos += delta_ms / 1000
cur_target = self._sc.target_position()
if cur_target < 0: cur_target = 0
if cur_target > len(self.songs) - 1: cur_target = len(self.songs) - 1
if self.input.buttons.app.left.pressed:
self._sc.scroll_left()
self._scroll_pos = 0.0
utils.play_crunch(self.app)
elif self.input.buttons.app.right.pressed:
self._sc.scroll_right()
self._scroll_pos = 0.0
utils.play_crunch(self.app)
pos = self._sc.target_position()
if pos < 0: pos = 0
if pos > len(self.songs) - 1: pos = len(self.songs) - 1
if pos != cur_target:
media.load(self.songs[pos].dirName + "/song.mp3")
if self.input.buttons.app.middle.pressed:
utils.play_go(self.app)
if self.songs:
self.vm.push(difficulty.DifficultyView(self.app, self.songs[pos]), ViewTransitionSwipeLeft())
if self.input.buttons.os.middle.pressed:
utils.play_back(self.app)
def on_enter(self, vm: Optional[ViewManager]) -> None:
super().on_enter(vm)
if self.app and self.app.after_score:
self.play()
self.app.after_score = False
def play(self):
if self.songs:
media.load(self.songs[self._sc.target_position()].dirName + "/song.mp3")
else:
media.stop()
def on_exit(self):
super().on_exit()