368 lines
16 KiB
Python
368 lines
16 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""舞萌DX (maimai DX) .ma2 谱面解析。
|
||
|
||
指令 -> 音符类型 的对应表来自反编译源码
|
||
source/Assembly-CSharp/Manager/Ma2fileRecordID.cs
|
||
（s_Ma2fileRecord_Data 里 Ma2Category.MA2_Note 的 102 条记录）
|
||
计分类别的划分复刻
|
||
source/Assembly-CSharp/Manager/GamePlayManager.cs NoteType2ScoreType()
|
||
理论分权重来自
|
||
source/Assembly-CSharp/Manager/NotesReader.cs calcTotal()
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
from dataclasses import dataclass, field
|
||
from typing import Dict, List, Optional
|
||
|
||
# ---------------------------------------------------------------- 计分权重
|
||
WEIGHT_TAP = 500 # TAP / XTP / STR / XST / TTP（Touch 与 Tap 同为 500）
|
||
WEIGHT_HOLD = 1000 # HLD / XHO / THO
|
||
WEIGHT_SLIDE = 1500 # SI_ SCL SCR SUL SUR SSL SSR SV_ SXL SXR SLL SLR SF_
|
||
WEIGHT_BREAK = 2500 # BRK / BST / BRTAP / BRHLD / BRSTR / BRSI_ ... 以及 ExBreak 系
|
||
BREAK_BONUS_THEORY = 100 # 每个绝赞(Break)的理论奖励分
|
||
|
||
CATEGORY_TAP = "Tap"
|
||
CATEGORY_HOLD = "Hold"
|
||
CATEGORY_SLIDE = "Slide"
|
||
CATEGORY_BREAK = "Break"
|
||
CATEGORY_CONNECT = "ConnectSlide"
|
||
|
||
CATEGORY_ORDER = (CATEGORY_TAP, CATEGORY_HOLD, CATEGORY_SLIDE, CATEGORY_BREAK)
|
||
CATEGORY_WEIGHT = {
|
||
CATEGORY_TAP: WEIGHT_TAP,
|
||
CATEGORY_HOLD: WEIGHT_HOLD,
|
||
CATEGORY_SLIDE: WEIGHT_SLIDE,
|
||
CATEGORY_BREAK: WEIGHT_BREAK,
|
||
}
|
||
CATEGORY_LABEL = {
|
||
CATEGORY_TAP: "Tap/Touch",
|
||
CATEGORY_HOLD: "Hold",
|
||
CATEGORY_SLIDE: "Slide",
|
||
CATEGORY_BREAK: "Break",
|
||
}
|
||
|
||
# ------------------------------------------------- ma2 指令 -> NotesTypeID.Def
|
||
NOTE_COMMAND_TYPES: Dict[str, str] = {
|
||
# --- 现行指令 ---
|
||
"TAP": "Begin", "BRK": "Break", "XTP": "ExTap", "HLD": "Hold", "XHO": "ExHold",
|
||
"STR": "Star", "BST": "BreakStar", "XST": "ExStar",
|
||
"TTP": "TouchTap", "THO": "TouchHold",
|
||
"SI_": "Slide", "SCL": "Slide", "SCR": "Slide", "SUL": "Slide", "SUR": "Slide",
|
||
"SSL": "Slide", "SSR": "Slide", "SV_": "Slide", "SXL": "Slide", "SXR": "Slide",
|
||
"SLL": "Slide", "SLR": "Slide", "SF_": "Slide",
|
||
# --- 旧式别名（游戏数据里真实存在）---
|
||
"NMTAP": "Begin", "BRTAP": "Break", "EXTAP": "ExTap", "BXTAP": "ExBreakTap",
|
||
"NMHLD": "Hold", "BRHLD": "BreakHold", "EXHLD": "ExHold", "BXHLD": "ExBreakHold",
|
||
"NMSTR": "Star", "BRSTR": "BreakStar", "EXSTR": "ExStar", "BXSTR": "ExBreakStar",
|
||
"NMTTP": "TouchTap", "NMTHO": "TouchHold",
|
||
"NMSI_": "Slide", "BRSI_": "BreakSlide", "EXSI_": "ExSlide", "BXSI_": "ExBreakSlide", "CNSI_": "ConnectSlide",
|
||
"NMSCL": "Slide", "BRSCL": "BreakSlide", "EXSCL": "ExSlide", "BXSCL": "ExBreakSlide", "CNSCL": "ConnectSlide",
|
||
"NMSCR": "Slide", "BRSCR": "BreakSlide", "EXSCR": "ExSlide", "BXSCR": "ExBreakSlide", "CNSCR": "ConnectSlide",
|
||
"NMSUL": "Slide", "BRSUL": "BreakSlide", "EXSUL": "ExSlide", "BXSUL": "ExBreakSlide", "CNSUL": "ConnectSlide",
|
||
"NMSUR": "Slide", "BRSUR": "BreakSlide", "EXSUR": "ExSlide", "BXSUR": "ExBreakSlide", "CNSUR": "ConnectSlide",
|
||
"NMSSL": "Slide", "BRSSL": "BreakSlide", "EXSSL": "ExSlide", "BXSSL": "ExBreakSlide", "CNSSL": "ConnectSlide",
|
||
"NMSSR": "Slide", "BRSSR": "BreakSlide", "EXSSR": "ExSlide", "BXSSR": "ExBreakSlide", "CNSSR": "ConnectSlide",
|
||
"NMSV_": "Slide", "BRSV_": "BreakSlide", "EXSV_": "ExSlide", "BXSV_": "ExBreakSlide", "CNSV_": "ConnectSlide",
|
||
"NMSXL": "Slide", "BRSXL": "BreakSlide", "EXSXL": "ExSlide", "BXSXL": "ExBreakSlide", "CNSXL": "ConnectSlide",
|
||
"NMSXR": "Slide", "BRSXR": "BreakSlide", "EXSXR": "ExSlide", "BXSXR": "ExBreakSlide", "CNSXR": "ConnectSlide",
|
||
"NMSLL": "Slide", "BRSLL": "BreakSlide", "EXSLL": "ExSlide", "BXSLL": "ExBreakSlide", "CNSLL": "ConnectSlide",
|
||
"NMSLR": "Slide", "BRSLR": "BreakSlide", "EXSLR": "ExSlide", "BXSLR": "ExBreakSlide", "CNSLR": "ConnectSlide",
|
||
"NMSF_": "Slide", "BRSF_": "BreakSlide", "EXSF_": "ExSlide", "BXSF_": "ExBreakSlide", "CNSF_": "ConnectSlide",
|
||
}
|
||
|
||
_BREAK_DEFS = frozenset((
|
||
"Break", "BreakHold", "BreakStar", "BreakSlide",
|
||
"ExBreakTap", "ExBreakHold", "ExBreakStar", "ExBreakSlide",
|
||
))
|
||
_SLIDE_DEFS = frozenset(("Slide", "BreakSlide", "ExSlide", "ExBreakSlide"))
|
||
_HOLD_DEFS = frozenset(("Hold", "BreakHold", "ExHold", "ExBreakHold", "TouchHold"))
|
||
_TOUCH_DEFS = frozenset(("TouchTap", "TouchHold"))
|
||
_CONNECT_DEFS = frozenset(("ConnectSlide",))
|
||
|
||
# 判定时机只有 Critical / TooLate 的 EX 音符（ExTap/ExStar/ExBreakTap/ExBreakStar）
|
||
EX_DEFS = frozenset(("ExTap", "ExStar", "ExBreakTap", "ExBreakStar"))
|
||
|
||
# ---------------------------------------------------------------- 音符组
|
||
# 「音符组」= 计分行为完全相同的一类音符：权重、可选判定、失分档位、绝赞奖励档位都一样。
|
||
# 划分依据（反编译源码）：
|
||
# * 计分类型 NoteScore.EScoreType 由 GamePlayManager.NoteType2ScoreType() 决定
|
||
# （Break 系 -> Break，滑条 -> Slide，长按 -> Hold，其余 -> Tap/Touch）
|
||
# * 判定时机表 NoteJudge.EJudgeType 由各音符类决定：
|
||
# TapNote/StarNote -> Tap（或 ExNote 时 ExTap）
|
||
# HoldNote/BreakHoldNote/TouchHoldC -> 长按结算 JudgeHoldTotal（不分 2nd/3rd 档）
|
||
# SlideRoot/SlideFan -> SlideOut（不分 2nd/3rd 档，也没有非 Critical 的 Perfect）
|
||
# BreakNote/BreakStarNote -> Break 表（Great 有 1250/1500/2000 三档）
|
||
# * ExTap 判定表 [-9,-9,-9,-9,-9,-9,-9,0,9,9,9,9,9,9,9] 使非 Critical 的档位全部不可达
|
||
# -> ExTap/ExStar（含 ExBreak 版）只能 Critical 或 Miss
|
||
# * ExHold/ExBreakHold 的头部用 ExTap 表，但结果还要过 JudgeHoldTotal 重映射，
|
||
# 最终仍可得到 Critical/Perfect/Great/Good/Miss -> 与普通 Hold / BreakHold 相同
|
||
# * ExSlide / ExBreakSlide 的 JudgeType 恒为 SlideOut，且 isBreak() 不含 ExBreakSlide
|
||
# -> 计分与普通 Slide 完全一致（游戏数据里这两种指令出现次数为 0）
|
||
GROUP_TAP = "tap" # TAP NMTAP STR NMSTR TTP NMTTP 权重 500
|
||
GROUP_HOLD = "hold" # HLD NMHLD XHO EXHLD THO NMTHO 权重 1000
|
||
GROUP_SLIDE = "slide" # SI_ SCL SCR SUL SUR SSL SSR SV_ SXL SXR SLL SLR SF_ + NM/EX/CN 变体 权重 1500
|
||
GROUP_BREAK = "break" # BRK BRTAP BST BRSTR 权重 2500，Great 三档、Perfect 奖励两档
|
||
GROUP_BREAK_FLAT = "break_flat" # BRHLD BXHLD 权重 2500，长按判定，不分档
|
||
GROUP_BREAK_SLIDE = "break_slide" # BRSI_ BRSCL BRSCR BRSUL BRSUR BRSSL BRSSR BRSV_ BRSXL BRSXR BRSLL BRSLR BRSF_
|
||
GROUP_EX = "ex" # XTP EXTAP XST EXSTR 权重 500，只能 Critical / Miss
|
||
GROUP_EX_BREAK = "ex_break" # BXTAP BXSTR 权重 2500，只能 Critical / Miss
|
||
|
||
GROUP_ORDER = (GROUP_TAP, GROUP_HOLD, GROUP_SLIDE, GROUP_BREAK,
|
||
GROUP_BREAK_FLAT, GROUP_BREAK_SLIDE, GROUP_EX, GROUP_EX_BREAK)
|
||
GROUP_LABEL = {
|
||
GROUP_TAP: "Tap/Star/Touch",
|
||
GROUP_HOLD: "Hold",
|
||
GROUP_SLIDE: "Slide",
|
||
GROUP_BREAK: "Break",
|
||
GROUP_BREAK_FLAT: "Break(长按判定)",
|
||
GROUP_BREAK_SLIDE: "Break(滑条判定)",
|
||
GROUP_EX: "EX Tap/Star",
|
||
GROUP_EX_BREAK: "EX Break",
|
||
}
|
||
GROUP_WEIGHT = {
|
||
GROUP_TAP: WEIGHT_TAP,
|
||
GROUP_HOLD: WEIGHT_HOLD,
|
||
GROUP_SLIDE: WEIGHT_SLIDE,
|
||
GROUP_BREAK: WEIGHT_BREAK,
|
||
GROUP_BREAK_FLAT: WEIGHT_BREAK,
|
||
GROUP_BREAK_SLIDE: WEIGHT_BREAK,
|
||
GROUP_EX: WEIGHT_TAP,
|
||
GROUP_EX_BREAK: WEIGHT_BREAK,
|
||
}
|
||
GROUP_IS_BREAK = frozenset((GROUP_BREAK, GROUP_BREAK_FLAT, GROUP_BREAK_SLIDE, GROUP_EX_BREAK))
|
||
|
||
_GROUP_BY_DEF = {
|
||
"Begin": GROUP_TAP, "Star": GROUP_TAP, "TouchTap": GROUP_TAP,
|
||
"Hold": GROUP_HOLD, "ExHold": GROUP_HOLD, "TouchHold": GROUP_HOLD,
|
||
"Slide": GROUP_SLIDE, "ExSlide": GROUP_SLIDE, "ExBreakSlide": GROUP_SLIDE,
|
||
"Break": GROUP_BREAK, "BreakStar": GROUP_BREAK,
|
||
"BreakHold": GROUP_BREAK_FLAT, "ExBreakHold": GROUP_BREAK_FLAT,
|
||
"BreakSlide": GROUP_BREAK_SLIDE,
|
||
"ExTap": GROUP_EX, "ExStar": GROUP_EX,
|
||
"ExBreakTap": GROUP_EX_BREAK, "ExBreakStar": GROUP_EX_BREAK,
|
||
}
|
||
_CATEGORY_BY_GROUP = {
|
||
GROUP_TAP: CATEGORY_TAP, GROUP_HOLD: CATEGORY_HOLD, GROUP_SLIDE: CATEGORY_SLIDE,
|
||
GROUP_BREAK: CATEGORY_BREAK, GROUP_BREAK_FLAT: CATEGORY_BREAK,
|
||
GROUP_BREAK_SLIDE: CATEGORY_BREAK, GROUP_EX: CATEGORY_TAP, GROUP_EX_BREAK: CATEGORY_BREAK,
|
||
}
|
||
|
||
|
||
def note_group(def_name: str) -> Optional[str]:
|
||
"""NotesTypeID.Def 名 -> 音符组；接续滑条等不计分音符返回 None。"""
|
||
return _GROUP_BY_DEF.get(def_name)
|
||
|
||
|
||
def score_category(def_name: str) -> Optional[str]:
|
||
"""复刻 GamePlayManager.NoteType2ScoreType() 的粗分类（Tap/Hold/Slide/Break）。"""
|
||
grp = note_group(def_name)
|
||
if grp is None:
|
||
return None
|
||
return _CATEGORY_BY_GROUP[grp]
|
||
|
||
|
||
# ------------------------------------------------------------------ 谱面对象
|
||
@dataclass
|
||
class Chart:
|
||
path: str = "<memory>"
|
||
command_counts: Dict[str, int] = field(default_factory=dict)
|
||
category_counts: Dict[str, int] = field(default_factory=dict)
|
||
group_counts: Dict[str, int] = field(default_factory=dict)
|
||
connect_count: int = 0
|
||
header: Dict[str, str] = field(default_factory=dict)
|
||
totals: Dict[str, int] = field(default_factory=dict)
|
||
warnings: List[str] = field(default_factory=list)
|
||
# 外部补充的信息（游戏路径模式）
|
||
music_id: Optional[int] = None
|
||
difficulty_index: Optional[int] = None
|
||
difficulty_name: Optional[str] = None
|
||
song_name: Optional[str] = None
|
||
level_text: Optional[str] = None
|
||
max_notes_xml: Optional[int] = None
|
||
|
||
@property
|
||
def total_notes(self) -> int:
|
||
"""计分音符总数 N（不含接续滑条）。"""
|
||
return sum(self.group_counts.get(g, 0) for g in GROUP_ORDER)
|
||
|
||
@property
|
||
def declared_total(self) -> Optional[int]:
|
||
return self.totals.get("T_NUM_ALL")
|
||
|
||
def counts(self) -> Dict[str, int]:
|
||
"""按音符组统计（供计分模型使用）。"""
|
||
return {g: self.group_counts.get(g, 0) for g in GROUP_ORDER}
|
||
|
||
def category_summary(self) -> Dict[str, int]:
|
||
return {c: self.category_counts.get(c, 0) for c in CATEGORY_ORDER}
|
||
|
||
|
||
def parse_ma2(path: str) -> Chart:
|
||
with open(path, "rb") as fp:
|
||
raw = fp.read()
|
||
chart = parse_ma2_bytes(raw, path)
|
||
return chart
|
||
|
||
|
||
def parse_ma2_bytes(raw: bytes, path: str = "<memory>") -> Chart:
|
||
chart = Chart(path=path)
|
||
text = raw.decode("latin1")
|
||
unknown = set()
|
||
for line in text.replace("\r\n", "\n").replace("\r", "\n").split("\n"):
|
||
line = line.strip()
|
||
if not line:
|
||
continue
|
||
parts = line.split("\t")
|
||
if len(parts) == 1:
|
||
parts = line.split()
|
||
if not parts:
|
||
continue
|
||
cmd = parts[0]
|
||
args = parts[1:]
|
||
|
||
def_name = NOTE_COMMAND_TYPES.get(cmd)
|
||
if def_name is not None:
|
||
chart.command_counts[cmd] = chart.command_counts.get(cmd, 0) + 1
|
||
grp = note_group(def_name)
|
||
if grp is None:
|
||
chart.connect_count += 1
|
||
else:
|
||
chart.group_counts[grp] = chart.group_counts.get(grp, 0) + 1
|
||
cat = score_category(def_name)
|
||
chart.category_counts[cat] = chart.category_counts.get(cat, 0) + 1
|
||
continue
|
||
|
||
if cmd.startswith("T_") or cmd.startswith("TTM_"):
|
||
if args:
|
||
try:
|
||
chart.totals[cmd] = int(args[0])
|
||
except ValueError:
|
||
pass
|
||
continue
|
||
|
||
if cmd not in chart.header and args:
|
||
chart.header[cmd] = "\t".join(args)
|
||
elif not args and cmd.isupper() and len(cmd) > 2:
|
||
unknown.add(cmd)
|
||
|
||
declared = chart.declared_total
|
||
if declared is not None and declared != chart.total_notes:
|
||
chart.warnings.append(
|
||
"T_NUM_ALL=%d 与解析出的计分音符数 %d 不一致" % (declared, chart.total_notes)
|
||
)
|
||
return chart
|
||
|
||
|
||
# ------------------------------------------------------------ 游戏路径解析
|
||
DIFFICULTY_NAMES = ("Basic", "Advanced", "Expert", "Master", "Re:Master")
|
||
_DIFFICULTY_ALIASES = {
|
||
"basic": 0, "bas": 0, "bsc": 0, "0": 0, "00": 0,
|
||
"advanced": 1, "adv": 1, "1": 1, "01": 1,
|
||
"expert": 2, "exp": 2, "2": 2, "02": 2,
|
||
"master": 3, "mas": 3, "mst": 3, "3": 3, "03": 3,
|
||
"remaster": 4, "re:master": 4, "remas": 4, "re": 4, "4": 4, "04": 4,
|
||
}
|
||
|
||
|
||
def parse_difficulty(text) -> int:
|
||
if isinstance(text, int):
|
||
if 0 <= text <= 4:
|
||
return text
|
||
raise ValueError("难度序号必须是 0~4")
|
||
key = str(text).strip().lower()
|
||
if key in _DIFFICULTY_ALIASES:
|
||
return _DIFFICULTY_ALIASES[key]
|
||
raise ValueError("无法识别的难度：%s（可用 0=BASIC 1=ADVANCED 2=EXPERT 3=MASTER 4=RE:MASTER，"
|
||
"或直接写名字）" % text)
|
||
|
||
|
||
def find_music_dir(game_path: str) -> str:
|
||
"""把「游戏根目录 / Sinmai_Data / StreamingAssets / A000」都归一化到 A000/music。"""
|
||
p = os.path.abspath(os.path.expanduser(game_path))
|
||
if not os.path.isdir(p):
|
||
raise FileNotFoundError("路径不存在：%s" % p)
|
||
if os.path.basename(p).lower() == "music" and os.path.basename(os.path.dirname(p)).upper() == "A000":
|
||
return p
|
||
candidates = [
|
||
p,
|
||
os.path.join(p, "music"),
|
||
os.path.join(p, "A000", "music"),
|
||
os.path.join(p, "StreamingAssets", "A000", "music"),
|
||
os.path.join(p, "Sinmai_Data", "StreamingAssets", "A000", "music"),
|
||
os.path.join(p, "Package", "Sinmai_Data", "StreamingAssets", "A000", "music"),
|
||
]
|
||
for c in candidates:
|
||
if os.path.isdir(c) and os.path.basename(c).lower() == "music":
|
||
return c
|
||
raise FileNotFoundError("在 %s 下找不到 A000/music 目录" % p)
|
||
|
||
|
||
def music_dir_path(music_dir: str, music_id: int) -> str:
|
||
return os.path.join(music_dir, "music%06d" % music_id)
|
||
|
||
|
||
def chart_path(music_dir: str, music_id: int, difficulty: int) -> str:
|
||
return os.path.join(music_dir_path(music_dir, music_id), "%06d_%02d.ma2" % (music_id, difficulty))
|
||
|
||
|
||
def parse_music_xml(path: str) -> Dict[str, object]:
|
||
"""读取 Music.xml（只用标准库），取出曲名与各难度信息。"""
|
||
import xml.etree.ElementTree as ET
|
||
|
||
info: Dict[str, object] = {"notes": []}
|
||
try:
|
||
root = ET.parse(path).getroot()
|
||
except Exception as exc: # noqa: BLE001
|
||
return {"error": str(exc), "notes": []}
|
||
|
||
def text_of(node, tag):
|
||
child = node.find(tag)
|
||
if child is None:
|
||
return None
|
||
if child.text and child.text.strip():
|
||
return child.text.strip()
|
||
sub = child.find("str")
|
||
if sub is not None and sub.text and sub.text.strip():
|
||
return sub.text.strip()
|
||
return None
|
||
|
||
info["name"] = text_of(root, "name")
|
||
info["artist"] = text_of(root, "artistName")
|
||
info["genre"] = text_of(root, "genreName")
|
||
info["bpm"] = text_of(root, "bpm")
|
||
notes_data = root.find("notesData")
|
||
notes = []
|
||
if notes_data is not None:
|
||
for node in notes_data.findall("Notes"):
|
||
f = node.find("file/path")
|
||
notes.append({
|
||
"file": f.text.strip() if f is not None and f.text else None,
|
||
"level": text_of(node, "level"),
|
||
"levelDecimal": text_of(node, "levelDecimal"),
|
||
"musicLevelID": text_of(node, "musicLevelID"),
|
||
"maxNotes": text_of(node, "maxNotes"),
|
||
"designer": text_of(node, "notesDesigner"),
|
||
"isEnable": text_of(node, "isEnable"),
|
||
})
|
||
info["notes"] = notes
|
||
return info
|
||
|
||
|
||
def level_display(level, level_decimal) -> str:
|
||
"""12 + levelDecimal 4 -> '12+'；10 + 2 -> '10+'；7 + 0 -> '7'。"""
|
||
try:
|
||
lv = int(level)
|
||
except (TypeError, ValueError):
|
||
return ""
|
||
try:
|
||
dec = int(level_decimal or 0)
|
||
except (TypeError, ValueError):
|
||
dec = 0
|
||
if dec >= 3:
|
||
return "%d+" % lv
|
||
if dec > 0:
|
||
return "%d.%d" % (lv, dec)
|
||
return "%d" % lv
|