/* Glazed Lists (c) 2003-2005 */
/* http://publicobject.com/glazedlists/ publicobject.com,*/
/* O'Dell Engineering Ltd.*/
package com.publicobject.misc.swing;
import ca.odell.glazedlists.swing.EventTableModel;
import ca.odell.glazedlists.SeparatorList;
import javax.swing.*;
import javax.swing.event.TableModelEvent;
import javax.swing.plaf.basic.BasicTableUI;
import javax.swing.table.*;
import java.awt.*;
/**
* @author <a href="mailto:jesse@swank.ca">Jesse Wilson</a>
*/
public class JSeparatorTable extends JTable {
/** working with separator cells */
private TableCellRenderer separatorRenderer;
private TableCellEditor separatorEditor;
public JSeparatorTable(EventTableModel tableModel) {
super(tableModel);
setUI(new SpanTableUI());
// use a toString() renderer for the separator
this.separatorRenderer = getDefaultRenderer(Object.class);
}
/** {@inheritDoc} */
public void setModel(TableModel tableModel) {
if(!(tableModel instanceof EventTableModel)) throw new IllegalArgumentException();
super.setModel(tableModel);
}
/** {@inheritDoc} */
public Rectangle getCellRect(int row, int column, boolean includeSpacing) {
EventTableModel eventTableModel = (EventTableModel) getModel();
// sometimes JTable asks for a cellrect that doesn't exist anymore, due
// to an editor being installed before a bunch of rows were removed.
// In this case, just return an empty rectangle, since it's going to
// be discarded anyway
if(row >= eventTableModel.getRowCount()) {
return new Rectangle(0, 0, 0, 0);
}
// if it's the separator row, return the entire row as one big rectangle
Object rowValue = eventTableModel.getElementAt(row);
if(rowValue instanceof SeparatorList.Separator) {
Rectangle firstColumn = super.getCellRect(row, 0, includeSpacing);
Rectangle lastColumn = super.getCellRect(row, getColumnCount() - 1, includeSpacing);
return firstColumn.union(lastColumn);
// otherwise it's business as usual
} else {
return super.getCellRect(row, column, includeSpacing);
}
}
public Rectangle getCellRectWithoutSpanning(int row, int column, boolean includeSpacing) {
return super.getCellRect(row, column, includeSpacing);
}
/** {@inheritDoc} */
public Object getValueAt(int row, int column) {
Object rowValue = ((EventTableModel)getModel()).getElementAt(row);
// if it's the separator row, return the entire row as one big rectangle
if(rowValue instanceof SeparatorList.Separator) {
return rowValue;
// otherwise it's business as usual
} else {
return super.getValueAt(row, column);
}
}
/** {@inheritDoc} */
public TableCellRenderer getCellRenderer(int row, int column) {
Object rowValue = ((EventTableModel)getModel()).getElementAt(row);
// if it's the separator row, use the separator renderer
if(rowValue instanceof SeparatorList.Separator) {
return separatorRenderer;
// otherwise it's business as usual
} else {
return super.getCellRenderer(row, column);
}
}
/** {@inheritDoc} */
public TableCellEditor getCellEditor(int row, int column) {
Object rowValue = ((EventTableModel)getModel()).getElementAt(row);
// if it's the separator row, use the separator editor
if(rowValue instanceof SeparatorList.Separator) {
return separatorEditor;
// otherwise it's business as usual
} else {
return super.getCellEditor(row, column);
}
}
/**
* Get the renderer for separator rows.
*/
public TableCellRenderer getSeparatorRenderer() {
return separatorRenderer;
}
public void setSeparatorRenderer(TableCellRenderer separatorRenderer) {
this.separatorRenderer = separatorRenderer;
}
/**
* Get the editor for separator rows.
*/
public TableCellEditor getSeparatorEditor() {
return separatorEditor;
}
public void setSeparatorEditor(TableCellEditor separatorEditor) {
this.separatorEditor = separatorEditor;
}
/** {@inheritDoc} */
public void tableChanged(TableModelEvent e) {
// stop edits when the table changes, or else we might
// get a relocated edit in the wrong cell!
if(isEditing()) {
super.getCellEditor().cancelCellEditing();
}
// handle the change event
super.tableChanged(e);
}
}
/**
* Modified from BasicTableUI to allow for spanning cells.
*/
class SpanTableUI extends BasicTableUI {
private JSeparatorTable separatorTable;
public void installUI(JComponent c) {
this.separatorTable = (JSeparatorTable) c;
super.installUI(c);
}
/** Paint a representation of the <code>table</code> instance
* that was set in installUI().
*/
public void paint(Graphics g, JComponent c) {
Rectangle clip = g.getClipBounds();
Rectangle bounds = table.getBounds();
// account for the fact that the graphics has already been translated
// into the table's bounds
bounds.x = bounds.y = 0;
if (table.getRowCount() <= 0 || table.getColumnCount() <= 0 ||
// this check prevents us from painting the entire table
// when the clip doesn't intersect our bounds at all
!bounds.intersects(clip)) {
return;
}
Point upperLeft = clip.getLocation();
Point lowerRight = new Point(clip.x + clip.width - 1, clip.y + clip.height - 1);
int rMin = table.rowAtPoint(upperLeft);
int rMax = table.rowAtPoint(lowerRight);
// This should never happen (as long as our bounds intersect the clip,
// which is why we bail above if that is the case).
if (rMin == -1) {
rMin = 0;
}
// If the table does not have enough rows to fill the view we'll get -1.
// (We could also get -1 if our bounds don't intersect the clip,
// which is why we bail above if that is the case).
// Replace this with the index of the last row.
if (rMax == -1) {
rMax = table.getRowCount()-1;
}
boolean ltr = table.getComponentOrientation().isLeftToRight();
int cMin = table.columnAtPoint(ltr ? upperLeft : lowerRight);
int cMax = table.columnAtPoint(ltr ? lowerRight : upperLeft);
// This should never happen.
if (cMin == -1) {
cMin = 0;
}
// If the table does not have enough columns to fill the view we'll get -1.
// Replace this with the index of the last column.
if (cMax == -1) {
cMax = table.getColumnCount()-1;
}
// Paint the grid.
paintGrid(g, rMin, rMax, cMin, cMax);
// Paint the cells.
paintCells(g, rMin, rMax, cMin, cMax);
}
private void paintCell(Graphics g, Rectangle cellRect, int row, int column) {
if (table.isEditing() && table.getEditingRow()==row && table.getEditingColumn()==column) {
Component component = table.getEditorComponent();
component.setBounds(cellRect);
component.validate();
} else {
TableCellRenderer renderer = table.getCellRenderer(row, column);
Component component = table.prepareRenderer(renderer, row, column);
rendererPane.paintComponent(g, component, table, cellRect.x, cellRect.y,
cellRect.width, cellRect.height, true);
}
}
private void paintCells(Graphics g, int rMin, int rMax, int cMin, int cMax) {
JTableHeader header = table.getTableHeader();
TableColumn draggedColumn = (header == null) ? null : header.getDraggedColumn();
TableColumnModel cm = table.getColumnModel();
int columnMargin = cm.getColumnMargin();
Rectangle cellRect;
TableColumn aColumn;
int columnWidth;
if (table.getComponentOrientation().isLeftToRight()) {
for(int row = rMin; row <= rMax; row++) {
for(int column = cMin; column <= cMax; column++) {
aColumn = cm.getColumn(column);
cellRect = table.getCellRect(row, column, false);
if (aColumn != draggedColumn) {
paintCell(g, cellRect, row, column);
}
}
}
} else {
for(int row = rMin; row <= rMax; row++) {
cellRect = table.getCellRect(row, cMin, false);
aColumn = cm.getColumn(cMin);
if (aColumn != draggedColumn) {
columnWidth = aColumn.getWidth();
cellRect.width = columnWidth - columnMargin;
paintCell(g, cellRect, row, cMin);
}
for(int column = cMin+1; column <= cMax; column++) {
aColumn = cm.getColumn(column);
columnWidth = aColumn.getWidth();
cellRect.width = columnWidth - columnMargin;
cellRect.x -= columnWidth;
if (aColumn != draggedColumn) {
paintCell(g, cellRect, row, column);
}
}
}
}
// Paint the dragged column if we are dragging.
if (draggedColumn != null) {
paintDraggedArea(g, rMin, rMax, draggedColumn, header.getDraggedDistance());
}
// Remove any renderers that may be left in the rendererPane.
rendererPane.removeAll();
}
/*
* Paints the grid lines within <I>aRect</I>, using the grid
* color set with <I>setGridColor</I>. Paints vertical lines
* if <code>getShowVerticalLines()</code> returns true and paints
* horizontal lines if <code>getShowHorizontalLines()</code>
* returns true.
*/
private void paintGrid(Graphics g, int rMin, int rMax, int cMin, int cMax) {
g.setColor(table.getGridColor());
Rectangle minCell = table.getCellRect(rMin, cMin, true);
Rectangle maxCell = table.getCellRect(rMax, cMax, true);
Rectangle damagedArea = minCell.union( maxCell );
if (table.getShowHorizontalLines()) {
int tableWidth = damagedArea.x + damagedArea.width;
int y = damagedArea.y;
for (int row = rMin; row <= rMax; row++) {
y += table.getRowHeight(row);
g.drawLine(damagedArea.x, y - 1, tableWidth - 1, y - 1);
}
}
if (table.getShowVerticalLines()) {
TableColumnModel cm = table.getColumnModel();
int tableHeight = damagedArea.y + damagedArea.height;
int x;
if (table.getComponentOrientation().isLeftToRight()) {
x = damagedArea.x;
for (int column = cMin; column <= cMax; column++) {
x += cm.getColumn(column).getWidth();
g.drawLine(x - 1, 0, x - 1, tableHeight - 1);
}
} else {
x = damagedArea.x + damagedArea.width;
for (int column = cMin; column < cMax; column++) {
x -= cm.getColumn(column).getWidth();
g.drawLine(x - 1, 0, x - 1, tableHeight - 1);
}
x -= cm.getColumn(cMax).getWidth();
g.drawLine(x, 0, x, tableHeight - 1);
}
}
}
private int viewIndexForColumn(TableColumn aColumn) {
TableColumnModel cm = table.getColumnModel();
for (int column = 0; column < cm.getColumnCount(); column++) {
if (cm.getColumn(column) == aColumn) {
return column;
}
}
return -1;
}
private void paintDraggedArea(Graphics g, int rMin, int rMax, TableColumn draggedColumn, int distance) {
int draggedColumnIndex = viewIndexForColumn(draggedColumn);
for(int row = rMin; row <= rMax; row++) {
// skip separator rows
Object rowValue = ((EventTableModel)separatorTable.getModel()).getElementAt(row);
// only paint the cell on non-separator rows
if(!(rowValue instanceof SeparatorList.Separator)) {
Rectangle cellRect = table.getCellRect(row, draggedColumnIndex, false);
// Paint a gray well in place of the moving column.
g.setColor(table.getParent().getBackground());
g.fillRect(cellRect.x, cellRect.y, cellRect.width, cellRect.height);
// Move to the where the cell has been dragged.
cellRect.x += distance;
// Fill the background.
g.setColor(table.getBackground());
g.fillRect(cellRect.x, cellRect.y, cellRect.width, cellRect.height);
// Paint the vertical grid lines if necessary.
if (table.getShowVerticalLines()) {
g.setColor(table.getGridColor());
int x1 = cellRect.x;
int y1 = cellRect.y;
int x2 = x1 + cellRect.width - 1;
int y2 = y1 + cellRect.height - 1;
// Left
g.drawLine(x1-1, y1, x1-1, y2);
// Right
g.drawLine(x2, y1, x2, y2);
}
// Render the cell value
paintCell(g, cellRect, row, draggedColumnIndex);
}
// Paint the (lower) horizontal grid line if necessary.
if (table.getShowHorizontalLines()) {
g.setColor(table.getGridColor());
Rectangle rcr = table.getCellRect(row, draggedColumnIndex, true);
rcr.x += distance;
int x1 = rcr.x;
int y1 = rcr.y;
int x2 = x1 + rcr.width - 1;
int y2 = y1 + rcr.height - 1;
g.drawLine(x1, y2, x2, y2);
}
}
}
}