gvSIG projects pool

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.vcsgis.app;

import java.io.File;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import java.util.Set;

import javax.swing.ComboBoxModel;

import javax.swing.DefaultComboBoxModel;

import javax.swing.DefaultListModel;

import javax.swing.ListModel;

import javax.swing.tree.TreeModel;

import org.apache.commons.lang.StringUtils;

import org.gvsig.andami.PluginsLocator;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

import org.gvsig.app.project.Project;

import org.gvsig.app.project.ProjectManager;

import org.gvsig.app.project.documents.Document;

import org.gvsig.app.project.documents.DocumentManager;

import org.gvsig.app.project.documents.table.TableDocument;

import org.gvsig.app.project.documents.table.TableManager;

import org.gvsig.app.project.documents.view.ViewDocument;

import org.gvsig.app.project.documents.view.ViewManager;

import org.gvsig.app.project.documents.view.gui.IView;

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

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.aggregate.MultiCurve;

import org.gvsig.fmap.geom.aggregate.MultiPoint;

import org.gvsig.fmap.geom.aggregate.MultiSurface;

import org.gvsig.fmap.geom.primitive.Curve;

import org.gvsig.fmap.geom.primitive.Point;

import org.gvsig.fmap.geom.primitive.Surface;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.fmap.mapcontext.layers.vectorial.GraphicLayer;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol_v2;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.util.LabeledValue;

import org.gvsig.tools.util.LabeledValueImpl;

import org.gvsig.vcsgis.lib.VCSGisLocator;

import org.gvsig.vcsgis.lib.VCSGisManager;

import org.gvsig.vcsgis.lib.VCSGisRuntimeException;

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

import org.gvsig.vcsgis.swing.VCSGisSwingServices;

import org.slf4j.LoggerFactory;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class VCSGisSwingServicesImpl implements VCSGisSwingServices {

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

    private Map<Integer,ISymbol> highlightedPolygonSymbols;

    private Map<Integer,ISymbol> highlightedLineSymbols;

    private Map<Integer,ISymbol> highlightedPointSymbols;

    @Override

    public TreeModel getFeatureStoresTreeModel() {

        return new FeatureStoresTreeModel();

    }

    @Override

    public ListModel getFeatureStoresListModel() {

        DefaultListModel<FeatureStore> model = new DefaultListModel();

        ApplicationManager appManager = ApplicationLocator.getApplicationManager();

        Project project = appManager.getCurrentProject();

        List<Document> tables = project.getDocuments(TableManager.TYPENAME);

        for (Document document : tables) {

            TableDocument table = (TableDocument) document;

            model.addElement(table.getFeatureStore());

        }

        return model;

    }

    @Override

    public void addTableToProject(VCSGisWorkspace ws, FeatureStore store) {

        ApplicationManager appManager = ApplicationLocator.getApplicationManager();

        DocumentManager tableManager = appManager.getProjectManager().getDocumentManager(

                TableManager.TYPENAME

        );

        Project project = appManager.getCurrentProject();

        TableDocument tableDoc = (TableDocument) tableManager.createDocument();

        tableDoc.setName(store.getName());

        tableDoc.setStore(store);

        project.addDocument(tableDoc);

    }

    @Override

    public ComboBoxModel getViewDocumentsComboBoxModel() {

        I18nManager i18n = ToolsLocator.getI18nManager();

        DefaultComboBoxModel<LabeledValue> model = new DefaultComboBoxModel();

        model.addElement(new LabeledValueImpl(i18n.getTranslation("_Select_a_view"), null));

        ApplicationManager appManager = ApplicationLocator.getApplicationManager();

        Project project = appManager.getCurrentProject();

        List<Document> views = project.getDocuments(ViewManager.TYPENAME);

        for (Document document : views) {

            ViewDocument view = (ViewDocument) document;

            model.addElement(new LabeledValueImpl(view.getName(), view));

        }

        return model;

    }

    @Override

    public void addLayerToView(FeatureStore store, LabeledValue labeledView) {

        addLayerToView(store, labeledView, null);

    }

    @Override

    public void addLayerToView(FeatureStore store, LabeledValue labeledView, String name) {

        if (labeledView.getValue() == null){

            return;

        }

        ViewDocument view = (ViewDocument) labeledView.getValue();

        addLayerToView(store, view, name);

    }

    private void addLayerToView(FeatureStore store, ViewDocument view, String name) {

        try {

            String layerName = name;

            if(StringUtils.isBlank(layerName)){

                layerName = store.getName();

            }

            FLayer layer = MapContextLocator.getMapContextManager().createLayer(layerName, store);

            view.getMapContext().getLayers().add(layer);

        } catch (Exception ex) {

            VCSGisManager manager = VCSGisLocator.getManager();

            throw new VCSGisRuntimeException(

                    VCSGisManager.ERR_CANT_ADD_LAYER, 

                    manager.getErrorMessage(VCSGisManager.ERR_CANT_ADD_LAYER),

                    ex);

        }

    }

    @Override

    public void highlight(int mode, Geometry geom) {

        highlight(mode, geom, null);

    }

    @Override

    public void highlight(int mode, Geometry geom, FeatureStore store) {

        if (this.highlightedPointSymbols == null) {

            this.highlightedPointSymbols = new HashMap<>();

            this.highlightedLineSymbols = new HashMap<>();

            this.highlightedPolygonSymbols = new HashMap<>();

            try {

                File pluginfolder = PluginsLocator.getManager().getPlugin(this).getPluginDirectory();

                File folder = new File(pluginfolder, "symbols");

                ISymbol[] symbols = MapContextLocator.getSymbolManager().loadSymbols(folder);

                for (ISymbol symbol : symbols) {

                    if (symbol instanceof ISymbol_v2) {

                        String symbolid = ((ISymbol_v2) symbol).getID();

                        switch(symbolid) {

                            case "vcsgis-repository-polygon":

                                this.highlightedPolygonSymbols.put(HIGHLIGHT_REPOSITORY, symbol);

                                break;

                            case "vcsgis-repository-line":

                                this.highlightedLineSymbols.put(HIGHLIGHT_REPOSITORY, symbol);

                                break;

                            case "vcsgis-repository-point":

                                this.highlightedPointSymbols.put(HIGHLIGHT_REPOSITORY, symbol);

                                break;

                            case "vcsgis-workspace-polygon":

                                this.highlightedPolygonSymbols.put(HIGHLIGHT_WORKSPACE, symbol);

                                break;

                            case "vcsgis-workspace-line":

                                this.highlightedLineSymbols.put(HIGHLIGHT_WORKSPACE, symbol);

                                break;

                            case "vcsgis-workspace-point":

                                this.highlightedPointSymbols.put(HIGHLIGHT_WORKSPACE, symbol);

                                break;

                            case "vcsgis-workspace-previous-polygon":

                                this.highlightedPolygonSymbols.put(HIGHLIGHT_WORKSPACE_PREVIOUS, symbol);

                                break;

                            case "vcsgis-workspace-previous-line":

                                this.highlightedLineSymbols.put(HIGHLIGHT_WORKSPACE_PREVIOUS, symbol);

                                break;

                            case "vcsgis-workspace-previous-point":

                                this.highlightedPointSymbols.put(HIGHLIGHT_WORKSPACE_PREVIOUS, symbol);

                                break;

                        }

                    }

                }

            } catch (Exception ex) {

            }

        }

        List<ViewDocument> viewList = null;

        if(store == null){

            ApplicationManager application = ApplicationLocator.getManager();

            ViewDocument viewdoc = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

            viewList = new ArrayList<>();

            viewList.add(viewdoc);

        } else {

            viewList = getViewDocumentsHavingAStore(store);

        }

        for (ViewDocument viewDoc : viewList) {

            MapContext mapContext = viewDoc.getMapContext();

            GraphicLayer gl = mapContext.getGraphicsLayer();

            if (geom != null) {

                ISymbol symbol = null;

                if (geom instanceof Point || geom instanceof MultiPoint) {

                    symbol = this.highlightedPointSymbols.get(mode);

                } else if (geom instanceof Curve || geom instanceof MultiCurve) {

                    symbol = this.highlightedLineSymbols.get(mode);

                } else if (geom instanceof Surface || geom instanceof MultiSurface) {

                    symbol = this.highlightedPolygonSymbols.get(mode);

                }

                if (symbol != null) {

                    int symbolid = gl.getSymbolId(symbol);

                    if (symbolid < 0) {

                        gl.addSymbol(symbol);

                        symbolid = gl.getSymbolId(symbol);

                    }

                    gl.addGraphic("vcsgis-highlighted", geom, symbolid);

                }

            }

            mapContext.invalidate();

        }

    }

    @Override

    public void centerActiveViewToGeometry(Geometry geometry) {

        if(geometry != null){

            IView view = getActiveView();

            if(view != null){

                ViewDocument viewDocument = view.getViewDocument();

                viewDocument.center(geometry.getEnvelope());

            }

        }

    }

    @Override

    public void centerViewsHavingAStoreToGeometry(FeatureStore store, Geometry geometry) {

        if(geometry != null){

            List<ViewDocument> views = getViewDocumentsHavingAStore(store);

            for (ViewDocument view : views) {

                if(view != null){

                    view.center(geometry.getEnvelope());

                }

            }

        }

    }

    @Override

    public void zoomActiveViewToGeometry(Geometry geometry) {

        if(geometry != null){

            IView view = getActiveView();

            if(view != null){

                ViewDocument viewDocument = view.getViewDocument();

                viewDocument.getMapContext().getViewPort().setEnvelope(geometry.getEnvelope());

            }

        }

    }

    @Override

    public void zoomViewsHavingAStoreToGeometry(FeatureStore store, Geometry geometry) {

        if(geometry != null){

            List<ViewDocument> views = getViewDocumentsHavingAStore(store);

            for (ViewDocument view : views) {

                if(view != null){

                    view.getMapContext().getViewPort().setEnvelope(geometry.getEnvelope());

                }

            }

        }

    }

    private IView getActiveView() {

        ApplicationManager application = ApplicationLocator.getManager();

        IView view = (IView) application.getActiveComponent(ViewDocument.class);

        return view;

    }

    @Override

    public void cleanHighligthed() {

        ApplicationManager application = ApplicationLocator.getManager();

        ProjectManager projectManager = application.getProjectManager();

        Project project = projectManager.getCurrentProject();

        List<Document> views = project.getDocuments(ViewManager.TYPENAME);

        for (Document doc : views) {

            ViewDocument viewdoc = (ViewDocument)doc;

            MapContext mapContext = viewdoc.getMapContext();

            GraphicLayer gl = mapContext.getGraphicsLayer();

            //FIXME: Refrescar el mapContext solo cuando se ha borrado el graphics  

            if(gl.removeGraphics("vcsgis-highlighted")) {

                mapContext.invalidate();

            }

        }

    }

    @Override

    public void cleanActiveViewHighligthed() {

        ApplicationManager application = ApplicationLocator.getManager();

        ViewDocument viewdoc = (ViewDocument) application.getActiveDocument(ViewManager.TYPENAME);

        if(viewdoc != null){

            MapContext mapContext = viewdoc.getMapContext();

            GraphicLayer gl = mapContext.getGraphicsLayer();

            gl.removeGraphics("vcsgis-highlighted");

            mapContext.invalidate();

        }

    }

    @Override

    public void refreshDocument(FeatureStore store){

        if(store == null){

            return;

        }

        ApplicationManager application = ApplicationLocator.getManager();

        ProjectManager projectManager = application.getProjectManager();

        Project project = projectManager.getCurrentProject();

        String storeFullName = store.getFullName();

        List<Document> views = project.getDocuments(ViewManager.TYPENAME);

        for (Document doc : views) {

            ViewDocument view = (ViewDocument)doc;

            for (Iterator<FLayer> it = view.deepiterator(); it.hasNext();) {

                FLayer layer = it.next();

                if(layer instanceof FLyrVect){

                    if( StringUtils.equals(((FLyrVect) layer).getFeatureStore().getFullName(), storeFullName)){

                        view.getMapContext().invalidate();

                        break;

                    }

                }

            }

        }

        List<Document> tables = project.getDocuments(TableManager.TYPENAME);

        for (Document doc : tables) {

            TableDocument table = (TableDocument)doc;

            FeatureStore featureStore = table.getStore();

            if( StringUtils.equals(featureStore.getFullName(), storeFullName)){

                try {

                    featureStore.refresh();

                } catch (DataException e) {

                    LOGGER.warn("Error refreshing table", e);

                }

            }

        }

    }

    @Override

    public void refreshDocument(Set<FeatureStore> stores){

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

            return;

        }

        ApplicationManager application = ApplicationLocator.getManager();

        ProjectManager projectManager = application.getProjectManager();

        Project project = projectManager.getCurrentProject();

        Set<ViewDocument> viewsToRefresh = new HashSet<>();

        for (FeatureStore store : stores) {

            String storeFullName = store.getFullName();

            List<Document> views = project.getDocuments(ViewManager.TYPENAME);

            for (Document doc : views) {

                ViewDocument view = (ViewDocument)doc;

                for (Iterator<FLayer> it = view.deepiterator(); it.hasNext();) {

                    FLayer layer = it.next();

                    if(layer instanceof FLyrVect){

                        if( StringUtils.equals(((FLyrVect) layer).getFeatureStore().getFullName(), storeFullName)){

                            viewsToRefresh.add(view);

                            break;

                        }

                    }

                }

            }

            List<Document> tables = project.getDocuments(TableManager.TYPENAME);

            for (Document doc : tables) {

                TableDocument table = (TableDocument)doc;

                FeatureStore featureStore = table.getStore();

                if( StringUtils.equals(featureStore.getFullName(), storeFullName)){

                    try {

                        featureStore.refresh();

                    } catch (DataException e) {

                        LOGGER.warn("Error refreshing table", e);

                    }

                }

            }

        }

        for (ViewDocument viewDocument : viewsToRefresh) {

            viewDocument.getMapContext().invalidate();

        }

    }

    public List<ViewDocument> getViewDocumentsHavingAStore(FeatureStore store){

        if(store == null){

            return null;

        }

        List<ViewDocument> viewList = new ArrayList<>();

        String storeFullName = store.getFullName();

        ApplicationManager application = ApplicationLocator.getManager();

        ProjectManager projectManager = application.getProjectManager();

        Project project = projectManager.getCurrentProject();

        List<Document> views = project.getDocuments(ViewManager.TYPENAME);

        for (Document doc : views) {

            ViewDocument view = (ViewDocument) doc;

            for (Iterator<FLayer> it = view.deepiterator(); it.hasNext();) {

                FLayer layer = it.next();

                if (layer instanceof FLyrVect) {

                    if (StringUtils.equals(

                            ((FLyrVect) layer).getFeatureStore().getFullName(),

                            store.getFullName()

                    )) {

                        viewList.add(view);

                    }

                    break;

                }

            }

        }

        return viewList;

    }

    @Override

    public void addLayerToActiveView(FeatureStore store, String name) {

        if(store != null && StringUtils.isNotBlank(name)){

            IView view = getActiveView();

            if(view != null){

                ViewDocument viewDocument = view.getViewDocument();

                this.addLayerToView(store, viewDocument, name);

            }

        }

    }

    @Override

    public boolean isThereAnyActiveView() {

        return this.getActiveView() != null;

    }

    @Override

    public void addTableToProject(VCSGisWorkspace ws, FeatureStore store, String tableName) {

        ApplicationManager appManager = ApplicationLocator.getApplicationManager();

        DocumentManager tableManager = appManager.getProjectManager().getDocumentManager(

                TableManager.TYPENAME

        );

        Project project = appManager.getCurrentProject();

        TableDocument tableDoc = (TableDocument) tableManager.createDocument();

        tableDoc.setName(tableName);

        tableDoc.setStore(store);

        project.addDocument(tableDoc);

    }

}