package org.swingml.tablebrowser.ext;
import java.util.*;
import javax.swing.*;
import javax.swing.table.*;
import org.swingml.model.TableColumnModel;
/**
* @author Crosslogic
*/
public class TableBrowserCellLabelDecorator {
public TableBrowserCellLabelDecorator (TableBrowserComponent aTable) {
super();
TableBrowserModel theTableModel = (TableBrowserModel) aTable.getModel();
Iterator theColumns = theTableModel.getColumns().iterator();
TableColumnModel theColumn = null;
while (theColumns.hasNext()) {
theColumn = (TableColumnModel) theColumns.next();
TableCellRenderer originalRenderer = aTable.getDefaultRenderer(theColumn.getType().getClass());
TableCellEditor originalEditor = aTable.getDefaultEditor(theColumn.getType().getClass());
if (originalRenderer instanceof JLabel) {
Class clazz = theColumn.getType().getClass();
if (clazz == JButton.class) {
aTable.setDefaultRenderer(theColumn.getType().getClass(), new IconComponentCellEditor());
} else {
aTable.setDefaultRenderer(theColumn.getType().getClass(), new TableBrowserCellLabelRenderer(originalRenderer));
aTable.setDefaultEditor(theColumn.getType().getClass(), new TableBrowserCellLabelEditor(originalEditor));
}
}
}
}
}