robob/PyTaiko
1
0
Fork 0
mirror of https://github.com/Yonokid/PyTaiko.git synced 2026-02-04 11:40:13 +01:00
Code Issues Packages Projects Releases Wiki Activity

Cleaned up parser a little, more feature complete implementation soon

Browse Source
This commit is contained in:
Valerio
2026-01-05 18:38:33 -05:00
parent b766a9e170
commit 299ac2c47b
3 changed files with 1448 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

114
osz.py
View File

@@ -13,67 +13,77 @@ from libs.global_data import Modifiers
from libs.utils import strip_comments
from libs.tja import TimelineObject, Note, NoteType, Drumroll, Balloon, NoteList, CourseData, ParserState
osu_file = Path("./PNames.osu")
import re
osu_file = Path("./Renatus.osu")
contents = osu_file.open(mode='r', encoding='utf-8').read()
import re
def read_osu_note_data(file_path, target_header="HitObjects"):
note_data = []
current_header = None
class OsuParser:
def __init__(self):
with file_path.open(mode='r', encoding='utf-8') as f:
def read_osu_data(self, file_path, target_header="HitObjects"):
data = []
current_header = None
for line in f:
line = line.rstrip("\n")
with file_path.open(mode='r', encoding='utf-8') as f:
if re.match(r"\[\w*\]", line): # header pattern
current_header = line[1:-1]
for line in f:
line = line.rstrip("\n")
if current_header == target_header:
if re.match(r"[-+]?\d*\.?\d+" , line):
string_array = re.findall(r"[-+]?\d*\.?\d+" , line) # search for floats
int_array = [int(num_str) for num_str in string_array]
note_data.append(int_array)
else:
continue
if re.match(r"\[\w*\]", line): # header pattern
current_header = line[1:-1]
return note_data
note_data = read_osu_note_data(osu_file)
def note_data_to_NoteList(note_data):
osu_NoteList = NoteList()
counter = 0
for line in note_data:
if line[3] == 1 and line[4] == 0: # DON
don = Note()
don.type = NoteType(1)
don.hit_ms = line[2]
don.bpm = 207
don.scroll_x = 1
don.scroll_y = 0
don.display = True
don.index = counter
counter = counter + 1
don.moji = 0
osu_NoteList.play_notes.append(don)
if line[3] == 1 and line[4] == 2: # KAT
kat = Note()
kat.type = NoteType(2)
kat.hit_ms = line[2]
kat.bpm = 207
kat.scroll_x = 1
kat.scroll_y = 0
kat.display = True
kat.index = counter
counter = counter + 1
kat.moji = 1
osu_NoteList.play_notes.append(kat)
osu_NoteList.draw_notes = osu_NoteList.play_notes.copy()
return osu_NoteList
if current_header == target_header:
if re.match(r"[-+]?\d*\.?\d+" , line): # Events, TimingPoints, HitObjects
string_array = re.findall(r"[-+]?\d*\.?\d+" , line) # search for floats
int_array = [float(num_str) for num_str in string_array]
data.append(int_array)
if re.match(r'\w*\:\s(\w*.?\w*)', line): # General, Editor, Metadata, Difficulty
match = re.search(r'\w*\:\s(\w*.?\w*)', line)
if match:
data.append(match.group(1))
else:
continue
return data
def note_data_to_NoteList(self, note_data):
osu_NoteList = NoteList()
counter = 0
for line in note_data:
if line[3] == 1 and line[4] == 0: # DON
don = Note()
don.type = NoteType(1)
don.hit_ms = line[2]
don.bpm = 207
don.scroll_x = 1
don.scroll_y = 0
don.display = True
don.index = counter
counter = counter + 1
don.moji = 0
osu_NoteList.play_notes.append(don)
if line[3] == 1 and line[4] != 0: # KAT
kat = Note()
kat.type = NoteType(2)
kat.hit_ms = line[2]
kat.bpm = 207
kat.scroll_x = 1
kat.scroll_y = 0
kat.display = True
kat.index = counter
counter = counter + 1
kat.moji = 1
osu_NoteList.play_notes.append(kat)
osu_NoteList.draw_notes = osu_NoteList.play_notes.copy()
return osu_NoteList
myparse = OsuParser()
print(myparse.read_osu_data(osu_file, target_header="TimingPoints"))