summaryrefslogtreecommitdiff
path: root/ext/spl/examples/recursiveiteratoriterator.inc
blob: 95d52d7f7626fee313ce8fc21cb51a03dbfc4507 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
<?php

/**
 * @brief   Iterates through recursive iterators
 * @author  Marcus Boerger
 * @version 1.0
 *
 */
class RecursiveIteratorIterator implements Iterator
{
	protected $ait = array();
	protected $count = 0;

	function __construct(RecursiveIterator $it) {
		$this->count = 1;
		$this->ait[0] = $it;
	}


	function rewind() {
		while ($this->count > 1) {
			unset($this->ait[--$this->count]);
		}
		$this->ait[0]->rewind();
		$this->ait[0]->recursed = false;
	}
	
	function hasMore() {
		$count = $this->count;
		while ($count--) {
			$it = $this->ait[$count];
			if ($it->hasMore()) {// || (!$it->recursed && $it->isRecursive())) {
				return true;
			}
		}
		return false;
	}
	
	function key() {
		$it = $this->ait[$this->count-1];
		return $it->key();
	}
	
	function current() {
		$it = $this->ait[$this->count-1];
		return $it->current();
	}
	
	function next() {
		while ($this->count) {
			$it = $this->ait[$this->count-1];
			if ($it->hasMore()) {
				if (!$it->recursed && $it->hasChildren()) {
					$it->recursed = true;
					$sub = $it->getChildren();
					$sub->recursed = false;
					$sub->rewind();
					if ($sub->hasMore()) {
						$this->ait[$this->count++] = $sub;
						if (!is_a($sub, 'RecursiveIterator')) {
							throw new Exception(get_class($sub).'::getChildren() must return an object that implements RecursiveIterator');
						} 
						return;
					}
					unset($sub);
				}
				$it->next();
				$it->recursed = false;
				if ($it->hasMore()) {
					return;
				}
				$it->recursed = false;
			}
			if ($this->count <= 1) {
				return;
			}
			unset($this->ait[--$this->count]);
			$it = $this->ait[$this->count-1];
		}
	}
	
	function getCurrentIterator() {
		return $this->ait[$this->count-1];
	}
}

?>