Taiko_Stuff/tja2fumen
1
0
Fork 0
mirror of synced 2025-02-03 13:13:26 +01:00
Code Releases Wiki Activity

constants.py: Add support for 0-4 course strings

Browse Source
This commit is contained in:
Viv 2023-06-03 09:45:53 -04:00
parent ffc62c73a9
commit f5aa185ec2
2 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tja2fumen/constants.py
View File

@ -96,6 +96,20 @@ unknownHeaderSample[76] = 78
unknownHeaderSample[77] = 97 unknownHeaderSample[77] = 97
unknownHeaderSample[78] = 188 unknownHeaderSample[78] = 188
NORMALIZE_COURSE = {
'0': 'Easy',
'Easy': 'Easy',
'1': 'Normal',
'Normal': 'Normal',
'2': 'Hard',
'Hard': 'Hard',
'3': 'Oni',
'Oni': 'Oni',
'4': 'Ura',
'Ura': 'Ura',
'Edit': 'Ura'
}
DIFFICULTY_BYTES = { DIFFICULTY_BYTES = {
'Easy': [112, 23], 'Easy': [112, 23],
'Normal': [88, 27], 'Normal': [88, 27],

6
tja2fumen/parsers.py
View File

@ -4,7 +4,7 @@ import re
from tja2fumen.utils import readStruct, getBool, shortHex, nameValue, debugPrint, checkValidHeader, validateHeaderMetadata from tja2fumen.utils import readStruct, getBool, shortHex, nameValue, debugPrint, checkValidHeader, validateHeaderMetadata
from tja2fumen.constants import ( from tja2fumen.constants import (
# TJA constants # TJA constants
HEADER_GLOBAL, HEADER_COURSE, BRANCH_COMMANDS, MEASURE_COMMANDS, COMMAND, HEADER_GLOBAL, HEADER_COURSE, BRANCH_COMMANDS, MEASURE_COMMANDS, COMMAND, NORMALIZE_COURSE,
# Fumen constants # Fumen constants
branchNames, noteTypes branchNames, noteTypes
) )
@ -36,7 +36,7 @@ def parseTJA(tja):
else: else:
# Check to see if we're starting a new course # Check to see if we're starting a new course
if parsed['type'] == 'header' and parsed['scope'] == 'course' and parsed['name'] == 'COURSE': if parsed['type'] == 'header' and parsed['scope'] == 'course' and parsed['name'] == 'COURSE':
currentCourse = parsed['value'] currentCourse = NORMALIZE_COURSE[parsed['value']]
if currentCourse not in courses.keys(): if currentCourse not in courses.keys():
courses[currentCourse] = [] courses[currentCourse] = []
# Append the line to the current course # Append the line to the current course
@ -85,7 +85,7 @@ def getCourse(tjaHeaders, lines):
def parseHeaderMetadata(line): def parseHeaderMetadata(line):
nonlocal headers nonlocal headers
if line["name"] == 'COURSE': if line["name"] == 'COURSE':
headers['course'] = line['value'] headers['course'] = NORMALIZE_COURSE[line['value']]
elif line["name"] == 'LEVEL': elif line["name"] == 'LEVEL':
headers['level'] = int(line['value']) if line['value'] else 0 headers['level'] = int(line['value']) if line['value'] else 0
elif line["name"] == 'SCOREINIT': elif line["name"] == 'SCOREINIT':