自己实现的android树控件，android TreeView

　　1.开发原因

　　在项目中经常需要一个需要一个树状框架，这是非常常见的控件。不过可能是谷歌考虑到android是手机系统，界面宽度有限，

所以只提供了只有二级的ExpandableListView。虽然这个控件可以满足很多需求，但是无数级的树在某些情况下还是需要的，所以我花了一天时间

（大部分时间都在调试动画去了，不过现在动画还有点问题，具体原因不明。。如果某位大神能找到原因灰常感谢）。

注：今早起来终于修复了最后一个bug，现在的动画效果已经非常完美了，等下就把加了注释的代码贴上来。

　　2.原理

　　网上很多都是扩展listview实现的，不过listview貌似不支持复杂控件的事件？而且做动画也不方便，所有我决定扩展linearlayout，在里面增加子节点的方式实现。

　　3.上图

图片是gif，真实效果要更加流畅一些，但是bug也还在的，就是开始隐藏的节点无法获取高度，可以通过自己事先设置解决。

　　3.代码

TreeView.java:

 package net.memornote.android.ui.view;

 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
 import java.util.Timer;
 import java.util.TimerTask;

 import android.content.Context;
 import android.graphics.Rect;
 import android.util.AttributeSet;
 import android.view.View;
 import android.view.animation.Animation;
 import android.view.animation.Animation.AnimationListener;
 import android.view.animation.ScaleAnimation;
 import android.view.animation.TranslateAnimation;
 import android.widget.LinearLayout;

 public class TreeView extends LinearLayout{

 //    private List<TreeItem> items;
     private List<TreeItem> sortedItems;
     private int animTime;

     public TreeView(Context context, AttributeSet attrs) {
         super(context, attrs);
         setOrientation(LinearLayout.VERTICAL);
     }
     /**
      * initialize data,you must make sure that each item has parent except the top ones.
      * @param items the data to show
      * @param index the index of the tree to insert to
      * @param viewHeight each view's height
      * @param animTime if you want expand animation,
      * you can set the time(ms) of animation,otherwise you can set 0.
      *
      */
     public void initData(Collection<TreeItem> items,int index){

         if(items==null||items.size()==0){
             return ;
         }

         sortItemList(items);

         int size=sortedItems.size();

         initAddIndex=index<0?-Integer.MAX_VALUE:index;

         for (int i=0;i<size;i++) {
             TreeItem item=sortedItems.get(i);
             recuseShow(item);
         }

     }

     private boolean isAnim=false;
     /**
      * 这个方法还有很 严重的bug，无法使用。。
      * 设置为0则关闭动画
      * @param animTime
      */
     public void enabledAnim(int animTime) {
         if(animTime<0){
             isAnim=false;
             return ;
         }
         this.animTime=animTime;
         isAnim=true;
     }

     private int initAddIndex; 

     private void recuseShow(TreeItem item){
         View view=item.getView();
         addView(view,initAddIndex);
         if(item.getParent()!=null){
             view.setVisibility(View.GONE);
         }else {
             view.setVisibility(View.VISIBLE);
         }
         initAddIndex++;
         List<TreeItem> childrens=item.getChildrens();
         if(childrens.size()>0){
             for (TreeItem it : childrens) {
                 recuseShow(it);
             }
         }
     }

     private void sortItemList(Collection<TreeItem> items) {
         //把items按照层级关系存放，sortedItems只存放顶层的item
         sortedItems=new ArrayList<TreeItem>(5);
         for (TreeItem item : items) {
             if(item.getParent()==null){
                 sortedItems.add(item);
             }else {
                 item.getParent().getChildrens().add(item);
             }
         }

     }

     private int viewIndex=0;
     private int animHeight=0;
     private void addChild(TreeItem item,boolean isRecurse){
         if(item.getChildrens().size()>0){
             List<TreeItem> list=item.getChildrens();
             for (TreeItem it :    list) {
                 View view=it.getView();
                 if(view.isShown()){
                     continue;
                 }
                 viewIndex++;
                 view.setVisibility(View.VISIBLE);
                 if(isAnim){
                     animHeight-=it.getViewHeight();
                 }
                 it.nextIsExpand=true;
                 if(isRecurse){
                     addChild(it,true);
                 }
             }
         }
     }
     private int removeCount=0;
     private synchronized void removeChild(TreeItem item,boolean isRecurse){
         if(item.getChildrens().size()>0){
             List<TreeItem> list=item.getChildrens();
             for (TreeItem it :    list) {
                 View view=it.getView();
                 if(!view.isShown()){
                     continue;
                 }

                 TranslateAnimation ta=new TranslateAnimation(0, 0, 0, 0);
                 ta.setFillBefore(true);
                 ta.setDuration(1000);
                 view.startAnimation(ta);
                 removeCount++;
                 view.setVisibility(View.GONE);
                 if(isAnim){
                     animHeight+=it.getViewHeight();
                 }
                 if(isRecurse){
                     removeChild(it,true);
                 }
             }
         }
     }

     private void animAdd(){
         TranslateAnimation ta=new TranslateAnimation(
                 Animation.ABSOLUTE, 0,
                 Animation.ABSOLUTE, 0,
                 Animation.ABSOLUTE, animHeight,
                 Animation.ABSOLUTE, 0);
         ta.setFillBefore(true);
         ta.setFillAfter(false);
         ta.setDuration(animTime);

         for (int i = viewIndex+1; i < getChildCount(); i++) {
             View view=getChildAt(i);
             if(view.isShown()){
                 view.startAnimation(ta);
             }
         }
         animHeight=0;
     }
     private void animRemove(){
         TranslateAnimation ta=new TranslateAnimation(
                 Animation.ABSOLUTE, 0,
                 Animation.ABSOLUTE, 0,
                 Animation.ABSOLUTE, animHeight,
                 Animation.ABSOLUTE, 0);
         ta.setFillAfter(false);
         ta.setFillBefore(true);
         ta.setDuration(animTime);

         int startAnimIndex;
         startAnimIndex=viewIndex+1;
         for (int i = startAnimIndex; i < getChildCount(); i++) {
             View view=getChildAt(i);
             if(view.isShown()){
                 view.startAnimation(ta);
             }
         }
         animHeight=0;
     }
     public void expand(TreeItem item){
         viewIndex=indexOfChild(item.getView());
         addChild(item,false);
         if(isAnim){
             animAdd();
         }
     }

     public void expandAllChildren(TreeItem item) {
         viewIndex=indexOfChild(item.getView());
         addChild(item,true);
         if(isAnim){
             animAdd();
         }
     }

     public void expandAll(){
         if(sortedItems==null){
             return ;
         }
         for (TreeItem item : sortedItems) {
             expandAllChildren(item);
         }
     }

     public void contractAllChildren(TreeItem item) {
         viewIndex=indexOfChild(item.getView())+1;
         removeChild(item,true);
         if(isAnim){
             animRemove();
         }
     }

     public void contractAll(){
         if(sortedItems==null){
             return ;
         }
         for (TreeItem item : sortedItems) {
             contractAllChildren(item);
         }
     }

     public void bind(TreeItem item) {
         if(item.nextIsExpand){
             expand(item);
         }else {
             contractAllChildren(item);
         }
         item.nextIsExpand=!item.nextIsExpand;
     }

 }

TreeView

TreeItem.java：

package net.memornote.android.ui.view;

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

import android.view.View;

public class TreeItem {
    private View view;
    private TreeItem parent;
    private List<TreeItem> childrens=new ArrayList<TreeItem>(0);
    public boolean nextIsExpand=true;
    private int viewHeight;  

    public TreeItem(){}
    /**
     * 初始化TreeItem
     * @param view
     * @param parent
     */
    public TreeItem(View view, TreeItem parent) {
        super();
        this.view = view;
        this.parent = parent;
    }
    /**
     * 初始化TreeItem
     * @param view
     * @param parent
     */
    public TreeItem(View view, TreeItem parent,int viewHeight) {
        super();
        this.view = view;
        this.parent = parent;
        this.viewHeight=viewHeight;
    }

    public View getView() {
        if(view!=null){
            view.setPadding(getLevel()*20,0,0,0);
        }
        return view;
    }

    public void setView(View view) {
        this.view = view;
    }

    public TreeItem getParent() {
        return parent;
    }

    public void setParent(TreeItem parent) {
        this.parent = parent;
    }

    /**
     * 动态获取该节点的级数
     * @return
     */
    public int getLevel() {
        int level=0;
        TreeItem localParent=parent;

        while (localParent!=null) {
            level++;
            localParent=localParent.getParent();
        }

        return level;
    }
    public List<TreeItem> getChildrens() {
        return childrens;
    }
    public int getViewHeight() {
        if(view==null||view.getHeight()==0){
            return viewHeight;
        }
        return view.getHeight();
    }
    public void setViewHeight(int viewHeight) {
        this.viewHeight = viewHeight;
    }

}

TreeItem

测试代码:

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

 import net.yunstudio.util.view.treeview.TreeItem;
 import net.yunstudio.util.view.treeview.TreeView;
 import android.os.Bundle;
 import android.app.Activity;
 import android.view.Menu;
 import android.view.View;
 import android.view.View.OnClickListener;
 import android.widget.Button;

 public class MainActivity extends Activity {

     private TreeView treeView;
     @Override
     protected void onCreate(Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
         setContentView(R.layout.activity_main);
         treeView=(TreeView) findViewById(R.id.treeview);
         List<TreeItem> items=new ArrayList<TreeItem>();

         initData(items);
         treeView.initData(items, 0);
         treeView.enabledAnim(500);

     }
     //初始化一些测试数据
     public void initData(List<TreeItem> items) {
         for (int i=0;i<10;i++) {
 //            TextView tv=new TextView(getActivity());
             Button button=new Button(this);
             button.setText("item"+i);
             final TreeItem item=new TreeItem(button, null);

             button.setOnClickListener(new OnClickListener() {
                 @Override
                 public void onClick(View v) {
                     treeView.bind(item);
                 }
             });

             items.add(item);

             if(i%2==0){
                 Button bt1=new Button(this);
                 bt1.setText("item"+i);
                 bt1.setClickable(true);
                 final TreeItem item_1=new TreeItem(bt1, item);

                 bt1.setOnClickListener(new OnClickListener() {
                     @Override
                     public void onClick(View v) {

                         treeView.bind(item_1);
                     }
                 });
                 items.add(item_1);

                 if(i%4==0){
                     Button bt_2=new Button(this);
                     bt_2.setText("item"+i);
                     bt_2.setClickable(true);
                     final TreeItem item_2=new TreeItem( bt_2, item_1);

                     bt_2.setOnClickListener(new OnClickListener() {
                         @Override
                         public void onClick(View v) {
                             treeView.bind(item_2);
                         }
                     });
                     items.add(item_2);
                 }

             }

         }
     }

     @Override
     public boolean onCreateOptionsMenu(Menu menu) {
         // Inflate the menu; this adds items to the action bar if it is present.
         getMenuInflater().inflate(R.menu.main, menu);
         return true;
     }

 }

源码下载:https://github.com/yzhen334/android_treeview