org.datacleaner.widgets.properties.AbstractMultipleNumberPropertyWidget Maven / Gradle / Ivy

Go to download
/**
 * DataCleaner (community edition)
 * Copyright (C) 2014 Neopost - Customer Information Management
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.datacleaner.widgets.properties;

import java.awt.BorderLayout;
import java.awt.Component;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.lang.reflect.Array;
import java.text.NumberFormat;
import java.text.ParseException;

import javax.swing.JButton;
import javax.swing.border.EmptyBorder;
import javax.swing.event.DocumentEvent;
import javax.swing.text.JTextComponent;

import org.apache.metamodel.util.EqualsBuilder;
import org.datacleaner.descriptors.ConfiguredPropertyDescriptor;
import org.datacleaner.job.builder.ComponentBuilder;
import org.datacleaner.panels.DCPanel;
import org.datacleaner.util.DCDocumentListener;
import org.datacleaner.util.IconUtils;
import org.datacleaner.util.ReflectionUtils;
import org.datacleaner.util.StringUtils;
import org.datacleaner.util.WidgetFactory;
import org.jdesktop.swingx.JXTextField;
import org.jdesktop.swingx.VerticalLayout;

/**
 * {@link PropertyWidget} for Number arrays. Displays number arrays as a set of
 * text boxes and plus/minus buttons to grow/shrink the array.
 */
public abstract class AbstractMultipleNumberPropertyWidget extends AbstractPropertyWidget {

	private final NumberFormat _numberFormat = NumberFormat.getInstance();
	private final DCPanel _textFieldPanel;

	protected AbstractMultipleNumberPropertyWidget(ComponentBuilder componentBuilder, ConfiguredPropertyDescriptor propertyDescriptor) {
		super(componentBuilder, propertyDescriptor);

		_textFieldPanel = new DCPanel();
		_textFieldPanel.setLayout(new VerticalLayout(2));

		N currentValue = getCurrentValue();
		updateComponents(currentValue);

		final JButton addButton = WidgetFactory.createSmallButton(IconUtils.ACTION_ADD);
		addButton.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(ActionEvent e) {
				addTextField(null, true);
				fireValueChanged();
			}
		});

		final JButton removeButton = WidgetFactory.createSmallButton(IconUtils.ACTION_REMOVE);
		removeButton.addActionListener(new ActionListener() {
			@Override
			public void actionPerformed(ActionEvent e) {
				int componentCount = _textFieldPanel.getComponentCount();
				if (componentCount > 0) {
					_textFieldPanel.remove(componentCount - 1);
					_textFieldPanel.updateUI();
					fireValueChanged();
				}
			}
		});

		final DCPanel buttonPanel = new DCPanel();
		buttonPanel.setBorder(new EmptyBorder(0, 4, 0, 0));
		buttonPanel.setLayout(new VerticalLayout(2));
		buttonPanel.add(addButton);
		buttonPanel.add(removeButton);

		final DCPanel outerPanel = new DCPanel();
		outerPanel.setLayout(new BorderLayout());

		outerPanel.add(_textFieldPanel, BorderLayout.CENTER);
		outerPanel.add(buttonPanel, BorderLayout.EAST);

		add(outerPanel);
	}

	@SuppressWarnings("unchecked")
	private N createArray(int length) {
		return (N) Array.newInstance(getPropertyDescriptor().getBaseType(), length);
	}

	public void updateComponents(N values) {
		if (values == null) {
			values = createArray(2);
		}
		final N previousValues = getValue();
		if (!EqualsBuilder.equals(values, previousValues)) {
			for (int i = 0; i < Math.min(Array.getLength(previousValues), Array.getLength(values)); i++) {
				// modify text boxes
				if (!EqualsBuilder.equals(Array.get(previousValues, i), Array.get(values, i))) {
					JTextComponent component = (JTextComponent) _textFieldPanel.getComponent(i);
					component.setText(_numberFormat.format(Array.get(values, i)));
				}
			}

			while (_textFieldPanel.getComponentCount() < Array.getLength(values)) {
				// add text boxes if there are too few
				Object nextValue = Array.get(values, _textFieldPanel.getComponentCount());
				addTextField(nextValue, false);
			}

			while (_textFieldPanel.getComponentCount() > Array.getLength(values)) {
				// remove text boxes if there are too many
				_textFieldPanel.remove(_textFieldPanel.getComponentCount() - 1);
			}
			_textFieldPanel.updateUI();
		}
	}

	private void addTextField(Object value, boolean updateUI) {
		JXTextField textField = WidgetFactory.createTextField();
		if (value != null) {
			textField.setText(_numberFormat.format(value));
		}
		textField.getDocument().addDocumentListener(new DCDocumentListener() {
			@Override
			protected void onChange(DocumentEvent e) {
				fireValueChanged();
			}
		});
		_textFieldPanel.add(textField);
		if (updateUI) {
			_textFieldPanel.updateUI();
		}
	}

	@Override
	public N getValue() {
		Component[] components = _textFieldPanel.getComponents();

		final N result = createArray(components.length);
		for (int i = 0; i < components.length; i++) {
			JXTextField textField = (JXTextField) components[i];
			String stringValue = textField.getText();
			setInArray(result, i, convertToNumber(stringValue));
		}
		return result;
	}

	private void setInArray(Object array, int index, Number number) {
		Class baseType = getPropertyDescriptor().getBaseType();
		if (baseType.isPrimitive()) {
			if (number == null) {
				number = 0;
			}
			if (baseType == byte.class) {
				Array.setByte(array, index, number.byteValue());
			} else if (baseType == short.class) {
				Array.setShort(array, index, number.shortValue());
			} else if (baseType == int.class) {
				Array.setInt(array, index, number.intValue());
			} else if (baseType == long.class) {
				Array.setLong(array, index, number.longValue());
			} else if (baseType == float.class) {
				Array.setFloat(array, index, number.floatValue());
			} else if (baseType == double.class) {
				Array.setDouble(array, index, number.doubleValue());
			}
		} else {
			Array.set(array, index, number);
		}
	}

	private Number convertToNumber(String stringValue) {
		if (StringUtils.isNullOrEmpty(stringValue)) {
			return null;
		}
		Class type = getPropertyDescriptor().getBaseType();
		try {
			final Number number = _numberFormat.parse(stringValue);
			if (type != Number.class) {
				if (ReflectionUtils.isInteger(type)) {
					return Integer.valueOf(number.intValue());
				} else if (ReflectionUtils.isLong(type)) {
					return Long.valueOf(number.longValue());
				} else if (ReflectionUtils.isDouble(type)) {
					return Double.valueOf(number.doubleValue());
				} else if (ReflectionUtils.isShort(type)) {
					return Short.valueOf(number.shortValue());
				} else if (ReflectionUtils.isByte(type)) {
					return Byte.valueOf(number.byteValue());
				} else if (ReflectionUtils.isFloat(type)) {
					return Float.valueOf(number.floatValue());
				}
			}
			return number;
		} catch (ParseException e) {
			throw new IllegalStateException("Cannot parse to number: " + stringValue);
		}
	}

	@Override
	protected void setValue(N value) {
		updateComponents(value);
	}

}