// Near Infinity - An Infinity Engine Browser and Editor // Copyright (C) 2001 - 2005 Jon Olav Hauglid // See LICENSE.txt for license information package org.infinity.gui.treetable; import javax.swing.JTree; import javax.swing.SwingUtilities; import javax.swing.event.TreeExpansionEvent; import javax.swing.event.TreeExpansionListener; import javax.swing.event.TreeModelEvent; import javax.swing.event.TreeModelListener; import javax.swing.table.AbstractTableModel; import javax.swing.tree.TreePath; final class TreeTableModelAdapter extends AbstractTableModel { private final JTree tree; private final TreeTableModel treeTableModel; 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. @Override public void treeExpanded(TreeExpansionEvent event) { fireTableDataChanged(); } @Override 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() { @Override public void treeNodesChanged(TreeModelEvent e) { delayedFireTableDataChanged(); } @Override public void treeNodesInserted(TreeModelEvent e) { delayedFireTableDataChanged(); } @Override public void treeNodesRemoved(TreeModelEvent e) { delayedFireTableDataChanged(); } @Override public void treeStructureChanged(TreeModelEvent e) { delayedFireTableDataChanged(); } }); } // --------------------- Begin Interface TableModel --------------------- @Override public int getRowCount() { return tree.getRowCount(); } // Wrappers, implementing TableModel interface. @Override public int getColumnCount() { return treeTableModel.getColumnCount(); } @Override public Object getValueAt(int row, int column) { return treeTableModel.getValueAt(nodeForRow(row), column); } // --------------------- End Interface TableModel --------------------- @Override public Class<? extends Object> getColumnClass(int column) { return treeTableModel.getColumnClass(column); } @Override public String getColumnName(int column) { return treeTableModel.getColumnName(column); } @Override public boolean isCellEditable(int row, int column) { return treeTableModel.isCellEditable(nodeForRow(row), column); } @Override 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. */ private void delayedFireTableDataChanged() { SwingUtilities.invokeLater(new Runnable() { @Override public void run() { fireTableDataChanged(); } }); } private Object nodeForRow(int row) { TreePath treePath = tree.getPathForRow(row); return treePath.getLastPathComponent(); } }