Render Splitted States information on output file

lib/lrama/reporter.rb

@@ -29,7 +29,7 @@ def report(io, states)
         @terms.report(io, states)
         @conflicts.report(io, states)
         @grammar.report(io, states)
-        @states.report(io, states)
+        @states.report(io, states, ielr: states.ielr_defined?)
       end
     end
   end

lib/lrama/reporter/states.rb

@@ -13,10 +13,12 @@
         @verbose = verbose
       end
 
-      # @rbs (IO io, Lrama::States states) -> void
-      def report(io, states)
+      # @rbs (IO io, Lrama::States states, ielr: bool) -> void
+      def report(io, states, ielr: false)
         cex = Counterexamples.new(states) if @counterexamples
 
+        report_splitted_states(io, states.states) if ielr
+
         states.states.each do |state|
           report_state_header(io, state)
           report_items(io, state)
@@ -35,6 +37,21 @@
 
       private
 
+      # @rbs (IO io, Array[Lrama::State] states) -> void
+      def report_splitted_states(io, states)
+        ss = states.select(&:splitted_state?)
+
+        return if ss.empty?
+
+        io << "Splitted States\n\n"
+
+        ss.each do |state|
+          io << "    State #{state.id} is splitted from state #{state.lalr_isocore.id}\n"
+        end
+
+        io << "\n"
+      end
+
       # @rbs (IO io, Lrama::State state) -> void
       def report_state_header(io, state)
         io << "State #{state.id}\n\n"

lib/lrama/state.rb

@@ -235,6 +235,11 @@ def rr_conflicts
       end
     end
 
+    # @rbs () -> bool
+    def splitted_state?
+      @lalr_isocore != self
+    end
+
     # Definition 3.40 (propagate_lookaheads)
     #
     # @rbs (State next_state) -> lookahead_set

lib/lrama/states.rb

@@ -40,7 +40,7 @@ class States
     include Lrama::Tracer::Duration
 
     def_delegators "@grammar", :symbols, :terms, :nterms, :rules,
-      :accept_symbol, :eof_symbol, :undef_symbol, :find_symbol_by_s_value!
+      :accept_symbol, :eof_symbol, :undef_symbol, :find_symbol_by_s_value!, :ielr_defined?
 
     attr_reader :states #: Array[State]
     attr_reader :reads_relation #: Hash[transition, Array[transition]]

sig/generated/lrama/reporter/states.rbs

@@ -6,11 +6,14 @@ module Lrama
       # @rbs (itemsets: bool, lookaheads: bool, solved: bool, counterexamples: bool, verbose: bool, **untyped _) -> void
       def initialize: (itemsets: bool, lookaheads: bool, solved: bool, counterexamples: bool, verbose: bool, **untyped _) -> void
 
-      # @rbs (IO io, Lrama::States states) -> void
-      def report: (IO io, Lrama::States states) -> void
+      # @rbs (IO io, Lrama::States states, ielr: bool) -> void
+      def report: (IO io, Lrama::States states, ielr: bool) -> void
 
       private
 
+      # @rbs (IO io, Array[Lrama::State] states) -> void
+      def report_splitted_states: (IO io, Array[Lrama::State] states) -> void
+
       # @rbs (IO io, Lrama::State state) -> void
       def report_state_header: (IO io, Lrama::State state) -> void
 

sig/generated/lrama/state.rbs

@@ -133,6 +133,9 @@ module Lrama
     # @rbs () -> Array[conflict]
     def rr_conflicts: () -> Array[conflict]
 
+    # @rbs () -> bool
+    def splitted_state?: () -> bool
+
     # Definition 3.40 (propagate_lookaheads)
     #
     # @rbs (State next_state) -> lookahead_set

spec/lrama/states_spec.rb

@@ -1815,6 +1815,12 @@
       Lrama::Reporter.new(states: true).report(io, states)
 
       expect(io.string).to eq(<<~STR)
+        Splitted States
+
+            State 19 is splitted from state 4
+            State 20 is splitted from state 9
+            State 21 is splitted from state 14
+
         State 0
 
             0 $accept: • S "end of file"