---

TTree.java

/*
*  Name:        TTree.java
*  Author:      Philip Milne
*               Scott Violet
*  Maintainer:  Rafael Jesus Alcantara Perez
*  Summary:     Tree used to render tree columns.
*  Date:        $Date: 2003/09/28 23:22:05 $
*  Revision:    $Revision: 1.2 $
*
*  Copyright (C) 2003  Rafael Jesus Alcantara Perez <rafa@dedalo-ing.com>
*
*  This program is free software; you can redistribute it and/or modify
*  it under the terms of the GNU General Public License as published by
*  the Free Software Foundation; either version 2 of the License, or
*  (at your option) any later version.
*
*  This program is distributed in the hope that it will be useful,
*  but WITHOUT ANY WARRANTY; without even the implied warranty of
*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
*  GNU General Public License for more details.
*
*  You should have received a copy of the GNU General Public License
*  along with this program; if not, write to the Free Software
*  Foundation, Inc., 59 Temple Place, Suite 330, Boston,
*  MA 02111-1307, USA.
*
*
*  Copyright 1997-2000 Sun Microsystems, Inc. All Rights Reserved.
*
*  Redistribution and use in source and binary forms, with or
*  without modification, are permitted provided that the following
*  conditions are met:
*
*  - Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
*
*  - Redistribution in binary form must reproduce the above
*    copyright notice, this list of conditions and the following
*    disclaimer in the documentation and/or other materials
*    provided with the distribution.
*
*  Neither the name of Sun Microsystems, Inc. or the names of
*  contributors may be used to endorse or promote products derived
*  from this software without specific prior written permission.
*
*  This software is provided "AS IS," without a warranty of any
*  kind. ALL EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND
*  WARRANTIES, INCLUDING ANY IMPLIED WARRANTY OF MERCHANTABILITY,
*  FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE HEREBY
*  EXCLUDED. SUN AND ITS LICENSORS SHALL NOT BE LIABLE FOR ANY
*  DAMAGES OR LIABILITIES SUFFERED BY LICENSEE AS A RESULT OF OR
*  RELATING TO USE, MODIFICATION OR DISTRIBUTION OF THIS SOFTWARE OR
*  ITS DERIVATIVES. IN NO EVENT WILL SUN OR ITS LICENSORS BE LIABLE
*  FOR ANY LOST REVENUE, PROFIT OR DATA, OR FOR DIRECT, INDIRECT,
*  SPECIAL, CONSEQUENTIAL, INCIDENTAL OR PUNITIVE DAMAGES, HOWEVER
*  CAUSED AND REGARDLESS OF THE THEORY OF LIABILITY, ARISING OUT OF
*  THE USE OF OR INABILITY TO USE THIS SOFTWARE, EVEN IF SUN HAS
*  BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
*
*  You acknowledge that this software is not designed, licensed or
*  intended for use in the design, construction, operation or
*  maintenance of any nuclear facility.
*/

package org.mpcl.nui.treetable;

import java.awt.Color;
import java.awt.Component;
import java.awt.Graphics;
import java.util.Enumeration;
import javax.swing.JTable;
import javax.swing.JTree;
import javax.swing.UIManager;
import javax.swing.table.TableCellRenderer;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.TreeCellRenderer;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreeNode;
import javax.swing.tree.TreePath;
import org.mpcl.nui.TTreeTable;


public class TTree extends JTree
{

  protected int iVisibleRow;

  protected TTreeTable   tTreeTable;


  //
  //  C O N S T R U C T O R S
  //

  public TTree (TTreeTable tTREE_TABLE, TreeModel tTREE_MODEL)
  {
    super (tTREE_MODEL);
    tTreeTable = tTREE_TABLE;
    setCellRenderer (new TTreeTableCellRenderer (tTREE_TABLE));
  }

  public void collapseAll (TreePath tPARENT_TREE_PATH)
  {

    Enumeration   I;
    TreeNode      tChildNode;
    TreePath      tChildPath;
    TreeNode      tLastNode = (TreeNode) tPARENT_TREE_PATH.getLastPathComponent();

    if ( tLastNode.getChildCount() > 0 )
    {
      //
      //  Collapses recursively all child nodes.
      //
      I = tLastNode.children();
      while ( I.hasMoreElements() )
      {
        tChildNode = (TreeNode) I.nextElement();
        tChildPath = tPARENT_TREE_PATH.pathByAddingChild (tChildNode);
        collapseAll (tChildPath);
      }
    }

    //
    //  Collapse must be done bottom-up.
    //
    collapsePath (tPARENT_TREE_PATH);

  }  // collapseAll()

  public void expandAll (TreePath tPARENT_TREE_PATH)
  {

    Enumeration   I;
    TreeNode      tChildNode;
    TreePath      tChildPath;
    TreeNode      tLastNode = (TreeNode) tPARENT_TREE_PATH.getLastPathComponent();

    if ( tLastNode.getChildCount() > 0 )
    {
      //
      //  Expands recursively all child nodes.
      //
      I = tLastNode.children();
      while ( I.hasMoreElements() )
      {
        tChildNode = (TreeNode) I.nextElement();
        tChildPath = tPARENT_TREE_PATH.pathByAddingChild (tChildNode);
        expandAll (tChildPath);
      }
    }

    //
    //  Expansion must be done bottom-up.
    //
    expandPath (tPARENT_TREE_PATH);

  }  // expandAll()

  public void setBounds (int iX, int iY, int iWIDTH, int iHEIGHT)
  {
    super.setBounds (iX, 0, iWIDTH, tTreeTable.getHeight());
  }

  public void updateUI()
  {
    super.updateUI();

    //
    //  Make the tree's cell renderer use the table's cell selection colors.
    //

    DefaultTreeCellRenderer   tDefaultTreeCellRenderer; 
    TreeCellRenderer          tTreeCellRenderer = getCellRenderer();

    if ( tTreeCellRenderer instanceof DefaultTreeCellRenderer )
    {
      tDefaultTreeCellRenderer = (DefaultTreeCellRenderer) tTreeCellRenderer;
      tDefaultTreeCellRenderer.setTextSelectionColor (UIManager.getColor ("Table.selectionForeground"));
      tDefaultTreeCellRenderer.setBackgroundSelectionColor (UIManager.getColor ("Table.selectionBackground"));
    }
  }

  public void setLastVisibleRow (int iVISIBLE_ROW)
  {
    iVisibleRow = iVISIBLE_ROW;
  }

  public void setRowHeight (int iROW_HEIGHT)
  {
    if ( iROW_HEIGHT > 0 )
    {
      super.setRowHeight (iROW_HEIGHT); 
      if ( tTreeTable.getRowHeight() != iROW_HEIGHT )
      {
        tTreeTable.setRowHeight (getRowHeight()); 
      }
    }
  }


  //
  //  S E L E C T O R S
  //

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

}  // class TTree

---