// TreeTableModelAdapter.java
//
// Created: December 2005
//
// This file is a component of the BioMoby project.
// Copyright Martin Senger (martin.senger@gmail.com).
//

package org.biomoby.service.dashboard.treetable;

import javax.swing.JTree;
import javax.swing.SwingUtilities;
import javax.swing.table.AbstractTableModel;
import javax.swing.tree.TreePath;
import javax.swing.event.TreeExpansionEvent;
import javax.swing.event.TreeExpansionListener;
import javax.swing.event.TreeModelEvent;
import javax.swing.event.TreeModelListener;

/**
 * This is a wrapper class that takes a TreeTableModel and implements
 * the table model interface. The implementation is trivial, with all
 * of the event dispatching support provided by the superclass: the
 * AbstractTableModel. <p>
 *
 * Inspired by <a
 * href="http://java.sun.com/products/jfc/tsc/articles/treetable1/index.html">http://java.sun.com/products/jfc/tsc/articles/treetable1/index.html</a>. <p>
 *
 * @author Philip Milne 
 * @author Scott Violet
 * @author <A HREF="mailto:martin.senger@gmail.com">Martin Senger</A>
 * @version $Id: TreeTableModelAdapter.java,v 1.4 2006/02/04 09:24:42 senger Exp $
 */

public class TreeTableModelAdapter
    extends AbstractTableModel {

    JTree tree;
    TreeTableModel treeTableModel;

    public TreeTableModelAdapter (TreeTableModel treeTableModel, JTree tree) {
        this.tree = tree;
        this.treeTableModel = treeTableModel;

	tree.addTreeExpansionListener (new TreeExpansionListener() {
	    // Don't use fireTableRowsInserted() here; the selection model
	    // would get updated twice. 
	    public void treeExpanded (TreeExpansionEvent event) {  
	      fireTableDataChanged(); 
	    }
            public void treeCollapsed (TreeExpansionEvent event) {  
	      fireTableDataChanged(); 
	    }
	});

	// Install a TreeModelListener that can update the table when
	// tree changes. We use delayedFireTableDataChanged as we can
	// not be guaranteed the tree will have finished processing
	// the event before us.
	treeTableModel.addTreeModelListener (new TreeModelListener() {
	    public void treeNodesChanged (TreeModelEvent e) {
		delayedFireTableDataChanged();
	    }

	    public void treeNodesInserted (TreeModelEvent e) {
		delayedFireTableDataChanged();
	    }

	    public void treeNodesRemoved (TreeModelEvent e) {
		delayedFireTableDataChanged();
	    }

	    public void treeStructureChanged (TreeModelEvent e) {
		delayedFireTableDataChanged();
	    }
	});
    }

    // Wrappers, implementing TableModel interface. 

    public int getColumnCount() {
	return treeTableModel.getColumnCount();
    }

    public String getColumnName (int column) {
	return treeTableModel.getColumnName(column);
    }

    public Class getColumnClass (int column) {
	return treeTableModel.getColumnClass(column);
    }

    public int getRowCount() {
	return tree.getRowCount();
    }

    public Object nodeForRow (int row) {
	TreePath treePath = tree.getPathForRow(row);
	if (treePath == null) return null;
	return treePath.getLastPathComponent();         
    }

    public Object getValueAt (int row, int column) {
	return treeTableModel.getValueAt (nodeForRow(row), column);
    }

    public boolean isCellEditable (int row, int column) {
         return treeTableModel.isCellEditable (nodeForRow(row), column); 
    }

    public void setValueAt (Object value, int row, int column) {
	treeTableModel.setValueAt (value, nodeForRow(row), column);
    }

    /**
     * Invokes fireTableDataChanged after all the pending events have been
     * processed. SwingUtilities.invokeLater is used to handle this.
     */
    protected void delayedFireTableDataChanged() {
 	SwingUtilities.invokeLater (new Runnable() {
 	    public void run() {
		fireTableDataChanged();
 	    }
 	});
    }
}