tree.php

<?php
 
 
class XoopsObjectTree
{

    public $_parentId;
    public $_myId;
    public $_rootId = null;
    public $_tree = [];
    public $_objects;

    public function __construct(&$objectArr, $myId, $parentId, $rootId = null)
    {
        $this->_objects =& $objectArr;
        $this->_myId = $myId;
        $this->_parentId = $parentId;
        if (isset($rootId)) {
            $this->_rootId = $rootId;
        }
        $this->_initialize();
    }

    public function _initialize()
    {
        foreach (array_keys($this->_objects) as $i) {
            $key1 = $this->_objects[$i]->getVar($this->_myId);
            $this->_tree[$key1]['obj'] =& $this->_objects[$i];
            $key2 = $this->_objects[$i]->getVar($this->_parentId);
            $this->_tree[$key1]['parent'] = $key2;
            $this->_tree[$key2]['child'][] = $key1;
            if (isset($this->_rootId)) {
                $this->_tree[$key1]['root'] = $this->_objects[$i]->getVar($this->_rootId);
            }
        }
    }

    public function &getTree()
    {
        return $this->_tree;
    }

    public function &getByKey($key)
    {
        return $this->_tree[$key]['obj'];
    }

    public function &getFirstChild($key)
    {
        $ret = [];
        if (isset($this->_tree[$key]['child'])) {
            foreach ($this->_tree[$key]['child'] as $childkey) {
                $ret[$childkey] =& $this->_tree[$childkey]['obj'];
            }
        }
        return $ret;
    }

    public function &getAllChild($key, $ret = [])
    {
        if (isset($this->_tree[$key]['child'])) {
            foreach ($this->_tree[$key]['child'] as $childkey) {
                $ret[$childkey] =& $this->_tree[$childkey]['obj'];
                $children =& $this->getAllChild($childkey, $ret);
                foreach (array_keys($children) as $newkey) {
                    $ret[$newkey] =& $children[$newkey];
                }
            }
        }
        return $ret;
    }

    public function &getAllParent($key, $ret = [], $uplevel = 1)
    {
        if (isset($this->_tree[$key]['parent'], $this->_tree[$this->_tree[$key]['parent']]['obj'])) {
            $ret[$uplevel] =& $this->_tree[$this->_tree[$key]['parent']]['obj'];
            $parents =& $this->getAllParent($this->_tree[$key]['parent'], $ret, $uplevel+1);
            foreach (array_keys($parents) as $newkey) {
                $ret[$newkey] =& $parents[$newkey];
            }
        }
        return $ret;
    }

    public function _makeSelBoxOptions($fieldName, $selected, $key, &$ret, $prefix_orig, $prefix_curr = '')
    {
        if ($key > 0) {
            $value = $this->_tree[$key]['obj']->getVar($this->_myId);
            $ret .= '<option value="'.$value.'"';
            if ($value == $selected) {
                $ret .= ' selected="selected"';
            }
            $ret .= '>'.$prefix_curr.$this->_tree[$key]['obj']->getVar($fieldName).'</option>';
            $prefix_curr .= $prefix_orig;
        }
        if (isset($this->_tree[$key]['child']) && !empty($this->_tree[$key]['child'])) {
            foreach ($this->_tree[$key]['child'] as $childkey) {
                $this->_makeSelBoxOptions($fieldName, $selected, $childkey, $ret, $prefix_orig, $prefix_curr);
            }
        }
    }

    public function &makeSelBox($name, $fieldName, $prefix='-', $selected='', $addEmptyOption = false, $key=0)
    {
        $ret = '<select name="'.$name.'" id="'.$name.'">';
        if (false !== $addEmptyOption) {
            $ret .= '<option value="0"></option>';
        }
        $this->_makeSelBoxOptions($fieldName, $selected, $key, $ret, $prefix);
        $ret .= '</select>';
        return $ret;
    }
}