无限级分类信息格式化工具类

    分类功能在很多web应用中都是必须的, 尤其是商城中尤为重要. 而对无限级分类信息的格式化也是web编程的基本的技能之一. 这里分享一个无限级分类的格式化工具类. 

    本来是将其中的方法都作为静态方法直接使用, 回来发现很多时候大家分类信息中的分类id的名称各不相同, 所以又加了一个构造方法, 指定分类id字段.

<?php
/**
 * 分类信息格式化工具类
 */

class Category {
    protected $sonName;
    protected $parentName;

    /**
     * 初始化工作
     * @param string $son    子类标识的名称
     * @param string $parent 父类标识的名称
     */
    public function __construct($son = 'id', $parent = 'pid') {
        $this->sonName = $son;
        $this->parentName = $parent;
    }

    /**
     * 根据传递的父类ID获取所有的子级分类
     * 组合一维数组
     * @param  [type]  $data 分类数组
     * @param  integer $pid  父类id
     * @param  integer $lev  父类所属层级
     * @return [type]        格式化后的数组信息
     */
    public function unlimitedForLevel($data, $pid = 0, $lev = 0) {
        $arr = array();

        foreach($data as $v) {
            if($v[$this->parentName] == $pid) {
                $v['lev'] = $lev + 1;
                $arr[] = $v;
                $arr = array_merge($arr, self::unlimitedForLevel($data, $v[$this->sonName], $lev + 1));
            }
        }

        return $arr;
    }

    /**
     * 根据传递的父类ID获取所有的子级分类
     * 组合多维数组
     * @param  [type]  $data 分类数组
     * @param  integer $pid  父类id
     * @return [type]        格式化后的数组信息
     */
    public function unlimitedForLayer($data, $pid = 0) {
        $arr = array();

        foreach($data as $v) {
            if($v[$this->parentName] == $pid) {
                $v['children'] = self::unlimitedForLayer($data, $v[$this->sonName]);
                $arr[] = $v;
            }
        }

        return $arr;
    }

    /**
     * 根据传递子类ID获取所有的父级分类
     * @param  [type]  $data 分类数组
     * @param  integer $id   子类id
     * @return [type]        父类数组信息
     */
    public function getParents($data, $id) {
        $arr = array();

        foreach($data as $v) {
            if($v[$this->sonName] == $id) {
                $arr[] = $v;
                $arr = array_merge(self::getParents($data, $v[$this->parentName]), $arr);
            }
        }

        return $arr;
    }

    /**
     * 根据传递的父类ID获取所有的子级分类ID
     * 注意返回值中不包括传递进来的父类ID
     * @param  [type] $data 分类数组
     * @param  [type] $pid  父类id
     * @return [type]       子类id数组
     */
    public function getChildsID($data, $pid) {
        $arr =array();

        foreach($data as $v) {
            if($v[$this->parentName] == $pid) {
                $arr[] = $v[$this->sonName];
                $arr = array_merge($arr, self::getChildsID($data, $v[$this->sonName]));
            }
        }

        return $arr;
    }

    /**
     * 根据传递的子类ID获取所有的父类ID
     * 注意返回值中不包括传递进来的子类ID
     * @param  [type] $data 分类数组
     * @param  [type] $id   子类id
     * @return [type]       父类id数组
     */
    public function getParentsID($data, $id) {
        $arr = array();

        foreach($data as $v) {
            if($v[$this->sonName] == $id) {
                $arr[] = $v[$this->parentName];
                $arr = array_merge($arr, self::getParentsID($data, $v[$this->parentName]));
            }
        }

        return $arr;
    }
}