THUAI6/CAPI/python/PyAPI/structures.py

from enum import Enum
from typing import List, Dict
import sys

if sys.version_info < (3, 9):
    from typing import Tuple
else:
    Tuple = tuple


class GameState(Enum):
    NullGameState = 0
    GameStart = 1
    GameRunning = 2
    GameEnd = 3


class PlaceType(Enum):
    NullPlaceType = 0
    Land = 1
    Wall = 2
    Grass = 3
    ClassRoom = 4
    Gate = 5
    HiddenGate = 6
    Window = 7
    Door3 = 8
    Door5 = 9
    Door6 = 10
    Chest = 11


class ShapeType(Enum):
    NullShapeType = 0
    Square = 1
    Circle = 2


class PlayerType(Enum):
    NullPlayerType = 0
    StudentPlayer = 1
    TrickerPlayer = 2


class PropType(Enum):
    NullPropType = 0
    Key3 = 1
    Key5 = 2
    Key6 = 3
    AddSpeed = 4
    AddLifeOrClairaudience = 5
    AddHpOrAp = 6
    ShieldOrSpear = 7
    RecoveryFromDizziness = 8


class BulletType(Enum):
    NullBulletType = 0
    FlyingKnife = 1
    CommonAttackOfTricker = 2
    BombBomb = 3
    JumpyDumpty = 4
    AtomBomb = 5


class StudentType(Enum):
    NullStudentType = 0
    Athlete = 1
    Teacher = 2
    StraightAStudent = 3
    Robot = 4
    TechOtaku = 5
    Sunshine = 6


class TrickerType(Enum):
    NullTrickerType = 0
    Assassin = 1
    Klee = 2
    ANoisyPerson = 3
    Idol = 4


class StudentBuffType(Enum):
    NullStudentBuffType = 0
    AddSpeed = 1
    AddLife = 2
    Shield = 3
    Invisible = 4


class TrickerBuffType(Enum):
    NullTrickerBuffType = 0
    AddSpeed = 1
    Spear = 2
    AddAp = 3
    Clairaudience = 4
    Invisible = 5


class PlayerState(Enum):
    NullState = 0
    Idle = 1
    Learning = 2
    Addicted = 3
    Quit = 4
    Graduated = 5
    Encouraged = 6
    Roused = 7
    Stunned = 8
    Encouraging = 9
    Rousing = 10
    Swinging = 11
    Attacking = 12
    Locking = 13
    # Rummaging = 14
    Climbing = 15
    OpeningAChest = 16
    UsingSpecialSkill = 17
    OpeningAGate = 18


class MessageOfObj(Enum):
    NullMessageOfObj = 0
    StudentMessage = 1
    TrickerMessage = 2
    PropMessage = 3
    BulletMessage = 4
    BombedBulletMessage = 5
    ClassroomMessage = 6
    GateMessage = 7
    ChestMessage = 8
    DoorMessage = 9
    MapMessage = 10
    NewsMessage = 11
    HiddenGateMessage = 12


class HiddenGateState(Enum):
    Null = 0
    Refreshed = 1
    Opened = 2


class Player:
    def __init__(self, **kwargs) -> None:
        self.x = 0
        self.y = 0
        self.speed = 0
        self.viewRange = 0
        self.playerID = 0
        self.guid = 0
        self.radius = 0
        self.score = 0
        self.facingDirection = 0.0
        self.timeUntilSkillAvailable = []
        self.playerType = PlayerType.NullPlayerType
        self.prop = []
        self.place = PlaceType.NullPlaceType
        self.bulletType = BulletType.NullBulletType
        self.playerState = PlayerState.NullState
    x: int
    y: int
    speed: int
    viewRange: int
    playerID: int
    guid: int
    radius: int
    score: int
    facingDirection: float
    timeUntilSkillAvailable: List[float]
    playerType: PlayerType
    prop: List[PropType]
    place: PlaceType
    bulletType: BulletType
    playerState: PlayerState


class Student(Player):
    def __init__(self, **kwargs) -> None:
        super().__init__()
        self.studentType = StudentType.NullStudentType
        self.determination = 0
        self.addiction = 0
        self.encourageProgress = 0
        self.rouseProgress = 0
        self.learningSpeed = 0
        self.encourageSpeed = 0
        self.dangerAlert = 0.0
        self.buff = []
    studentType: StudentType
    determination: int
    addiction: int
    encourageProgress: int
    rouseProgress: int
    learningSpeed: int
    encourageSpeed: int
    dangerAlert: float
    buff: List[StudentBuffType]


class Tricker(Player):
    def __init__(self, **kwargs) -> None:
        super().__init__()
        self.trickerType = TrickerType.NullTrickerType
        self.trickDesire = 0.0
        self.classVolume = 0.0
        self.buff = []
    trickerType: TrickerType
    trickDesire: float
    classVolume: float
    buff: List[TrickerBuffType]


class Prop:
    def __init__(self, **kwargs) -> None:
        self.x = 0
        self.y = 0
        self.guid = 0
        self.type = PropType.NullPropType
        self.place = PlaceType.NullPlaceType
        self.facingDirection = 0.0
    x: int
    y: int
    guid: int
    type: PropType
    place: PlaceType
    facingDirection: float


class Bullet:
    def __init__(self, **kwargs) -> None:
        self.bulletType = BulletType.NullBulletType
        self.x = 0
        self.y = 0
        self.facingDirection = 0.0
        self.guid = 0
        self.team = PlayerType.NullPlayerType
        self.place = PlaceType.NullPlaceType
        self.bombRange = 0.0
        self.speed = 0
    bulletType: BulletType
    x: int
    y: int
    facingDirection: float
    guid: int
    team: PlayerType
    place: PlaceType
    bombRange: float
    speed: int


class BombedBullet:
    def __init__(self, **kwargs) -> None:
        self.bulletType = BulletType.NullBulletType
        self.x = 0
        self.y = 0
        self.facingDirection = 0.0
        self.mappingID = 0
        self.bombRange = 0.0
    bulletType: BulletType
    x: int
    y: int
    facingDirection: float
    mappingID: int
    bombRange: float


class GameMap:
    def __init__(self, **kwargs) -> None:
        self.classroomState = {}
        self.gateState = {}
        self.chestState = {}
        self.doorState = {}
        self.doorProgress = {}
        self.hiddenGateState = {}
    classroomState: Dict[Tuple[int, int], int]
    gateState: Dict[Tuple[int, int], int]
    chestState: Dict[Tuple[int, int], int]
    doorState: Dict[Tuple[int, int], bool]
    doorProgress: Dict[Tuple[int, int], int]
    hiddenGateState: Dict[Tuple[int, int], HiddenGateState]


class GameInfo:
    def __init__(self, **kwargs) -> None:
        self.gameTime = 0
        self.subjectFinished = 0
        self.studentGraduated = 0
        self.studentQuited = 0
        self.studentScore = 0
        self.trickerScore = 0
    gameTime: int
    subjectFinished: int
    studentGraduated: int
    studentQuited: int
    studentScore: int
    trickerScore: int