SuperNova 2 Support

bemani/backend/ddr/base.py

@@ -127,6 +127,13 @@ def format_profile(self, userid: UserID, profile: Profile) -> Node:
         """
         return Node.void("game")
 
+    def format_profile_part(self, userid: UserID, profile: Profile, part: str) -> Node:
+        """
+        Base handler for a profile. Given a userid and a profile dictionary,
+        return a Node representing a profile. Should be overridden.
+        """
+        return Node.void("game")
+
     def format_scores(self, userid: UserID, profile: Profile, scores: List[Score]) -> Node:
         """
         Base handler for a score list. Given a userid, profile and a score list,
@@ -142,6 +149,24 @@ def unformat_profile(self, userid: UserID, request: Node, oldprofile: Profile) -
         """
         return oldprofile
 
+    def get_profile_by_refid_and_part(self, refid: Optional[str], part: Optional[str]) -> Optional[Node]:
+        """
+        Given a RefID, return a formatted profile node. Basically every game
+        needs a profile lookup, even if it handles where that happens in
+        a different request. This is provided for code deduplication.
+        """
+        if refid is None:
+            return None
+
+        # First try to load the actual profile
+        userid = self.data.remote.user.from_refid(self.game, self.version, refid)
+        profile = self.get_profile(userid)
+        if profile is None:
+            return None
+
+        # Now, return it
+        return self.format_profile_part(userid, profile, part)
+
     def get_profile_by_refid(self, refid: Optional[str]) -> Optional[Node]:
         """
         Given a RefID, return a formatted profile node. Basically every game

bemani/backend/ddr/ddrsn2/__init__.py

@@ -0,0 +1,3 @@
+from bemani.backend.ddr.ddrsn2.ddrsn2 import DDRSuperNova2
+
+__all__ = ["DDRSuperNova2"]

bemani/backend/ddr/ddrsn2/battlerecord.py

@@ -0,0 +1,26 @@
+import logging
+from ctypes import *
+
+
+class BattleRecordStruct(Structure):
+    _pack_ = 1
+    _fields_ = [
+        ("name", c_char * 14),
+        ("wins", c_uint16),
+        ("loses", c_uint16),
+        ("draws", c_uint16),
+    ]
+
+
+class BattleRecord:
+    @staticmethod
+    def create(name: str, wins: int, losses: int, draws: int) -> BattleRecordStruct:
+        this_name = name
+        if len(this_name) > 8:
+            this_name = this_name[:8]
+            logging.warning("name {} too long, truncating to {}", name, this_name)
+        elif len(this_name) < 8:
+            logging.warning("name too short, padding with spaces")
+            this_name = this_name.ljust(8, " ")
+
+        return BattleRecordStruct(this_name.encode("ascii"), wins, losses, draws)