/* * RapidMiner * * Copyright (C) 2001-2011 by Rapid-I and the contributors * * Complete list of developers available at our web site: * * http://rapid-i.com * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * 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 Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see http://www.gnu.org/licenses/. */ package com.rapidminer.gui.properties.celleditors.value; import java.awt.Component; import java.awt.Insets; import java.awt.event.ActionEvent; import java.util.Collection; import java.util.LinkedList; import javax.swing.AbstractCellEditor; import javax.swing.JButton; import javax.swing.JTable; import com.rapidminer.gui.properties.AttributesPropertyDialog; import com.rapidminer.gui.tools.ResourceAction; import com.rapidminer.operator.Operator; import com.rapidminer.parameter.ParameterTypeAttributes; import com.rapidminer.parameter.UndefinedParameterError; /** * The value cell editor for the {@link ParameterTypeAttributes}. Displays * a button and opens a dialog to select attribute subsets in the case that * button is clicked. * * @author Tobias Malbrecht */ public class AttributesValueCellEditor extends AbstractCellEditor implements PropertyValueCellEditor { private static final long serialVersionUID = -2387465714767785072L; private String attributeListString = ""; private final JButton button; private Operator operator; public AttributesValueCellEditor(final ParameterTypeAttributes type) { this.button = new JButton(new ResourceAction(true, "attributes") { private static final long serialVersionUID = -4890375754223285831L; @Override public void actionPerformed(ActionEvent e) { LinkedList<String> preSelectedAttributeNames = new LinkedList<String>(); String combinedNames = null; try { if (operator != null) combinedNames = operator.getParameter(type.getKey()); } catch (UndefinedParameterError er) { } if (combinedNames != null) { for (String attributeName : combinedNames.split("\\|")) { preSelectedAttributeNames.add(attributeName); } } AttributesPropertyDialog dialog = new AttributesPropertyDialog(type, preSelectedAttributeNames); dialog.setVisible(true); if (dialog.isOk()) { StringBuilder builder = new StringBuilder(); boolean first = true; Collection<String> attributeNames = dialog.getSelectedAttributeNames(); for (String attributeName : attributeNames) { if (!first) { builder.append("|"); } builder.append(attributeName); first = false; } attributeListString = builder.toString(); fireEditingStopped(); } else { fireEditingCanceled(); } } }); button.setMargin(new Insets(0, 0, 0, 0)); } @Override public Object getCellEditorValue() { return attributeListString; } @Override public void setOperator(Operator operator) { this.operator = operator; } @Override public boolean rendersLabel() { return false; } @Override public boolean useEditorAsRenderer() { return false; } @Override public Component getTableCellEditorComponent(JTable table, Object value, boolean isSelected, int row, int column) { return button; } @Override public Component getTableCellRendererComponent(JTable table, Object value, boolean isSelected, boolean hasFocus, int row, int column) { return button; } }