sc2analyzer 0.1.0a1__py3-none-any.whl

co_op/event.py

@@ -0,0 +1,55 @@
+from sc2reader.events import Event
+
+
+class CoOpEvent:
+    PLAYER_EVENT = "PlayerEvent"
+    GAME_EVENT = "GameEvent"
+    UNITBORN_EVENT = "UnitBornEvent"
+    def __init__(self, events:list[Event]):
+        self.events = events
+        self.event_types = self.get_event_types()
+        self.players_events = self.arrange_events()
+
+    def get_event_types(self)->dict[str, set[str]]:
+        game_events = set()
+        player_events = set()
+        for event in self.events:
+            value = getattr(event, "control_pid", None)
+            if value == None:
+                game_events.add(event.name)
+            else:
+                player_events.add(event.name)
+
+        types = dict()
+        types[self.GAME_EVENT] = game_events
+        types[self.PLAYER_EVENT] = player_events
+
+        return types
+
+    def arrange_events(self)->dict[int, list]:
+        events = {}
+        for event in self.events:
+            if event.name in self.event_types[self.PLAYER_EVENT]:
+                id = event.control_pid
+                event_list = []
+                if id in events:
+                    event_list = events[id]
+                event_list.append(event)
+                events[id] = event_list
+
+        return events
+    def get_unit_born_events(self):
+        events = list[Event]()
+        for event in self.events:
+            if event.name == self.UNITBORN_EVENT:
+                events.append(event)
+
+        return events
+
+    def get_unit_born_events_for_player(self, id:int)->list[Event]:
+        events = list[Event]()
+        for event in self.players_events[id]:
+            if event.name == self.UNITBORN_EVENT:
+                events.append(event)
+
+        return events

co_op/models/__init__.py

@@ -0,0 +1,88 @@
+HEROS = {
+    "Dehaka",
+    "Tychus"
+}
+
+COMMANDER_POWER_UNITS = {
+    "TychusCoop",
+    "TychusWarhound",
+    "TychusMedic",
+    "TychusHERC",
+    "TychusFirebat",
+
+    "DehakaCoop",
+    "DehakaCoopClone",
+    "DehakaMurvar",
+    "DehakaDakrun",
+    "DehakaGlevig",
+
+    #aba
+    "Brutalisk",
+    "Leviathan",
+}
+
+POWER_VALUES = {
+    "DehakaCoop": 1000,
+    "DehakaCoopClone": 300,
+    "DehakaMurvar": 800,
+    "DehakaDakrun": 800,
+    "DehakaGlevig": 800,
+
+    "TychusCoop": 1000,
+    "TychusWarhound": 800,
+    "TychusMedic": 700,
+    "TychusHERC": 700,
+    "TychusFirebat": 800,
+}
+
+HERO_IGNOR = [
+    "DehakaTrainEggRoach",
+    "DehakaTrainEggDrone",
+    "DehakaDrone",
+    "DehakaHatchery",
+    "DehakaCoopReviveCocoonFootPrint",
+    "DehakaTrainEggHydralisk",
+    "TychusSCV",
+    "TychusCommandCenter",
+    "TychusResearchCenter",
+    "TychusWarhoundAutoTurret",
+    "TychusResearchCenterUnlocked",
+    "TychusResearchCenterLocked",
+    "SupplyDepotLowered",
+    "Egg",
+    "SiegeTankSieged",
+    "Larva",
+    "ToxicNestBurrowed",
+
+    #"BiomassPickup",
+]
+DEHAKA = {
+    "DehakaTrainEggRoach",
+    "DehakaTrainEggDrone",
+    "DehakaDrone",
+    "DehakaCreeperFlying",
+}
+TYCHUS_OUTLAWS = {
+    "TychusMedic",
+    "TychusSCV",
+    "TychusWarhoundAutoTurret",
+    "TychusResearchCenterUnlocked",
+    "TychusResearchCenterLocked",
+    "TychusHERC",
+    "TychusOdin",
+    "TychusCommandCenter",
+    "TychusResearchCenter"
+    "CoopCasterTychus"
+    "TychusCoop",
+    "TychusWarhound",
+    "TychusFirebat",
+    "TychusHERC"
+    "Tychus",
+    "Sirius",
+    "Nux",
+    "Sam",
+    "Blaze",
+    "Rattlesnake",
+    "Vega",
+    "Cannonball"
+}

co_op/models/commander.py

@@ -0,0 +1,8 @@
+from dataclasses import dataclass, field
+
+@dataclass
+class Commander:
+    unit_id: int
+    owner: int
+    name: str
+    units_ids: set[int] = field(default_factory=set)

co_op/models/player.py

@@ -0,0 +1,13 @@
+from dataclasses import dataclass
+
+from sc2reader.events import Event
+
+
+@dataclass
+class Player:
+    id: int
+    name: str
+    commander: str
+    race: str
+    events: list[Event]
+

co_op/models/unit.py

@@ -0,0 +1,25 @@
+from dataclasses import dataclass, field
+
+
+@dataclass
+class Unit:
+    unit_id: int
+    owner: int
+
+    current_type: str
+
+    # TODO: refact to is_commander_ability
+    commander_ability = False
+
+    is_army = False
+    is_building = False
+    is_worker = False
+    is_temporary = False
+    is_commander_unit = False
+    is_commander = False
+
+
+    birth_frame: int
+    death_frame: int | None = None
+
+    type_history: list = field(default_factory=list)