一朋友写的,分享给大家。
文章来源:http://www.midnight-soft.cn/index.php/781
<?
php
/* *
+------------------------------------------------
* 通用的树型类
+------------------------------------------------
* @author yangyunzhou@foxmail.com
+------------------------------------------------
* @date 2010年11月23日10:09:31
+------------------------------------------------
*/
class Tree
{
/* *
+------------------------------------------------
* 生成树型结构所需要的2维数组
+------------------------------------------------
* @author yangyunzhou@foxmail.com
+------------------------------------------------
* @var Array
*/
var $arr = array ();
/* *
+------------------------------------------------
* 生成树型结构所需修饰符号,可以换成图片
+------------------------------------------------
* @author yangyunzhou@foxmail.com
+------------------------------------------------
* @var Array
*/
var $icon = array ( ' │ ' , ' ├ ' , ' └ ' );
/* *
* @access private
*/
var $ret = '' ;
/* *
* 构造函数,初始化类
* @param array 2维数组,例如:
* array(
* 1 => array('id'=>'1','parentid'=>0,'name'=>'一级栏目一'),
* 2 => array('id'=>'2','parentid'=>0,'name'=>'一级栏目二'),
* 3 => array('id'=>'3','parentid'=>1,'name'=>'二级栏目一'),
* 4 => array('id'=>'4','parentid'=>1,'name'=>'二级栏目二'),
* 5 => array('id'=>'5','parentid'=>2,'name'=>'二级栏目三'),
* 6 => array('id'=>'6','parentid'=>3,'name'=>'三级栏目一'),
* 7 => array('id'=>'7','parentid'=>3,'name'=>'三级栏目二')
* )
*/
function tree( $arr = array ())
{
$this -> arr = $arr ;
$this -> ret = '' ;
return is_array ( $arr );
}
/* *
* 得到父级数组
* @param int
* @return array
*/
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
*/
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
*/
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 ;
}
/* *
* -------------------------------------
* 得到树型结构
* -------------------------------------
* @author yangyunzhou@foxmail.com
* @param $myid 表示获得这个ID下的所有子级
* @param $str 生成树形结构基本代码, 例如: "<option value=\$id \$select>\$spacer\$name</option>"
* @param $sid 被选中的ID, 比如在做树形下拉框的时候需要用到
* @param $adds
* @param $str_group
*/
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 => $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 = $id == $sid ? ' selected ' : '' ;
@ extract ( $a );
$parentid == 0 && $str_group ? eval ( " \$nstr = \" $str_group \"; " ) : eval ( " \$nstr = \" $str \"; " );
$this -> ret .= $nstr ;
$this -> get_tree( $id , $str , $sid , $adds . $k . ' ' , $str_group );
$number ++ ;
}
}
return $this -> ret;
}
/* *
* 同上一方法类似,但允许多选
*/
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 . ' ' );
$number ++ ;
}
}
return $this -> ret;
}
function have( $list , $item ){
return ( strpos ( ' ,, ' . $list . ' , ' , ' , ' . $item . ' , ' ));
}
/* *
+------------------------------------------------
* 格式化数组
+------------------------------------------------
* @author yangyunzhou@foxmail.com
+------------------------------------------------
*/
function getArray( $myid = 0 , $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 : '' ;
@ extract ( $a );
$a [ ' title ' ] = $spacer . ' ' . $a [ ' title ' ];
$this -> ret[ $a [ ' id ' ]] = $a ;
$fd = $adds . $k . ' ' ;
$this -> getArray( $id , $sid , $fd );
$number ++ ;
}
}
return $this -> ret;
}
}
?>
/* *
+------------------------------------------------
* 通用的树型类
+------------------------------------------------
* @author yangyunzhou@foxmail.com
+------------------------------------------------
* @date 2010年11月23日10:09:31
+------------------------------------------------
*/
class Tree
{
/* *
+------------------------------------------------
* 生成树型结构所需要的2维数组
+------------------------------------------------
* @author yangyunzhou@foxmail.com
+------------------------------------------------
* @var Array
*/
var $arr = array ();
/* *
+------------------------------------------------
* 生成树型结构所需修饰符号,可以换成图片
+------------------------------------------------
* @author yangyunzhou@foxmail.com
+------------------------------------------------
* @var Array
*/
var $icon = array ( ' │ ' , ' ├ ' , ' └ ' );
/* *
* @access private
*/
var $ret = '' ;
/* *
* 构造函数,初始化类
* @param array 2维数组,例如:
* array(
* 1 => array('id'=>'1','parentid'=>0,'name'=>'一级栏目一'),
* 2 => array('id'=>'2','parentid'=>0,'name'=>'一级栏目二'),
* 3 => array('id'=>'3','parentid'=>1,'name'=>'二级栏目一'),
* 4 => array('id'=>'4','parentid'=>1,'name'=>'二级栏目二'),
* 5 => array('id'=>'5','parentid'=>2,'name'=>'二级栏目三'),
* 6 => array('id'=>'6','parentid'=>3,'name'=>'三级栏目一'),
* 7 => array('id'=>'7','parentid'=>3,'name'=>'三级栏目二')
* )
*/
function tree( $arr = array ())
{
$this -> arr = $arr ;
$this -> ret = '' ;
return is_array ( $arr );
}
/* *
* 得到父级数组
* @param int
* @return array
*/
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
*/
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
*/
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 ;
}
/* *
* -------------------------------------
* 得到树型结构
* -------------------------------------
* @author yangyunzhou@foxmail.com
* @param $myid 表示获得这个ID下的所有子级
* @param $str 生成树形结构基本代码, 例如: "<option value=\$id \$select>\$spacer\$name</option>"
* @param $sid 被选中的ID, 比如在做树形下拉框的时候需要用到
* @param $adds
* @param $str_group
*/
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 => $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 = $id == $sid ? ' selected ' : '' ;
@ extract ( $a );
$parentid == 0 && $str_group ? eval ( " \$nstr = \" $str_group \"; " ) : eval ( " \$nstr = \" $str \"; " );
$this -> ret .= $nstr ;
$this -> get_tree( $id , $str , $sid , $adds . $k . ' ' , $str_group );
$number ++ ;
}
}
return $this -> ret;
}
/* *
* 同上一方法类似,但允许多选
*/
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 . ' ' );
$number ++ ;
}
}
return $this -> ret;
}
function have( $list , $item ){
return ( strpos ( ' ,, ' . $list . ' , ' , ' , ' . $item . ' , ' ));
}
/* *
+------------------------------------------------
* 格式化数组
+------------------------------------------------
* @author yangyunzhou@foxmail.com
+------------------------------------------------
*/
function getArray( $myid = 0 , $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 : '' ;
@ extract ( $a );
$a [ ' title ' ] = $spacer . ' ' . $a [ ' title ' ];
$this -> ret[ $a [ ' id ' ]] = $a ;
$fd = $adds . $k . ' ' ;
$this -> getArray( $id , $sid , $fd );
$number ++ ;
}
}
return $this -> ret;
}
}
?>