Application: gvSIG desktop

package org.gvsig.export.swing.impl.panels;

import java.awt.Component;

import java.awt.Font;

import java.util.ArrayList;

import java.util.Comparator;

import java.util.HashMap;

import java.util.List;

import javax.swing.DefaultCellEditor;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JComboBox;

import javax.swing.JComponent;

import javax.swing.JLabel;

import javax.swing.JList;

import javax.swing.JOptionPane;

import javax.swing.JTable;

import javax.swing.ListCellRenderer;

import javax.swing.table.DefaultTableModel;

import javax.swing.table.TableModel;

import org.apache.commons.lang.StringUtils;

import org.gvsig.export.ExportAttributes;

import org.gvsig.export.ExportParameters;

import org.gvsig.export.swing.JExportProcessPanel;

import org.gvsig.export.swing.spi.ExportPanel;

import org.gvsig.export.swing.spi.ExportPanelValidationException;

import org.gvsig.export.swing.spi.ExportPanelsManager;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.DataTypesManager;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.swing.api.ListElement;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.api.ToolsSwingManager;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

public class SelectAttributesPanel

        extends SelectAttributesPanelView

        implements ExportPanel {

    private static final Logger LOG = LoggerFactory.getLogger(SelectAttributesPanel.class);

    public static final int COLUMN_FIELD_NAME = 0;

    public static final int COLUMN_FIELD_NEW_NAME = 1;

    public static final int COLUMN_FIELD_TYPE = 2;

    public static final int COLUMN_FIELD_SIZE = 3;

    public static final int COLUMN_FIELD_EXPORT = 4;

    private final ExportParameters parameters;

    private final JExportProcessPanel processPanel;

    private HashMap<String, String> layerAttributeDescriptor;

    public SelectAttributesPanel(

            JExportProcessPanel processPanel,

            ExportParameters parameters

    ) {

        this.processPanel = processPanel;

        this.parameters = parameters;

        this.parameters.getExportAttributes().setActive(true);

        this.initComponents();

    }

    @Override

    public String getIdPanel() {

        return ExportPanelsManager.PANEL_ATTRIBUTES_SELECTION;

    }

    @Override

    public String getTitlePanel() {

        I18nManager i18n = ToolsLocator.getI18nManager();

        return i18n.getTranslation("_Attribute_Selection");

    }

    @Override

    public boolean validatePanel() throws ExportPanelValidationException {

        ExportAttributes expAttrs;

        expAttrs = this.parameters.getExportAttributes();

        DataTypesManager dataManager = ToolsLocator.getDataTypesManager();

        TableModel model = this.tblAttributes.getModel();

        for (int i = 0; i < model.getRowCount(); i++) {

            String object = (String) model.getValueAt(i, COLUMN_FIELD_NAME);

            for (ExportAttributes.ExportAttribute expAttr : expAttrs) { // assign valuen from table to ExportAttribute

                if (object == null ? expAttr.getName() == null : object.equals(expAttr.getName())) {

                    try {

                        expAttr.setNewName((String) model.getValueAt(i, COLUMN_FIELD_NEW_NAME));

                        Object value = model.getValueAt(i, COLUMN_FIELD_TYPE);

                        int dataType = dataManager.getType(value.toString());

                        expAttr.setNewType(dataType);

                        int sizeValue;

                        try {

                            sizeValue = Integer.parseInt(model.getValueAt(i, COLUMN_FIELD_SIZE).toString());

                        } catch (NumberFormatException ex) {

                            LOG.error("Not been able to set size value" + expAttr.getName() + ".Exception:" + ex);

                            sizeValue = expAttr.getSize();

                        }

                        expAttr.setSize(sizeValue);

                        expAttr.setExported((boolean) model.getValueAt(i, COLUMN_FIELD_EXPORT));

                    } catch (Exception ex) {

                        // TODO

                        // break for incorrect values

                        LOG.error("Not been able to set Export attribute for field" + expAttr.getName() + ".Exception:" + ex);

                    }

                }

            }

        }

        if (!expAttrs.isAttributeNamesValid()) {

            I18nManager i18nManager = ToolsLocator.getI18nManager();

            int as = JOptionPane.showConfirmDialog(

                    this,

                    i18nManager.getTranslation("_Not_valid_attribute_names_Do_you_want_a_suggest_names?"),

                    i18nManager.getTranslation("_Not_valid_attriute_names"),

                    JOptionPane.YES_NO_OPTION

            );

            if (as == JOptionPane.YES_OPTION) {

                expAttrs.fixAttributeNames();

                this.fillTableWithAttributes();

            } else if (as == JOptionPane.NO_OPTION) {

            }

            return false;

        }

        return true;

    }

    @Override

    public void enterPanel() {

        this.fillTableWithAttributes();

    }

    @Override

    public void previousPanel() {

        this.fillTableWithAttributes();

    }

    @Override

    public void nextPanel() {

        //get params are in the validate method

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    private void fillTableWithAttributes() {

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        String[] columnNames = new String[]{

            i18nManager.getTranslation("_Field_name"),

            i18nManager.getTranslation("_Change_field_name"),

            i18nManager.getTranslation("_Change_field_type"),

            i18nManager.getTranslation("_Change_field_size"),

            i18nManager.getTranslation("_Check_export")};

        DataTypesManager dataManager = ToolsLocator.getDataTypesManager();

        List<ListElement<Integer>> fieldTypes = new ArrayList<>();

        fieldTypes.add(new ListElement<>(dataManager.getTypeName(DataTypes.BOOLEAN), DataTypes.BOOLEAN));

        fieldTypes.add(new ListElement<>(dataManager.getTypeName(DataTypes.DATE), DataTypes.DATE));

        fieldTypes.add(new ListElement<>(dataManager.getTypeName(DataTypes.INT), DataTypes.INT));

        fieldTypes.add(new ListElement<>(dataManager.getTypeName(DataTypes.DOUBLE), DataTypes.DOUBLE));

        fieldTypes.add(new ListElement<>(dataManager.getTypeName(DataTypes.LONG), DataTypes.LONG));

        fieldTypes.add(new ListElement<>(dataManager.getTypeName(DataTypes.STRING), DataTypes.STRING));

        fieldTypes.add(new ListElement<>(dataManager.getTypeName(DataTypes.GEOMETRY), DataTypes.GEOMETRY));

        fieldTypes.sort(new Comparator<ListElement<Integer>>() {

            @Override

            public int compare(ListElement<Integer> o1, ListElement<Integer> o2) {

                return o1.toString().compareTo(o2.toString());

            }

        });

//        model = new ComboModelDataType(fieldTypes, fad.getType());

        DefaultComboBoxModel<ListElement<Integer>> modelTypes = new DefaultComboBoxModel<>();

        for (ListElement<Integer> element : fieldTypes) {

            modelTypes.addElement(element);

        }

        JComboBox comboBoxFieldTypes = new JComboBox();

        comboBoxFieldTypes.setModel(modelTypes);

        // Values to field the table are taken from the parameter.getExportAttributes

        Object[][] propertyFields = new Object[this.parameters.getExportAttributes().size()][5];

        this.layerAttributeDescriptor = new HashMap<>();

        for (int nAttr = 0; nAttr < this.parameters.getExportAttributes().size(); nAttr++) {

            //ExportAttributes.ExportAttribute exportAttr : this.parameters.getExportAttributes()) {

            ExportAttributes.ExportAttribute exportAttr = this.parameters.getExportAttributes().get(nAttr);

            propertyFields[nAttr][COLUMN_FIELD_NAME] = exportAttr.getName();

            propertyFields[nAttr][COLUMN_FIELD_NEW_NAME] = exportAttr.getNewName();

            propertyFields[nAttr][COLUMN_FIELD_TYPE] = dataManager.getTypeName(exportAttr.getDataType());

            propertyFields[nAttr][COLUMN_FIELD_SIZE] = exportAttr.getSize();

            propertyFields[nAttr][COLUMN_FIELD_EXPORT] = exportAttr.isExported();

            this.layerAttributeDescriptor.put(

                    exportAttr.getName(),

                    dataManager.getTypeName(exportAttr.getDataType())

            );

        }

        DefaultTableModel model = new DefaultTableModel(propertyFields, columnNames) {

            private static final long serialVersionUID = 1L;

            @Override

            public boolean isCellEditable(int row, int col) {

                return col != 0;

            }

            @Override

            public Class getColumnClass(int column) {

                switch (column) {

                    case 0:

                    case 1:

                    case 3:

                        return String.class;

                    case 2:

                        return JComboBox.class;

                    case 4:

                        return Boolean.class;

                }

                return getValueAt(0, column).getClass();

            }

        };

        this.tblAttributes.setModel(model);

        class TypeDefaultCellEditor extends DefaultCellEditor {

            private final HashMap<String, String> attributes;

            public TypeDefaultCellEditor(JComboBox comboBox, HashMap attributes) {

                super(comboBox);

                this.attributes = attributes;

            }

            @Override

            public Component getTableCellEditorComponent(JTable table, Object value, boolean isSelected,

                    int row, int column) {

                JComboBox editor = (JComboBox) super.getTableCellEditorComponent(table, value, isSelected, row, column);

                Object labelField = table.getModel().getValueAt(row, 0);

                String initialType = this.attributes.get(labelField.toString());

                editor.getModel().setSelectedItem(value);

                class ComboBoxRenderer extends JLabel implements ListCellRenderer {

                    private final String actualValue;

                    private final String initialType;

                    public ComboBoxRenderer(String value, String initialType) {

                        this.actualValue = value;

                        this.initialType = initialType;

                    }

                    @Override

                    public Component getListCellRendererComponent(JList list,

                            Object value,

                            int index,

                            boolean isSelected,

                            boolean cellHasFocus) {

                        if (value != null) {

                            this.setText(value.toString());

                        }

                        Font f = this.getFont();

                        if (StringUtils.equals(value.toString(), initialType)) {

                            this.setFont(f.deriveFont(f.getStyle() | Font.BOLD));

                        } else {

                            this.setFont(f.deriveFont(f.getStyle() & ~Font.BOLD));

                        }

                        return this;

                    }

                }

                ListCellRenderer cmbRenderer = new ComboBoxRenderer(value.toString(), initialType);

                editor.setRenderer(cmbRenderer);

                return editor;

            }

        }

        DefaultCellEditor cellEditor = new TypeDefaultCellEditor(comboBoxFieldTypes, this.layerAttributeDescriptor);

        this.tblAttributes.getColumnModel().getColumn(COLUMN_FIELD_TYPE).setCellEditor(cellEditor);

        this.translate();

    }

    private void initComponents() {

        // init default values

        // init all UI

        this.fillTableWithAttributes();

    }

    private void translate() {

        ToolsSwingManager manager = ToolsSwingLocator.getToolsSwingManager();

        manager.translate(this.lblHeader);

    }

}