graph-games-proto 0.3.1915__py3-none-any.whl → 0.3.1917__py3-none-any.whl

graph_games_proto/fns.py

@@ -1932,7 +1932,110 @@ class BonusStatus(PClass):
         )
 
 
+class StateKernal(PClass):
+    rng = field(type=random.Random)
+    game_config = field(type=GameConfig)
+    edges = field(type=list)  # List[BiEdge]
+    decks = field(type=list)  # List[Deck]
+    piles = field(type=list)  # List[Pile]
+    players = field(type=list)  # List[Player]
+    player_idxs = field(type=list)  # List[int]
+    history = field(type=list)  # List[PublicAction]
+    player_scores = field(type=list)  # List[PlayerScore]
+    uuid2edge = field(type=dict)  # Dict[str, BiEdge]
+    idx2path = field(type=list)  # List[Path2]
+    uuid2segment = field(type=dict)  # Dict[str, Segment]
+    pieceuuid2piece = field(type=dict)  # Dict[str, Piece]
+    edgeuuid2idx = field(type=dict)  # Dict[str, int]
+    carduuid2card = field(type=dict)  # Dict[str, Card]
+    def __todict__(self):
+        return {
+            "rng": rng2json(self.rng),
+            "game_config": self.game_config.__todict__(),
+            "edges": [edge.__todict__() for edge in self.edges],
+            "decks": [deck.__todict__() for deck in self.decks],
+            "piles": [pile.__todict__() for pile in self.piles],
+            "players": [player.__todict__() for player in self.players],
+            "player_idxs": self.player_idxs,
+            "history": [action.__todict__() for action in self.history],
+            "player_scores": [score.__todict__() for score in self.player_scores],
+            "uuid2edge": {k: v.__todict() for k, v in self.uuid2edge.items()},
+            "idx2path": [v.__todict__() for v in self.idx2path],
+            "uuid2segment": {k: v.__todict__() for k, v in self.uuid2segment.items()},
+            "pieceuuid2piece": {k: v.__todict__() for k, v in self.pieceuuid2piece.items()},
+            "edgeuuid2idx": self.edgeuuid2idx,
+            "carduuid2card": {k: v.__todict__() for k, v in self.carduuid2card.items()},
+        }
+    @staticmethod
+    def __fromdict__(d):
+        return StateKernal(
+            rng=json2rng(d["rng"]),
+            game_config=GameConfig.__fromdict__(d["game_config"]),
+            edges=[BiEdge.__fromdict__(edge) for edge in d["edges"]],
+            decks=[Deck.__fromdict__(deck) for deck in d["decks"]],
+            piles=[Pile.__fromdict__(pile) for pile in d["piles"]],
+            players=[Player.__fromdict__(player) for player in d["players"]],
+            player_idxs=d["player_idxs"],
+            history=[PublicAction.__fromdict__(action) for action in d["history"]],
+            player_scores=[PlayerScore.__fromdict__(score) for score in d["player_scores"]],
+            uuid2edge={k: BiEdge.__fromdict(v) for k, v in d["uuid2edge"].items()},
+            idx2path=[Path2.__fromdict(v) for v in d["idx2path"]],
+            uuid2segment={k: Segment2.__fromdict(v) for k, v in d["uuid2segment"].items()},
+            pieceuuid2piece={k: Piece.__fromdict(v) for k, v in d["pieceuuid2piece"].items()},
+            edgeuuid2idx=d["edgeuuid2idx"],
+            carduuid2card={k: Card.__fromdict(v) for k, v in d["carduuid2card"].items()},
+        )
+
+
+def init_state_kernal(**kwargs):
+    game_config = kwargs.get('game_config')
+    fig = game_config.fig
+    board_config = fig.board_config
+    edges = kwargs.get('edges', [])
+    uuid2edge = {}
+    idx2path = []
+    uuid2segment = {}
+    edgeuuid2idx = {edge.uuid: idx for idx, edge in enumerate(edges)}
+    for edge in edges:
+        uuid2edge[edge.uuid] = edge
+        for path in edge.paths:
+            idx2path.append(path)
+            for segment in path.segments:
+                uuid2segment[segment.uuid] = segment
+    pieceuuid2piece = {}
+    for piece_template in board_config.piece_templates:
+        if piece_template.has_player:
+            for player_idx in range(game_config.num_players):
+                pieces = generate_pieces(piece_template, player_idx)
+                for piece in pieces:
+                    pieceuuid2piece[piece.uuid] = piece
+    carduuid2card = {}
+    for dek in board_config.deks:
+        cards = generate_cards(dek)
+        for card in cards:
+            carduuid2card[card.uuid] = card
+    
+    return StateKernal(
+        rng=kwargs.get('rng'),
+        game_config=game_config,
+        edges=edges,
+        decks=kwargs.get('decks'),
+        piles=kwargs.get('piles'),
+        players=kwargs.get('players'),
+        player_idxs=kwargs.get('player_idxs'),
+        history=kwargs.get('history'),
+        player_scores=kwargs.get('player_scores'),
+        uuid2edge=uuid2edge,
+        idx2path=idx2path,
+        uuid2segment=uuid2segment,
+        edgeuuid2idx=edgeuuid2idx,
+        carduuid2card=carduuid2card,
+        pieceuuid2piece=pieceuuid2piece,
+    )
+
+
 class State(PClass):
+    kernal = field(type=StateKernal)
     uuid2edge = field(type=dict)  # Dict[str, BiEdge]
     idx2path = field(type=list)  # List[Path2]
     uuid2segment = field(type=dict)  # Dict[str, Segment]
@@ -1964,6 +2067,7 @@ class State(PClass):
     winners = field(type=list)  # List[int]
     def __todict__(self):
         return {
+            "kernal": self.kernal.__todict__(),
             "uuid2edge": {k: v.__todict__() for k, v in self.uuid2edge.items()},
             "idx2path": [v.__todict__() for v in self.idx2path],
             "uuid2segment": {k: v.__todict__() for k, v in self.uuid2segment.items()},
@@ -1997,6 +2101,7 @@ class State(PClass):
     @staticmethod
     def __fromdict__(d):
         return State(
+            kernal=StateKernal.__fromdict__(d["kernal"]),
             uuid2edge={k: BiEdge.__fromdict__(v) for k, v in d["uuid2edge"].items()},
             idx2path=[Path2.__fromdict__(v) for v in d["idx2path"]],
             uuid2segment={k: Segment2.__fromdict__(v) for k, v in d["uuid2segment"].items()},
@@ -2606,7 +2711,9 @@ def getinitialstate(game_config):
 
     state = run_state_hooks(state, INITIALIZATION_HOOKS, True)
     
-    state_kernal = init_state_kernal(
+    kernal = init_state_kernal(
+        rng=rng,
+        game_config=state.game_config,
         edges=state.edges,
         decks=state.decks,
         piles=state.piles,
@@ -2614,9 +2721,10 @@ def getinitialstate(game_config):
         player_idxs=state.player_idxs,
         history=state.history,
         player_scores=state.player_scores,
-        game_config=state.game_config,
     )
-    state = state.set(legal_actions_3=calc_legal_actions3(state_kernal))
+
+    state = state.set(kernal=kernal)
+    state = state.set(legal_actions_3=calc_legal_actions3(kernal))
 
     return state
 
@@ -4141,67 +4249,28 @@ def get_next_player_shuffled_idx(state_kernal, last_action):
     return next_player_shuffled_idx
 
 
-class StateKernal(PClass):
-    game_config = field(type=GameConfig)
-    edges = field(type=list)  # List[BiEdge]
-    decks = field(type=list)  # List[Deck]
-    piles = field(type=list)  # List[Pile]
-    players = field(type=list)  # List[Player]
-    player_idxs = field(type=list)  # List[int]
-    history = field(type=list)  # List[PublicAction]
-    player_scores = field(type=list)  # List[PlayerScore]
-    uuid2edge = field(type=dict)  # Dict[str, BiEdge]
-    idx2path = field(type=list)  # List[Path2]
-    uuid2segment = field(type=dict)  # Dict[str, Segment]
-    pieceuuid2piece = field(type=dict)  # Dict[str, Piece]
-    edgeuuid2idx = field(type=dict)  # Dict[str, int]
-    carduuid2card = field(type=dict)  # Dict[str, Card]
-
-
-def init_state_kernal(**kwargs):
-    game_config = kwargs.get('game_config')
-    fig = game_config.fig
-    board_config = fig.board_config
-    edges = kwargs.get('edges', [])
-    uuid2edge = {}
-    idx2path = []
-    uuid2segment = {}
-    edgeuuid2idx = {edge.uuid: idx for idx, edge in enumerate(edges)}
-    for edge in edges:
-        uuid2edge[edge.uuid] = edge
-        for path in edge.paths:
-            idx2path.append(path)
-            for segment in path.segments:
-                uuid2segment[segment.uuid] = segment
-    pieceuuid2piece = {}
-    for piece_template in board_config.piece_templates:
-        if piece_template.has_player:
-            for player_idx in range(game_config.num_players):
-                pieces = generate_pieces(piece_template, player_idx)
-                for piece in pieces:
-                    pieceuuid2piece[piece.uuid] = piece
-    carduuid2card = {}
-    for dek in board_config.deks:
-        cards = generate_cards(dek)
-        for card in cards:
-            carduuid2card[card.uuid] = card
-    
-    return StateKernal(
-        game_config=game_config,
-        edges=edges,
-        decks=kwargs.get('decks'),
-        piles=kwargs.get('piles'),
-        players=kwargs.get('players'),
-        player_idxs=kwargs.get('player_idxs'),
-        history=kwargs.get('history'),
-        player_scores=kwargs.get('player_scores'),
-        uuid2edge=uuid2edge,
-        idx2path=idx2path,
-        uuid2segment=uuid2segment,
-        edgeuuid2idx=edgeuuid2idx,
-        carduuid2card=carduuid2card,
-        pieceuuid2piece=pieceuuid2piece,
+def init_state(kernal):
+    state = State(
+        rng=kernal.rng,
+        game_config=kernal.game_config,
+        edges=kernal.edges,
+        decks=kernal.decks,
+        piles=kernal.piles,
+        players=kernal.players,
+        player_idxs=kernal.player_idxs,
+        history=kernal.history,
+        player_scores=kernal.player_scores,
+        legal_actions_3=[],
+        is_terminal=False,
+        uuid2edge=kernal.uuid2edge,
+        idx2path=kernal.idx2path,
+        uuid2segment=kernal.uuid2segment,
+        pieceuuid2piece=kernal.pieceuuid2piece,
+        edgeuuid2idx=kernal.edgeuuid2idx,
+        carduuid2card=kernal.carduuid2card,
     )
+    state = state.set(legal_actions_3=calc_legal_actions3(kernal))
+    return state
 
 
 def getnextstate2(s, a, log=False):
@@ -4214,6 +4283,7 @@ def getnextstate2(s, a, log=False):
     s = run_state_hooks(s, HANDLE_TERMINAL_HOOKS, log)
 
     state_kernal = init_state_kernal(
+        rng=s.rng,
         game_config=s.game_config,
         edges=s.edges,
         decks=s.decks,
@@ -4344,10 +4414,10 @@ def get_public_player_scores(s):
 # goal_completions = field(type=list, initial=[])  # List[GoalCompletion]
 
 
-
 def get_public_history(s):
     return [action.get_public(s) for action in s.history]
 
+
 def imagine_history(public_state, private_state):
     pass
 
@@ -4363,41 +4433,21 @@ def imagine_players(public_state, private_state):
 def imagine_decks(public_state, private_state):
     pass
 
-def imagine_rng(public_state, private_state):
-    pass
-
 
 def imagine_state(public_state, private_state):
-    return State(
-        uuid2edge = public_state.uuid2edge,
-        idx2path = public_state.idx2path,
-        uuid2segment = public_state.uuid2segment,
-        pieceuuid2piece = public_state.pieceuuid2piece,
-        carduuid2card = public_state.carduuid2card,
-        bonus_statuses = public_state.bonus_statuses,
-        bonusuuid2bonusidx = public_state.bonusuuid2bonusidx,
-        carduuid2deckidx = public_state.carduuid2deckidx,
-        starting_decks = public_state.starting_decks,
-        starting_piles = public_state.starting_piles,
-        player_graphs = public_state.player_graphs,
-        goals = public_state.goals,
-        nodes = public_state.nodes,
-        nodeuuid2idx = public_state.nodeuuid2idx,
-        edges = public_state.edges,
-        edgeuuid2idx = public_state.edgeuuid2idx,
-        edgetuple2uuid = public_state.edgetuple2uuid,
-        regions = public_state.regions,
-        last_to_play = public_state.last_to_play,
-        winners = public_state.winners,
-        piles = public_state.piles,
-        player_idxs = public_state.player_idxs,
-        game_config = public_state.game_config,
-        players = imagine_players(public_state, private_state),
-        decks = imagine_decks(public_state, private_state),
-        rng = imagine_rng(public_state, private_state),
-        history = imagine_history(public_state, private_state),
-        player_scores = imagine_player_scores(public_state, private_state),
+    kernal = init_state_kernal(
+        rng=random.Random(),
+        game_config=public_state.game_config,
+        player_idxs=public_state.player_idxs,
+        piles=public_state.piles,
+        edges=public_state.edges,
+        decks=imagine_decks(public_state, private_state),
+        players=imagine_players(public_state, private_state),
+        history=imagine_history(public_state, private_state),
+        player_scores=imagine_player_scores(public_state, private_state),
     )
+    imagined_state = init_state(kernal)
+    return imagined_state
 
 
 def isterminal(s):

{graph_games_proto-0.3.1915.dist-info → graph_games_proto-0.3.1917.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graph_games_proto
-Version: 0.3.1915
+Version: 0.3.1917
 Requires-Dist: multipledispatch==1.0.0
 Requires-Dist: pyrsistent==0.20.0
 Requires-Dist: numpy==2.2.4

{graph_games_proto-0.3.1915.dist-info → graph_games_proto-0.3.1917.dist-info}/RECORD

@@ -1,9 +1,9 @@
 graph_games_proto/__init__.py,sha256=_EVQR-51XehfH45XZlba1WPdx3omS3Gm1nTwrgGyn2Q,667
 graph_games_proto/all_types.py,sha256=IpbwftEcHS5Ewz-saFNk0lO9FvcbuHG36odRTayCXUk,54911
-graph_games_proto/fns.py,sha256=Rj9ALtDUBPp5C0sn7ExUcuXrzOabAw7A1Xsf-LKL7Sg,191027
+graph_games_proto/fns.py,sha256=yH7roM_wtAdsgFfeQA8wahPykXQrdCoySKm3f8U5Diw,193316
 graph_games_proto/main.py,sha256=fj2U7KcwrpZtuUhjOX5yVxY18LZvvsxDFYZ_S5mxe04,145
 graph_games_proto/state.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graph_games_proto-0.3.1915.dist-info/METADATA,sha256=8DzNZJm4Rg3_o3wwY0t6DEQIjF8wFibw9dK6jrRntd4,188
-graph_games_proto-0.3.1915.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graph_games_proto-0.3.1915.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
-graph_games_proto-0.3.1915.dist-info/RECORD,,
+graph_games_proto-0.3.1917.dist-info/METADATA,sha256=Nzsbrk3Z0TxOO5hmViO9MlS4_Lw-kTunuJoJiB3N9FU,188
+graph_games_proto-0.3.1917.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graph_games_proto-0.3.1917.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
+graph_games_proto-0.3.1917.dist-info/RECORD,,