lib_tree.php
1.88 KB
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
91
92
93
<?php
/**
* [Discuz!] (C)2001-2099 Comsenz Inc.
* This is NOT a freeware, use is subject to license terms
*
* $Id: lib_tree.php 23684 2011-08-04 02:24:59Z cnteacher $
*/
if (!defined('IN_DISCUZ')) {
exit('Access Denied');
}
class lib_tree {
public $data = array();
public $child = array(-1 => array());
public $layer = array(-1 => -1);
public $parent = array();
public $countid = 0;
public function __construct() {
}
public function setNode($id, $parent, $value) {
$parent = $parent ? $parent : 0;
$this->data[$id] = $value;
$this->child[$parent][] = $id;
$this->parent[$id] = $parent;
if (!isset($this->layer[$parent])) {
$this->layer[$id] = 0;
} else {
$this->layer[$id] = $this->layer[$parent] + 1;
}
}
public function getList(&$tree, $root= 0) {
foreach ($this->child[$root] as $key => $id) {
$tree[] = $id;
if ($this->child[$id])
$this->getList($tree, $id);
}
}
public function getValue($id) {
return $this->data[$id];
}
public function reSetLayer($id) {
if ($this->parent[$id]) {
$this->layer[$this->countid] = $this->layer[$this->countid] + 1;
$this->reSetLayer($this->parent[$id]);
}
}
public function getLayer($id, $space = false) {
$this->layer[$id] = 0;
$this->countid = $id;
$this->reSetLayer($id);
return $space ? str_repeat($space, $this->layer[$id]) : $this->layer[$id];
}
public function getParent($id) {
return $this->parent[$id];
}
public function getParents($id) {
while ($this->parent[$id] != -1) {
$id = $parent[$this->layer[$id]] = $this->parent[$id];
}
ksort($parent);
reset($parent);
return $parent;
}
public function getChild($id) {
return $this->child[$id];
}
public function getChilds($id = 0) {
$child = array();
$this->getList($child, $id);
return $child;
}
}
?>