Application: gvSIG desktop

package org.gvsig.raster.tools.app.mainplugin;

import java.io.File;

import java.io.IOException;

import javax.swing.JOptionPane;

import org.apache.commons.io.FileUtils;

import org.apache.commons.io.FilenameUtils;

import org.apache.commons.lang3.StringUtils;

import org.cresques.cts.ICRSFactory;

import org.cresques.cts.IProjection;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.andami.plugins.Extension;

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.ApplicationManager;

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

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

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.raster.RasterStore;

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

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

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

import org.gvsig.fmap.mapcontext.raster.api.RasterLayer;

import org.gvsig.fmap.mapcontrol.MapControl;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

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

import org.gvsig.tools.swing.api.threadsafedialogs.ThreadSafeDialogsManager;

/**

 *

 * @author fdiaz

 *

 */

public class AssignProjectionExtension extends Extension {

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

    @Override

    public void initialize() {

        IconThemeHelper.registerIcon("action", "setviewprojection-raster-icon", this);

    }

    @Override

    public void execute(String actionCommand) {

        ApplicationManager applicationManager = ApplicationLocator.getManager();

        I18nManager i18nManager = ToolsLocator.getI18nManager();

        if (StringUtils.equalsIgnoreCase(actionCommand, "assign-projection")) {

            IView view = getActiveView();

            final MapControl mapControl = view.getMapControl();

            IProjection projection = mapControl.getProjection();

            RasterStore rasterStore = getFirstFileRasterLayerActive().getRasterStore();

            DataStoreParameters storeParams = rasterStore.getParameters();

            if (storeParams instanceof FilesystemStoreParameters) {

                FilesystemStoreParameters fileSystemStoreParameters = (FilesystemStoreParameters) storeParams;

                File file = fileSystemStoreParameters.getFile();

                File prjFile = getPrjFile(file);

                if (prjFile.exists()) {

                    String message = "_the_content_of_the_file_XFILEX_will_be_replaced_do_you_want_to_continue";

                    int reply =

                        applicationManager.confirmDialog(

                            i18nManager.getTranslation(message, new String[] { prjFile.getAbsolutePath() }),

                            i18nManager.getTranslation("Assign_projection"),

                            JOptionPane.YES_NO_OPTION,

                            JOptionPane.QUESTION_MESSAGE);

                    if (reply != JOptionPane.YES_OPTION) {

                        return;

                    }

                }

                try {

                    savePrjFile(prjFile, projection);

                } catch (IOException e) {

                    logger.warn("Can't save prj file.", e);

                    String message2 = "_cant_save_prj_file";

                    applicationManager.messageDialog(message2, "Assing_projection", JOptionPane.WARNING_MESSAGE);

                }

            }

        }

    }

    private void savePrjFile(File prjFile, IProjection proj) throws IOException {

        String export = proj.export(ICRSFactory.FORMAT_WKT_ESRI);

        if (export != null) {

            FileUtils.writeStringToFile(prjFile, export);

        }

    }

    /**

     * @param dataFile

     * @return

     */

    private File getPrjFile(File dataFile) {

        File file = new File(FilenameUtils.removeExtension(dataFile.getAbsolutePath()) + ".prj");

        return file;

    }

    @Override

    public boolean isEnabled() {

        return (getFirstFileRasterLayerActive() != null);

    }

    @Override

    public boolean isVisible() {

        IView view = getActiveView();

        if (view != null) {

            ViewDocument viewDocument = view.getViewDocument();

            FLayers layers = viewDocument.getMapContext().getLayers();

            for (FLayer layer : layers) {

                if (layer instanceof RasterLayer) {

                    return true;

                }

            }

        }

        return false;

    }

    private IView getActiveView() {

        ApplicationManager application = ApplicationLocator.getManager();

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

        return view;

    }

    private RasterLayer getFirstFileRasterLayerActive() {

        IView view = getActiveView();

        if (view != null) {

            ViewDocument viewDocument = view.getViewDocument();

            FLayers layers = viewDocument.getMapContext().getLayers();

            for (FLayer layer : layers) {

                if (layer instanceof RasterLayer && layer.isActive()) {

                    RasterLayer rasterLayer = (RasterLayer) layer;

                    RasterStore rasterStore = rasterLayer.getRasterStore();

                    DataStoreParameters storeParams = rasterStore.getParameters();

                    if (storeParams instanceof FilesystemStoreParameters) {

                        return rasterLayer;

                    }

                }

            }

        }

        return null;

    }

}