Interscript.transliterate_each

Author: webdev778

ruby/lib/interscript.rb

@@ -36,6 +36,11 @@ def transliterate(system_code, string, maps={}, compiler: Interscript::Interpret
       load(system_code, maps, compiler: compiler).(string)
     end
 
+    # Gives each possible value of the transliteration.
+    def transliterate_each(system_code, string, maps={}, &block)
+      load(system_code, maps).(string, each: true, &block)
+    end
+
     def transliterate_file(system_code, input_file, output_file, maps={})
       input = File.read(input_file)
       output = transliterate(system_code, input, maps)

ruby/lib/interscript/interpreter.rb

@@ -5,9 +5,68 @@ def compile(map, _:nil)
     self
   end
 
-  def call(str, stage=:main)
+  def call(str, stage=:main, each: false, &block)
     stage = @map.stages[stage]
-    Stage.new(@map, str).execute_rule(stage)
+    s =
+    if each
+      e = Enumerator.new do |yielder|
+        options = []
+        options_set = false
+        choices = nil
+
+        i = 0
+
+        loop do
+          result = nil
+
+          f = Fiber.new do
+            $select_nth_string = true
+            result = Stage.new(@map, str).execute_rule(stage)
+            $select_nth_string = false
+            Fiber.yield(:end)
+          end
+
+          iter = 0
+
+          loop do
+            break if f.resume == :end
+            # hash is unused for now... some problems may arise in certain
+            # scenarios that are not a danger right now, but i'm genuinely
+            # unsure how it can be handled.
+            #
+            # This scenario is described in a commented out test.
+            type, value, hash = f.resume
+            if options_set
+              f.resume(choices[i][iter])
+            else
+              options[iter] = value
+              f.resume(0)
+            end
+            iter += 1
+          end
+
+          unless options_set
+            options_set = true
+
+            opts = options.map { |i| (0...i).to_a }
+            choices = opts[0].product(*opts[1..-1])
+          end
+
+          yielder.yield(result)
+
+          i += 1
+          break if i == choices.length
+        end
+      end
+
+      if block_given?
+        e.each(&block)
+      else
+        e
+      end
+    else
+      Stage.new(@map, str).execute_rule(stage)
+    end
   end
 
   class Stage
@@ -106,7 +165,7 @@ def build_regexp(r)
     end
 
     def build_item i, target=nil, doc=@map
-      i = i.first_string if %i[str parstr].include? target
+      i = i.nth_string if %i[str parstr].include? target
       i = Interscript::Node::Item.try_convert(i)
       target = :par if target == :parstr
 

ruby/lib/interscript/node/item/alias.rb

@@ -23,6 +23,8 @@ def first_string
     self
   end
 
+  alias nth_string first_string
+
   def to_hash
     { :class => self.class.to_s,
       :name => name,

ruby/lib/interscript/node/item/any.rb

@@ -36,6 +36,16 @@ def first_string
     end
   end
 
+  def nth_string
+    return first_string unless $select_nth_string
+
+    d = data
+    Fiber.yield(:prepare)
+    id = Fiber.yield(:select_nth_string, d.count, self.hash)
+    Fiber.yield(:selection)
+    Interscript::Node::Item.try_convert(value[id]).nth_string
+  end
+
   def max_length
     self.data.map(&:max_length).max
   end

ruby/lib/interscript/node/item/capture.rb

@@ -11,6 +11,10 @@ def first_string
     data.first_string
   end
 
+  def nth_string
+    data.nth_string
+  end
+
   def to_hash
     { :class => self.class.to_s,
       :data => self.data.to_hash }
@@ -33,6 +37,8 @@ def first_string
     self
   end
 
+  alias nth_string first_string
+
   def to_hash
     { :class => self.class.to_s,
       :id => self.id }

ruby/lib/interscript/node/item/group.rb

@@ -32,6 +32,10 @@ def first_string
     self.children.map(&:first_string).reduce(&:+)
   end
 
+  def nth_string
+    self.children.map(&:nth_string).reduce(&:+)
+  end
+
   def max_length
     @children.map { |i| i.max_length }.sum
   end

ruby/lib/interscript/node/item/repeat.rb

@@ -9,6 +9,10 @@ def first_string
     data.first_string
   end
 
+  def nth_string
+    data.nth_string
+  end
+
   def max_length
     data.max_length
   end

ruby/lib/interscript/node/item/string.rb

@@ -17,6 +17,8 @@ def first_string
     self.data
   end
 
+  alias nth_string first_string
+
   def + other
     if self.data == ""
       Interscript::Node::Item.try_convert(other)

ruby/spec/spec_helper.rb

@@ -51,8 +51,8 @@ def each_compiler &block
 end
 
 class Interscript::Node::Document
-  def call(str, stage=:main, compiler=$compiler || Interscript::Interpreter)
-    compiler.(self).(str, stage)
+  def call(str, stage=:main, compiler=$compiler || Interscript::Interpreter, **kwargs)
+    compiler.(self).(str, stage, **kwargs)
   end
 end
 

ruby/spec/transliterate_each_spec.rb

@@ -0,0 +1,23 @@
+RSpec.describe "Interscript#transliterate_each" do
+  before :example do
+    $compiler = Interscript::Interpreter
+  end
+
+  it "works" do
+    s = stage {
+      sub "X", any("abcd")
+      sub "Y", any("defg")
+      sub "Z", any("ghij")
+    }
+
+    expect(s.("XYZ", each: true).take(5)).to eq(%w[adg adh adi adj aeg])
+  end
+
+  # it "supports a certain scenario" do
+  #   s = stage {
+  #    sub "X", any(["x", any("abc"), "d"])
+  #   }
+  #
+  #   expect(s.("X", each: true).to_a).to eq(%w[x x x a b c d d d])
+  # end
+end