pacer 0.9.1.1-java

data/lib/pacer/graph.rb

@@ -0,0 +1,14 @@
+module Pacer
+  import com.tinkerpop.blueprints.pgm.Graph
+  import com.tinkerpop.blueprints.pgm.Element
+  import com.tinkerpop.blueprints.pgm.Vertex
+  import com.tinkerpop.blueprints.pgm.Edge
+end
+
+require 'pacer/graph/graph_mixin'
+require 'pacer/graph/element_mixin'
+require 'pacer/graph/vertex_mixin'
+require 'pacer/graph/edge_mixin'
+require 'pacer/graph/index_mixin'
+require 'pacer/graph/graph_indices_mixin'
+require 'pacer/graph/graph_transactions_mixin'

data/lib/pacer/pipe/blackbox_pipeline.rb

@@ -0,0 +1,48 @@
+module Pacer::Pipes
+  # Same concept as the regular pipeline, but this one
+  # does not touch the intermediate pipes. They must be
+  # wired together before being passed in to this object.
+  # This allows me to build a pipeline in Pacer and then
+  # pass it on to a pipe like FutureFilterPipe that only
+  # knows how to act on a single pipe.
+  class BlackboxPipeline
+    include com.tinkerpop.pipes.Pipe 
+    
+    def initialize(start_pipe, end_pipe)
+      @start_pipe = start_pipe
+      @end_pipe = end_pipe
+    end
+
+    def setStarts(pipe)
+      if pipe.respond_to? :iterator
+        @start_pipe.setStarts pipe.iterator
+      else
+        @start_pipe.setStarts pipe
+      end
+    end
+
+    def next
+      @end_pipe.next
+    end
+
+    def hasNext
+      @end_pipe.hasNext
+    end
+
+    def reset
+      @end_pipe.reset
+    end
+
+    def getPath
+      @end_pipe.getPath
+    end
+
+    def iterator
+      @end_pipe.iterator
+    end
+
+    def to_s
+      "[#{ @start_pipe }...#{ @end_pipe }]"
+    end
+  end
+end

data/lib/pacer/pipe/block_filter_pipe.rb

@@ -0,0 +1,38 @@
+module Pacer::Pipes
+  class BlockFilterPipe < AbstractPipe
+    field_reader :starts
+
+    def initialize(back, block, invert = false)
+      super()
+      @back = back
+      @block = block
+      @graph = back.graph
+      @invert = invert
+
+      @exts = @back.extensions + [Pacer::Extensions::BlockFilterElement]
+      @is_element = @graph.element_type?(back.element_type) if @graph
+    end
+
+    def processNextStart()
+      while raw_element = starts.next
+        if @is_element
+          extended_element = raw_element.add_extensions(@exts)
+          extended_element.back = @back
+          extended_element.graph = @back.graph
+          ok = @block.call extended_element
+        else
+          ok = @block.call raw_element
+        end
+        ok = !ok if @invert
+        return raw_element if ok
+      end
+      raise Pacer::NoSuchElementException
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+  end
+end

data/lib/pacer/pipe/collection_filter_pipe.rb

@@ -0,0 +1,10 @@
+module Pacer
+  module Pipes
+    module Renamed
+      AbstractCollectionFilterPipe = com.tinkerpop.pipes.filter.CollectionFilterPipe
+    end
+
+    class CollectionFilterPipe < Renamed::AbstractCollectionFilterPipe
+    end
+  end
+end

data/lib/pacer/pipe/cross_product_transform_pipe.rb

@@ -0,0 +1,48 @@
+module Pacer::Pipes
+  class CrossProductTransformPipe < UnaryTransformPipe
+    attr_reader :branch_b
+
+    def initialize(method, branch_a, branch_b)
+      super(method, branch_a)
+      if branch_b.is_a? Pipe
+        @branch_b = branch_b
+        @b = nil
+      else
+        @b = branch_b
+      end
+    end
+
+    def processNextStart
+      next_pair
+      a.send method, b rescue nil if a.respond_to? method
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+
+    protected
+
+    attr_accessor :b
+
+    def next_pair
+      if branch_b
+        begin
+          self.b = branch_b.next
+        rescue NativeException => e
+          if e.cause.getClass == Pacer::NoSuchElementException.getClass or @first
+            next_a
+            branch_b.setStarts SingleIterator.new(element)
+            retry
+          else
+            raise e
+          end
+        end
+      else
+        next_a
+      end
+    end
+  end
+end

data/lib/pacer/pipe/enumerable_pipe.rb

@@ -0,0 +1,30 @@
+module Pacer::Pipes
+  class EnumerablePipe < RubyPipe
+    def initialize(enumerable)
+      super()
+      case enumerable
+      when Enumerator
+        starts = enumerable
+      when Pacer::ElementMixin
+        starts = [enumerable].to_enum
+      when Enumerable
+        starts = enumerable.to_enum
+      else
+        starts = [enumerable].to_enum
+      end
+      set_starts starts
+    end
+
+    def processNextStart()
+      @starts.next
+    rescue StopIteration
+      raise Pacer::NoSuchElementException
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+  end
+end

data/lib/pacer/pipe/expandable_pipe.rb

@@ -0,0 +1,63 @@
+module Pacer::Pipes
+  class ExpandablePipe < RubyPipe
+    def initialize
+      super()
+      @queue = java.util.LinkedList.new
+    end
+
+    def add(element, metadata = nil, path = nil)
+      @queue.add [element, metadata, path]
+    end
+
+    def metadata
+      @metadata
+    end
+
+    def next
+      super
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    ensure
+      @path = @next_path
+      @metadata = @next_metadata
+    end
+
+    protected
+
+    def processNextStart
+      if @queue.isEmpty
+        @next_metadata = nil
+        r = @starts.next
+        if @starts.respond_to? :getPath
+          @next_path = @starts.getPath
+        else
+          @next_path = java.util.ArrayList.new
+        end
+        r
+      else
+        element, @next_metadata, @next_path = @queue.remove
+        element
+      end
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+
+    def getPathToHere
+      path = java.util.ArrayList.new
+      if @path
+        @path.each do |e|
+          path.add e
+        end
+      end
+      path
+    end
+  end
+end

data/lib/pacer/pipe/id_collection_filter_pipe.rb

@@ -0,0 +1,33 @@
+module Pacer::Pipes
+  class IdCollectionFilterPipe < RubyPipe
+    def initialize(ids, comparison)
+      super()
+      @ids = Set[*ids]
+      @comparison = comparison
+    end
+
+    def processNextStart
+      if @comparison == Pacer::Pipes::EQUAL
+        while true
+          element = @starts.next
+          if element and @ids.include? element.element_id
+            return element
+          end
+        end
+      else
+        while true
+          element = @starts.next
+          if element and not @ids.include? element.element_id
+            return element
+          end
+        end
+      end
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+  end
+end

data/lib/pacer/pipe/is_empty_pipe.rb

@@ -0,0 +1,30 @@
+module Pacer::Pipes
+  class IsEmptyPipe < RubyPipe
+    def initialize
+      super
+      @raise = false
+    end
+
+    def processNextStart
+      raise Pacer::NoSuchElementException if @raise
+      @starts.next
+      @raise = true
+    rescue NativeException => e
+      if e.cause.getClass == NoSuchElementException.getClass
+        # This is the only case where we return true.
+        # The only time we get here is if the first call to next
+        # has no results.
+        true
+      else
+        raise e
+      end
+    else
+      raise Pacer::NoSuchElementException
+    end
+
+    def reset
+      @raise = false
+      super()
+    end
+  end
+end

data/lib/pacer/pipe/is_unique_pipe.rb

@@ -0,0 +1,61 @@
+module Pacer::Pipes
+  class IsUniquePipe < RubyPipe
+    import java.util.ArrayList
+    import com.tinkerpop.pipes.sideeffect.SideEffectPipe
+    import com.tinkerpop.pipes.util.ExpandableIterator
+    import com.tinkerpop.pipes.filter.DuplicateFilterPipe
+
+    include SideEffectPipe
+
+    def initialize
+      super()
+      prepare_state
+    end
+
+    def processNextStart
+      value = starts.next
+      check_uniqueness value if @unique
+      value
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+
+    def reset
+      super()
+      prepare_state
+    end
+
+    def unique?
+      @unique
+    end
+
+    def getSideEffect
+      @unique
+    end
+
+    protected
+
+    def check_uniqueness(value)
+      @expando.add value
+      @unique_pipe.next
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        @unique = false
+      else
+        raise e
+      end
+    end
+
+    def prepare_state
+      @unique = true
+      @expando = ExpandableIterator.new ArrayList.new.iterator
+      @unique_pipe = DuplicateFilterPipe.new
+      @unique_pipe.setStarts @expando
+    end
+
+  end
+end

data/lib/pacer/pipe/label_collection_filter_pipe.rb

@@ -0,0 +1,21 @@
+module Pacer::Pipes
+  class LabelCollectionFilterPipe < RubyPipe
+    def initialize(labels)
+      super()
+      @labels = Set[*labels.map(&:to_s)]
+    end
+
+    def processNextStart
+      while true
+        edge = @starts.next
+        return edge if edge and @labels.include? edge.label
+      end
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+  end
+end

data/lib/pacer/pipe/label_prefix_pipe.rb

@@ -0,0 +1,21 @@
+module Pacer::Pipes
+  class LabelPrefixPipe < RubyPipe
+    def initialize(prefix)
+      super()
+      @prefix = /^#{prefix}/
+    end
+
+    def processNextStart
+      while true
+        edge = @starts.next
+        return edge if edge.label =~ @prefix
+      end
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+  end
+end

data/lib/pacer/pipe/loop_pipe.rb

@@ -0,0 +1,86 @@
+module Pacer::Pipes
+  class LoopPipe < RubyPipe
+    import java.util.ArrayList
+
+    def initialize(graph, looping_pipe, control_block)
+      super()
+      @control_block = control_block
+
+      @expando = ExpandablePipe.new
+      empty = ArrayList.new
+      @expando.setStarts empty.iterator
+      looping_pipe.setStarts(@expando)
+      @looping_pipe = looping_pipe
+    end
+
+    def next
+      super
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    ensure
+      @path = @next_path
+    end
+
+    def setStarts(starts)
+      @starts_has_path = starts.respond_to? :getPath
+      super
+    end
+
+    protected
+
+    def processNextStart
+      while true
+        # FIXME: hasNext shouldn't be raising an exception...
+        has_next = @looping_pipe.hasNext rescue nil
+        if has_next
+          element = @looping_pipe.next
+          depth = (@expando.metadata || 0) + 1
+          @next_path = @looping_pipe.getPath
+        else
+          element = @starts.next
+          if @starts_has_path
+            @next_path = @starts.getPath
+          else
+            @next_path = ArrayList.new
+            @next_path.add element
+          end
+          depth = 0
+        end
+        element.graph ||= @graph if element.respond_to? :graph=
+        case @control_block.call element, depth, @next_path
+        when :loop
+          @expando.add element, depth, @next_path
+        when :emit
+          return element
+        when :emit_and_loop, :loop_and_emit
+          @expando.add element, depth, @next_path
+          return element
+        when false, nil
+        else
+          @expando.add element, depth, @next_path
+          return element
+        end
+      end
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+
+    def getPathToHere
+      path = ArrayList.new
+      if @path
+        @path.each do |e|
+          path.add e
+        end
+      end
+      path
+    end
+  end
+end

data/lib/pacer/pipe/map_pipe.rb

@@ -0,0 +1,36 @@
+module Pacer::Pipes
+  class MapPipe < AbstractPipe
+    field_reader :starts
+
+    def initialize(back, block)
+      super()
+      @block = block
+      @back = back
+      @graph = back.graph if back
+      @extensions = back.extensions + [Pacer::Extensions::BlockFilterElement]
+      @is_element = @graph.element_type?(back.element_type) if @graph
+    end
+
+    def processNextStart
+      while true
+        obj = starts.next
+        begin
+          if @is_element
+            obj = obj.add_extensions(@extensions)
+            obj.back = @back
+            obj.graph = @back.graph
+          end
+        rescue
+        end
+        result = @block.call(obj)
+        return result
+      end
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+  end
+end

data/lib/pacer/pipe/never_pipe.rb

@@ -0,0 +1,9 @@
+module Pacer::Pipes
+  class NeverPipe < RubyPipe
+    protected
+
+    def processNextStart
+      raise NoSuchElementException
+    end
+  end
+end

data/lib/pacer/pipe/process_pipe.rb

@@ -0,0 +1,37 @@
+module Pacer::Pipes
+  class ProcessPipe < AbstractPipe
+    field_reader :starts
+    attr_reader :is_element, :extensions, :back, :block, :graph
+
+    def initialize(back, block)
+      super()
+      @block = block
+      @back = back
+      @graph = back.graph if back
+      @extensions = back.extensions + [Pacer::Extensions::BlockFilterElement]
+      @is_element = graph.element_type?(back.element_type) if graph
+    end
+
+    def processNextStart
+      while true
+        obj = starts.next
+        begin
+          if is_element
+            obj = obj.add_extensions(extensions)
+            obj.back = back
+            obj.graph = graph
+          end
+        rescue
+        end
+        result = block.call(obj)
+        return obj
+      end
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+  end
+end

data/lib/pacer/pipe/property_comparison_pipe.rb

@@ -0,0 +1,40 @@
+module Pacer::Pipes
+  class PropertyComparisonFilterPipe < RubyPipe
+    def initialize(left, right, filter)
+      super()
+      @filter = filter
+      @left = left.to_s
+      @right = right.to_s
+    end
+
+    protected
+
+    def processNextStart
+      while true
+        obj = @starts.next
+        l = obj.getProperty(@left)
+        r = obj.getProperty(@right)
+        case @filter
+        when FilterPipe::Filter::EQUAL
+          return obj if l == r
+        when FilterPipe::Filter::NOT_EQUAL
+          return obj if l != r
+        when FilterPipe::Filter::GREATER_THAN
+          return obj if l and r and l > r
+        when FilterPipe::Filter::LESS_THAN
+          return obj if l and r and l < r
+        when FilterPipe::Filter::GREATER_THAN_EQUAL
+          return obj if l and r and l >= r
+        when FilterPipe::Filter::LESS_THAN_EQUAL
+          return obj if l and r and l <= r
+        end
+      end
+    rescue NativeException => e
+      if e.cause.getClass == Pacer::NoSuchElementException.getClass
+        raise e.cause
+      else
+        raise e
+      end
+    end
+  end
+end

data/lib/pacer/pipe/ruby_pipe.rb

@@ -0,0 +1,25 @@
+module Pacer::Pipes
+  class RubyPipe < AbstractPipe
+    attr_reader :starts
+
+    def setStarts(starts)
+      @starts = starts
+    end
+    alias set_starts setStarts
+
+    def reset
+      super
+      @starts.reset if starts.respond_to? :reset
+    end
+
+    protected
+
+    def getPathToHere
+      if starts.respond_to? :getPath
+        starts.getPath
+      else
+        java.util.ArrayList.new
+      end
+    end
+  end
+end