graph-games-proto 0.3.1986__py3-none-any.whl → 0.3.1989__py3-none-any.whl

graph_games_proto/fns.py

@@ -1952,6 +1952,7 @@ class StateKernel(PClass):
     bonusuuid2bonusidx = field(type=dict)  # Dict[str, int]
     starting_decks = field(type=list)  # List[Deck]
     starting_piles = field(type=list) # List[Pile]
+    goals = field(type=list)  # List[FrozenGoal]
     def __todict__(self):
         return {
             "rng": rng2json(self.rng),
@@ -1973,6 +1974,7 @@ class StateKernel(PClass):
             "bonusuuid2bonusidx": self.bonusuuid2bonusidx,
             "starting_decks": [deck.__todict__() for deck in self.starting_decks],
             "starting_piles": [pile.__todict__() for pile in self.starting_piles],
+            "goals": [goal.__todict__() for goal in self.goals],
         }
     @staticmethod
     def __fromdict__(d):
@@ -1996,6 +1998,7 @@ class StateKernel(PClass):
             bonusuuid2bonusidx=d["bonusuuid2bonusidx"],
             starting_decks=[Deck.__fromdict__(x) for x in d["starting_decks"]],
             starting_piles=[Pile.__fromdict__(x) for x in d["starting_piles"]],
+            goals=[FrozenGoal.__fromdict__(goal) for goal in d["goals"]],
         )
 
 
@@ -2058,28 +2061,35 @@ def init_state_kernel(**kwargs):
         bonusuuid2bonusidx=bonusuuid2bonusidx,
         starting_decks=kwargs.get('starting_decks'),
         starting_piles=kwargs.get('starting_piles'),
+        goals=kwargs.get('goals'),
     )
 
 
 class State(PClass):
+    # Identity
     kernel = field(type=StateKernel)
-    idx2path = field(type=list)  # List[Path2]
-    bonus_statuses = field(type=list)  # List[BonusStatus]
     history = field(type=list)  # List[Action2]
-    player_scores = field(type=list)  # List[PlayerScore2]
-    player_graphs = field(type=list)  # List[PlayerGraph]
-    goals = field(type=list)  # List[Goal]
+    players = field(type=list)  # List[Player]
     nodes = field(type=list)  # List[Node]
     edges = field(type=list)  # List[BiEdge]
     regions = field(type=list)  # List[Region]
-    legal_actions_3 = field(type=list)  # List[LegalAction]
     piles = field(type=list)  # List[Pile]
-    players = field(type=list)  # List[Player]
+    last_to_play = field(type=(int, type(None)), initial=None)
+    
+    idx2path = field(type=list)  # List[Path2]
+
+    # Scoring
     player_idxs = field(type=list)  # List[int]
     decks = field(type=list)  # List[Deck]
     rng = field(type=random.Random)
-    last_to_play = field(type=(int, type(None)), initial=None)
+    
+    # Memoized
+    player_graphs = field(type=list)  # List[PlayerGraph]
+    legal_actions_3 = field(type=list)  # List[LegalAction]
+    player_scores = field(type=list)  # List[PlayerScore2]
+    bonus_statuses = field(type=list)  # List[BonusStatus]
     winners = field(type=list)  # List[int]
+
     def __todict__(self):
         return {
             "kernel": self.kernel.__todict__(),
@@ -2088,7 +2098,6 @@ class State(PClass):
             "history": [x.__todict__() for x in self.history],
             "player_scores": [x.__todict__() for x in self.player_scores],
             "player_graphs": [x.__todict__() for x in self.player_graphs],
-            "goals": [goal.__todict__() for goal in self.goals],
             "nodes": [node.__todict__() for node in self.nodes],
             "edges": [edge.__todict__() for edge in self.edges],
             "regions": [region.__todict__() for region in self.regions],
@@ -2110,7 +2119,6 @@ class State(PClass):
             history=[Action2.__fromdict__(x) for x in d["history"]],
             player_scores=[PlayerScore2.__fromdict__(x) for x in d["player_scores"]],
             player_graphs=[PlayerGraph.__fromdict__(x) for x in d["player_graphs"]],
-            goals=[FrozenGoal.__fromdict__(goal) for goal in d["goals"]],
             nodes=[Node.__fromdict__(n) for n in d["nodes"]],
             edges=[BiEdge.__fromdict__(e) for e in d["edges"]],
             regions=[Region.__fromdict__(r) for r in d["regions"]],
@@ -2654,7 +2662,6 @@ def getinitialstate(game_config):
                 neighbors=[[] for _ in range(len(nodes))]
             ) for player_idx in range(game_config.num_players)
         ],
-        goals = board_config.goals,
         nodes = nodes,
         edges = edges,
         regions = get_regions(board_config),
@@ -2682,6 +2689,7 @@ def getinitialstate(game_config):
         player_scores=state.player_scores,
         starting_decks=starting_decks,
         starting_piles=starting_piles,
+        goals=board_config.goals,
     )
 
     state = state.set(kernel=kernel)
@@ -2753,10 +2761,11 @@ def score_public_items(state, player_idx):
     return items
 
 
-def score_private_items(game, player_idx):
+@dispatch(State, int)
+def score_private_items(state, player_idx):
     items = []
-    goaluuid2goal = {goal.uuid: goal for goal in game.goals}
-    goal_completions = get_goal_completions(game, player_idx)
+    goaluuid2goal = {goal.uuid: goal for goal in state.kernel.goals}
+    goal_completions = get_goal_completions(state, player_idx)
     complete_goal_uuids = [gc.goal_uuid for gc in goal_completions if gc.complete]
     incomplete_goal_uuids = [gc.goal_uuid for gc in goal_completions if not gc.complete]
     for complete_goal_uuid in complete_goal_uuids:
@@ -4104,6 +4113,7 @@ def getnextstate2(s, a, log=False):
         player_scores=s.player_scores,
         starting_decks=s.kernel.starting_decks,
         starting_piles=s.kernel.starting_piles,
+        goals=s.kernel.goals,
     )
     s = s.set(legal_actions_3=calc_legal_actions3(state_kernel))
 
@@ -4509,9 +4519,10 @@ def get_player_graph(s, player_idx):
     return nodes
 
 
+@dispatch(State, int)
 def get_goals(s, player_idx):
     player = s.players[player_idx]
-    goaluuid2goal = {goal.uuid: goal for goal in s.goals}
+    goaluuid2goal = {goal.uuid: goal for goal in s.kernel.goals}
     goal_uuids = [s.kernel.carduuid2card[card_uuid].goal_uuid for card_uuid in player.cards if s.kernel.carduuid2card[card_uuid].goal_uuid]
     return [
         goaluuid2goal[goal_uuid] for goal_uuid in goal_uuids if goal_uuid in goaluuid2goal

{graph_games_proto-0.3.1986.dist-info → graph_games_proto-0.3.1989.dist-info}/METADATA

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

{graph_games_proto-0.3.1986.dist-info → graph_games_proto-0.3.1989.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=CfVDjz6q53feWv71XmCFGOGFI40s8Op0jiq6BzEuVLI,182494
+graph_games_proto/fns.py,sha256=I5TKswJWFPFWBORV9VOArGWBlXgUW3GXb-1GP0HVhxE,182680
 graph_games_proto/main.py,sha256=fj2U7KcwrpZtuUhjOX5yVxY18LZvvsxDFYZ_S5mxe04,145
 graph_games_proto/state.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graph_games_proto-0.3.1986.dist-info/METADATA,sha256=idZgHPywS8D8vXzhRPJefE30f5kr5NrL1MeSjcbDtgI,188
-graph_games_proto-0.3.1986.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graph_games_proto-0.3.1986.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
-graph_games_proto-0.3.1986.dist-info/RECORD,,
+graph_games_proto-0.3.1989.dist-info/METADATA,sha256=Mv0DRZjDx42RK5lO5oFiwkJofNq6YmNqokhkaH2rwVA,188
+graph_games_proto-0.3.1989.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graph_games_proto-0.3.1989.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
+graph_games_proto-0.3.1989.dist-info/RECORD,,