Application: gvSIG desktop » gvSIG projections

package org.gvsig.proj.swing.impl;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.beans.PropertyChangeListener;

import java.text.ParseException;

import java.util.ArrayList;

import javax.swing.Action;

import javax.swing.Icon;

import javax.swing.ImageIcon;

import javax.swing.JComponent;

import javax.swing.JTree;

import javax.swing.event.TreeSelectionEvent;

import javax.swing.event.TreeSelectionListener;

import javax.swing.tree.DefaultTreeCellRenderer;

import javax.swing.tree.DefaultTreeModel;

import javax.swing.tree.TreeNode;

import javax.swing.tree.TreePath;

import javax.swing.tree.TreeSelectionModel;

import org.gvsig.proj.CoordinateReferenceSystem;

import org.gvsig.proj.catalog.CRSDefinition;

import org.gvsig.proj.catalog.CRSType;

import org.gvsig.proj.catalog.exception.UnsupportedCoordinateReferenceSystemException;

import org.gvsig.proj.swing.CoordinateReferenceSystemSelectorComponent;

import org.gvsig.proj.swing.impl.tree.AsyncLoadedGroup;

import org.gvsig.proj.swing.impl.tree.BranchNode;

import org.gvsig.proj.swing.impl.tree.CRSSearchFilter;

import org.gvsig.proj.swing.impl.tree.CrsLoader;

import org.gvsig.proj.swing.impl.tree.CrsTreeNode;

import org.gvsig.proj.swing.impl.tree.GroupLoader;

import org.gvsig.proj.swing.impl.tree.LeafNode;

import org.gvsig.proj.swing.impl.tree.LoadingNode;

import org.gvsig.proj.swing.impl.tree.TreeContainer;

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

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class CrsSelectorComponent extends CrsPanel

        implements CoordinateReferenceSystemSelectorComponent, 

                                                                                                Component, TreeSelectionListener,

                                                                                                TreeContainer {

        private CoordinateReferenceSystem selectedCrs = null;

        private ActionListenerSupport listenerSupport = ToolsSwingLocator.getToolsSwingManager().createActionListenerSupport();

        public static final String CRS_SELECTED_ACTION_COMMAND = "CrsSelected";

        protected ImageIcon RECENT_ICON;

        protected ImageIcon FAVOURITE_ICON;

        public ImageIcon FOLDER_OPEN_ICON;

        public ImageIcon FOLDER_CLOSED_ICON;

        public ImageIcon CRS_ICON;

        public ImageIcon LOADING_ICON;

        private static Logger logger = LoggerFactory.getLogger(CrsSelectorComponent.class);

        protected DefaultTreeModel model;

        protected BranchNode root;

        protected RecentCRSs recent;

        public CrsSelectorComponent(DefaultCoordinateReferenceSystemSwingManager manager) {

                super(manager);

                initIcons();

                initComponents();

                initTree();

                resetSearchCombo(null);

                initSearchActions();

        }

        @Override

        public ImageIcon loadImage( String imageName ) {

                if (imageName.toLowerCase().endsWith(".png") || imageName.toLowerCase().endsWith(".gif")) {

                        return this.manager.getIconTheme().get(imageName.substring(0, (imageName.length()-4)));

                }

                return this.manager.getIconTheme().get(imageName);

        }

        protected void initComponents() {

                // hide spatial filter since it is currently not supported

                this.btnFilterDropdown.setVisible(false);

                this.pnl_filterType.setVisible(false);

                this.cboFilterSpatial.setVisible(false);

                this.btnSearch.setToolTipText(manager.getTranslation("Search"));

            this.btnSearchRemove.setToolTipText(manager.getTranslation("Clear search"));

            this.btnFavoritesAdd.setToolTipText(manager.getTranslation("Add selected CRS to favorites"));

            this.btnFavoritesAdd1.setToolTipText(manager.getTranslation("Remove selected CRS from favorites"));

            this.btnCrsAdd.setToolTipText(manager.getTranslation("Create a new CRS"));

            this.btnFavoritesAdd2.setToolTipText(manager.getTranslation("Remove selected CRS from recents"));

            this.lbl_currentCrs.setText(manager.getTranslation("Current coordinate system:"));

            this.jtxt_description.setEditable(false);

                this.jtxt_description.setText("");

                this.jtxt_description.setLineWrap(true);

                this.jtxt_description.setWrapStyleWord(true);

                this.txtCrsWkt.setEditable(false);

                this.txtCrsWkt.setText("");

        }

        protected void search() {

                String filter = getAlphanumericFilter();

                //this.cboFilterAlpha.removei

                //this.cboFilterAlpha.insertItemAt(filter, 0);

                if (filter != null && !filter.equals("")) {

                        //this.cboFilterAlpha.addItem(filter);

                        manager.getCoordinateReferenceSystemTextFilterHistory().add(filter);

                }

                for (int i=0; i<root.getChildCount(); i++) {

                        TreeNode node = root.getChildAt(i);

                        if (node instanceof BranchNode) {

                                ((BranchNode)node).reloadChildren();

                        }

                }

                resetSearchCombo(filter);

                model.reload();

            expandBranches();

        }

        protected void expandBranches() {

                expandBranches(this.root);

        }

        private void expandBranches(TreeNode root) {

                for (int i=0; i<root.getChildCount(); i++) {

                        TreeNode node = root.getChildAt(i);

                        if (node instanceof BranchNode) {

                                TreeNode[] path = this.model.getPathToRoot(node);

                                if (path != null) {

                                        this.treeResults.expandPath(new TreePath(path));

                                }

                                expandBranches(node);

                        }

                }

        }

        protected void resetSearchCombo(String selectedItem) {

                this.cboFilterAlpha.removeAllItems();

                if ("".equals(selectedItem)) {

                        this.cboFilterAlpha.insertItemAt(selectedItem, 0);

                }

                for (String filter: manager.getCoordinateReferenceSystemTextFilterHistory()) {

                        this.cboFilterAlpha.addItem(filter);        

                }

                this.cboFilterAlpha.setSelectedItem(selectedItem);

        }

        protected void initSearchActions() {

                this.cboFilterAlpha.getActionMap().put("enterPressed", new Action() {

                        @Override

                        public void actionPerformed(ActionEvent e) {

                                search();        

                        }

                        @Override

                        public void setEnabled(boolean b) {

                                // TODO Auto-generated method stub

                        }

                        @Override

                        public void removePropertyChangeListener(PropertyChangeListener listener) {

                                // TODO Auto-generated method stub

                        }

                        @Override

                        public void putValue(String key, Object value) {

                                // TODO Auto-generated method stub

                        }

                        @Override

                        public boolean isEnabled() {

                                return true;

                        }

                        @Override

                        public Object getValue(String key) {

                                // TODO Auto-generated method stub

                                return null;

                        }

                        @Override

                        public void addPropertyChangeListener(PropertyChangeListener listener) {

                                // TODO Auto-generated method stub

                        }

                });

                this.btnSearch.addActionListener(new ActionListener() {

                        @Override

                        public void actionPerformed(ActionEvent e) {

                                search();

                        }

                });

                this.btnSearchRemove.addActionListener(new ActionListener() {

                        @Override

                        public void actionPerformed(ActionEvent e) {

                                resetSearchCombo("");

                                search();

                        }

                });

        }

        protected void reloadBranchNodes() {

        }

        protected void initIcons() {

                RECENT_ICON = manager.getIconTheme().get("recent");

                FAVOURITE_ICON = manager.getIconTheme().get("favourite");

                FOLDER_OPEN_ICON = manager.getIconTheme().get("folder_open");

                FOLDER_CLOSED_ICON = manager.getIconTheme().get("folder_closed");

                CRS_ICON = manager.getIconTheme().get("crs");

                LOADING_ICON = manager.getIconTheme().get("spinner");

        }

        protected void initTree() {

                root = new BranchNode(null, this, "root");

                model = new DefaultTreeModel(root, false);

                JTree tree = this.treeResults;

                tree.getSelectionModel().setSelectionMode

                (TreeSelectionModel.SINGLE_TREE_SELECTION);

                listenerSupport = ToolsSwingLocator.getToolsSwingManager().createActionListenerSupport();

                tree.addTreeSelectionListener(this);

                tree.setCellRenderer(new CellRenderer());

                recent = new RecentCRSs(root, manager.getTranslation("Recent"));

                recent.loadChildren();

                root.add(recent);

                BranchNode epsg = new BranchNode(root, this, manager.getTranslation("EPSG"));

                root.add(epsg);

                CRSSearchFilter projectedCRSs = new CRSSearchFilter();

                projectedCRSs.setAuthority("EPSG");

                projectedCRSs.setIncludedTypes(new CRSType[]{CRSType.ProjectedCRSType});

                epsg.add(new AuthorityLoadedGroup(epsg, projectedCRSs, manager.getTranslation("Projected")));

                CRSSearchFilter geographicCRSs = new CRSSearchFilter();

                geographicCRSs.setAuthority("EPSG");

                geographicCRSs.setIncludedTypes(new CRSType[]{CRSType.GeographicCRSType});

                epsg.add(new AuthorityLoadedGroup(epsg, geographicCRSs, manager.getTranslation("Geographic")));

                CRSSearchFilter otherCRSs = new CRSSearchFilter();

                otherCRSs.setAuthority("EPSG");

                otherCRSs.setExcludedTypes(new CRSType[]{CRSType.ProjectedCRSType, CRSType.GeographicCRSType});

                epsg.add(new AuthorityLoadedGroup(epsg, otherCRSs, manager.getTranslation("Other")));

                tree.setModel(model);

                tree.expandPath(new TreePath(model.getPathToRoot(epsg)));

        }

        @Override

        public void valueChanged(TreeSelectionEvent e) {

                TreeNode selection = (TreeNode) this.treeResults.getLastSelectedPathComponent();

                if (selection==null) {

                        return;

                }

                if (selection instanceof CrsTreeNode) {

                        CRSDefinition def = ((CrsTreeNode)selection).getCRS();

                        try {

                                selectedCrs = manager.getCRSManager().getCoordinateReferenceSystem(def);

                                this.jtxt_description.setText(def.getDescription());

                                this.txtCrsWkt.setText(def.toWKT());

                                listenerSupport.fireActionEvent(new ActionEvent(selection, ActionEvent.ACTION_FIRST, CRS_SELECTED_ACTION_COMMAND));

                        } catch (UnsupportedCoordinateReferenceSystemException | ParseException e1) {

                                logger.debug("CRS not supported", e);

                        }                        

                }

        }

        @Override

        public JComponent asJComponent() {

                return this;

        }

        @Override

        public CoordinateReferenceSystem getCoordinateReferenceSystem() {

                return selectedCrs;

        }

        @Override

        public void setCoordinateReferenceSystem(CoordinateReferenceSystem crs) {

                selectedCrs = crs;

                // add the provided node to recent nodes and select it

                CrsTreeNode node = new CrsTreeNode(crs.getDefinition(), recent);

                recent.add(node);

                model.reload(recent);

                this.treeResults.setSelectionPath(new TreePath(model.getPathToRoot(node)));

        }

        @Override

        public void setCustomGroupLabel(String label) {

                // TODO Auto-generated method stub

        }

        @Override

        public void addCustomCRS(CRSDefinition customCRS) {

                // TODO Auto-generated method stub

        }

        @Override

        public void addCRSSpatialFilter(String label, double minLon, double maxLon, double minLat, double maxLat) {

                // TODO Auto-generated method stub

        }

        @Override

        public void addCRSSelectionListener(ActionListener listener) {

                // TODO Auto-generated method stub

        }

        @Override

        public void enableSpatialFilter(String label) {

                // TODO Auto-generated method stub

        }

        @Override

        public void enableAlphanumericFilter(String searchString) {

                cboFilterAlpha.addItem(searchString);

                cboFilterAlpha.setSelectedItem(searchString);

        }

        protected String getAlphanumericFilter() {

                return (String) cboFilterAlpha.getSelectedItem();

        }

        public class RecentCRSs extends BranchNode {

                public RecentCRSs(TreeNode parent, String message) {

                        super(parent, CrsSelectorComponent.this, message);

                }                

                @Override

                public Icon getIcon(boolean expanded) {

                        return RECENT_ICON;

                }

                @Override

                public void reloadChildren() {

                        super.children.clear();

                        // FIXME: maybe history should also use CRSDefinitions

                        String filter;

                        if (CrsSelectorComponent.this.getAlphanumericFilter()!=null) {

                                filter = CrsSelectorComponent.this.getAlphanumericFilter().toLowerCase();

                        }

                        else {

                                filter = null;

                        }

                        for (CoordinateReferenceSystem crs: manager.getCoordinateReferenceSystemHistory()) {

                                if (filter==null || crs.getDefinition().getName().toLowerCase().contains(filter)) {

                                        super.children.add(new CrsTreeNode(crs.getDefinition(), this));

                                }

                        }

                }

        }

        public class AuthorityLoadedGroup extends AsyncLoadedGroup {

                protected final CRSSearchFilter filter;

                public AuthorityLoadedGroup(TreeNode parent, CRSSearchFilter filter, String message) {

                        super(parent, CrsSelectorComponent.this, message);

                        this.filter = filter;

                }

                protected GroupLoader doCreateLoader() {

                        this.filter.setTextFilter(CrsSelectorComponent.this.getAlphanumericFilter());

                        // FIXME: set also spatial filter when supported by API and UI

                        this.filter.setSpatialFilter(null);

                        return new CrsLoader(filter, this, CrsSelectorComponent.this.manager.getCatalogManager());

                }

                @Override

                public void loaded(ArrayList<TreeNode> loadedData) {

                        this.children.clear();

                        this.children.addAll(loadedData);

                        CrsSelectorComponent.this.model.reload(this);

                }

                @Override

                protected void cancelled() {

                        // TODO Auto-generated method stub

                }

        }

        public class CellRenderer extends DefaultTreeCellRenderer {

                @Override

                public java.awt.Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded,

                                boolean leaf, int row, boolean hasFocus) {

                        java.awt.Component c = super.getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, hasFocus);

                        if (value instanceof BranchNode) {

                                this.setIcon(((BranchNode)value).getIcon(expanded));

                        }

                        else if (value instanceof CrsTreeNode) {

                                this.setIcon(CRS_ICON);

                        }

                        else if (value instanceof LeafNode) {

                                ImageIcon icon = ((LeafNode)value).getIcon();

                                if (icon != null) {

                                        this.setIcon(icon);

                                }

                        }

                        return this;

                }

        }

        public DefaultCoordinateReferenceSystemSwingManager getManager() {

                return manager;

        }

        public DefaultTreeModel getModel() {

                return this.model;

        }

        public LoadingNode createLoadingNode(BranchNode parent) {

                return new LoadingNode(parent,

                                this,

                                this.manager.getTranslation("Loading..."));

        }

        @Override

        public JTree getTree() {

                return this.treeResults;

        }

}