/*
 * MJTreeTable.java
 *
 * Created on 2007年6月16日, 上午6:11
 *
 * To change this template, choose Tools | Template Manager
 * and open the template in the editor.
 */

package org.joy.mjtreetable;
import javax.swing.tree.*;
import javax.swing.table.*;
import javax.swing.*;
import java.awt.Dimension;
import java.awt.Component;
import java.awt.Graphics;
/**
 * MJTreeTable
 * @version 1.0.0
 * @author molinus
 */
public class MJTreeTable extends JTable{
    /**
     * 樹
     */
    protected TreeTableCellRenderer tree;
    /** Creates a new instance of MJTreeTable */
    public MJTreeTable(TreeTableModel treeTableModel) {
        super();
        
        // Create the tree. It will be used as a renderer and editor.
        tree = new TreeTableCellRenderer(treeTableModel);
        
        // Install a tableModel representing the visible rows in the tree.
        super.setModel(new TreeTableModelAdapter(treeTableModel, tree));
        
        // Force the JTable and JTree to share their row selection models.
        tree.setSelectionModel(new DefaultTreeSelectionModel() {
            // Extend the implementation of the constructor, as if:
            /* public this() */ {
                setSelectionModel(listSelectionModel);
            }
        });
        // Make the tree and table row heights the same.
        tree.setRowHeight(getRowHeight());
        
        // Install the tree editor renderer and editor.
        setDefaultRenderer(TreeTableModel.class, tree);
        setDefaultEditor(TreeTableModel.class, new TreeTableCellEditor());
        
        setShowGrid(false);
        setIntercellSpacing(new Dimension(0, 0));
    }
    /**
     * 取得正在編輯的行
     */
    public int getEditingRow() {
        return (getColumnClass(editingColumn) == TreeTableModel.class) ? -1 : editingRow;  
    }

    /**
    * The renderer used to display the tree nodes, a JTree.  
    */

    public class TreeTableCellRenderer extends JTree implements TableCellRenderer {

	protected int visibleRow;
   
	public TreeTableCellRenderer(TreeModel model) { 
	    super(model); 
	}

	public void setBounds(int x, int y, int w, int h) {
	    super.setBounds(x, 0, w, MJTreeTable.this.getHeight());
	}

	public void paint(Graphics g) {
	    g.translate(0, -visibleRow * getRowHeight());
	    super.paint(g);
	}

	public Component getTableCellRendererComponent(JTable table,
						       Object value,
						       boolean isSelected,
						       boolean hasFocus,
						       int row, int column) {
	    if(isSelected)
		setBackground(table.getSelectionBackground());
	    else
		setBackground(table.getBackground());
       
	    visibleRow = row;
	    return this;
	}
    }

    /** 
    *  The editor used to interact with tree nodes, a JTree.  
    */

    public class TreeTableCellEditor extends AbstractCellEditor implements TableCellEditor {
	public Component getTableCellEditorComponent(JTable table, Object value,
						     boolean isSelected, int r, int c) {
	    return tree;
	}
    }
    
}