<?php

namespace application\extensions;

/**
 * 通用的树型类,可以生成任何树型结构
 */
class Tree
{

 /**
 * 生成树型结构所需要的2维数组
 * @var array
 */
 public $arr = array();

 /**
 * 生成树型结构所需修饰符号,可以换成图片
 * @var array
 */
 public $icon = array('│', '├', '└');
 public $nbsp = "&nbsp;";

 /**
 * @access private
 */
 public $ret = '';

 /**
 * 构造函数,初始化类
 * @param array 2维数组,例如:
 * array(
 * 1 => array('id'=>'1','pid'=>0,'name'=>'一级栏目一'),
 * 2 => array('id'=>'2','pid'=>0,'name'=>'一级栏目二'),
 * 3 => array('id'=>'3','pid'=>1,'name'=>'二级栏目一'),
 * 4 => array('id'=>'4','pid'=>1,'name'=>'二级栏目二'),
 * 5 => array('id'=>'5','pid'=>2,'name'=>'二级栏目三'),
 * 6 => array('id'=>'6','pid'=>3,'name'=>'三级栏目一'),
 * 7 => array('id'=>'7','pid'=>3,'name'=>'三级栏目二')
 * )
 */
 public function init($arr = array())
 {
 $this->arr = $arr;
 $this->ret = '';
 return is_array($arr);
 }

 /**
 * 得到父级数组
 * @param int
 * @return array
 */
 public function get_parent($myid)
 {
 $newarr = array();
 if (!isset($this->arr[$myid]))
 return false;
 $pid = $this->arr[$myid]['pid'];
 $pid = $this->arr[$pid]['pid'];
 if (is_array($this->arr)) {
 foreach ($this->arr as $id => $a) {
 if ($a['pid'] == $pid)
 $newarr[$id] = $a;
 }
 }
 return $newarr;
 }

 /**
 * 得到子级数组
 * @param int
 * @return array
 */
 public function get_child($myid)
 {
 $a = $newarr = array();
 if (is_array($this->arr)) {
 foreach ($this->arr as $id => $a) {
 if ($a['pid'] == $myid)
 $newarr[$id] = $a;
 }
 }
 return $newarr ? $newarr : false;
 }

 /**
 * 得到当前位置数组
 * @param int
 * @return array
 */
 public function get_pos($myid, &$newarr)
 {
 $a = array();
 if (!isset($this->arr[$myid]))
 return false;
 $newarr[] = $this->arr[$myid];
 $pid = $this->arr[$myid]['pid'];
 if (isset($this->arr[$pid])) {
 $this->get_pos($pid, $newarr);
 }
 if (is_array($newarr)) {
 krsort($newarr);
 foreach ($newarr as $v) {
 $a[$v['id']] = $v;
 }
 }
 return $a;
 }

 /**
 * 得到树型结构
 * @param int ID,表示获得这个ID下的所有子级
 * @param string 生成树型结构的基本代码,例如:"<option value=\$id \$selected>\$spacer\$name</option>"
 * @param int 被选中的ID,比如在做树型下拉框的时候需要用到
 * @return string
 */
 public function get_tree($myid, $str, $sid = 0, $adds = '', $str_group = '')
 {
 $number = 1;
 $child = $this->get_child($myid);
 if (is_array($child)) {
 $total = count($child);
 foreach ($child as $id => $value) {
 $j = $k = '';
 if ($number == $total) {
 $j .= $this->icon[2];
 $k = $adds ? $this->nbsp : '';
 } else {
 $j .= $this->icon[1];
 $k = $adds ? $this->icon[0] : '';
 }
 $spacer = $adds ? $adds . $j : '';
 $selected = $id == $sid ? 'selected' : '';
 @extract($value);
 if ($pid == 0 && $str_group) {
 eval("\$nstr = \"$str_group\";");
 } else {
 eval("\$nstr = \"$str\";");
 }
 $this->ret .= $nstr;
 $nbsp = $this->nbsp;
 $this->get_tree($id, $str, $sid, $adds . $k . $nbsp, $str_group);
 $number++;
 }
 }
 return $this->ret;
 }

 /**
 * 同上一方法类似,但允许多选
 */
 public function get_tree_multi($myid, $str, $sid = 0, $adds = '')
 {
 $number = 1;
 $child = $this->get_child($myid);
 if (is_array($child)) {
 $total = count($child);
 foreach ($child as $id => $a) {
 $j = $k = '';
 if ($number == $total) {
 $j .= $this->icon[2];
 } else {
 $j .= $this->icon[1];
 $k = $adds ? $this->icon[0] : '';
 }
 $spacer = $adds ? $adds . $j : '';

 $selected = $this->have($sid, $id) ? 'selected' : '';
 @extract($a);
 eval("\$nstr = \"$str\";");
 $this->ret .= $nstr;
 $this->get_tree_multi($id, $str, $sid, $adds . $k . '&nbsp;');
 $number++;
 }
 }
 return $this->ret;
 }

 /**
 * @param integer $myid 要查询的ID
 * @param string $str 第一种HTML代码方式
 * @param string $str2 第二种HTML代码方式
 * @param integer $sid 默认选中
 * @param integer $adds 前缀
 */
 public function get_tree_category($myid, $str, $str2, $sid = 0, $adds = '')
 {
 $number = 1;
 $child = $this->get_child($myid);
 if (is_array($child)) {
 $total = count($child);
 foreach ($child as $id => $a) {
 $j = $k = '';
 if ($number == $total) {
 $j .= $this->icon[2];
 } else {
 $j .= $this->icon[1];
 $k = $adds ? $this->icon[0] : '';
 }
 $spacer = $adds ? $adds . $j : '';

 $selected = $this->have($sid, $id) ? 'selected' : '';
 @extract($a);
 if (empty($html_disabled)) {
 eval("\$nstr = \"$str\";");
 } else {
 eval("\$nstr = \"$str2\";");
 }
 $this->ret .= $nstr;
 $this->get_tree_category($id, $str, $str2, $sid, $adds . $k . '&nbsp;');
 $number++;
 }
 }
 return $this->ret;
 }

 /**
 * 同上一类方法,jquery treeview 风格,可伸缩样式(需要treeview插件支持)
 * @param $myid 表示获得这个ID下的所有子级
 * @param $effected_id 需要生成treeview目录数的id
 * @param $str 末级样式
 * @param $str2 目录级别样式
 * @param $showlevel 直接显示层级数,其余为异步显示,0为全部限制
 * @param $style 目录样式 默认 filetree 可增加其他样式如'filetree treeview-famfamfam'
 * @param $currentlevel 计算当前层级,递归使用 适用改函数时不需要用该参数
 * @param $recursion 递归使用 外部调用时为FALSE
 */
 function get_treeview($myid, $effected_id = 'example', $str = "<span class='file'>\$name</span>", $str2 = "<span class='folder'>\$name</span>", $showlevel = 0, $style = 'filetree ', $currentlevel = 1, $recursion = FALSE)
 {
 $child = $this->get_child($myid);
 if (!defined('EFFECTED_INIT')) {
 $effected = ' id="' . $effected_id . '"';
 define('EFFECTED_INIT', 1);
 } else {
 $effected = '';
 }
 $placeholder = '<ul><li><span class="placeholder"></span></li></ul>';
 if (!$recursion)
 $this->str .= '<ul' . $effected . ' class="' . $style . '">';
 foreach ($child as $id => $a) {

 @extract($a);
 if ($showlevel > 0 && $showlevel == $currentlevel && $this->get_child($id))
 $folder = 'hasChildren'; //如设置显示层级模式@2011年07月01日
 $floder_status = isset($folder) ? ' class="' . $folder . '"' : '';
 $this->str .= $recursion ? '<ul><li' . $floder_status . ' id=\'' . $id . '\'>' : '<li' . $floder_status . ' id=\'' . $id . '\'>';
 $recursion = FALSE;
 if ($this->get_child($id)) {
 eval("\$nstr = \"$str2\";");
 $this->str .= $nstr;
 if ($showlevel == 0 || ($showlevel > 0 && $showlevel > $currentlevel)) {
 $this->get_treeview($id, $effected_id, $str, $str2, $showlevel, $style, $currentlevel + 1, TRUE);
 } elseif ($showlevel > 0 && $showlevel == $currentlevel) {
 $this->str .= $placeholder;
 }
 } else {
 eval("\$nstr = \"$str\";");
 $this->str .= $nstr;
 }
 $this->str .= $recursion ? '</li></ul>' : '</li>';
 }
 if (!$recursion)
 $this->str .= '</ul>';
 return $this->str;
 }

 /**
 * 获取子栏目json
 * Enter description here ...
 * @param unknown_type $myid
 */
 public function creat_sub_json($myid, $str = '')
 {
 $sub_cats = $this->get_child($myid);
 $n = 0;
 if (is_array($sub_cats))
 foreach ($sub_cats as $c) {
 $data[$n]['id'] = iconv(CHARSET, 'utf-8', $c['catid']);
 if ($this->get_child($c['catid'])) {
 $data[$n]['liclass'] = 'hasChildren';
 $data[$n]['children'] = array(array('text' => '&nbsp;', 'classes' => 'placeholder'));
 $data[$n]['classes'] = 'folder';
 $data[$n]['text'] = iconv(CHARSET, 'utf-8', $c['catname']);
 } else {
 if ($str) {
 @extract(array_iconv($c, CHARSET, 'utf-8'));
 eval("\$data[$n]['text'] = \"$str\";");
 } else {
 $data[$n]['text'] = iconv(CHARSET, 'utf-8', $c['catname']);
 }
 }
 $n++;
 }
 return json_encode($data);
 }

 private function have($list, $item)
 {
 return (strpos(',,' . $list . ',', ',' . $item . ','));
 }

}