Application: gvSIG desktop

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

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import java.net.URL;

import java.util.List;

import javax.swing.DefaultComboBoxModel;

import javax.swing.DefaultListModel;

import javax.swing.ImageIcon;

import javax.swing.JButton;

import javax.swing.JComboBox;

import javax.swing.JComponent;

import javax.swing.JList;

import javax.swing.ListModel;

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.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.featurequery.FeatureQueryCalculatedColumnsPanel;

import org.gvsig.fmap.dal.swing.impl.featuretype.FeatureAttributeListCellRenderer;

import org.gvsig.tools.dataTypes.DataType;

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

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.icontheme.IconTheme;

import org.gvsig.tools.util.LabeledValueImpl;

/**

 *

 * @author jjdelcerro

 */

public class DefaultFeatureQueryCalculatedColumnsPanel

        extends DefaultFeatureQueryCalculatedColumnsPanelView

        implements FeatureQueryCalculatedColumnsPanel {

    private EditableFeatureAttributeDescriptor actualEditableAttribute;

    private DefaultListModel lstAttributesModel;

    private static class ColumnController {

        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 EditableFeatureAttributeDescriptor attr;

        public ColumnController(

                JTextComponent txtName,

                JComboBox cboDataType,

                JButton btnDataType,

                JTextComponent txtExpression,

                JButton btnExpression,

                JButton btnExpressionHistory,

                JButton btnExpressionBookmarks

        ) {

            this.txtName = txtName;

            this.cboDataType = cboDataType;

            this.btnDataType = btnDataType;

            this.txtExpression = txtExpression;

            this.btnExpression = btnExpression;

            this.btnExpressionHistory = btnExpressionHistory;

            this.btnExpressionBookmarks = btnExpressionBookmarks;

            this.initComponents();

            this.setEnabled(false);

        }

        public void initComponents() {

            List<DataType> dataTypes = DALLocator.getDataManager().getDataTypes();

            DefaultComboBoxModel<LabeledValueImpl<DataType>> modelColumn = new DefaultComboBoxModel<LabeledValueImpl<DataType>>();

            for (DataType dataType : dataTypes) {

                LabeledValueImpl<DataType> labeled = new LabeledValueImpl<DataType>(dataType.getName(), dataType);

                modelColumn.addElement(labeled);

            }

            this.cboDataType.setModel(modelColumn);

            ExpressionEvaluatorSwingManager managerExpSwing = ExpressionEvaluatorSwingLocator.getManager();

            this.expPicker = managerExpSwing.createExpressionPickerController(

                    txtExpression,

                    btnExpression,

                    btnExpressionBookmarks,

                    btnExpressionHistory

            );

        }

        public void setEnabled(boolean enabled) {

            this.txtName.setEnabled(enabled);

            this.expPicker.setEnabled(enabled);

            this.cboDataType.setEnabled(enabled);

        }

        public void clean() {

            this.txtName.setText("");

            this.expPicker.set(null);

            this.cboDataType.setSelectedIndex(0);

        }

        public void put(EditableFeatureAttributeDescriptor attr) {

            this.attr = attr;

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

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

            if (emu!=null) {

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

            }

            this.cboDataType.setSelectedItem(attr.getDataType());

        }

        public EditableFeatureAttributeDescriptor fetch(EditableFeatureAttributeDescriptor attr) {

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

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

            LabeledValueImpl<DataType> dataTypeValue = (LabeledValueImpl<DataType>) this.cboDataType.getSelectedItem();

            this.attr.setDataType(dataTypeValue.getValue());

            return null;

        }

    }

    private FeatureStore store;

    private FeatureType featureType;

    private FeatureQuery query;

    private ColumnController colummController;

    private JExpressionBuilder pckExpression;

    private static String COLUMN_DEFAULT_NAME = "Field";

    public DefaultFeatureQueryCalculatedColumnsPanel() {

        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.pckExpression = DALSwingLocator.getManager().createQueryFilterExpresion(store);

        } catch (DataException ex) {

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

        }

    }

    private void initComponents() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        this.colummController = new ColumnController(

                this.txtColumnName,

                this.cboColumnDataType,

                this.btnColumnDataType,

                this.txtColumnExpression,

                this.btnColumnExpression,

                this.btnColumnExpressionHistory,

                this.btnColumnExpressionBookmarks

        );

//        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.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) -> {

            //

        });

    }

    private void doSelectAttribute() {

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

        if (value == null) {

            this.colummController.clean();

            this.colummController.setEnabled(false);

            this.actualEditableAttribute = null;

        } else {

            //EditableFeatureAttributeDescriptor value = node.getValue();

            this.actualEditableAttribute = value;

            colummController.put(value);

            colummController.setEnabled(true);

        }

    }

    @Override

    public FeatureQuery fetch(FeatureQuery query) {

        if (query == null) {

            return this.query.getCopy();

        }

        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;

        }

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

        for (EditableFeatureAttributeDescriptor col : cols) {

            //LabeledValueImpl<EditableFeatureAttributeDescriptor> lf = new LabeledValueImpl<EditableFeatureAttributeDescriptor>(col.getLabel(), col);

            this.lstAttributesModel.addElement(col);

        }

        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);

        int id = 0;

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

            String newName = COLUMN_DEFAULT_NAME + "_" + id;

            id+=1;

            newAttr.setName(newName);

        }

        this.colummController.put(newAttr);

        this.lstAttributesModel.addElement(new ListElement(newAttr.getName(), 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() {

        throw new UnsupportedOperationException("Not supported yet.");

    }

    private void doApplyChanges() {

        this.colummController.fetch(this.actualEditableAttribute);

    }

    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);

    }

}