001    /* JAPI - (Yet anothr (hopefully) useful) Java API
002     *
003     * Copyright (C) 2004-2006 Christian Hujer
004     *
005     * This program is free software; you can redistribute it and/or
006     * modify it under the terms of the GNU General Public License as
007     * published by the Free Software Foundation; either version 2 of the
008     * License, or (at your option) any later version.
009     *
010     * This program is distributed in the hope that it will be useful, but
011     * WITHOUT ANY WARRANTY; without even the implied warranty of
012     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
013     * General Public License for more details.
014     *
015     * You should have received a copy of the GNU General Public License
016     * along with this program; if not, write to the Free Software
017     * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
018     * 02111-1307, USA.
019     */
020    
021    package net.sf.japi.swing.treetable;
022    
023    import java.awt.Graphics;
024    import java.awt.Component;
025    import java.awt.Dimension;
026    import javax.swing.JTable;
027    import javax.swing.AbstractCellEditor;
028    import javax.swing.JTree;
029    import javax.swing.tree.TreeModel;
030    import javax.swing.tree.DefaultTreeSelectionModel;
031    import javax.swing.table.TableCellEditor;
032    import javax.swing.table.TableCellRenderer;
033    
034    /** This example shows how to create a simple JTreeTable component,
035     * by using a JTree as a renderer (and editor) for the cells in a
036     * particular column in the JTable.
037     */
038    public class JTreeTable<R, T> extends JTable {
039    
040        /** TreeTableCellRenderer. */
041        private TreeTableCellRenderer tree;
042    
043        /** Create a JTreeTable. */
044        public JTreeTable(final TreeTableModel<R, T> treeTableModel) {
045            tree = new TreeTableCellRenderer(new TreeTableModelTreeModelAdapter<R, T>(treeTableModel));
046            setModel(new TreeTableModelTableModelAdapter(treeTableModel, tree));
047    
048            // Force the JTable and JTree to share their row selection models.
049            tree.setSelectionModel(new DefaultTreeSelectionModel() {
050    
051                {
052                    setSelectionModel(listSelectionModel);
053                }
054    
055            });
056            // Make the tree and table row heights the same.
057            tree.setRowHeight(getRowHeight());
058    
059            // Install the tree editor renderer and editor.
060            setDefaultRenderer(TreeTableModel.class, tree);
061            setDefaultEditor(TreeTableModel.class, new TreeTableCellEditor());
062    
063            setShowGrid(false);
064            setIntercellSpacing(new Dimension(0, 0));
065        }
066    
067        /* Workaround for BasicTableUI anomaly. Make sure the UI never tries to
068         * paint the editor. The UI currently uses different techniques to
069         * paint the renderers and editors and overriding setBounds() below
070         * is not the right thing to do for an editor. Returning -1 for the
071         * editing row in this case, ensures the editor is never painted.
072         */
073        @Override public int getEditingRow() {
074            return getColumnClass(editingColumn) == TreeTableModel.class ? -1 : editingRow;
075        }
076    
077        /** Renderer for TreeTableCells. */
078        public class TreeTableCellRenderer extends JTree implements TableCellRenderer {
079    
080            private int visibleRow;
081    
082            public TreeTableCellRenderer(final TreeModel model) {
083                super(model);
084            }
085    
086            @Override public void setBounds(final int x, final int y, final int width, final int height) {
087                super.setBounds(x, 0, width, JTreeTable.this.getHeight());
088            }
089    
090            @Override public void paint(final Graphics g) {
091                g.translate(0, -visibleRow * getRowHeight());
092                super.paint(g);
093            }
094    
095            public Component getTableCellRendererComponent(final JTable table, final Object value, final boolean isSelected, final boolean hasFocus, final int row, final int column) {
096                if (isSelected) {
097                    setBackground(table.getSelectionBackground());
098                } else {
099                    setBackground(table.getBackground());
100                }
101    
102                visibleRow = row;
103                return this;
104            }
105    
106        } // class TreeTableCellRenderer
107    
108        /** Editor for TreeTableCells. */
109        public class TreeTableCellEditor extends AbstractCellEditor implements TableCellEditor {
110    
111            /** {@inheritDoc} */
112            public Component getTableCellEditorComponent(final JTable table, final Object value, final boolean isSelected, final int row, final int column) {
113                return tree;
114            }
115    
116            /** {@inheritDoc} */
117            public Object getCellEditorValue() {
118                return "bla";  //TODO
119            }
120    
121        } // class TreeTableCellEditor
122    
123    } // class JTreeTable