added gimmick

This commit is contained in:
Yonokid
2025-07-14 22:50:30 -04:00
parent ebe991ecf1
commit aed0c530a4
4 changed files with 124 additions and 36 deletions

View File

@@ -18,6 +18,8 @@ def process_tja_file(tja_file):
all_notes.extend( all_notes.extend(
TJAParser.notes_to_position(TJAParser(tja.file_path), diff) TJAParser.notes_to_position(TJAParser(tja.file_path), diff)
) )
if all_notes == []:
return ''
hash = tja.hash_note_data(all_notes[0], all_notes[2]) hash = tja.hash_note_data(all_notes[0], all_notes[2])
return hash return hash
@@ -71,6 +73,8 @@ def build_song_hashes(output_file="cache/song_hashes.json"):
all_notes.extend( all_notes.extend(
TJAParser.notes_to_position(TJAParser(tja.file_path), diff) TJAParser.notes_to_position(TJAParser(tja.file_path), diff)
) )
if all_notes == []:
continue
hash_val = tja.hash_note_data(all_notes[0], all_notes[2]) hash_val = tja.hash_note_data(all_notes[0], all_notes[2])
song_hashes[hash_val] = { song_hashes[hash_val] = {
"file_path": str(tja_file), "file_path": str(tja_file),

View File

@@ -25,7 +25,8 @@ class Note:
type: int = field(init=False) type: int = field(init=False)
hit_ms: float = field(init=False) hit_ms: float = field(init=False)
load_ms: float = field(init=False) load_ms: float = field(init=False)
pixels_per_frame: float = field(init=False) pixels_per_frame_x: float = field(init=False)
pixels_per_frame_y: float = field(init=False)
display: bool = field(init=False) display: bool = field(init=False)
index: int = field(init=False) index: int = field(init=False)
bpm: float = field(init=False) bpm: float = field(init=False)
@@ -175,6 +176,7 @@ class TJAParser:
DIFFS = {0: "easy", 1: "normal", 2: "hard", 3: "oni", 4: "edit", 5: "tower", 6: "dan"} DIFFS = {0: "easy", 1: "normal", 2: "hard", 3: "oni", 4: "edit", 5: "tower", 6: "dan"}
def __init__(self, path: Path, start_delay: int = 0, distance: int = 866): def __init__(self, path: Path, start_delay: int = 0, distance: int = 866):
self.file_path: Path = path self.file_path: Path = path
print(self.file_path)
lines = self.file_path.read_text(encoding='utf-8-sig').splitlines() lines = self.file_path.read_text(encoding='utf-8-sig').splitlines()
self.data = [cleaned for line in lines self.data = [cleaned for line in lines
@@ -244,22 +246,22 @@ class TJAParser:
balloon_data = item.split(':')[1] balloon_data = item.split(':')[1]
if balloon_data == '': if balloon_data == '':
continue continue
self.metadata.course_data[current_diff].balloon.extend([int(x) for x in balloon_data.split(',')]) self.metadata.course_data[current_diff].balloon.extend([int(x) for x in balloon_data.split(',') if x != ''])
elif item.startswith('BALLOONEXP'): elif item.startswith('BALLOONEXP'):
balloon_data = item.split(':')[1] balloon_data = item.split(':')[1]
if balloon_data == '': if balloon_data == '':
continue continue
self.metadata.course_data[current_diff].balloon.extend([int(x) for x in balloon_data.split(',')]) self.metadata.course_data[current_diff].balloon.extend([int(x) for x in balloon_data.split(',') if x != ''])
elif item.startswith('BALLOONMAS'): elif item.startswith('BALLOONMAS'):
balloon_data = item.split(':')[1] balloon_data = item.split(':')[1]
if balloon_data == '': if balloon_data == '':
continue continue
self.metadata.course_data[current_diff].balloon = ([int(x) for x in balloon_data.split(',')]) self.metadata.course_data[current_diff].balloon = [int(x) for x in balloon_data.split(',') if x != '']
elif item.startswith('BALLOON'): elif item.startswith('BALLOON'):
balloon_data = item.split(':')[1] balloon_data = item.split(':')[1]
if balloon_data == '': if balloon_data == '':
continue continue
self.metadata.course_data[current_diff].balloon = [int(x) for x in balloon_data.split(',')] self.metadata.course_data[current_diff].balloon = [int(x) for x in balloon_data.split(',') if x != '']
elif item.startswith('SCOREINIT'): elif item.startswith('SCOREINIT'):
score_init = item.split(':')[1] score_init = item.split(':')[1]
if score_init == '': if score_init == '':
@@ -282,7 +284,7 @@ class TJAParser:
elif '限定' in self.metadata.title[region_code]: elif '限定' in self.metadata.title[region_code]:
self.ex_data.limited_time = True self.ex_data.limited_time = True
def data_to_notes(self, diff): def data_to_notes(self, diff) -> list[list[str]]:
diff_name = self.DIFFS.get(diff, "").lower() diff_name = self.DIFFS.get(diff, "").lower()
# Use enumerate for single iteration # Use enumerate for single iteration
@@ -400,7 +402,8 @@ class TJAParser:
index = 0 index = 0
time_signature = 4/4 time_signature = 4/4
bpm = self.metadata.bpm bpm = self.metadata.bpm
scroll_modifier = 1 x_scroll_modifier = 1
y_scroll_modifier = 0
barline_display = True barline_display = True
gogo_time = False gogo_time = False
skip_branch = False skip_branch = False
@@ -409,6 +412,8 @@ class TJAParser:
bar_length = sum(len(part) for part in bar if '#' not in part) bar_length = sum(len(part) for part in bar if '#' not in part)
barline_added = False barline_added = False
for part in bar: for part in bar:
if '#LYRIC' in part:
continue
if part.startswith('#BRANCHSTART'): if part.startswith('#BRANCHSTART'):
skip_branch = True skip_branch = True
continue continue
@@ -421,7 +426,74 @@ class TJAParser:
time_signature = float(part[9:divisor]) / float(part[divisor+1:]) time_signature = float(part[9:divisor]) / float(part[divisor+1:])
continue continue
elif '#SCROLL' in part: elif '#SCROLL' in part:
scroll_modifier = float(part[7:]) # Extract the value after '#SCROLL '
scroll_value = part[7:].strip() # Remove '#SCROLL' and whitespace
# Initialize default values
x_scroll_modifier = 0
y_scroll_modifier = 0
# Handle empty value
if not scroll_value:
continue
# Check if it's a complex number (contains 'i')
if 'i' in scroll_value:
# Handle different imaginary number formats
if scroll_value == 'i':
x_scroll_modifier = 0
y_scroll_modifier = 1
elif scroll_value == '-i':
x_scroll_modifier = 0
y_scroll_modifier = -1
elif scroll_value.endswith('i') or scroll_value.endswith('.i'):
# Remove the 'i' or '.i' suffix
if scroll_value.endswith('.i'):
complex_part = scroll_value[:-2]
else:
complex_part = scroll_value[:-1]
# Look for + or - that separates real and imaginary parts
# Find the rightmost + or - (excluding position 0 for negative numbers)
plus_pos = complex_part.rfind('+')
minus_pos = complex_part.rfind('-')
separator_pos = -1
if plus_pos > 0: # Ignore + at position 0
separator_pos = plus_pos
if minus_pos > 0 and minus_pos > separator_pos: # Ignore - at position 0
separator_pos = minus_pos
if separator_pos > 0:
# Complex number like '1+i', '3+4i', '2-5i', '-1+2i', etc.
real_part = complex_part[:separator_pos]
imag_part = complex_part[separator_pos:]
x_scroll_modifier = float(real_part) if real_part else 0
# Handle imaginary part
if imag_part == '+' or imag_part == '':
y_scroll_modifier = 1
elif imag_part == '-':
y_scroll_modifier = -1
else:
y_scroll_modifier = float(imag_part)
else:
# Pure imaginary like '5i', '-3i', '2.5i'
if complex_part == '' or complex_part == '+':
y_scroll_modifier = 1
elif complex_part == '-':
y_scroll_modifier = -1
else:
y_scroll_modifier = float(complex_part)
x_scroll_modifier = 0
else:
# 'i' is somewhere in the middle - invalid format
continue
else:
# Pure real number
x_scroll_modifier = float(scroll_value)
y_scroll_modifier = 0
continue continue
elif '#BPMCHANGE' in part: elif '#BPMCHANGE' in part:
bpm = float(part[11:]) bpm = float(part[11:])
@@ -438,8 +510,6 @@ class TJAParser:
elif '#GOGOEND' in part: elif '#GOGOEND' in part:
gogo_time = False gogo_time = False
continue continue
elif '#LYRIC' in part:
continue
elif part.startswith('#M'): elif part.startswith('#M'):
skip_branch = False skip_branch = False
continue continue
@@ -455,8 +525,9 @@ class TJAParser:
bar_line = Note() bar_line = Note()
#Determines how quickly the notes need to move across the screen to reach the judgment circle in time #Determines how quickly the notes need to move across the screen to reach the judgment circle in time
bar_line.pixels_per_frame = get_pixels_per_frame(bpm * time_signature * scroll_modifier, time_signature*4, self.distance) bar_line.pixels_per_frame_x = get_pixels_per_frame(bpm * time_signature * x_scroll_modifier, time_signature*4, self.distance)
pixels_per_ms = get_pixels_per_ms(bar_line.pixels_per_frame) bar_line.pixels_per_frame_y = get_pixels_per_frame(bpm * time_signature * y_scroll_modifier, time_signature*4, self.distance)
pixels_per_ms = get_pixels_per_ms(bar_line.pixels_per_frame_x)
bar_line.hit_ms = self.current_ms bar_line.hit_ms = self.current_ms
if pixels_per_ms == 0: if pixels_per_ms == 0:
@@ -480,13 +551,16 @@ class TJAParser:
increment = ms_per_measure / bar_length increment = ms_per_measure / bar_length
for item in part: for item in part:
if item == '0': if item == '.':
continue
if item == '0' or (not item.isdigit()):
self.current_ms += increment self.current_ms += increment
continue continue
note = Note() note = Note()
note.hit_ms = self.current_ms note.hit_ms = self.current_ms
note.pixels_per_frame = bar_line.pixels_per_frame note.pixels_per_frame_x = bar_line.pixels_per_frame_x
pixels_per_ms = get_pixels_per_ms(note.pixels_per_frame) note.pixels_per_frame_y = bar_line.pixels_per_frame_y
pixels_per_ms = get_pixels_per_ms(note.pixels_per_frame_x)
note.load_ms = (note.hit_ms if pixels_per_ms == 0 note.load_ms = (note.hit_ms if pixels_per_ms == 0
else note.hit_ms - (self.distance / pixels_per_ms)) else note.hit_ms - (self.distance / pixels_per_ms))
note.type = int(item) note.type = int(item)
@@ -504,12 +578,12 @@ class TJAParser:
note = Balloon(note) note = Balloon(note)
note.count = 1 if not balloon else balloon.pop(0) note.count = 1 if not balloon else balloon.pop(0)
elif item == '8': elif item == '8':
new_pixels_per_ms = play_note_list[-1].pixels_per_frame / (1000 / 60) new_pixels_per_ms = play_note_list[-1].pixels_per_frame_x / (1000 / 60)
if new_pixels_per_ms == 0: if new_pixels_per_ms == 0:
note.load_ms = note.hit_ms note.load_ms = note.hit_ms
else: else:
note.load_ms = note.hit_ms - (self.distance / new_pixels_per_ms) note.load_ms = note.hit_ms - (self.distance / new_pixels_per_ms)
note.pixels_per_frame = play_note_list[-1].pixels_per_frame note.pixels_per_frame_x = play_note_list[-1].pixels_per_frame_x
self.current_ms += increment self.current_ms += increment
play_note_list.append(note) play_note_list.append(note)
bisect.insort(draw_note_list, note, key=lambda x: x.load_ms) bisect.insort(draw_note_list, note, key=lambda x: x.load_ms)
@@ -519,6 +593,7 @@ class TJAParser:
if isinstance(play_note_list[-2], Drumroll) and play_note_list[-1].type != 8: if isinstance(play_note_list[-2], Drumroll) and play_note_list[-1].type != 8:
print(self.file_path, diff) print(self.file_path, diff)
print(bar) print(bar)
continue
raise Exception(f"{play_note_list[-2]}") raise Exception(f"{play_note_list[-2]}")
# https://stackoverflow.com/questions/72899/how-to-sort-a-list-of-dictionaries-by-a-value-of-the-dictionary-in-python # https://stackoverflow.com/questions/72899/how-to-sort-a-list-of-dictionaries-by-a-value-of-the-dictionary-in-python
# Sorting by load_ms is necessary for drawing, as some notes appear on the # Sorting by load_ms is necessary for drawing, as some notes appear on the

View File

@@ -113,8 +113,11 @@ class GameScreen:
self.player_1 = Player(self, 1, difficulty) self.player_1 = Player(self, 1, difficulty)
if not hasattr(self, 'song_music'): if not hasattr(self, 'song_music'):
self.song_music = audio.load_sound(self.tja.metadata.wave) if self.tja.metadata.wave.exists() and self.tja.metadata.wave.is_file():
audio.normalize_sound(self.song_music, 0.1935) self.song_music = audio.load_sound(self.tja.metadata.wave)
audio.normalize_sound(self.song_music, 0.1935)
else:
self.song_music = None
self.start_ms = (get_current_ms() - self.tja.metadata.offset*1000) self.start_ms = (get_current_ms() - self.tja.metadata.offset*1000)
def on_screen_start(self): def on_screen_start(self):
@@ -279,8 +282,11 @@ class Player:
def get_result_score(self): def get_result_score(self):
return self.score, self.good_count, self.ok_count, self.bad_count, self.total_drumroll, self.max_combo return self.score, self.good_count, self.ok_count, self.bad_count, self.total_drumroll, self.max_combo
def get_position(self, game_screen: GameScreen, ms: float, pixels_per_frame: float) -> int: def get_position_x(self, width: int, current_ms: float, load_ms: float, pixels_per_frame: float) -> int:
return int(game_screen.width + pixels_per_frame * 60 / 1000 * (ms - game_screen.current_ms) - 64) - self.visual_offset return int(width + pixels_per_frame * (60 / 1000) * (load_ms - current_ms) - 64) - self.visual_offset
def get_position_y(self, current_ms: float, load_ms: float, pixels_per_frame: float, pixels_per_frame_x) -> int:
return int((pixels_per_frame * (60 / 1000) * (load_ms - current_ms)) + (((1280 - GameScreen.JUDGE_X) * pixels_per_frame) / pixels_per_frame_x))
def animation_manager(self, animation_list: list): def animation_manager(self, animation_list: list):
if len(animation_list) <= 0: if len(animation_list) <= 0:
@@ -303,7 +309,7 @@ class Player:
for i in range(len(self.current_bars)-1, -1, -1): for i in range(len(self.current_bars)-1, -1, -1):
bar = self.current_bars[i] bar = self.current_bars[i]
position = self.get_position(game_screen, bar.hit_ms, bar.pixels_per_frame) position = self.get_position_x(game_screen.width, game_screen.current_ms, bar.hit_ms, bar.pixels_per_frame_x)
if position < GameScreen.JUDGE_X + 650: if position < GameScreen.JUDGE_X + 650:
self.current_bars.pop(i) self.current_bars.pop(i)
@@ -357,7 +363,7 @@ class Player:
note = self.current_notes_draw[0] note = self.current_notes_draw[0]
if note.type in {5, 6, 7} and len(self.current_notes_draw) > 1: if note.type in {5, 6, 7} and len(self.current_notes_draw) > 1:
note = self.current_notes_draw[1] note = self.current_notes_draw[1]
position = self.get_position(game_screen, note.hit_ms, note.pixels_per_frame) position = self.get_position_x(game_screen.width, game_screen.current_ms, note.hit_ms, note.pixels_per_frame_x)
if position < GameScreen.JUDGE_X + 650: if position < GameScreen.JUDGE_X + 650:
self.current_notes_draw.pop(0) self.current_notes_draw.pop(0)
@@ -575,12 +581,12 @@ class Player:
self.gauge.update(get_current_ms(), self.good_count, self.ok_count, self.bad_count, self.total_notes) self.gauge.update(get_current_ms(), self.good_count, self.ok_count, self.bad_count, self.total_notes)
def draw_drumroll(self, game_screen: GameScreen, head: Drumroll, current_eighth: int): def draw_drumroll(self, game_screen: GameScreen, head: Drumroll, current_eighth: int):
start_position = self.get_position(game_screen, head.load_ms, head.pixels_per_frame) start_position = self.get_position_x(game_screen.width, game_screen.current_ms, head.load_ms, head.pixels_per_frame_x)
tail = next((note for note in self.current_notes_draw[1:] if note.type == 8 and note.index > head.index), None) tail = next((note for note in self.current_notes_draw[1:] if note.type == 8 and note.index > head.index), None)
if tail is None: if tail is None:
raise Exception("Tail for Balloon not found") raise Exception("Tail for Balloon not found")
is_big = int(head.type == 6) * 2 is_big = int(head.type == 6) * 2
end_position = self.get_position(game_screen, tail.load_ms, tail.pixels_per_frame) end_position = self.get_position_x(game_screen.width, game_screen.current_ms, tail.load_ms, tail.pixels_per_frame_x)
length = (end_position - start_position - 50) length = (end_position - start_position - 50)
if length <= 0: if length <= 0:
end_position += 50 end_position += 50
@@ -601,11 +607,11 @@ class Player:
def draw_balloon(self, game_screen: GameScreen, head: Balloon, current_eighth: int): def draw_balloon(self, game_screen: GameScreen, head: Balloon, current_eighth: int):
offset = 12 offset = 12
start_position = self.get_position(game_screen, head.load_ms, head.pixels_per_frame) start_position = self.get_position_x(game_screen.width, game_screen.current_ms, head.load_ms, head.pixels_per_frame_x)
tail = next((note for note in self.current_notes_draw[1:] if note.type == 8 and note.index > head.index), None) tail = next((note for note in self.current_notes_draw[1:] if note.type == 8 and note.index > head.index), None)
if tail is None: if tail is None:
raise Exception("Tail for Balloon not found") raise Exception("Tail for Balloon not found")
end_position = self.get_position(game_screen, tail.load_ms, tail.pixels_per_frame) end_position = self.get_position_x(game_screen.width, game_screen.current_ms, tail.load_ms, tail.pixels_per_frame_x)
pause_position = 349 pause_position = 349
if game_screen.current_ms >= tail.hit_ms: if game_screen.current_ms >= tail.hit_ms:
position = end_position position = end_position
@@ -623,8 +629,9 @@ class Player:
for bar in reversed(self.current_bars): for bar in reversed(self.current_bars):
if not bar.display: if not bar.display:
continue continue
position = self.get_position(game_screen, bar.load_ms, bar.pixels_per_frame) x_position = self.get_position_x(game_screen.width, game_screen.current_ms, bar.load_ms, bar.pixels_per_frame_x)
ray.draw_texture(game_screen.note_type_list[bar.type], position+60, 190, ray.WHITE) y_position = self.get_position_y(game_screen.current_ms, bar.load_ms, bar.pixels_per_frame_y, bar.pixels_per_frame_x)
ray.draw_texture(game_screen.note_type_list[bar.type], x_position + 60, y_position + 190, ray.WHITE)
def draw_notes(self, game_screen: GameScreen): def draw_notes(self, game_screen: GameScreen):
if len(self.current_notes_draw) <= 0: if len(self.current_notes_draw) <= 0:
@@ -649,17 +656,18 @@ class Player:
continue continue
if note.type == 8: if note.type == 8:
continue continue
position = self.get_position(game_screen, note.load_ms, note.pixels_per_frame) x_position = self.get_position_x(game_screen.width, game_screen.current_ms, note.load_ms, note.pixels_per_frame_x)
y_position = self.get_position_y(game_screen.current_ms, note.load_ms, note.pixels_per_frame_y, note.pixels_per_frame_x)
if isinstance(note, Drumroll): if isinstance(note, Drumroll):
self.draw_drumroll(game_screen, note, current_eighth) self.draw_drumroll(game_screen, note, current_eighth)
elif isinstance(note, Balloon): elif isinstance(note, Balloon):
self.draw_balloon(game_screen, note, current_eighth) self.draw_balloon(game_screen, note, current_eighth)
moji_texture = game_screen.texture_se_moji[note.moji] moji_texture = game_screen.texture_se_moji[note.moji]
ray.draw_texture(moji_texture, position - (moji_texture.width//2) + 64, 323, ray.WHITE) ray.draw_texture(moji_texture, x_position - (moji_texture.width//2) + 64, 323 + y_position, ray.WHITE)
else: else:
ray.draw_texture(game_screen.note_type_list[note.type][current_eighth % 2], position, 192, ray.WHITE) ray.draw_texture(game_screen.note_type_list[note.type][current_eighth % 2], x_position, y_position + 192, ray.WHITE)
moji_texture = game_screen.texture_se_moji[note.moji] moji_texture = game_screen.texture_se_moji[note.moji]
ray.draw_texture(moji_texture, position - (moji_texture.width//2) + 64, 323, ray.WHITE) ray.draw_texture(moji_texture, x_position - (moji_texture.width//2) + 64, 323 + y_position, ray.WHITE)
#ray.draw_text(str(note.index), position+64, 192, 25, ray.GREEN) #ray.draw_text(str(note.index), position+64, 192, 25, ray.GREEN)
def draw(self, game_screen: GameScreen): def draw(self, game_screen: GameScreen):

View File

@@ -223,9 +223,10 @@ class SongSelectScreen:
if not isinstance(song, Directory) and song.box.is_open: if not isinstance(song, Directory) and song.box.is_open:
if self.demo_song is None and get_current_ms() >= song.box.wait + (83.33*3): if self.demo_song is None and get_current_ms() >= song.box.wait + (83.33*3):
song.box.get_scores() song.box.get_scores()
self.demo_song = audio.load_music_stream(song.tja.metadata.wave, preview=song.tja.metadata.demostart, normalize=0.1935) if song.tja.metadata.wave.exists() and song.tja.metadata.wave.is_file():
audio.play_music_stream(self.demo_song) self.demo_song = audio.load_music_stream(song.tja.metadata.wave, preview=song.tja.metadata.demostart, normalize=0.1935)
audio.stop_sound(self.sound_bgm) audio.play_music_stream(self.demo_song)
audio.stop_sound(self.sound_bgm)
if song.box.is_open: if song.box.is_open:
current_box = song.box current_box = song.box
if current_box.texture_index != 552 and get_current_ms() >= song.box.wait + (83.33*3): if current_box.texture_index != 552 and get_current_ms() >= song.box.wait + (83.33*3):