用Java2D画出树的结构_v0.1.0

先看效果图：

 

定义树的数据结构

/** * 2010-11-8 * John */package tree;import java.util.ArrayList;import java.util.List;/** * 树的结构 * @author John * */public class Node {private String name;//该结点名字private int layer = 0;//该结点层级private List<Node> childs = null;//保存该结点的孩子public Node(String name){this.name = name;}/** * 增加一个孩子 * @param n 要作为孩子增加的结点 */public void add(Node n){if(childs == null)childs = new ArrayList<Node>();n.setLayer(layer+1);setChildLayout(n);childs.add(n);}/** * 递归设置孩子的层级 * @param n */private void setChildLayout(Node n){if(n.hasChild()){List<Node> c = n.getChilds();for(Node node : c){node.setLayer(node.getLayer()+1);setChildLayout(node);}}}/** * 获取结点名 * @return 结点名 */public String getName() {return name;}/** * 设置结点名 * @param name 结点名 */public void setName(String name) {this.name = name;}/** * 获取该结点的层级 * @return 该结点的层级 */public int getLayer() {return layer;}/** * 设置该结点的层级 * @param layer 该结点的层级 */public void setLayer(int layer) {this.layer = layer;}/** * 获取该结点的孩子 * @return 所有孩子结点 */public List<Node> getChilds() {return childs;}/** * 检查是否存在孩子 * @return 是则返回true，否则返回false */public boolean hasChild(){return childs == null ? false : true;}/** * 递归打印所有的结点（包括子结点） * @param n 要打印的根结点 */public void printAllNode(Node n){System.out.println(n);if(n.hasChild()){List<Node> c = n.getChilds();for(Node node : c){printAllNode(node);}}}public String getAllNodeName(Node n){String s = n.toString()+"/n";if(n.hasChild()){List<Node> c = n.getChilds();for(Node node : c){s+=getAllNodeName(node)+"/n";}}return s;}public String toString(){return layer+"层/t: "+name;}}

 

Node的测试类：

/** * 2010-11-9 * John */package tree.demo;import tree.Node;/** * @author John * */public class TestPrintNode {public static void main(String[] args){Node n = new Node("root");n.add(new Node("a1"));n.add(new Node("a2"));Node n2 = new Node("a3");n2.add(new Node("b1"));n2.add(new Node("b2"));n2.add(new Node("b3"));Node n3 = new Node("b4");n2.add(n3);n3.add(new Node("c1"));n3.add(new Node("c2"));n.add(n2);n.printAllNode(n);//输出树}} 

 

画树的Panel

 /** * 2010-11-9 * John */package tree;import java.awt.Color;import java.awt.Font;import java.awt.FontMetrics;import java.awt.Graphics;import java.util.List;import javax.swing.JPanel;/** * TODO 同一层结点过多有BUG，应该对每一层的所有结点都进行个数统计，之后才绘制。 * @author John * */public class TreePanel extends JPanel {private Node tree;//保存整棵树private int gridWidth = 80;//每个结点的宽度private int gridHeight = 20;//每个结点的高度private int vGap = 50;//每2个结点的垂直距离private int hGap = 30;//每2个结点的水平距离private int startY = 10;//根结点的Y，默认距离顶部10像素private int startX = 0;//根结点的X，默认水平居中对齐private int childAlign;//孩子对齐方式public static int CHILD_ALIGN_ABSOLUTE = 0;//相对Panel居中public static int CHILD_ALIGN_RELATIVE = 1;//相对父结点居中private Font font = new Font("微软雅黑",Font.BOLD,14);//描述结点的字体private Color gridColor = Color.BLACK;//结点背景颜色private Color linkLineColor = Color.BLACK;//结点连线颜色private Color stringColor = Color.WHITE;//结点描述文字的颜色/** * 默认构造 */public TreePanel(){this(null,CHILD_ALIGN_ABSOLUTE);}/** * 根据传入的Node绘制树，以绝对居中的方式绘制 * @param n 要绘制的树 */public TreePanel(Node n){this(n,CHILD_ALIGN_ABSOLUTE);}/** * 设置要绘制时候的对齐策略 * @param childAlign 对齐策略 * @see tree.TreePanel#CHILD_ALIGN_RELATIVE * @see tree.TreePanel#CHILD_ALIGN_ABSOLUTE */public TreePanel(int childAlign){this(null,childAlign);}/** * 根据孩子对齐策略childAlign绘制的树的根结点n * @param n 要绘制的树的根结点 * @param childAlign 对齐策略 */public TreePanel(Node n, int childAlign){super();setTree(n);this.childAlign = childAlign;}/** * 设置用于绘制的树 * @param n 用于绘制的树的 */public void setTree(Node n) {tree = n;}//重写而已，调用自己的绘制方法public void paintComponent(Graphics g){startX = (getWidth()-gridWidth)/2;super.paintComponent(g);g.setFont(font);drawAllNode(tree, startX, g);}/** * 递归绘制整棵树 * @param n 被绘制的Node * @param xPos 根节点的绘制X位置 * @param g 绘图上下文环境 */public void drawAllNode(Node n, int x, Graphics g){int y = n.getLayer()*(vGap+gridHeight)+startY;int fontY = y + gridHeight - 5;//5为测试得出的值，你可以通过FM计算更精确的，但会影响速度g.setColor(gridColor);g.fillRect(x, y, gridWidth, gridHeight);//画结点的格子g.setColor(stringColor);g.drawString(n.toString(), x, fontY);//画结点的名字if(n.hasChild()){List<Node> c = n.getChilds();int size = n.getChilds().size();int tempPosx = childAlign == CHILD_ALIGN_RELATIVE ? x+gridWidth/2 - (size*(gridWidth+hGap)-hGap)/2 : (getWidth() - size*(gridWidth+hGap)+hGap)/2; int i = 0;for(Node node : c){int newX = tempPosx+(gridWidth+hGap)*i;//孩子结点起始Xg.setColor(linkLineColor);g.drawLine(x+gridWidth/2, y+gridHeight, newX+gridWidth/2, y+gridHeight+vGap);//画连接结点的线drawAllNode(node, newX, g);i++;}}}public Color getGridColor() {return gridColor;}/** * 设置结点背景颜色 * @param gridColor 结点背景颜色 */public void setGridColor(Color gridColor) {this.gridColor = gridColor;}public Color getLinkLineColor() {return linkLineColor;}/** * 设置结点连接线的颜色 * @param gridLinkLine 结点连接线的颜色 */public void setLinkLineColor(Color gridLinkLine) {this.linkLineColor = gridLinkLine;}public Color getStringColor() {return stringColor;}/** * 设置结点描述的颜色 * @param stringColor 结点描述的颜色 */public void setStringColor(Color stringColor) {this.stringColor = stringColor;}public int getStartY() {return startY;}/** * 设置根结点的Y位置 * @param startY 根结点的Y位置 */public void setStartY(int startY) {this.startY = startY;}public int getStartX() {return startX;}/** * 设置根结点的X位置 * @param startX 根结点的X位置 */public void setStartX(int startX) {this.startX = startX;}}

 

测试TreePanel

/** * 2010-11-9 * John */package tree.demo;import java.awt.BorderLayout;import java.awt.Color;import java.awt.GridLayout;import java.awt.Panel;import javax.swing.JFrame;import javax.swing.JPanel;import javax.swing.JScrollPane;import javax.swing.JTextArea;import tree.Node;import tree.TreePanel;/** * @author John * */public class TestDrawTree extends JFrame{public TestDrawTree(){super("Test Draw Tree");initComponents();}public static void main(String[] args){TestDrawTree frame = new TestDrawTree();frame.setSize(800, 600);frame.setVisible(true);frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);} public void initComponents(){/* * 初始化树的数据 */Node n = new Node("root");Node a1 = new Node("a1");Node a2 = new Node("a2");n.add(a1);n.add(a2);Node a3 = new Node("a3");Node b1 = new Node("b1");Node d1 = new Node("d1");Node d2 = new Node("d2");b1.add(d1);b1.add(d2);a3.add(b1);a3.add(new Node("b2"));a3.add(new Node("b3"));Node n3 = new Node("b4");a3.add(n3);n3.add(new Node("c1"));n3.add(new Node("c2"));n.add(a3);//n.printAllNode(n);//输出树/* * 创建一个用于绘制树的面板并将树传入,使用相对对齐方式 */TreePanel panel1 = new TreePanel(TreePanel.CHILD_ALIGN_RELATIVE);panel1.setTree(n);/* * 创建一个用于绘制树的面板并将树传入,使用绝对对齐方式 */TreePanel panel2 = new TreePanel(TreePanel.CHILD_ALIGN_ABSOLUTE);panel2.setTree(n);panel2.setBackground(Color.BLACK);panel2.setGridColor(Color.WHITE);panel2.setLinkLineColor(Color.WHITE);panel2.setStringColor(Color.BLACK);JPanel contentPane = new JPanel();contentPane.setLayout(new GridLayout(2,1));contentPane.add(panel1);contentPane.add(panel2);add(contentPane,BorderLayout.CENTER);}}