SortedTreeNodeCollection.php

Go to the documentation of this file.

<?php
namespace TYPO3\CMS\Backend\Tree;

class SortedTreeNodeCollection extends \TYPO3\CMS\Backend\Tree\TreeNodeCollection {

    public function contains(\TYPO3\CMS\Backend\Tree\TreeNode $node) {
        return $this->offsetOf($node) !== -1;
    }

    protected function offsetOf(\TYPO3\CMS\Backend\Tree\TreeNode $node) {
        return $this->binarySearch($node, 0, $this->count() - 1);
    }

    protected function binarySearch(\TYPO3\CMS\Backend\Tree\TreeNode $node, $start, $end) {
        if (!$start && $end - $start >= 2 || $end - $start > 2) {
            $divider = ceil(($end - $start) / 2);
            if ($this->offsetGet($divider)->equals($node)) {
                return $divider;
            } elseif ($this->offsetGet($divider)->compareTo($node) > 0) {
                return $this->binarySearch($node, $start, $divider - 1);
            } else {
                return $this->binarySearch($node, $divider + 1, $end);
            }
        } else {
            if ($this->offsetGet($start)->equals($node)) {
                return $start;
            } elseif ($this->offsetGet($end)->equals($node)) {
                return $end;
            } else {
                return -1;
            }
        }
    }

    protected function normalize() {
        $nodes = array();
        foreach ($this as $node) {
            $nodes[] = $node;
        }
        $this->exchangeArray($nodes);
    }

    public function append(\TYPO3\CMS\Backend\Tree\TreeNode $node) {
        parent::append($node);
        $this->asort();
        $this->normalize();
    }

}