Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.geoprocess.lib.sextante.core;

import java.awt.Component;

import java.util.List;

import javax.swing.BorderFactory;

import javax.swing.JOptionPane;

import javax.swing.JScrollPane;

import javax.swing.JTextPane;

import javax.swing.ScrollPaneConstants;

import javax.swing.SwingUtilities;

import javax.swing.border.BevelBorder;

import es.unex.sextante.core.GeoAlgorithm;

import es.unex.sextante.core.ObjectAndDescription;

import es.unex.sextante.core.OutputObjectsSet;

import es.unex.sextante.core.ParametersSet;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.ILayer;

import es.unex.sextante.dataObjects.IRasterLayer;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.gui.additionalResults.AdditionalResults;

import es.unex.sextante.gui.algorithm.iterative.SingleFeatureVectorLayer;

import es.unex.sextante.gui.core.SextanteGUI;

import es.unex.sextante.gui.settings.SextanteGeneralSettings;

import es.unex.sextante.outputs.FileOutputChannel;

import es.unex.sextante.outputs.IOutputChannel;

import es.unex.sextante.outputs.NullOutputChannel;

import es.unex.sextante.outputs.Output;

import es.unex.sextante.outputs.Output3DRasterLayer;

import es.unex.sextante.outputs.OutputRasterLayer;

import es.unex.sextante.outputs.OutputTable;

import es.unex.sextante.outputs.OutputText;

import es.unex.sextante.outputs.OutputVectorLayer;

import es.unex.sextante.outputs.OverwriteOutputChannel;

import es.unex.sextante.parameters.Parameter;

import es.unex.sextante.parameters.RasterLayerAndBand;

import org.cresques.cts.IProjection;

import org.gvsig.andami.PluginServices;

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

import org.gvsig.app.ApplicationLocator;

import org.gvsig.app.extension.ProjectExtension;

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.table.TableDocument;

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.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.datastruct.NoData;

import org.gvsig.fmap.mapcontext.MapContext;

import org.gvsig.fmap.mapcontext.exceptions.ReloadLayerException;

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

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

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

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

import org.gvsig.geoprocess.lib.sextante.dataObjects.FileTools;

import org.gvsig.raster.fmap.layers.FLyrRaster;

public class PostProcessTaskRunnable implements Runnable {

    private MapContext m_MapContext;

    private final GeoAlgorithm m_Algorithm;

    private final OutputObjectsSet m_Output;

    private final boolean m_bShowResultsWindows;

    public PostProcessTaskRunnable(final GeoAlgorithm algorithm,

        final boolean bShowResultsWindow) {

        m_Output = algorithm.getOutputObjects();

        m_Algorithm = algorithm;

        m_bShowResultsWindows = bShowResultsWindow;

    }

    public void run() {

        if (m_Output.hasLayers()) {

            setOutputView();

        }

        addResults();

    }

    private void setOutputView() {

        ViewDocument view = null;

        final ParametersSet parameters = m_Algorithm.getParameters();

        for (int i = 0; i < parameters.getNumberOfParameters(); i++) {

            final Parameter param = parameters.getParameter(i);

            final Object object = param.getParameterValueAsObject();

            if (object instanceof ILayer) {

                view = getViewFromLayer((ILayer) object);

                m_MapContext = view.getMapContext();

                return;

            } else

                if (object instanceof List) {

                    final List list = (List) object;

                    for (int j = 0; j < list.size(); j++) {

                        final Object obj = list.get(j);

                        if (obj instanceof ILayer) {

                            view = getViewFromLayer((ILayer) obj);

                            m_MapContext = view.getMapContext();

                            return;

                        } else

                            if (obj instanceof RasterLayerAndBand) {

                                final RasterLayerAndBand rlab =

                                    (RasterLayerAndBand) obj;

                                view = getViewFromLayer(rlab.getRasterLayer());

                                m_MapContext = view.getMapContext();

                                return;

                            }

                    }

                }

        }

        // if there is no input view, ask the user

        final Project project =

            ((ProjectExtension) PluginServices

                .getExtension(ProjectExtension.class)).getProject();

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

        final Object[] options = new Object[views.size() + 1];

        options[0] = Sextante.getText("Create_new_view");

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

            options[i + 1] = views.get(i);

        }

        final Object selectedObject =

            JOptionPane.showInputDialog(null,

                Sextante.getText("Select_output_view"),

                Sextante.getText("Output_view"), JOptionPane.PLAIN_MESSAGE,

                null, options, null);

        if (selectedObject instanceof IView) {

            m_MapContext =

                ((IView) selectedObject).getMapControl().getMapContext();

        } else {

            final Document newView =

                ProjectManager.getInstance().createDocument(

                    ViewManager.TYPENAME,

                    "SEXTANTE (" + m_Algorithm.getName() + ")");

            ((ProjectExtension) PluginServices

                .getExtension(ProjectExtension.class)).getProject()

                .add(newView);

            final IWindow window = newView.getFactory().getMainWindow(newView);

            final Runnable doWorkRunnable = new Runnable() {

                public void run() {

                    PluginServices.getMDIManager().addWindow(window);

                    m_MapContext = ((ViewDocument) newView).getMapContext();

                }

            };

            try {

                SwingUtilities.invokeAndWait(doWorkRunnable);

            } catch (final Exception e) {

                Sextante.addErrorToLog(e);

            }

        }

    }

    private ViewDocument getViewFromLayer(ILayer layer) {

        if (layer instanceof SingleFeatureVectorLayer) {

            layer = ((SingleFeatureVectorLayer) layer).getOriginalLayer();

        }

        final FLayer gvSIGBaseLayer = (FLayer) layer.getBaseDataObject();

        final Project project =

            ((ProjectExtension) PluginServices

                .getExtension(ProjectExtension.class)).getProject();

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

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

            final ViewDocument viewDoc = (ViewDocument) docs.get(i);

            final FLayers layers = viewDoc.getMapContext().getLayers();

            final LayersIterator iter = new LayersIterator(layers);

            while (iter.hasNext()) {

                final FLayer gvSIGLayer = iter.nextLayer();

                if (gvSIGLayer.equals(gvSIGBaseLayer)) {

                    return viewDoc;

                }

            }

        }

        return null;

    }

    private void addResults() {

        final boolean bUseInternalNames =

            new Boolean(

                SextanteGUI

                    .getSettingParameterValue(SextanteGeneralSettings.USE_INTERNAL_NAMES))

                .booleanValue();

        final boolean bModiFyResultsNames =

            new Boolean(

                SextanteGUI

                    .getSettingParameterValue(SextanteGeneralSettings.MODIFY_NAMES))

                .booleanValue();

        String sDescription;

        boolean bInvalidate = false;

        boolean bShowAdditionalPanel = false;

        if (m_MapContext != null) {

            m_MapContext.beginAtomicEvent();

        }

        for (int i = 0; i < m_Output.getOutputObjectsCount(); i++) {

            final Output out = m_Output.getOutput(i);

            sDescription = out.getDescription();

            final IOutputChannel channel = out.getOutputChannel();

            final Object object = out.getOutputObject();

            if ((out instanceof OutputRasterLayer)

                || (out instanceof Output3DRasterLayer)

                || (out instanceof OutputTable)

                || (out instanceof OutputVectorLayer)) {

                if (bUseInternalNames) {

                    sDescription = out.getName();

                } else

                    if (bModiFyResultsNames) {

                        sDescription =

                            SextanteGUI.modifyResultName(sDescription);

                    }

                if ((channel instanceof NullOutputChannel) || (channel == null)) {

                    continue;

                }

            }

            if (out instanceof OutputVectorLayer) {

                String sFilename = null;

                if (channel instanceof FileOutputChannel) {

                    sFilename = ((FileOutputChannel) channel).getFilename();

                    final FLyrVect flayer =

                        (FLyrVect) FileTools.openLayer(sFilename, sDescription,

                            (IProjection) m_Algorithm.getOutputCRS());

                    if (flayer != null) {

                        flayer.setName(sDescription);

                        m_MapContext.getLayers().addLayer(flayer);

                        bInvalidate = true;

                    }

                } else

                    if (channel instanceof OverwriteOutputChannel) {

                        // TODO:add support for non file based layer

                        final FLyrVect flayer =

                            (FLyrVect) ((OverwriteOutputChannel) channel)

                                .getLayer().getBaseDataObject();

                        try {

                            flayer.reload();

                            bInvalidate = true;

                        } catch (final ReloadLayerException e) {

                        }

                    }

                if (object != null) {

                    ((IVectorLayer) object).close();

                }

            } else

                if (out instanceof OutputTable) {

                    try {

                        final String sFilename =

                            ((FileOutputChannel) channel).getFilename();

                        final TableDocument table =

                            FileTools.openTable(sFilename, sDescription);

                        if (table != null) {

                            table.setName(sDescription);

                            ApplicationLocator.getManager().getProjectManager()

                                .getCurrentProject().add(table);

                            // final JScrollPane jScrollPane =

                            // TableTools

                            // .getScrollableTablePanelFromITable((ITable)

                            // object);

                            // AdditionalResults

                            // .addComponent(new ObjectAndDescription(

                            // sDescription, jScrollPane));

                            // bShowAdditionalPanel = true;

                        }

                    } catch (final Exception e) {

                        Sextante.addErrorToLog(e);

                    }

                } else

                    if (out instanceof OutputRasterLayer) {

                        final IRasterLayer rasterLayer = (IRasterLayer) object;

                        if (channel instanceof FileOutputChannel) {

                            final String sFilename =

                                ((FileOutputChannel) channel).getFilename();

                            final FLyrRaster flayer =

                                (FLyrRaster) FileTools.openLayer(

                                    sFilename,

                                    sDescription,

                                    (IProjection) m_Algorithm.getOutputCRS());

                            if (flayer != null) {

                                if (rasterLayer != null) {

                                    try {

                                        NoData nodata =

                                            RasterLocator

                                                .getManager()

                                                .getDataStructFactory()

                                                .createNoData(

                                                    rasterLayer

                                                        .getNoDataValue(),

                                                    null, sFilename);

                                        flayer.setNoDataValue(nodata);

                                        nodata.save();

                                        rasterLayer.close();

                                    } catch (final Exception e) {

                                        Sextante.addErrorToLog(e);

                                    }

                                }

                                flayer.setName(sDescription);

                                m_MapContext.getLayers().addLayer(flayer);

                                bInvalidate = true;

                            }

                        }

                    } else

                        if (out instanceof OutputText) {

                            JTextPane jTextPane;

                            JScrollPane jScrollPane;

                            jTextPane = new JTextPane();

                            jTextPane.setEditable(false);

                            jTextPane.setContentType("text/html");

                            jTextPane.setText((String) object);

                            jScrollPane = new JScrollPane();

                            jScrollPane.setViewportView(jTextPane);

                            jScrollPane

                                .setHorizontalScrollBarPolicy(ScrollPaneConstants.HORIZONTAL_SCROLLBAR_ALWAYS);

                            jTextPane.setBorder(BorderFactory

                                .createEtchedBorder(BevelBorder.LOWERED));

                            AdditionalResults

                                .addComponent(new ObjectAndDescription(

                                    sDescription, jScrollPane));

                            bShowAdditionalPanel = true;

                        } else

                            if (object instanceof Component) {

                                AdditionalResults

                                    .addComponent(new ObjectAndDescription(

                                        sDescription, object));

                                bShowAdditionalPanel = true;

                            } else

                                if (out instanceof Output3DRasterLayer) {

                                    JOptionPane.showMessageDialog(

                                        SextanteGUI.getMainFrame(),

                                        Sextante.getText("3d_not_supported"),

                                        Sextante.getText("Warning"),

                                        JOptionPane.WARNING_MESSAGE);

                                }

        }

        if (m_MapContext != null) {

            m_MapContext.endAtomicEvent();

        }

        if (bInvalidate) {

            m_MapContext.invalidate();

        }

        if (bShowAdditionalPanel && m_bShowResultsWindows) {

            AdditionalResults.showPanel();

        }

    }

}