graph-games-proto 0.3.1889__py3-none-any.whl → 0.3.1897__py3-none-any.whl

graph_games_proto/fns.py

@@ -2484,11 +2484,6 @@ def getsettingvalue(s, setting_name):
     return getsettingvalue(s.game_config.fig, setting_name)
 
 
-def calclegalactions3(state):
-    legal_actions = []
-    return legal_actions
-
-
 @dispatch(GameConfig)
 def getinitialstate(game_config):
     fig = game_config.fig
@@ -2617,7 +2612,19 @@ def getinitialstate(game_config):
     )
 
     state = run_state_hooks(state, INITIALIZATION_HOOKS, True)
-    state = state.set(legal_actions_3=calclegalactions3(state))
+    
+    state_kernal = init_state_kernal(
+        edges=state.edges,
+        decks=state.decks,
+        piles=state.piles,
+        players=state.players,
+        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))
+
     return state
 
 
@@ -4027,6 +4034,65 @@ def isactionlegal2(s, a):
     return run_accept_action_hooks(s, a, ACCEPT_ACTION_HOOKS, False)
 
 
+def calc_legal_actions3(state_kernal):
+    history = state_kernal.history
+
+    if len(history) == 0:
+        # Initial
+        return [
+            LegalAction(
+                player_idx=player.idx,
+                name="INITIAL-GOAL-KEEP",
+                instruction="Your move. Please select the routes you want to keep.",
+                allotted_seconds=INITIAL_ALLOTTED_SECONDS,
+                allotted_since_action_idx=(len(history) - 1),
+                keep=LegalActionKeep(
+                    deck_idx=1,
+                    min=2,
+                )
+            )
+            for player in state_kernal.players
+        ]
+    
+    if False:
+        pass
+    elif True:
+        last_action = history[-1]
+        if last_action.legal_action.name == "INITIAL-GOAL-KEEP":
+            return get_default_legal_actions(state_kernal, state_kernal.player_idxs[0])
+        else:
+            last_player_idx = last_action.legal_action.player_idx
+            last_player_shuffled_idx = last_action.player_idxs.index(last_player_idx)
+            next_player_shuffled_idx = (last_player_shuffled_idx + 1) % len(last_action.player_idxs)
+            return get_default_legal_actions(state_kernal, state_kernal.player_idxs[next_player_shuffled_idx])
+    
+    return []
+
+
+class StateKernal(PClass):
+    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]
+    game_config = field(type=GameConfig)
+
+def init_state_kernal(**kwargs):
+    return StateKernal(
+        edges=kwargs.get('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'),
+        game_config=kwargs.get('game_config'),
+    )
+
+
+
 def getnextstate2(s, a, log=False):
     is_legal, reason = isactionlegal2(s, a)
     if not is_legal:
@@ -4037,6 +4103,19 @@ def getnextstate2(s, a, log=False):
     s = run_state_hooks(s, HANDLE_TERMINAL_HOOKS, log)
     if not s.legal_actions_2:
         s = run_state_action_hooks(s, a, EMPTY_LEGAL_ACTIONS_HOOKS, log)
+
+    state_kernal = init_state_kernal(
+        edges=s.edges,
+        decks=s.decks,
+        piles=s.piles,
+        players=s.players,
+        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))
+
     return s
 
 
@@ -4400,6 +4479,10 @@ def getfirstplayeridx(rng, num_players):
     return rng.randint(0, num_players-1)
 
 
+def get_legal_actions3(s, player_idx):
+    return [a for a in s.legal_actions_3 if a.player_idx == player_idx]
+
+
 @dispatch(State, int)
 def get_legal_actions(s, player_idx):
     return [a for a in s.legal_actions_2 if a.player_idx == player_idx]
@@ -4894,6 +4977,7 @@ def getprivatestate(s, player_idx):
         player_score=getprivateplayerscore(s, s.player_scores[player_idx]),
         player=s.players[player_idx],
         legal_actions_2 = get_legal_actions(s, player_idx),
+        legal_actions_3 = get_legal_actions3(s, player_idx),
         goal_completions=get_goal_completions(s, player_idx),
     )
 
@@ -5375,4 +5459,78 @@ HANDLE_TERMINAL_HOOKS = [
         when="HANDLE_TERMINAL",
         code=HANDLE_TERMINAL_HOOK_1,
     ),
-]
+]
+
+def get_default_legal_actions(state_kernal, player_idx):
+    allotted_seconds = DEFAULT_ALLOTTED_SECONDS
+    allotted_since_action_idx = len(state_kernal.history) - 1
+    legal_actions = []
+
+    if len(state_kernal.decks[1].facedown_stack) >= 1:
+        legal_actions.append(
+            LegalAction(
+                player_idx=player_idx,
+                name="DRAW-GOAL",
+                instruction="draw a route",
+                allotted_seconds=allotted_seconds,
+                allotted_since_action_idx=allotted_since_action_idx,
+                draw_discard=LegalActionDrawDiscard(
+                    deck_idx=1,
+                    quantity=3,
+                    min=1,
+                )
+            )
+        )
+
+    if len(state_kernal.decks[0].facedown_stack) >= 1:
+        legal_actions.append(
+            LegalAction(
+                auto_preferred=True,
+                player_idx=player_idx,
+                name="DRAW",
+                instruction="draw a unit",
+                allotted_seconds=allotted_seconds,
+                allotted_since_action_idx=allotted_since_action_idx,
+                draw=LegalActionDraw(
+                    deck_idx=0,
+                    quantity=1,
+                )
+            )
+        )
+
+    if len(state_kernal.decks[0].facedown_stack) >= 2:
+        state_kernal = append_to_legal_actions(
+            state_kernal,
+            LegalAction(
+                player_idx=player_idx,
+                name="DRAW-DOUBLE",
+                instruction="draw 2 units",
+                allotted_seconds=allotted_seconds,
+                allotted_since_action_idx=allotted_since_action_idx,
+                draw=LegalActionDraw(
+                    deck_idx=0,
+                    quantity=2,
+                )
+            )
+        )
+
+    for card_uuid in state_kernal.decks[0].faceup_spots:
+        if card_uuid:
+            legal_actions.append(
+                LegalAction(
+                    player_idx=player_idx,
+                    name="FACEUP-DRAW",
+                    instruction="draw a faceup unit",
+                    allotted_seconds=allotted_seconds,
+                    allotted_since_action_idx=allotted_since_action_idx,
+                    faceup_draw=LegalActionFaceupDraw(
+                        deck_idx=0,
+                        card_uuid=card_uuid,
+                    )
+                )
+            )
+
+    for legal_action in get_legal_actions_for_paths(state_kernal, player_idx):
+        legal_actions.append(legal_action)
+    
+    return legal_actions

{graph_games_proto-0.3.1889.dist-info → graph_games_proto-0.3.1897.dist-info}/METADATA

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

{graph_games_proto-0.3.1889.dist-info → graph_games_proto-0.3.1897.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=w4Hkn6z0bUuZeInMvSbbNpFjzAlp1UgZsamFBsuN2to,181960
+graph_games_proto/fns.py,sha256=Aqcoiw2rungLAiTgtj4Su_BpliNWO__s70knUaWkcIE,187324
 graph_games_proto/main.py,sha256=fj2U7KcwrpZtuUhjOX5yVxY18LZvvsxDFYZ_S5mxe04,145
 graph_games_proto/state.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-graph_games_proto-0.3.1889.dist-info/METADATA,sha256=F9asK6sb4QsoRU5ecRsQpSd-2kjESiDI9EfSosyE1uQ,188
-graph_games_proto-0.3.1889.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-graph_games_proto-0.3.1889.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
-graph_games_proto-0.3.1889.dist-info/RECORD,,
+graph_games_proto-0.3.1897.dist-info/METADATA,sha256=wPb_gd1bIy9puPul8HzXMOixr0wqyY0XIF_dV2dfePE,188
+graph_games_proto-0.3.1897.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+graph_games_proto-0.3.1897.dist-info/top_level.txt,sha256=-4QSrBMf_MM4BGsr2QXBpqDx8c8k_OPnzGyFjqjakes,18
+graph_games_proto-0.3.1897.dist-info/RECORD,,