Swing自定义组件: JCheckTree

一个Renderer 一个Listener 应该可以搞定.. 自定义组件..关键就在交互事件上..


4个类.分别为:CheckTreeDemo,CheckTree,Node,CheckRenderer,CheckListener

从类名可以看出来. CheckTreeDemo是JFrame类 CheckTree是组件,Node是数据模型,CheckRenderer是CheckTree得渲染,CheckListener是CheckTree的事件


package tree;

import java.awt.BorderLayout;
import javax.swing.JFrame;
import javax.swing.JPanel;
import javax.swing.SwingUtilities;

/**
*
* @author chensiyu
* @createDate 2011/8/24
*/
public class CheckTreeDemo extends JPanel {

private CheckTree checkTree;

public CheckTreeDemo() {
init();
}

private void init(){
setLayout(new BorderLayout());

Node rootNode = new Node("Root");
Node []nodes = new Node[10];
for(int i = 0;i <nodes.length;i++){
Node node = new Node("Node-"+i);
rootNode.add(node);
}
checkTree = new CheckTree(rootNode);
add(checkTree,BorderLayout.CENTER);

}

public static void main(String[] args) {
final JFrame frame = new JFrame();
frame.setTitle("CheckTree");
frame.setLayout(new BorderLayout());
frame.setSize(200, 400);
frame.getContentPane().add(new CheckTreeDemo());
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
SwingUtilities.invokeLater(new Runnable() {

@Override
public void run() {
frame.setVisible(true);
}
});
}
}


package tree;

import javax.swing.JTree;
import javax.swing.tree.TreeNode;

/**
*
* @author chensiyu
* @createDate 2011/8/24
*/
public class CheckTree extends JTree {

private CheckRenderer checkRenderer;
private CheckListener checkListener;

public CheckTree() {
init();
}

public CheckTree(TreeNode node) {
super(node);
init();
}

private void init() {
checkRenderer = new CheckRenderer();
checkListener = new CheckListener();
setCellRenderer(checkRenderer);
addMouseListener(checkListener);
}
}

package tree;

import javax.swing.tree.DefaultMutableTreeNode;

/**
*
* @author chensiyu
* @createDate 2011/8/24
*/
public class Node extends DefaultMutableTreeNode{

private String name = "";
private boolean selected = false;

public Node(String name){
super(name);
setName(name);
}
public String getName() {
return name;
}

public void setName(String name) {
this.name = name;
}

public boolean isSelected() {
return selected;
}

public void setSelected(boolean selected) {
this.selected = selected;
}
}

package tree;

/**
*
* @author chensiyu
* @createDate 2011/8/24
*/
import java.awt.*;
import javax.swing.*;
import javax.swing.tree.*;

class CheckRenderer extends JPanel implements TreeCellRenderer {

protected static JCheckBox check;
protected JLabel label;
private static Rectangle checkBounds;

public CheckRenderer() {
setLayout(null);
add(check = new JCheckBox());
add(label = new JLabel(""));
check.setBackground(UIManager.getColor("Tree.textBackground"));
label.setForeground(UIManager.getColor("Tree.textForeground"));
}

@Override
public Component getTreeCellRendererComponent(JTree tree, Object value,
boolean isSelected, boolean expanded, boolean leaf, int row,
boolean hasFocus) {
String stringValue = tree.convertValueToText(value, isSelected,
expanded, leaf, row, hasFocus);
setEnabled(tree.isEnabled());
label.setFont(tree.getFont());
label.setText(stringValue);
setBackground(tree.getBackground());
setForeground(tree.getForeground());

if(value instanceof Node){
Node node = (Node)value;
check.setSelected(node.isSelected());
}
return this;
}

@Override
public Dimension getPreferredSize() {
Dimension d_check = check.getPreferredSize();
Dimension d_label = label.getPreferredSize();
return new Dimension(d_check.width + d_label.width,
(d_check.height < d_label.height ? d_label.height
: d_check.height));
}

@Override
public void doLayout() {
Dimension d_check = check.getPreferredSize();
Dimension d_label = label.getPreferredSize();
int y_check = 0;
int y_label = 0;
if (d_check.height < d_label.height) {
y_check = (d_label.height - d_check.height) / 2;
} else {
y_label = (d_check.height - d_label.height) / 2;
}
check.setLocation(0, y_check);
check.setBounds(0, y_check, d_check.width, d_check.height);
label.setLocation(d_check.width, y_label);
label.setBounds(d_check.width, y_label, d_label.width, d_label.height);
if (checkBounds == null){
checkBounds = check.getBounds();
}

}

public static Rectangle getCheckBoxRectangle() {
if( null == checkBounds )
return new Rectangle(0,0,0,0);
return (Rectangle) checkBounds.clone();
}
}

package tree;

import java.awt.Point;
import java.awt.Rectangle;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import javax.swing.JTree;
import javax.swing.tree.TreePath;

/**
*
* @author chensiyu
* @createDate 2011/8/24
*/
class CheckListener extends MouseAdapter {

public CheckListener() {
}

@Override
public void mouseClicked(MouseEvent e) {
JTree tree = (JTree) e.getSource();
Point p = e.getPoint();
int x = e.getX();
int y = e.getY();
int row = tree.getRowForLocation(x, y);
TreePath path = tree.getPathForRow(row);

if (null == path) {
return;
}

Object component = path.getLastPathComponent();
if (null == component) {
return;
}

if (component instanceof Node) {
Rectangle chRect = CheckRenderer.getCheckBoxRectangle();
Rectangle rowRect = tree.getPathBounds(path);
chRect.setLocation(chRect.x + rowRect.x, chRect.y + rowRect.y);
if (e.getClickCount() == 1 && chRect.contains(p)) {
Node node = (Node)component;
node.setSelected(!node.isSelected());
tree.repaint();
}
}
}
}