Contents

module Flutterby
  # A helper module with methods to walk across a node tree in various
  # directions and variations and perform a block of code on each passed node.
  #
  module TreeWalker
    extend self

    # Walk the tree up, invoking the passed block for every node
    # found on the way, passing the node as its only argument.
    #
    def walk_up(node, val = nil, &blk)
      val = blk.call(node, val)
      node.parent ? walk_up(node.parent, val, &blk) : val
    end

    # Walk the graph from the root to the specified node. Just like {#walk_up},
    # except the block will be called on higher level nodes first.
    #
    def walk_down(node, val = nil, &blk)
      val = node.parent ? walk_up(node.parent, val, &blk) : val
      blk.call(node, val)
    end

    # Walk the entire tree, top to bottom, starting with its root, and then
    # descending into its child layers.
    #
    def walk_tree(node, val = nil, &blk)
      val = blk.call(node, val)

      node.children.each do |child|
        val = walk_tree(child, val, &blk)
      end

      val
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
flutterby-0.6.2	lib/flutterby/tree_walker.rb
flutterby-0.6.1	lib/flutterby/tree_walker.rb
flutterby-0.6.0	lib/flutterby/tree_walker.rb
flutterby-0.5.2	lib/flutterby/tree_walker.rb
flutterby-0.5.1	lib/flutterby/tree_walker.rb
flutterby-0.5.0	lib/flutterby/tree_walker.rb