Application: gvSIG desktop

package org.gvsig.fmap.dal.swing.impl.featurequery;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.net.URL;

import java.util.List;

import javax.swing.DefaultListModel;

import javax.swing.DefaultListSelectionModel;

import javax.swing.ImageIcon;

import javax.swing.JButton;

import javax.swing.JComboBox;

import javax.swing.JComponent;

import javax.swing.JList;

import javax.swing.JOptionPane;

import javax.swing.ListModel;

import javax.swing.ListSelectionModel;

import javax.swing.event.ChangeEvent;

import javax.swing.event.ChangeListener;

import javax.swing.event.DocumentEvent;

import javax.swing.event.DocumentListener;

import javax.swing.event.ListSelectionEvent;

import javax.swing.text.JTextComponent;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang.StringUtils;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingLocator;

import org.gvsig.expressionevaluator.swing.ExpressionEvaluatorSwingManager;

import org.gvsig.expressionevaluator.swing.ExpressionPickerController;

import org.gvsig.expressionevaluator.swing.JExpressionBuilder;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.expressionevaluator.FeatureAttributeEmulatorExpression;

import org.gvsig.fmap.dal.feature.EditableFeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureQuery;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.swing.DALSwingLocator;

import org.gvsig.fmap.dal.swing.expressionevaluator.FeatureStoreElement;

import org.gvsig.fmap.dal.swing.featurequery.FeatureQueryCalculatedColumnsPanel;

import org.gvsig.fmap.dal.swing.featuretype.FeatureTypeAttributePanel;

import org.gvsig.fmap.dal.swing.featuretype.FeatureTypePanel;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dataTypes.Coercion;

import org.gvsig.tools.dataTypes.CoercionException;

import org.gvsig.tools.dataTypes.DataType;

import org.gvsig.tools.dataTypes.DataTypes;

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.gvsig.tools.swing.api.pickercontroller.DataTypePickerController;

import org.gvsig.tools.swing.api.windowmanager.Dialog;

import org.gvsig.tools.swing.api.windowmanager.WindowManager;

import org.gvsig.tools.swing.api.windowmanager.WindowManager_v2;

import org.gvsig.tools.swing.icontheme.IconTheme;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 *

 * @author jjdelcerro

 */

public class DefaultFeatureQueryCalculatedColumnsPanel

        extends DefaultFeatureQueryCalculatedColumnsPanelView

        implements FeatureQueryCalculatedColumnsPanel {

    private class ColumnController implements DocumentListener, ChangeListener, ActionListener {

        private final JTextComponent txtName;

        private final JComboBox cboDataType;

        private final JButton btnDataType;

        private final JButton btnExpressionBookmarks;

        private final JButton btnExpressionHistory;

        private final JButton btnExpression;

        private final JTextComponent txtExpression;

        private ExpressionPickerController expPicker;

        private DataTypePickerController pickerDataType;

        private boolean hasChanges;

        private final JTextComponent txtSize;

        private final JTextComponent txtPrecision;

        private final JTextComponent txtScale;

        private final Logger LOGGER = LoggerFactory.getLogger(ColumnController.class);

        public ColumnController(

                JTextComponent txtName,

                JComboBox cboDataType,

                JButton btnDataType,

                JTextComponent txtExpression,

                JButton btnExpression,

                JButton btnExpressionHistory,

                JButton btnExpressionBookmarks,

                JTextComponent txtSize,

                JTextComponent txtPrecision,

                JTextComponent txtScale

        ) {

            this.txtName = txtName;

            this.cboDataType = cboDataType;

            this.btnDataType = btnDataType;

            this.txtExpression = txtExpression;

            this.btnExpression = btnExpression;

            this.btnExpressionHistory = btnExpressionHistory;

            this.btnExpressionBookmarks = btnExpressionBookmarks;

            this.txtSize = txtSize;

            this.txtPrecision = txtPrecision;

            this.txtScale = txtScale;

            this.hasChanges = false;

            this.initComponents();

            this.setEnabled(false);

        }

        public void initComponents() {

            this.pickerDataType = ToolsSwingLocator.getToolsSwingManager().createDataTypePickerController(this.cboDataType, this.btnDataType, false);

            ExpressionEvaluatorSwingManager managerExpSwing = ExpressionEvaluatorSwingLocator.getManager();

            this.expPicker = managerExpSwing.createExpressionPickerController(

                    txtExpression,

                    btnExpression,

                    btnExpressionBookmarks,

                    btnExpressionHistory

            );

            this.txtName.getDocument().addDocumentListener(this);

            this.cboDataType.addActionListener(this);

            this.txtExpression.getDocument().addDocumentListener(this);

            this.txtSize.getDocument().addDocumentListener(this);

            this.txtPrecision.getDocument().addDocumentListener(this);

            this.txtScale.getDocument().addDocumentListener(this);

            //Todo: que el listener expresion lo gestion el picker

            this.cboDataType.addActionListener(new ActionListener() {

                @Override

                public void actionPerformed(ActionEvent ae) {

                    doSetDefaultValuesForDataType();

                    doEnabledByDataType();

                }

            });

        }

        public void doEnabledByDataType() {

            JComboBox comboBox = cboDataType;

            if (comboBox.getSelectedItem()==null) {

                return;

            }

            DataType dataType = (DataType) comboBox.getSelectedItem();

            txtSize.setEnabled(dataType.supportSize());

            txtPrecision.setEnabled(dataType.supportPrecision());

            txtScale.setEnabled(dataType.supportScale());

        }

        public void doSetDefaultValuesForDataType() {

            if (this.cboDataType.getSelectedItem()==null) {

                return;

            }

            DataType dataType = (DataType) this.cboDataType.getSelectedItem();

            if (dataType.supportSize() && txtSize.getText().isEmpty()) {

                txtSize.setText(String.valueOf(dataType.getDefaultSize()));

            } else if (!dataType.supportSize()) {

                txtSize.setText("");

            }

            if (dataType.supportPrecision() && txtPrecision.getText().isEmpty()) {

                txtPrecision.setText(String.valueOf(dataType.getDefaultPrecision()));

            } else if (!dataType.supportPrecision()) {

                txtPrecision.setText("");

            }

            if (dataType.supportScale() && txtScale.getText().isEmpty()) {

                txtScale.setText(String.valueOf(dataType.getDefaultScale()));

            } else if (!dataType.supportScale()) {

                txtScale.setText("");

            }

        }

        public boolean hasChanges() {

            return this.hasChanges;

        }

        public void setEnabled(boolean enabled) {

            this.txtName.setEnabled(enabled);

            this.expPicker.setEnabled(enabled);

            this.cboDataType.setEnabled(enabled);

            this.txtSize.setEnabled(enabled);

            this.txtPrecision.setEnabled(enabled);

            this.txtScale.setEnabled(enabled);

        }

        public void clean() {

            this.txtName.setText("");

            this.expPicker.set(null);

            this.cboDataType.setSelectedIndex(0);

            this.txtSize.setText("");

            this.txtPrecision.setText("");

            this.txtScale.setText("");

        }

        public void put(EditableFeatureAttributeDescriptor attr) {

            this.clean();

            this.txtName.setText(attr.getName());

            FeatureAttributeEmulatorExpression emu = (FeatureAttributeEmulatorExpression) attr.getFeatureAttributeEmulator();

            if (emu != null) {

                this.expPicker.set(emu.getExpression());

            }

            this.pickerDataType.set(attr.getDataType());

            Coercion coerceString = ToolsLocator.getDataTypesManager().get(DataTypes.STRING).getCoercion();

            txtSize.setEnabled(attr.getDataType().supportSize());

            if (attr.getDataType().supportSize()) {

                try {

                    this.txtSize.setText((String) coerceString.coerce(attr.getSize()));

                } catch (CoercionException ex) {

                    LOGGER.warn("Not able to coerce text to integer from size box.", ex);

                }

            }

            txtPrecision.setEnabled(attr.getDataType().supportPrecision());

            if (attr.getDataType().supportPrecision()) {

                try {

                    this.txtPrecision.setText((String) coerceString.coerce(attr.getPrecision()));

                } catch (CoercionException ex) {

                    LOGGER.warn("Not able to coerce text to integer from precision box.", ex);

                }

            }

            txtScale.setEnabled(attr.getDataType().supportScale());

            if (attr.getDataType().supportScale()) {

                try {

                    this.txtScale.setText((String) coerceString.coerce(attr.getScale()));

                } catch (CoercionException ex) {

                    LOGGER.warn("Not able to coerce text to integer from scale box.", ex);

                }

            }

//            doEnabledByDataType();

            this.hasChanges = false;

        }

        public EditableFeatureAttributeDescriptor fetch(EditableFeatureAttributeDescriptor attr) {

            attr.setName(this.txtName.getText());

            attr.setFeatureAttributeEmulator(this.expPicker.get());

            attr.setDataType(this.pickerDataType.get());

            Coercion coerceInteger = ToolsLocator.getDataTypesManager().get(DataTypes.INTEGER).getCoercion();

            try {

                if (attr.getDataType().supportSize()) {

                    attr.setSize((int) coerceInteger.coerce(this.txtSize.getText()));

                }

            } catch (CoercionException ex) {

                LOGGER.warn("Not able to coerce text to integer from size box.", ex);

            }

            try {

                if (attr.getDataType().supportPrecision()) {

                    attr.setPrecision((int) coerceInteger.coerce(this.txtPrecision.getText()));

                }

            } catch (CoercionException ex) {

                LOGGER.warn("Not able to coerce text to integer from precision box.", ex);

            }

            try {

                if (attr.getDataType().supportScale()) {

                    attr.setScale((int) coerceInteger.coerce(this.txtScale.getText()));

                }

            } catch (CoercionException ex) {

                LOGGER.warn("Not able to coerce text to integer from scale box.", ex);

            }

            this.hasChanges = false;

            return attr;

        }

        private void doHasChanges() {

            this.hasChanges = true;

        }

        @Override

        public void insertUpdate(DocumentEvent de) {

            doHasChanges();

        }

        @Override

        public void removeUpdate(DocumentEvent de) {

            doHasChanges();

        }

        @Override

        public void changedUpdate(DocumentEvent de) {

            doHasChanges();

        }

        @Override

        public void stateChanged(ChangeEvent ce) {

            doHasChanges();

        }

        @Override

        public void actionPerformed(ActionEvent ae) {

            doHasChanges();

        }

        public void setStore(FeatureStore store) {

             FeatureStoreElement calculatedStoreElement = DALSwingLocator.getSwingManager()

                    .createFeatureStoreElement();

            calculatedStoreElement.setFeatureStore(store);

            this.expPicker.removeAllSymbolTables();

            this.expPicker.addElement(calculatedStoreElement);

        }

    }

    private ColumnController columnController;

    private FeatureStore store;

    private FeatureType featureType;

    private FeatureQuery query;

    private JExpressionBuilder pckExpression;

    private static String COLUMN_DEFAULT_NAME = "Field";

    private EditableFeatureAttributeDescriptor actualEditableAttribute;

    private DefaultListModel lstAttributesModel;

    private static final Logger LOGGER = LoggerFactory.getLogger(DefaultFeatureQueryCalculatedColumnsPanel.class);

    public DefaultFeatureQueryCalculatedColumnsPanel() {

        this.columnController = null;

        this.initComponents();

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    @Override

    public ImageIcon loadImage(String imageName) {

        String name = FilenameUtils.getBaseName(imageName);

        IconTheme theme = ToolsSwingLocator.getIconThemeManager().getDefault();

        if (theme.exists(name)) {

            return theme.get(name);

        }

        URL url = this.getClass().getResource(name + ".png");

        if (url == null) {

            return null;

        }

        return new ImageIcon(url);

    }

    @Override

    public void setStore(FeatureStore store) {

        try {

            this.featureType = store.getDefaultFeatureType();

            this.store = store;

            this.query = store.createFeatureQuery();

            this.columnController.setStore(store);

            this.pckExpression = DALSwingLocator.getManager().createQueryFilterExpresion(store);

        } catch (DataException ex) {

            throw new RuntimeException("Can't assign store", ex);

        }

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        IconTheme theme = ToolsSwingLocator.getIconThemeManager().getCurrent();

        ImageIcon icon = theme.get("picker-datatype");

        this.btnColumnDataType.setIcon(icon);

        toolsSwingManager.translate(this.lblExpression);

        toolsSwingManager.translate(this.lblName);

        toolsSwingManager.translate(this.lblType);

        toolsSwingManager.translate(this.lblSize);

        toolsSwingManager.translate(this.lblPrecision);

        toolsSwingManager.translate(this.lblScale);

        toolsSwingManager.translate(this.btnApplyChanges);

        toolsSwingManager.translate(this.btnColumnMore);

        this.columnController = new ColumnController(

                this.txtColumnName,

                this.cboColumnDataType,

                this.btnColumnDataType,

                this.txtColumnExpression,

                this.btnColumnExpression,

                this.btnColumnExpressionHistory,

                this.btnColumnExpressionBookmarks,

                this.txtColumnSize,

                this.txtColumnPrecision,

                this.txtColumnScale

        );

//        this.lstAttributes.setCellRenderer(new FeatureAttributeListCellRenderer());

        this.lstAttributes.addListSelectionListener((ListSelectionEvent e) -> {

            if (e.getValueIsAdjusting()) {

                return;

            }

            doSelectAttribute();

        });

        Dimension sz = this.getPreferredSize();

        if (sz.width < DEFAULT_WIDTH) {

            sz.width = DEFAULT_WIDTH;

        }

        if (sz.height < DEFAULT_HEIGHT) {

            sz.height = DEFAULT_HEIGHT;

        }

        this.setPreferredSize(sz);

        this.lstAttributesModel = new DefaultListModel<ListElement<EditableFeatureAttributeDescriptor>>();

//        this.lstAttributesModel = ToolsSwingLocator.getToolsSwingManager().createFilteredListModel();

        this.lstAttributes.setModel(lstAttributesModel);

        this.lstAttributes.setSelectionModel(new DefaultListSelectionModel() {

            @Override

            public void setSelectionInterval(int index0, int index1) {

                if (checkColumnControllerHasChangesWithUser()) {

                    super.setSelectionInterval(index0, index1);

                    doSelectAttribute();

                }

            }

        });

        this.btnAdd.addActionListener((ActionEvent e) -> {

            doAdd();

        });

        this.btnRemove.addActionListener((ActionEvent e) -> {

            doRemove();

        });

        this.btnUp.addActionListener((ActionEvent e) -> {

            doUp(lstAttributes);

        });

        this.btnDown.addActionListener((ActionEvent e) -> {

            doDown(lstAttributes);

        });

        this.btnApplyChanges.addActionListener((ActionEvent e) -> {

            doApplyChanges();

        });

        this.btnColumnMore.addActionListener((ActionEvent e) -> {

            doColumnMore();

        });

    }

    private boolean checkColumnControllerHasChangesWithUser() {

        if (columnController.hasChanges()) {

            I18nManager i18n = ToolsLocator.getI18nManager();

            String message = i18n.getTranslation("_Changes_has_not_been_saved_Are_you_sure_you_want_to_continue");

            String title = i18n.getTranslation("_Changes_not_saved");

            int r = ToolsSwingLocator.getThreadSafeDialogsManager().confirmDialog(message, title,

                    JOptionPane.YES_NO_OPTION,

                    JOptionPane.QUESTION_MESSAGE);

            if (r == JOptionPane.NO_OPTION) {

                return false;

            }

        }

        return true;

    }

    private void doColumnMore() {

        if (this.columnController == null) {

            return;

        }

        final I18nManager i18n = ToolsLocator.getI18nManager();

        WindowManager_v2 winManager = (WindowManager_v2) ToolsSwingLocator.getWindowManager();

        final FeatureTypeAttributePanel panel = DALSwingLocator.getSwingManager().createFeatureTypeAttributePanel();

        panel.clean();

//                panel.setNameAsNew(featureType);

        panel.put(actualEditableAttribute);

        panel.setMode(FeatureTypePanel.MODE_EDIT_ALL);

        final Dialog dialog = winManager.createDialog(

                panel.asJComponent(),

                i18n.getTranslation("_Extra_column"),

                null,

                WindowManager_v2.BUTTONS_OK_CANCEL

        );

        dialog.addActionListener(new ActionListener() {

            @Override

            public void actionPerformed(ActionEvent e) {

                if (dialog.getAction() != WindowManager_v2.BUTTONS_OK) {

                    return;

                }

                try {

                    FeatureAttributeDescriptor prev = actualEditableAttribute.getCopy();

                    panel.fetch(actualEditableAttribute);

                    columnController.put(actualEditableAttribute);

                    if (!prev.equals(actualEditableAttribute)) {

                        columnController.doHasChanges();

                    }

                } catch (Exception ex) {

                    LOGGER.warn("Problems updating feature descriptor.", ex);

                }

            }

        });

        dialog.show(WindowManager.MODE.DIALOG);

    }

    public boolean checkPanel(StringBuilder msg) {

        I18nManager i18n = ToolsLocator.getI18nManager();

        if (this.hasChanges()) {

            int r = ToolsSwingLocator.getThreadSafeDialogsManager().confirmDialog(

                    i18n.getTranslation("_Changes_has_not_been_saved_Are_you_sure_you_want_to_continue"),

                    i18n.getTranslation("_Changes_not_saved"),

                    JOptionPane.YES_NO_OPTION,

                    JOptionPane.QUESTION_MESSAGE);

            if (r == JOptionPane.NO_OPTION) {

                return false;

            }

        }

        return true;

    }

    public boolean hasChanges() {

        return this.columnController.hasChanges();

    }

    private void doSelectAttribute() {

        EditableFeatureAttributeDescriptor value = null;

        if (this.lstAttributes.getSelectedValue() != null) {

            ListElement<EditableFeatureAttributeDescriptor> selectedValue = (ListElement<EditableFeatureAttributeDescriptor>) this.lstAttributes.getSelectedValue();

            if (selectedValue != null) {

                value = selectedValue.getValue();

            }

        }

        if (value == null) {

            this.columnController.clean();

            this.columnController.setEnabled(false);

            this.actualEditableAttribute = null;

        } else {

            //EditableFeatureAttributeDescriptor value = node.getValue();

            this.actualEditableAttribute = value;

            columnController.setEnabled(true);

            columnController.put(value);

        }

    }

    @Override

    public FeatureQuery fetch(FeatureQuery query) {

        if (query == null) {

            return this.query.getCopy();

        }

        this.query.getExtraColumn().clear();

        for (int i = 0; i < lstAttributesModel.getSize(); i++) {

            ListElement<EditableFeatureAttributeDescriptor> element = (ListElement<EditableFeatureAttributeDescriptor>) lstAttributesModel.get(i);

            EditableFeatureAttributeDescriptor newExtraColumn = this.query.getExtraColumn().add(element.getValue().getName());

            newExtraColumn.copyFrom(element.getValue());

        }

        query.copyFrom(this.query);

        return query;

    }

    @Override

    public FeatureQuery fetch() {

        return this.fetch(null);

    }

    @Override

    public void put(FeatureQuery query) {

        this.query.copyFrom(query);

        addExtraColumns(this.query);

    }

    private void addExtraColumns(FeatureQuery query) {

        List<EditableFeatureAttributeDescriptor> cols = query.getExtraColumn().getColumns();

        if (cols == null || cols.isEmpty()) {

            return;

        }

        for (EditableFeatureAttributeDescriptor col : cols) {

            ListElement lf = new ListElement(col.getLabel(), col);

            this.lstAttributesModel.addElement(lf);

        }

        if (this.lstAttributesModel.getSize() > 0) {

            this.lstAttributes.setSelectedIndex(0);

        }

    }

    public static void selfRegister() {

        String[][] iconNames = new String[][]{

            new String[]{"dalswing", "common-applychanges"},

            new String[]{"dalswing", "common-more"},

            new String[]{"dalswing", "featurequery-column-add"},

            new String[]{"dalswing", "featurequery-column-remove"},

            new String[]{"dalswing", "featurequery-column-down"},

            new String[]{"dalswing", "featurequery-column-up"}

        };

        IconTheme theme = ToolsSwingLocator.getIconThemeManager().getCurrent();

        for (String[] icon : iconNames) {

            URL url = DefaultFeatureQueryOrderPanel.class.getResource(icon[1] + ".png");

            theme.registerDefault("DALSwing", icon[0], icon[1], null, url);

        }

    }

    private void doAdd() {

        EditableFeatureAttributeDescriptor newAttr = DALLocator.getDataManager().createFeatureAttributeDescriptor();

        newAttr.setName(COLUMN_DEFAULT_NAME);

        newAttr.setDataType(DataTypes.STRING);

        if (newAttr.getDataType().supportSize()) {

            newAttr.setSize(newAttr.getDataType().getDefaultSize());

        }

        int id = 0;

        while (checkIfValueExistInModel(newAttr.getName(), this.lstAttributesModel)) {

            String newName = COLUMN_DEFAULT_NAME + "_" + id;

            id += 1;

            newAttr.setName(newName);

        }

        this.columnController.put(newAttr);

        this.columnController.setEnabled(true);

        this.lstAttributesModel.addElement(new ListElement(newAttr.getName(), newAttr));

        ListElement.setSelected(this.lstAttributes, newAttr);

        this.actualEditableAttribute = newAttr;

    }

    private boolean checkIfValueExistInModel(String name, ListModel<ListElement<EditableFeatureAttributeDescriptor>> model) {

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

            EditableFeatureAttributeDescriptor element = model.getElementAt(i).getValue();

            if (StringUtils.equals(element.getLabel(), name)) {

                return true;

            }

        }

        return false;

    }

    private void doRemove() {

        ListSelectionModel lsm = lstAttributes.getSelectionModel();

        DefaultListModel model = (DefaultListModel) this.lstAttributes.getModel();

        int actualIndex = lsm.getMinSelectionIndex();

        Object selectedValue = this.lstAttributes.getSelectedValue();

        this.lstAttributesModel.removeElement(selectedValue);

        this.columnController.clean();

        int size = model.size();

        if (size == 0) {

            //List is empty: disable delete, up, and down buttons.

        } else {

            //Adjust the selection.

            if (model.getSize() == 0) {

            } else if (actualIndex >= model.getSize()) {

                lstAttributes.setSelectedIndex(model.getSize() - 1);

            } else {

                lstAttributes.setSelectedIndex(actualIndex);

            }

        }

    }

    private void doApplyChanges() {

        this.columnController.fetch(this.actualEditableAttribute);

        this.lstAttributes.invalidate();

    }

    private void doUp(JList lstColumns) {

        int moveMe = lstColumns.getSelectedIndex();

        if (moveMe != 0) {

            //not already at top

            swap(lstColumns, moveMe, moveMe - 1);

            lstColumns.setSelectedIndex(moveMe - 1);

            lstColumns.ensureIndexIsVisible(moveMe - 1);

        }

    }

    private void doDown(JList lstColumns) {

        int moveMe = lstColumns.getSelectedIndex();

        if (moveMe != lstColumns.getModel().getSize() - 1) {

            //not already at bottom

            swap(lstColumns, moveMe, moveMe + 1);

            lstColumns.setSelectedIndex(moveMe + 1);

            lstColumns.ensureIndexIsVisible(moveMe + 1);

        }

    }

    private void swap(JList lstColumns, int a, int b) {

        DefaultListModel model = (DefaultListModel) lstColumns.getModel();

        Object aObject = model.getElementAt(a);

        Object bObject = model.getElementAt(b);

        model.set(a, bObject);

        model.set(b, aObject);

    }

}