sc2analyzer 0.1.0a1__tar.gz

sc2analyzer-0.1.0a1/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: sc2analyzer
+Version: 0.1.0a1
+Requires-Dist: numpy>=1.26
+Requires-Dist: build
+Requires-Dist: twine
+Requires-Dist: pytest
+Requires-Dist: sc2reader>=1.8
+Requires-Dist: pandas
+Requires-Dist: matplotlib

sc2analyzer-0.1.0a1/README.md

@@ -0,0 +1,3 @@
+# CO-OP games
+## Links
+[s2protocol code](https://github.com/Blizzard/s2protocol)

sc2analyzer-0.1.0a1/pyproject.toml

@@ -0,0 +1,25 @@
+[build-system]
+requires = ["setuptools>=61"]
+build-backend = "setuptools.build_meta"
+
+
+[project]
+name = "sc2analyzer"
+version = "0.1.0a1"
+
+dependencies = [
+    #"pandas>=2.2",
+    "numpy>=1.26",
+    "build",
+    "twine",
+    "pytest",
+    "sc2reader>=1.8",
+    "pandas",
+    "matplotlib"
+]
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-dir]
+"" = "src"

sc2analyzer-0.1.0a1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

sc2analyzer-0.1.0a1/src/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

sc2analyzer-0.1.0a1/src/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"
+}

sc2analyzer-0.1.0a1/src/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)

sc2analyzer-0.1.0a1/src/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]
+

sc2analyzer-0.1.0a1/src/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)