95 lines
2.7 KiB
PHP
95 lines
2.7 KiB
PHP
<?php
|
|
namespace com;
|
|
/**
|
|
* 通用的树型类，可以生成任何树型结构
|
|
*/
|
|
class Tree {
|
|
|
|
protected $formatTree;
|
|
|
|
/**
|
|
* 把返回的数据集转换成Tree
|
|
* @param array $list 要转换的数据集
|
|
* @param string $pid parent标记字段
|
|
* @param string $level level标记字段
|
|
* @return array
|
|
* @author 麦当苗儿 <zuojiazi@vip.qq.com>
|
|
*/
|
|
protected function list_to_tree($list, $pk='id', $pid = 'pid', $child = '_child', $root = 0) {
|
|
// 创建Tree
|
|
$tree = array();
|
|
if(is_array($list)) {
|
|
// 创建基于主键的数组引用
|
|
$refer = array();
|
|
foreach ($list as $key => $data) {
|
|
$refer[$data[$pk]] =& $list[$key];
|
|
}
|
|
foreach ($list as $key => $data) {
|
|
// 判断是否存在parent
|
|
$parentId = $data[$pid];
|
|
if ($root == $parentId) {
|
|
$tree[] =& $list[$key];
|
|
}else{
|
|
if (isset($refer[$parentId])) {
|
|
$parent =& $refer[$parentId];
|
|
$parent['childs'][] = $data['id'];
|
|
$parent[$child][] =& $list[$key];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return $tree;
|
|
}
|
|
|
|
/**
|
|
* 将树子节点加层级成列表
|
|
*/
|
|
protected function _toFormatTree($tree, $level = 1) {
|
|
foreach ($tree as $key => $value) {
|
|
$temp = $value;
|
|
if (isset($temp['_child'])) {
|
|
$temp['_child'] = true;
|
|
$temp['level'] = $level;
|
|
} else {
|
|
$temp['_child'] = false;
|
|
$temp['level'] = $level;
|
|
}
|
|
array_push($this->formatTree, $temp);
|
|
if (isset($value['_child'])) {
|
|
$this->_toFormatTree($value['_child'], ($level + 1));
|
|
}
|
|
}
|
|
}
|
|
|
|
protected function cat_empty_deal($cat, $next_parentid, $pid='pid', $empty = " ") {
|
|
$str = "";
|
|
if ($cat[$pid]) {
|
|
for ($i=2; $i < $cat['level']; $i++) {
|
|
$str .= $empty."│";
|
|
}
|
|
if ($cat[$pid] != $next_parentid && !$cat['_child']) {
|
|
$str .= $empty."└─ ";
|
|
} else {
|
|
$str .= $empty."├─ ";
|
|
}
|
|
}
|
|
return $str;
|
|
}
|
|
|
|
public function toFormatTree($list,$title = 'title',$pk='id',$pid = 'pid',$root = 0){
|
|
if (empty($list)) {
|
|
return false;
|
|
}
|
|
$list = $this->list_to_tree($list,$pk,$pid,'_child',$root);
|
|
$this->formatTree = array();
|
|
$this->_toFormatTree($list);
|
|
foreach ($this->formatTree as $key => $value) {
|
|
$index = ($key+1);
|
|
$next_parentid = isset($this->formatTree[$index][$pid]) ? $this->formatTree[$index][$pid] : '';
|
|
$value['level_show'] = $this->cat_empty_deal($value, $next_parentid);
|
|
$value['title_show'] = $value['level_show'].$value[$title];
|
|
$data[] = $value;
|
|
}
|
|
return $data;
|
|
}
|
|
} |