235 lines
9.8 KiB
Python
235 lines
9.8 KiB
Python
|
# vim: set fileencoding=utf-8
|
||
|
|
from typing import Any, Dict, Iterator, Optional, Tuple
|
||
|
|
|
||
|
|
from flask_caching import Cache # type: ignore
|
||
|
|
|
||
|
|
from bemani.backend.iidx import IIDXFactory, IIDXBase
|
||
|
|
from bemani.common import ValidatedDict, GameConstants
|
||
|
|
from bemani.data import Attempt, Data, Score, Song, UserID
|
||
|
|
from bemani.frontend.base import FrontendBase
|
||
|
|
|
||
|
|
|
||
|
|
class IIDXFrontend(FrontendBase):
|
||
|
|
|
||
|
|
game = GameConstants.IIDX
|
||
|
|
|
||
|
|
valid_charts = [
|
||
|
|
IIDXBase.CHART_TYPE_N7,
|
||
|
|
IIDXBase.CHART_TYPE_H7,
|
||
|
|
IIDXBase.CHART_TYPE_A7,
|
||
|
|
IIDXBase.CHART_TYPE_N14,
|
||
|
|
IIDXBase.CHART_TYPE_H14,
|
||
|
|
IIDXBase.CHART_TYPE_A14,
|
||
|
|
]
|
||
|
|
|
||
|
|
valid_rival_types = [
|
||
|
|
'sp_rival',
|
||
|
|
'dp_rival',
|
||
|
|
]
|
||
|
|
|
||
|
|
def __init__(self, data: Data, config: Dict[str, Any], cache: Cache) -> None:
|
||
|
|
super().__init__(data, config, cache)
|
||
|
|
self.machines: Dict[int, str] = {}
|
||
|
|
|
||
|
|
def all_games(self) -> Iterator[Tuple[str, int, str]]:
|
||
|
|
yield from IIDXFactory.all_games()
|
||
|
|
|
||
|
|
def get_duplicate_id(self, musicid: int, chart: int) -> Optional[Tuple[int, int]]:
|
||
|
|
modern_to_legacy_map = {
|
||
|
|
23066: 4213,
|
||
|
|
22068: 9203,
|
||
|
|
22052: 10203,
|
||
|
|
22039: 12201,
|
||
|
|
21201: 12204,
|
||
|
|
21064: 12206,
|
||
|
|
23077: 13215,
|
||
|
|
22025: 14202,
|
||
|
|
21068: 14210,
|
||
|
|
22069: 14211,
|
||
|
|
23070: 14214,
|
||
|
|
23069: 15202,
|
||
|
|
21063: 15204,
|
||
|
|
21065: 15205,
|
||
|
|
22028: 15207,
|
||
|
|
22049: 15208,
|
||
|
|
22043: 15209,
|
||
|
|
23060: 15211,
|
||
|
|
21062: 15215,
|
||
|
|
21067: 16207,
|
||
|
|
23062: 16209,
|
||
|
|
21066: 16212,
|
||
|
|
23030: 22096,
|
||
|
|
23051: 22097,
|
||
|
|
}
|
||
|
|
# Some charts were changed, and others kept the same on these
|
||
|
|
if chart in [0, 1, 2]:
|
||
|
|
modern_to_legacy_map[23065] = 9206
|
||
|
|
oldid = modern_to_legacy_map.get(musicid)
|
||
|
|
oldchart = chart
|
||
|
|
if oldid == 12204:
|
||
|
|
if oldchart == 1:
|
||
|
|
oldchart = 2
|
||
|
|
elif oldchart == 2:
|
||
|
|
oldchart = 1
|
||
|
|
if oldid is not None:
|
||
|
|
return (oldid, oldchart)
|
||
|
|
else:
|
||
|
|
return None
|
||
|
|
|
||
|
|
def format_dan_rank(self, rank: int) -> str:
|
||
|
|
if rank == -1:
|
||
|
|
return '--'
|
||
|
|
|
||
|
|
return {
|
||
|
|
IIDXBase.DAN_RANK_7_KYU: '七級',
|
||
|
|
IIDXBase.DAN_RANK_6_KYU: '六級',
|
||
|
|
IIDXBase.DAN_RANK_5_KYU: '五級',
|
||
|
|
IIDXBase.DAN_RANK_4_KYU: '四級',
|
||
|
|
IIDXBase.DAN_RANK_3_KYU: '三級',
|
||
|
|
IIDXBase.DAN_RANK_2_KYU: '二級',
|
||
|
|
IIDXBase.DAN_RANK_1_KYU: '一級',
|
||
|
|
IIDXBase.DAN_RANK_1_DAN: '初段',
|
||
|
|
IIDXBase.DAN_RANK_2_DAN: '二段',
|
||
|
|
IIDXBase.DAN_RANK_3_DAN: '三段',
|
||
|
|
IIDXBase.DAN_RANK_4_DAN: '四段',
|
||
|
|
IIDXBase.DAN_RANK_5_DAN: '五段',
|
||
|
|
IIDXBase.DAN_RANK_6_DAN: '六段',
|
||
|
|
IIDXBase.DAN_RANK_7_DAN: '七段',
|
||
|
|
IIDXBase.DAN_RANK_8_DAN: '八段',
|
||
|
|
IIDXBase.DAN_RANK_9_DAN: '九段',
|
||
|
|
IIDXBase.DAN_RANK_10_DAN: '十段',
|
||
|
|
IIDXBase.DAN_RANK_CHUDEN: '中伝',
|
||
|
|
IIDXBase.DAN_RANK_KAIDEN: '皆伝',
|
||
|
|
}[rank]
|
||
|
|
|
||
|
|
def format_flags(self, settings_dict: ValidatedDict) -> Dict[str, Any]:
|
||
|
|
flags = settings_dict.get_int('flags')
|
||
|
|
return {
|
||
|
|
'grade': (flags & 0x001) != 0,
|
||
|
|
'status': (flags & 0x002) != 0,
|
||
|
|
'difficulty': (flags & 0x004) != 0,
|
||
|
|
'alphabet': (flags & 0x008) != 0,
|
||
|
|
'rival_played': (flags & 0x010) != 0,
|
||
|
|
'rival_win_lose': (flags & 0x040) != 0,
|
||
|
|
'rival_info': (flags & 0x080) != 0,
|
||
|
|
'hide_play_count': (flags & 0x100) != 0,
|
||
|
|
'disable_song_preview': settings_dict.get_int('disable_song_preview') != 0,
|
||
|
|
'effector_lock': settings_dict.get_int('effector_lock') != 0,
|
||
|
|
}
|
||
|
|
|
||
|
|
def format_settings(self, settings_dict: ValidatedDict) -> Dict[str, Any]:
|
||
|
|
return {
|
||
|
|
'frame': settings_dict.get_int('frame'),
|
||
|
|
'turntable': settings_dict.get_int('turntable'),
|
||
|
|
'burst': settings_dict.get_int('burst'),
|
||
|
|
'bgm': settings_dict.get_int('bgm'),
|
||
|
|
'towel': settings_dict.get_int('towel'),
|
||
|
|
'judge_pos': settings_dict.get_int('judge_pos'),
|
||
|
|
'voice': settings_dict.get_int('voice'),
|
||
|
|
'noteskin': settings_dict.get_int('noteskin'),
|
||
|
|
'full_combo': settings_dict.get_int('full_combo'),
|
||
|
|
'beam': settings_dict.get_int('beam'),
|
||
|
|
'judge': settings_dict.get_int('judge'),
|
||
|
|
'pacemaker': settings_dict.get_int('pacemaker'),
|
||
|
|
'effector_preset': settings_dict.get_int('effector_preset'),
|
||
|
|
}
|
||
|
|
|
||
|
|
def format_profile(self, profile: ValidatedDict, playstats: ValidatedDict) -> Dict[str, Any]:
|
||
|
|
formatted_profile = super().format_profile(profile, playstats)
|
||
|
|
formatted_profile.update({
|
||
|
|
'arcade': "",
|
||
|
|
'prefecture': profile.get_int('pid', 51),
|
||
|
|
'settings': self.format_settings(profile.get_dict('settings')),
|
||
|
|
'flags': self.format_flags(profile.get_dict('settings')),
|
||
|
|
'sdjp': playstats.get_int('single_dj_points'),
|
||
|
|
'ddjp': playstats.get_int('double_dj_points'),
|
||
|
|
'sp': playstats.get_int('single_plays'),
|
||
|
|
'dp': playstats.get_int('double_plays'),
|
||
|
|
'sdan': self.format_dan_rank(profile.get_int('sgrade', -1)),
|
||
|
|
'ddan': self.format_dan_rank(profile.get_int('dgrade', -1)),
|
||
|
|
'srank': profile.get_int('sgrade', -1),
|
||
|
|
'drank': profile.get_int('dgrade', -1),
|
||
|
|
})
|
||
|
|
if 'shop_location' in profile:
|
||
|
|
shop_id = profile.get_int('shop_location')
|
||
|
|
if shop_id in self.machines:
|
||
|
|
formatted_profile['arcade'] = self.machines[shop_id]
|
||
|
|
else:
|
||
|
|
pcbid = self.data.local.machine.from_machine_id(shop_id)
|
||
|
|
if pcbid is not None:
|
||
|
|
machine = self.data.local.machine.get_machine(pcbid)
|
||
|
|
self.machines[shop_id] = machine.name
|
||
|
|
formatted_profile['arcade'] = machine.name
|
||
|
|
return formatted_profile
|
||
|
|
|
||
|
|
def format_score(self, userid: UserID, score: Score) -> Dict[str, Any]:
|
||
|
|
formatted_score = super().format_score(userid, score)
|
||
|
|
formatted_score['miss_count'] = score.data.get_int('miss_count')
|
||
|
|
formatted_score['lamp'] = score.data.get_int('clear_status')
|
||
|
|
formatted_score['status'] = {
|
||
|
|
IIDXBase.CLEAR_STATUS_NO_PLAY: 'NO PLAY',
|
||
|
|
IIDXBase.CLEAR_STATUS_FAILED: 'FAILED',
|
||
|
|
IIDXBase.CLEAR_STATUS_ASSIST_CLEAR: 'ASSIST CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_EASY_CLEAR: 'EASY CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_CLEAR: 'CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_HARD_CLEAR: 'HARD CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_EX_HARD_CLEAR: 'EX HARD CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_FULL_COMBO: 'FULL COMBO',
|
||
|
|
}.get(score.data.get_int('clear_status'), 'NO PLAY')
|
||
|
|
return formatted_score
|
||
|
|
|
||
|
|
def format_top_score(self, userid: UserID, score: Score) -> Dict[str, Any]:
|
||
|
|
formatted_score = super().format_score(userid, score)
|
||
|
|
formatted_score['miss_count'] = score.data.get_int('miss_count')
|
||
|
|
formatted_score['lamp'] = score.data.get_int('clear_status')
|
||
|
|
formatted_score['ghost'] = [x for x in (score.data.get_bytes('ghost') or b'')]
|
||
|
|
formatted_score['status'] = {
|
||
|
|
IIDXBase.CLEAR_STATUS_NO_PLAY: 'NO PLAY',
|
||
|
|
IIDXBase.CLEAR_STATUS_FAILED: 'FAILED',
|
||
|
|
IIDXBase.CLEAR_STATUS_ASSIST_CLEAR: 'ASSIST CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_EASY_CLEAR: 'EASY CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_CLEAR: 'CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_HARD_CLEAR: 'HARD CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_EX_HARD_CLEAR: 'EX HARD CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_FULL_COMBO: 'FULL COMBO',
|
||
|
|
}.get(score.data.get_int('clear_status'), 'NO PLAY')
|
||
|
|
return formatted_score
|
||
|
|
|
||
|
|
def format_attempt(self, userid: UserID, attempt: Attempt) -> Dict[str, Any]:
|
||
|
|
formatted_attempt = super().format_attempt(userid, attempt)
|
||
|
|
formatted_attempt['miss_count'] = attempt.data.get_int('miss_count')
|
||
|
|
formatted_attempt['status'] = {
|
||
|
|
IIDXBase.CLEAR_STATUS_NO_PLAY: 'NO PLAY',
|
||
|
|
IIDXBase.CLEAR_STATUS_FAILED: 'FAILED',
|
||
|
|
IIDXBase.CLEAR_STATUS_ASSIST_CLEAR: 'ASSIST CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_EASY_CLEAR: 'EASY CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_CLEAR: 'CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_HARD_CLEAR: 'HARD CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_EX_HARD_CLEAR: 'EX HARD CLEAR',
|
||
|
|
IIDXBase.CLEAR_STATUS_FULL_COMBO: 'FULL COMBO',
|
||
|
|
}.get(attempt.data.get_int('clear_status'), 'NO PLAY')
|
||
|
|
return formatted_attempt
|
||
|
|
|
||
|
|
def format_song(self, song: Song) -> Dict[str, Any]:
|
||
|
|
difficulties = [0, 0, 0, 0, 0, 0]
|
||
|
|
notecounts = [0, 0, 0, 0, 0, 0]
|
||
|
|
difficulties[song.chart] = song.data.get_int('difficulty', 13)
|
||
|
|
notecounts[song.chart] = song.data.get_int('notecount', 5730)
|
||
|
|
|
||
|
|
formatted_song = super().format_song(song)
|
||
|
|
formatted_song['bpm_min'] = song.data.get_int('bpm_min', 120)
|
||
|
|
formatted_song['bpm_max'] = song.data.get_int('bpm_max', 120)
|
||
|
|
formatted_song['difficulties'] = difficulties
|
||
|
|
formatted_song['notecounts'] = notecounts
|
||
|
|
return formatted_song
|
||
|
|
|
||
|
|
def merge_song(self, existing: Dict[str, Any], new: Song) -> Dict[str, Any]:
|
||
|
|
new_song = super().merge_song(existing, new)
|
||
|
|
if (
|
||
|
|
existing['difficulties'][new.chart] == 0 or
|
||
|
|
existing['notecounts'][new.chart] == 0
|
||
|
|
):
|
||
|
|
new_song['difficulties'][new.chart] = new.data.get_int('difficulty', 13)
|
||
|
|
new_song['notecounts'][new.chart] = new.data.get_int('notecount', 5730)
|
||
|
|
return new_song
|