/*
* Copyright 2000-2007 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.util.ui.treetable;
import com.intellij.util.ui.Tree;
import com.intellij.util.ui.UIUtil;
import javax.swing.*;
import javax.swing.border.Border;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.TreeCellRenderer;
import javax.swing.tree.TreeModel;
import java.awt.*;
import java.awt.event.KeyEvent;
/**
* author: lesya
*/
public class TreeTableTree extends Tree{
private Border myBorder;
private final TreeTable myTreeTable;
protected int myVisibleRow;
private boolean myCellFocused;
public TreeTableTree(TreeModel model, TreeTable treeTable) {
super(model);
myTreeTable = treeTable;
setCellRenderer(getCellRenderer());
}
public void updateUI() {
super.updateUI();
TreeCellRenderer tcr = super.getCellRenderer();
if (tcr instanceof DefaultTreeCellRenderer) {
DefaultTreeCellRenderer dtcr = ((DefaultTreeCellRenderer)tcr);
dtcr.setTextSelectionColor(UIUtil.getTableSelectionForeground());
dtcr.setBackgroundSelectionColor(UIUtil.getTableSelectionBackground());
}
}
public void setRowHeight(int rowHeight) {
if (rowHeight > 0) {
super.setRowHeight(rowHeight);
if (myTreeTable != null && myTreeTable.getRowHeight() != rowHeight) {
myTreeTable.setRowHeight(getRowHeight());
}
}
}
public void setBounds(int x, int y, int w, int h) {
super.setBounds(x, 0, w, myTreeTable.getHeight());
}
public void paint(Graphics g) {
Graphics g1 = g.create();
g1.translate(0, -myVisibleRow * getRowHeight());
super.paint(g1);
g1.dispose();
if (myBorder != null){
myBorder.paintBorder(this, g, 0, 0, myTreeTable.getWidth(), getRowHeight());
}
}
public void setBorder(Border border) {
super.setBorder(border);
myBorder = border;
}
public void setTreeTableTreeBorder(Border border) {
myBorder = border;
}
public void setVisibleRow(int row) {
myVisibleRow = row;
setPreferredSize(new Dimension(getPreferredSize().height, getRowBounds(myVisibleRow).width));
}
public void _processKeyEvent(KeyEvent e){
super.processKeyEvent(e);
}
public void setCellFocused(boolean focused) {
myCellFocused = focused;
}
public void setCellRenderer(final TreeCellRenderer x) {
super.setCellRenderer(
new TreeCellRenderer() {
public Component getTreeCellRendererComponent(JTree tree, Object value,
boolean selected, boolean expanded,
boolean leaf, int row, boolean hasFocus) {
return x.getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, myCellFocused);
}
}
);
}
}