summaryrefslogtreecommitdiff
path: root/www/wiki/extensions/SemanticResultFormats/src/Outline/OutlineTree.php
blob: 3a3ecabb8f14d0ce1c820189347fbac1984887eb (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
88
89
90
<?php

namespace SRF\Outline;

/**
 * A tree structure for holding the outline data
 *
 * @license GNU GPL v2+
 * @since 3.1
 */
class OutlineTree {

	/**
	 * @var []
	 */
	public $tree;

	/**
	 * @var []
	 */
	public $items;

	/**
	 * @var integer
	 */
	public $itemCount = 0;

	/**
	 * @var integer
	 */
	public $leafCount = 0;

	/**
	 * @since 3.1
	 *
	 * @param array $items
	 */
	public function __construct( $items = [] ) {
		$this->tree = [];
		$this->items = $items;
	}

	/**
	 * @since 3.1
	 *
	 * @param $item
	 */
	public function addItem( $item ) {
		$this->items[] = $item;
		$this->itemCount++;
	}

	/**
	 * @since 3.1
	 *
	 * @param $vals
	 * @param $item
	 */
	public function categorizeItem( $vals, $item ) {
		foreach ( $vals as $val ) {
			if ( array_key_exists( $val, $this->tree ) ) {
				$this->tree[$val]->items[] = $item;
				$this->tree[$val]->leafCount++;
			} else {
				$this->tree[$val] = new self( [ $item ] );
				$this->tree[$val]->leafCount++;
			}
		}
	}

	/**
	 * @since 3.1
	 *
	 * @param $property
	 */
	public function addProperty( $property ) {
		if ( $this->items !== null && count( $this->items ) > 0 ) {
			foreach ( $this->items as $item ) {
				$cur_vals = $item->getFieldValues( $property );
				$this->categorizeItem( $cur_vals, $item );
			}
			$this->items = null;
		} else {
			foreach ( $this->tree as $i => $node ) {
				$this->tree[$i]->addProperty( $property );
			}
		}
	}

}