#!/usr/bin/env ruby
# -*- coding: utf-8 -*-

#==========================
# elementlist.rb
#==========================
#
# Contains a list of unordered tree elements with a defined parent
# Copyright (c) 2007-2021 Yoichiro Hasebe <yohasebe@gmail.com>

require 'element'

module RSyntaxTree
  class ElementList

    attr_accessor :elements, :iterator 
    def initialize
      @elements = Array.new # The element array
      @iterator = -1 # Iterator index (used for get_first / get_next)
    end

    def set_hierarchy
      @elements.each do |e|
        get_id(e.parent).add_child(e.id) unless e.parent == 0
      end
    end

    def add(element)
      @elements << element
      if(element.parent != 0)
        parent = get_id(element.parent)
        parent.type = ETYPE_NODE
      end
    end

    def get_first
      if(@elements.length == 0)
        return nil
      else
        @iterator = 0
        return @elements[@iterator]
      end
    end

    def get_next
      @iterator += 1
      if @elements[@iterator]
        return @elements[@iterator]
      else
        return nil
      end
    end

    def get_id(id)
      @elements.each do |element|
        if(element.id == id)
          return element 
        end
      end  
      return nil;
    end

    def get_elements
      @elements
    end

    def get_child_count(id)
      get_children(id).length
    end

    def get_children(id)
      children = Array.new
      @elements.each do |element|
        if(element.parent == id)
          children << element.id
        end
      end
      children
    end

    def get_element_width(id)
      element = get_id(id)
      if element
        return element.width
      else
        return -1;
      end
    end

    def set_element_width(id, width)
      element = get_id(id)
      if element
        element.width = width
      end
    end

    def get_indent(id)
      element = get_id(id)
      if element
        return element.indent
      else
        return -1
      end  
    end

    def set_indent(id, indent)
      element = get_id(id)
      if element
        element.indent = indent
      end
    end

    def get_level_height
      maxlevel = 0
      @elements.each do |element|
        level = element.level
        if(level > maxlevel)
          maxlevel = level
        end
      end
      return maxlevel + 1;
    end
  end
end