Defines lr1 parsing table

Author: Maxcode123

src/syntactes/parsing_table/__init__.py

@@ -1,3 +1,3 @@
 from .entry import Entry
 from .conflict import Conflict, ConflictType
-from .table import LR0ParsingTable, SLRParsingTable
+from .table import LR0ParsingTable, SLRParsingTable, LR1ParsingTable, ParsingTable

src/syntactes/parsing_table/table.py

@@ -1,13 +1,22 @@
-from typing import Iterable, Optional, TypeAlias
+from typing import Iterable, Optional, Protocol, TypeAlias
+
+from typing_extensions import Self
 
 from syntactes import Grammar, Token
 from syntactes._action import Action
-from syntactes._state import LR0State
+from syntactes._state import LR0State, LR1State, State
 from syntactes.parsing_table import Conflict, Entry
 
 Row: TypeAlias = dict[Token, list[Action]]
 
 
+class ParsingTable(Protocol):
+    rows: dict[State, Row]
+
+    @staticmethod
+    def from_entries(entries: Iterable[Entry], grammar: Grammar) -> Self: ...
+
+
 class LR0ParsingTable:
     """
     Table that contains all the transitions from state to state with a symbol.
@@ -19,13 +28,11 @@ def __init__(self, grammar: Grammar) -> None:
         self._initial_state = None
 
     @staticmethod
-    def from_entries(
-        entries: Iterable[Entry], tokens: Iterable[Token]
-    ) -> "LR0ParsingTable":
+    def from_entries(entries: Iterable[Entry], grammar: Grammar) -> "LR0ParsingTable":
         """
         Create a parsing table from the given entries.
         """
-        table = LR0ParsingTable(tokens)
+        table = LR0ParsingTable(grammar)
         {table.add_entry(entry) for entry in entries}
         return table
 
@@ -133,3 +140,46 @@ def from_entries(
 
     def _header_str(self) -> str:
         return "SLR PARSING TABLE"
+
+
+class LR1ParsingTable(LR0ParsingTable):
+    """
+    Table that contains all the transitions from state to state with a symbol.
+    """
+
+    def __init__(self, grammar: Grammar) -> None:
+        self.rows: dict[LR1State, Row] = dict()
+        self._grammar = grammar
+        self._initial_state = None
+
+    @staticmethod
+    def from_entries(
+        entries: Iterable[Entry], tokens: Iterable[Token]
+    ) -> "LR1ParsingTable":
+        """
+        Create a parsing table from the given entries.
+        """
+        table = LR1ParsingTable(tokens)
+        {table.add_entry(entry) for entry in entries}
+        return table
+
+    @property
+    def initial_state(self) -> LR1State:
+        return self._initial_state
+
+    def get_actions(self, state: LR1State, token: Token) -> Optional[list[Action]]:
+        """
+        Get the actions from state with given number with `token`.
+        If there are no actions, returns None.
+        """
+        return self.rows.get(state, {}).get(token, None)
+
+    def get(self, state: LR1State) -> Optional[Row]:
+        """
+        Get the mapping of tokens to actions for the given state number.
+        Returns None if the state is not found.
+        """
+        return self.rows.get(state, None)
+
+    def _header_str(self) -> str:
+        return "LR1 PARSING TABLE"