components/visual-panels/core/src/java/util/com/oracle/solaris/vp/util/swing/tree/AbstractTreeTableModel.java
changeset 827 0944d8c0158b
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/components/visual-panels/core/src/java/util/com/oracle/solaris/vp/util/swing/tree/AbstractTreeTableModel.java	Thu May 24 04:16:47 2012 -0400
@@ -0,0 +1,209 @@
+/*
+ * CDDL HEADER START
+ *
+ * The contents of this file are subject to the terms of the
+ * Common Development and Distribution License (the "License").
+ * You may not use this file except in compliance with the License.
+ *
+ * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
+ * or http://www.opensolaris.org/os/licensing.
+ * See the License for the specific language governing permissions
+ * and limitations under the License.
+ *
+ * When distributing Covered Code, include this CDDL HEADER in each
+ * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
+ * If applicable, add the following below this CDDL HEADER, with the
+ * fields enclosed by brackets "[]" replaced with your own identifying
+ * information: Portions Copyright [yyyy] [name of copyright owner]
+ *
+ * CDDL HEADER END
+ */
+
+/*
+ * Copyright (c) 2009, 2012, Oracle and/or its affiliates. All rights reserved.
+ */
+
+package com.oracle.solaris.vp.util.swing.tree;
+
+import java.util.Comparator;
+import javax.swing.event.*;
+import javax.swing.table.*;
+import javax.swing.tree.*;
+import com.oracle.solaris.vp.util.common.propinfo.PropInfo;
+import com.oracle.solaris.vp.util.swing.propinfo.SwingPropInfo;
+
+@SuppressWarnings({"serial"})
+public abstract class AbstractTreeTableModel extends DefaultTreeModel
+    implements TreeTableModel, TreeModelListener {
+
+    //
+    // Instance data
+    //
+
+    private PropInfo<?, ?>[] props;
+
+    //
+    // Constructors
+    //
+
+    public AbstractTreeTableModel(TreeNode root, PropInfo<?, ?>[] props) {
+	super(root);
+	this.props = props;
+    }
+
+    public AbstractTreeTableModel(TreeNode root, boolean asksAllowsChildren,
+	PropInfo<?, ?>[] props) {
+
+	super(root, asksAllowsChildren);
+	this.props = props;
+    }
+
+    //
+    // TreeModel methods
+    //
+
+    @Override
+    public TreeNode getRoot() {
+	return (TreeNode)super.getRoot();
+    }
+
+    //
+    // TreeTableModel methods
+    //
+
+    @Override
+    public Class<?> getColumnClass(int column) {
+	TreeNode root = getRoot();
+	try {
+	    return getValueAt(root, column).getClass();
+	}
+
+	// Most likely NullPointerException
+	catch (Throwable e) {
+	    return Object.class;
+	}
+    }
+
+    @Override
+    public int getColumnCount() {
+	return props == null ? 0 : props.length;
+    }
+
+    @Override
+    public String getColumnName(int column) {
+	return props[column].getLabel();
+    }
+
+    @Override
+    @SuppressWarnings({"unchecked"})
+    public Object getValueAt(TreeNode node, int column) {
+	Object o = getUserObject(node);
+	PropInfo prop = props[column];
+	return prop.getValue(o);
+    }
+
+    @Override
+    @SuppressWarnings({"unchecked"})
+    public boolean isCellEditable(TreeNode node, int column) {
+	Object o = getUserObject(node);
+	PropInfo prop = props[column];
+	return prop.isEditable(o);
+    }
+
+    @Override
+    public boolean isTreeColumn(int column) {
+	return column == 0;
+    }
+
+    @Override
+    @SuppressWarnings({"unchecked"})
+    public void setValueAt(Object value, TreeNode node, int column) {
+	Object o = getUserObject(node);
+	PropInfo prop = props[column];
+	prop.setValue(o, value);
+    }
+
+    //
+    // AbstractTreeTableModel methods
+    //
+
+    public Comparator<?> getComparator(int column) {
+	return props[column].getComparator();
+    }
+
+    public PropInfo<?, ?>[] getPropInfos() {
+	return props;
+    }
+
+    public TableCellEditor getTableCellEditor(int column) {
+	TableCellEditor editor = null;
+
+	PropInfo prop = props[column];
+	if (prop instanceof SwingPropInfo) {
+	    editor = ((SwingPropInfo)prop).getTableCellEditor();
+	}
+
+	return editor;
+    }
+
+    public TableCellRenderer getTableCellRenderer(int column) {
+	TableCellRenderer renderer = null;
+
+	PropInfo prop = props[column];
+	if (prop instanceof SwingPropInfo) {
+	    renderer = ((SwingPropInfo)prop).getTableCellRenderer();
+	}
+
+	return renderer;
+    }
+
+    protected abstract Object getUserObject(TreeNode node);
+
+    //
+    // TreeModelListener methods
+    //
+
+    /**
+     * Propagates the given event to registered listeners.
+     */
+    @Override
+    public void treeNodesChanged(TreeModelEvent e) {
+	TreeModelListener[] listeners = getTreeModelListeners();
+	for (int i = 0; i < listeners.length; i++) {
+	    listeners[i].treeNodesChanged(e);
+	}
+    }
+
+    /**
+     * Propagates the given event to registered listeners.
+     */
+    @Override
+    public void treeNodesInserted(TreeModelEvent e) {
+	TreeModelListener[] listeners = getTreeModelListeners();
+	for (int i = 0; i < listeners.length; i++) {
+	    listeners[i].treeNodesInserted(e);
+	}
+    }
+
+    /**
+     * Propagates the given event to registered listeners.
+     */
+    @Override
+    public void treeNodesRemoved(TreeModelEvent e) {
+	TreeModelListener[] listeners = getTreeModelListeners();
+	for (int i = 0; i < listeners.length; i++) {
+	    listeners[i].treeNodesRemoved(e);
+	}
+    }
+
+    /**
+     * Propagates the given event to registered listeners.
+     */
+    @Override
+    public void treeStructureChanged(TreeModelEvent e) {
+	TreeModelListener[] listeners = getTreeModelListeners();
+	for (int i = 0; i < listeners.length; i++) {
+	    listeners[i].treeStructureChanged(e);
+	}
+    }
+}