graph-games-proto 0.3.1900__py3-none-any.whl → 0.3.1902__py3-none-any.whl

graph_games_proto/fns.py

@@ -4070,6 +4070,7 @@ def calc_legal_actions3(state_kernal):
 
 
 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]
@@ -4077,14 +4078,18 @@ class StateKernal(PClass):
     player_idxs = field(type=list)  # List[int]
     history = field(type=list)  # List[PublicAction]
     player_scores = field(type=list)  # List[PlayerScore]
-    game_config = field(type=GameConfig)
     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 = []
@@ -4096,7 +4101,21 @@ def init_state_kernal(**kwargs):
             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'),
@@ -4104,11 +4123,12 @@ def init_state_kernal(**kwargs):
         player_idxs=kwargs.get('player_idxs'),
         history=kwargs.get('history'),
         player_scores=kwargs.get('player_scores'),
-        game_config=kwargs.get('game_config'),
         uuid2edge=uuid2edge,
         idx2path=idx2path,
         uuid2segment=uuid2segment,
         edgeuuid2idx=edgeuuid2idx,
+        carduuid2card=carduuid2card,
+        pieceuuid2piece=pieceuuid2piece,
     )
 
 
@@ -4124,6 +4144,7 @@ def getnextstate2(s, a, log=False):
         s = run_state_action_hooks(s, a, EMPTY_LEGAL_ACTIONS_HOOKS, log)
 
     state_kernal = init_state_kernal(
+        game_config=s.game_config,
         edges=s.edges,
         decks=s.decks,
         piles=s.piles,
@@ -4131,7 +4152,6 @@ def getnextstate2(s, a, log=False):
         player_idxs=s.player_idxs,
         history=s.history,
         player_scores=s.player_scores,
-        game_config=s.game_config,
     )
     s = s.set(legal_actions_3=calc_legal_actions3(state_kernal))
 

{graph_games_proto-0.3.1900.dist-info → graph_games_proto-0.3.1902.dist-info}/METADATA

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

{graph_games_proto-0.3.1900.dist-info → graph_games_proto-0.3.1902.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=UaOPciy5NGI80NFPQ7UkCAh76MTngEOQfwdd9v-Zgs4,187990
+graph_games_proto/fns.py,sha256=2KcO1B2jFlcwcddwGFzkjV5d9nxZz9m5sm0kOmZ9NkI,188787
 graph_games_proto/main.py,sha256=fj2U7KcwrpZtuUhjOX5yVxY18LZvvsxDFYZ_S5mxe04,145
 graph_games_proto/state.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graph_games_proto-0.3.1900.dist-info/METADATA,sha256=AHrt9_-p3mdMs6pDQ0nrl_Vzly7iMgnU7IeqofXelbk,188
-graph_games_proto-0.3.1900.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graph_games_proto-0.3.1900.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
-graph_games_proto-0.3.1900.dist-info/RECORD,,
+graph_games_proto-0.3.1902.dist-info/METADATA,sha256=gvmEGj3QttF1wrG5ynpGKh3fOsSr6tBeRmWDSFElNIo,188
+graph_games_proto-0.3.1902.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graph_games_proto-0.3.1902.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
+graph_games_proto-0.3.1902.dist-info/RECORD,,