package org.swingml.tablebrowser.ext;
import java.util.*;
import javax.swing.*;
import javax.swing.table.*;
import org.swingml.model.TableColumnModel;
public class TableBrowserCellIconDecorator {
public TableBrowserCellIconDecorator (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());
if (!(originalRenderer instanceof TableBrowserCellIconRenderer)) {
Class clazz = theColumn.getType().getClass();
if (clazz == JButton.class) {
aTable.setDefaultRenderer(theColumn.getType().getClass(), new IconComponentCellEditor());
} else {
aTable.setDefaultRenderer(theColumn.getType().getClass(), new TableBrowserCellIconRenderer(originalRenderer));
}
}
}
}
}