Contents

# frozen_string_literal: true

module RShade
  class EventTree
    include Enumerable

    attr_reader :current, :head

    def initialize
      @current = @head = EventTreeNode.new(nil, 0, nil)
    end

    # rubocop:disable Metrics/AbcSize
    def add(value, level)
      if current.level + 1 == level
        current.children << EventTreeNode.new(value, level, current)
        return
      end
      if current.level + 1 < level

        last = current.children.last
        current.children << EventTreeNode.new(nil, current.level + 1, current) unless last
        @current = current.children.last
        add(value, level)
        return
      end

      return unless current.level + 1 > level
      return unless current.parent

      @current = current.parent
      add(value, level)
    end
    # rubocop:enable Metrics/AbcSize

    def current!(&block)
      block.call(current.children.last) if current.children.last
    end

    def each(&block)
      @head.each(&block)
    end
  end

  class EventTreeNode
    include Enumerable
    attr_reader :children, :level, :vlevel, :value
    attr_accessor :parent

    def initialize(value, level, parent = nil)
      @children = []
      @level = level
      @parent = parent
      @value = value
      @vlevel = vlevel!(parent)
    end

    def each(&block)
      block.call(self) if !parent.nil? || !value.nil?
      children.each { |item| item.each(&block) }
    end

    private

    def vlevel!(node)
      return 0 if node.nil?

      if node.value || node.level.zero?
        node.vlevel + 1
      else
        vlevel!(node.parent)
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
rshade-0.2.2	lib/rshade/event_tree.rb
rshade-0.2.1	lib/rshade/event_tree.rb