2394 lines
95 KiB
Python
2394 lines
95 KiB
Python
# vim: set fileencoding=utf-8
|
|
import random
|
|
import struct
|
|
from typing import Optional, Dict, Any, List, Tuple
|
|
from typing_extensions import Final
|
|
|
|
from bemani.backend.iidx.base import IIDXBase
|
|
from bemani.backend.iidx.course import IIDXCourse
|
|
from bemani.backend.iidx.pendual import IIDXPendual
|
|
|
|
from bemani.common import (
|
|
Profile,
|
|
ValidatedDict,
|
|
VersionConstants,
|
|
BroadcastConstants,
|
|
Time,
|
|
ID,
|
|
)
|
|
from bemani.data import Data, UserID
|
|
from bemani.protocol import Node
|
|
|
|
|
|
class IIDXCopula(IIDXCourse, IIDXBase):
|
|
|
|
name: str = "Beatmania IIDX copula"
|
|
version: int = VersionConstants.IIDX_COPULA
|
|
|
|
GAME_CLTYPE_SINGLE: Final[int] = 0
|
|
GAME_CLTYPE_DOUBLE: Final[int] = 1
|
|
|
|
DAN_STAGES: Final[int] = 4
|
|
|
|
GAME_CLEAR_STATUS_NO_PLAY: Final[int] = 0
|
|
GAME_CLEAR_STATUS_FAILED: Final[int] = 1
|
|
GAME_CLEAR_STATUS_ASSIST_CLEAR: Final[int] = 2
|
|
GAME_CLEAR_STATUS_EASY_CLEAR: Final[int] = 3
|
|
GAME_CLEAR_STATUS_CLEAR: Final[int] = 4
|
|
GAME_CLEAR_STATUS_HARD_CLEAR: Final[int] = 5
|
|
GAME_CLEAR_STATUS_EX_HARD_CLEAR: Final[int] = 6
|
|
GAME_CLEAR_STATUS_FULL_COMBO: Final[int] = 7
|
|
|
|
GAME_GHOST_TYPE_RIVAL: Final[int] = 1
|
|
GAME_GHOST_TYPE_GLOBAL_TOP: Final[int] = 2
|
|
GAME_GHOST_TYPE_GLOBAL_AVERAGE: Final[int] = 3
|
|
GAME_GHOST_TYPE_LOCAL_TOP: Final[int] = 4
|
|
GAME_GHOST_TYPE_LOCAL_AVERAGE: Final[int] = 5
|
|
GAME_GHOST_TYPE_DAN_TOP: Final[int] = 6
|
|
GAME_GHOST_TYPE_DAN_AVERAGE: Final[int] = 7
|
|
GAME_GHOST_TYPE_RIVAL_TOP: Final[int] = 8
|
|
GAME_GHOST_TYPE_RIVAL_AVERAGE: Final[int] = 9
|
|
|
|
GAME_GHOST_LENGTH: Final[int] = 64
|
|
|
|
GAME_SP_DAN_RANK_7_KYU: Final[int] = 0
|
|
GAME_SP_DAN_RANK_6_KYU: Final[int] = 1
|
|
GAME_SP_DAN_RANK_5_KYU: Final[int] = 2
|
|
GAME_SP_DAN_RANK_4_KYU: Final[int] = 3
|
|
GAME_SP_DAN_RANK_3_KYU: Final[int] = 4
|
|
GAME_SP_DAN_RANK_2_KYU: Final[int] = 5
|
|
GAME_SP_DAN_RANK_1_KYU: Final[int] = 6
|
|
GAME_SP_DAN_RANK_1_DAN: Final[int] = 7
|
|
GAME_SP_DAN_RANK_2_DAN: Final[int] = 8
|
|
GAME_SP_DAN_RANK_3_DAN: Final[int] = 9
|
|
GAME_SP_DAN_RANK_4_DAN: Final[int] = 10
|
|
GAME_SP_DAN_RANK_5_DAN: Final[int] = 11
|
|
GAME_SP_DAN_RANK_6_DAN: Final[int] = 12
|
|
GAME_SP_DAN_RANK_7_DAN: Final[int] = 13
|
|
GAME_SP_DAN_RANK_8_DAN: Final[int] = 14
|
|
GAME_SP_DAN_RANK_9_DAN: Final[int] = 15
|
|
GAME_SP_DAN_RANK_10_DAN: Final[int] = 16
|
|
GAME_SP_DAN_RANK_CHUDEN: Final[int] = 17
|
|
GAME_SP_DAN_RANK_KAIDEN: Final[int] = 18
|
|
|
|
GAME_DP_DAN_RANK_7_KYU: Final[int] = 0
|
|
GAME_DP_DAN_RANK_6_KYU: Final[int] = 1
|
|
GAME_DP_DAN_RANK_5_KYU: Final[int] = 2
|
|
GAME_DP_DAN_RANK_4_KYU: Final[int] = 3
|
|
GAME_DP_DAN_RANK_3_KYU: Final[int] = 4
|
|
GAME_DP_DAN_RANK_2_KYU: Final[int] = 5
|
|
GAME_DP_DAN_RANK_1_KYU: Final[int] = 6
|
|
GAME_DP_DAN_RANK_1_DAN: Final[int] = 7
|
|
GAME_DP_DAN_RANK_2_DAN: Final[int] = 8
|
|
GAME_DP_DAN_RANK_3_DAN: Final[int] = 9
|
|
GAME_DP_DAN_RANK_4_DAN: Final[int] = 10
|
|
GAME_DP_DAN_RANK_5_DAN: Final[int] = 11
|
|
GAME_DP_DAN_RANK_6_DAN: Final[int] = 12
|
|
GAME_DP_DAN_RANK_7_DAN: Final[int] = 13
|
|
GAME_DP_DAN_RANK_8_DAN: Final[int] = 14
|
|
GAME_DP_DAN_RANK_9_DAN: Final[int] = 15
|
|
GAME_DP_DAN_RANK_10_DAN: Final[int] = 16
|
|
GAME_DP_DAN_RANK_CHUDEN: Final[int] = 17
|
|
GAME_DP_DAN_RANK_KAIDEN: Final[int] = 18
|
|
|
|
FAVORITE_LIST_LENGTH: Final[int] = 20
|
|
|
|
GAME_CHART_TYPE_N7: Final[int] = 0
|
|
GAME_CHART_TYPE_H7: Final[int] = 1
|
|
GAME_CHART_TYPE_A7: Final[int] = 2
|
|
GAME_CHART_TYPE_N14: Final[int] = 3
|
|
GAME_CHART_TYPE_H14: Final[int] = 4
|
|
GAME_CHART_TYPE_A14: Final[int] = 5
|
|
GAME_CHART_TYPE_B7: Final[int] = 6
|
|
|
|
def previous_version(self) -> Optional[IIDXBase]:
|
|
return IIDXPendual(self.data, self.config, self.model)
|
|
|
|
@classmethod
|
|
def run_scheduled_work(
|
|
cls, data: Data, config: Dict[str, Any]
|
|
) -> List[Tuple[str, Dict[str, Any]]]:
|
|
"""
|
|
Insert dailies into the DB.
|
|
"""
|
|
events = []
|
|
if data.local.network.should_schedule(
|
|
cls.game, cls.version, "daily_charts", "daily"
|
|
):
|
|
# Generate a new list of three dailies.
|
|
start_time, end_time = data.local.network.get_schedule_duration("daily")
|
|
all_songs = list(
|
|
set(
|
|
[
|
|
song.id
|
|
for song in data.local.music.get_all_songs(
|
|
cls.game, cls.version
|
|
)
|
|
]
|
|
)
|
|
)
|
|
if len(all_songs) >= 3:
|
|
daily_songs = random.sample(all_songs, 3)
|
|
data.local.game.put_time_sensitive_settings(
|
|
cls.game,
|
|
cls.version,
|
|
"dailies",
|
|
{
|
|
"start_time": start_time,
|
|
"end_time": end_time,
|
|
"music": daily_songs,
|
|
},
|
|
)
|
|
events.append(
|
|
(
|
|
"iidx_daily_charts",
|
|
{
|
|
"version": cls.version,
|
|
"music": daily_songs,
|
|
},
|
|
)
|
|
)
|
|
|
|
# Mark that we did some actual work here.
|
|
data.local.network.mark_scheduled(
|
|
cls.game, cls.version, "daily_charts", "daily"
|
|
)
|
|
return events
|
|
|
|
@classmethod
|
|
def get_settings(cls) -> Dict[str, Any]:
|
|
"""
|
|
Return all of our front-end modifiably settings.
|
|
"""
|
|
return {
|
|
"bools": [
|
|
{
|
|
"name": "Global Shop Ranking",
|
|
"tip": "Return network-wide ranking instead of shop ranking on results screen.",
|
|
"category": "game_config",
|
|
"setting": "global_shop_ranking",
|
|
},
|
|
{
|
|
"name": "Events In Omnimix",
|
|
"tip": "Allow events to be enabled at all for Omnimix.",
|
|
"category": "game_config",
|
|
"setting": "omnimix_events_enabled",
|
|
},
|
|
],
|
|
"ints": [
|
|
{
|
|
"name": "Event Phase",
|
|
"tip": "Event phase for all players.",
|
|
"category": "game_config",
|
|
"setting": "event_phase",
|
|
"values": {
|
|
0: "No Event",
|
|
1: "Tokotoko Line",
|
|
2: "Mystery Line Phase 1",
|
|
3: "Mystery Line Phase 2",
|
|
4: "Mystery Line Phase 3",
|
|
},
|
|
},
|
|
],
|
|
}
|
|
|
|
def db_to_game_status(self, db_status: int) -> int:
|
|
return {
|
|
self.CLEAR_STATUS_NO_PLAY: self.GAME_CLEAR_STATUS_NO_PLAY,
|
|
self.CLEAR_STATUS_FAILED: self.GAME_CLEAR_STATUS_FAILED,
|
|
self.CLEAR_STATUS_ASSIST_CLEAR: self.GAME_CLEAR_STATUS_ASSIST_CLEAR,
|
|
self.CLEAR_STATUS_EASY_CLEAR: self.GAME_CLEAR_STATUS_EASY_CLEAR,
|
|
self.CLEAR_STATUS_CLEAR: self.GAME_CLEAR_STATUS_CLEAR,
|
|
self.CLEAR_STATUS_HARD_CLEAR: self.GAME_CLEAR_STATUS_HARD_CLEAR,
|
|
self.CLEAR_STATUS_EX_HARD_CLEAR: self.GAME_CLEAR_STATUS_EX_HARD_CLEAR,
|
|
self.CLEAR_STATUS_FULL_COMBO: self.GAME_CLEAR_STATUS_FULL_COMBO,
|
|
}[db_status]
|
|
|
|
def game_to_db_status(self, game_status: int) -> int:
|
|
return {
|
|
self.GAME_CLEAR_STATUS_NO_PLAY: self.CLEAR_STATUS_NO_PLAY,
|
|
self.GAME_CLEAR_STATUS_FAILED: self.CLEAR_STATUS_FAILED,
|
|
self.GAME_CLEAR_STATUS_ASSIST_CLEAR: self.CLEAR_STATUS_ASSIST_CLEAR,
|
|
self.GAME_CLEAR_STATUS_EASY_CLEAR: self.CLEAR_STATUS_EASY_CLEAR,
|
|
self.GAME_CLEAR_STATUS_CLEAR: self.CLEAR_STATUS_CLEAR,
|
|
self.GAME_CLEAR_STATUS_HARD_CLEAR: self.CLEAR_STATUS_HARD_CLEAR,
|
|
self.GAME_CLEAR_STATUS_EX_HARD_CLEAR: self.CLEAR_STATUS_EX_HARD_CLEAR,
|
|
self.GAME_CLEAR_STATUS_FULL_COMBO: self.CLEAR_STATUS_FULL_COMBO,
|
|
}[game_status]
|
|
|
|
def db_to_game_rank(self, db_dan: int, cltype: int) -> int:
|
|
# Special case for no DAN rank
|
|
if db_dan == -1:
|
|
return -1
|
|
|
|
if cltype == self.GAME_CLTYPE_SINGLE:
|
|
return {
|
|
self.DAN_RANK_7_KYU: self.GAME_SP_DAN_RANK_7_KYU,
|
|
self.DAN_RANK_6_KYU: self.GAME_SP_DAN_RANK_6_KYU,
|
|
self.DAN_RANK_5_KYU: self.GAME_SP_DAN_RANK_5_KYU,
|
|
self.DAN_RANK_4_KYU: self.GAME_SP_DAN_RANK_4_KYU,
|
|
self.DAN_RANK_3_KYU: self.GAME_SP_DAN_RANK_3_KYU,
|
|
self.DAN_RANK_2_KYU: self.GAME_SP_DAN_RANK_2_KYU,
|
|
self.DAN_RANK_1_KYU: self.GAME_SP_DAN_RANK_1_KYU,
|
|
self.DAN_RANK_1_DAN: self.GAME_SP_DAN_RANK_1_DAN,
|
|
self.DAN_RANK_2_DAN: self.GAME_SP_DAN_RANK_2_DAN,
|
|
self.DAN_RANK_3_DAN: self.GAME_SP_DAN_RANK_3_DAN,
|
|
self.DAN_RANK_4_DAN: self.GAME_SP_DAN_RANK_4_DAN,
|
|
self.DAN_RANK_5_DAN: self.GAME_SP_DAN_RANK_5_DAN,
|
|
self.DAN_RANK_6_DAN: self.GAME_SP_DAN_RANK_6_DAN,
|
|
self.DAN_RANK_7_DAN: self.GAME_SP_DAN_RANK_7_DAN,
|
|
self.DAN_RANK_8_DAN: self.GAME_SP_DAN_RANK_8_DAN,
|
|
self.DAN_RANK_9_DAN: self.GAME_SP_DAN_RANK_9_DAN,
|
|
self.DAN_RANK_10_DAN: self.GAME_SP_DAN_RANK_10_DAN,
|
|
self.DAN_RANK_CHUDEN: self.GAME_SP_DAN_RANK_CHUDEN,
|
|
self.DAN_RANK_KAIDEN: self.GAME_SP_DAN_RANK_KAIDEN,
|
|
}[db_dan]
|
|
elif cltype == self.GAME_CLTYPE_DOUBLE:
|
|
return {
|
|
self.DAN_RANK_7_KYU: self.GAME_DP_DAN_RANK_7_KYU,
|
|
self.DAN_RANK_6_KYU: self.GAME_DP_DAN_RANK_6_KYU,
|
|
self.DAN_RANK_5_KYU: self.GAME_DP_DAN_RANK_5_KYU,
|
|
self.DAN_RANK_4_KYU: self.GAME_DP_DAN_RANK_4_KYU,
|
|
self.DAN_RANK_3_KYU: self.GAME_DP_DAN_RANK_3_KYU,
|
|
self.DAN_RANK_2_KYU: self.GAME_DP_DAN_RANK_2_KYU,
|
|
self.DAN_RANK_1_KYU: self.GAME_DP_DAN_RANK_1_KYU,
|
|
self.DAN_RANK_1_DAN: self.GAME_DP_DAN_RANK_1_DAN,
|
|
self.DAN_RANK_2_DAN: self.GAME_DP_DAN_RANK_2_DAN,
|
|
self.DAN_RANK_3_DAN: self.GAME_DP_DAN_RANK_3_DAN,
|
|
self.DAN_RANK_4_DAN: self.GAME_DP_DAN_RANK_4_DAN,
|
|
self.DAN_RANK_5_DAN: self.GAME_DP_DAN_RANK_5_DAN,
|
|
self.DAN_RANK_6_DAN: self.GAME_DP_DAN_RANK_6_DAN,
|
|
self.DAN_RANK_7_DAN: self.GAME_DP_DAN_RANK_7_DAN,
|
|
self.DAN_RANK_8_DAN: self.GAME_DP_DAN_RANK_8_DAN,
|
|
self.DAN_RANK_9_DAN: self.GAME_DP_DAN_RANK_9_DAN,
|
|
self.DAN_RANK_10_DAN: self.GAME_DP_DAN_RANK_10_DAN,
|
|
self.DAN_RANK_CHUDEN: self.GAME_DP_DAN_RANK_CHUDEN,
|
|
self.DAN_RANK_KAIDEN: self.GAME_DP_DAN_RANK_KAIDEN,
|
|
}[db_dan]
|
|
else:
|
|
raise Exception("Invalid cltype!")
|
|
|
|
def game_to_db_rank(self, game_dan: int, cltype: int) -> int:
|
|
# Special case for no DAN rank
|
|
if game_dan == -1:
|
|
return -1
|
|
|
|
if cltype == self.GAME_CLTYPE_SINGLE:
|
|
return {
|
|
self.GAME_SP_DAN_RANK_7_KYU: self.DAN_RANK_7_KYU,
|
|
self.GAME_SP_DAN_RANK_6_KYU: self.DAN_RANK_6_KYU,
|
|
self.GAME_SP_DAN_RANK_5_KYU: self.DAN_RANK_5_KYU,
|
|
self.GAME_SP_DAN_RANK_4_KYU: self.DAN_RANK_4_KYU,
|
|
self.GAME_SP_DAN_RANK_3_KYU: self.DAN_RANK_3_KYU,
|
|
self.GAME_SP_DAN_RANK_2_KYU: self.DAN_RANK_2_KYU,
|
|
self.GAME_SP_DAN_RANK_1_KYU: self.DAN_RANK_1_KYU,
|
|
self.GAME_SP_DAN_RANK_1_DAN: self.DAN_RANK_1_DAN,
|
|
self.GAME_SP_DAN_RANK_2_DAN: self.DAN_RANK_2_DAN,
|
|
self.GAME_SP_DAN_RANK_3_DAN: self.DAN_RANK_3_DAN,
|
|
self.GAME_SP_DAN_RANK_4_DAN: self.DAN_RANK_4_DAN,
|
|
self.GAME_SP_DAN_RANK_5_DAN: self.DAN_RANK_5_DAN,
|
|
self.GAME_SP_DAN_RANK_6_DAN: self.DAN_RANK_6_DAN,
|
|
self.GAME_SP_DAN_RANK_7_DAN: self.DAN_RANK_7_DAN,
|
|
self.GAME_SP_DAN_RANK_8_DAN: self.DAN_RANK_8_DAN,
|
|
self.GAME_SP_DAN_RANK_9_DAN: self.DAN_RANK_9_DAN,
|
|
self.GAME_SP_DAN_RANK_10_DAN: self.DAN_RANK_10_DAN,
|
|
self.GAME_SP_DAN_RANK_CHUDEN: self.DAN_RANK_CHUDEN,
|
|
self.GAME_SP_DAN_RANK_KAIDEN: self.DAN_RANK_KAIDEN,
|
|
}[game_dan]
|
|
elif cltype == self.GAME_CLTYPE_DOUBLE:
|
|
return {
|
|
self.GAME_DP_DAN_RANK_7_KYU: self.DAN_RANK_7_KYU,
|
|
self.GAME_DP_DAN_RANK_6_KYU: self.DAN_RANK_6_KYU,
|
|
self.GAME_DP_DAN_RANK_5_KYU: self.DAN_RANK_5_KYU,
|
|
self.GAME_DP_DAN_RANK_4_KYU: self.DAN_RANK_4_KYU,
|
|
self.GAME_DP_DAN_RANK_3_KYU: self.DAN_RANK_3_KYU,
|
|
self.GAME_DP_DAN_RANK_2_KYU: self.DAN_RANK_2_KYU,
|
|
self.GAME_DP_DAN_RANK_1_KYU: self.DAN_RANK_1_KYU,
|
|
self.GAME_DP_DAN_RANK_1_DAN: self.DAN_RANK_1_DAN,
|
|
self.GAME_DP_DAN_RANK_2_DAN: self.DAN_RANK_2_DAN,
|
|
self.GAME_DP_DAN_RANK_3_DAN: self.DAN_RANK_3_DAN,
|
|
self.GAME_DP_DAN_RANK_4_DAN: self.DAN_RANK_4_DAN,
|
|
self.GAME_DP_DAN_RANK_5_DAN: self.DAN_RANK_5_DAN,
|
|
self.GAME_DP_DAN_RANK_6_DAN: self.DAN_RANK_6_DAN,
|
|
self.GAME_DP_DAN_RANK_7_DAN: self.DAN_RANK_7_DAN,
|
|
self.GAME_DP_DAN_RANK_8_DAN: self.DAN_RANK_8_DAN,
|
|
self.GAME_DP_DAN_RANK_9_DAN: self.DAN_RANK_9_DAN,
|
|
self.GAME_DP_DAN_RANK_10_DAN: self.DAN_RANK_10_DAN,
|
|
self.GAME_DP_DAN_RANK_CHUDEN: self.DAN_RANK_CHUDEN,
|
|
self.GAME_DP_DAN_RANK_KAIDEN: self.DAN_RANK_KAIDEN,
|
|
}[game_dan]
|
|
else:
|
|
raise Exception("Invalid cltype!")
|
|
|
|
def game_to_db_chart(self, db_chart: int) -> int:
|
|
return {
|
|
self.GAME_CHART_TYPE_B7: self.CHART_TYPE_B7,
|
|
self.GAME_CHART_TYPE_N7: self.CHART_TYPE_N7,
|
|
self.GAME_CHART_TYPE_H7: self.CHART_TYPE_H7,
|
|
self.GAME_CHART_TYPE_A7: self.CHART_TYPE_A7,
|
|
self.GAME_CHART_TYPE_N14: self.CHART_TYPE_N14,
|
|
self.GAME_CHART_TYPE_H14: self.CHART_TYPE_H14,
|
|
self.GAME_CHART_TYPE_A14: self.CHART_TYPE_A14,
|
|
}[db_chart]
|
|
|
|
def handle_IIDX23shop_getname_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23shop")
|
|
root.set_attribute("cls_opt", "0")
|
|
machine = self.data.local.machine.get_machine(self.config.machine.pcbid)
|
|
root.set_attribute("opname", machine.name)
|
|
root.set_attribute("pid", str(self.get_machine_region()))
|
|
return root
|
|
|
|
def handle_IIDX23shop_savename_request(self, request: Node) -> Node:
|
|
self.update_machine_name(request.attribute("opname"))
|
|
root = Node.void("IIDX23shop")
|
|
return root
|
|
|
|
def handle_IIDX23shop_sentinfo_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23shop")
|
|
return root
|
|
|
|
def handle_IIDX23shop_getconvention_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23shop")
|
|
machine = self.data.local.machine.get_machine(self.config.machine.pcbid)
|
|
if machine.arcade is not None:
|
|
course = self.data.local.machine.get_settings(
|
|
machine.arcade, self.game, self.music_version, "shop_course"
|
|
)
|
|
else:
|
|
course = None
|
|
|
|
if course is None:
|
|
course = ValidatedDict()
|
|
|
|
root.set_attribute("music_0", str(course.get_int("music_0", 20032)))
|
|
root.set_attribute("music_1", str(course.get_int("music_1", 20009)))
|
|
root.set_attribute("music_2", str(course.get_int("music_2", 20015)))
|
|
root.set_attribute("music_3", str(course.get_int("music_3", 20064)))
|
|
root.add_child(Node.bool("valid", course.get_bool("valid")))
|
|
return root
|
|
|
|
def handle_IIDX23shop_setconvention_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23shop")
|
|
machine = self.data.local.machine.get_machine(self.config.machine.pcbid)
|
|
if machine.arcade is not None:
|
|
course = ValidatedDict()
|
|
course.replace_int("music_0", request.child_value("music_0"))
|
|
course.replace_int("music_1", request.child_value("music_1"))
|
|
course.replace_int("music_2", request.child_value("music_2"))
|
|
course.replace_int("music_3", request.child_value("music_3"))
|
|
course.replace_bool("valid", request.child_value("valid"))
|
|
self.data.local.machine.put_settings(
|
|
machine.arcade, self.game, self.music_version, "shop_course", course
|
|
)
|
|
|
|
return root
|
|
|
|
def handle_IIDX23shop_sendescapepackageinfo_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23shop")
|
|
root.set_attribute("expire", str((Time.now() + 86400 * 365) * 1000))
|
|
return root
|
|
|
|
def handle_IIDX23ranking_getranker_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23ranking")
|
|
chart = self.game_to_db_chart(int(request.attribute("clid")))
|
|
if chart not in [
|
|
self.CHART_TYPE_N7,
|
|
self.CHART_TYPE_H7,
|
|
self.CHART_TYPE_A7,
|
|
self.CHART_TYPE_N14,
|
|
self.CHART_TYPE_H14,
|
|
self.CHART_TYPE_A14,
|
|
]:
|
|
# Chart type 6 is presumably beginner mode, but it crashes the game
|
|
return root
|
|
|
|
machine = self.data.local.machine.get_machine(self.config.machine.pcbid)
|
|
if machine.arcade is not None:
|
|
course = self.data.local.machine.get_settings(
|
|
machine.arcade, self.game, self.music_version, "shop_course"
|
|
)
|
|
else:
|
|
course = None
|
|
|
|
if course is None:
|
|
course = ValidatedDict()
|
|
|
|
if not course.get_bool("valid"):
|
|
# Shop course not enabled or not present
|
|
return root
|
|
|
|
convention = Node.void("convention")
|
|
root.add_child(convention)
|
|
convention.set_attribute("clid", str(chart))
|
|
convention.set_attribute("update_date", str(Time.now() * 1000))
|
|
|
|
# Grab all scores for each of the four songs, filter out people who haven't
|
|
# set us as their arcade and then return the top 20 scores (adding all 4 songs).
|
|
songids = [
|
|
course.get_int("music_0"),
|
|
course.get_int("music_1"),
|
|
course.get_int("music_2"),
|
|
course.get_int("music_3"),
|
|
]
|
|
|
|
totalscores: Dict[UserID, int] = {}
|
|
profiles: Dict[UserID, Profile] = {}
|
|
for songid in songids:
|
|
scores = self.data.local.music.get_all_scores(
|
|
self.game,
|
|
self.music_version,
|
|
songid=songid,
|
|
songchart=chart,
|
|
)
|
|
|
|
for score in scores:
|
|
if score[0] not in totalscores:
|
|
totalscores[score[0]] = 0
|
|
profile = self.get_any_profile(score[0])
|
|
if profile is None:
|
|
profile = Profile(self.game, self.version, "", 0)
|
|
profiles[score[0]] = profile
|
|
|
|
totalscores[score[0]] += score[1].points
|
|
|
|
topscores = sorted(
|
|
[
|
|
(totalscores[userid], profiles[userid])
|
|
for userid in totalscores
|
|
if self.user_joined_arcade(machine, profiles[userid])
|
|
],
|
|
key=lambda tup: tup[0],
|
|
reverse=True,
|
|
)[:20]
|
|
|
|
rank = 0
|
|
for topscore in topscores:
|
|
rank = rank + 1
|
|
|
|
detail = Node.void("detail")
|
|
convention.add_child(detail)
|
|
detail.set_attribute("name", topscore[1].get_str("name"))
|
|
detail.set_attribute("rank", str(rank))
|
|
detail.set_attribute("score", str(topscore[0]))
|
|
detail.set_attribute("pid", str(topscore[1].get_int("pid")))
|
|
|
|
qpro = topscore[1].get_dict("qpro")
|
|
detail.set_attribute("head", str(qpro.get_int("head")))
|
|
detail.set_attribute("hair", str(qpro.get_int("hair")))
|
|
detail.set_attribute("face", str(qpro.get_int("face")))
|
|
detail.set_attribute("body", str(qpro.get_int("body")))
|
|
detail.set_attribute("hand", str(qpro.get_int("hand")))
|
|
|
|
return root
|
|
|
|
def handle_IIDX23ranking_entry_request(self, request: Node) -> Node:
|
|
extid = int(request.attribute("iidxid"))
|
|
courseid = int(request.attribute("coid"))
|
|
chart = self.game_to_db_chart(int(request.attribute("clid")))
|
|
course_type = int(request.attribute("regist_type"))
|
|
clear_status = self.game_to_db_status(int(request.attribute("clr")))
|
|
pgreats = int(request.attribute("pgnum"))
|
|
greats = int(request.attribute("gnum"))
|
|
|
|
if course_type == 0:
|
|
index = self.COURSE_TYPE_INTERNET_RANKING
|
|
elif course_type == 1:
|
|
index = self.COURSE_TYPE_SECRET
|
|
else:
|
|
raise Exception("Unknown registration type for course entry!")
|
|
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
if userid is not None:
|
|
# Update achievement to track course statistics
|
|
self.update_course(
|
|
userid,
|
|
index,
|
|
courseid,
|
|
chart,
|
|
clear_status,
|
|
pgreats,
|
|
greats,
|
|
)
|
|
|
|
# We should return the user's position, but its not displayed anywhere
|
|
# so fuck it.
|
|
root = Node.void("IIDX23ranking")
|
|
root.set_attribute("anum", "1")
|
|
root.set_attribute("jun", "1")
|
|
return root
|
|
|
|
def handle_IIDX23music_crate_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23music")
|
|
attempts = self.get_clear_rates()
|
|
|
|
all_songs = list(
|
|
set(
|
|
[
|
|
song.id
|
|
for song in self.data.local.music.get_all_songs(
|
|
self.game, self.music_version
|
|
)
|
|
]
|
|
)
|
|
)
|
|
for song in all_songs:
|
|
clears = []
|
|
fcs = []
|
|
|
|
for chart in [0, 1, 2, 3, 4, 5]:
|
|
placed = False
|
|
if song in attempts and chart in attempts[song]:
|
|
values = attempts[song][chart]
|
|
if values["total"] > 0:
|
|
clears.append(int((100 * values["clears"]) / values["total"]))
|
|
fcs.append(int((100 * values["fcs"]) / values["total"]))
|
|
placed = True
|
|
if not placed:
|
|
clears.append(101)
|
|
fcs.append(101)
|
|
|
|
clearnode = Node.u8_array("c", clears + fcs)
|
|
clearnode.set_attribute("mid", str(song))
|
|
root.add_child(clearnode)
|
|
|
|
return root
|
|
|
|
def handle_IIDX23music_getrank_request(self, request: Node) -> Node:
|
|
cltype = int(request.attribute("cltype"))
|
|
|
|
root = Node.void("IIDX23music")
|
|
style = Node.void("style")
|
|
root.add_child(style)
|
|
style.set_attribute("type", str(cltype))
|
|
|
|
for rivalid in [-1, 0, 1, 2, 3, 4]:
|
|
if rivalid == -1:
|
|
attr = "iidxid"
|
|
else:
|
|
attr = f"iidxid{rivalid}"
|
|
|
|
try:
|
|
extid = int(request.attribute(attr))
|
|
except Exception:
|
|
# Invalid extid
|
|
continue
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
if userid is not None:
|
|
scores = self.data.remote.music.get_scores(
|
|
self.game, self.music_version, userid
|
|
)
|
|
|
|
# Grab score data for user/rival
|
|
scoredata = self.make_score_struct(
|
|
scores,
|
|
self.CLEAR_TYPE_SINGLE
|
|
if cltype == self.GAME_CLTYPE_SINGLE
|
|
else self.CLEAR_TYPE_DOUBLE,
|
|
rivalid,
|
|
)
|
|
for s in scoredata:
|
|
root.add_child(Node.s16_array("m", s))
|
|
|
|
# Grab most played for user/rival
|
|
most_played = [
|
|
play[0]
|
|
for play in self.data.local.music.get_most_played(
|
|
self.game, self.music_version, userid, 20
|
|
)
|
|
]
|
|
if len(most_played) < 20:
|
|
most_played.extend([0] * (20 - len(most_played)))
|
|
best = Node.u16_array("best", most_played)
|
|
best.set_attribute("rno", str(rivalid))
|
|
root.add_child(best)
|
|
|
|
if rivalid == -1:
|
|
# Grab beginner statuses for user only
|
|
beginnerdata = self.make_beginner_struct(scores)
|
|
for b in beginnerdata:
|
|
root.add_child(Node.u16_array("b", b))
|
|
|
|
return root
|
|
|
|
def handle_IIDX23music_reg_request(self, request: Node) -> Node:
|
|
extid = int(request.attribute("iidxid"))
|
|
musicid = int(request.attribute("mid"))
|
|
chart = self.game_to_db_chart(int(request.attribute("clid")))
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
|
|
# See if we need to report global or shop scores
|
|
if self.machine_joined_arcade():
|
|
game_config = self.get_game_config()
|
|
global_scores = game_config.get_bool("global_shop_ranking")
|
|
machine = self.data.local.machine.get_machine(self.config.machine.pcbid)
|
|
else:
|
|
# If we aren't in an arcade, we can only show global scores
|
|
global_scores = True
|
|
machine = None
|
|
|
|
# First, determine our current ranking before saving the new score
|
|
all_scores = sorted(
|
|
self.data.remote.music.get_all_scores(
|
|
game=self.game,
|
|
version=self.music_version,
|
|
songid=musicid,
|
|
songchart=chart,
|
|
),
|
|
key=lambda s: (s[1].points, s[1].timestamp),
|
|
reverse=True,
|
|
)
|
|
all_players = {
|
|
uid: prof
|
|
for (uid, prof) in self.get_any_profiles([s[0] for s in all_scores])
|
|
}
|
|
|
|
if not global_scores:
|
|
all_scores = [
|
|
score
|
|
for score in all_scores
|
|
if (
|
|
score[0] == userid
|
|
or self.user_joined_arcade(machine, all_players[score[0]])
|
|
)
|
|
]
|
|
|
|
# Find our actual index
|
|
oldindex = None
|
|
for i in range(len(all_scores)):
|
|
if all_scores[i][0] == userid:
|
|
oldindex = i
|
|
break
|
|
|
|
if userid is not None:
|
|
clear_status = self.game_to_db_status(int(request.attribute("cflg")))
|
|
pgreats = int(request.attribute("pgnum"))
|
|
greats = int(request.attribute("gnum"))
|
|
miss_count = int(request.attribute("mnum"))
|
|
ghost = request.child_value("ghost")
|
|
shopid = ID.parse_machine_id(request.attribute("shopconvid"))
|
|
|
|
self.update_score(
|
|
userid,
|
|
musicid,
|
|
chart,
|
|
clear_status,
|
|
pgreats,
|
|
greats,
|
|
miss_count,
|
|
ghost,
|
|
shopid,
|
|
)
|
|
|
|
# Calculate and return statistics about this song
|
|
root = Node.void("IIDX23music")
|
|
root.set_attribute("clid", request.attribute("clid"))
|
|
root.set_attribute("mid", request.attribute("mid"))
|
|
|
|
attempts = self.get_clear_rates(musicid, chart)
|
|
count = attempts[musicid][chart]["total"]
|
|
clear = attempts[musicid][chart]["clears"]
|
|
full_combo = attempts[musicid][chart]["fcs"]
|
|
|
|
if count > 0:
|
|
root.set_attribute("crate", str(int((100 * clear) / count)))
|
|
root.set_attribute("frate", str(int((100 * full_combo) / count)))
|
|
else:
|
|
root.set_attribute("crate", "0")
|
|
root.set_attribute("frate", "0")
|
|
root.set_attribute("rankside", "0")
|
|
|
|
if userid is not None:
|
|
# Shop ranking
|
|
shopdata = Node.void("shopdata")
|
|
root.add_child(shopdata)
|
|
shopdata.set_attribute(
|
|
"rank", "-1" if oldindex is None else str(oldindex + 1)
|
|
)
|
|
|
|
# Grab the rank of some other players on this song
|
|
ranklist = Node.void("ranklist")
|
|
root.add_child(ranklist)
|
|
|
|
all_scores = sorted(
|
|
self.data.remote.music.get_all_scores(
|
|
game=self.game,
|
|
version=self.music_version,
|
|
songid=musicid,
|
|
songchart=chart,
|
|
),
|
|
key=lambda s: (s[1].points, s[1].timestamp),
|
|
reverse=True,
|
|
)
|
|
missing_players = [uid for (uid, _) in all_scores if uid not in all_players]
|
|
for (uid, prof) in self.get_any_profiles(missing_players):
|
|
all_players[uid] = prof
|
|
|
|
if not global_scores:
|
|
all_scores = [
|
|
score
|
|
for score in all_scores
|
|
if (
|
|
score[0] == userid
|
|
or self.user_joined_arcade(machine, all_players[score[0]])
|
|
)
|
|
]
|
|
|
|
# Find our actual index
|
|
ourindex = None
|
|
for i in range(len(all_scores)):
|
|
if all_scores[i][0] == userid:
|
|
ourindex = i
|
|
break
|
|
if ourindex is None:
|
|
raise Exception("Cannot find our own score after saving to DB!")
|
|
start = ourindex - 4
|
|
end = ourindex + 4
|
|
if start < 0:
|
|
start = 0
|
|
if end >= len(all_scores):
|
|
end = len(all_scores) - 1
|
|
relevant_scores = all_scores[start : (end + 1)]
|
|
|
|
record_num = start + 1
|
|
for score in relevant_scores:
|
|
profile = all_players[score[0]]
|
|
|
|
data = Node.void("data")
|
|
ranklist.add_child(data)
|
|
data.set_attribute("iidx_id", str(profile.extid))
|
|
data.set_attribute("name", profile.get_str("name"))
|
|
|
|
machine_name = ""
|
|
if "shop_location" in profile:
|
|
shop_id = profile.get_int("shop_location")
|
|
machine = self.get_machine_by_id(shop_id)
|
|
if machine is not None:
|
|
machine_name = machine.name
|
|
data.set_attribute("opname", machine_name)
|
|
data.set_attribute("rnum", str(record_num))
|
|
data.set_attribute("score", str(score[1].points))
|
|
data.set_attribute(
|
|
"clflg",
|
|
str(self.db_to_game_status(score[1].data.get_int("clear_status"))),
|
|
)
|
|
data.set_attribute("pid", str(profile.get_int("pid")))
|
|
data.set_attribute("myFlg", "1" if score[0] == userid else "0")
|
|
data.set_attribute("update", "0")
|
|
|
|
data.set_attribute(
|
|
"sgrade",
|
|
str(
|
|
self.db_to_game_rank(
|
|
profile.get_int(self.DAN_RANKING_SINGLE, -1),
|
|
self.GAME_CLTYPE_SINGLE,
|
|
),
|
|
),
|
|
)
|
|
data.set_attribute(
|
|
"dgrade",
|
|
str(
|
|
self.db_to_game_rank(
|
|
profile.get_int(self.DAN_RANKING_DOUBLE, -1),
|
|
self.GAME_CLTYPE_DOUBLE,
|
|
),
|
|
),
|
|
)
|
|
|
|
qpro = profile.get_dict("qpro")
|
|
data.set_attribute("head", str(qpro.get_int("head")))
|
|
data.set_attribute("hair", str(qpro.get_int("hair")))
|
|
data.set_attribute("face", str(qpro.get_int("face")))
|
|
data.set_attribute("body", str(qpro.get_int("body")))
|
|
data.set_attribute("hand", str(qpro.get_int("hand")))
|
|
|
|
record_num = record_num + 1
|
|
|
|
return root
|
|
|
|
def handle_IIDX23music_breg_request(self, request: Node) -> Node:
|
|
extid = int(request.attribute("iidxid"))
|
|
musicid = int(request.attribute("mid"))
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
|
|
if userid is not None:
|
|
clear_status = self.game_to_db_status(int(request.attribute("cflg")))
|
|
pgreats = int(request.attribute("pgnum"))
|
|
greats = int(request.attribute("gnum"))
|
|
|
|
self.update_score(
|
|
userid,
|
|
musicid,
|
|
self.CHART_TYPE_B7,
|
|
clear_status,
|
|
pgreats,
|
|
greats,
|
|
-1,
|
|
b"",
|
|
None,
|
|
)
|
|
|
|
# Return nothing.
|
|
root = Node.void("IIDX23music")
|
|
return root
|
|
|
|
def handle_IIDX23music_play_request(self, request: Node) -> Node:
|
|
musicid = int(request.attribute("mid"))
|
|
chart = self.game_to_db_chart(int(request.attribute("clid")))
|
|
clear_status = self.game_to_db_status(int(request.attribute("cflg")))
|
|
|
|
self.update_score(
|
|
None, # No userid since its anonymous
|
|
musicid,
|
|
chart,
|
|
clear_status,
|
|
0, # No ex score
|
|
0, # No ex score
|
|
0, # No miss count
|
|
None, # No ghost
|
|
None, # No shop for this user
|
|
)
|
|
|
|
# Calculate and return statistics about this song
|
|
root = Node.void("IIDX23music")
|
|
root.set_attribute("clid", request.attribute("clid"))
|
|
root.set_attribute("mid", request.attribute("mid"))
|
|
|
|
attempts = self.get_clear_rates(musicid, chart)
|
|
count = attempts[musicid][chart]["total"]
|
|
clear = attempts[musicid][chart]["clears"]
|
|
full_combo = attempts[musicid][chart]["fcs"]
|
|
|
|
if count > 0:
|
|
root.set_attribute("crate", str(int((100 * clear) / count)))
|
|
root.set_attribute("frate", str(int((100 * full_combo) / count)))
|
|
else:
|
|
root.set_attribute("crate", "0")
|
|
root.set_attribute("frate", "0")
|
|
|
|
return root
|
|
|
|
def handle_IIDX23music_appoint_request(self, request: Node) -> Node:
|
|
musicid = int(request.attribute("mid"))
|
|
chart = self.game_to_db_chart(int(request.attribute("clid")))
|
|
ghost_type = int(request.attribute("ctype"))
|
|
extid = int(request.attribute("iidxid"))
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
|
|
root = Node.void("IIDX23music")
|
|
|
|
if userid is not None:
|
|
# Try to look up previous ghost for user
|
|
my_score = self.data.remote.music.get_score(
|
|
self.game, self.music_version, userid, musicid, chart
|
|
)
|
|
if my_score is not None:
|
|
mydata = Node.binary("mydata", my_score.data.get_bytes("ghost"))
|
|
mydata.set_attribute("score", str(my_score.points))
|
|
root.add_child(mydata)
|
|
|
|
ghost_score = self.get_ghost(
|
|
{
|
|
self.GAME_GHOST_TYPE_RIVAL: self.GHOST_TYPE_RIVAL,
|
|
self.GAME_GHOST_TYPE_GLOBAL_TOP: self.GHOST_TYPE_GLOBAL_TOP,
|
|
self.GAME_GHOST_TYPE_GLOBAL_AVERAGE: self.GHOST_TYPE_GLOBAL_AVERAGE,
|
|
self.GAME_GHOST_TYPE_LOCAL_TOP: self.GHOST_TYPE_LOCAL_TOP,
|
|
self.GAME_GHOST_TYPE_LOCAL_AVERAGE: self.GHOST_TYPE_LOCAL_AVERAGE,
|
|
self.GAME_GHOST_TYPE_DAN_TOP: self.GHOST_TYPE_DAN_TOP,
|
|
self.GAME_GHOST_TYPE_DAN_AVERAGE: self.GHOST_TYPE_DAN_AVERAGE,
|
|
self.GAME_GHOST_TYPE_RIVAL_TOP: self.GHOST_TYPE_RIVAL_TOP,
|
|
self.GAME_GHOST_TYPE_RIVAL_AVERAGE: self.GHOST_TYPE_RIVAL_AVERAGE,
|
|
}.get(ghost_type, self.GHOST_TYPE_NONE),
|
|
request.attribute("subtype"),
|
|
self.GAME_GHOST_LENGTH,
|
|
musicid,
|
|
chart,
|
|
userid,
|
|
)
|
|
|
|
# Add ghost score if we support it
|
|
if ghost_score is not None:
|
|
sdata = Node.binary("sdata", ghost_score["ghost"])
|
|
sdata.set_attribute("score", str(ghost_score["score"]))
|
|
if "name" in ghost_score:
|
|
sdata.set_attribute("name", ghost_score["name"])
|
|
if "pid" in ghost_score:
|
|
sdata.set_attribute("pid", str(ghost_score["pid"]))
|
|
if "extid" in ghost_score:
|
|
sdata.set_attribute("riidxid", str(ghost_score["extid"]))
|
|
root.add_child(sdata)
|
|
|
|
return root
|
|
|
|
def handle_IIDX23grade_raised_request(self, request: Node) -> Node:
|
|
extid = int(request.attribute("iidxid"))
|
|
cltype = int(request.attribute("gtype"))
|
|
rank = self.game_to_db_rank(int(request.attribute("gid")), cltype)
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
if userid is not None:
|
|
percent = int(request.attribute("achi"))
|
|
stages_cleared = int(request.attribute("cstage"))
|
|
cleared = stages_cleared == self.DAN_STAGES
|
|
|
|
if cltype == self.GAME_CLTYPE_SINGLE:
|
|
index = self.DAN_RANKING_SINGLE
|
|
else:
|
|
index = self.DAN_RANKING_DOUBLE
|
|
|
|
self.update_rank(
|
|
userid,
|
|
index,
|
|
rank,
|
|
percent,
|
|
cleared,
|
|
stages_cleared,
|
|
)
|
|
|
|
# Figure out number of players that played this ranking
|
|
all_achievements = self.data.local.user.get_all_achievements(
|
|
self.game, self.version, achievementid=rank, achievementtype=index
|
|
)
|
|
root = Node.void("IIDX23grade")
|
|
root.set_attribute("pnum", str(len(all_achievements)))
|
|
return root
|
|
|
|
def handle_IIDX23pc_common_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23pc")
|
|
root.set_attribute("expire", "600")
|
|
|
|
ir = Node.void("ir")
|
|
root.add_child(ir)
|
|
ir.set_attribute("beat", "2")
|
|
|
|
# See if we configured event overrides
|
|
if self.machine_joined_arcade():
|
|
game_config = self.get_game_config()
|
|
event_phase = game_config.get_int("event_phase")
|
|
omni_events = game_config.get_bool("omnimix_events_enabled")
|
|
else:
|
|
# If we aren't in an arcade, we turn off events
|
|
event_phase = 0
|
|
omni_events = False
|
|
|
|
if event_phase == 0 or (self.omnimix and (not omni_events)):
|
|
boss_phase = 0
|
|
event1 = 0
|
|
event2 = 0
|
|
elif event_phase == 1:
|
|
boss_phase = 1
|
|
event1 = 1
|
|
event2 = 0
|
|
elif event_phase in [2, 3, 4]:
|
|
boss_phase = 2
|
|
event1 = 0
|
|
event2 = event_phase - 1
|
|
|
|
boss = Node.void("boss")
|
|
root.add_child(boss)
|
|
boss.set_attribute("phase", str(boss_phase))
|
|
|
|
event1_phase = Node.void("event1_phase")
|
|
root.add_child(event1_phase)
|
|
event1_phase.set_attribute("phase", str(event1))
|
|
|
|
event2_phase = Node.void("event2_phase")
|
|
root.add_child(event2_phase)
|
|
event2_phase.set_attribute("phase", str(event2))
|
|
|
|
extra_boss_event = Node.void("extra_boss_event")
|
|
root.add_child(extra_boss_event)
|
|
extra_boss_event.set_attribute("phase", "1")
|
|
|
|
bemani_summer2016 = Node.void("bemani_summer2016")
|
|
root.add_child(bemani_summer2016)
|
|
bemani_summer2016.set_attribute("phase", "1")
|
|
|
|
vip_black_pass = Node.void("vip_pass_black")
|
|
root.add_child(vip_black_pass)
|
|
|
|
event1_rainbow_ticket = Node.void("event1_rainbow_ticket")
|
|
root.add_child(event1_rainbow_ticket)
|
|
|
|
djlevel_result = Node.void("djlevel_result")
|
|
root.add_child(djlevel_result)
|
|
|
|
newsong_another = Node.void("newsong_another")
|
|
root.add_child(newsong_another)
|
|
newsong_another.set_attribute("open", "1")
|
|
|
|
# Course definitions
|
|
courses: List[Dict[str, Any]] = [
|
|
{
|
|
"name": "POP",
|
|
"id": 1,
|
|
"songs": [
|
|
23034,
|
|
23012,
|
|
23011,
|
|
23032,
|
|
],
|
|
},
|
|
{
|
|
"name": "TRANCE",
|
|
"id": 2,
|
|
"songs": [
|
|
23014,
|
|
23033,
|
|
23038,
|
|
23013,
|
|
],
|
|
},
|
|
{
|
|
"name": "DJ",
|
|
"id": 3,
|
|
"songs": [
|
|
23025,
|
|
23026,
|
|
23024,
|
|
23061,
|
|
],
|
|
},
|
|
{
|
|
"name": "HCN",
|
|
"id": 4,
|
|
"songs": [
|
|
23016,
|
|
23010,
|
|
23057,
|
|
23000,
|
|
],
|
|
},
|
|
{
|
|
"name": "TRAIN",
|
|
"id": 5,
|
|
"songs": [
|
|
23023,
|
|
6029,
|
|
23047,
|
|
7008,
|
|
],
|
|
},
|
|
{
|
|
"name": "USAO",
|
|
"id": 6,
|
|
"songs": [
|
|
20064,
|
|
21015,
|
|
22044,
|
|
23027,
|
|
],
|
|
},
|
|
{
|
|
"name": "New Face",
|
|
"id": 7,
|
|
"songs": [
|
|
23054,
|
|
23039,
|
|
23036,
|
|
23045,
|
|
],
|
|
},
|
|
{
|
|
"name": "CANDY",
|
|
"id": 8,
|
|
"songs": [
|
|
16021,
|
|
18069,
|
|
19073,
|
|
23008,
|
|
],
|
|
},
|
|
{
|
|
"name": "ROCK",
|
|
"id": 9,
|
|
"songs": [
|
|
8026,
|
|
23002,
|
|
22072,
|
|
23050,
|
|
],
|
|
},
|
|
{
|
|
"name": "JAZZ",
|
|
"id": 10,
|
|
"songs": [
|
|
1017,
|
|
20088,
|
|
23035,
|
|
23030,
|
|
],
|
|
},
|
|
{
|
|
"name": "TAIYO",
|
|
"id": 11,
|
|
"songs": [
|
|
11007,
|
|
23020,
|
|
13029,
|
|
23051,
|
|
],
|
|
},
|
|
{
|
|
"name": "RHYZE",
|
|
"id": 12,
|
|
"songs": [
|
|
23029,
|
|
21070,
|
|
21071,
|
|
20063,
|
|
],
|
|
},
|
|
{
|
|
"name": "COLLABORATION",
|
|
"id": 13,
|
|
"songs": [
|
|
20094,
|
|
20008,
|
|
20020,
|
|
21082,
|
|
],
|
|
},
|
|
]
|
|
|
|
# Secret course definitions
|
|
secret_courses: List[Dict[str, Any]] = [
|
|
{
|
|
"name": "COLORS",
|
|
"id": 1,
|
|
"songs": [
|
|
20038,
|
|
20012,
|
|
20007,
|
|
22012,
|
|
],
|
|
},
|
|
{
|
|
"name": "BROKEN",
|
|
"id": 2,
|
|
"songs": [
|
|
4003,
|
|
18028,
|
|
18068,
|
|
22001,
|
|
],
|
|
},
|
|
{
|
|
"name": "PENDUAL",
|
|
"id": 3,
|
|
"songs": [
|
|
22013,
|
|
22008,
|
|
22054,
|
|
22100,
|
|
],
|
|
},
|
|
{
|
|
"name": "SYMMETRY",
|
|
"id": 4,
|
|
"songs": [
|
|
9052,
|
|
10024,
|
|
12054,
|
|
22017,
|
|
],
|
|
},
|
|
{
|
|
"name": "SEVEN",
|
|
"id": 5,
|
|
"songs": [
|
|
13014,
|
|
11014,
|
|
17059,
|
|
22011,
|
|
],
|
|
},
|
|
{
|
|
"name": "RAVE",
|
|
"id": 6,
|
|
"songs": [
|
|
21051,
|
|
22078,
|
|
21083,
|
|
23078,
|
|
],
|
|
},
|
|
{
|
|
"name": "P*Light",
|
|
"id": 7,
|
|
"songs": [
|
|
22061,
|
|
23031,
|
|
22080,
|
|
23079,
|
|
],
|
|
},
|
|
{
|
|
"name": "GRAND FINAL",
|
|
"id": 8,
|
|
"songs": [
|
|
14021,
|
|
21032,
|
|
21045,
|
|
23075,
|
|
],
|
|
},
|
|
{
|
|
"name": "SAY RYU",
|
|
"id": 9,
|
|
"songs": [
|
|
13038,
|
|
15026,
|
|
21007,
|
|
23082,
|
|
],
|
|
},
|
|
{
|
|
"name": "SUMMER",
|
|
"id": 10,
|
|
"songs": [
|
|
15048,
|
|
18005,
|
|
18021,
|
|
23091,
|
|
],
|
|
},
|
|
{
|
|
"name": "ART CORE",
|
|
"id": 11,
|
|
"songs": [
|
|
10023,
|
|
22051,
|
|
20097,
|
|
23090,
|
|
],
|
|
},
|
|
{
|
|
"name": "HAPPY",
|
|
"id": 12,
|
|
"songs": [
|
|
11036,
|
|
19070,
|
|
20040,
|
|
23093,
|
|
],
|
|
},
|
|
{
|
|
"name": "TAG",
|
|
"id": 13,
|
|
"songs": [
|
|
21058,
|
|
20015,
|
|
18056,
|
|
23095,
|
|
],
|
|
},
|
|
]
|
|
|
|
# For some reason, copula omnimix crashes on course mode, so don't enable it
|
|
if not self.omnimix:
|
|
internet_ranking = Node.void("internet_ranking")
|
|
root.add_child(internet_ranking)
|
|
|
|
used_ids: List[int] = []
|
|
for c in courses:
|
|
if c["id"] in used_ids:
|
|
raise Exception("Cannot have multiple courses with the same ID!")
|
|
elif c["id"] < 0 or c["id"] >= 20:
|
|
raise Exception("Course ID is out of bounds!")
|
|
else:
|
|
used_ids.append(c["id"])
|
|
course = Node.void("course")
|
|
internet_ranking.add_child(course)
|
|
course.set_attribute("opflg", "1")
|
|
course.set_attribute("course_id", str(c["id"]))
|
|
course.set_attribute("mid0", str(c["songs"][0]))
|
|
course.set_attribute("mid1", str(c["songs"][1]))
|
|
course.set_attribute("mid2", str(c["songs"][2]))
|
|
course.set_attribute("mid3", str(c["songs"][3]))
|
|
course.set_attribute("name", c["name"])
|
|
|
|
secret_ex_course = Node.void("secret_ex_course")
|
|
root.add_child(secret_ex_course)
|
|
|
|
used_secret_ids: List[int] = []
|
|
for c in secret_courses:
|
|
if c["id"] in used_secret_ids:
|
|
raise Exception(
|
|
"Cannot have multiple secret courses with the same ID!"
|
|
)
|
|
elif c["id"] < 0 or c["id"] >= 20:
|
|
raise Exception("Secret course ID is out of bounds!")
|
|
else:
|
|
used_secret_ids.append(c["id"])
|
|
course = Node.void("course")
|
|
secret_ex_course.add_child(course)
|
|
course.set_attribute("course_id", str(c["id"]))
|
|
course.set_attribute("mid0", str(c["songs"][0]))
|
|
course.set_attribute("mid1", str(c["songs"][1]))
|
|
course.set_attribute("mid2", str(c["songs"][2]))
|
|
course.set_attribute("mid3", str(c["songs"][3]))
|
|
course.set_attribute("name", c["name"])
|
|
|
|
expert = Node.void("expert")
|
|
root.add_child(expert)
|
|
expert.set_attribute("phase", "1")
|
|
|
|
expert_random_select = Node.void("expert_random_select")
|
|
root.add_child(expert_random_select)
|
|
expert_random_select.set_attribute("phase", "1")
|
|
|
|
expert_full = Node.void("expert_secret_full_open")
|
|
root.add_child(expert_full)
|
|
|
|
return root
|
|
|
|
def handle_IIDX23pc_delete_request(self, request: Node) -> Node:
|
|
return Node.void("IIDX23pc")
|
|
|
|
def handle_IIDX23pc_playstart_request(self, request: Node) -> Node:
|
|
return Node.void("IIDX23pc")
|
|
|
|
def handle_IIDX23pc_playend_request(self, request: Node) -> Node:
|
|
return Node.void("IIDX23pc")
|
|
|
|
def handle_IIDX23pc_oldget_request(self, request: Node) -> Node:
|
|
refid = request.attribute("rid")
|
|
userid = self.data.remote.user.from_refid(self.game, self.version, refid)
|
|
if userid is not None:
|
|
oldversion = self.previous_version()
|
|
profile = oldversion.get_profile(userid)
|
|
else:
|
|
profile = None
|
|
|
|
root = Node.void("IIDX23pc")
|
|
root.set_attribute("status", "1" if profile is None else "0")
|
|
return root
|
|
|
|
def handle_IIDX23pc_getname_request(self, request: Node) -> Node:
|
|
refid = request.attribute("rid")
|
|
userid = self.data.remote.user.from_refid(self.game, self.version, refid)
|
|
if userid is not None:
|
|
oldversion = self.previous_version()
|
|
profile = oldversion.get_profile(userid)
|
|
else:
|
|
profile = None
|
|
if profile is None:
|
|
raise Exception(
|
|
"Should not get here if we have no profile, we should "
|
|
+ "have returned '1' in the 'oldget' method above "
|
|
+ "which should tell the game not to present a migration."
|
|
)
|
|
|
|
root = Node.void("IIDX23pc")
|
|
root.set_attribute("name", profile.get_str("name"))
|
|
root.set_attribute("idstr", ID.format_extid(profile.extid))
|
|
root.set_attribute("pid", str(profile.get_int("pid")))
|
|
return root
|
|
|
|
def handle_IIDX23pc_takeover_request(self, request: Node) -> Node:
|
|
refid = request.attribute("rid")
|
|
name = request.attribute("name")
|
|
pid = int(request.attribute("pid"))
|
|
newprofile = self.new_profile_by_refid(refid, name, pid)
|
|
|
|
root = Node.void("IIDX23pc")
|
|
if newprofile is not None:
|
|
root.set_attribute("id", str(newprofile.extid))
|
|
return root
|
|
|
|
def handle_IIDX23pc_reg_request(self, request: Node) -> Node:
|
|
refid = request.attribute("rid")
|
|
name = request.attribute("name")
|
|
pid = int(request.attribute("pid"))
|
|
profile = self.new_profile_by_refid(refid, name, pid)
|
|
|
|
root = Node.void("IIDX23pc")
|
|
if profile is not None:
|
|
root.set_attribute("id", str(profile.extid))
|
|
root.set_attribute("id_str", ID.format_extid(profile.extid))
|
|
return root
|
|
|
|
def handle_IIDX23pc_get_request(self, request: Node) -> Node:
|
|
refid = request.attribute("rid")
|
|
root = self.get_profile_by_refid(refid)
|
|
if root is None:
|
|
root = Node.void("IIDX23pc")
|
|
return root
|
|
|
|
def handle_IIDX23pc_save_request(self, request: Node) -> Node:
|
|
extid = int(request.attribute("iidxid"))
|
|
self.put_profile_by_extid(extid, request)
|
|
|
|
root = Node.void("IIDX23pc")
|
|
return root
|
|
|
|
def handle_IIDX23pc_visit_request(self, request: Node) -> Node:
|
|
root = Node.void("IIDX23pc")
|
|
root.set_attribute("anum", "0")
|
|
root.set_attribute("pnum", "0")
|
|
root.set_attribute("sflg", "0")
|
|
root.set_attribute("pflg", "0")
|
|
root.set_attribute("aflg", "0")
|
|
root.set_attribute("snum", "0")
|
|
return root
|
|
|
|
def handle_IIDX23pc_shopregister_request(self, request: Node) -> Node:
|
|
extid = int(request.child_value("iidx_id"))
|
|
location = ID.parse_machine_id(request.child_value("location_id"))
|
|
|
|
userid = self.data.remote.user.from_extid(self.game, self.version, extid)
|
|
if userid is not None:
|
|
profile = self.get_profile(userid)
|
|
if profile is None:
|
|
profile = Profile(self.game, self.version, "", extid)
|
|
profile.replace_int("shop_location", location)
|
|
self.put_profile(userid, profile)
|
|
|
|
root = Node.void("IIDX23pc")
|
|
return root
|
|
|
|
def handle_IIDX23pc_eaappliresult_request(self, request: Node) -> Node:
|
|
clear_map = {
|
|
self.GAME_CLEAR_STATUS_NO_PLAY: "NO PLAY",
|
|
self.GAME_CLEAR_STATUS_FAILED: "FAILED",
|
|
self.GAME_CLEAR_STATUS_ASSIST_CLEAR: "ASSIST CLEAR",
|
|
self.GAME_CLEAR_STATUS_EASY_CLEAR: "EASY CLEAR",
|
|
self.GAME_CLEAR_STATUS_CLEAR: "CLEAR",
|
|
self.GAME_CLEAR_STATUS_HARD_CLEAR: "HARD CLEAR",
|
|
self.GAME_CLEAR_STATUS_EX_HARD_CLEAR: "EX HARD CLEAR",
|
|
self.GAME_CLEAR_STATUS_FULL_COMBO: "FULL COMBO",
|
|
}
|
|
# first we'll grab the data from the packet
|
|
# did = request.child_value('did')
|
|
# rid = request.child_value('rid')
|
|
name = request.child_value("name")
|
|
# qpro_hair = request.child_value('qpro_hair')
|
|
# qpro_head = request.child_value('qpro_head')
|
|
# qpro_body = request.child_value('qpro_body')
|
|
# qpro_hand = request.child_value('qpro_hand')
|
|
music_id = request.child_value("music_id")
|
|
class_id = request.child_value("class_id")
|
|
# no_save = request.child_value('no_save')
|
|
# is_couple = request.child_value('is_couple')
|
|
# target_graph = request.child_value('target_graph')
|
|
target_exscore = request.child_value("target_exscore")
|
|
# pacemaker = request.child_value('pacemaker')
|
|
best_clear = request.child_value("best_clear")
|
|
# best_djlevel = request.child_value('best_djlevel')
|
|
# best_exscore = request.child_value('best_exscore')
|
|
# best_misscount = request.child_value('best_misscount')
|
|
now_clear = request.child_value("now_clear")
|
|
# now_djlevel = request.child_value('now_djlevel')
|
|
now_exscore = request.child_value("now_exscore")
|
|
# now_misscount = request.child_value('now_misscount')
|
|
now_pgreat = request.child_value("now_pgreat")
|
|
now_great = request.child_value("now_great")
|
|
now_good = request.child_value("now_good")
|
|
now_bad = request.child_value("now_bad")
|
|
now_poor = request.child_value("now_poor")
|
|
now_combo = request.child_value("now_combo")
|
|
now_fast = request.child_value("now_fast")
|
|
now_slow = request.child_value("now_slow")
|
|
best_clear_string = clear_map.get(best_clear, "NO PLAY")
|
|
now_clear_string = clear_map.get(now_clear, "NO PLAY")
|
|
# let's get the song info first
|
|
song = self.data.local.music.get_song(
|
|
self.game, self.music_version, music_id, self.game_to_db_chart(class_id)
|
|
)
|
|
notecount = song.data.get("notecount", 0)
|
|
# Construct the dictionary for the broadcast
|
|
card_data = {
|
|
BroadcastConstants.DJ_NAME: name,
|
|
BroadcastConstants.SONG_NAME: song.name,
|
|
BroadcastConstants.ARTIST_NAME: song.artist,
|
|
BroadcastConstants.DIFFICULTY: song.data.get("difficulty", 0),
|
|
BroadcastConstants.TARGET_EXSCORE: target_exscore,
|
|
BroadcastConstants.EXSCORE: now_exscore,
|
|
BroadcastConstants.BEST_CLEAR_STATUS: best_clear_string,
|
|
BroadcastConstants.CLEAR_STATUS: now_clear_string,
|
|
BroadcastConstants.PLAY_STATS_HEADER: "How did you do?",
|
|
BroadcastConstants.PERFECT_GREATS: now_pgreat,
|
|
BroadcastConstants.GREATS: now_great,
|
|
BroadcastConstants.GOODS: now_good,
|
|
BroadcastConstants.BADS: now_bad,
|
|
BroadcastConstants.POORS: now_poor,
|
|
BroadcastConstants.COMBO_BREAKS: now_combo,
|
|
BroadcastConstants.SLOWS: now_slow,
|
|
BroadcastConstants.FASTS: now_fast,
|
|
}
|
|
if notecount != 0:
|
|
max_score = notecount * 2
|
|
percent = now_exscore / max_score
|
|
grade = int(9 * percent)
|
|
grades = ["F", "F", "E", "D", "C", "B", "A", "AA", "AAA", "MAX"]
|
|
card_data[BroadcastConstants.GRADE] = grades[grade]
|
|
card_data[BroadcastConstants.RATE] = str(round(percent, 2))
|
|
|
|
# Try to broadcast out the score to our webhook(s)
|
|
self.data.triggers.broadcast_score(card_data, self.game, song)
|
|
|
|
return Node.void("IIDX23pc")
|
|
|
|
def format_profile(self, userid: UserID, profile: Profile) -> Node:
|
|
root = Node.void("IIDX23pc")
|
|
|
|
# Look up play stats we bridge to every mix
|
|
play_stats = self.get_play_statistics(userid)
|
|
|
|
# Look up judge window adjustments
|
|
judge_dict = profile.get_dict("machine_judge_adjust")
|
|
machine_judge = judge_dict.get_dict(self.config.machine.pcbid)
|
|
|
|
# Profile data
|
|
pcdata = Node.void("pcdata")
|
|
root.add_child(pcdata)
|
|
pcdata.set_attribute("id", str(profile.extid))
|
|
pcdata.set_attribute("idstr", ID.format_extid(profile.extid))
|
|
pcdata.set_attribute("name", profile.get_str("name"))
|
|
pcdata.set_attribute("pid", str(profile.get_int("pid")))
|
|
pcdata.set_attribute("spnum", str(play_stats.get_int("single_plays")))
|
|
pcdata.set_attribute("dpnum", str(play_stats.get_int("double_plays")))
|
|
pcdata.set_attribute("sach", str(play_stats.get_int("single_dj_points")))
|
|
pcdata.set_attribute("dach", str(play_stats.get_int("double_dj_points")))
|
|
pcdata.set_attribute("mode", str(profile.get_int("mode")))
|
|
pcdata.set_attribute("pmode", str(profile.get_int("pmode")))
|
|
pcdata.set_attribute("rtype", str(profile.get_int("rtype")))
|
|
pcdata.set_attribute("sp_opt", str(profile.get_int("sp_opt")))
|
|
pcdata.set_attribute("dp_opt", str(profile.get_int("dp_opt")))
|
|
pcdata.set_attribute("dp_opt2", str(profile.get_int("dp_opt2")))
|
|
pcdata.set_attribute("gpos", str(profile.get_int("gpos")))
|
|
pcdata.set_attribute("s_sorttype", str(profile.get_int("s_sorttype")))
|
|
pcdata.set_attribute("d_sorttype", str(profile.get_int("d_sorttype")))
|
|
pcdata.set_attribute("s_pace", str(profile.get_int("s_pace")))
|
|
pcdata.set_attribute("d_pace", str(profile.get_int("d_pace")))
|
|
pcdata.set_attribute("s_gno", str(profile.get_int("s_gno")))
|
|
pcdata.set_attribute("d_gno", str(profile.get_int("d_gno")))
|
|
pcdata.set_attribute("s_gtype", str(profile.get_int("s_gtype")))
|
|
pcdata.set_attribute("d_gtype", str(profile.get_int("d_gtype")))
|
|
pcdata.set_attribute("s_sdlen", str(profile.get_int("s_sdlen")))
|
|
pcdata.set_attribute("d_sdlen", str(profile.get_int("d_sdlen")))
|
|
pcdata.set_attribute("s_sdtype", str(profile.get_int("s_sdtype")))
|
|
pcdata.set_attribute("d_sdtype", str(profile.get_int("d_sdtype")))
|
|
pcdata.set_attribute("s_timing", str(profile.get_int("s_timing")))
|
|
pcdata.set_attribute("d_timing", str(profile.get_int("d_timing")))
|
|
pcdata.set_attribute("s_notes", str(profile.get_float("s_notes")))
|
|
pcdata.set_attribute("d_notes", str(profile.get_float("d_notes")))
|
|
pcdata.set_attribute("s_judge", str(profile.get_int("s_judge")))
|
|
pcdata.set_attribute("d_judge", str(profile.get_int("d_judge")))
|
|
pcdata.set_attribute("s_judgeAdj", str(machine_judge.get_int("single")))
|
|
pcdata.set_attribute("d_judgeAdj", str(machine_judge.get_int("double")))
|
|
pcdata.set_attribute("s_hispeed", str(profile.get_float("s_hispeed")))
|
|
pcdata.set_attribute("d_hispeed", str(profile.get_float("d_hispeed")))
|
|
pcdata.set_attribute("s_liflen", str(profile.get_int("s_lift")))
|
|
pcdata.set_attribute("d_liflen", str(profile.get_int("d_lift")))
|
|
pcdata.set_attribute("s_disp_judge", str(profile.get_int("s_disp_judge")))
|
|
pcdata.set_attribute("d_disp_judge", str(profile.get_int("d_disp_judge")))
|
|
pcdata.set_attribute("s_opstyle", str(profile.get_int("s_opstyle")))
|
|
pcdata.set_attribute("d_opstyle", str(profile.get_int("d_opstyle")))
|
|
pcdata.set_attribute("s_exscore", str(profile.get_int("s_exscore")))
|
|
pcdata.set_attribute("d_exscore", str(profile.get_int("d_exscore")))
|
|
pcdata.set_attribute("s_largejudge", str(profile.get_int("s_largejudge")))
|
|
pcdata.set_attribute("d_largejudge", str(profile.get_int("d_largejudge")))
|
|
|
|
premium_unlocks = Node.void("ea_premium_course")
|
|
root.add_child(premium_unlocks)
|
|
|
|
# Secret flags (shh!)
|
|
secret_dict = profile.get_dict("secret")
|
|
secret = Node.void("secret")
|
|
root.add_child(secret)
|
|
secret.add_child(Node.s64_array("flg1", secret_dict.get_int_array("flg1", 4)))
|
|
secret.add_child(Node.s64_array("flg2", secret_dict.get_int_array("flg2", 4)))
|
|
secret.add_child(Node.s64_array("flg3", secret_dict.get_int_array("flg3", 4)))
|
|
|
|
# Favorites
|
|
for folder in ["favorite1", "favorite2", "favorite3"]:
|
|
favorite_dict = profile.get_dict(folder)
|
|
sp_mlist = b""
|
|
sp_clist = b""
|
|
singles_list = favorite_dict["single"] if "single" in favorite_dict else []
|
|
for single in singles_list:
|
|
sp_mlist = sp_mlist + struct.pack("<L", single["id"])
|
|
sp_clist = sp_clist + struct.pack("B", single["chart"])
|
|
while len(sp_mlist) < (4 * self.FAVORITE_LIST_LENGTH):
|
|
sp_mlist = sp_mlist + b"\x00\x00\x00\x00"
|
|
while len(sp_clist) < self.FAVORITE_LIST_LENGTH:
|
|
sp_clist = sp_clist + b"\x00"
|
|
|
|
dp_mlist = b""
|
|
dp_clist = b""
|
|
doubles_list = favorite_dict["double"] if "double" in favorite_dict else []
|
|
for double in doubles_list:
|
|
dp_mlist = dp_mlist + struct.pack("<L", double["id"])
|
|
dp_clist = dp_clist + struct.pack("B", double["chart"])
|
|
while len(dp_mlist) < (4 * self.FAVORITE_LIST_LENGTH):
|
|
dp_mlist = dp_mlist + b"\x00\x00\x00\x00"
|
|
while len(dp_clist) < self.FAVORITE_LIST_LENGTH:
|
|
dp_clist = dp_clist + b"\x00"
|
|
|
|
if folder == "favorite1":
|
|
favorite = Node.void("favorite")
|
|
elif folder == "favorite2":
|
|
favorite = Node.void("extra_favorite")
|
|
favorite.set_attribute("folder_id", "0")
|
|
elif folder == "favorite3":
|
|
favorite = Node.void("extra_favorite")
|
|
favorite.set_attribute("folder_id", "1")
|
|
root.add_child(favorite)
|
|
favorite.add_child(Node.binary("sp_mlist", sp_mlist))
|
|
favorite.add_child(Node.binary("sp_clist", sp_clist))
|
|
favorite.add_child(Node.binary("dp_mlist", dp_mlist))
|
|
favorite.add_child(Node.binary("dp_clist", dp_clist))
|
|
|
|
# DAN rankings
|
|
grade = Node.void("grade")
|
|
root.add_child(grade)
|
|
grade.set_attribute(
|
|
"sgid",
|
|
str(
|
|
self.db_to_game_rank(
|
|
profile.get_int(self.DAN_RANKING_SINGLE, -1),
|
|
self.GAME_CLTYPE_SINGLE,
|
|
)
|
|
),
|
|
)
|
|
grade.set_attribute(
|
|
"dgid",
|
|
str(
|
|
self.db_to_game_rank(
|
|
profile.get_int(self.DAN_RANKING_DOUBLE, -1),
|
|
self.GAME_CLTYPE_DOUBLE,
|
|
)
|
|
),
|
|
)
|
|
rankings = self.data.local.user.get_achievements(
|
|
self.game, self.version, userid
|
|
)
|
|
for rank in rankings:
|
|
if rank.type == self.DAN_RANKING_SINGLE:
|
|
grade.add_child(
|
|
Node.u8_array(
|
|
"g",
|
|
[
|
|
self.GAME_CLTYPE_SINGLE,
|
|
self.db_to_game_rank(rank.id, self.GAME_CLTYPE_SINGLE),
|
|
rank.data.get_int("stages_cleared"),
|
|
rank.data.get_int("percent"),
|
|
],
|
|
)
|
|
)
|
|
if rank.type == self.DAN_RANKING_DOUBLE:
|
|
grade.add_child(
|
|
Node.u8_array(
|
|
"g",
|
|
[
|
|
self.GAME_CLTYPE_DOUBLE,
|
|
self.db_to_game_rank(rank.id, self.GAME_CLTYPE_DOUBLE),
|
|
rank.data.get_int("stages_cleared"),
|
|
rank.data.get_int("percent"),
|
|
],
|
|
)
|
|
)
|
|
|
|
# User settings
|
|
settings_dict = profile.get_dict("settings")
|
|
skin = Node.s16_array(
|
|
"skin",
|
|
[
|
|
settings_dict.get_int("frame"),
|
|
settings_dict.get_int("turntable"),
|
|
settings_dict.get_int("burst"),
|
|
settings_dict.get_int("bgm"),
|
|
settings_dict.get_int("flags"),
|
|
settings_dict.get_int("towel"),
|
|
settings_dict.get_int("judge_pos"),
|
|
settings_dict.get_int("voice"),
|
|
settings_dict.get_int("noteskin"),
|
|
settings_dict.get_int("full_combo"),
|
|
settings_dict.get_int("beam"),
|
|
settings_dict.get_int("judge"),
|
|
0,
|
|
settings_dict.get_int("disable_song_preview"),
|
|
],
|
|
)
|
|
root.add_child(skin)
|
|
|
|
# Qpro data
|
|
qpro_dict = profile.get_dict("qpro")
|
|
root.add_child(
|
|
Node.u32_array(
|
|
"qprodata",
|
|
[
|
|
qpro_dict.get_int("head"),
|
|
qpro_dict.get_int("hair"),
|
|
qpro_dict.get_int("face"),
|
|
qpro_dict.get_int("hand"),
|
|
qpro_dict.get_int("body"),
|
|
],
|
|
)
|
|
)
|
|
|
|
# Rivals
|
|
rlist = Node.void("rlist")
|
|
root.add_child(rlist)
|
|
links = self.data.local.user.get_links(self.game, self.version, userid)
|
|
for link in links:
|
|
rival_type = None
|
|
if link.type == "sp_rival":
|
|
rival_type = "1"
|
|
elif link.type == "dp_rival":
|
|
rival_type = "2"
|
|
else:
|
|
# No business with this link type
|
|
continue
|
|
|
|
other_profile = self.get_profile(link.other_userid)
|
|
if other_profile is None:
|
|
continue
|
|
other_play_stats = self.get_play_statistics(link.other_userid)
|
|
|
|
rival = Node.void("rival")
|
|
rlist.add_child(rival)
|
|
rival.set_attribute("spdp", rival_type)
|
|
rival.set_attribute("id", str(other_profile.extid))
|
|
rival.set_attribute("id_str", ID.format_extid(other_profile.extid))
|
|
rival.set_attribute("djname", other_profile.get_str("name"))
|
|
rival.set_attribute("pid", str(other_profile.get_int("pid")))
|
|
rival.set_attribute(
|
|
"sg",
|
|
str(
|
|
self.db_to_game_rank(
|
|
other_profile.get_int(self.DAN_RANKING_SINGLE, -1),
|
|
self.GAME_CLTYPE_SINGLE,
|
|
)
|
|
),
|
|
)
|
|
rival.set_attribute(
|
|
"dg",
|
|
str(
|
|
self.db_to_game_rank(
|
|
other_profile.get_int(self.DAN_RANKING_DOUBLE, -1),
|
|
self.GAME_CLTYPE_DOUBLE,
|
|
)
|
|
),
|
|
)
|
|
rival.set_attribute("sa", str(other_play_stats.get_int("single_dj_points")))
|
|
rival.set_attribute("da", str(other_play_stats.get_int("double_dj_points")))
|
|
|
|
# If the user joined a particular shop, let the game know.
|
|
if "shop_location" in other_profile:
|
|
shop_id = other_profile.get_int("shop_location")
|
|
machine = self.get_machine_by_id(shop_id)
|
|
if machine is not None:
|
|
shop = Node.void("shop")
|
|
rival.add_child(shop)
|
|
shop.set_attribute("name", machine.name)
|
|
|
|
qprodata = Node.void("qprodata")
|
|
rival.add_child(qprodata)
|
|
qpro = other_profile.get_dict("qpro")
|
|
qprodata.set_attribute("head", str(qpro.get_int("head")))
|
|
qprodata.set_attribute("hair", str(qpro.get_int("hair")))
|
|
qprodata.set_attribute("face", str(qpro.get_int("face")))
|
|
qprodata.set_attribute("body", str(qpro.get_int("body")))
|
|
qprodata.set_attribute("hand", str(qpro.get_int("hand")))
|
|
|
|
# Expert courses
|
|
ir_data = Node.void("ir_data")
|
|
root.add_child(ir_data)
|
|
for rank in rankings:
|
|
if rank.type == self.COURSE_TYPE_INTERNET_RANKING:
|
|
ir_data.add_child(
|
|
Node.s32_array(
|
|
"e",
|
|
[
|
|
int(rank.id / 6), # course ID
|
|
rank.id % 6, # course chart
|
|
self.db_to_game_status(
|
|
rank.data.get_int("clear_status")
|
|
), # course clear status
|
|
rank.data.get_int("pgnum"), # flashing great count
|
|
rank.data.get_int("gnum"), # great count
|
|
],
|
|
)
|
|
)
|
|
|
|
secret_course_data = Node.void("secret_course_data")
|
|
root.add_child(secret_course_data)
|
|
for rank in rankings:
|
|
if rank.type == self.COURSE_TYPE_SECRET:
|
|
secret_course_data.add_child(
|
|
Node.s32_array(
|
|
"e",
|
|
[
|
|
int(rank.id / 6), # course ID
|
|
rank.id % 6, # course chart
|
|
self.db_to_game_status(
|
|
rank.data.get_int("clear_status")
|
|
), # course clear status
|
|
rank.data.get_int("pgnum"), # flashing great count
|
|
rank.data.get_int("gnum"), # great count
|
|
],
|
|
)
|
|
)
|
|
|
|
# If the user joined a particular shop, let the game know.
|
|
if "shop_location" in profile:
|
|
shop_id = profile.get_int("shop_location")
|
|
machine = self.get_machine_by_id(shop_id)
|
|
if machine is not None:
|
|
join_shop = Node.void("join_shop")
|
|
root.add_child(join_shop)
|
|
join_shop.set_attribute("joinflg", "1")
|
|
join_shop.set_attribute("join_cflg", "1")
|
|
join_shop.set_attribute("join_id", ID.format_machine_id(machine.id))
|
|
join_shop.set_attribute("join_name", machine.name)
|
|
|
|
# Step up mode
|
|
step_dict = profile.get_dict("step")
|
|
step = Node.void("step")
|
|
root.add_child(step)
|
|
step.set_attribute("friendship", str(step_dict.get_int("friendship")))
|
|
step.set_attribute("progress", str(step_dict.get_int("progress")))
|
|
step.set_attribute("station_clear", str(step_dict.get_int("station_clear")))
|
|
step.set_attribute("station_play", str(step_dict.get_int("station_play")))
|
|
step.set_attribute("sp_mission", str(step_dict.get_int("sp_mission")))
|
|
step.set_attribute("dp_mission", str(step_dict.get_int("dp_mission")))
|
|
step.set_attribute("sp_level", str(step_dict.get_int("sp_level")))
|
|
step.set_attribute("dp_level", str(step_dict.get_int("dp_level")))
|
|
step.set_attribute("sp_mplay", str(step_dict.get_int("sp_mplay")))
|
|
step.set_attribute("dp_mplay", str(step_dict.get_int("dp_mplay")))
|
|
step.set_attribute("mission_gauge", str(step_dict.get_int("mission_gauge")))
|
|
if "tokimeki" in step_dict:
|
|
step.add_child(
|
|
Node.binary(
|
|
"tokimeki",
|
|
step_dict.get_bytes(
|
|
"tokimeki", b"\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"
|
|
),
|
|
)
|
|
)
|
|
|
|
# Daily recommendations
|
|
entry = self.data.local.game.get_time_sensitive_settings(
|
|
self.game, self.version, "dailies"
|
|
)
|
|
if entry is not None:
|
|
packinfo = Node.void("packinfo")
|
|
root.add_child(packinfo)
|
|
|
|
pack_id = int(entry["start_time"] / 86400)
|
|
packinfo.set_attribute("pack_id", str(pack_id))
|
|
packinfo.set_attribute("music_0", str(entry["music"][0]))
|
|
packinfo.set_attribute("music_1", str(entry["music"][1]))
|
|
packinfo.set_attribute("music_2", str(entry["music"][2]))
|
|
else:
|
|
# No dailies :(
|
|
pack_id = None
|
|
|
|
# Tran medals and shit
|
|
achievements = Node.void("achievements")
|
|
root.add_child(achievements)
|
|
|
|
# Dailies
|
|
if pack_id is None:
|
|
achievements.set_attribute("pack", "0")
|
|
achievements.set_attribute("pack_comp", "0")
|
|
else:
|
|
daily_played = self.data.local.user.get_achievement(
|
|
self.game, self.version, userid, pack_id, "daily"
|
|
)
|
|
if daily_played is None:
|
|
daily_played = ValidatedDict()
|
|
achievements.set_attribute("pack", str(daily_played.get_int("pack_flg")))
|
|
achievements.set_attribute(
|
|
"pack_comp", str(daily_played.get_int("pack_comp"))
|
|
)
|
|
|
|
# Weeklies
|
|
achievements.set_attribute("last_weekly", str(profile.get_int("last_weekly")))
|
|
achievements.set_attribute("weekly_num", str(profile.get_int("weekly_num")))
|
|
|
|
# Prefecture visit flag
|
|
achievements.set_attribute("visit_flg", str(profile.get_int("visit_flg")))
|
|
|
|
# Number of rivals beaten
|
|
achievements.set_attribute("rival_crush", str(profile.get_int("rival_crush")))
|
|
|
|
# Tran medals
|
|
achievements.add_child(
|
|
Node.s64_array("trophy", profile.get_int_array("trophy", 10))
|
|
)
|
|
|
|
# Track deller
|
|
deller = Node.void("deller")
|
|
root.add_child(deller)
|
|
deller.set_attribute("deller", str(profile.get_int("deller")))
|
|
deller.set_attribute("rate", "0")
|
|
|
|
# Orb data
|
|
orb_data = Node.void("orb_data")
|
|
root.add_child(orb_data)
|
|
orb_data.set_attribute("rest_orb", str(profile.get_int("orbs")))
|
|
|
|
# Expert points
|
|
expert_point = Node.void("expert_point")
|
|
root.add_child(expert_point)
|
|
for rank in rankings:
|
|
if rank.type == "expert_point":
|
|
detail = Node.void("detail")
|
|
expert_point.add_child(detail)
|
|
detail.set_attribute("course_id", str(rank.id))
|
|
detail.set_attribute("n_point", str(rank.data.get_int("normal_points")))
|
|
detail.set_attribute("h_point", str(rank.data.get_int("hyper_points")))
|
|
detail.set_attribute(
|
|
"a_point", str(rank.data.get_int("another_points"))
|
|
)
|
|
|
|
# Tokotoko Line Event
|
|
if "event1" in profile:
|
|
# Only provide this if we've already saved it, so the game can init it's own way.
|
|
event1_dict = profile.get_dict("event1")
|
|
event1_data = Node.void("event1_data")
|
|
root.add_child(event1_data)
|
|
event1_data.set_attribute(
|
|
"point_map_0", str(event1_dict.get_int("point_map_0"))
|
|
)
|
|
event1_data.set_attribute(
|
|
"point_map_1", str(event1_dict.get_int("point_map_1"))
|
|
)
|
|
event1_data.set_attribute(
|
|
"point_map_2", str(event1_dict.get_int("point_map_2"))
|
|
)
|
|
event1_data.set_attribute(
|
|
"point_map_3", str(event1_dict.get_int("point_map_3"))
|
|
)
|
|
event1_data.set_attribute(
|
|
"point_map_4", str(event1_dict.get_int("point_map_4"))
|
|
)
|
|
event1_data.set_attribute(
|
|
"hold_point", str(event1_dict.get_int("hold_point"))
|
|
)
|
|
event1_data.set_attribute("last_map", str(event1_dict.get_int("last_map")))
|
|
event1_data.set_attribute(
|
|
"rank_point", str(event1_dict.get_int("rank_point"))
|
|
)
|
|
event1_data.set_attribute(
|
|
"tips_list", str(event1_dict.get_int("tips_list"))
|
|
)
|
|
event1_data.set_attribute("gift_point", "0")
|
|
|
|
# Mystery Line Event
|
|
if "event2" in profile:
|
|
# Only provide this if we've already saved it, so the game can init it's own way.
|
|
event2_dict = profile.get_dict("event2")
|
|
event2_data = Node.void("event2_data")
|
|
root.add_child(event2_data)
|
|
event2_data.set_attribute("play_num", str(event2_dict.get_int("play_num")))
|
|
event2_data.set_attribute("now_area", str(event2_dict.get_int("now_area")))
|
|
event2_data.set_attribute(
|
|
"now_note_grade", str(event2_dict.get_int("now_note_grade"))
|
|
)
|
|
event2_data.set_attribute(
|
|
"stop_area_time", str(event2_dict.get_int("stop_area_time"))
|
|
)
|
|
|
|
areas = self.data.local.user.get_achievements(
|
|
self.game, self.version, userid
|
|
)
|
|
for area in areas:
|
|
if area.type != "event2_area_data":
|
|
continue
|
|
|
|
event2_area_data = Node.void("event2_area_data")
|
|
event2_data.add_child(event2_area_data)
|
|
event2_area_data.set_attribute("area_no", str(area.id))
|
|
event2_area_data.set_attribute(
|
|
"area_play", str(area.data.get_int("play_num"))
|
|
)
|
|
event2_area_data.set_attribute(
|
|
"normal_point", str(area.data.get_int("normal_point"))
|
|
)
|
|
event2_area_data.set_attribute(
|
|
"hyper_point", str(area.data.get_int("hyper_point"))
|
|
)
|
|
event2_area_data.set_attribute(
|
|
"another_point", str(area.data.get_int("another_point"))
|
|
)
|
|
|
|
# One More Event
|
|
if "onemore" in profile:
|
|
# Only provide this if we've already saved it, so the game can init it's own way.
|
|
onemore_dict = profile.get_dict("onemore")
|
|
onemore_data = Node.void("onemore_data")
|
|
root.add_child(onemore_data)
|
|
onemore_data.set_attribute(
|
|
"defeat_0", str(onemore_dict.get_int("defeat_0"))
|
|
)
|
|
onemore_data.set_attribute(
|
|
"defeat_1", str(onemore_dict.get_int("defeat_1"))
|
|
)
|
|
onemore_data.set_attribute(
|
|
"defeat_2", str(onemore_dict.get_int("defeat_2"))
|
|
)
|
|
onemore_data.set_attribute(
|
|
"defeat_3", str(onemore_dict.get_int("defeat_3"))
|
|
)
|
|
onemore_data.set_attribute(
|
|
"defeat_4", str(onemore_dict.get_int("defeat_4"))
|
|
)
|
|
onemore_data.set_attribute(
|
|
"defeat_5", str(onemore_dict.get_int("defeat_5"))
|
|
)
|
|
onemore_data.set_attribute(
|
|
"challenge_num_n", str(onemore_dict.get_int("challenge_num_n"))
|
|
)
|
|
onemore_data.set_attribute(
|
|
"challenge_num_h", str(onemore_dict.get_int("challenge_num_h"))
|
|
)
|
|
onemore_data.set_attribute(
|
|
"challenge_num_a", str(onemore_dict.get_int("challenge_num_a"))
|
|
)
|
|
|
|
# Ea app features
|
|
if self.data.triggers.has_broadcast_destination(self.game):
|
|
root.add_child(Node.void("bind_eaappli"))
|
|
return root
|
|
|
|
def unformat_profile(
|
|
self, userid: UserID, request: Node, oldprofile: Profile
|
|
) -> Profile:
|
|
newprofile = oldprofile.clone()
|
|
play_stats = self.get_play_statistics(userid)
|
|
|
|
# Track play counts
|
|
cltype = int(request.attribute("cltype"))
|
|
if cltype == self.GAME_CLTYPE_SINGLE:
|
|
play_stats.increment_int("single_plays")
|
|
if cltype == self.GAME_CLTYPE_DOUBLE:
|
|
play_stats.increment_int("double_plays")
|
|
|
|
# Track DJ points
|
|
play_stats.replace_int("single_dj_points", int(request.attribute("s_achi")))
|
|
play_stats.replace_int("double_dj_points", int(request.attribute("d_achi")))
|
|
|
|
# Profile settings
|
|
newprofile.replace_int("mode", int(request.attribute("mode")))
|
|
newprofile.replace_int("pmode", int(request.attribute("pmode")))
|
|
newprofile.replace_int("rtype", int(request.attribute("rtype")))
|
|
newprofile.replace_int("s_lift", int(request.attribute("s_lift")))
|
|
newprofile.replace_int("d_lift", int(request.attribute("d_lift")))
|
|
newprofile.replace_int("sp_opt", int(request.attribute("sp_opt")))
|
|
newprofile.replace_int("dp_opt", int(request.attribute("dp_opt")))
|
|
newprofile.replace_int("dp_opt2", int(request.attribute("dp_opt2")))
|
|
newprofile.replace_int("gpos", int(request.attribute("gpos")))
|
|
newprofile.replace_int("s_sorttype", int(request.attribute("s_sorttype")))
|
|
newprofile.replace_int("d_sorttype", int(request.attribute("d_sorttype")))
|
|
newprofile.replace_int("s_pace", int(request.attribute("s_pace")))
|
|
newprofile.replace_int("d_pace", int(request.attribute("d_pace")))
|
|
newprofile.replace_int("s_gno", int(request.attribute("s_gno")))
|
|
newprofile.replace_int("d_gno", int(request.attribute("d_gno")))
|
|
newprofile.replace_int("s_gtype", int(request.attribute("s_gtype")))
|
|
newprofile.replace_int("d_gtype", int(request.attribute("d_gtype")))
|
|
newprofile.replace_int("s_sdlen", int(request.attribute("s_sdlen")))
|
|
newprofile.replace_int("d_sdlen", int(request.attribute("d_sdlen")))
|
|
newprofile.replace_int("s_sdtype", int(request.attribute("s_sdtype")))
|
|
newprofile.replace_int("d_sdtype", int(request.attribute("d_sdtype")))
|
|
newprofile.replace_int("s_timing", int(request.attribute("s_timing")))
|
|
newprofile.replace_int("d_timing", int(request.attribute("d_timing")))
|
|
newprofile.replace_float("s_notes", float(request.attribute("s_notes")))
|
|
newprofile.replace_float("d_notes", float(request.attribute("d_notes")))
|
|
newprofile.replace_int("s_judge", int(request.attribute("s_judge")))
|
|
newprofile.replace_int("d_judge", int(request.attribute("d_judge")))
|
|
newprofile.replace_float("s_hispeed", float(request.attribute("s_hispeed")))
|
|
newprofile.replace_float("d_hispeed", float(request.attribute("d_hispeed")))
|
|
newprofile.replace_int("s_disp_judge", int(request.attribute("s_disp_judge")))
|
|
newprofile.replace_int("d_disp_judge", int(request.attribute("d_disp_judge")))
|
|
newprofile.replace_int("s_opstyle", int(request.attribute("s_opstyle")))
|
|
newprofile.replace_int("d_opstyle", int(request.attribute("d_opstyle")))
|
|
newprofile.replace_int("s_exscore", int(request.attribute("s_exscore")))
|
|
newprofile.replace_int("d_exscore", int(request.attribute("d_exscore")))
|
|
newprofile.replace_int("s_largejudge", int(request.attribute("s_largejudge")))
|
|
newprofile.replace_int("d_largejudge", int(request.attribute("d_largejudge")))
|
|
|
|
# Update judge window adjustments per-machine
|
|
judge_dict = newprofile.get_dict("machine_judge_adjust")
|
|
machine_judge = judge_dict.get_dict(self.config.machine.pcbid)
|
|
machine_judge.replace_int("single", int(request.attribute("s_judgeAdj")))
|
|
machine_judge.replace_int("double", int(request.attribute("d_judgeAdj")))
|
|
judge_dict.replace_dict(self.config.machine.pcbid, machine_judge)
|
|
newprofile.replace_dict("machine_judge_adjust", judge_dict)
|
|
|
|
# Secret flags saving
|
|
secret = request.child("secret")
|
|
if secret is not None:
|
|
secret_dict = newprofile.get_dict("secret")
|
|
secret_dict.replace_int_array("flg1", 4, secret.child_value("flg1"))
|
|
secret_dict.replace_int_array("flg2", 4, secret.child_value("flg2"))
|
|
secret_dict.replace_int_array("flg3", 4, secret.child_value("flg3"))
|
|
newprofile.replace_dict("secret", secret_dict)
|
|
|
|
# Basic achievements
|
|
achievements = request.child("achievements")
|
|
if achievements is not None:
|
|
newprofile.replace_int(
|
|
"visit_flg", int(achievements.attribute("visit_flg"))
|
|
)
|
|
newprofile.replace_int(
|
|
"last_weekly", int(achievements.attribute("last_weekly"))
|
|
)
|
|
newprofile.replace_int(
|
|
"weekly_num", int(achievements.attribute("weekly_num"))
|
|
)
|
|
|
|
pack_id = int(achievements.attribute("pack_id"))
|
|
if pack_id > 0:
|
|
self.data.local.user.put_achievement(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
pack_id,
|
|
"daily",
|
|
{
|
|
"pack_flg": int(achievements.attribute("pack_flg")),
|
|
"pack_comp": int(achievements.attribute("pack_comp")),
|
|
},
|
|
)
|
|
|
|
trophies = achievements.child("trophy")
|
|
if trophies is not None:
|
|
# We only load the first 10 in profile load.
|
|
newprofile.replace_int_array("trophy", 10, trophies.value[:10])
|
|
|
|
# Deller saving
|
|
deller = request.child("deller")
|
|
if deller is not None:
|
|
newprofile.replace_int(
|
|
"deller", newprofile.get_int("deller") + int(deller.attribute("deller"))
|
|
)
|
|
|
|
# Secret course expert point saving
|
|
expert_point = request.child("expert_point")
|
|
if expert_point is not None:
|
|
courseid = int(expert_point.attribute("course_id"))
|
|
|
|
# Update achievement to track expert points
|
|
expert_point_achievement = self.data.local.user.get_achievement(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
courseid,
|
|
"expert_point",
|
|
)
|
|
if expert_point_achievement is None:
|
|
expert_point_achievement = ValidatedDict()
|
|
expert_point_achievement.replace_int(
|
|
"normal_points",
|
|
int(expert_point.attribute("n_point")),
|
|
)
|
|
expert_point_achievement.replace_int(
|
|
"hyper_points",
|
|
int(expert_point.attribute("h_point")),
|
|
)
|
|
expert_point_achievement.replace_int(
|
|
"another_points",
|
|
int(expert_point.attribute("a_point")),
|
|
)
|
|
|
|
self.data.local.user.put_achievement(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
courseid,
|
|
"expert_point",
|
|
expert_point_achievement,
|
|
)
|
|
|
|
# Favorites saving
|
|
for favorite in request.children:
|
|
singles = []
|
|
doubles = []
|
|
name = None
|
|
if favorite.name in ["favorite", "extra_favorite"]:
|
|
if favorite.name == "favorite":
|
|
name = "favorite1"
|
|
elif favorite.name == "extra_favorite":
|
|
folder = favorite.attribute("folder_id")
|
|
if folder == "0":
|
|
name = "favorite2"
|
|
if folder == "1":
|
|
name = "favorite3"
|
|
if name is None:
|
|
continue
|
|
|
|
single_music_bin = favorite.child_value("sp_mlist")
|
|
single_chart_bin = favorite.child_value("sp_clist")
|
|
double_music_bin = favorite.child_value("dp_mlist")
|
|
double_chart_bin = favorite.child_value("dp_clist")
|
|
|
|
for i in range(self.FAVORITE_LIST_LENGTH):
|
|
singles.append(
|
|
{
|
|
"id": struct.unpack(
|
|
"<L", single_music_bin[(i * 4) : ((i + 1) * 4)]
|
|
)[0],
|
|
"chart": struct.unpack("B", single_chart_bin[i : (i + 1)])[
|
|
0
|
|
],
|
|
}
|
|
)
|
|
doubles.append(
|
|
{
|
|
"id": struct.unpack(
|
|
"<L", double_music_bin[(i * 4) : ((i + 1) * 4)]
|
|
)[0],
|
|
"chart": struct.unpack("B", double_chart_bin[i : (i + 1)])[
|
|
0
|
|
],
|
|
}
|
|
)
|
|
|
|
# Filter out empty charts
|
|
if name is not None:
|
|
singles = [single for single in singles if single["id"] != 0]
|
|
doubles = [double for double in doubles if double["id"] != 0]
|
|
|
|
newprofile.replace_dict(
|
|
name,
|
|
{
|
|
"single": singles,
|
|
"double": doubles,
|
|
},
|
|
)
|
|
|
|
# Step-up mode
|
|
step = request.child("step")
|
|
if step is not None:
|
|
step_dict = newprofile.get_dict("step")
|
|
step_dict.replace_int("friendship", int(step.attribute("friendship")))
|
|
step_dict.replace_int("progress", int(step.attribute("progress")))
|
|
step_dict.replace_int("station_clear", int(step.attribute("station_clear")))
|
|
step_dict.replace_int("station_play", int(step.attribute("station_play")))
|
|
step_dict.replace_int("sp_mission", int(step.attribute("sp_mission")))
|
|
step_dict.replace_int("dp_mission", int(step.attribute("dp_mission")))
|
|
step_dict.replace_int("sp_level", int(step.attribute("sp_level")))
|
|
step_dict.replace_int("dp_level", int(step.attribute("dp_level")))
|
|
step_dict.replace_int("sp_mplay", int(step.attribute("sp_mplay")))
|
|
step_dict.replace_int("dp_mplay", int(step.attribute("dp_mplay")))
|
|
step_dict.replace_int("mission_gauge", int(step.attribute("mission_gauge")))
|
|
step_dict.replace_bytes("tokimeki", step.value)
|
|
newprofile.replace_dict("step", step_dict)
|
|
|
|
# Orb data saving
|
|
orb_data = request.child("orb_data")
|
|
if orb_data is not None:
|
|
orbs = newprofile.get_int("orbs")
|
|
orbs = orbs + int(orb_data.attribute("add_orb"))
|
|
if orb_data.child_value("use_vip_pass"):
|
|
orbs = 0
|
|
newprofile.replace_int("orbs", orbs)
|
|
|
|
# Tokotoko Line Event
|
|
event1_data = request.child("event1_data")
|
|
if event1_data is not None:
|
|
event1_dict = newprofile.get_dict("event1")
|
|
event1_dict.replace_int(
|
|
"hold_point", int(event1_data.attribute("hold_point"))
|
|
)
|
|
event1_dict.replace_int("last_map", int(event1_data.attribute("last_map")))
|
|
event1_dict.replace_int(
|
|
"rank_point", int(event1_data.attribute("rank_point"))
|
|
)
|
|
event1_dict.replace_int(
|
|
"tips_list", int(event1_data.attribute("tips_list"))
|
|
)
|
|
event1_dict.replace_int(
|
|
"point_map_0", int(event1_data.attribute("point_map_0"))
|
|
)
|
|
event1_dict.replace_int(
|
|
"point_map_1", int(event1_data.attribute("point_map_1"))
|
|
)
|
|
event1_dict.replace_int(
|
|
"point_map_2", int(event1_data.attribute("point_map_2"))
|
|
)
|
|
event1_dict.replace_int(
|
|
"point_map_3", int(event1_data.attribute("point_map_3"))
|
|
)
|
|
event1_dict.replace_int(
|
|
"point_map_4", int(event1_data.attribute("point_map_4"))
|
|
)
|
|
newprofile.replace_dict("event1", event1_dict)
|
|
|
|
# Mystery Line Event
|
|
event2_data = request.child("event2_data")
|
|
if event2_data is not None:
|
|
event2_dict = newprofile.get_dict("event2")
|
|
event2_dict.replace_int(
|
|
"now_area", int(event2_data.attribute("now_stay_area"))
|
|
)
|
|
event2_dict.replace_int(
|
|
"now_note_grade", int(event2_data.attribute("now_stay_note_grade"))
|
|
)
|
|
event2_dict.replace_int("play_num", int(event2_data.attribute("play_num")))
|
|
event2_dict.replace_int(
|
|
"stop_area_time", int(event2_data.attribute("stop_area_time"))
|
|
)
|
|
newprofile.replace_dict("event2", event2_dict)
|
|
|
|
for area_data in event2_data.children:
|
|
if area_data.name != "event2_area_data":
|
|
continue
|
|
|
|
area_id = int(area_data.attribute("area_no"))
|
|
self.data.local.user.put_achievement(
|
|
self.game,
|
|
self.version,
|
|
userid,
|
|
area_id,
|
|
"event2_area_data",
|
|
{
|
|
"play_num": int(area_data.attribute("area_play_num")),
|
|
"normal_point": int(area_data.attribute("normal_point")),
|
|
"hyper_point": int(area_data.attribute("hyper_point")),
|
|
"another_point": int(area_data.attribute("another_point")),
|
|
},
|
|
)
|
|
|
|
# Onemore Event
|
|
onemore_data = request.child("onemore_data")
|
|
if onemore_data is not None:
|
|
onemore_dict = newprofile.get_dict("onemore")
|
|
onemore_dict.replace_int(
|
|
"defeat_0", int(onemore_data.attribute("defeat_0"))
|
|
)
|
|
onemore_dict.replace_int(
|
|
"defeat_1", int(onemore_data.attribute("defeat_1"))
|
|
)
|
|
onemore_dict.replace_int(
|
|
"defeat_2", int(onemore_data.attribute("defeat_2"))
|
|
)
|
|
onemore_dict.replace_int(
|
|
"defeat_3", int(onemore_data.attribute("defeat_3"))
|
|
)
|
|
onemore_dict.replace_int(
|
|
"defeat_4", int(onemore_data.attribute("defeat_4"))
|
|
)
|
|
onemore_dict.replace_int(
|
|
"defeat_5", int(onemore_data.attribute("defeat_5"))
|
|
)
|
|
onemore_dict.replace_int(
|
|
"challenge_num_n", int(onemore_data.attribute("challenge_num_n"))
|
|
)
|
|
onemore_dict.replace_int(
|
|
"challenge_num_h", int(onemore_data.attribute("challenge_num_h"))
|
|
)
|
|
onemore_dict.replace_int(
|
|
"challenge_num_a", int(onemore_data.attribute("challenge_num_a"))
|
|
)
|
|
newprofile.replace_dict("onemore", onemore_dict)
|
|
|
|
# Keep track of play statistics across all mixes
|
|
self.update_play_statistics(userid, play_stats)
|
|
|
|
return newprofile
|