WholeNumberCellEditor.java

package dr.app.gui.table;

import dr.app.gui.components.WholeNumberField;

import javax.swing.*;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;


public class WholeNumberCellEditor extends DefaultCellEditor {

    private WholeNumberField editor;

    public WholeNumberCellEditor(int minValue, int maxValue) {
        super(new WholeNumberField(minValue, maxValue));

        editor = (WholeNumberField) getComponent();

        setClickCountToStart(2); //This is usually 1 or 2.

        // Must do this so that editing stops when appropriate.
        editor.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                fireEditingStopped();
            }
        });
    }

    protected void fireEditingStopped() {
        super.fireEditingStopped();
    }

    public Object getCellEditorValue() {
        return editor.getValue();
    }

    public Component getTableCellEditorComponent(JTable table,
                                                 Object value,
                                                 boolean isSelected,
                                                 int row,
                                                 int column) {
        editor.setValue(((Integer) value).intValue());
        return editor;
    }
}