Android TreeView实现带复选框树形组织结构

public class MainActivity extends AppCompatActivity {
 
 List<Node  list = new ArrayList<Node ();
 private TreeListView listView;
 private RelativeLayout relativeLayout, rl;
 
 @Override
 protected void onCreate(Bundle savedInstanceState) {
 super.onCreate(savedInstanceState);
 setContentView(R.layout.activity_main);
 relativeLayout = (RelativeLayout) findViewById(R.id.main_relative_layout);
 Context context=MainActivity.this;
 rl = new RelativeLayout(context);
 rl.setLayoutParams(new RelativeLayout.LayoutParams(RelativeLayout.LayoutParams.MATCH_PARENT, RelativeLayout.LayoutParams.MATCH_PARENT));
 listView = new TreeListView(context, initNodeTree());
 listView.setLayoutParams(new RelativeLayout.LayoutParams(RelativeLayout.LayoutParams.MATCH_PARENT, RelativeLayout.LayoutParams.MATCH_PARENT));
 relativeLayout.addView(listView);
 }
 public List<Node  initNodeTree() {
 
 List<Node  member_list =new ArrayList<Node ();
// -1表示为根节点,id的作用为标识对象身份，第三个参数此例子中是text文本
 member_list.add(new Node("" + -1, "1" , "111"));
 member_list.add(new Node(""+1 , "2" , "222"));
 member_list.add(new Node("" + -1, "3" , "333"));
 member_list.add(new Node("" + 1, "4" , "444"));
 member_list.add(new Node("" + 4, "5" , "555"));
 member_list.add(new Node("" + 4, "6" , "666"));
 member_list.add(new Node("" + 4, "7" , "777"));
 member_list.add(new Node("" + 7, "8" , "888"));
 member_list.add(new Node("" + 8, "9" , "999"));
 member_list.add(new Node("" + 8, "10" , "101010"));
 list.addAll(member_list);
 return list;
 }
}

public class Node implements Serializable {
private Node parent = null; // 父节点
private List<Node  childrens = new ArrayList<Node ();//子节点
private String value;//节点显示值
private boolean isChecked = false; //是否被选中
private boolean isExpand = true;//是否处于扩展状态
private boolean hasCheckBox = true;//是否有复选框
private String parentId = null;
private String curId = null;
//父节点集合
private List<Node  parents = new ArrayList< ();
/**
* 设置节点值
*
* @param parentId
* TODO
* @param curId
* TODO
*/
public Node( String parentId, String curId, String value) {
// TODO Auto-generated constructor stub
this.value = value;
this.parentId = parentId;
this.curId = curId;
}
public List<Node  getParents() {
return parents;
}
public void setParents(Node node) {
if(node != null) {
if (!parents.contains(node)) {
parents.add(node);
}
}
}
/**
* 得到父节点
*/
public Node getParent() {
return parent;
}
/**
* 设置父节点
* @param parent
*/
public void setParent(Node parent) {
this.parent = parent;
}
/**
* 得到子节点
* @return
*/
public List<Node  getChildrens() {
return childrens;
}
/**
* pandu是否根节点
* @return
*
*/
public boolean isRoot(){
return parent ==null?true:false;
}
/**
* 是否被选中
* @return
*
*/
public boolean isChecked() {
return isChecked;
}
public void setChecked(boolean isChecked) {
this.isChecked = isChecked;
}
/**
* 是否是展开状态
* @return
*
*/
public boolean isExplaned() {
return isExpand;
}
/**
* 设置展开状态
* @param isExplaned
*
*/
public void setExplaned(boolean isExplaned) {
this.isExpand = isExplaned;
}
/**
* 是否有复选框
* @return
*
*/
public boolean hasCheckBox() {
return hasCheckBox;
}
/**
* 设置是否有复选框
* @param hasCheckBox
*
*/
public void setHasCheckBox(boolean hasCheckBox) {
this.hasCheckBox = hasCheckBox;
}
/**
* 得到节点值
* @return
*
*/
public String getValue() {
return value;
}
/**
* 设置节点值
* @param value
*
*/
public void setValue(String value) {
this.value = value;
}
/**
* 增加一个子节点
* @param node
*
*/
public void addNode(Node node){
if(!childrens.contains(node)){
childrens.add(node);
}
}
/**
* 移除一个子节点
* @param node
*
*/
public void removeNode(Node node){
if(childrens.contains(node))
childrens.remove(node);
}
/**
* 移除指定位置的子节点
* @param location
*
*/
public void removeNode(int location){
childrens.remove(location);
}
/**
* 清除所有子节点
*
*/
public void clears(){
childrens.clear();
}
/**
* 判断给出的节点是否当前节点的父节点
* @param node
* @return
*
*/
public boolean isParent(Node node){
if(parent == null)return false;
if(parent.equals(node))return true;
return parent.isParent(node);
}
/**
* 递归获取当前节点级别
* @return
*
*/
public int getLevel(){
return parent ==null?0:parent.getLevel()+1;
}
/**
* 父节点是否处于折叠的状态
* @return
*
*/
public boolean isParentCollapsed(){
if(parent ==null)return false;
if(!parent.isExplaned())return true;
return parent.isParentCollapsed();
}
/**
* 是否叶节点（没有展开下级的几点）
* @return
*
*/
public boolean isLeaf(){
return childrens.size()<1?true:false;
}
/**
* 返回自己的id
* @return
**/
public String getCurId() {
// TODO Auto-generated method stub
return curId;
}
/**
* 返回的父id
* @return
**/
public String getParentId() {
// TODO Auto-generated method stub
return parentId;
}
}

package com.example.administrator.treeview.treeView;
import android.content.Context;
import android.util.Log;
import android.view.LayoutInflater;
import android.view.View;
import android.view.View.OnClickListener;
import android.view.ViewGroup;
import android.widget.BaseAdapter;
import android.widget.CheckBox;
import android.widget.ImageView;
import android.widget.TextView;
import com.example.administrator.treeview.R;
import java.util.ArrayList;
import java.util.List;
public class TreeAdapter extends BaseAdapter {
private Context con;
private LayoutInflater lif;
public List<Node  all = new ArrayList<Node ();//展示
private List<Node  cache = new ArrayList<Node ();//缓存,记录点状态
private TreeAdapter tree = this;
boolean hasCheckBox;
private int expandIcon = -1;//展开图标
private int collapseIcon = -1;//收缩图标
ViewItem vi = null;
// //存储checkbox选中的集合
// private List< 
/**
* 构造方法
*/
public TreeAdapter(Context context, List<Node  rootNodes){
this.con = context;
this.lif = (LayoutInflater)con.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
for(int i=0;i<rootNodes.size();i++){
addNode(rootNodes.get(i));
}
}
/**
* 把一个节点上的所有的内容都挂上去
* @param node
*/
public void addNode(Node node){
all.add(node);
cache.add(node);
if(node.isLeaf())return;
for(int i = 0;i<node.getChildrens().size();i++){
addNode(node.getChildrens().get(i));
}
}
/**
* 设置展开收缩图标
* @param expandIcon
* @param collapseIcon
*/
public void setCollapseAndExpandIcon(int expandIcon,int collapseIcon){
this.collapseIcon = collapseIcon;
this.expandIcon = expandIcon;
}
/**
* 一次性对某节点的所有节点进行选中or取消操作
*/
public void checkNode(Node n,boolean isChecked){
n.setChecked(isChecked);
checkChildren(n,isChecked);
// 有一个子节点选中，则父节点选中
if (n.getParent()!=null)
checkParent(n,isChecked);
// 有一个子节点未选中，则父节点未选中
// unCheckNode(n, isChecked);
}
/**
* 对父节点操作时，同步操作子节点
*/
public void checkChildren(Node n,boolean isChecked){
for(int i =0 ;i<n.getChildrens().size();i++){
n.getChildrens().get(i).setChecked(isChecked);
checkChildren(n.getChildrens().get(i),isChecked);
}
}
/**
* 有一个子节点选中，则父节点选中
*/
public void checkParent(Node n,boolean isChecked){
// 有一个子节点选中，则父节点选中
if (n.getParent()!=null&&isChecked){
n.getParent().setChecked(isChecked);
checkParent(n.getParent(),isChecked);
}
// 全部子节点取消选中，则父节点取消选中
if (n.getParent()!=null &&!isChecked){
for (int i = 0; i < n.getParent().getChildrens().size(); i++) {
if (n.getParent().getChildrens().get(i).isChecked()) {
checkParent(n.getParent(),!isChecked);
return ;
}
}
n.getParent().setChecked(isChecked);
checkParent(n.getParent(),isChecked);
}
}
/**
* 有一个子节点未选中，则父节点未选中
*/
public void unCheckNode(Node n, boolean isChecked){
boolean flag = false;
n.setChecked(isChecked);
if(n.getParent() != null ){
Log.d("parentSize", n.getParent().getChildrens().get(0).isChecked() + "");
for (int i = 0; i < n.getParent().getChildrens().size(); i++) {
if((n.getParent().getChildrens().get(i)) != n && (n.getParent().getChildrens().get(i).isChecked() != true)){
flag = true;
break;
}
}
if(!flag) {
unCheckNode(n.getParent(), isChecked);
}
}
}
/**
* 获取所有选中节点
* @return
*
*/
public List<Node  getSelectedNode(){
Log.d("getSelectedNode", "我被执行了！");
List<Node  checks =new ArrayList<Node () ;
for(int i = 0;i<cache.size();i++){
Node n =(Node)cache.get(i);
if(n.isChecked())
checks.add(n);
}
return checks;
}
public void setSelectedNode(List<String  selectedNode){
for (int i=0;i<cache.size();i++) {
if(selectedNode.contains(cache.get(i).getCurId())) {
cache.get(i).setChecked(true);
cache.get(i).getParent().setChecked(true);
}
}
}
/**
* 设置是否有复选框
* @param hasCheckBox
*
*/
public void setCheckBox(boolean hasCheckBox){
this.hasCheckBox = hasCheckBox;
}
/**
* 控制展开缩放某节点
* @param location
*
*/
public void ExpandOrCollapse(int location){
Node n = all.get(location);//获得当前视图需要处理的节点 
if(n!=null)//排除传入参数错误异常
{
if(!n.isLeaf()){
n.setExplaned(!n.isExplaned());// 由于该方法是用来控制展开和收缩的，所以取反即可
filterNode();//遍历一下，将所有上级节点展开的节点重新挂上去
this.notifyDataSetChanged();//刷新视图
}
}
}
/**
* 设置展开等级
* @param level
*
*/
public void setExpandLevel(int level){
all.clear();
for(int i = 0;i<cache.size();i++){
Node n = cache.get(i);
if(n.getLevel()<=level){
if(n.getLevel()<level)
n.setExplaned(true);
else
n.setExplaned(false);
all.add(n);
}
}
}
/* 清理all,从缓存中将所有父节点不为收缩状态的都挂上去*/
public void filterNode(){
all.clear();
for(int i = 0;i<cache.size();i++){
Node n = cache.get(i);
if(!n.isParentCollapsed()||n.isRoot())//凡是父节点不收缩或者不是根节点的都挂上去
all.add(n);
}
}
@Override
public int getCount() {
// TODO Auto-generated method stub
return all.size();
}
@Override
public Object getItem(int location) {
// TODO Auto-generated method stub
return all.get(location);
}
@Override
public long getItemId(int location) {
// TODO Auto-generated method stub
return location;
}
@Override
public View getView(final int location, View view, ViewGroup viewgroup) {
final Node n = all.get(location);
//ViewItem vi = null;
if(view == null){
view = lif.inflate(R.layout.member_item, null);
vi = new ViewItem();
vi.cb = (CheckBox)view.findViewById(R.id.checkBox);
vi.flagIcon = (ImageView)view.findViewById(R.id.disclosureImg);
vi.tv = (TextView)view.findViewById(R.id.contentText);
vi.cb.setOnClickListener(new OnClickListener() {
private Node mCheckBoxN;
@Override
public void onClick(View v) {
mCheckBoxN = (Node) v.getTag();
checkNode(mCheckBoxN, ((CheckBox) v).isChecked());
//unCheckNode(n, ((CheckBox) v).isChecked());
tree.notifyDataSetChanged(); //只有点击部门后刷新页面，不然刷新频繁导致卡顿
}
});
view.setTag(vi);
}
else{
vi = (ViewItem)view.getTag();
}
if(n!=null){
if(vi==null||vi.cb==null)
System.out.println();
vi.cb.setTag(n);
vi.cb.setChecked(n.isChecked());
//叶节点不显示展开收缩图标
if(n.isExplaned()){
if(expandIcon!=-1){
vi.flagIcon.setImageResource(expandIcon);
}
}
else{
if(collapseIcon!=-1){
vi.flagIcon.setImageResource(collapseIcon);
}
}
//显示文本
vi.tv.setText(n.getValue());
// 控制缩进
vi.flagIcon.setPadding(100*n.getLevel(), 3,3, 3);
if(n.isLeaf()){
vi.flagIcon.setVisibility(View.INVISIBLE);
}
else{
vi.flagIcon.setVisibility(View.VISIBLE);
}
//设置是否显示复选框
if(n.hasCheckBox()){
vi.cb.setVisibility(View.VISIBLE);
}
else{
vi.cb.setVisibility(View.GONE);
}
}
return view;
}
public class ViewItem{
private CheckBox cb;
private ImageView flagIcon;
private TextView tv;
}
}

package com.example.administrator.treeview.treeView;
import android.content.Context;
import android.util.Log;
import android.view.View;
import android.view.ViewGroup;
import android.widget.AdapterView;
import android.widget.LinearLayout;
import android.widget.ListView;
import android.widget.RelativeLayout;
import com.example.administrator.treeview.R;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
public class TreeListView extends ListView {
ListView treelist = null;
TreeAdapter ta = null;
public List<Node  mNodeList;
private List<Node  checkList;
public TreeListView(final Context context, List<Node  res) {
super(context);
treelist = this;
treelist.setFocusable(false);
treelist.setBackgroundColor(0xffffff);
treelist.setFadingEdgeLength(0);
treelist.setLayoutParams(new ViewGroup.LayoutParams(LinearLayout.LayoutParams.MATCH_PARENT, RelativeLayout.LayoutParams.MATCH_PARENT));
treelist.setOnItemClickListener(new OnItemClickListener() {
@Override
public void onItemClick(AdapterView<?  parent, View view,
int position, long id) {
((TreeAdapter) parent.getAdapter()).ExpandOrCollapse(position);
}
});
initNode(context, initNodRoot(res), true, -1, -1, 0);
}
// 使用 onMeasure 方法，来解决尺寸高度的问题，以及事件冲突的问题；
protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec) {
heightMeasureSpec = MeasureSpec.makeMeasureSpec(
Integer.MAX_VALUE  2,
MeasureSpec.AT_MOST
);
super.onMeasure(widthMeasureSpec, heightMeasureSpec);
}
// /**
// *
// * @param context
// * 响应监听的上下文
// * @param root
// * 已经挂好树的根节点
// * @param hasCheckBox
// * 是否整个树有复选框
// * @param tree_ex_id
// * 展开iconid -1会使用默认的
// * @param tree_ec_id
// * 收缩iconid -1会使用默认的
// * @param expandLevel
// * 初始展开等级
// *
// */
public List<Node  initNodRoot(List<Node  res) {
ArrayList<Node  list = new ArrayList<Node ();
ArrayList<Node  roots = new ArrayList<Node ();
Map<String, Node  nodemap = new LinkedHashMap<String, Node ();
for (int i = 0; i < res.size(); i++) {
Node nr = res.get(i);
Node n = new Node( nr.getParentId(), nr.getCurId(), nr.getValue());
nodemap.put(n.getCurId(), n);// 生成map树
}
Set<String  set = nodemap.keySet();
Collection<Node  collections = nodemap.values();
Iterator<Node  iterator = collections.iterator();
while (iterator.hasNext()) {// 添加所有根节点到root中
Node n = iterator.next();
if (!set.contains(n.getParentId()))
roots.add(n);
list.add(n);
}
for (int i = 0; i < list.size(); i++) {
Node n = list.get(i);
for (int j = i + 1; j < list.size(); j++) {
Node m = list.get(j);
if (m.getParentId() .equals( n.getCurId())) {
n.addNode(m);
m.setParent(n);
m.setParents(n);
} else if (m.getCurId() .equals( n.getParentId())) {
m.addNode(n);
n.setParent(m);
m.setParents(m);
}
}
}
return roots;
}
public void initNode(Context context, List<Node  root, boolean hasCheckBox,
int tree_ex_id, int tree_ec_id, int expandLevel) {
ta = new TreeAdapter(context, root);
//获取
mNodeList = ta.all;
// 设置整个树是否显示复选框
ta.setCheckBox(true);
// 设置展开和折叠时图标
int tree_ex_id_ = (tree_ex_id == -1) ? R.drawable.down_icon : tree_ex_id;
int tree_ec_id_ = (tree_ec_id == -1) ? R.drawable.right_icon : tree_ec_id;
ta.setCollapseAndExpandIcon(tree_ex_id_, tree_ec_id_);
// 设置默认展开级别
ta.setExpandLevel(expandLevel);
this.setAdapter(ta);
}
/* 返回当前所有选中节点的List数组 */
public List<Node  get() {
Log.d("get", ta.getSelectedNode().size() + "");
return ta.getSelectedNode();
}
public void setSelect(List<String  allSelect){
ta.setSelectedNode(allSelect);
}}