EESAST
/
THUAI6
Not watched
1
0
Fork 0
Code
Releases 3 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 0572c63a90
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '0572c63a90'
${ noResults }
THUAI6/CAPI/python/PyAPI/utils.py

432 lines
18 kB
Raw Blame History 

  1. import proto.MessageType_pb2 as MessageType

  2. import proto.Message2Server_pb2 as Message2Server

  3. import proto.Message2Clients_pb2 as Message2Clients

  4. import PyAPI.structures as THUAI6

  5. from typing import Final, List, Union

  6. 

  7. numOfGridPerCell: Final[int] = 1000

  8. 

  9. # 起到NameSpace的作用

  10. 

  11. 

  12. class NoInstance:

  13.     def __call__(self):

  14.         raise TypeError("This class cannot be instantiated.")

  15. 

  16. 

  17. class AssistFunction(NoInstance):

  18.     @staticmethod

  19.     def CellToGrid(cell: int) -> int:

  20.         return cell * numOfGridPerCell + numOfGridPerCell // 2

  21. 

  22.     @staticmethod

  23.     def GridToCell(grid: int) -> int:

  24.         return grid // numOfGridPerCell

  25. 

  26.     @staticmethod

  27.     def HaveView(

  28.         viewRange: int,

  29.         x: int,

  30.         y: int,

  31.         newX: int,

  32.         newY: int,

  33.         map: List[List[THUAI6.PlaceType]],

  34.     ) -> bool:

  35.         deltaX: int = newX - x

  36.         deltaY: int = newY - y

  37.         distance: float = deltaX**2 + deltaY**2

  38.         myPlace = map[AssistFunction.GridToCell(x)][AssistFunction.GridToCell(y)]

  39.         newPlace = map[AssistFunction.GridToCell(newX)][AssistFunction.GridToCell(newY)]

  40.         if myPlace != THUAI6.PlaceType.Grass and newPlace == THUAI6.PlaceType.Grass:

  41.             return False

  42.         if distance <= viewRange * viewRange:

  43.             divide: int = max(abs(deltaX), abs(deltaY)) // 100

  44.             if divide == 0:

  45.                 return True

  46.             dx: float = deltaX / divide

  47.             dy: float = deltaY / divide

  48.             selfX: float = float(x)

  49.             selfY: float = float(y)

  50.             if newPlace == THUAI6.PlaceType.Grass and myPlace == THUAI6.PlaceType.Grass:

  51.                 for i in range(divide):

  52.                     selfX += dx

  53.                     selfY += dy

  54.                     if (

  55.                         map[AssistFunction.GridToCell(int(selfX))][

  56.                             AssistFunction.GridToCell(int(selfY))

  57.                         ]

  58.                         != THUAI6.PlaceType.Grass

  59.                     ):

  60.                         return False

  61.                 else:

  62.                     return True

  63.             else:

  64.                 for i in range(divide):

  65.                     selfX += dx

  66.                     selfY += dy

  67.                     if (

  68.                         map[AssistFunction.GridToCell(int(selfX))][

  69.                             AssistFunction.GridToCell(int(selfY))

  70.                         ]

  71.                         == THUAI6.PlaceType.Wall

  72.                     ):

  73.                         return False

  74.                 else:

  75.                     return True

  76.         else:

  77.             return False

  78. 

  79. 

  80. class Proto2THUAI6(NoInstance):

  81.     placeTypeDict: Final[dict] = {

  82.         MessageType.NULL_PLACE_TYPE: THUAI6.PlaceType.NullPlaceType,

  83.         MessageType.WALL: THUAI6.PlaceType.Wall,

  84.         MessageType.LAND: THUAI6.PlaceType.Land,

  85.         MessageType.GRASS: THUAI6.PlaceType.Grass,

  86.         MessageType.CLASSROOM: THUAI6.PlaceType.ClassRoom,

  87.         MessageType.GATE: THUAI6.PlaceType.Gate,

  88.         MessageType.HIDDEN_GATE: THUAI6.PlaceType.HiddenGate,

  89.         MessageType.WINDOW: THUAI6.PlaceType.Window,

  90.         MessageType.DOOR3: THUAI6.PlaceType.Door3,

  91.         MessageType.DOOR5: THUAI6.PlaceType.Door5,

  92.         MessageType.DOOR6: THUAI6.PlaceType.Door6,

  93.         MessageType.CHEST: THUAI6.PlaceType.Chest,

  94.     }

  95. 

  96.     shapeTypeDict: Final[dict] = {

  97.         MessageType.NULL_SHAPE_TYPE: THUAI6.ShapeType.NullShapeType,

  98.         MessageType.SQUARE: THUAI6.ShapeType.Square,

  99.         MessageType.CIRCLE: THUAI6.ShapeType.Circle,

  100.     }

  101. 

  102.     propTypeDict: Final[dict] = {

  103.         MessageType.NULL_PROP_TYPE: THUAI6.PropType.NullPropType,

  104.         MessageType.KEY3: THUAI6.PropType.Key3,

  105.         MessageType.KEY5: THUAI6.PropType.Key5,

  106.         MessageType.KEY6: THUAI6.PropType.Key6,

  107.         MessageType.ADD_SPEED: THUAI6.PropType.AddSpeed,

  108.         MessageType.ADD_HP_OR_AP: THUAI6.PropType.AddHpOrAp,

  109.         MessageType.ADD_LIFE_OR_CLAIRAUDIENCE: THUAI6.PropType.AddLifeOrClairaudience,

  110.         MessageType.SHIELD_OR_SPEAR: THUAI6.PropType.ShieldOrSpear,

  111.         MessageType.RECOVERY_FROM_DIZZINESS: THUAI6.PropType.RecoveryFromDizziness,

  112.         MessageType.CRAFTING_BENCH: THUAI6.PropType.CraftingBench,

  113.     }

  114. 

  115.     playerTypeDict: Final[dict] = {

  116.         MessageType.NULL_PLAYER_TYPE: THUAI6.PlayerType.NullPlayerType,

  117.         MessageType.STUDENT_PLAYER: THUAI6.PlayerType.StudentPlayer,

  118.         MessageType.TRICKER_PLAYER: THUAI6.PlayerType.TrickerPlayer,

  119.     }

  120. 

  121.     studentTypeDict: Final[dict] = {

  122.         MessageType.NULL_STUDENT_TYPE: THUAI6.StudentType.NullStudentType,

  123.         MessageType.ATHLETE: THUAI6.StudentType.Athlete,

  124.         MessageType.TEACHER: THUAI6.StudentType.Teacher,

  125.         MessageType.STRAIGHT_A_STUDENT: THUAI6.StudentType.StraightAStudent,

  126.         MessageType.ROBOT: THUAI6.StudentType.Robot,

  127.         MessageType.TECH_OTAKU: THUAI6.StudentType.TechOtaku,

  128.         MessageType.SUNSHINE: THUAI6.StudentType.Sunshine,

  129.     }

  130. 

  131.     trickerTypeDict: Final[dict] = {

  132.         MessageType.NULL_TRICKER_TYPE: THUAI6.TrickerType.NullTrickerType,

  133.         MessageType.ASSASSIN: THUAI6.TrickerType.Assassin,

  134.         MessageType.KLEE: THUAI6.TrickerType.Klee,

  135.         MessageType.A_NOISY_PERSON: THUAI6.TrickerType.ANoisyPerson,

  136.         MessageType.IDOL: THUAI6.TrickerType.Idol,

  137.     }

  138. 

  139.     studentBuffTypeDict: Final[dict] = {

  140.         MessageType.NULL_SBUFF_TYPE: THUAI6.StudentBuffType.NullStudentBuffType,

  141.         MessageType.STUDENT_ADD_SPEED: THUAI6.StudentBuffType.AddSpeed,

  142.         MessageType.ADD_LIFE: THUAI6.StudentBuffType.AddLife,

  143.         MessageType.SHIELD: THUAI6.StudentBuffType.Shield,

  144.         MessageType.STUDENT_INVISIBLE: THUAI6.StudentBuffType.Invisible,

  145.     }

  146. 

  147.     trickerBuffTypeDict: Final[dict] = {

  148.         MessageType.NULL_TBUFF_TYPE: THUAI6.TrickerBuffType.NullTrickerBuffType,

  149.         MessageType.TRICKER_ADD_SPEED: THUAI6.TrickerBuffType.AddSpeed,

  150.         MessageType.SPEAR: THUAI6.TrickerBuffType.Spear,

  151.         MessageType.ADD_AP: THUAI6.TrickerBuffType.AddAp,

  152.         MessageType.CLAIRAUDIENCE: THUAI6.TrickerBuffType.Clairaudience,

  153.         MessageType.TRICKER_INVISIBLE: THUAI6.TrickerBuffType.Invisible,

  154.     }

  155. 

  156.     playerStateDict: Final[dict] = {

  157.         MessageType.NULL_STATUS: THUAI6.PlayerState.NullState,

  158.         MessageType.IDLE: THUAI6.PlayerState.Idle,

  159.         MessageType.LEARNING: THUAI6.PlayerState.Learning,

  160.         MessageType.ADDICTED: THUAI6.PlayerState.Addicted,

  161.         MessageType.QUIT: THUAI6.PlayerState.Quit,

  162.         MessageType.GRADUATED: THUAI6.PlayerState.Graduated,

  163.         MessageType.RESCUED: THUAI6.PlayerState.Roused,

  164.         MessageType.TREATED: THUAI6.PlayerState.Encouraged,

  165.         MessageType.STUNNED: THUAI6.PlayerState.Stunned,

  166.         MessageType.RESCUING: THUAI6.PlayerState.Rousing,

  167.         MessageType.TREATING: THUAI6.PlayerState.Encouraging,

  168.         MessageType.SWINGING: THUAI6.PlayerState.Swinging,

  169.         MessageType.ATTACKING: THUAI6.PlayerState.Attacking,

  170.         MessageType.LOCKING: THUAI6.PlayerState.Locking,

  171.         # MessageType.RUMMAGING: THUAI6.PlayerState.Rummaging,

  172.         MessageType.CLIMBING: THUAI6.PlayerState.Climbing,

  173.         MessageType.OPENING_A_CHEST: THUAI6.PlayerState.OpeningAChest,

  174.         MessageType.USING_SPECIAL_SKILL: THUAI6.PlayerState.UsingSpecialSkill,

  175.         MessageType.OPENING_A_GATE: THUAI6.PlayerState.OpeningAGate,

  176.     }

  177. 

  178.     gameStateDict: Final[dict] = {

  179.         MessageType.NULL_GAME_STATE: THUAI6.GameState.NullGameState,

  180.         MessageType.GAME_START: THUAI6.GameState.GameStart,

  181.         MessageType.GAME_RUNNING: THUAI6.GameState.GameRunning,

  182.         MessageType.GAME_END: THUAI6.GameState.GameEnd,

  183.     }

  184. 

  185.     bulletTypeDict: Final[dict] = {

  186.         MessageType.NULL_BULLET_TYPE: THUAI6.BulletType.NullBulletType,

  187.         MessageType.FLYING_KNIFE: THUAI6.BulletType.FlyingKnife,

  188.         MessageType.BOMB_BOMB: THUAI6.BulletType.BombBomb,

  189.         MessageType.COMMON_ATTACK_OF_TRICKER: THUAI6.BulletType.CommonAttackOfTricker,

  190.         MessageType.JUMPY_DUMPTY: THUAI6.BulletType.JumpyDumpty,

  191.         MessageType.STRIKE: THUAI6.BulletType.Strike,

  192.     }

  193. 

  194.     # 用于将Proto的对象转为THUAI6的对象

  195.     @staticmethod

  196.     def Protobuf2THUAI6Tricker(

  197.         trickerMsg: Message2Clients.MessageOfTricker,

  198.     ) -> THUAI6.Tricker:

  199.         tricker = THUAI6.Tricker()

  200.         tricker.x = trickerMsg.x

  201.         tricker.y = trickerMsg.y

  202.         tricker.speed = trickerMsg.speed

  203.         tricker.score = trickerMsg.score

  204.         tricker.facingDirection = trickerMsg.facing_direction

  205.         tricker.trickDesire = trickerMsg.trick_desire

  206.         tricker.classVolume = trickerMsg.class_volume

  207.         tricker.bulletType = Proto2THUAI6.bulletTypeDict[trickerMsg.bullet_type]

  208.         for time in trickerMsg.time_until_skill_available:

  209.             tricker.timeUntilSkillAvailable.append(time)

  210.         tricker.playerState = Proto2THUAI6.playerStateDict[trickerMsg.player_state]

  211.         for item in trickerMsg.prop:

  212.             tricker.prop.append(Proto2THUAI6.propTypeDict[item])

  213.         tricker.trickerType = Proto2THUAI6.trickerTypeDict[trickerMsg.tricker_type]

  214.         tricker.guid = trickerMsg.guid

  215.         tricker.playerID = trickerMsg.player_id

  216.         tricker.viewRange = trickerMsg.view_range

  217.         tricker.radius = trickerMsg.radius

  218.         for buff in trickerMsg.buff:

  219.             tricker.buff.append(Proto2THUAI6.trickerBuffTypeDict[buff])

  220.         tricker.playerType = THUAI6.PlayerType.TrickerPlayer

  221.         return tricker

  222. 

  223.     @staticmethod

  224.     def Protobuf2THUAI6Student(

  225.         studentMsg: Message2Clients.MessageOfStudent,

  226.     ) -> THUAI6.Student:

  227.         student = THUAI6.Student()

  228.         student.x = studentMsg.x

  229.         student.y = studentMsg.y

  230.         student.speed = studentMsg.speed

  231.         student.determination = studentMsg.determination

  232.         student.addiction = studentMsg.addiction

  233.         student.score = studentMsg.score

  234.         student.facingDirection = studentMsg.facing_direction

  235.         student.bulletType = Proto2THUAI6.bulletTypeDict[studentMsg.bullet_type]

  236.         student.learningSpeed = studentMsg.learning_speed

  237.         student.encourageSpeed = studentMsg.treat_speed

  238.         student.encourageProgress = studentMsg.treat_progress

  239.         student.rouseProgress = studentMsg.rescue_progress

  240.         student.dangerAlert = studentMsg.danger_alert

  241.         for time in studentMsg.time_until_skill_available:

  242.             student.timeUntilSkillAvailable.append(time)

  243.         for item in studentMsg.prop:

  244.             student.prop.append(Proto2THUAI6.propTypeDict[item])

  245.         student.studentType = Proto2THUAI6.studentTypeDict[studentMsg.student_type]

  246.         student.guid = studentMsg.guid

  247.         student.playerState = Proto2THUAI6.playerStateDict[studentMsg.player_state]

  248.         student.playerID = studentMsg.player_id

  249.         student.viewRange = studentMsg.view_range

  250.         student.radius = studentMsg.radius

  251.         for buff in studentMsg.buff:

  252.             student.buff.append(Proto2THUAI6.studentBuffTypeDict[buff])

  253.         student.playerType = THUAI6.PlayerType.StudentPlayer

  254.         return student

  255. 

  256.     @staticmethod

  257.     def Protobuf2THUAI6Prop(propMsg: Message2Clients.MessageOfProp) -> THUAI6.Prop:

  258.         prop = THUAI6.Prop()

  259.         prop.x = propMsg.x

  260.         prop.y = propMsg.y

  261.         prop.type = Proto2THUAI6.propTypeDict[propMsg.type]

  262.         prop.guid = propMsg.guid

  263.         prop.facingDirection = propMsg.facing_direction

  264.         return prop

  265. 

  266.     @staticmethod

  267.     def Protobuf2THUAI6GameInfo(allMsg: Message2Clients.MessageOfAll):

  268.         gameInfo = THUAI6.GameInfo()

  269.         gameInfo.gameTime = allMsg.game_time

  270.         gameInfo.subjectFinished = allMsg.subject_finished

  271.         gameInfo.studentGraduated = allMsg.student_graduated

  272.         gameInfo.studentQuited = allMsg.student_quited

  273.         gameInfo.studentScore = allMsg.student_score

  274.         gameInfo.trickerScore = allMsg.tricker_score

  275.         return gameInfo

  276. 

  277.     @staticmethod

  278.     def Protobuf2THUAI6Bullet(

  279.         bulletMsg: Message2Clients.MessageOfBullet,

  280.     ) -> THUAI6.Bullet:

  281.         bullet = THUAI6.Bullet()

  282.         bullet.x = bulletMsg.x

  283.         bullet.y = bulletMsg.y

  284.         bullet.bulletType = Proto2THUAI6.bulletTypeDict[bulletMsg.type]

  285.         bullet.facingDirection = bulletMsg.facing_direction

  286.         bullet.guid = bulletMsg.guid

  287.         bullet.team = Proto2THUAI6.playerTypeDict[bulletMsg.team]

  288.         bullet.bombRange = bulletMsg.bomb_range

  289.         return bullet

  290. 

  291.     @staticmethod

  292.     def Protobuf2THUAI6BombedBullet(

  293.         bulletMsg: Message2Clients.MessageOfBombedBullet,

  294.     ) -> THUAI6.BombedBullet:

  295.         bullet = THUAI6.BombedBullet()

  296.         bullet.x = bulletMsg.x

  297.         bullet.y = bulletMsg.y

  298.         bullet.bulletType = Proto2THUAI6.bulletTypeDict[bulletMsg.type]

  299.         bullet.facingDirection = bulletMsg.facing_direction

  300.         bullet.mappingID = bulletMsg.mapping_id

  301.         bullet.bombRange = bulletMsg.bomb_range

  302.         return bullet

  303. 

  304.     @staticmethod

  305.     def Bool2HiddenGateState(gateMsg: bool) -> THUAI6.HiddenGateState:

  306.         if gateMsg:

  307.             return THUAI6.HiddenGateState.Opened

  308.         else:

  309.             return THUAI6.HiddenGateState.Refreshed

  310. 

  311. 

  312. class THUAI62Proto(NoInstance):

  313.     placeTypeDict: Final[dict] = {

  314.         THUAI6.PlaceType.NullPlaceType: MessageType.NULL_PLACE_TYPE,

  315.         THUAI6.PlaceType.Wall: MessageType.WALL,

  316.         THUAI6.PlaceType.Land: MessageType.LAND,

  317.         THUAI6.PlaceType.Grass: MessageType.GRASS,

  318.         THUAI6.PlaceType.ClassRoom: MessageType.CLASSROOM,

  319.         THUAI6.PlaceType.Gate: MessageType.GATE,

  320.         THUAI6.PlaceType.HiddenGate: MessageType.HIDDEN_GATE,

  321.         THUAI6.PlaceType.Door3: MessageType.DOOR3,

  322.         THUAI6.PlaceType.Door5: MessageType.DOOR5,

  323.         THUAI6.PlaceType.Door6: MessageType.DOOR6,

  324.         THUAI6.PlaceType.Chest: MessageType.CHEST,

  325.         THUAI6.PlaceType.Window: MessageType.WINDOW,

  326.     }

  327. 

  328.     playerTypeDict: Final[dict] = {

  329.         THUAI6.PlayerType.NullPlayerType: MessageType.NULL_PLAYER_TYPE,

  330.         THUAI6.PlayerType.StudentPlayer: MessageType.STUDENT_PLAYER,

  331.         THUAI6.PlayerType.TrickerPlayer: MessageType.TRICKER_PLAYER,

  332.     }

  333. 

  334.     studentTypeDict: Final[dict] = {

  335.         THUAI6.StudentType.NullStudentType: MessageType.NULL_STUDENT_TYPE,

  336.         THUAI6.StudentType.Athlete: MessageType.ATHLETE,

  337.         THUAI6.StudentType.Teacher: MessageType.TEACHER,

  338.         THUAI6.StudentType.StraightAStudent: MessageType.STRAIGHT_A_STUDENT,

  339.         THUAI6.StudentType.Robot: MessageType.ROBOT,

  340.         THUAI6.StudentType.TechOtaku: MessageType.TECH_OTAKU,

  341.         THUAI6.StudentType.Sunshine: MessageType.SUNSHINE,

  342.     }

  343. 

  344.     trickerTypeDict: Final[dict] = {

  345.         THUAI6.TrickerType.NullTrickerType: MessageType.NULL_TRICKER_TYPE,

  346.         THUAI6.TrickerType.Assassin: MessageType.ASSASSIN,

  347.         THUAI6.TrickerType.Klee: MessageType.KLEE,

  348.         THUAI6.TrickerType.ANoisyPerson: MessageType.A_NOISY_PERSON,

  349.         THUAI6.TrickerType.Idol: MessageType.IDOL,

  350.     }

  351. 

  352.     propTypeDict: Final[dict] = {

  353.         THUAI6.PropType.NullPropType: MessageType.NULL_PROP_TYPE,

  354.         THUAI6.PropType.Key3: MessageType.KEY3,

  355.         THUAI6.PropType.Key5: MessageType.KEY5,

  356.         THUAI6.PropType.Key6: MessageType.KEY6,

  357.         THUAI6.PropType.AddHpOrAp: MessageType.ADD_HP_OR_AP,

  358.         THUAI6.PropType.AddLifeOrClairaudience: MessageType.ADD_LIFE_OR_CLAIRAUDIENCE,

  359.         THUAI6.PropType.AddSpeed: MessageType.ADD_SPEED,

  360.         THUAI6.PropType.ShieldOrSpear: MessageType.SHIELD_OR_SPEAR,

  361.         THUAI6.PropType.CraftingBench: MessageType.CRAFTING_BENCH,

  362.     }

  363. 

  364.     # 用于将THUAI6的对象转为Proto的对象

  365. 

  366.     @staticmethod

  367.     def THUAI62ProtobufPlayer(

  368.         playerID: int,

  369.         playerType: THUAI6.PlayerType,

  370.         studentType: THUAI6.StudentType,

  371.         trickerType: THUAI6.TrickerType,

  372.     ) -> Message2Server.PlayerMsg:

  373.         if playerType == THUAI6.PlayerType.StudentPlayer:

  374.             return Message2Server.PlayerMsg(

  375.                 player_id=playerID,

  376.                 player_type=MessageType.STUDENT_PLAYER,

  377.                 student_type=THUAI62Proto.studentTypeDict[studentType],

  378.             )

  379.         else:

  380.             return Message2Server.PlayerMsg(

  381.                 player_id=playerID,

  382.                 player_type=MessageType.TRICKER_PLAYER,

  383.                 tricker_type=THUAI62Proto.trickerTypeDict[trickerType],

  384.             )

  385. 

  386.     @staticmethod

  387.     def THUAI62ProtobufID(playerID: int) -> Message2Server.IDMsg:

  388.         return Message2Server.IDMsg(player_id=playerID)

  389. 

  390.     @staticmethod

  391.     def THUAI62ProtobufMove(time: int, angle: float, id: int) -> Message2Server.MoveMsg:

  392.         return Message2Server.MoveMsg(

  393.             player_id=id, angle=angle, time_in_milliseconds=time

  394.         )

  395. 

  396.     @staticmethod

  397.     def THUAI62ProtobufTreatAndRescue(

  398.         playerID: int, mateID: int

  399.     ) -> Message2Server.TreatAndRescueMsg:

  400.         return Message2Server.TreatAndRescueMsg(player_id=playerID, to_player_id=mateID)

  401. 

  402.     @staticmethod

  403.     def THUAI62ProtobufProp(prop: THUAI6.PropType, id: int) -> Message2Server.PropMsg:

  404.         return Message2Server.PropMsg(

  405.             player_id=id, prop_type=THUAI62Proto.propTypeDict[prop]

  406.         )

  407. 

  408.     @staticmethod

  409.     def THUAI62ProtobufSend(

  410.         msg: Union[str, bytes], toID: int, id: int

  411.     ) -> Message2Server.SendMsg:

  412.         if isinstance(msg, str):

  413.             return Message2Server.SendMsg(

  414.                 player_id=id, to_player_id=toID, text_message=msg

  415.             )

  416.         elif isinstance(msg, bytes):

  417.             return Message2Server.SendMsg(

  418.                 player_id=id, to_player_id=toID, binary_message=msg

  419.             )

  420. 

  421.     @staticmethod

  422.     def THUAI62ProtobufAttack(angle: float, id: int) -> Message2Server.AttackMsg:

  423.         return Message2Server.AttackMsg(player_id=id, angle=angle)

  424. 

  425.     @staticmethod

  426.     def THUAI62ProtobufSkill(

  427.         skillID: int, skillParam: int, id: int

  428.     ) -> Message2Server.SkillMsg:

  429.         return Message2Server.SkillMsg(

  430.             player_id=id, skill_id=skillID, skill_param=skillParam

  431.         )