109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
|
from typing import Dict, Optional, Any
|
||
|
|
|
||
|
|
from bemani.backend.base import Base, Factory
|
||
|
|
from bemani.backend.ddr.stubs import (
|
||
|
|
DDRX,
|
||
|
|
DDRSuperNova2,
|
||
|
|
DDRSuperNova,
|
||
|
|
DDRExtreme,
|
||
|
|
DDR7thMix,
|
||
|
|
DDR6thMix,
|
||
|
|
DDR5thMix,
|
||
|
|
DDR4thMix,
|
||
|
|
DDR3rdMix,
|
||
|
|
DDR2ndMix,
|
||
|
|
DDR1stMix,
|
||
|
|
)
|
||
|
|
from bemani.backend.ddr.ddrx2 import DDRX2
|
||
|
|
from bemani.backend.ddr.ddrx3 import DDRX3
|
||
|
|
from bemani.backend.ddr.ddr2013 import DDR2013
|
||
|
|
from bemani.backend.ddr.ddr2014 import DDR2014
|
||
|
|
from bemani.backend.ddr.ddrace import DDRAce
|
||
|
|
from bemani.backend.ddr.ddra20 import DDRA20
|
||
|
|
from bemani.common import Model, VersionConstants
|
||
|
|
from bemani.data import Data
|
||
|
|
|
||
|
|
|
||
|
|
class DDRFactory(Factory):
|
||
|
|
|
||
|
|
MANAGED_CLASSES = [
|
||
|
|
DDR1stMix,
|
||
|
|
DDR2ndMix,
|
||
|
|
DDR4thMix,
|
||
|
|
DDR3rdMix,
|
||
|
|
DDR5thMix,
|
||
|
|
DDR6thMix,
|
||
|
|
DDR7thMix,
|
||
|
|
DDRExtreme,
|
||
|
|
DDRSuperNova,
|
||
|
|
DDRSuperNova2,
|
||
|
|
DDRX,
|
||
|
|
DDRX2,
|
||
|
|
DDRX3,
|
||
|
|
DDR2013,
|
||
|
|
DDR2014,
|
||
|
|
DDRAce,
|
||
|
|
DDRA20,
|
||
|
|
]
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def register_all(cls) -> None:
|
||
|
|
for game in ['HDX', 'JDX', 'KDX', 'MDX']:
|
||
|
|
Base.register(game, DDRFactory)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def create(cls, data: Data, config: Dict[str, Any], model: Model, parentmodel: Optional[Model]=None) -> Optional[Base]:
|
||
|
|
|
||
|
|
def version_from_date(date: int) -> Optional[int]:
|
||
|
|
if date < 2014051200:
|
||
|
|
return VersionConstants.DDR_2013
|
||
|
|
elif date >= 2014051200 and date < 2016033000:
|
||
|
|
return VersionConstants.DDR_2014
|
||
|
|
elif date >= 2016033000 and date < 2019042300:
|
||
|
|
return VersionConstants.DDR_ACE
|
||
|
|
elif date >= 2019042300:
|
||
|
|
return VersionConstants.DDR_A20
|
||
|
|
return None
|
||
|
|
|
||
|
|
if model.game == 'HDX':
|
||
|
|
return DDRX(data, config, model)
|
||
|
|
if model.game == 'JDX':
|
||
|
|
return DDRX2(data, config, model)
|
||
|
|
if model.game == 'KDX':
|
||
|
|
return DDRX3(data, config, model)
|
||
|
|
if model.game == 'MDX':
|
||
|
|
if model.version is None:
|
||
|
|
if parentmodel is None:
|
||
|
|
return None
|
||
|
|
|
||
|
|
# We have no way to tell apart newer versions. However, we can make
|
||
|
|
# an educated guess if we happen to be summoned for old profile lookup.
|
||
|
|
if parentmodel.game not in ['HDX', 'JDX', 'KDX', 'MDX']:
|
||
|
|
return None
|
||
|
|
|
||
|
|
parentversion = version_from_date(parentmodel.version)
|
||
|
|
if parentversion == VersionConstants.DDR_A20:
|
||
|
|
return DDRAce(data, config, model)
|
||
|
|
if parentversion == VersionConstants.DDR_ACE:
|
||
|
|
return DDR2014(data, config, model)
|
||
|
|
if parentversion == VersionConstants.DDR_2014:
|
||
|
|
return DDR2013(data, config, model)
|
||
|
|
if parentversion == VersionConstants.DDR_2013:
|
||
|
|
return DDRX3(data, config, model)
|
||
|
|
|
||
|
|
# Unknown older version
|
||
|
|
return None
|
||
|
|
|
||
|
|
version = version_from_date(model.version)
|
||
|
|
if version == VersionConstants.DDR_2013:
|
||
|
|
return DDR2013(data, config, model)
|
||
|
|
if version == VersionConstants.DDR_2014:
|
||
|
|
return DDR2014(data, config, model)
|
||
|
|
if version == VersionConstants.DDR_ACE:
|
||
|
|
return DDRAce(data, config, model)
|
||
|
|
if version == VersionConstants.DDR_A20:
|
||
|
|
return DDRA20(data, config, model)
|
||
|
|
|
||
|
|
# Unknown game version
|
||
|
|
return None
|