From ef1504d70e6ed05a3eefacac60a945951fe4478f Mon Sep 17 00:00:00 2001 From: shima004 Date: Wed, 2 Oct 2024 13:17:14 +0900 Subject: [PATCH 1/5] feat: Add typings for rcrs_core entities, connection, commands, and messages --- poetry.lock | 2 +- typings/rcrs_core/agents/agent.pyi | 94 ++++++++++++ typings/rcrs_core/commands/AKClear.pyi | 16 ++ typings/rcrs_core/commands/AKClearArea.pyi | 16 ++ typings/rcrs_core/commands/AKLoad.pyi | 16 ++ typings/rcrs_core/commands/AKMove.pyi | 17 +++ typings/rcrs_core/commands/AKRescue.pyi | 16 ++ typings/rcrs_core/commands/AKRest.pyi | 16 ++ typings/rcrs_core/commands/AKSay.pyi | 16 ++ typings/rcrs_core/commands/AKSpeak.pyi | 16 ++ typings/rcrs_core/commands/AKSubscribe.pyi | 17 +++ typings/rcrs_core/commands/AKTell.pyi | 16 ++ typings/rcrs_core/commands/AKUnload.pyi | 16 ++ typings/rcrs_core/commands/Command.pyi | 16 ++ typings/rcrs_core/config/config.pyi | 55 +++++++ .../rcrs_core/config/config_constraint.pyi | 25 ++++ .../rcrs_core/connection/RCRSProto_pb2.pyi | 42 ++++++ typings/rcrs_core/connection/URN.pyi | 141 ++++++++++++++++++ typings/rcrs_core/connection/__init__.pyi | 4 + .../connection/componentLauncher.pyi | 22 +++ typings/rcrs_core/connection/connection.pyi | 22 +++ .../connection/rcrs_encoding_utils.pyi | 20 +++ typings/rcrs_core/entities/__init__.pyi | 4 + .../rcrs_core/entities/ambulanceCenter.pyi | 22 +++ typings/rcrs_core/entities/ambulanceTeam.pyi | 22 +++ typings/rcrs_core/entities/area.pyi | 52 +++++++ typings/rcrs_core/entities/blockade.pyi | 73 +++++++++ typings/rcrs_core/entities/building.pyi | 124 +++++++++++++++ typings/rcrs_core/entities/civilian.pyi | 22 +++ typings/rcrs_core/entities/edge.pyi | 31 ++++ typings/rcrs_core/entities/entity.pyi | 61 ++++++++ typings/rcrs_core/entities/fireBrigade.pyi | 34 +++++ typings/rcrs_core/entities/fireStation.pyi | 22 +++ typings/rcrs_core/entities/gassStation.pyi | 22 +++ typings/rcrs_core/entities/human.pyi | 114 ++++++++++++++ typings/rcrs_core/entities/hydrant.pyi | 22 +++ typings/rcrs_core/entities/policeForce.pyi | 22 +++ typings/rcrs_core/entities/policeOffice.pyi | 22 +++ typings/rcrs_core/entities/refuge.pyi | 61 ++++++++ typings/rcrs_core/entities/road.pyi | 22 +++ .../entities/standardEntityFactory.pyi | 17 +++ typings/rcrs_core/log/logger.pyi | 28 ++++ typings/rcrs_core/messages/AKAcknowledge.pyi | 18 +++ typings/rcrs_core/messages/AKConnect.pyi | 18 +++ typings/rcrs_core/messages/KAConnectError.pyi | 18 +++ typings/rcrs_core/messages/KAConnectOK.pyi | 18 +++ typings/rcrs_core/messages/KASense.pyi | 19 +++ typings/rcrs_core/messages/Shutdown.pyi | 18 +++ .../messages/controlMessageFactory.pyi | 13 ++ typings/rcrs_core/messages/message.pyi | 23 +++ .../rcrs_core/properties/booleanProperty.pyi | 21 +++ .../rcrs_core/properties/edgeListProperty.pyi | 38 +++++ .../properties/entityIDListProperty.pyi | 26 ++++ .../rcrs_core/properties/entityIDProperty.pyi | 21 +++ .../rcrs_core/properties/intArrayProperty.pyi | 24 +++ typings/rcrs_core/properties/intProperty.pyi | 21 +++ typings/rcrs_core/properties/property.pyi | 22 +++ .../properties/standardPropertyFactory.pyi | 13 ++ typings/rcrs_core/worldmodel/changeSet.pyi | 34 +++++ typings/rcrs_core/worldmodel/entityID.pyi | 25 ++++ typings/rcrs_core/worldmodel/worldmodel.pyi | 51 +++++++ 61 files changed, 1858 insertions(+), 1 deletion(-) create mode 100644 typings/rcrs_core/agents/agent.pyi create mode 100644 typings/rcrs_core/commands/AKClear.pyi create mode 100644 typings/rcrs_core/commands/AKClearArea.pyi create mode 100644 typings/rcrs_core/commands/AKLoad.pyi create mode 100644 typings/rcrs_core/commands/AKMove.pyi create mode 100644 typings/rcrs_core/commands/AKRescue.pyi create mode 100644 typings/rcrs_core/commands/AKRest.pyi create mode 100644 typings/rcrs_core/commands/AKSay.pyi create mode 100644 typings/rcrs_core/commands/AKSpeak.pyi create mode 100644 typings/rcrs_core/commands/AKSubscribe.pyi create mode 100644 typings/rcrs_core/commands/AKTell.pyi create mode 100644 typings/rcrs_core/commands/AKUnload.pyi create mode 100644 typings/rcrs_core/commands/Command.pyi create mode 100644 typings/rcrs_core/config/config.pyi create mode 100644 typings/rcrs_core/config/config_constraint.pyi create mode 100644 typings/rcrs_core/connection/RCRSProto_pb2.pyi create mode 100644 typings/rcrs_core/connection/URN.pyi create mode 100644 typings/rcrs_core/connection/__init__.pyi create mode 100644 typings/rcrs_core/connection/componentLauncher.pyi create mode 100644 typings/rcrs_core/connection/connection.pyi create mode 100644 typings/rcrs_core/connection/rcrs_encoding_utils.pyi create mode 100644 typings/rcrs_core/entities/__init__.pyi create mode 100644 typings/rcrs_core/entities/ambulanceCenter.pyi create mode 100644 typings/rcrs_core/entities/ambulanceTeam.pyi create mode 100644 typings/rcrs_core/entities/area.pyi create mode 100644 typings/rcrs_core/entities/blockade.pyi create mode 100644 typings/rcrs_core/entities/building.pyi create mode 100644 typings/rcrs_core/entities/civilian.pyi create mode 100644 typings/rcrs_core/entities/edge.pyi create mode 100644 typings/rcrs_core/entities/entity.pyi create mode 100644 typings/rcrs_core/entities/fireBrigade.pyi create mode 100644 typings/rcrs_core/entities/fireStation.pyi create mode 100644 typings/rcrs_core/entities/gassStation.pyi create mode 100644 typings/rcrs_core/entities/human.pyi create mode 100644 typings/rcrs_core/entities/hydrant.pyi create mode 100644 typings/rcrs_core/entities/policeForce.pyi create mode 100644 typings/rcrs_core/entities/policeOffice.pyi create mode 100644 typings/rcrs_core/entities/refuge.pyi create mode 100644 typings/rcrs_core/entities/road.pyi create mode 100644 typings/rcrs_core/entities/standardEntityFactory.pyi create mode 100644 typings/rcrs_core/log/logger.pyi create mode 100644 typings/rcrs_core/messages/AKAcknowledge.pyi create mode 100644 typings/rcrs_core/messages/AKConnect.pyi create mode 100644 typings/rcrs_core/messages/KAConnectError.pyi create mode 100644 typings/rcrs_core/messages/KAConnectOK.pyi create mode 100644 typings/rcrs_core/messages/KASense.pyi create mode 100644 typings/rcrs_core/messages/Shutdown.pyi create mode 100644 typings/rcrs_core/messages/controlMessageFactory.pyi create mode 100644 typings/rcrs_core/messages/message.pyi create mode 100644 typings/rcrs_core/properties/booleanProperty.pyi create mode 100644 typings/rcrs_core/properties/edgeListProperty.pyi create mode 100644 typings/rcrs_core/properties/entityIDListProperty.pyi create mode 100644 typings/rcrs_core/properties/entityIDProperty.pyi create mode 100644 typings/rcrs_core/properties/intArrayProperty.pyi create mode 100644 typings/rcrs_core/properties/intProperty.pyi create mode 100644 typings/rcrs_core/properties/property.pyi create mode 100644 typings/rcrs_core/properties/standardPropertyFactory.pyi create mode 100644 typings/rcrs_core/worldmodel/changeSet.pyi create mode 100644 typings/rcrs_core/worldmodel/entityID.pyi create mode 100644 typings/rcrs_core/worldmodel/worldmodel.pyi diff --git a/poetry.lock b/poetry.lock index 2f0e6c9..d7a68ff 100644 --- a/poetry.lock +++ b/poetry.lock @@ -347,7 +347,7 @@ rtree = "*" type = "git" url = "https://github.com/adf-python/rcrs-core-python" reference = "HEAD" -resolved_reference = "7b0d4ef8d6c1205c00f9e197c5833a3939f482b2" +resolved_reference = "bd42a563fee818496ce4c020a7ec82c4b5c67042" [[package]] name = "rtree" diff --git a/typings/rcrs_core/agents/agent.pyi b/typings/rcrs_core/agents/agent.pyi new file mode 100644 index 0000000..e19674a --- /dev/null +++ b/typings/rcrs_core/agents/agent.pyi @@ -0,0 +1,94 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.connection.URN import Entity +from abc import ABC, abstractmethod +from rcrs_core.entities.human import Human + +class Agent(ABC): + def __init__(self, pre) -> None: + ... + + @abstractmethod + def precompute(self): # -> None: + ... + + @abstractmethod + def think(self, time, change_set, hear): # -> None: + ... + + def get_name(self): # -> str: + ... + + def get_id(self): # -> EntityID | None: + ... + + def set_send_msg(self, connection_send_func): # -> None: + ... + + def start_up(self, request_id): # -> None: + ... + + def post_connect(self): # -> None: + ... + + def message_received(self, msg): # -> None: + ... + + def handle_connect_error(self, msg): + ... + + def handle_connect_ok(self, msg): # -> None: + ... + + def sendAKAcknowledge(self, request_id): # -> None: + ... + + def process_sense(self, msg): # -> None: + ... + + def me(self) -> Human: + ... + + def location(self) -> Entity: + ... + + def random_walk(self): # -> list[Any]: + ... + + def send_clear(self, time, target): # -> None: + ... + + def send_clear_area(self, time, x=..., y=...): # -> None: + ... + + def send_load(self, time, target): # -> None: + ... + + def send_move(self, time, path, x=..., y=...): # -> None: + ... + + def send_rescue(self, time, target): # -> None: + ... + + def send_rest(self, time_step): # -> None: + ... + + def send_say(self, time_step: int, message: str): # -> None: + ... + + def send_speak(self, time_step: int, message: str, channel: int): # -> None: + ... + + def send_subscribe(self, time, channel): # -> None: + ... + + def send_tell(self, time_step: int, message: str): # -> None: + ... + + def send_unload(self, time): # -> None: + ... + + + diff --git a/typings/rcrs_core/commands/AKClear.pyi b/typings/rcrs_core/commands/AKClear.pyi new file mode 100644 index 0000000..1f3119a --- /dev/null +++ b/typings/rcrs_core/commands/AKClear.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKClear(Command): + def __init__(self, agent_id: EntityID, time: int, target: EntityID) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKClearArea.pyi b/typings/rcrs_core/commands/AKClearArea.pyi new file mode 100644 index 0000000..cde4e0d --- /dev/null +++ b/typings/rcrs_core/commands/AKClearArea.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKClearArea(Command): + def __init__(self, agent_id: EntityID, time: int, destinationX: int, destinationY: int) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKLoad.pyi b/typings/rcrs_core/commands/AKLoad.pyi new file mode 100644 index 0000000..1dda8d4 --- /dev/null +++ b/typings/rcrs_core/commands/AKLoad.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKLoad(Command): + def __init__(self, agent_id: EntityID, time: int, target: EntityID) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKMove.pyi b/typings/rcrs_core/commands/AKMove.pyi new file mode 100644 index 0000000..32f0c27 --- /dev/null +++ b/typings/rcrs_core/commands/AKMove.pyi @@ -0,0 +1,17 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID +from typing import List + +class AKMove(Command): + def __init__(self, agent_id: EntityID, time: int, path: List[int], destinationX=..., destinationY=...) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKRescue.pyi b/typings/rcrs_core/commands/AKRescue.pyi new file mode 100644 index 0000000..e1b18d6 --- /dev/null +++ b/typings/rcrs_core/commands/AKRescue.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKRescue(Command): + def __init__(self, agent_id: EntityID, time: int, target: EntityID) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKRest.pyi b/typings/rcrs_core/commands/AKRest.pyi new file mode 100644 index 0000000..78540d6 --- /dev/null +++ b/typings/rcrs_core/commands/AKRest.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKRest(Command): + def __init__(self, agent_id: EntityID, time: int) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKSay.pyi b/typings/rcrs_core/commands/AKSay.pyi new file mode 100644 index 0000000..d523688 --- /dev/null +++ b/typings/rcrs_core/commands/AKSay.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKSay(Command): + def __init__(self, agent_id: EntityID, time: int, message: str) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKSpeak.pyi b/typings/rcrs_core/commands/AKSpeak.pyi new file mode 100644 index 0000000..a72aada --- /dev/null +++ b/typings/rcrs_core/commands/AKSpeak.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKSpeak(Command): + def __init__(self, agent_id: EntityID, time: int, message: str, channel: int) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKSubscribe.pyi b/typings/rcrs_core/commands/AKSubscribe.pyi new file mode 100644 index 0000000..b6b2717 --- /dev/null +++ b/typings/rcrs_core/commands/AKSubscribe.pyi @@ -0,0 +1,17 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID +from typing import List + +class AKSubscribe(Command): + def __init__(self, agent_id: EntityID, time: int, channels: List[int]) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKTell.pyi b/typings/rcrs_core/commands/AKTell.pyi new file mode 100644 index 0000000..e35dfd1 --- /dev/null +++ b/typings/rcrs_core/commands/AKTell.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKTell(Command): + def __init__(self, agent_id: EntityID, time: int, message: str) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/AKUnload.pyi b/typings/rcrs_core/commands/AKUnload.pyi new file mode 100644 index 0000000..689bc11 --- /dev/null +++ b/typings/rcrs_core/commands/AKUnload.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.commands.Command import Command +from rcrs_core.worldmodel.entityID import EntityID + +class AKUnload(Command): + def __init__(self, agent_id: EntityID, time: int) -> None: + ... + + def prepare_cmd(self): # -> MessageProto: + ... + + + diff --git a/typings/rcrs_core/commands/Command.pyi b/typings/rcrs_core/commands/Command.pyi new file mode 100644 index 0000000..e98df35 --- /dev/null +++ b/typings/rcrs_core/commands/Command.pyi @@ -0,0 +1,16 @@ +""" +This type stub file was generated by pyright. +""" + +from abc import ABC, abstractmethod + +class Command(ABC): + def __init__(self) -> None: + ... + + @abstractmethod + def prepare_cmd(): # -> None: + ... + + + diff --git a/typings/rcrs_core/config/config.pyi b/typings/rcrs_core/config/config.pyi new file mode 100644 index 0000000..c6942e2 --- /dev/null +++ b/typings/rcrs_core/config/config.pyi @@ -0,0 +1,55 @@ +""" +This type stub file was generated by pyright. +""" + +class Config: + def __init__(self) -> None: + ... + + def get_value(self, key: str) -> str | None: + ... + + def get_value_or_default(self, key: str, default: str) -> str: + ... + + def get_int_value(self, key: str) -> int | None: + ... + + def get_int_value_or_default(self, key: str, default: int) -> int: + ... + + def get_float_value(self, key) -> float | None: + ... + + def get_float_value_or_default(self, key: str, default: float) -> float: + ... + + def get_boolean_value(self, key) -> bool | None: + ... + + def get_boolean_value_or_default(self, key: str, default: bool) -> bool: + ... + + def get_array_value(self, key) -> list | None: + ... + + def get_array_value_or_default(self, key: str, default: list) -> list: + ... + + def set_value(self, key: str, value: str) -> None: + ... + + def set_int_value(self, key: str, value: int) -> None: + ... + + def set_float_value(self, key: str, value: float) -> None: + ... + + def set_boolean_value(self, key: str, value: bool) -> None: + ... + + def set_array_value(self, key: str, value: list) -> None: + ... + + + diff --git a/typings/rcrs_core/config/config_constraint.pyi b/typings/rcrs_core/config/config_constraint.pyi new file mode 100644 index 0000000..4b018e9 --- /dev/null +++ b/typings/rcrs_core/config/config_constraint.pyi @@ -0,0 +1,25 @@ +""" +This type stub file was generated by pyright. +""" + +from abc import ABC, abstractmethod +from typing import Set, TYPE_CHECKING +from rcrs_core.config.config import Config + +if TYPE_CHECKING: + ... +class ConfigConstraint(ABC): + @abstractmethod + def is_violated(self, config: Config) -> bool: + ... + + @abstractmethod + def get_description(self) -> str: + ... + + @abstractmethod + def get_keys(self) -> Set[str]: + ... + + + diff --git a/typings/rcrs_core/connection/RCRSProto_pb2.pyi b/typings/rcrs_core/connection/RCRSProto_pb2.pyi new file mode 100644 index 0000000..b0a1b66 --- /dev/null +++ b/typings/rcrs_core/connection/RCRSProto_pb2.pyi @@ -0,0 +1,42 @@ +""" +This type stub file was generated by pyright. +""" + +"""Generated protocol buffer code.""" +_sym_db = ... +DESCRIPTOR = ... +_MESSAGEPROTO_COMPONENTSENTRY = ... +_MESSAGEPROTO = ... +_MESSAGELISTPROTO = ... +_MESSAGECOMPONENTPROTO = ... +_STRLISTPROTO = ... +_INTLISTPROTO = ... +_FLOATLISTPROTO = ... +_INTMATRIXPROTO = ... +_VALUEPROTO = ... +_PROPERTYPROTO = ... +_POINT2DPROTO = ... +_ENTITYPROTO = ... +_ENTITYLISTPROTO = ... +_CONFIGPROTO_DATAENTRY = ... +_CONFIGPROTO = ... +_EDGELISTPROTO = ... +_EDGEPROTO = ... +_CHANGESETPROTO_ENTITYCHANGEPROTO = ... +_CHANGESETPROTO = ... +MessageProto = ... +MessageListProto = ... +MessageComponentProto = ... +StrListProto = ... +IntListProto = ... +FloatListProto = ... +IntMatrixProto = ... +ValueProto = ... +PropertyProto = ... +Point2DProto = ... +EntityProto = ... +EntityListProto = ... +ConfigProto = ... +EdgeListProto = ... +EdgeProto = ... +ChangeSetProto = ... diff --git a/typings/rcrs_core/connection/URN.pyi b/typings/rcrs_core/connection/URN.pyi new file mode 100644 index 0000000..c43d1b2 --- /dev/null +++ b/typings/rcrs_core/connection/URN.pyi @@ -0,0 +1,141 @@ +""" +This type stub file was generated by pyright. +""" + +from enum import IntEnum + +class Entity(IntEnum): + WORLD = ... + ROAD = ... + BLOCKADE = ... + BUILDING = ... + REFUGE = ... + HYDRANT = ... + GAS_STATION = ... + FIRE_STATION = ... + AMBULANCE_CENTRE = ... + POLICE_OFFICE = ... + CIVILIAN = ... + FIRE_BRIGADE = ... + AMBULANCE_TEAM = ... + POLICE_FORCE = ... + + +class Property(IntEnum): + START_TIME = ... + LONGITUDE = ... + LATITUDE = ... + WIND_FORCE = ... + WIND_DIRECTION = ... + X = ... + Y = ... + BLOCKADES = ... + REPAIR_COST = ... + FLOORS = ... + BUILDING_ATTRIBUTES = ... + IGNITION = ... + FIERYNESS = ... + BROKENNESS = ... + BUILDING_CODE = ... + BUILDING_AREA_GROUND = ... + BUILDING_AREA_TOTAL = ... + APEXES = ... + EDGES = ... + POSITION = ... + DIRECTION = ... + POSITION_HISTORY = ... + STAMINA = ... + HP = ... + DAMAGE = ... + BURIEDNESS = ... + TRAVEL_DISTANCE = ... + WATER_QUANTITY = ... + TEMPERATURE = ... + IMPORTANCE = ... + CAPACITY = ... + BEDCAPACITY = ... + OCCUPIEDBEDS = ... + REFILLCAPACITY = ... + WAITINGLISTSIZE = ... + + +class Command(IntEnum): + AK_REST = ... + AK_MOVE = ... + AK_LOAD = ... + AK_UNLOAD = ... + AK_SAY = ... + AK_TELL = ... + AK_EXTINGUISH = ... + AK_RESCUE = ... + AK_CLEAR = ... + AK_CLEAR_AREA = ... + AK_SUBSCRIBE = ... + AK_SPEAK = ... + + +class ComponentCommand(IntEnum): + Target = ... + DestinationX = ... + DestinationY = ... + Water = ... + Path = ... + Message = ... + Channel = ... + Channels = ... + + +class ControlMSG(IntEnum): + KG_CONNECT = ... + KG_ACKNOWLEDGE = ... + GK_CONNECT_OK = ... + GK_CONNECT_ERROR = ... + SK_CONNECT = ... + SK_ACKNOWLEDGE = ... + SK_UPDATE = ... + KS_CONNECT_OK = ... + KS_CONNECT_ERROR = ... + KS_UPDATE = ... + KS_COMMANDS = ... + KS_AFTERSHOCKS_INFO = ... + VK_CONNECT = ... + VK_ACKNOWLEDGE = ... + KV_CONNECT_OK = ... + KV_CONNECT_ERROR = ... + KV_TIMESTEP = ... + AK_CONNECT = ... + AK_ACKNOWLEDGE = ... + KA_CONNECT_OK = ... + KA_CONNECT_ERROR = ... + KA_SENSE = ... + SHUTDOWN = ... + ENTITY_ID_REQUEST = ... + ENTITY_ID_RESPONSE = ... + + +class ComponentControlMSG(IntEnum): + RequestID = ... + AgentID = ... + Version = ... + Name = ... + RequestedEntityTypes = ... + SimulatorID = ... + RequestNumber = ... + NumberOfIDs = ... + NewEntityIDs = ... + Reason = ... + Entities = ... + ViewerID = ... + AgentConfig = ... + Time = ... + Updates = ... + Hearing = ... + INTENSITIES = ... + TIMES = ... + ID = ... + Commands = ... + SimulatorConfig = ... + Changes = ... + + +MAP = ... diff --git a/typings/rcrs_core/connection/__init__.pyi b/typings/rcrs_core/connection/__init__.pyi new file mode 100644 index 0000000..006bc27 --- /dev/null +++ b/typings/rcrs_core/connection/__init__.pyi @@ -0,0 +1,4 @@ +""" +This type stub file was generated by pyright. +""" + diff --git a/typings/rcrs_core/connection/componentLauncher.pyi b/typings/rcrs_core/connection/componentLauncher.pyi new file mode 100644 index 0000000..edf2c44 --- /dev/null +++ b/typings/rcrs_core/connection/componentLauncher.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.agents.agent import Agent +from rcrs_core.connection.connection import Connection + +class ComponentLauncher: + def __init__(self, port, host) -> None: + ... + + def make_connection(self) -> Connection: + ... + + def connect(self, agent: Agent, _request_id): # -> None: + ... + + def generate_request_ID(self): # -> int: + ... + + + diff --git a/typings/rcrs_core/connection/connection.pyi b/typings/rcrs_core/connection/connection.pyi new file mode 100644 index 0000000..33218b8 --- /dev/null +++ b/typings/rcrs_core/connection/connection.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +class Connection: + def __init__(self, host, port) -> None: + ... + + def connect(self): # -> None: + ... + + def parseMessageFromKernel(self): # -> None: + ... + + def message_received(self, agent_message_received): # -> None: + ... + + def send_msg(self, msg): # -> None: + ... + + + diff --git a/typings/rcrs_core/connection/rcrs_encoding_utils.pyi b/typings/rcrs_core/connection/rcrs_encoding_utils.pyi new file mode 100644 index 0000000..947fc2b --- /dev/null +++ b/typings/rcrs_core/connection/rcrs_encoding_utils.pyi @@ -0,0 +1,20 @@ +""" +This type stub file was generated by pyright. +""" + +""" utils to handle connection """ +def write_int32(value, sock): # -> None: + ... + +def readnbytes(sock, n): # -> Literal[b""]: + ... + +def read_int32(sock): # -> int: + ... + +def write_msg(msg, sock): # -> None: + ... + +def read_msg(sock): # -> MessageProto: + ... + diff --git a/typings/rcrs_core/entities/__init__.pyi b/typings/rcrs_core/entities/__init__.pyi new file mode 100644 index 0000000..006bc27 --- /dev/null +++ b/typings/rcrs_core/entities/__init__.pyi @@ -0,0 +1,4 @@ +""" +This type stub file was generated by pyright. +""" + diff --git a/typings/rcrs_core/entities/ambulanceCenter.pyi b/typings/rcrs_core/entities/ambulanceCenter.pyi new file mode 100644 index 0000000..bde5db4 --- /dev/null +++ b/typings/rcrs_core/entities/ambulanceCenter.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.building import Building + +class AmbulanceCentreEntity(Building): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def set_entity(self, properties): # -> None: + ... + + def get_entity_name(self): # -> Literal['Ambulance Centre']: + ... + + def copy_impl(self): # -> AmbulanceCentreEntity: + ... + + + diff --git a/typings/rcrs_core/entities/ambulanceTeam.pyi b/typings/rcrs_core/entities/ambulanceTeam.pyi new file mode 100644 index 0000000..5c42faa --- /dev/null +++ b/typings/rcrs_core/entities/ambulanceTeam.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.human import Human + +class AmbulanceTeamEntity(Human): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def set_entity(self, properties): # -> None: + ... + + def get_entity_name(self): # -> Literal['Ambulance Team']: + ... + + def copy_impl(self): # -> AmbulanceTeamEntity: + ... + + + diff --git a/typings/rcrs_core/entities/area.pyi b/typings/rcrs_core/entities/area.pyi new file mode 100644 index 0000000..b3f1559 --- /dev/null +++ b/typings/rcrs_core/entities/area.pyi @@ -0,0 +1,52 @@ +""" +This type stub file was generated by pyright. +""" + +from typing import List +from rcrs_core.entities.entity import Entity + +class Area(Entity): + def __init__(self, entity_id) -> None: + ... + + def set_entity(self, properties): # -> None: + ... + + def get_location(self): # -> tuple[Any | None, Any | None]: + ... + + def get_neighbours(self): # -> list[Any]: + ... + + def get_edge_to(self, neighbour): # -> Entity | None: + ... + + def get_property(self, urn): # -> IntProperty | EdgeListProperty | EntityIDListProperty | None: + ... + + def get_edges_property(self): # -> EdgeListProperty: + ... + + def get_edges(self) -> List[Entity]: + ... + + def set_edges(self, value): # -> None: + ... + + def add_edge(self, edge): # -> None: + ... + + def get_blockades_property(self): # -> EntityIDListProperty: + ... + + def get_blockades(self): # -> None: + ... + + def set_blockades(self, value): # -> None: + ... + + def get_shape(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/blockade.pyi b/typings/rcrs_core/entities/blockade.pyi new file mode 100644 index 0000000..a9b1502 --- /dev/null +++ b/typings/rcrs_core/entities/blockade.pyi @@ -0,0 +1,73 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.entity import Entity + +class Blockade(Entity): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def get_entity_name(self): # -> Literal['Blockade']: + ... + + def set_entity(self, properties): # -> None: + ... + + def copy_impl(self): # -> Blockade: + ... + + def get_property(self, urn): # -> IntProperty | EntityIDProperty | IntArrayProperty | None: + ... + + def get_x_property(self): # -> IntProperty: + ... + + def get_x(self): # -> None: + ... + + def set_x(self, value): # -> None: + ... + + def get_y_property(self): # -> IntProperty: + ... + + def get_y(self): # -> None: + ... + + def set_y(self, value): # -> None: + ... + + def get_apexes_property(self): # -> IntArrayProperty: + ... + + def get_apexes(self): # -> None: + ... + + def set_apexes(self, value): # -> None: + ... + + def get_position_property(self): # -> EntityIDProperty: + ... + + def get_position(self): # -> None: + ... + + def set_position(self, value): # -> None: + ... + + def get_repaire_cost_property(self): # -> IntProperty: + ... + + def get_repaire_cost(self): # -> None: + ... + + def set_repaire_cost(self, value): # -> None: + ... + + def get_shape(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/building.pyi b/typings/rcrs_core/entities/building.pyi new file mode 100644 index 0000000..edf477d --- /dev/null +++ b/typings/rcrs_core/entities/building.pyi @@ -0,0 +1,124 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.area import Area + +class Building(Area): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def is__fiery(self): # -> bool: + ... + + def set_entity(self, properties): # -> None: + ... + + def copy_impl(self): # -> Building: + ... + + def get_entity_name(self): # -> Literal['Building']: + ... + + def get_property(self, urn): # -> IntProperty | EdgeListProperty | EntityIDListProperty | None: + ... + + def get_floors_property(self): # -> IntProperty: + ... + + def get_floors(self): # -> None: + ... + + def set_floors(self, value): # -> None: + ... + + def get_ignition_property(self): # -> IntProperty: + ... + + def get_ignition(self): # -> None: + ... + + def set_ignition(self, value): # -> None: + ... + + def get_fieryness_property(self): # -> IntProperty: + ... + + def get_fieryness(self): # -> None: + ... + + def set_fieryness(self, value): # -> None: + ... + + def get_fieryness_enum(self): # -> None: + ... + + def get_brokenness_property(self): # -> IntProperty: + ... + + def get_brokenness(self): # -> None: + ... + + def set_brokenness(self, value): # -> None: + ... + + def get_building_code_property(self): # -> IntProperty: + ... + + def get_building_code(self): # -> None: + ... + + def set_building_code(self, value): # -> None: + ... + + def get_building_code_enum(self): # -> None: + ... + + def get_attributes_property(self): # -> IntProperty: + ... + + def get_attributes(self): # -> None: + ... + + def set_attributes(self, value): # -> None: + ... + + def get_ground_area_property(self): # -> IntProperty: + ... + + def get_ground_area(self): # -> None: + ... + + def set_ground_area(self, value): # -> None: + ... + + def get_total_area_property(self): # -> IntProperty: + ... + + def get_total_area(self): # -> None: + ... + + def set_total_area(self, value): # -> None: + ... + + def get_temperature_property(self): # -> IntProperty: + ... + + def get_temperature(self): # -> None: + ... + + def set_temperature(self, value): # -> None: + ... + + def get_importance_property(self): # -> IntProperty: + ... + + def get_importance(self): # -> None: + ... + + def set_importance(self, value): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/civilian.pyi b/typings/rcrs_core/entities/civilian.pyi new file mode 100644 index 0000000..95f8ee3 --- /dev/null +++ b/typings/rcrs_core/entities/civilian.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.human import Human + +class Civilian(Human): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def copy_impl(self): # -> Civilian: + ... + + def get_entity_name(self): # -> Literal['Civilian']: + ... + + def set_entity(self, properties): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/edge.pyi b/typings/rcrs_core/entities/edge.pyi new file mode 100644 index 0000000..f35452b --- /dev/null +++ b/typings/rcrs_core/entities/edge.pyi @@ -0,0 +1,31 @@ +""" +This type stub file was generated by pyright. +""" + +class Edge: + def __init__(self, start_x, start_y, end_x, end_y, _neighbour) -> None: + ... + + def get_start_x(self): + ... + + def get_start_y(self): + ... + + def get_end_x(self): + ... + + def get_end_y(self): + ... + + def is_passable(self): # -> bool: + ... + + def get_neighbour(self): # -> None: + ... + + def to_string(self): + ... + + + diff --git a/typings/rcrs_core/entities/entity.pyi b/typings/rcrs_core/entities/entity.pyi new file mode 100644 index 0000000..1ebb7aa --- /dev/null +++ b/typings/rcrs_core/entities/entity.pyi @@ -0,0 +1,61 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.worldmodel.entityID import EntityID +from abc import ABC + +class Entity(ABC): + def __init__(self, _entity_id) -> None: + ... + + def set_entity(self, properties): # -> None: + ... + + def get_id(self) -> EntityID: + ... + + def get_properties(self): # -> dict[Any, Any]: + ... + + def register_properties(self, _properties): # -> None: + ... + + def copy_impl(self): # -> Entity: + ... + + def copy(self): # -> Entity: + ... + + def get_urn(self): + ... + + def __hash__(self) -> int: + ... + + def get_property(self, property_urn): # -> None: + ... + + def get_location(self): # -> tuple[Any | None, Any | None] | tuple[None, None]: + ... + + def get_x_property(self): # -> IntProperty: + ... + + def get_x(self): # -> None: + ... + + def set_x(self, value): # -> None: + ... + + def get_y_property(self): # -> IntProperty: + ... + + def get_y(self): # -> None: + ... + + def set_y(self, value): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/fireBrigade.pyi b/typings/rcrs_core/entities/fireBrigade.pyi new file mode 100644 index 0000000..850dc4d --- /dev/null +++ b/typings/rcrs_core/entities/fireBrigade.pyi @@ -0,0 +1,34 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.human import Human + +class FireBrigadeEntity(Human): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def set_entity(self, properties): # -> None: + ... + + def copy_impl(self): # -> FireBrigadeEntity: + ... + + def get_entity_name(self): # -> Literal['Fire brigade']: + ... + + def get_property(self, urn): # -> IntProperty | EntityIDProperty | IntArrayProperty | None: + ... + + def get_water_property(self): # -> IntProperty: + ... + + def get_water(self): # -> None: + ... + + def set_water(self, value): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/fireStation.pyi b/typings/rcrs_core/entities/fireStation.pyi new file mode 100644 index 0000000..46a7b74 --- /dev/null +++ b/typings/rcrs_core/entities/fireStation.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.building import Building + +class FireStationEntity(Building): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def copy_impl(self): # -> FireStationEntity: + ... + + def get_entity_name(self): # -> Literal['Fire Station']: + ... + + def set_entity(self, properties): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/gassStation.pyi b/typings/rcrs_core/entities/gassStation.pyi new file mode 100644 index 0000000..71aa885 --- /dev/null +++ b/typings/rcrs_core/entities/gassStation.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.building import Building + +class GasStation(Building): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def copy_impl(self): # -> GasStation: + ... + + def get_entity_name(self): # -> Literal['Gass Station']: + ... + + def set_entity(self, properties): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/human.pyi b/typings/rcrs_core/entities/human.pyi new file mode 100644 index 0000000..aebfd77 --- /dev/null +++ b/typings/rcrs_core/entities/human.pyi @@ -0,0 +1,114 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.entity import Entity +from rcrs_core.properties.entityIDProperty import EntityIDProperty +from rcrs_core.worldmodel.entityID import EntityID +from rcrs_core.worldmodel.worldmodel import WorldModel + +class Human(Entity): + def __init__(self, entity_id) -> None: + ... + + def get_location(self, world_model: WorldModel): # -> None: + ... + + def set_entity(self, properties: dict): # -> None: + ... + + def get_property(self, urn): # -> EntityIDProperty | IntArrayProperty | IntProperty | None: + ... + + def get_x_property(self): # -> IntProperty: + ... + + def get_x(self) -> int: + ... + + def set_x(self, value) -> None: + ... + + def get_y_property(self): # -> IntProperty: + ... + + def get_y(self): # -> None: + ... + + def set_y(self, value): # -> None: + ... + + def get_position_property(self) -> EntityIDProperty: + ... + + def get_position(self) -> EntityID: + ... + + def set_position(self, value): # -> None: + ... + + def get_position_history_property(self): # -> IntArrayProperty: + ... + + def get_position_history(self): # -> None: + ... + + def set_position_history(self, value): # -> None: + ... + + def get_direction_property(self): # -> IntProperty: + ... + + def get_direction(self): # -> None: + ... + + def set_direction(self, value): # -> None: + ... + + def get_stamina_property(self): # -> IntProperty: + ... + + def get_stamina(self): # -> None: + ... + + def set_stamina(self, value): # -> None: + ... + + def get_hp_property(self): # -> IntProperty: + ... + + def get_hp(self): # -> None: + ... + + def set_hp(self, value): # -> None: + ... + + def get_damage_property(self): # -> IntProperty: + ... + + def get_damage(self): # -> None: + ... + + def set_damage(self, value): # -> None: + ... + + def get_buriedness_property(self): # -> IntProperty: + ... + + def get_buriedness(self): # -> None: + ... + + def set_buriedness(self, value): # -> None: + ... + + def get_travel_distance_property(self): # -> IntProperty: + ... + + def get_travel_distance(self): # -> None: + ... + + def set_travel_distance(self, value): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/hydrant.pyi b/typings/rcrs_core/entities/hydrant.pyi new file mode 100644 index 0000000..4d564cd --- /dev/null +++ b/typings/rcrs_core/entities/hydrant.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.road import Road + +class Hydrant(Road): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def copy_impl(self): # -> Hydrant: + ... + + def get_entity_name(self): # -> Literal['Hydrant']: + ... + + def set_entity(self, properties): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/policeForce.pyi b/typings/rcrs_core/entities/policeForce.pyi new file mode 100644 index 0000000..1ee42f3 --- /dev/null +++ b/typings/rcrs_core/entities/policeForce.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.human import Human + +class PoliceForceEntity(Human): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def copy_impl(self): # -> PoliceForceEntity: + ... + + def get_entity_name(self): # -> Literal['Police force']: + ... + + def set_entity(self, properties): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/policeOffice.pyi b/typings/rcrs_core/entities/policeOffice.pyi new file mode 100644 index 0000000..f060e59 --- /dev/null +++ b/typings/rcrs_core/entities/policeOffice.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.building import Building + +class PoliceOfficeEntity(Building): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def copy_impl(self): # -> PoliceOfficeEntity: + ... + + def get_entity_name(self): # -> Literal['Police office']: + ... + + def set_entity(self, properties): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/refuge.pyi b/typings/rcrs_core/entities/refuge.pyi new file mode 100644 index 0000000..17ff0ed --- /dev/null +++ b/typings/rcrs_core/entities/refuge.pyi @@ -0,0 +1,61 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.building import Building + +class Refuge(Building): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def set_entity(self, properties): # -> None: + ... + + def copy_impl(self): # -> Refuge: + ... + + def get_entity_name(self) -> str: + ... + + def get_property(self, urn): # -> IntProperty | EdgeListProperty | EntityIDListProperty | None: + ... + + def get_bed_capacity_property(self): # -> IntProperty: + ... + + def get_bed_capacity(self): # -> None: + ... + + def set_bed_capacity(self, value): # -> None: + ... + + def get_occupied_beds_property(self): # -> IntProperty: + ... + + def get_occupied_beds(self): # -> None: + ... + + def set_occupied_beds(self, value): # -> None: + ... + + def get_refill_capacity_property(self): # -> IntProperty: + ... + + def get_refill_capacity(self): # -> None: + ... + + def set_refill_capacity(self, value): # -> None: + ... + + def get_waiting_list_size_property(self): # -> IntProperty: + ... + + def get_waiting_list_size(self): # -> None: + ... + + def set_waiting_list_size(self, value): # -> None: + ... + + + diff --git a/typings/rcrs_core/entities/road.pyi b/typings/rcrs_core/entities/road.pyi new file mode 100644 index 0000000..2ceaf0d --- /dev/null +++ b/typings/rcrs_core/entities/road.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.entities.area import Area + +class Road(Area): + urn = ... + def __init__(self, entity_id) -> None: + ... + + def set_entity(self, properties) -> None: + ... + + def get_entity_name(self) -> str: + ... + + def copy_impl(self): # -> Road: + ... + + + diff --git a/typings/rcrs_core/entities/standardEntityFactory.pyi b/typings/rcrs_core/entities/standardEntityFactory.pyi new file mode 100644 index 0000000..651fc76 --- /dev/null +++ b/typings/rcrs_core/entities/standardEntityFactory.pyi @@ -0,0 +1,17 @@ +""" +This type stub file was generated by pyright. +""" + +class StandardEntityFactory: + _instance = ... + def __new__(cls): # -> Self: + ... + + def __init__(self) -> None: + ... + + def make_entity(urn, id): # -> Building | Refuge | Road | Blockade | PoliceForceEntity | PoliceOfficeEntity | AmbulanceTeamEntity | AmbulanceCentreEntity | FireBrigadeEntity | FireStationEntity | Civilian | Hydrant | GasStation | None: + ... + + + diff --git a/typings/rcrs_core/log/logger.pyi b/typings/rcrs_core/log/logger.pyi new file mode 100644 index 0000000..bec3fd5 --- /dev/null +++ b/typings/rcrs_core/log/logger.pyi @@ -0,0 +1,28 @@ +""" +This type stub file was generated by pyright. +""" + +class Logger: + def __init__(self, _name, agent_id=...) -> None: + ... + + def info(self, msg): # -> None: + ... + + def debug(self, msg): # -> None: + ... + + def error(self, msg): # -> None: + ... + + def warning(self, msg): # -> None: + ... + + def warn(self, msg): # -> None: + ... + + def set_id(self, id): # -> None: + ... + + + diff --git a/typings/rcrs_core/messages/AKAcknowledge.pyi b/typings/rcrs_core/messages/AKAcknowledge.pyi new file mode 100644 index 0000000..1bea0e2 --- /dev/null +++ b/typings/rcrs_core/messages/AKAcknowledge.pyi @@ -0,0 +1,18 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.messages.message import Message + +class AKAcknowledge(Message): + def __init__(self) -> None: + ... + + def write(self, request_id, agent_id): # -> MessageProto: + ... + + def read(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/messages/AKConnect.pyi b/typings/rcrs_core/messages/AKConnect.pyi new file mode 100644 index 0000000..2c23c18 --- /dev/null +++ b/typings/rcrs_core/messages/AKConnect.pyi @@ -0,0 +1,18 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.messages.message import Message + +class AKConnect(Message): + def __init__(self) -> None: + ... + + def write(self, request_id, agent): # -> MessageProto: + ... + + def read(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/messages/KAConnectError.pyi b/typings/rcrs_core/messages/KAConnectError.pyi new file mode 100644 index 0000000..4fa547d --- /dev/null +++ b/typings/rcrs_core/messages/KAConnectError.pyi @@ -0,0 +1,18 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.messages.message import Message + +class KAConnectError(Message): + def __init__(self, data) -> None: + ... + + def read(self): # -> None: + ... + + def write(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/messages/KAConnectOK.pyi b/typings/rcrs_core/messages/KAConnectOK.pyi new file mode 100644 index 0000000..100e853 --- /dev/null +++ b/typings/rcrs_core/messages/KAConnectOK.pyi @@ -0,0 +1,18 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.messages.message import Message + +class KAConnectOK(Message): + def __init__(self, data) -> None: + ... + + def read(self): # -> None: + ... + + def write(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/messages/KASense.pyi b/typings/rcrs_core/messages/KASense.pyi new file mode 100644 index 0000000..163ac38 --- /dev/null +++ b/typings/rcrs_core/messages/KASense.pyi @@ -0,0 +1,19 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.connection import RCRSProto_pb2 +from rcrs_core.messages.message import Message + +class KASense(Message): + def __init__(self, data: RCRSProto_pb2) -> None: + ... + + def read(self): # -> None: + ... + + def write(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/messages/Shutdown.pyi b/typings/rcrs_core/messages/Shutdown.pyi new file mode 100644 index 0000000..e5a6335 --- /dev/null +++ b/typings/rcrs_core/messages/Shutdown.pyi @@ -0,0 +1,18 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.messages.message import Message + +class Shutdown(Message): + def __init__(self, data) -> None: + ... + + def read(self): # -> None: + ... + + def write(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/messages/controlMessageFactory.pyi b/typings/rcrs_core/messages/controlMessageFactory.pyi new file mode 100644 index 0000000..a3e403b --- /dev/null +++ b/typings/rcrs_core/messages/controlMessageFactory.pyi @@ -0,0 +1,13 @@ +""" +This type stub file was generated by pyright. +""" + +class ControlMessageFactory: + def __init__(self) -> None: + ... + + def make_message(self, msg): # -> KASense | KAConnectOK | KAConnectError | Shutdown | None: + ... + + + diff --git a/typings/rcrs_core/messages/message.pyi b/typings/rcrs_core/messages/message.pyi new file mode 100644 index 0000000..005a916 --- /dev/null +++ b/typings/rcrs_core/messages/message.pyi @@ -0,0 +1,23 @@ +""" +This type stub file was generated by pyright. +""" + +from abc import ABC, abstractmethod + +class Message(ABC): + def __init__(self, _urn) -> None: + ... + + @abstractmethod + def read(self): # -> None: + ... + + @abstractmethod + def write(self): # -> None: + ... + + def get_urn(self): # -> Any: + ... + + + diff --git a/typings/rcrs_core/properties/booleanProperty.pyi b/typings/rcrs_core/properties/booleanProperty.pyi new file mode 100644 index 0000000..7bd816f --- /dev/null +++ b/typings/rcrs_core/properties/booleanProperty.pyi @@ -0,0 +1,21 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.properties.property import Property + +class BooleanProperty(Property): + def __init__(self, urn) -> None: + ... + + def set_fields(self, value): # -> None: + ... + + def copy(self): # -> BooleanProperty: + ... + + def take_value(self, _property): # -> None: + ... + + + diff --git a/typings/rcrs_core/properties/edgeListProperty.pyi b/typings/rcrs_core/properties/edgeListProperty.pyi new file mode 100644 index 0000000..2af9f10 --- /dev/null +++ b/typings/rcrs_core/properties/edgeListProperty.pyi @@ -0,0 +1,38 @@ +""" +This type stub file was generated by pyright. +""" + +from typing import List +from rcrs_core.properties.property import Property +from rcrs_core.entities.edge import Edge + +class EdgeListProperty(Property): + def __init__(self, urn) -> None: + ... + + def get_fields(self): # -> None: + ... + + def set_fields(self, data): # -> None: + ... + + def set_value(self, _value: List[Edge]): # -> None: + ... + + def set_edges(self, _edges: List[Edge]): # -> None: + ... + + def add_edge(self, _edge): # -> None: + ... + + def clear_edges(self): # -> None: + ... + + def take_value(self, _value): # -> None: + ... + + def copy(self): # -> EdgeListProperty: + ... + + + diff --git a/typings/rcrs_core/properties/entityIDListProperty.pyi b/typings/rcrs_core/properties/entityIDListProperty.pyi new file mode 100644 index 0000000..9f629f1 --- /dev/null +++ b/typings/rcrs_core/properties/entityIDListProperty.pyi @@ -0,0 +1,26 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.properties.property import Property +from rcrs_core.worldmodel.entityID import EntityID +from typing import List + +class EntityIDListProperty(Property): + def __init__(self, urn) -> None: + ... + + def set_fields(self, data): # -> None: + ... + + def set_value(self, _value: List[EntityID]): # -> None: + ... + + def copy(self): # -> EntityIDListProperty: + ... + + def take_value(self, _property): # -> None: + ... + + + diff --git a/typings/rcrs_core/properties/entityIDProperty.pyi b/typings/rcrs_core/properties/entityIDProperty.pyi new file mode 100644 index 0000000..da7b261 --- /dev/null +++ b/typings/rcrs_core/properties/entityIDProperty.pyi @@ -0,0 +1,21 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.properties.property import Property + +class EntityIDProperty(Property): + def __init__(self, urn) -> None: + ... + + def set_fields(self, value): # -> None: + ... + + def copy(self): # -> EntityIDProperty: + ... + + def take_value(self, _property): # -> None: + ... + + + diff --git a/typings/rcrs_core/properties/intArrayProperty.pyi b/typings/rcrs_core/properties/intArrayProperty.pyi new file mode 100644 index 0000000..a4c7a6c --- /dev/null +++ b/typings/rcrs_core/properties/intArrayProperty.pyi @@ -0,0 +1,24 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.properties.property import Property + +class IntArrayProperty(Property): + def __init__(self, urn) -> None: + ... + + def set_fields(self, data): # -> None: + ... + + def set_value(self, data): # -> None: + ... + + def copy(self): # -> IntArrayProperty: + ... + + def take_value(self, _property): # -> None: + ... + + + diff --git a/typings/rcrs_core/properties/intProperty.pyi b/typings/rcrs_core/properties/intProperty.pyi new file mode 100644 index 0000000..83cf474 --- /dev/null +++ b/typings/rcrs_core/properties/intProperty.pyi @@ -0,0 +1,21 @@ +""" +This type stub file was generated by pyright. +""" + +from rcrs_core.properties.property import Property + +class IntProperty(Property): + def __init__(self, urn) -> None: + ... + + def set_fields(self, value): # -> None: + ... + + def copy(self): # -> IntProperty: + ... + + def take_value(self, _property): # -> None: + ... + + + diff --git a/typings/rcrs_core/properties/property.pyi b/typings/rcrs_core/properties/property.pyi new file mode 100644 index 0000000..cbff460 --- /dev/null +++ b/typings/rcrs_core/properties/property.pyi @@ -0,0 +1,22 @@ +""" +This type stub file was generated by pyright. +""" + +class Property: + def __init__(self, _urn) -> None: + ... + + def get_urn(self): # -> Any: + ... + + def get_value(self): # -> None: + ... + + def set_value(self, _value): # -> None: + ... + + def to_string(self): # -> None: + ... + + + diff --git a/typings/rcrs_core/properties/standardPropertyFactory.pyi b/typings/rcrs_core/properties/standardPropertyFactory.pyi new file mode 100644 index 0000000..be77e22 --- /dev/null +++ b/typings/rcrs_core/properties/standardPropertyFactory.pyi @@ -0,0 +1,13 @@ +""" +This type stub file was generated by pyright. +""" + +class StandardPropertyFactory: + def __init__(self) -> None: + ... + + def make_property(urn): # -> IntProperty | IntArrayProperty | BooleanProperty | EntityIDProperty | EntityIDListProperty | EdgeListProperty | None: + ... + + + diff --git a/typings/rcrs_core/worldmodel/changeSet.pyi b/typings/rcrs_core/worldmodel/changeSet.pyi new file mode 100644 index 0000000..d1788f1 --- /dev/null +++ b/typings/rcrs_core/worldmodel/changeSet.pyi @@ -0,0 +1,34 @@ +""" +This type stub file was generated by pyright. +""" + +class ChangeSet: + def __init__(self, _change_set=...) -> None: + ... + + def add_change(self, entity_id, entity_urn, property): # -> None: + ... + + def entity_deleted(self, _entity_id): # -> None: + ... + + def get_changed_properties(self, _entity_id): # -> list[Any]: + ... + + def get_changed_property(self, _entity_id, prop_urn): # -> None: + ... + + def get_changed_entities(self): # -> dict_keys[Any, Any]: + ... + + def get_deleted_entities(self): # -> set[Any]: + ... + + def get_entity_urn(self, _entity_id): # -> None: + ... + + def merge(self, _change_set): # -> None: + ... + + + diff --git a/typings/rcrs_core/worldmodel/entityID.pyi b/typings/rcrs_core/worldmodel/entityID.pyi new file mode 100644 index 0000000..e570b8e --- /dev/null +++ b/typings/rcrs_core/worldmodel/entityID.pyi @@ -0,0 +1,25 @@ +""" +This type stub file was generated by pyright. +""" + +class EntityID: + def __init__(self, id: int) -> None: + ... + + def __eq__(self, other) -> bool: + ... + + def __hash__(self) -> int: + ... + + def get_value(self): # -> | int: + ... + + def __str__(self) -> str: + ... + + def equals(self, o): # -> bool: + ... + + + diff --git a/typings/rcrs_core/worldmodel/worldmodel.pyi b/typings/rcrs_core/worldmodel/worldmodel.pyi new file mode 100644 index 0000000..e9ccce0 --- /dev/null +++ b/typings/rcrs_core/worldmodel/worldmodel.pyi @@ -0,0 +1,51 @@ +""" +This type stub file was generated by pyright. +""" + +from typing import List +from rcrs_core.entities.entity import Entity +from rcrs_core.worldmodel.changeSet import ChangeSet +from rcrs_core.worldmodel.entityID import EntityID + +class WorldModel: + def __init__(self) -> None: + ... + + def add_entities(self, entities: List[Entity]): # -> None: + ... + + def get_entity(self, entity_id: EntityID) -> Entity: + ... + + def add_entity(self, entity): # -> None: + ... + + def remove_entity(self, entity_id): # -> None: + ... + + def get_entities(self): # -> dict_values[Any, Any]: + ... + + def merge(self, change_set: ChangeSet): # -> None: + ... + + def index_entities(self): # -> None: + ... + + def make_rectangle(self, entity): # -> tuple[None, None, None, None] | tuple[float, float, float, float]: + ... + + def get_rect_bounds(self): # -> tuple[Any, Any, Any, Any]: + ... + + def get_world_bounds(self): # -> tuple[Any, Any, Any, Any]: + ... + + def get_objects_in_rectangle(self, x1, y1, x2, y2): # -> list[Any]: + ... + + def get_objects_in_range(self, entity, range): # -> list[Any]: + ... + + + From ae4e8fa2238e24b5dedf4b740a1e46c3bc30d2c1 Mon Sep 17 00:00:00 2001 From: shima004 Date: Wed, 2 Oct 2024 14:09:56 +0900 Subject: [PATCH 2/5] Delete unused type stub files --- typings/rcrs_core/agents/agent.pyi | 94 ------------ typings/rcrs_core/commands/AKClear.pyi | 16 -- typings/rcrs_core/commands/AKClearArea.pyi | 16 -- typings/rcrs_core/commands/AKLoad.pyi | 16 -- typings/rcrs_core/commands/AKMove.pyi | 17 --- typings/rcrs_core/commands/AKRescue.pyi | 16 -- typings/rcrs_core/commands/AKRest.pyi | 16 -- typings/rcrs_core/commands/AKSay.pyi | 16 -- typings/rcrs_core/commands/AKSpeak.pyi | 16 -- typings/rcrs_core/commands/AKSubscribe.pyi | 17 --- typings/rcrs_core/commands/AKTell.pyi | 16 -- typings/rcrs_core/commands/AKUnload.pyi | 16 -- typings/rcrs_core/commands/Command.pyi | 16 -- typings/rcrs_core/config/config.pyi | 55 ------- .../rcrs_core/config/config_constraint.pyi | 25 ---- .../rcrs_core/connection/RCRSProto_pb2.pyi | 42 ------ typings/rcrs_core/connection/URN.pyi | 141 ------------------ typings/rcrs_core/connection/__init__.pyi | 4 - .../connection/componentLauncher.pyi | 22 --- typings/rcrs_core/connection/connection.pyi | 22 --- .../connection/rcrs_encoding_utils.pyi | 20 --- typings/rcrs_core/entities/__init__.pyi | 4 - .../rcrs_core/entities/ambulanceCenter.pyi | 22 --- typings/rcrs_core/entities/ambulanceTeam.pyi | 22 --- typings/rcrs_core/entities/area.pyi | 52 ------- typings/rcrs_core/entities/blockade.pyi | 73 --------- typings/rcrs_core/entities/building.pyi | 124 --------------- typings/rcrs_core/entities/civilian.pyi | 22 --- typings/rcrs_core/entities/edge.pyi | 31 ---- typings/rcrs_core/entities/entity.pyi | 61 -------- typings/rcrs_core/entities/fireBrigade.pyi | 34 ----- typings/rcrs_core/entities/fireStation.pyi | 22 --- typings/rcrs_core/entities/gassStation.pyi | 22 --- typings/rcrs_core/entities/human.pyi | 114 -------------- typings/rcrs_core/entities/hydrant.pyi | 22 --- typings/rcrs_core/entities/policeForce.pyi | 22 --- typings/rcrs_core/entities/policeOffice.pyi | 22 --- typings/rcrs_core/entities/refuge.pyi | 61 -------- typings/rcrs_core/entities/road.pyi | 22 --- .../entities/standardEntityFactory.pyi | 17 --- typings/rcrs_core/log/logger.pyi | 28 ---- typings/rcrs_core/messages/AKAcknowledge.pyi | 18 --- typings/rcrs_core/messages/AKConnect.pyi | 18 --- typings/rcrs_core/messages/KAConnectError.pyi | 18 --- typings/rcrs_core/messages/KAConnectOK.pyi | 18 --- typings/rcrs_core/messages/KASense.pyi | 19 --- typings/rcrs_core/messages/Shutdown.pyi | 18 --- .../messages/controlMessageFactory.pyi | 13 -- typings/rcrs_core/messages/message.pyi | 23 --- .../rcrs_core/properties/booleanProperty.pyi | 21 --- .../rcrs_core/properties/edgeListProperty.pyi | 38 ----- .../properties/entityIDListProperty.pyi | 26 ---- .../rcrs_core/properties/entityIDProperty.pyi | 21 --- .../rcrs_core/properties/intArrayProperty.pyi | 24 --- typings/rcrs_core/properties/intProperty.pyi | 21 --- typings/rcrs_core/properties/property.pyi | 22 --- .../properties/standardPropertyFactory.pyi | 13 -- typings/rcrs_core/worldmodel/changeSet.pyi | 34 ----- typings/rcrs_core/worldmodel/entityID.pyi | 25 ---- typings/rcrs_core/worldmodel/worldmodel.pyi | 51 ------- 60 files changed, 1857 deletions(-) delete mode 100644 typings/rcrs_core/agents/agent.pyi delete mode 100644 typings/rcrs_core/commands/AKClear.pyi delete mode 100644 typings/rcrs_core/commands/AKClearArea.pyi delete mode 100644 typings/rcrs_core/commands/AKLoad.pyi delete mode 100644 typings/rcrs_core/commands/AKMove.pyi delete mode 100644 typings/rcrs_core/commands/AKRescue.pyi delete mode 100644 typings/rcrs_core/commands/AKRest.pyi delete mode 100644 typings/rcrs_core/commands/AKSay.pyi delete mode 100644 typings/rcrs_core/commands/AKSpeak.pyi delete mode 100644 typings/rcrs_core/commands/AKSubscribe.pyi delete mode 100644 typings/rcrs_core/commands/AKTell.pyi delete mode 100644 typings/rcrs_core/commands/AKUnload.pyi delete mode 100644 typings/rcrs_core/commands/Command.pyi delete mode 100644 typings/rcrs_core/config/config.pyi delete mode 100644 typings/rcrs_core/config/config_constraint.pyi delete mode 100644 typings/rcrs_core/connection/RCRSProto_pb2.pyi delete mode 100644 typings/rcrs_core/connection/URN.pyi delete mode 100644 typings/rcrs_core/connection/__init__.pyi delete mode 100644 typings/rcrs_core/connection/componentLauncher.pyi delete mode 100644 typings/rcrs_core/connection/connection.pyi delete mode 100644 typings/rcrs_core/connection/rcrs_encoding_utils.pyi delete mode 100644 typings/rcrs_core/entities/__init__.pyi delete mode 100644 typings/rcrs_core/entities/ambulanceCenter.pyi delete mode 100644 typings/rcrs_core/entities/ambulanceTeam.pyi delete mode 100644 typings/rcrs_core/entities/area.pyi delete mode 100644 typings/rcrs_core/entities/blockade.pyi delete mode 100644 typings/rcrs_core/entities/building.pyi delete mode 100644 typings/rcrs_core/entities/civilian.pyi delete mode 100644 typings/rcrs_core/entities/edge.pyi delete mode 100644 typings/rcrs_core/entities/entity.pyi delete mode 100644 typings/rcrs_core/entities/fireBrigade.pyi delete mode 100644 typings/rcrs_core/entities/fireStation.pyi delete mode 100644 typings/rcrs_core/entities/gassStation.pyi delete mode 100644 typings/rcrs_core/entities/human.pyi delete mode 100644 typings/rcrs_core/entities/hydrant.pyi delete mode 100644 typings/rcrs_core/entities/policeForce.pyi delete mode 100644 typings/rcrs_core/entities/policeOffice.pyi delete mode 100644 typings/rcrs_core/entities/refuge.pyi delete mode 100644 typings/rcrs_core/entities/road.pyi delete mode 100644 typings/rcrs_core/entities/standardEntityFactory.pyi delete mode 100644 typings/rcrs_core/log/logger.pyi delete mode 100644 typings/rcrs_core/messages/AKAcknowledge.pyi delete mode 100644 typings/rcrs_core/messages/AKConnect.pyi delete mode 100644 typings/rcrs_core/messages/KAConnectError.pyi delete mode 100644 typings/rcrs_core/messages/KAConnectOK.pyi delete mode 100644 typings/rcrs_core/messages/KASense.pyi delete mode 100644 typings/rcrs_core/messages/Shutdown.pyi delete mode 100644 typings/rcrs_core/messages/controlMessageFactory.pyi delete mode 100644 typings/rcrs_core/messages/message.pyi delete mode 100644 typings/rcrs_core/properties/booleanProperty.pyi delete mode 100644 typings/rcrs_core/properties/edgeListProperty.pyi delete mode 100644 typings/rcrs_core/properties/entityIDListProperty.pyi delete mode 100644 typings/rcrs_core/properties/entityIDProperty.pyi delete mode 100644 typings/rcrs_core/properties/intArrayProperty.pyi delete mode 100644 typings/rcrs_core/properties/intProperty.pyi delete mode 100644 typings/rcrs_core/properties/property.pyi delete mode 100644 typings/rcrs_core/properties/standardPropertyFactory.pyi delete mode 100644 typings/rcrs_core/worldmodel/changeSet.pyi delete mode 100644 typings/rcrs_core/worldmodel/entityID.pyi delete mode 100644 typings/rcrs_core/worldmodel/worldmodel.pyi diff --git a/typings/rcrs_core/agents/agent.pyi b/typings/rcrs_core/agents/agent.pyi deleted file mode 100644 index e19674a..0000000 --- a/typings/rcrs_core/agents/agent.pyi +++ /dev/null @@ -1,94 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.connection.URN import Entity -from abc import ABC, abstractmethod -from rcrs_core.entities.human import Human - -class Agent(ABC): - def __init__(self, pre) -> None: - ... - - @abstractmethod - def precompute(self): # -> None: - ... - - @abstractmethod - def think(self, time, change_set, hear): # -> None: - ... - - def get_name(self): # -> str: - ... - - def get_id(self): # -> EntityID | None: - ... - - def set_send_msg(self, connection_send_func): # -> None: - ... - - def start_up(self, request_id): # -> None: - ... - - def post_connect(self): # -> None: - ... - - def message_received(self, msg): # -> None: - ... - - def handle_connect_error(self, msg): - ... - - def handle_connect_ok(self, msg): # -> None: - ... - - def sendAKAcknowledge(self, request_id): # -> None: - ... - - def process_sense(self, msg): # -> None: - ... - - def me(self) -> Human: - ... - - def location(self) -> Entity: - ... - - def random_walk(self): # -> list[Any]: - ... - - def send_clear(self, time, target): # -> None: - ... - - def send_clear_area(self, time, x=..., y=...): # -> None: - ... - - def send_load(self, time, target): # -> None: - ... - - def send_move(self, time, path, x=..., y=...): # -> None: - ... - - def send_rescue(self, time, target): # -> None: - ... - - def send_rest(self, time_step): # -> None: - ... - - def send_say(self, time_step: int, message: str): # -> None: - ... - - def send_speak(self, time_step: int, message: str, channel: int): # -> None: - ... - - def send_subscribe(self, time, channel): # -> None: - ... - - def send_tell(self, time_step: int, message: str): # -> None: - ... - - def send_unload(self, time): # -> None: - ... - - - diff --git a/typings/rcrs_core/commands/AKClear.pyi b/typings/rcrs_core/commands/AKClear.pyi deleted file mode 100644 index 1f3119a..0000000 --- a/typings/rcrs_core/commands/AKClear.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKClear(Command): - def __init__(self, agent_id: EntityID, time: int, target: EntityID) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKClearArea.pyi b/typings/rcrs_core/commands/AKClearArea.pyi deleted file mode 100644 index cde4e0d..0000000 --- a/typings/rcrs_core/commands/AKClearArea.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKClearArea(Command): - def __init__(self, agent_id: EntityID, time: int, destinationX: int, destinationY: int) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKLoad.pyi b/typings/rcrs_core/commands/AKLoad.pyi deleted file mode 100644 index 1dda8d4..0000000 --- a/typings/rcrs_core/commands/AKLoad.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKLoad(Command): - def __init__(self, agent_id: EntityID, time: int, target: EntityID) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKMove.pyi b/typings/rcrs_core/commands/AKMove.pyi deleted file mode 100644 index 32f0c27..0000000 --- a/typings/rcrs_core/commands/AKMove.pyi +++ /dev/null @@ -1,17 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID -from typing import List - -class AKMove(Command): - def __init__(self, agent_id: EntityID, time: int, path: List[int], destinationX=..., destinationY=...) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKRescue.pyi b/typings/rcrs_core/commands/AKRescue.pyi deleted file mode 100644 index e1b18d6..0000000 --- a/typings/rcrs_core/commands/AKRescue.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKRescue(Command): - def __init__(self, agent_id: EntityID, time: int, target: EntityID) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKRest.pyi b/typings/rcrs_core/commands/AKRest.pyi deleted file mode 100644 index 78540d6..0000000 --- a/typings/rcrs_core/commands/AKRest.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKRest(Command): - def __init__(self, agent_id: EntityID, time: int) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKSay.pyi b/typings/rcrs_core/commands/AKSay.pyi deleted file mode 100644 index d523688..0000000 --- a/typings/rcrs_core/commands/AKSay.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKSay(Command): - def __init__(self, agent_id: EntityID, time: int, message: str) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKSpeak.pyi b/typings/rcrs_core/commands/AKSpeak.pyi deleted file mode 100644 index a72aada..0000000 --- a/typings/rcrs_core/commands/AKSpeak.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKSpeak(Command): - def __init__(self, agent_id: EntityID, time: int, message: str, channel: int) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKSubscribe.pyi b/typings/rcrs_core/commands/AKSubscribe.pyi deleted file mode 100644 index b6b2717..0000000 --- a/typings/rcrs_core/commands/AKSubscribe.pyi +++ /dev/null @@ -1,17 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID -from typing import List - -class AKSubscribe(Command): - def __init__(self, agent_id: EntityID, time: int, channels: List[int]) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKTell.pyi b/typings/rcrs_core/commands/AKTell.pyi deleted file mode 100644 index e35dfd1..0000000 --- a/typings/rcrs_core/commands/AKTell.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKTell(Command): - def __init__(self, agent_id: EntityID, time: int, message: str) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/AKUnload.pyi b/typings/rcrs_core/commands/AKUnload.pyi deleted file mode 100644 index 689bc11..0000000 --- a/typings/rcrs_core/commands/AKUnload.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.commands.Command import Command -from rcrs_core.worldmodel.entityID import EntityID - -class AKUnload(Command): - def __init__(self, agent_id: EntityID, time: int) -> None: - ... - - def prepare_cmd(self): # -> MessageProto: - ... - - - diff --git a/typings/rcrs_core/commands/Command.pyi b/typings/rcrs_core/commands/Command.pyi deleted file mode 100644 index e98df35..0000000 --- a/typings/rcrs_core/commands/Command.pyi +++ /dev/null @@ -1,16 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from abc import ABC, abstractmethod - -class Command(ABC): - def __init__(self) -> None: - ... - - @abstractmethod - def prepare_cmd(): # -> None: - ... - - - diff --git a/typings/rcrs_core/config/config.pyi b/typings/rcrs_core/config/config.pyi deleted file mode 100644 index c6942e2..0000000 --- a/typings/rcrs_core/config/config.pyi +++ /dev/null @@ -1,55 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class Config: - def __init__(self) -> None: - ... - - def get_value(self, key: str) -> str | None: - ... - - def get_value_or_default(self, key: str, default: str) -> str: - ... - - def get_int_value(self, key: str) -> int | None: - ... - - def get_int_value_or_default(self, key: str, default: int) -> int: - ... - - def get_float_value(self, key) -> float | None: - ... - - def get_float_value_or_default(self, key: str, default: float) -> float: - ... - - def get_boolean_value(self, key) -> bool | None: - ... - - def get_boolean_value_or_default(self, key: str, default: bool) -> bool: - ... - - def get_array_value(self, key) -> list | None: - ... - - def get_array_value_or_default(self, key: str, default: list) -> list: - ... - - def set_value(self, key: str, value: str) -> None: - ... - - def set_int_value(self, key: str, value: int) -> None: - ... - - def set_float_value(self, key: str, value: float) -> None: - ... - - def set_boolean_value(self, key: str, value: bool) -> None: - ... - - def set_array_value(self, key: str, value: list) -> None: - ... - - - diff --git a/typings/rcrs_core/config/config_constraint.pyi b/typings/rcrs_core/config/config_constraint.pyi deleted file mode 100644 index 4b018e9..0000000 --- a/typings/rcrs_core/config/config_constraint.pyi +++ /dev/null @@ -1,25 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from abc import ABC, abstractmethod -from typing import Set, TYPE_CHECKING -from rcrs_core.config.config import Config - -if TYPE_CHECKING: - ... -class ConfigConstraint(ABC): - @abstractmethod - def is_violated(self, config: Config) -> bool: - ... - - @abstractmethod - def get_description(self) -> str: - ... - - @abstractmethod - def get_keys(self) -> Set[str]: - ... - - - diff --git a/typings/rcrs_core/connection/RCRSProto_pb2.pyi b/typings/rcrs_core/connection/RCRSProto_pb2.pyi deleted file mode 100644 index b0a1b66..0000000 --- a/typings/rcrs_core/connection/RCRSProto_pb2.pyi +++ /dev/null @@ -1,42 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -"""Generated protocol buffer code.""" -_sym_db = ... -DESCRIPTOR = ... -_MESSAGEPROTO_COMPONENTSENTRY = ... -_MESSAGEPROTO = ... -_MESSAGELISTPROTO = ... -_MESSAGECOMPONENTPROTO = ... -_STRLISTPROTO = ... -_INTLISTPROTO = ... -_FLOATLISTPROTO = ... -_INTMATRIXPROTO = ... -_VALUEPROTO = ... -_PROPERTYPROTO = ... -_POINT2DPROTO = ... -_ENTITYPROTO = ... -_ENTITYLISTPROTO = ... -_CONFIGPROTO_DATAENTRY = ... -_CONFIGPROTO = ... -_EDGELISTPROTO = ... -_EDGEPROTO = ... -_CHANGESETPROTO_ENTITYCHANGEPROTO = ... -_CHANGESETPROTO = ... -MessageProto = ... -MessageListProto = ... -MessageComponentProto = ... -StrListProto = ... -IntListProto = ... -FloatListProto = ... -IntMatrixProto = ... -ValueProto = ... -PropertyProto = ... -Point2DProto = ... -EntityProto = ... -EntityListProto = ... -ConfigProto = ... -EdgeListProto = ... -EdgeProto = ... -ChangeSetProto = ... diff --git a/typings/rcrs_core/connection/URN.pyi b/typings/rcrs_core/connection/URN.pyi deleted file mode 100644 index c43d1b2..0000000 --- a/typings/rcrs_core/connection/URN.pyi +++ /dev/null @@ -1,141 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from enum import IntEnum - -class Entity(IntEnum): - WORLD = ... - ROAD = ... - BLOCKADE = ... - BUILDING = ... - REFUGE = ... - HYDRANT = ... - GAS_STATION = ... - FIRE_STATION = ... - AMBULANCE_CENTRE = ... - POLICE_OFFICE = ... - CIVILIAN = ... - FIRE_BRIGADE = ... - AMBULANCE_TEAM = ... - POLICE_FORCE = ... - - -class Property(IntEnum): - START_TIME = ... - LONGITUDE = ... - LATITUDE = ... - WIND_FORCE = ... - WIND_DIRECTION = ... - X = ... - Y = ... - BLOCKADES = ... - REPAIR_COST = ... - FLOORS = ... - BUILDING_ATTRIBUTES = ... - IGNITION = ... - FIERYNESS = ... - BROKENNESS = ... - BUILDING_CODE = ... - BUILDING_AREA_GROUND = ... - BUILDING_AREA_TOTAL = ... - APEXES = ... - EDGES = ... - POSITION = ... - DIRECTION = ... - POSITION_HISTORY = ... - STAMINA = ... - HP = ... - DAMAGE = ... - BURIEDNESS = ... - TRAVEL_DISTANCE = ... - WATER_QUANTITY = ... - TEMPERATURE = ... - IMPORTANCE = ... - CAPACITY = ... - BEDCAPACITY = ... - OCCUPIEDBEDS = ... - REFILLCAPACITY = ... - WAITINGLISTSIZE = ... - - -class Command(IntEnum): - AK_REST = ... - AK_MOVE = ... - AK_LOAD = ... - AK_UNLOAD = ... - AK_SAY = ... - AK_TELL = ... - AK_EXTINGUISH = ... - AK_RESCUE = ... - AK_CLEAR = ... - AK_CLEAR_AREA = ... - AK_SUBSCRIBE = ... - AK_SPEAK = ... - - -class ComponentCommand(IntEnum): - Target = ... - DestinationX = ... - DestinationY = ... - Water = ... - Path = ... - Message = ... - Channel = ... - Channels = ... - - -class ControlMSG(IntEnum): - KG_CONNECT = ... - KG_ACKNOWLEDGE = ... - GK_CONNECT_OK = ... - GK_CONNECT_ERROR = ... - SK_CONNECT = ... - SK_ACKNOWLEDGE = ... - SK_UPDATE = ... - KS_CONNECT_OK = ... - KS_CONNECT_ERROR = ... - KS_UPDATE = ... - KS_COMMANDS = ... - KS_AFTERSHOCKS_INFO = ... - VK_CONNECT = ... - VK_ACKNOWLEDGE = ... - KV_CONNECT_OK = ... - KV_CONNECT_ERROR = ... - KV_TIMESTEP = ... - AK_CONNECT = ... - AK_ACKNOWLEDGE = ... - KA_CONNECT_OK = ... - KA_CONNECT_ERROR = ... - KA_SENSE = ... - SHUTDOWN = ... - ENTITY_ID_REQUEST = ... - ENTITY_ID_RESPONSE = ... - - -class ComponentControlMSG(IntEnum): - RequestID = ... - AgentID = ... - Version = ... - Name = ... - RequestedEntityTypes = ... - SimulatorID = ... - RequestNumber = ... - NumberOfIDs = ... - NewEntityIDs = ... - Reason = ... - Entities = ... - ViewerID = ... - AgentConfig = ... - Time = ... - Updates = ... - Hearing = ... - INTENSITIES = ... - TIMES = ... - ID = ... - Commands = ... - SimulatorConfig = ... - Changes = ... - - -MAP = ... diff --git a/typings/rcrs_core/connection/__init__.pyi b/typings/rcrs_core/connection/__init__.pyi deleted file mode 100644 index 006bc27..0000000 --- a/typings/rcrs_core/connection/__init__.pyi +++ /dev/null @@ -1,4 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - diff --git a/typings/rcrs_core/connection/componentLauncher.pyi b/typings/rcrs_core/connection/componentLauncher.pyi deleted file mode 100644 index edf2c44..0000000 --- a/typings/rcrs_core/connection/componentLauncher.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.agents.agent import Agent -from rcrs_core.connection.connection import Connection - -class ComponentLauncher: - def __init__(self, port, host) -> None: - ... - - def make_connection(self) -> Connection: - ... - - def connect(self, agent: Agent, _request_id): # -> None: - ... - - def generate_request_ID(self): # -> int: - ... - - - diff --git a/typings/rcrs_core/connection/connection.pyi b/typings/rcrs_core/connection/connection.pyi deleted file mode 100644 index 33218b8..0000000 --- a/typings/rcrs_core/connection/connection.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class Connection: - def __init__(self, host, port) -> None: - ... - - def connect(self): # -> None: - ... - - def parseMessageFromKernel(self): # -> None: - ... - - def message_received(self, agent_message_received): # -> None: - ... - - def send_msg(self, msg): # -> None: - ... - - - diff --git a/typings/rcrs_core/connection/rcrs_encoding_utils.pyi b/typings/rcrs_core/connection/rcrs_encoding_utils.pyi deleted file mode 100644 index 947fc2b..0000000 --- a/typings/rcrs_core/connection/rcrs_encoding_utils.pyi +++ /dev/null @@ -1,20 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -""" utils to handle connection """ -def write_int32(value, sock): # -> None: - ... - -def readnbytes(sock, n): # -> Literal[b""]: - ... - -def read_int32(sock): # -> int: - ... - -def write_msg(msg, sock): # -> None: - ... - -def read_msg(sock): # -> MessageProto: - ... - diff --git a/typings/rcrs_core/entities/__init__.pyi b/typings/rcrs_core/entities/__init__.pyi deleted file mode 100644 index 006bc27..0000000 --- a/typings/rcrs_core/entities/__init__.pyi +++ /dev/null @@ -1,4 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - diff --git a/typings/rcrs_core/entities/ambulanceCenter.pyi b/typings/rcrs_core/entities/ambulanceCenter.pyi deleted file mode 100644 index bde5db4..0000000 --- a/typings/rcrs_core/entities/ambulanceCenter.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.building import Building - -class AmbulanceCentreEntity(Building): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def set_entity(self, properties): # -> None: - ... - - def get_entity_name(self): # -> Literal['Ambulance Centre']: - ... - - def copy_impl(self): # -> AmbulanceCentreEntity: - ... - - - diff --git a/typings/rcrs_core/entities/ambulanceTeam.pyi b/typings/rcrs_core/entities/ambulanceTeam.pyi deleted file mode 100644 index 5c42faa..0000000 --- a/typings/rcrs_core/entities/ambulanceTeam.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.human import Human - -class AmbulanceTeamEntity(Human): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def set_entity(self, properties): # -> None: - ... - - def get_entity_name(self): # -> Literal['Ambulance Team']: - ... - - def copy_impl(self): # -> AmbulanceTeamEntity: - ... - - - diff --git a/typings/rcrs_core/entities/area.pyi b/typings/rcrs_core/entities/area.pyi deleted file mode 100644 index b3f1559..0000000 --- a/typings/rcrs_core/entities/area.pyi +++ /dev/null @@ -1,52 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from typing import List -from rcrs_core.entities.entity import Entity - -class Area(Entity): - def __init__(self, entity_id) -> None: - ... - - def set_entity(self, properties): # -> None: - ... - - def get_location(self): # -> tuple[Any | None, Any | None]: - ... - - def get_neighbours(self): # -> list[Any]: - ... - - def get_edge_to(self, neighbour): # -> Entity | None: - ... - - def get_property(self, urn): # -> IntProperty | EdgeListProperty | EntityIDListProperty | None: - ... - - def get_edges_property(self): # -> EdgeListProperty: - ... - - def get_edges(self) -> List[Entity]: - ... - - def set_edges(self, value): # -> None: - ... - - def add_edge(self, edge): # -> None: - ... - - def get_blockades_property(self): # -> EntityIDListProperty: - ... - - def get_blockades(self): # -> None: - ... - - def set_blockades(self, value): # -> None: - ... - - def get_shape(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/blockade.pyi b/typings/rcrs_core/entities/blockade.pyi deleted file mode 100644 index a9b1502..0000000 --- a/typings/rcrs_core/entities/blockade.pyi +++ /dev/null @@ -1,73 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.entity import Entity - -class Blockade(Entity): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def get_entity_name(self): # -> Literal['Blockade']: - ... - - def set_entity(self, properties): # -> None: - ... - - def copy_impl(self): # -> Blockade: - ... - - def get_property(self, urn): # -> IntProperty | EntityIDProperty | IntArrayProperty | None: - ... - - def get_x_property(self): # -> IntProperty: - ... - - def get_x(self): # -> None: - ... - - def set_x(self, value): # -> None: - ... - - def get_y_property(self): # -> IntProperty: - ... - - def get_y(self): # -> None: - ... - - def set_y(self, value): # -> None: - ... - - def get_apexes_property(self): # -> IntArrayProperty: - ... - - def get_apexes(self): # -> None: - ... - - def set_apexes(self, value): # -> None: - ... - - def get_position_property(self): # -> EntityIDProperty: - ... - - def get_position(self): # -> None: - ... - - def set_position(self, value): # -> None: - ... - - def get_repaire_cost_property(self): # -> IntProperty: - ... - - def get_repaire_cost(self): # -> None: - ... - - def set_repaire_cost(self, value): # -> None: - ... - - def get_shape(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/building.pyi b/typings/rcrs_core/entities/building.pyi deleted file mode 100644 index edf477d..0000000 --- a/typings/rcrs_core/entities/building.pyi +++ /dev/null @@ -1,124 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.area import Area - -class Building(Area): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def is__fiery(self): # -> bool: - ... - - def set_entity(self, properties): # -> None: - ... - - def copy_impl(self): # -> Building: - ... - - def get_entity_name(self): # -> Literal['Building']: - ... - - def get_property(self, urn): # -> IntProperty | EdgeListProperty | EntityIDListProperty | None: - ... - - def get_floors_property(self): # -> IntProperty: - ... - - def get_floors(self): # -> None: - ... - - def set_floors(self, value): # -> None: - ... - - def get_ignition_property(self): # -> IntProperty: - ... - - def get_ignition(self): # -> None: - ... - - def set_ignition(self, value): # -> None: - ... - - def get_fieryness_property(self): # -> IntProperty: - ... - - def get_fieryness(self): # -> None: - ... - - def set_fieryness(self, value): # -> None: - ... - - def get_fieryness_enum(self): # -> None: - ... - - def get_brokenness_property(self): # -> IntProperty: - ... - - def get_brokenness(self): # -> None: - ... - - def set_brokenness(self, value): # -> None: - ... - - def get_building_code_property(self): # -> IntProperty: - ... - - def get_building_code(self): # -> None: - ... - - def set_building_code(self, value): # -> None: - ... - - def get_building_code_enum(self): # -> None: - ... - - def get_attributes_property(self): # -> IntProperty: - ... - - def get_attributes(self): # -> None: - ... - - def set_attributes(self, value): # -> None: - ... - - def get_ground_area_property(self): # -> IntProperty: - ... - - def get_ground_area(self): # -> None: - ... - - def set_ground_area(self, value): # -> None: - ... - - def get_total_area_property(self): # -> IntProperty: - ... - - def get_total_area(self): # -> None: - ... - - def set_total_area(self, value): # -> None: - ... - - def get_temperature_property(self): # -> IntProperty: - ... - - def get_temperature(self): # -> None: - ... - - def set_temperature(self, value): # -> None: - ... - - def get_importance_property(self): # -> IntProperty: - ... - - def get_importance(self): # -> None: - ... - - def set_importance(self, value): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/civilian.pyi b/typings/rcrs_core/entities/civilian.pyi deleted file mode 100644 index 95f8ee3..0000000 --- a/typings/rcrs_core/entities/civilian.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.human import Human - -class Civilian(Human): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def copy_impl(self): # -> Civilian: - ... - - def get_entity_name(self): # -> Literal['Civilian']: - ... - - def set_entity(self, properties): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/edge.pyi b/typings/rcrs_core/entities/edge.pyi deleted file mode 100644 index f35452b..0000000 --- a/typings/rcrs_core/entities/edge.pyi +++ /dev/null @@ -1,31 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class Edge: - def __init__(self, start_x, start_y, end_x, end_y, _neighbour) -> None: - ... - - def get_start_x(self): - ... - - def get_start_y(self): - ... - - def get_end_x(self): - ... - - def get_end_y(self): - ... - - def is_passable(self): # -> bool: - ... - - def get_neighbour(self): # -> None: - ... - - def to_string(self): - ... - - - diff --git a/typings/rcrs_core/entities/entity.pyi b/typings/rcrs_core/entities/entity.pyi deleted file mode 100644 index 1ebb7aa..0000000 --- a/typings/rcrs_core/entities/entity.pyi +++ /dev/null @@ -1,61 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.worldmodel.entityID import EntityID -from abc import ABC - -class Entity(ABC): - def __init__(self, _entity_id) -> None: - ... - - def set_entity(self, properties): # -> None: - ... - - def get_id(self) -> EntityID: - ... - - def get_properties(self): # -> dict[Any, Any]: - ... - - def register_properties(self, _properties): # -> None: - ... - - def copy_impl(self): # -> Entity: - ... - - def copy(self): # -> Entity: - ... - - def get_urn(self): - ... - - def __hash__(self) -> int: - ... - - def get_property(self, property_urn): # -> None: - ... - - def get_location(self): # -> tuple[Any | None, Any | None] | tuple[None, None]: - ... - - def get_x_property(self): # -> IntProperty: - ... - - def get_x(self): # -> None: - ... - - def set_x(self, value): # -> None: - ... - - def get_y_property(self): # -> IntProperty: - ... - - def get_y(self): # -> None: - ... - - def set_y(self, value): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/fireBrigade.pyi b/typings/rcrs_core/entities/fireBrigade.pyi deleted file mode 100644 index 850dc4d..0000000 --- a/typings/rcrs_core/entities/fireBrigade.pyi +++ /dev/null @@ -1,34 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.human import Human - -class FireBrigadeEntity(Human): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def set_entity(self, properties): # -> None: - ... - - def copy_impl(self): # -> FireBrigadeEntity: - ... - - def get_entity_name(self): # -> Literal['Fire brigade']: - ... - - def get_property(self, urn): # -> IntProperty | EntityIDProperty | IntArrayProperty | None: - ... - - def get_water_property(self): # -> IntProperty: - ... - - def get_water(self): # -> None: - ... - - def set_water(self, value): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/fireStation.pyi b/typings/rcrs_core/entities/fireStation.pyi deleted file mode 100644 index 46a7b74..0000000 --- a/typings/rcrs_core/entities/fireStation.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.building import Building - -class FireStationEntity(Building): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def copy_impl(self): # -> FireStationEntity: - ... - - def get_entity_name(self): # -> Literal['Fire Station']: - ... - - def set_entity(self, properties): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/gassStation.pyi b/typings/rcrs_core/entities/gassStation.pyi deleted file mode 100644 index 71aa885..0000000 --- a/typings/rcrs_core/entities/gassStation.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.building import Building - -class GasStation(Building): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def copy_impl(self): # -> GasStation: - ... - - def get_entity_name(self): # -> Literal['Gass Station']: - ... - - def set_entity(self, properties): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/human.pyi b/typings/rcrs_core/entities/human.pyi deleted file mode 100644 index aebfd77..0000000 --- a/typings/rcrs_core/entities/human.pyi +++ /dev/null @@ -1,114 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.entity import Entity -from rcrs_core.properties.entityIDProperty import EntityIDProperty -from rcrs_core.worldmodel.entityID import EntityID -from rcrs_core.worldmodel.worldmodel import WorldModel - -class Human(Entity): - def __init__(self, entity_id) -> None: - ... - - def get_location(self, world_model: WorldModel): # -> None: - ... - - def set_entity(self, properties: dict): # -> None: - ... - - def get_property(self, urn): # -> EntityIDProperty | IntArrayProperty | IntProperty | None: - ... - - def get_x_property(self): # -> IntProperty: - ... - - def get_x(self) -> int: - ... - - def set_x(self, value) -> None: - ... - - def get_y_property(self): # -> IntProperty: - ... - - def get_y(self): # -> None: - ... - - def set_y(self, value): # -> None: - ... - - def get_position_property(self) -> EntityIDProperty: - ... - - def get_position(self) -> EntityID: - ... - - def set_position(self, value): # -> None: - ... - - def get_position_history_property(self): # -> IntArrayProperty: - ... - - def get_position_history(self): # -> None: - ... - - def set_position_history(self, value): # -> None: - ... - - def get_direction_property(self): # -> IntProperty: - ... - - def get_direction(self): # -> None: - ... - - def set_direction(self, value): # -> None: - ... - - def get_stamina_property(self): # -> IntProperty: - ... - - def get_stamina(self): # -> None: - ... - - def set_stamina(self, value): # -> None: - ... - - def get_hp_property(self): # -> IntProperty: - ... - - def get_hp(self): # -> None: - ... - - def set_hp(self, value): # -> None: - ... - - def get_damage_property(self): # -> IntProperty: - ... - - def get_damage(self): # -> None: - ... - - def set_damage(self, value): # -> None: - ... - - def get_buriedness_property(self): # -> IntProperty: - ... - - def get_buriedness(self): # -> None: - ... - - def set_buriedness(self, value): # -> None: - ... - - def get_travel_distance_property(self): # -> IntProperty: - ... - - def get_travel_distance(self): # -> None: - ... - - def set_travel_distance(self, value): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/hydrant.pyi b/typings/rcrs_core/entities/hydrant.pyi deleted file mode 100644 index 4d564cd..0000000 --- a/typings/rcrs_core/entities/hydrant.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.road import Road - -class Hydrant(Road): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def copy_impl(self): # -> Hydrant: - ... - - def get_entity_name(self): # -> Literal['Hydrant']: - ... - - def set_entity(self, properties): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/policeForce.pyi b/typings/rcrs_core/entities/policeForce.pyi deleted file mode 100644 index 1ee42f3..0000000 --- a/typings/rcrs_core/entities/policeForce.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.human import Human - -class PoliceForceEntity(Human): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def copy_impl(self): # -> PoliceForceEntity: - ... - - def get_entity_name(self): # -> Literal['Police force']: - ... - - def set_entity(self, properties): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/policeOffice.pyi b/typings/rcrs_core/entities/policeOffice.pyi deleted file mode 100644 index f060e59..0000000 --- a/typings/rcrs_core/entities/policeOffice.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.building import Building - -class PoliceOfficeEntity(Building): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def copy_impl(self): # -> PoliceOfficeEntity: - ... - - def get_entity_name(self): # -> Literal['Police office']: - ... - - def set_entity(self, properties): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/refuge.pyi b/typings/rcrs_core/entities/refuge.pyi deleted file mode 100644 index 17ff0ed..0000000 --- a/typings/rcrs_core/entities/refuge.pyi +++ /dev/null @@ -1,61 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.building import Building - -class Refuge(Building): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def set_entity(self, properties): # -> None: - ... - - def copy_impl(self): # -> Refuge: - ... - - def get_entity_name(self) -> str: - ... - - def get_property(self, urn): # -> IntProperty | EdgeListProperty | EntityIDListProperty | None: - ... - - def get_bed_capacity_property(self): # -> IntProperty: - ... - - def get_bed_capacity(self): # -> None: - ... - - def set_bed_capacity(self, value): # -> None: - ... - - def get_occupied_beds_property(self): # -> IntProperty: - ... - - def get_occupied_beds(self): # -> None: - ... - - def set_occupied_beds(self, value): # -> None: - ... - - def get_refill_capacity_property(self): # -> IntProperty: - ... - - def get_refill_capacity(self): # -> None: - ... - - def set_refill_capacity(self, value): # -> None: - ... - - def get_waiting_list_size_property(self): # -> IntProperty: - ... - - def get_waiting_list_size(self): # -> None: - ... - - def set_waiting_list_size(self, value): # -> None: - ... - - - diff --git a/typings/rcrs_core/entities/road.pyi b/typings/rcrs_core/entities/road.pyi deleted file mode 100644 index 2ceaf0d..0000000 --- a/typings/rcrs_core/entities/road.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.entities.area import Area - -class Road(Area): - urn = ... - def __init__(self, entity_id) -> None: - ... - - def set_entity(self, properties) -> None: - ... - - def get_entity_name(self) -> str: - ... - - def copy_impl(self): # -> Road: - ... - - - diff --git a/typings/rcrs_core/entities/standardEntityFactory.pyi b/typings/rcrs_core/entities/standardEntityFactory.pyi deleted file mode 100644 index 651fc76..0000000 --- a/typings/rcrs_core/entities/standardEntityFactory.pyi +++ /dev/null @@ -1,17 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class StandardEntityFactory: - _instance = ... - def __new__(cls): # -> Self: - ... - - def __init__(self) -> None: - ... - - def make_entity(urn, id): # -> Building | Refuge | Road | Blockade | PoliceForceEntity | PoliceOfficeEntity | AmbulanceTeamEntity | AmbulanceCentreEntity | FireBrigadeEntity | FireStationEntity | Civilian | Hydrant | GasStation | None: - ... - - - diff --git a/typings/rcrs_core/log/logger.pyi b/typings/rcrs_core/log/logger.pyi deleted file mode 100644 index bec3fd5..0000000 --- a/typings/rcrs_core/log/logger.pyi +++ /dev/null @@ -1,28 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class Logger: - def __init__(self, _name, agent_id=...) -> None: - ... - - def info(self, msg): # -> None: - ... - - def debug(self, msg): # -> None: - ... - - def error(self, msg): # -> None: - ... - - def warning(self, msg): # -> None: - ... - - def warn(self, msg): # -> None: - ... - - def set_id(self, id): # -> None: - ... - - - diff --git a/typings/rcrs_core/messages/AKAcknowledge.pyi b/typings/rcrs_core/messages/AKAcknowledge.pyi deleted file mode 100644 index 1bea0e2..0000000 --- a/typings/rcrs_core/messages/AKAcknowledge.pyi +++ /dev/null @@ -1,18 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.messages.message import Message - -class AKAcknowledge(Message): - def __init__(self) -> None: - ... - - def write(self, request_id, agent_id): # -> MessageProto: - ... - - def read(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/messages/AKConnect.pyi b/typings/rcrs_core/messages/AKConnect.pyi deleted file mode 100644 index 2c23c18..0000000 --- a/typings/rcrs_core/messages/AKConnect.pyi +++ /dev/null @@ -1,18 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.messages.message import Message - -class AKConnect(Message): - def __init__(self) -> None: - ... - - def write(self, request_id, agent): # -> MessageProto: - ... - - def read(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/messages/KAConnectError.pyi b/typings/rcrs_core/messages/KAConnectError.pyi deleted file mode 100644 index 4fa547d..0000000 --- a/typings/rcrs_core/messages/KAConnectError.pyi +++ /dev/null @@ -1,18 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.messages.message import Message - -class KAConnectError(Message): - def __init__(self, data) -> None: - ... - - def read(self): # -> None: - ... - - def write(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/messages/KAConnectOK.pyi b/typings/rcrs_core/messages/KAConnectOK.pyi deleted file mode 100644 index 100e853..0000000 --- a/typings/rcrs_core/messages/KAConnectOK.pyi +++ /dev/null @@ -1,18 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.messages.message import Message - -class KAConnectOK(Message): - def __init__(self, data) -> None: - ... - - def read(self): # -> None: - ... - - def write(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/messages/KASense.pyi b/typings/rcrs_core/messages/KASense.pyi deleted file mode 100644 index 163ac38..0000000 --- a/typings/rcrs_core/messages/KASense.pyi +++ /dev/null @@ -1,19 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.connection import RCRSProto_pb2 -from rcrs_core.messages.message import Message - -class KASense(Message): - def __init__(self, data: RCRSProto_pb2) -> None: - ... - - def read(self): # -> None: - ... - - def write(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/messages/Shutdown.pyi b/typings/rcrs_core/messages/Shutdown.pyi deleted file mode 100644 index e5a6335..0000000 --- a/typings/rcrs_core/messages/Shutdown.pyi +++ /dev/null @@ -1,18 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.messages.message import Message - -class Shutdown(Message): - def __init__(self, data) -> None: - ... - - def read(self): # -> None: - ... - - def write(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/messages/controlMessageFactory.pyi b/typings/rcrs_core/messages/controlMessageFactory.pyi deleted file mode 100644 index a3e403b..0000000 --- a/typings/rcrs_core/messages/controlMessageFactory.pyi +++ /dev/null @@ -1,13 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class ControlMessageFactory: - def __init__(self) -> None: - ... - - def make_message(self, msg): # -> KASense | KAConnectOK | KAConnectError | Shutdown | None: - ... - - - diff --git a/typings/rcrs_core/messages/message.pyi b/typings/rcrs_core/messages/message.pyi deleted file mode 100644 index 005a916..0000000 --- a/typings/rcrs_core/messages/message.pyi +++ /dev/null @@ -1,23 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from abc import ABC, abstractmethod - -class Message(ABC): - def __init__(self, _urn) -> None: - ... - - @abstractmethod - def read(self): # -> None: - ... - - @abstractmethod - def write(self): # -> None: - ... - - def get_urn(self): # -> Any: - ... - - - diff --git a/typings/rcrs_core/properties/booleanProperty.pyi b/typings/rcrs_core/properties/booleanProperty.pyi deleted file mode 100644 index 7bd816f..0000000 --- a/typings/rcrs_core/properties/booleanProperty.pyi +++ /dev/null @@ -1,21 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.properties.property import Property - -class BooleanProperty(Property): - def __init__(self, urn) -> None: - ... - - def set_fields(self, value): # -> None: - ... - - def copy(self): # -> BooleanProperty: - ... - - def take_value(self, _property): # -> None: - ... - - - diff --git a/typings/rcrs_core/properties/edgeListProperty.pyi b/typings/rcrs_core/properties/edgeListProperty.pyi deleted file mode 100644 index 2af9f10..0000000 --- a/typings/rcrs_core/properties/edgeListProperty.pyi +++ /dev/null @@ -1,38 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from typing import List -from rcrs_core.properties.property import Property -from rcrs_core.entities.edge import Edge - -class EdgeListProperty(Property): - def __init__(self, urn) -> None: - ... - - def get_fields(self): # -> None: - ... - - def set_fields(self, data): # -> None: - ... - - def set_value(self, _value: List[Edge]): # -> None: - ... - - def set_edges(self, _edges: List[Edge]): # -> None: - ... - - def add_edge(self, _edge): # -> None: - ... - - def clear_edges(self): # -> None: - ... - - def take_value(self, _value): # -> None: - ... - - def copy(self): # -> EdgeListProperty: - ... - - - diff --git a/typings/rcrs_core/properties/entityIDListProperty.pyi b/typings/rcrs_core/properties/entityIDListProperty.pyi deleted file mode 100644 index 9f629f1..0000000 --- a/typings/rcrs_core/properties/entityIDListProperty.pyi +++ /dev/null @@ -1,26 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.properties.property import Property -from rcrs_core.worldmodel.entityID import EntityID -from typing import List - -class EntityIDListProperty(Property): - def __init__(self, urn) -> None: - ... - - def set_fields(self, data): # -> None: - ... - - def set_value(self, _value: List[EntityID]): # -> None: - ... - - def copy(self): # -> EntityIDListProperty: - ... - - def take_value(self, _property): # -> None: - ... - - - diff --git a/typings/rcrs_core/properties/entityIDProperty.pyi b/typings/rcrs_core/properties/entityIDProperty.pyi deleted file mode 100644 index da7b261..0000000 --- a/typings/rcrs_core/properties/entityIDProperty.pyi +++ /dev/null @@ -1,21 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.properties.property import Property - -class EntityIDProperty(Property): - def __init__(self, urn) -> None: - ... - - def set_fields(self, value): # -> None: - ... - - def copy(self): # -> EntityIDProperty: - ... - - def take_value(self, _property): # -> None: - ... - - - diff --git a/typings/rcrs_core/properties/intArrayProperty.pyi b/typings/rcrs_core/properties/intArrayProperty.pyi deleted file mode 100644 index a4c7a6c..0000000 --- a/typings/rcrs_core/properties/intArrayProperty.pyi +++ /dev/null @@ -1,24 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.properties.property import Property - -class IntArrayProperty(Property): - def __init__(self, urn) -> None: - ... - - def set_fields(self, data): # -> None: - ... - - def set_value(self, data): # -> None: - ... - - def copy(self): # -> IntArrayProperty: - ... - - def take_value(self, _property): # -> None: - ... - - - diff --git a/typings/rcrs_core/properties/intProperty.pyi b/typings/rcrs_core/properties/intProperty.pyi deleted file mode 100644 index 83cf474..0000000 --- a/typings/rcrs_core/properties/intProperty.pyi +++ /dev/null @@ -1,21 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from rcrs_core.properties.property import Property - -class IntProperty(Property): - def __init__(self, urn) -> None: - ... - - def set_fields(self, value): # -> None: - ... - - def copy(self): # -> IntProperty: - ... - - def take_value(self, _property): # -> None: - ... - - - diff --git a/typings/rcrs_core/properties/property.pyi b/typings/rcrs_core/properties/property.pyi deleted file mode 100644 index cbff460..0000000 --- a/typings/rcrs_core/properties/property.pyi +++ /dev/null @@ -1,22 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class Property: - def __init__(self, _urn) -> None: - ... - - def get_urn(self): # -> Any: - ... - - def get_value(self): # -> None: - ... - - def set_value(self, _value): # -> None: - ... - - def to_string(self): # -> None: - ... - - - diff --git a/typings/rcrs_core/properties/standardPropertyFactory.pyi b/typings/rcrs_core/properties/standardPropertyFactory.pyi deleted file mode 100644 index be77e22..0000000 --- a/typings/rcrs_core/properties/standardPropertyFactory.pyi +++ /dev/null @@ -1,13 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class StandardPropertyFactory: - def __init__(self) -> None: - ... - - def make_property(urn): # -> IntProperty | IntArrayProperty | BooleanProperty | EntityIDProperty | EntityIDListProperty | EdgeListProperty | None: - ... - - - diff --git a/typings/rcrs_core/worldmodel/changeSet.pyi b/typings/rcrs_core/worldmodel/changeSet.pyi deleted file mode 100644 index d1788f1..0000000 --- a/typings/rcrs_core/worldmodel/changeSet.pyi +++ /dev/null @@ -1,34 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class ChangeSet: - def __init__(self, _change_set=...) -> None: - ... - - def add_change(self, entity_id, entity_urn, property): # -> None: - ... - - def entity_deleted(self, _entity_id): # -> None: - ... - - def get_changed_properties(self, _entity_id): # -> list[Any]: - ... - - def get_changed_property(self, _entity_id, prop_urn): # -> None: - ... - - def get_changed_entities(self): # -> dict_keys[Any, Any]: - ... - - def get_deleted_entities(self): # -> set[Any]: - ... - - def get_entity_urn(self, _entity_id): # -> None: - ... - - def merge(self, _change_set): # -> None: - ... - - - diff --git a/typings/rcrs_core/worldmodel/entityID.pyi b/typings/rcrs_core/worldmodel/entityID.pyi deleted file mode 100644 index e570b8e..0000000 --- a/typings/rcrs_core/worldmodel/entityID.pyi +++ /dev/null @@ -1,25 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -class EntityID: - def __init__(self, id: int) -> None: - ... - - def __eq__(self, other) -> bool: - ... - - def __hash__(self) -> int: - ... - - def get_value(self): # -> | int: - ... - - def __str__(self) -> str: - ... - - def equals(self, o): # -> bool: - ... - - - diff --git a/typings/rcrs_core/worldmodel/worldmodel.pyi b/typings/rcrs_core/worldmodel/worldmodel.pyi deleted file mode 100644 index e9ccce0..0000000 --- a/typings/rcrs_core/worldmodel/worldmodel.pyi +++ /dev/null @@ -1,51 +0,0 @@ -""" -This type stub file was generated by pyright. -""" - -from typing import List -from rcrs_core.entities.entity import Entity -from rcrs_core.worldmodel.changeSet import ChangeSet -from rcrs_core.worldmodel.entityID import EntityID - -class WorldModel: - def __init__(self) -> None: - ... - - def add_entities(self, entities: List[Entity]): # -> None: - ... - - def get_entity(self, entity_id: EntityID) -> Entity: - ... - - def add_entity(self, entity): # -> None: - ... - - def remove_entity(self, entity_id): # -> None: - ... - - def get_entities(self): # -> dict_values[Any, Any]: - ... - - def merge(self, change_set: ChangeSet): # -> None: - ... - - def index_entities(self): # -> None: - ... - - def make_rectangle(self, entity): # -> tuple[None, None, None, None] | tuple[float, float, float, float]: - ... - - def get_rect_bounds(self): # -> tuple[Any, Any, Any, Any]: - ... - - def get_world_bounds(self): # -> tuple[Any, Any, Any, Any]: - ... - - def get_objects_in_rectangle(self, x1, y1, x2, y2): # -> list[Any]: - ... - - def get_objects_in_range(self, entity, range): # -> list[Any]: - ... - - - From 17f6f34db1b57be641bbf175edf9bcca4753170b Mon Sep 17 00:00:00 2001 From: shima004 Date: Wed, 2 Oct 2024 14:10:09 +0900 Subject: [PATCH 3/5] refactor: Remove unused Agent class and related imports --- adf_core_python/core/agent/agent.py | 30 ----------------------------- 1 file changed, 30 deletions(-) delete mode 100644 adf_core_python/core/agent/agent.py diff --git a/adf_core_python/core/agent/agent.py b/adf_core_python/core/agent/agent.py deleted file mode 100644 index ae19b1e..0000000 --- a/adf_core_python/core/agent/agent.py +++ /dev/null @@ -1,30 +0,0 @@ -from rcrs_core.entities.entity import Entity - -from adf_core_python.core.agent.config.module_config import ModuleConfig -from adf_core_python.core.agent.develop.develop_data import DevelopData -from adf_core_python.core.agent.info.scenario_info import Mode -from adf_core_python.core.agent.precompute.precompute_data import PrecomputeData - - -class Agent(Entity): - def __init__( - self, - team_name: str, - is_precompute: bool, - is_debug: bool, - data_storage_name: str, - module_config: ModuleConfig, - develop_data: DevelopData, - ): - self._tema_name = team_name - self._is_precompute = is_precompute - self._is_debug = is_debug - - if self._is_precompute: - # PrecomputeData.remove_data(data_storage_name) - self._mode = Mode.PRECOMPUTATION - - self._module_config: ModuleConfig = module_config - self._develop_data: DevelopData = develop_data - self._precompute_data: PrecomputeData = PrecomputeData(data_storage_name) - self._message_manager = None From cdea89918768af565958319935313d140aa7166f Mon Sep 17 00:00:00 2001 From: shima004 Date: Wed, 2 Oct 2024 14:10:23 +0900 Subject: [PATCH 4/5] feat: Add pyrightconfig.json for standard type checking mode and exclude site-packages --- pyrightconfig.json | 4 ++++ 1 file changed, 4 insertions(+) create mode 100644 pyrightconfig.json diff --git a/pyrightconfig.json b/pyrightconfig.json new file mode 100644 index 0000000..8a1eb70 --- /dev/null +++ b/pyrightconfig.json @@ -0,0 +1,4 @@ +{ + "typeCheckingMode": "standard", + "exclude": ["**/site-packages"] +} From 2b778558e1d20ab7f5136b8b0d12b5d7dcbdb274 Mon Sep 17 00:00:00 2001 From: shima004 Date: Wed, 2 Oct 2024 14:10:40 +0900 Subject: [PATCH 5/5] feat: fix type --- .../core/agent/config/module_config.py | 5 ++-- adf_core_python/core/agent/info/agent_info.py | 7 +++-- .../tactics/default_tactics_ambulance_team.py | 30 ++++++++++--------- .../tactics/default_tactics_fire_brigade.py | 30 ++++++++++--------- .../tactics/default_tactics_police_force.py | 30 ++++++++++--------- 5 files changed, 56 insertions(+), 46 deletions(-) diff --git a/adf_core_python/core/agent/config/module_config.py b/adf_core_python/core/agent/config/module_config.py index 095cb50..72a292a 100644 --- a/adf_core_python/core/agent/config/module_config.py +++ b/adf_core_python/core/agent/config/module_config.py @@ -79,11 +79,12 @@ def _flatten( dict[str, Any] Flattened dictionary """ - flatten_data = {} + flatten_data: dict[str, Any] = {} for key, value in data.items(): new_key = f"{parent_key}{sep}{key}" if parent_key else key if isinstance(value, dict): - flatten_data.update(self._flatten(value, new_key, sep=sep)) + v: dict[str, Any] = value + flatten_data.update(self._flatten(v, new_key, sep=sep)) else: flatten_data[new_key] = value return flatten_data diff --git a/adf_core_python/core/agent/info/agent_info.py b/adf_core_python/core/agent/info/agent_info.py index d65070c..030de23 100644 --- a/adf_core_python/core/agent/info/agent_info.py +++ b/adf_core_python/core/agent/info/agent_info.py @@ -2,8 +2,11 @@ from typing import Any from rcrs_core.agents.agent import Agent +from rcrs_core.entities.entity import Entity from rcrs_core.entities.human import Human -from rcrs_core.worldmodel.worldmodel import ChangeSet, Entity, EntityID, WorldModel +from rcrs_core.worldmodel.changeSet import ChangeSet +from rcrs_core.worldmodel.entityID import EntityID +from rcrs_core.worldmodel.worldmodel import WorldModel from adf_core_python.core.agent.action.action import Action @@ -97,7 +100,7 @@ def get_position_entity_id(self) -> EntityID: """ entity = self._world_model.get_entity(self.get_entity_id()) if isinstance(entity, Human): - return entity.get_position_property() + return entity.get_position() else: return entity.get_id() diff --git a/adf_core_python/implement/tactics/default_tactics_ambulance_team.py b/adf_core_python/implement/tactics/default_tactics_ambulance_team.py index 45c7707..8666e14 100644 --- a/adf_core_python/implement/tactics/default_tactics_ambulance_team.py +++ b/adf_core_python/implement/tactics/default_tactics_ambulance_team.py @@ -111,21 +111,23 @@ def think( entity_id = agent_info.get_entity_id() # noqa: F841 target_entity_id = self._human_detector.calculate().get_target_entity_id() - action = ( - self._action_transport.set_target_entity_id(target_entity_id) - .calc() - .get_action() - ) - if action is not None: - return action + if target_entity_id is not None: + action = ( + self._action_transport.set_target_entity_id(target_entity_id) + .calc() + .get_action() + ) + if action is not None: + return action target_entity_id = self._search.calculate().get_target_entity_id() - action = ( - self._action_ext_move.set_target_entity_id(target_entity_id) - .calc() - .get_action() - ) - if action is not None: - return action + if target_entity_id is not None: + action = ( + self._action_ext_move.set_target_entity_id(target_entity_id) + .calc() + .get_action() + ) + if action is not None: + return action return ActionRest() diff --git a/adf_core_python/implement/tactics/default_tactics_fire_brigade.py b/adf_core_python/implement/tactics/default_tactics_fire_brigade.py index 11b38af..2af3378 100644 --- a/adf_core_python/implement/tactics/default_tactics_fire_brigade.py +++ b/adf_core_python/implement/tactics/default_tactics_fire_brigade.py @@ -110,21 +110,23 @@ def think( entity_id = agent_info.get_entity_id() # noqa: F841 target_entity_id = self._human_detector.calculate().get_target_entity_id() - action = ( - self._action_fire_rescue.set_target_entity_id(target_entity_id) - .calc() - .get_action() - ) - if action is not None: - return action + if target_entity_id is not None: + action = ( + self._action_fire_rescue.set_target_entity_id(target_entity_id) + .calc() + .get_action() + ) + if action is not None: + return action target_entity_id = self._search.calculate().get_target_entity_id() - action = ( - self._action_ext_move.set_target_entity_id(target_entity_id) - .calc() - .get_action() - ) - if action is not None: - return action + if target_entity_id is not None: + action = ( + self._action_ext_move.set_target_entity_id(target_entity_id) + .calc() + .get_action() + ) + if action is not None: + return action return ActionRest() diff --git a/adf_core_python/implement/tactics/default_tactics_police_force.py b/adf_core_python/implement/tactics/default_tactics_police_force.py index bd4f29d..3253aeb 100644 --- a/adf_core_python/implement/tactics/default_tactics_police_force.py +++ b/adf_core_python/implement/tactics/default_tactics_police_force.py @@ -114,21 +114,23 @@ def think( entity_id = agent_info.get_entity_id() # noqa: F841 target_entity_id = self._road_detector.calculate().get_target_entity_id() - action = ( - self._action_ext_clear.set_target_entity_id(target_entity_id) - .calc() - .get_action() - ) - if action is not None: - return action + if target_entity_id is not None: + action = ( + self._action_ext_clear.set_target_entity_id(target_entity_id) + .calc() + .get_action() + ) + if action is not None: + return action target_entity_id = self._search.calculate().get_target_entity_id() - action = ( - self._action_ext_clear.set_target_entity_id(target_entity_id) - .calc() - .get_action() - ) - if action is not None: - return action + if target_entity_id is not None: + action = ( + self._action_ext_clear.set_target_entity_id(target_entity_id) + .calc() + .get_action() + ) + if action is not None: + return action return ActionRest()