mirror of
https://github.com/Yonokid/PyTaiko.git
synced 2026-02-04 19:50:12 +01:00
198 lines
8.6 KiB
Python
198 lines
8.6 KiB
Python
import logging
|
|
|
|
from libs.audio import audio
|
|
from libs.file_navigator import SongBox, SongFile
|
|
from libs.global_data import PlayerNum
|
|
from libs.transition import Transition
|
|
from libs.utils import get_current_ms, global_data
|
|
from scenes.song_select import (
|
|
DiffSortSelect,
|
|
SongSelectPlayer,
|
|
SongSelectScreen,
|
|
State,
|
|
)
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class TwoPlayerSongSelectScreen(SongSelectScreen):
|
|
def on_screen_start(self):
|
|
super().on_screen_start()
|
|
self.player_1 = SongSelectPlayer(PlayerNum.P1, self.text_fade_in)
|
|
self.player_2 = SongSelectPlayer(PlayerNum.P2, self.text_fade_in)
|
|
|
|
def finalize_song(self, current_item):
|
|
global_data.session_data[PlayerNum.P1].selected_song = current_item.path
|
|
global_data.session_data[PlayerNum.P1].genre_index = current_item.box.genre_index
|
|
logger.info(f"Finalized song selection: {global_data.session_data[PlayerNum.P1].selected_song}")
|
|
|
|
def handle_input(self):
|
|
if self.player_1.is_ready:
|
|
self.player_2.handle_input(self.state, self)
|
|
else:
|
|
self.player_1.handle_input(self.state, self)
|
|
|
|
def handle_input_browsing(self):
|
|
"""Handle input for browsing songs."""
|
|
action = self.player_1.handle_input_browsing(self.last_moved, self.navigator.items[self.navigator.selected_index] if self.navigator.items else None)
|
|
if action is None:
|
|
action = self.player_2.handle_input_browsing(self.last_moved, self.navigator.items[self.navigator.selected_index] if self.navigator.items else None)
|
|
if action is None:
|
|
return
|
|
|
|
current_time = get_current_ms()
|
|
if action == "skip_left":
|
|
self.reset_demo_music()
|
|
for _ in range(10):
|
|
self.navigator.navigate_left()
|
|
self.last_moved = current_time
|
|
elif action == "skip_right":
|
|
self.reset_demo_music()
|
|
for _ in range(10):
|
|
self.navigator.navigate_right()
|
|
self.last_moved = current_time
|
|
elif action == "navigate_left":
|
|
self.reset_demo_music()
|
|
self.navigator.navigate_left()
|
|
self.last_moved = current_time
|
|
elif action == "navigate_right":
|
|
self.reset_demo_music()
|
|
self.navigator.navigate_right()
|
|
self.last_moved = current_time
|
|
elif action == "go_back":
|
|
self.navigator.go_back()
|
|
elif action == "diff_sort":
|
|
self.state = State.DIFF_SORTING
|
|
self.diff_sort_selector = DiffSortSelect(self.navigator.diff_sort_statistics, self.navigator.diff_sort_diff, self.navigator.diff_sort_level)
|
|
self.text_fade_in.start()
|
|
self.text_fade_out.start()
|
|
elif action == "select_song":
|
|
current_song = self.navigator.get_current_item()
|
|
if not isinstance(current_song, SongFile):
|
|
self.navigator.select_current_item()
|
|
return
|
|
selected_song = self.navigator.select_current_item()
|
|
if selected_song:
|
|
self.state = State.SONG_SELECTED
|
|
self.player_1.on_song_selected(selected_song)
|
|
audio.play_sound('don', 'sound')
|
|
audio.play_sound('voice_select_diff', 'voice')
|
|
self.move_away.start()
|
|
self.diff_fade_out.start()
|
|
self.text_fade_out.start()
|
|
self.text_fade_in.start()
|
|
self.player_1.selected_diff_bounce.start()
|
|
self.player_1.selected_diff_fadein.start()
|
|
self.player_2.selected_diff_bounce.start()
|
|
self.player_2.selected_diff_fadein.start()
|
|
elif action == "add_favorite":
|
|
self.navigator.add_favorite()
|
|
current_box = self.navigator.get_current_item().box
|
|
if not isinstance(current_box, SongBox):
|
|
return
|
|
current_box.is_favorite = not current_box.is_favorite
|
|
|
|
def handle_input_selected(self):
|
|
"""Handle input for selecting difficulty."""
|
|
p2_result = False
|
|
result = self.player_1.handle_input_selected(self.navigator.get_current_item())
|
|
if result is None:
|
|
result = self.player_2.handle_input_selected(self.navigator.get_current_item())
|
|
if result is None:
|
|
return
|
|
p2_result = True
|
|
if result is not None:
|
|
logger.info(f"Difficulty selection result: {result}, p2_result={p2_result}")
|
|
if result == "cancel":
|
|
self._cancel_selection()
|
|
logger.info("Selection cancelled")
|
|
elif result == "confirm":
|
|
if p2_result:
|
|
self._confirm_selection(2)
|
|
else:
|
|
self._confirm_selection(1)
|
|
logger.info("Selection confirmed")
|
|
elif result == "ura_toggle":
|
|
if p2_result:
|
|
self.ura_switch_animation.start(not self.player_2.is_ura)
|
|
else:
|
|
self.ura_switch_animation.start(not self.player_1.is_ura)
|
|
logger.info("Ura toggled")
|
|
|
|
def handle_input_diff_sort(self):
|
|
"""Handle input for sorting difficulty."""
|
|
if self.diff_sort_selector is None:
|
|
raise Exception("Diff sort selector was not able to be created")
|
|
|
|
result = self.player_1.handle_input_diff_sort(self.diff_sort_selector)
|
|
|
|
if result is not None:
|
|
diff, level = result
|
|
self.diff_sort_selector = None
|
|
self.state = State.BROWSING
|
|
self.text_fade_out.reset()
|
|
self.text_fade_in.reset()
|
|
logger.info(f"Diff sort selected: diff={diff}, level={level}")
|
|
if diff != -1:
|
|
if level != -1:
|
|
self.navigator.diff_sort_diff = diff
|
|
self.navigator.diff_sort_level = level
|
|
self.navigator.select_current_item()
|
|
|
|
def _cancel_selection(self):
|
|
"""Reset to browsing state"""
|
|
super()._cancel_selection()
|
|
self.player_2.selected_song = False
|
|
|
|
def _confirm_selection(self, player_selected: int = 1):
|
|
"""Confirm song selection and create game transition"""
|
|
audio.play_sound('don', 'sound')
|
|
audio.play_sound(f'voice_start_song_{global_data.player_num}p', 'voice')
|
|
if player_selected == 1:
|
|
global_data.session_data[PlayerNum.P1].selected_difficulty = self.player_1.selected_difficulty
|
|
self.player_1.selected_diff_highlight_fade.start()
|
|
self.player_1.selected_diff_text_resize.start()
|
|
self.player_1.selected_diff_text_fadein.start()
|
|
elif player_selected == 2:
|
|
global_data.session_data[PlayerNum.P2].selected_difficulty = self.player_2.selected_difficulty
|
|
self.player_2.selected_diff_highlight_fade.start()
|
|
self.player_2.selected_diff_text_resize.start()
|
|
self.player_2.selected_diff_text_fadein.start()
|
|
logger.info(f"Confirmed selection for player {player_selected}")
|
|
|
|
def check_for_selection(self):
|
|
if (self.player_1.selected_diff_highlight_fade.is_finished and
|
|
self.player_2.selected_diff_highlight_fade.is_finished and
|
|
not audio.is_sound_playing(f'voice_start_song_{global_data.player_num}p') and self.game_transition is None):
|
|
selected_song = self.navigator.get_current_item()
|
|
if not isinstance(selected_song, SongFile):
|
|
raise Exception("picked directory")
|
|
|
|
title = selected_song.tja.metadata.title.get(
|
|
global_data.config['general']['language'], '')
|
|
subtitle = selected_song.tja.metadata.subtitle.get(
|
|
global_data.config['general']['language'], '')
|
|
self.game_transition = Transition(title, subtitle)
|
|
self.game_transition.start()
|
|
logger.info(f"Game transition started for song: {title} - {subtitle}")
|
|
|
|
def update_players(self, current_time) -> str:
|
|
self.player_1.update(current_time)
|
|
self.player_2.update(current_time)
|
|
if self.text_fade_out.is_finished:
|
|
self.player_1.selected_song = True
|
|
self.player_2.selected_song = True
|
|
next_screen = "GAME_2P"
|
|
return next_screen
|
|
|
|
def draw_background_diffs(self):
|
|
self.player_1.draw_background_diffs(self.state)
|
|
self.player_2.draw_background_diffs(self.state)
|
|
|
|
def draw_players(self):
|
|
if self.player_1.selected_difficulty == self.player_2.selected_difficulty:
|
|
self.player_1.draw(self.state, True)
|
|
self.player_2.draw(self.state, True)
|
|
else:
|
|
self.player_1.draw(self.state, False)
|
|
self.player_2.draw(self.state, False)
|