source: src/main/java/agents/anac/y2015/xianfa/Tree.java

package agents.anac.y2015.xianfa;

import java.util.ArrayList;

public class Tree {
        private Node root;
        //private int depth=0;
        private ArrayList<ArrayList<Node>> nodes = new ArrayList<ArrayList<Node>>();
        boolean setRoot = false;
        
        public Tree() {
                
        }
        
        public void addNewDepth() {
                if (!setRoot) {
                        setRoot = true;
                        root = new Node();
                        ArrayList<Node> lvl1 = new ArrayList<Node>();
                        lvl1.add(root);
                        nodes.add(lvl1);
                } else {
                        ArrayList<Node> newLvl = new ArrayList<Node>();
                        nodes.add(newLvl);
                }
        }
        
        public void addNodeInDepth(Node node, int depth) {
                nodes.get(depth).add(node);
        }
        
        public void setRoot(Node node) {
                root = new Node();
                ArrayList<Node> lvl1 = new ArrayList<Node>();
                lvl1.add(root);
                nodes.add(lvl1);
        }
        
        public Node getRoot() {
                return root;
        }
        
        public void addNode(Node node) {
                
        }
        
        public int getSizeOfLevel(int depth) {
                return nodes.get(depth).size();
        }
        
        public Node getMemberInLevel(int depth, int member) {
                return nodes.get(depth).get(member);
        }
        
        public int getLevels() {
                return nodes.size();
        }

}