Adds SLRParsingTable class

Author: Maxcode123

src/syntactes/generator.py

@@ -2,7 +2,7 @@
 from syntactes._item import LR0Item
 from syntactes._state import LR0State
 from syntactes.grammar import Grammar
-from syntactes.table import Entry, ParsingTable
+from syntactes.table import Entry, LR0ParsingTable, SLRParsingTable
 from syntactes.token import Token
 
 
@@ -14,7 +14,7 @@ class LR0Generator:
     def __init__(self, grammar: Grammar) -> None:
         self.grammar = grammar
 
-    def generate(self) -> ParsingTable:
+    def generate(self) -> LR0ParsingTable:
         """
         Generates an LR0 parsing table for the configured grammar.
         """
@@ -23,7 +23,7 @@ def generate(self) -> ParsingTable:
 
         entries = shift_entries | reduce_entries
 
-        table = ParsingTable.from_entries(entries, self.grammar)
+        table = LR0ParsingTable.from_entries(entries, self.grammar)
 
         return table
 
@@ -233,6 +233,19 @@ def _create_reduce_entries(self, states: set[LR0State]) -> set[Entry]:
 
 
 class SLRGenerator(LR0Generator):
+    def generate(self) -> SLRParsingTable:
+        """
+        Generates an SLR parsing table for the configured grammar.
+        """
+        states, shift_entries = self._create_states_and_shift_entries()
+        reduce_entries = self._create_reduce_entries(states)
+
+        entries = shift_entries | reduce_entries
+
+        table = SLRParsingTable.from_entries(entries, self.grammar)
+
+        return table
+
     def _create_reduce_entries(self, states: set[LR0State]) -> set[Entry]:
         """
         Computes and returns the entries for reduce actions and the accept action.

src/syntactes/table.py

@@ -1,7 +1,7 @@
 from typing import Iterable, Optional, TypeAlias
 
 from syntactes._action import Action
-from syntactes._state import State
+from syntactes._state import LR0State
 from syntactes.grammar import Grammar
 from syntactes.token import Token
 
@@ -14,7 +14,7 @@ class Entry:
     a state to another state via a symbol.
     """
 
-    def __init__(self, from_state: State, token: Token, action: Action) -> None:
+    def __init__(self, from_state: LR0State, token: Token, action: Action) -> None:
         self.from_state = from_state
         self.token = token
         self.action = action
@@ -39,23 +39,23 @@ def __eq__(self, other) -> bool:
         )
 
 
-class ParsingTable:
+class LR0ParsingTable:
     """
     Table that contains all the transitions from state to state with a symbol.
     """
 
     def __init__(self, grammar: Grammar) -> None:
-        self.rows: dict[State, Row] = dict()
+        self.rows: dict[LR0State, Row] = dict()
         self._grammar = grammar
 
-    def get_actions(self, state: State, token: Token) -> Optional[list[Action]]:
+    def get_actions(self, state: LR0State, 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: State) -> Optional[Row]:
+    def get(self, state: LR0State) -> Optional[Row]:
         """
         Get the mapping of tokens to actions for the given state number.
         Returns None if the state is not found.
@@ -73,11 +73,11 @@ def add_entry(self, entry: Entry) -> None:
     @staticmethod
     def from_entries(
         entries: Iterable[Entry], tokens: Iterable[Token]
-    ) -> "ParsingTable":
+    ) -> "LR0ParsingTable":
         """
         Create a parsing table from the given entries.
         """
-        table = ParsingTable(tokens)
+        table = LR0ParsingTable(tokens)
         {table.add_entry(entry) for entry in entries}
         return table
 
@@ -128,3 +128,19 @@ def _table_pretty_str(self) -> str:
 
     def _header_str(self) -> str:
         return "LR0 PARSING TABLE"
+
+
+class SLRParsingTable(LR0ParsingTable):
+    @staticmethod
+    def from_entries(
+        entries: Iterable[Entry], tokens: Iterable[Token]
+    ) -> "SLRParsingTable":
+        """
+        Create a parsing table from the given entries.
+        """
+        table = SLRParsingTable(tokens)
+        {table.add_entry(entry) for entry in entries}
+        return table
+
+    def _header_str(self) -> str:
+        return "SLR PARSING TABLE"