com.centit.support.common.TreeNode Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of centit-utils Show documentation
java 常用工具类,作为 apache-commons的补充
There is a newer version: 5.3.2302
package com.centit.support.common;

import java.util.ArrayList;
import java.util.List;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.centit.support.algorithm.ReflectionOpt;
import com.centit.support.algorithm.StringBaseOpt;
@SuppressWarnings("unused")
public class TreeNode {
    /**
     * 存储节点的值
     */
    private T value;
    /**
     * 指向父节点，null表示根节点
     */
    private TreeNode praent;
    /**
     * 子节点列表
     */
    private List> children;

    public TreeNode(){
        this.children = null;
        this.praent = null;
    }

    public TreeNode(T value){
        this.value = value;
        this.children = null;
        this.praent = null;
    }

    public T getValue() {
        return value;
    }

    public void setValue(T value) {
        this.value = value;
    }

    public List> getChildren() {
        return children;
    }

    public void setChildren(List> children) {
        this.children = children;
    }

    public TreeNode addChild(T child) {
        TreeNode treeNode = new TreeNode<>(child);
        this.addChild(treeNode);
        return treeNode;
    }

    public void addChild(TreeNode child) {
        if(this.children ==null)
            this.children = new ArrayList<>();
        child.praent = this;
        this.children.add(child);
    }

    public TreeNode getPraent() {
        return praent;
    }

    public void setPraent(TreeNode praent) {
        this.praent = praent;
    }

    public boolean isLeaf(){
        return children==null || children.size()==0;
    }
    /**
     * 判断是否为根节点
     * @return boolean 是否为跟节点
     */
    public boolean isRoot(){
        return praent==null;
    }

    public JSONObject toJSONObject(String childrenPropertyName){
        JSONObject jo ;

        if(ReflectionOpt.isScalarType(this.getValue().getClass())){
            jo = new JSONObject();
            jo.put("value", StringBaseOpt.objectToString(this.getValue()));
        }else
            jo = (JSONObject) JSON.toJSON(this.getValue());

        if(this.children!=null && this.children.size()>0){
            JSONArray ja = new JSONArray();
            for(TreeNode c: this.children){
                ja.add( c.toJSONObject(childrenPropertyName));
            }
            jo.put(childrenPropertyName, ja);
        }

        return jo;
    }

    public JSONObject toJSONObject(){
        return toJSONObject("children");
    }

    public String toString(){
        return toJSONObject("children").toJSONString();
    }
    /**
     * 计算节点到根节点之间的层数，包括自己和根节点
     * @return 节点到根节点之间的层数，包括自己和根节点
     */
    public int getPathCount() {
        int result = 0;
        for(TreeNode path = this; path != null; path = path.getPraent()) {
            result++;
        }
        return result;
    }
    /**
     *  获得从当前节点一直到跟节点路径上所有节点
     * @return 从当前节点一直到跟节点路径上所有节点
     */
    public TreeNode[] getTreeNodePath() {
        int i = getPathCount();
        @SuppressWarnings("unchecked")
        TreeNode[]  result = new TreeNode [i--];
        for(TreeNode path = this; path != null; path = path.getPraent()) {
            result[i--] = path;
        }
        return result;
    }
    /**
     * 获得根节点
     * @return 根节点
     */
    public TreeNode getRootTreeNode() {
         for(TreeNode path = this; path != null; path = path.getPraent()) {
            if(path.isRoot())
                return path;
        }
        return this;
    }

    /**
     * 获得从当前节点一直到跟节点路径上所有节点的值
     * @return 从当前节点一直到跟节点路径上所有节点的值
     */
    public T [] getPath() {
        int i = getPathCount();
        @SuppressWarnings("unchecked")
        T []  result =(T[]) new Object[i--];
        for(TreeNode path = this; path != null; path = path.getPraent()) {
            result[i--] = path.getValue();
        }
        return result;
    }
    /**
     * 获得根节点的值
     * @return 根节点的值
     */
    public T getRootValue() {

        T rootValue = this.getValue();
        TreeNode path = this.getPraent();
        while(path != null ){
            rootValue = path.getValue();
            path = path.getPraent();
        }
        return rootValue;
   }

}