mort666-wongi-engine 0.2.9

data/lib/wongi-engine/beta.rb

@@ -0,0 +1,11 @@
+require 'wongi-engine/beta/beta_node'
+require 'wongi-engine/beta/beta_memory'
+require 'wongi-engine/beta/filter_node'
+require 'wongi-engine/beta/assignment_node'
+require 'wongi-engine/beta/join_node'
+require 'wongi-engine/beta/ncc_partner'
+require 'wongi-engine/beta/ncc_node'
+require 'wongi-engine/beta/neg_node'
+require 'wongi-engine/beta/optional_node'
+require 'wongi-engine/beta/or_node'
+require 'wongi-engine/beta/production_node'

data/lib/wongi-engine/beta/assignment_node.rb

@@ -0,0 +1,40 @@
+module Wongi::Engine
+
+  class AssignmentNode < BetaNode
+
+    def initialize parent, variable, body
+      super parent
+      @variable, @body = variable, body
+    end
+
+    def beta_activate token, wme = nil, assignments = { }
+      children.each do |child|
+        child.beta_activate Token.new( child, token, nil, { @variable => @body.respond_to?(:call) ? @body.call(token) : @body } )
+      end
+    end
+
+    def beta_deactivate token
+      children.each do |child|
+        child.tokens.each do |t|
+          if t.parent == token
+            child.beta_deactivate t
+            #token.destroy
+          end
+        end
+      end
+    end
+
+    def refresh_child child
+      tmp = children
+      self.children = [ child ]
+      parent.tokens.each do |token|
+        children.each do |child|
+          child.beta_activate Token.new( child, token, nil, { @variable => @body.respond_to?(:call) ? @body.call(token) : @body } )
+        end
+      end
+      self.children = tmp
+    end
+
+  end
+
+end

data/lib/wongi-engine/beta/beta_memory.rb

@@ -0,0 +1,49 @@
+module Wongi::Engine
+
+  class BetaMemory < BetaNode
+    include TokenContainer
+
+    def seed assignments = {}
+      @seed = assignments
+      t = Token.new( self, nil, nil, assignments )
+      rete.default_overlay.add_token(t, self)
+    end
+
+    def subst valuations
+      beta_deactivate(tokens.first)
+      token = Token.new( self, nil, nil, @seed )
+      valuations.each { |variable, value| token.subst variable, value }
+      beta_activate(token)
+    end
+
+    def beta_activate token
+      existing = tokens.find { |et| et.duplicate? token }
+      return if existing # TODO really?
+      token.overlay.add_token(token, self)
+      children.each do |child|
+        child.beta_activate token
+      end
+      token
+    end
+
+    def beta_deactivate token
+      return nil unless tokens.find token
+      token.overlay.remove_token(token, self)
+      token.deleted!
+      if token.parent
+        token.parent.children.delete token # should this go into Token#destroy?
+      end
+      children.each do |child|
+        child.beta_deactivate token
+      end
+      token
+    end
+
+    def refresh_child child
+      tokens.each do |token|
+        child.beta_activate token
+      end
+    end
+
+  end
+end

data/lib/wongi-engine/beta/beta_node.rb

@@ -0,0 +1,94 @@
+module Wongi::Engine
+
+  class BetaNode
+
+    module TokenContainer
+      def tokens
+        Enumerator.new do |y|
+          rete.overlays.each do |overlay|
+            overlay.raw_tokens(self).dup.each do |token|
+              y << token unless token.deleted?
+            end
+            overlay.raw_tokens(self).reject! &:deleted?
+          end
+        end
+      end
+
+      def empty?
+        tokens.first.nil?
+      end
+
+      def size
+        tokens.count
+      end
+    end
+
+    include CoreExt
+
+    attr_writer :rete
+    attr_reader :parent
+    attr_accessor :children
+    attr_predicate :debug
+
+    def initialize parent = nil
+      @parent = parent
+      @children = []
+      if parent
+        parent.children << self
+      end
+    end
+
+    def root?
+      parent.nil?
+    end
+
+    def depth
+      @depth ||= if parent.nil?
+        0
+      else
+        parent.depth + 1
+      end
+    end
+
+    def rete
+      @rete ||= if parent
+        parent.rete
+      end
+    end
+
+    abstract :alpha_activate
+    abstract :alpha_deactivate
+    abstract :alpha_reactivate
+    abstract :beta_activate
+    abstract :beta_deactivate
+    abstract :beta_reactivate
+
+    def assignment_node variable, body
+      node = AssignmentNode.new self, variable, body
+      node.refresh
+      node
+    end
+
+    def refresh
+      parent.refresh_child self
+    end
+
+    def refresh_child node
+      raise "#{self.class} must implement refresh_child"
+    end
+
+    private
+
+    def dp message
+      if debug?
+        puts "#{indent}#{message}"
+      end
+    end
+
+    def indent
+      '  ' * depth
+    end
+
+  end
+
+end

data/lib/wongi-engine/beta/filter_node.rb

@@ -0,0 +1,48 @@
+module Wongi
+  module Engine
+
+    class FilterNode < BetaNode
+
+      attr_accessor :test
+
+      def initialize parent, test
+        super parent
+        self.test = test
+      end
+
+      def beta_activate token
+        if test.passes? token
+          children.each do |child|
+            child.beta_activate Token.new( child, token, nil, {} )
+          end
+        end
+      end
+
+      def beta_deactivate token
+        children.each do |child|
+          child.tokens.each do |t|
+            if t.parent == token
+              child.beta_deactivate t
+              #token.destroy
+            end
+          end
+        end
+      end
+
+      def equivalent? test
+        test == self.test
+      end
+
+      def refresh_child child
+        tmp = children
+        self.children = [ child ]
+        parent.tokens.each do |token|
+          beta_activate token
+        end
+        self.children = tmp
+      end
+
+    end
+
+  end
+end

data/lib/wongi-engine/beta/join_node.rb

@@ -0,0 +1,140 @@
+module Wongi
+  module Engine
+
+    TokenAssignment = Struct.new(:wme, :field) do
+      def call token = nil
+        wme.send field
+      end
+      def inspect
+        "#{field} of #{wme}"
+      end
+      def to_s
+        inspect
+      end
+    end
+
+    class BetaTest
+
+      attr_reader :field
+      attr_reader :variable
+
+      def initialize field, variable
+        @field, @variable = field, variable
+      end
+
+      def matches? token, wme
+        assignment = token[ self.variable ]
+        field = wme.send( self.field )
+        #field.nil? ||
+        assignment && field == assignment
+      end
+
+      def equivalent? other
+        other.field == self.field && other.variable == self.variable
+      end
+
+    end
+
+    class JoinNode < BetaNode
+
+      attr_accessor :alpha
+      attr_reader :tests
+      attr_reader :assignment_pattern
+
+      def initialize parent, tests, assignment
+        super(parent)
+        @tests = tests
+        @assignment_pattern = assignment
+      end
+
+      def equivalent? alpha, tests, assignment_pattern
+        return false unless self.alpha == alpha
+        return false unless self.assignment_pattern == assignment_pattern
+        return false unless (self.tests.empty? && tests.empty?) || self.tests.length == tests.length && self.tests.all? { |my_test|
+          tests.any? { |new_test|
+            my_test.equivalent? new_test
+          }
+        }
+        true
+      end
+
+      def alpha_activate wme
+        assignments = collect_assignments( wme )
+        parent.tokens.each do |token|
+          if matches?( token, wme )
+            children.each do |beta|
+              beta.beta_activate Token.new( beta, token, wme, assignments )
+            end
+          end
+        end
+      end
+
+      def alpha_deactivate wme
+        children.each do |child|
+          child.tokens.each do |token|
+            if token.wme == wme
+              child.beta_deactivate token
+            end
+          end
+        end
+      end
+
+      def beta_activate token
+        self.alpha.wmes.each do |wme|
+          if matches?( token, wme )
+            assignments = collect_assignments( wme )
+            children.each do |beta|
+              beta.beta_activate Token.new( beta, token, wme, assignments )
+            end
+          end
+        end
+      end
+
+      def beta_deactivate token
+        children.each do |child|
+          child.tokens.each do |t|
+            if t.parent == token
+              child.beta_deactivate t
+            end
+          end
+        end
+      end
+
+      def refresh_child child
+        alpha.wmes.each do |wme|
+          assignments = collect_assignments( wme )
+          parent.tokens.each do |token|
+            if matches?( token, wme )
+              child.beta_activate Token.new( child, token, wme, assignments )
+            end
+          end
+        end
+      end
+
+      protected
+
+      def matches? token, wme
+        @tests.each do |test|
+          return false unless test.matches?( token, wme )
+        end
+        true
+      end
+
+      def collect_assignments wme
+        assignments = {}
+        return assignments if assignment_pattern.nil?
+        if assignment_pattern.subject != :_
+          assignments[ assignment_pattern.subject ] = TokenAssignment.new(wme, :subject)
+        end
+        if assignment_pattern.predicate != :_
+          assignments[ assignment_pattern.predicate ] = TokenAssignment.new(wme, :predicate)
+        end
+        if assignment_pattern.object != :_
+          assignments[ assignment_pattern.object ] = TokenAssignment.new(wme, :object)
+        end
+        assignments
+      end
+
+    end
+  end
+end

data/lib/wongi-engine/beta/ncc_node.rb

@@ -0,0 +1,67 @@
+module Wongi
+  module Engine
+    class NccNode < BetaNode
+      include TokenContainer
+
+      attr_accessor :partner
+
+      def initialize parent
+        super
+      end
+
+      def beta_activate token
+        return if tokens.find { |t| t.parent == token }
+        t = Token.new self, token, nil, {}
+        t.overlay.add_token(t, self)
+        partner.tokens.each do |ncc_token|
+          next unless ncc_token.ancestors.find { |a| a.equal? token }
+          t.ncc_results << ncc_token
+          ncc_token.owner = t
+        end
+        if t.ncc_results.empty?
+          children.each do |child|
+            child.beta_activate Token.new( child, t, nil, { } )
+          end
+        end
+      end
+
+      def beta_deactivate token
+        t = tokens.find { |tok| tok.parent == token }
+        return unless t
+        t.overlay.remove_token(t, self)
+        t.deleted!
+        partner.tokens.select { |ncc| ncc.owner == t }.each do |ncc_token|
+          ncc_token.owner = nil
+          t.ncc_results.delete ncc_token
+        end
+        children.each do |beta|
+          beta.tokens.select { |token| token.parent == t }.each do |token|
+            beta.beta_deactivate token
+          end
+        end
+      end
+
+      def ncc_activate token
+        children.each do |child|
+          child.beta_activate Token.new( child, token, nil, { } )
+        end
+      end
+
+      def ncc_deactivate token
+        children.each do |beta|
+          beta.tokens.select { |t| t.parent == token }.each do |t|
+            beta.beta_deactivate t
+          end
+        end
+      end
+
+      def refresh_child child
+        tokens.each do |token|
+          if token.ncc_results.empty?
+            child.beta_activate Token.new( child, token, nil, { } )
+          end
+        end
+      end
+    end
+  end
+end