gvSIG projects pool

package org.gvsig.topology.swing.impl;

import java.awt.Dimension;

import java.awt.event.MouseAdapter;

import java.awt.event.MouseEvent;

import java.util.ArrayList;

import java.util.List;

import java.util.Objects;

import javax.swing.ImageIcon;

import javax.swing.JLabel;

import javax.swing.JTree;

import javax.swing.event.TreeSelectionEvent;

import javax.swing.event.TreeSelectionListener;

import javax.swing.tree.DefaultTreeCellRenderer;

import javax.swing.tree.TreePath;

import org.apache.commons.lang3.StringUtils;

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

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

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

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

import org.gvsig.topology.lib.api.TopologyDataSet;

import org.gvsig.topology.swing.api.TopologySwingLocator;

import org.gvsig.topology.swing.api.TopologySwingServices;

/**

 *

 * @author jjdelcerro

 */

public class SelectDataSetDialog extends SelectDataSetDialogView {

    private final TopologySwingServices services;

    private Dialog dialog = null;

    SelectDataSetDialog() {

        this.services = TopologySwingLocator.getTopologySwingManager().getDefaultServices();

        this.initComponents();

    }

    private void initComponents() {

        this.treeDataSets.setModel(this.services.getDataSetTreeModel());

        this.expandAll(this.treeDataSets);

        this.treeDataSets.addTreeSelectionListener(new TreeSelectionListener() {

            @Override

            public void valueChanged(TreeSelectionEvent e) {

                performDataSetSelected();

            }

        });

        this.treeDataSets.setCellRenderer(new DefaultTreeCellRenderer() {

            @Override

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

                JLabel label = (JLabel) super.getTreeCellRendererComponent(tree,value, selected, expanded, leaf, row, hasFocus);

                ImageIcon icon = services.getTreeIcon(tree.getModel(),value);

                label.setIcon(icon);

                label.setText(Objects.toString(value, ""));

                return label;

            }

        });

//        this.treeDataSets.addMouseListener(new MouseAdapter() {

//            @Override

//            public void mouseClicked(MouseEvent e) {

//                if( e.getButton()==MouseEvent.BUTTON1 && e.getClickCount()==2 ) {

//                    setVisible(false);

//                }

//            }

//        });

        this.setPreferredSize(new Dimension(350, 400));

        this.translate();

    }

    private void expandAll(JTree tree) {

        int r = 0;

        while (r < tree.getRowCount()) {

            tree.expandRow(r);

            r++;

        }

    }

    private void translate() {

        ToolsSwingManager tsm = ToolsSwingLocator.getToolsSwingManager();

        tsm.translate(this.lblName);

    }

    public void setDialog(Dialog dialog) {

        this.dialog = dialog;

    }

    private void setOkButtonEnabled(boolean enabled) {

        if (this.dialog == null) {

            return;

        }

        this.dialog.setButtonEnabled(WindowManager_v2.BUTTONS_OK, enabled);

    }

    private List<TopologyDataSet> getSelectedDataSets() {

        TreePath[] paths = treeDataSets.getSelectionPaths();

        List<TopologyDataSet> datasets = new ArrayList<>();

        for (TreePath path : paths) {

            Object element = path.getLastPathComponent();

            if (element instanceof TopologyDataSet) {

                TopologyDataSet dataSet = (TopologyDataSet) element;

                datasets.add(dataSet);

            }

        }

        return datasets;

    }

    public List<TopologyDataSet> getDataSets() {

        List<TopologyDataSet> dataSets = this.getSelectedDataSets();

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

            return null;

        }

        if( dataSets.size()==1 ) {

            String name = this.txtName.getText();

            if (!StringUtils.isEmpty(name)) {

                dataSets.get(0).setName(name);

            }

        }

        return dataSets;

    }

    public void performDataSetSelected() {

        List<TopologyDataSet> dataSets = this.getSelectedDataSets();

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

            this.txtName.setText("");

            this.setOkButtonEnabled(false);

        } else {

            this.txtName.setText(dataSets.get(0).getName());

            this.setOkButtonEnabled(true);

        }

    }

}