shortdeck1.0

shortdeck_arena/simulation.py

@@ -0,0 +1,65 @@
+from __future__ import annotations
+
+import json
+import random
+from pathlib import Path
+from typing import List, Dict, Optional
+
+from typing import TYPE_CHECKING
+if TYPE_CHECKING:
+    from .agent import Agent
+from .card import Card
+
+
+class Simulation:
+    def __init__(self, agents: List[Agent]):
+        self.agents = agents
+        self.history: List[Dict] = []
+        self.cards: List[Card] = []
+        self.saved = False
+        self.new_round()
+
+    def new_round(self):
+        self.history = []
+        self.cards = Card.all_short()
+        random.shuffle(self.cards)
+        self.saved = False
+
+    def player_cards(self, pid: int) -> List[Card]:
+        return self.cards[pid * 2 : pid * 2 + 2]
+
+    def board_cards(self, street: str) -> List[Card]:
+        nplayers = len(self.agents)
+        idx_start = nplayers * 2
+        if street == "flop":
+            return self.cards[idx_start: idx_start + 3]
+        if street == "turn":
+            return self.cards[idx_start: idx_start + 4]
+        if street == "river":
+            return self.cards[idx_start: idx_start + 5]
+        return []
+
+    def node_info(self) -> Dict:
+        return {"bet_min": 1, "bet_max": 100}
+
+    def apply_action(self, pid: int, action: str, amount: Optional[int] = None):
+        self.history.append({"pid": pid, "action": action, "amount": amount})
+
+    def to_save_data(self) -> Dict:
+        players = [f"Agent{a.pid}" for a in self.agents]
+        return {
+            "history": self.history,
+            "players": players,
+            "player_cards": ["".join(str(c) for c in self.player_cards(i)) for i in range(len(self.agents))],
+            "board": "".join(str(c) for c in self.board_cards("river")),
+        }
+
+    def dump_data(self, path: Path | None = None):
+        if self.saved:
+            return
+        if path is None:
+            path = Path.cwd() / "shortdeck_arena_history.jsonl"
+        with path.open("a", encoding="utf-8") as f:
+            f.write(json.dumps(self.to_save_data(), ensure_ascii=False))
+            f.write("\n")
+        self.saved = True