EnabledComponentBasedOnWidgets.java

package es.icarto.gvsig.navtableforms.ormlite.widgetsdependency;

import java.awt.Color;
import java.awt.Font;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseListener;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.swing.JCheckBox;
import javax.swing.JComboBox;
import javax.swing.JComponent;
import javax.swing.JFormattedTextField;
import javax.swing.JTable;
import javax.swing.JTextArea;
import javax.swing.JTextField;

import com.toedter.calendar.JDateChooser;

import es.icarto.gvsig.navtableforms.IValidatableForm;
import es.udc.cartolab.gvsig.navtable.format.DateFormatNT;

public class EnabledComponentBasedOnWidgets implements ActionListener {

    private final JComponent component;
    private final Map<JComponent, List<String>> conditions = new HashMap<JComponent, List<String>>();
    private boolean removeDependentValues;
    private final IValidatableForm form;
    // to store the table listeners when the component is a jtable
    private MouseListener[] listeners = new MouseListener[0];

    public EnabledComponentBasedOnWidgets(JComponent component,
	    IValidatableForm form) {
	this.component = component;
	this.form = form;
    }

    public void addCondition(JComponent widget, String value) {
	if (!conditions.containsKey(widget)) {
	    conditions.put(widget, new ArrayList<String>());
	}
	conditions.get(widget).add(value);
    }

    public void addConditions(JComponent widget, List<String> values) {
	if (!conditions.containsKey(widget)) {
	    conditions.put(widget, new ArrayList<String>());
	}
	conditions.get(widget).addAll(values);
    }

    public void setRemoveDependentValues(boolean removeDependentValues) {
	this.removeDependentValues = removeDependentValues;
    }

    public void setListeners() {
	for (JComponent widget : conditions.keySet()) {
	    if (widget instanceof JCheckBox) {
		((JCheckBox) widget).addActionListener(this);
	    } else if (widget instanceof JComboBox) {
		((JComboBox) widget).addActionListener(this);
	    }
	}
	listeners = component.getMouseListeners() != null ? component
		.getMouseListeners() : new MouseListener[0];
    }

    @Override
    public void actionPerformed(ActionEvent e) {
	if (!form.isFillingValues()) {
	    enableComponent(false);
	}
    }

    private void enableComponent(boolean initialLoad) {
	Set<JComponent> widgets = conditions.keySet();
	boolean enable = true;
	for (JComponent widget : widgets) {
	    if (widget instanceof JCheckBox) {
		enable = enable
			&& checkCheckBoxCondition((JCheckBox) widget,
				conditions.get(widget));
	    } else if (widget instanceof JComboBox) {
		enable = enable
			&& checkComboBoxCondition((JComboBox) widget,
				conditions.get(widget));
	    }
	}
	changeComponentState(enable, initialLoad);
    }

    private boolean checkComboBoxCondition(JComboBox widget, List<String> values) {
	if (widget.getSelectedItem() != null) {
	    String selected = widget.getSelectedItem().toString();
	    for (String value : values) {
		if (value.equalsIgnoreCase(selected)) {
		    return true;
		}
	    }
	}
	return false;
    }

    private boolean checkCheckBoxCondition(JCheckBox widget, List<String> values) {
	String selected = String.valueOf(widget.isSelected());
	for (String value : values) {
	    if (value.equalsIgnoreCase(selected)) {
		return true;
	    }
	}
	return false;
    }

    private void changeComponentState(boolean enabled, boolean initialLoad) {
	// We don't have to do anything if state hasn't changed
	if (enabled != component.isEnabled()) {
	    if (component instanceof JTable) {
		// If the component is a table, we adjust its viewport and
		// remove/restore its listeners.
		if (enabled) {
		    ((JTable) component).setFillsViewportHeight(true);
		    for (MouseListener l : listeners) {
			component.addMouseListener(l);
		    }
		} else {
		    ((JTable) component).setFillsViewportHeight(false);
		    listeners = component.getMouseListeners();
		    for (MouseListener l : listeners) {
			component.removeMouseListener(l);
		    }
		}
	    } else if ((component instanceof JComboBox) && !enabled) {
		    // If the component is a combobox and we are disabling
		    // it, prior to that we select the default item.
		    if (((JComboBox) component).getItemCount() > 0) {
			((JComboBox) component).setSelectedIndex(0);
		    }
	    } else if ((component instanceof JCheckBox) && !enabled
			    && ((JCheckBox) component).isSelected()) {
			// If the component is a checkbox, we are disabling
			// it and it was checked, prior to that we uncheck it.
			((JCheckBox) component).doClick();
	    } 
		
	    
	    component.setEnabled(enabled);
	    
	    if ((component instanceof JDateChooser) && enabled) {
		initDateChooser((JDateChooser) component);
	    }
	    if (removeDependentValues && !initialLoad) {
		removeValue(component);
	    }
	}
    }
    
    private void initDateChooser(JDateChooser c) {
	SimpleDateFormat dateFormat = DateFormatNT.getDateFormat();
	c.setDateFormatString(dateFormat.toPattern());
	c.getDateEditor().setEnabled(false);
	c.getDateEditor().getUiComponent()
	.setBackground(new Color(255, 255, 255));
	c.getDateEditor().getUiComponent()
	.setFont(new Font("Arial", Font.PLAIN, 11));
	c.getDateEditor().getUiComponent().setToolTipText(null);
    }

    private void removeValue(JComponent c) {
	String fieldName = c.getName();
	if (c instanceof JFormattedTextField) {
	    ((JFormattedTextField) c).setText("");
	    form.getFormController().setValue(fieldName, "");
	} else if (c instanceof JTextField) {
	    ((JTextField) c).setText("");
	    form.getFormController().setValue(fieldName, "");
	} else if (c instanceof JComboBox) {
	    ((JComboBox) c).setSelectedIndex(0);
	    form.getFormController().setValue(fieldName, " ");
	} else if (c instanceof JCheckBox) {
	    ((JCheckBox) c).setSelected(false);
	    form.getFormController().setValue(fieldName, "false");
	} else if (c instanceof JTextArea) {
	    ((JTextArea) c).setText("");
	    form.getFormController().setValue(fieldName, "");
	}
    }

    public void removeListeners() {
	for (JComponent widget : conditions.keySet()) {
	    if (widget instanceof JCheckBox) {
		((JCheckBox) widget).removeActionListener(this);
	    }
	    if (widget instanceof JComboBox) {
		((JComboBox) widget).removeActionListener(this);
	    }
	}

	listeners = new MouseListener[0];
    }

    public void fillValues() {
	enableComponent(true);
    }

}