gvSIG projects pool

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (c) 2007-2020 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.vcsgis.swing.impl.addToWorkspace;

import java.awt.Dimension;

import java.awt.event.ActionEvent;

import javax.swing.ComboBoxModel;

import javax.swing.DefaultListCellRenderer;

import javax.swing.JComponent;

import javax.swing.JLabel;

import javax.swing.JList;

import javax.swing.JTree;

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.event.TreeSelectionEvent;

import javax.swing.tree.DefaultTreeCellRenderer;

import javax.swing.tree.TreePath;

import javax.swing.tree.TreeSelectionModel;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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

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

import org.gvsig.tools.swing.api.pickercontroller.PickerController;

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

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

import org.gvsig.tools.util.LabeledValue;

import static org.gvsig.vcsgis.lib.VCSGisManager.ERR_OK;

import org.gvsig.vcsgis.lib.workspace.VCSGisWorkspace;

import org.gvsig.vcsgis.swing.VCSGisJAddToWorkspace;

import org.gvsig.vcsgis.swing.VCSGisSwingLocator;

import org.gvsig.vcsgis.swing.VCSGisSwingManager;

import org.gvsig.vcsgis.swing.VCSGisSwingServices;

import org.slf4j.LoggerFactory;

/**

 *

 * @author gvSIG Team

 */

public class VCSGisJAddToWorkspaceImpl extends VCSGisJAddToWorkspaceView

        implements Component, VCSGisJAddToWorkspace {

    private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(VCSGisJAddToWorkspaceImpl.class);

    private static final int TAB_LAYERS_INDEX = 0;

    private static final int TAB_TABLES_INDEX = 1;

    private PickerController<VCSGisWorkspace> workspacePicker;

    private Dialog dialog;

    private AttributeDescriptorPickerController fieldForLabelPicker;

    public VCSGisJAddToWorkspaceImpl() {

        initializeComponents();

    }

    private void initializeComponents() {

        translate();

        VCSGisSwingManager swingManager = VCSGisSwingLocator.getVCSGisSwingManager();

        this.workspacePicker = swingManager.createWorkspacePickerController(

                cboWorkspace,

                btnWorkspace

        );

        this.workspacePicker.addChangeListener((ChangeEvent e) -> {

            doUpdateEnableComponents();

        });

        this.setPreferredSize(new Dimension(520, 260));

        tabAddToWorkspace.addChangeListener((ChangeEvent e) -> {

            updateTextName();

            updateFieldsForLabel();

            doUpdateEnableComponents();

        });

        VCSGisSwingServices services = swingManager.getDefaultServices();

        ComboBoxModel<LabeledValue> viewDocumentsModel = services.getViewDocumentsComboBoxModel();

        cboView.setModel(viewDocumentsModel);

        cboView.addActionListener((ActionEvent e) -> {

            doUpdateEnableComponents();

        });

        chkAddLayerToView.addChangeListener((ChangeEvent e) -> {

            doUpdateEnableComponents();

        });

        treeLayers.setModel(services.getFeatureStoresTreeModel());

        treeLayers.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);

        treeLayers.getSelectionModel().addTreeSelectionListener((TreeSelectionEvent e) -> {

            updateTextName();

            updateFieldsForLabel();

            doUpdateEnableComponents();

        });

        treeLayers.setCellRenderer(new DefaultTreeCellRenderer() {

            @Override

            public java.awt.Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, boolean leaf, int row, boolean hasFocus) {

                JLabel label = (JLabel) super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus); //To change body of generated methods, choose Tools | Templates.

                if (value instanceof FeatureStore) {

                    label.setText(((FeatureStore) value).getName());

                }

                return label;

            }

        });

        lstTables.setModel(services.getFeatureStoresListModel());

        lstTables.getSelectionModel().setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

        lstTables.getSelectionModel().addListSelectionListener((ListSelectionEvent e) -> {

            updateTextName();

            updateFieldsForLabel();

            doUpdateEnableComponents();

        });

        lstTables.setCellRenderer(new DefaultListCellRenderer(){

            @Override

            public java.awt.Component getListCellRendererComponent(JList<?> list, Object value, int index, boolean isSelected, boolean cellHasFocus) {

                JLabel label = (JLabel) super.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);

                if (value instanceof FeatureStore) {

                    label.setText(((FeatureStore) value).getName());

                }

                return label;

            }

        });

        txtName.getDocument().addDocumentListener(new DocumentListener() {

            @Override

            public void insertUpdate(DocumentEvent e) {

                doUpdateEnableComponents();

            }

            @Override

            public void removeUpdate(DocumentEvent e) {

                doUpdateEnableComponents();

            }

            @Override

            public void changedUpdate(DocumentEvent e) {

                doUpdateEnableComponents();

            }

        });

        this.fieldForLabelPicker = DALSwingLocator.getDataSwingManager().createAttributeDescriptorPickerController(

                cboFieldForLabel, 

                (FeatureAttributeDescriptor t) -> {

            if (t.getType() == DataTypes.GEOMETRY) {

                return false;

            }

            return !t.isComputed();

        });

        this.fieldForLabelPicker.addChangeListener(new ChangeListener() {

            @Override

            public void stateChanged(ChangeEvent e) {

                doUpdateEnableComponents();

            }

        });

        if(this.cboView.getModel().getSize() > 0){

            this.cboView.setEnabled(true);

            this.cboView.setSelectedIndex(0);

            this.chkAddLayerToView.setSelected(true);

        } else {

            this.cboView.setEnabled(false);

            this.chkAddLayerToView.setSelected(false);

        }

        this.chkAddTableToProject.setSelected(true);

    }

    @Override

    public JComponent asJComponent() {

        return this;

    }

    @Override

    public void setDialog(Dialog dialog) {

        this.dialog = dialog;

        this.doUpdateEnableComponents();

    }

    private void updateTextName() {

        FeatureStore value = getFeatureStore();

        if (value != null) {

            txtName.setText(value.getName());

        } else {

            txtName.setText("");

        }

    }

    private void updateFieldsForLabel() {

        FeatureStore store = getFeatureStore();

        if (store != null){

            this.fieldForLabelPicker.setFeatureType(store.getDefaultFeatureTypeQuietly());

        }

    }

    private void doUpdateEnableComponents() {

        VCSGisWorkspace workspace = getWorkspace();

        LabeledValue selectedView = (LabeledValue)cboView.getSelectedItem();

        boolean enableInit = workspace != null && 

                getFeatureStore() != null &&

                StringUtils.isNotBlank(txtName.getText());

        if (tabAddToWorkspace.getSelectedIndex() == TAB_LAYERS_INDEX) {

            enableInit = enableInit && (!chkAddLayerToView.isSelected()

                    || (chkAddLayerToView.isSelected() && selectedView != null && selectedView.getValue() != null));

        }

        if (dialog != null) {

            this.dialog.setButtonEnabled(WindowManager_v2.BUTTON_OK,

                    enableInit

            );

        }

    }

    private void translate() {

        ToolsSwingManager toolsSwingManager = ToolsSwingLocator.getToolsSwingManager();

        toolsSwingManager.translate(this.lblWorkspace);

        toolsSwingManager.translate(this.lblName);

        toolsSwingManager.translate(this.tabAddToWorkspace);

        toolsSwingManager.translate(this.lblFieldForLabel);

        toolsSwingManager.translate(this.chkAddLayerToView);

        toolsSwingManager.translate(this.chkAddTableToProject);

    }

    @Override

    public VCSGisWorkspace getWorkspace() {

        VCSGisWorkspace workspace = this.workspacePicker.get();

        return workspace;

    }

    @Override

    public FeatureStore getFeatureStore() {

        switch (tabAddToWorkspace.getSelectedIndex()) {

            case TAB_LAYERS_INDEX:

                TreePath selection = treeLayers.getSelectionModel().getSelectionPath();

                if(selection != null) {

                    Object[] path = selection.getPath();

                    boolean isLeaf = treeLayers.getModel().isLeaf(path[path.length - 1]);

                    Object obj = path[path.length - 1];

                    if (isLeaf && obj instanceof FeatureStore) {

                        return (FeatureStore) obj;

                    }

                }

            case TAB_TABLES_INDEX:

                Object selected = lstTables.getSelectedValue();

                if(selected instanceof FeatureStore) {

                    return (FeatureStore) selected;

                }

        }

        return null;

    }

    @Override

    public String getTabletName() {

        return this.txtName.getText();

    }

    @Override

    public int add() {

        VCSGisSwingManager swingManager = VCSGisSwingLocator.getVCSGisSwingManager();

        VCSGisSwingServices services = swingManager.getDefaultServices();

        VCSGisWorkspace workspace = getWorkspace();

        if (workspace != null) {

            FeatureStore featureStore = getFeatureStore();

            if (featureStore != null) {

                FeatureAttributeDescriptor descriptor = this.fieldForLabelPicker.get();

                //TODO: Adicionamos pkName si hace falta

                int result = workspace.add(txtName.getText(), featureStore, descriptor == null ? null : descriptor.getName());

                if(result == ERR_OK){

                    switch (tabAddToWorkspace.getSelectedIndex()) {

                        case TAB_LAYERS_INDEX:

                            if(chkAddLayerToView.isSelected()){

                                FeatureStore store = workspace.getFeatureStore(txtName.getText());

                                String layerName = workspace.getLabel()+" - " + store.getName();

                                services.addLayerToView(store, (LabeledValue)cboView.getSelectedItem(), layerName);

                            }

                            break;

                        case TAB_TABLES_INDEX:

                            if(chkAddTableToProject.isSelected()){

                                FeatureStore store = workspace.getFeatureStore(txtName.getText());

                                services.addTableToProject(workspace, store);

                            }

                            break;

                    }

                }

                return result;

            }

        }

        return -1;

    }

}