Add TJA parser (translated from JavaScript)
Source: https://github.com/WHMHammer/tja-tools/blob/master/src/js/parseTJA.js
This commit is contained in:
Viv
parent
8fa6295efa
commit
93dd1355f8
379
tja2fumen/parsetja.py
Normal file
379
tja2fumen/parsetja.py
Normal file
@ -0,0 +1,379 @@
|
||||
# Original source: https://github.com/WHMHammer/tja-tools/blob/master/src/js/parseTJA.js
|
||||
|
||||
import re
|
||||
|
||||
HEADER_GLOBAL = [
|
||||
'TITLE',
|
||||
'TITLEJA',
|
||||
'SUBTITLE',
|
||||
'SUBTITLEJA',
|
||||
'BPM',
|
||||
'WAVE',
|
||||
'OFFSET',
|
||||
'DEMOSTART',
|
||||
'GENRE',
|
||||
]
|
||||
|
||||
HEADER_COURSE = [
|
||||
'COURSE',
|
||||
'LEVEL',
|
||||
'BALLOON',
|
||||
'SCOREINIT',
|
||||
'SCOREDIFF',
|
||||
'TTRO'
|
||||
'WBEAT',
|
||||
]
|
||||
|
||||
COMMAND = [
|
||||
'START',
|
||||
'END',
|
||||
'GOGOSTART',
|
||||
'GOGOEND',
|
||||
'MEASURE',
|
||||
'SCROLL',
|
||||
'BPMCHANGE',
|
||||
'DELAY',
|
||||
'BRANCHSTART',
|
||||
'BRANCHEND',
|
||||
'SECTION',
|
||||
'N',
|
||||
'E',
|
||||
'M',
|
||||
'LEVELHOLD',
|
||||
'BMSCROLL',
|
||||
'HBSCROLL',
|
||||
'BARLINEOFF',
|
||||
'BARLINEON',
|
||||
'TTBREAK',
|
||||
]
|
||||
|
||||
|
||||
def parseLine(line):
|
||||
match_comment = re.match(r"//.*", line)
|
||||
match_header = re.match(r"^([A-Z]+):(.+)", line)
|
||||
match_command = re.match(r"^#([A-Z]+)(?:\s+(.+))?", line)
|
||||
match_data = re.match(r"^(([0-9]|A|B|C|F|G)*,?)$", line)
|
||||
|
||||
# comment
|
||||
if match_comment:
|
||||
return {
|
||||
"type": 'comment',
|
||||
"value": line
|
||||
} # js: line = line.substr(0, match.index).strip()
|
||||
|
||||
# header
|
||||
elif match_header:
|
||||
nameUpper = match_header.group(1).upper()
|
||||
value = match_header.group(2)
|
||||
|
||||
if nameUpper in HEADER_GLOBAL:
|
||||
return {
|
||||
"type": 'header',
|
||||
"scope": 'global',
|
||||
"name": nameUpper,
|
||||
"value": value.strip(),
|
||||
}
|
||||
elif nameUpper in HEADER_COURSE:
|
||||
return {
|
||||
"type": 'header',
|
||||
"scope": 'course',
|
||||
"name": nameUpper,
|
||||
"value": value.strip(),
|
||||
}
|
||||
else:
|
||||
breakpoint()
|
||||
|
||||
# command
|
||||
elif match_command:
|
||||
nameUpper = match_command.group(1).upper()
|
||||
value = match_command.group(2)
|
||||
if not value:
|
||||
value = ''
|
||||
|
||||
if nameUpper in COMMAND:
|
||||
return {
|
||||
"type": 'command',
|
||||
"name": nameUpper,
|
||||
"value": value.strip(),
|
||||
}
|
||||
|
||||
# data
|
||||
elif match_data:
|
||||
data = match_data.group(1)
|
||||
|
||||
return {
|
||||
"type": 'data',
|
||||
"data": data,
|
||||
}
|
||||
|
||||
else:
|
||||
return {
|
||||
"type": 'unknown',
|
||||
"value": line,
|
||||
}
|
||||
|
||||
|
||||
def getCourse(tjaHeaders, lines):
|
||||
headers = {
|
||||
"course": 'Oni',
|
||||
"level": 0,
|
||||
"balloon": [],
|
||||
"scoreInit": 100,
|
||||
"scoreDiff": 100,
|
||||
"ttRowBeat": 16,
|
||||
}
|
||||
|
||||
measures = []
|
||||
measureDividend = 4
|
||||
measureDivisor = 4
|
||||
measureProperties = {}
|
||||
measureData = ''
|
||||
measureEvents = []
|
||||
currentBranch = 'N'
|
||||
targetBranch = 'N'
|
||||
flagLevelhold = False
|
||||
|
||||
# Process lines
|
||||
for line in lines:
|
||||
if line["type"] == 'header':
|
||||
if line["name"] == 'COURSE':
|
||||
headers['course'] = line['value']
|
||||
|
||||
elif line["name"] == 'LEVEL':
|
||||
headers['level'] = int(line['value'])
|
||||
|
||||
elif line["name"] == 'BALLOON':
|
||||
if line['value']:
|
||||
balloons = [int(v) for v in line['value'].split(",")]
|
||||
else:
|
||||
balloons = []
|
||||
headers['balloon'] = balloons
|
||||
|
||||
elif line["name"] == 'SCOREINIT':
|
||||
headers['scoreInit'] = int(line['value'])
|
||||
|
||||
elif line["name"] == 'SCOREDIFF':
|
||||
headers['scoreDiff'] = int(line['value'])
|
||||
|
||||
elif line["name"] == 'TTROWBEAT':
|
||||
headers['ttRowBeat'] = int(line['value'])
|
||||
|
||||
elif line["type"] == 'command':
|
||||
if line["name"] == 'BRANCHSTART':
|
||||
if flagLevelhold:
|
||||
continue
|
||||
values = line['value'].split(',')
|
||||
if values[0] == 'r':
|
||||
if len(values) >= 3:
|
||||
targetBranch = 'M'
|
||||
elif len(values) == 2:
|
||||
targetBranch = 'E'
|
||||
else:
|
||||
targetBranch = 'N'
|
||||
elif values[0] == 'p':
|
||||
if len(values) >= 3 and float(values[2]) <= 100:
|
||||
targetBranch = 'M'
|
||||
elif len(values) >= 2 and float(values[1]) <= 100:
|
||||
targetBranch = 'E'
|
||||
else:
|
||||
targetBranch = 'N'
|
||||
|
||||
elif line["name"] == 'BRANCHEND':
|
||||
currentBranch = targetBranch
|
||||
|
||||
elif line["name"] == 'N':
|
||||
currentBranch = 'N'
|
||||
|
||||
elif line["name"] == 'E':
|
||||
currentBranch = 'E'
|
||||
|
||||
elif line["name"] == 'M':
|
||||
currentBranch = 'M'
|
||||
|
||||
elif line["name"] == 'START':
|
||||
currentBranch = 'N'
|
||||
targetBranch = 'N'
|
||||
flagLevelhold = False
|
||||
|
||||
elif line["name"] == 'END':
|
||||
currentBranch = 'N'
|
||||
targetBranch = 'N'
|
||||
flagLevelhold = False
|
||||
|
||||
else:
|
||||
if currentBranch != targetBranch:
|
||||
continue
|
||||
|
||||
if line['name'] == 'MEASURE':
|
||||
matchMeasure = re.match(r"(\d+)/(\d+)", line['value'])
|
||||
if not matchMeasure:
|
||||
continue
|
||||
measureDividend = int(matchMeasure.group(1))
|
||||
measureDivisor = int(matchMeasure.group(2))
|
||||
|
||||
elif line['name'] == 'GOGOSTART':
|
||||
measureEvents.append({
|
||||
"name": 'gogoStart',
|
||||
"position": len(measureData),
|
||||
})
|
||||
|
||||
elif line['name'] == 'GOGOEND':
|
||||
measureEvents.append({
|
||||
"name": 'gogoEnd',
|
||||
"position": len(measureData),
|
||||
})
|
||||
|
||||
elif line['name'] == 'SCROLL':
|
||||
measureEvents.append({
|
||||
"name": 'scroll',
|
||||
"position": len(measureData),
|
||||
"value": float(line['value']),
|
||||
})
|
||||
|
||||
elif line['name'] == 'BPMCHANGE':
|
||||
measureEvents.append({
|
||||
"name": 'bpm',
|
||||
"position": len(measureData),
|
||||
"value": float(line['value']),
|
||||
})
|
||||
|
||||
elif line['name'] == 'TTBREAK':
|
||||
measureProperties['ttBreak'] = True
|
||||
|
||||
elif line['name'] == 'LEVELHOLD':
|
||||
flagLevelhold = True
|
||||
|
||||
else:
|
||||
print(line['name']) # Unknown: BARLINEOFF, BARLINEON
|
||||
|
||||
elif line['type'] == 'data' and currentBranch is targetBranch:
|
||||
data = line['data']
|
||||
if data.endswith(','):
|
||||
measureData += data[0:-1]
|
||||
measure = {
|
||||
"length": [measureDividend, measureDivisor],
|
||||
"properties": measureProperties,
|
||||
"data": measureData,
|
||||
"events": measureEvents,
|
||||
}
|
||||
measures.append(measure)
|
||||
measureData = ''
|
||||
measureEvents = []
|
||||
measureProperties = {}
|
||||
else:
|
||||
measureData += data
|
||||
|
||||
if len(measures):
|
||||
# Make first BPM event
|
||||
firstBPMEventFound = False
|
||||
# Search for BPM event in the first measure
|
||||
for i in range(len(measures[0]['events'])):
|
||||
evt = measures[0]['events'][i]
|
||||
if evt.name == 'bpm' and evt.position == 0:
|
||||
firstBPMEventFound = True
|
||||
if not firstBPMEventFound:
|
||||
# noinspection PyTypeChecker
|
||||
measures[0]['events'].insert(0, {
|
||||
"name": 'bpm',
|
||||
"position": 0,
|
||||
"value": tjaHeaders['bpm'],
|
||||
})
|
||||
|
||||
# Helper values
|
||||
course = 0
|
||||
courseValue = headers['course'].lower()
|
||||
|
||||
if courseValue in ['easy', '0']:
|
||||
course = 0
|
||||
elif courseValue in ['normal', '1']:
|
||||
course = 1
|
||||
elif courseValue in ['hard', '2']:
|
||||
course = 2
|
||||
elif courseValue in ['oni', '3']:
|
||||
course = 3
|
||||
elif courseValue in ['ura', 'edit', '4']:
|
||||
course = 4
|
||||
|
||||
if measureData:
|
||||
measures.append({
|
||||
"length": [measureDividend, measureDivisor],
|
||||
"properties": measureProperties,
|
||||
"data": measureData,
|
||||
"events": measureEvents,
|
||||
})
|
||||
else:
|
||||
for event in measureEvents:
|
||||
event['position'] = len(measures[len(measures) - 1]['data'])
|
||||
# noinspection PyTypeChecker
|
||||
measures[len(measures) - 1]['events'].append(event)
|
||||
|
||||
# Output
|
||||
print(measures[len(measures) - 1])
|
||||
return course, headers, measures
|
||||
|
||||
|
||||
def parseTJA(tja):
|
||||
# Split by lines
|
||||
lines = tja.read().splitlines()
|
||||
|
||||
headers = {
|
||||
"title": '',
|
||||
"subtitle": '',
|
||||
"bpm": 120,
|
||||
"wave": '',
|
||||
"offset": 0,
|
||||
"demoStart": 0,
|
||||
"genre": '',
|
||||
}
|
||||
|
||||
# Line by line
|
||||
courses = {}
|
||||
courseLines = []
|
||||
for line in lines:
|
||||
if line == '':
|
||||
continue
|
||||
|
||||
parsed = parseLine(line)
|
||||
if parsed['type'] == 'header' and parsed['scope'] == 'global':
|
||||
if parsed['name'] == 'TITLE':
|
||||
headers['title'] = parsed['value']
|
||||
elif parsed['name'] == 'TITLEJA':
|
||||
headers['titleja'] = parsed['value']
|
||||
elif parsed['name'] == 'SUBTITLE':
|
||||
headers['subtitle'] = parsed['value']
|
||||
elif parsed['name'] == 'SUBTITLEJA':
|
||||
headers['subtitleja'] = parsed['value']
|
||||
elif parsed['name'] == 'BPM':
|
||||
headers['bpm'] = float(parsed['value'])
|
||||
elif parsed['name'] == 'WAVE':
|
||||
headers['wave'] = parsed['value']
|
||||
elif parsed['name'] == 'OFFSET':
|
||||
headers['offset'] = float(parsed['value'])
|
||||
elif parsed['name'] == 'DEMOSTART':
|
||||
headers['demoStart'] = float(parsed['value'])
|
||||
elif parsed['name'] == 'GENRE':
|
||||
headers['genre'] = parsed['value']
|
||||
|
||||
elif parsed['type'] == 'header' and parsed['scope'] == 'course':
|
||||
if parsed['name'] == 'COURSE':
|
||||
if courseLines:
|
||||
course = getCourse(headers, courseLines)
|
||||
courses[course[1]['course']] = course
|
||||
else:
|
||||
pass # This is the first course, so we haven't parsed its lines ye
|
||||
|
||||
courseLines.append(parsed)
|
||||
|
||||
elif parsed['type'] == 'command':
|
||||
courseLines.append(parsed)
|
||||
|
||||
elif parsed['type'] == 'data':
|
||||
courseLines.append(parsed)
|
||||
|
||||
# Parse the lines for the last course (since there are no more course headers left)
|
||||
if courseLines:
|
||||
course = getCourse(headers, courseLines)
|
||||
courses[course[1]['course']] = course
|
||||
|
||||
# Return
|
||||
return headers, courses
|
||||
Loading…
Reference in New Issue
Block a user