Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

* 

* 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.

* 

*/

package org.gvsig.raster.wms.app.wmsclient.wmc;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileWriter;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.andami.PluginServices;

import org.gvsig.andami.messages.NotificationManager;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.andami.plugins.IExtension;

import org.gvsig.andami.ui.mdiManager.IWindow;

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.view.DefaultViewDocument;

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

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

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

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

import org.gvsig.raster.wms.app.wmsclient.gui.panel.WebMapContextSettingsPanel;

import org.gvsig.raster.wms.app.wmsclient.layer.FLyrWMS;

/**

 * Extension to export a view with WMS layers to a OGC's Web Map Context XML

 * file

 *

 * @author jaume dom?nguez faus - jaume.dominguez@iver.es

 * @author laura d?az s?nchez - laura.diaz@iver.es

 */

public class ExportWebMapContextExtension extends Extension {

        private static ArrayList<String>   supportedVersions;

        private AbstractViewPanel          viewToExport;

        private WebMapContextSettingsPanel mc;

        private static IExtension          thisExtension;

        {

                supportedVersions = new ArrayList<String>();

                supportedVersions.add("1.1.0");

        }

        public void initialize() {

                thisExtension = PluginServices

                                .getExtension(ExportWebMapContextExtension.class);

        }

        public void execute(String actionCommand) {

                if (actionCommand.equals("EXPORT")) {

                        // Here we collect the info

                        DefaultViewDocument[] views = getExportableViews();

                        if (views.length <= 0) {

                                return;

                        }

                        mc = new WebMapContextSettingsPanel(views);

                        PluginServices.getMDIManager().addWindow(mc);

                } else if (actionCommand.equals("DO_EXPORT")) {

                        // Here the target file is produced (called from the WebMapContextSettingsPanel)

                        String xml = mc.getXML();

                        File f = mc.getTargetFile();

                        if (xml != null) {

                                createFile(f, xml);

                        }

                }

        }

        /**

         * Takes a File object and its XML contents and stores it as a regular

         * file in the file system.

         * @param f

         * @param xml

         */

        public static void createFile(File f, String xml) {

                if (xml != null) {

                        try {

                                if (!f.exists()) {

                                        f.createNewFile();

                                }

                                BufferedWriter bw = new BufferedWriter(new FileWriter(f));

                                bw.write(xml);

                                bw.close();

                                bw = null;

                        } catch (IOException e) {

                                NotificationManager.addError(PluginServices.getText(

                                                thisExtension, "error_writting_file"), e);

                        }

                }

        }

        public boolean isEnabled() {

                return true;

        }

        public boolean isVisible() {

                // Will be visible if the current project has, at least, one FLyrWMS.

//                Project project = ((ProjectExtension) PluginServices.getExtension(ProjectExtension.class)).getProject();

                final Project project = ProjectManager.getInstance().getCurrentProject();

                if (project == null) {

                        return false;

                }

                IWindow f = PluginServices.getMDIManager().getActiveWindow();

                if (f instanceof AbstractViewPanel) {

                        AbstractViewPanel v = (AbstractViewPanel) f;

                        if (v != null && v  instanceof AbstractViewPanel) {

                                // Check if the active contains WMS layers. If so, this view

                                // will be the one to be exported.

                                FLayers lyrs = v.getMapControl().getMapContext().getLayers();

                                for (int i = 0; i < lyrs.getLayersCount(); i++) {

                                        FLayer lyr = lyrs.getLayer(i);

                                        if (WebMapContext.containsExportableLayers(lyr)) {

                                                viewToExport = v;

                                                return true;

                                        }

                                }

                        }

                }

                // Since the active view does not contain WMS layers then will

                // see what about the others. In this case, no view is set to be

                // the exported one.

                viewToExport = null;

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

                for (int i = 0; i < views.size(); i++) {

                        DefaultViewDocument v = ((DefaultViewDocument) views.get(i));

                        if (v != null) {

                                FLayers lyrs = v.getMapContext().getLayers();

                                for (int j = 0; j < lyrs.getLayersCount(); j++) {

                                        FLayer lyr = lyrs.getLayer(j);

                                        if (WebMapContext.containsExportableLayers(lyr)) {

                                                return true;

                                        }

                                }

                        }

                }

                return false;

        }

        /**

         * <p>

         * Searches the views in the current project that can be exported to a

         * WebMapContext file (with ".cml" extension) and return them in a

         * ProjectView array.<br>

         * </p>

         * <p>

         * A view is exportable to WebMapContext if it contains at least one FLyrWMS

         * and in a near future, any other OGC layer such as WebCoverageService, WFS, and so on. Only

         * these layers will be exported. Other kind of layers are ignored since they

         * are out of the OGC premises.

         * </p>

         * @return

         */

        @SuppressWarnings({ "unchecked", "deprecation" })

        private DefaultViewDocument[] getExportableViews() {

//                Project project = ((ProjectExtension) PluginServices.getExtension(ProjectExtension.class)).getProject();

//                ArrayList views = project.getDocumentsByType(ProjectViewFactory.registerName);

                final Project project = ProjectManager.getInstance().getCurrentProject();

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

                List exportableViews = new ArrayList();

                if (viewToExport!=null) {

                        exportableViews.add(viewToExport.getModel());

                }

                for (int i = 0; i < views.size(); i++) {

                        DefaultViewDocument v = ((DefaultViewDocument) views.get(i));

                        if (v != null) {

                                FLayers lyrs = v.getMapContext().getLayers();

                                for (int j = 0; j < lyrs.getLayersCount(); j++) {

                                        FLayer lyr = lyrs.getLayer(j);

                                        if (lyr instanceof FLyrWMS && !exportableViews.contains(v)) {

                                                exportableViews.add(v);

                                                break;

                                        }

                                }

                        }

                }

                return (DefaultViewDocument[]) exportableViews.toArray(new DefaultViewDocument[0]);

        }

}