mirror of
https://github.com/Yonokid/PyTaiko.git
synced 2026-02-04 03:30:13 +01:00
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
import hashlib
|
|
import logging
|
|
import math
|
|
import random
|
|
from collections import deque
|
|
from dataclasses import dataclass, field, fields
|
|
from enum import IntEnum
|
|
from functools import lru_cache
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
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")
|
|
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
|
|
|
|
with file_path.open(mode='r', encoding='utf-8') as f:
|
|
|
|
for line in f:
|
|
line = line.rstrip("\n")
|
|
|
|
if re.match(r"\[\w*\]", line): # header pattern
|
|
current_header = line[1:-1]
|
|
|
|
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
|
|
|
|
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
|
|
|
|
|
|
|
|
|
|
|