TableBrowserCellIconDecorator.java :  » XML-UI » SwingML » org » swingml » tablebrowser » ext » Java Open Source

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));                    
                }
            }
        }
    }
}