Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.multifile.app;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.andami.PluginServices;

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

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

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.fmap.dal.coverage.dataset.Buffer;

import org.gvsig.fmap.dal.coverage.exception.RmfSerializerException;

import org.gvsig.fmap.dal.coverage.store.RasterDataStore;

import org.gvsig.fmap.dal.coverage.store.props.ColorInterpretation;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

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

import org.gvsig.gui.beans.table.exceptions.NotInitializeException;

import org.gvsig.i18n.Messages;

import org.gvsig.raster.algorithm.RasterBaseAlgorithmLibrary;

import org.gvsig.raster.algorithm.process.DataProcess;

import org.gvsig.raster.algorithm.process.IProcessActions;

import org.gvsig.raster.algorithm.process.ProcessException;

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

import org.gvsig.raster.impl.provider.RasterProvider;

import org.gvsig.raster.multifile.app.panel.BandSelectorPanel;

import org.gvsig.raster.multifile.io.MultiFileDataParameters;

import org.gvsig.raster.multifile.io.MultiFileFormat;

import org.gvsig.raster.swing.RasterSwingLibrary;

public class MainDialogActions implements IProcessActions {

        private String                          file                 = null;

        private String                          folder               = null;

        private BandSelectorPanel               panel                = null;

        private AbstractViewPanel               selectedView         = null;

        public MainDialogActions(BandSelectorPanel panel, String file, String folder) {

                this.file = file;

                this.folder = folder;

                this.panel = panel;

        }

        public void buildOneLayer() {

                RasterDataStore mainRasterStore = panel.getResult();

                saveMetadata();

                DataProcess clippingProcess = null;

                try {

                        clippingProcess = RasterBaseAlgorithmLibrary.getManager().createRasterTask("ClippingProcess");

                } catch (ProcessException e) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("error_processing_layer"), panel, e);

                }

                clippingProcess.setActions(this);

                String tit = PluginServices.getMDIManager().getWindowInfo(getView()).getTitle();

                clippingProcess.addParam("viewname", tit);

                clippingProcess.addParam("pixelcoordinates", new int[] { 0, (int) mainRasterStore.getHeight(), (int) mainRasterStore.getWidth(), 0 });

                clippingProcess.addParam("resolution", new int[]{(int) mainRasterStore.getWidth(),

                                                                                                                          (int) mainRasterStore.getHeight()});

                clippingProcess.addParam("suffix", ".tif");

                clippingProcess.addParam("filename", folder + File.separator + file);

                clippingProcess.addParam("layer", mainRasterStore);

                int[] drawableBands = new int[mainRasterStore.getBandCount()];

                for (int i = 0; i < drawableBands.length; i++) {

                        drawableBands[i] = i;

                }

                clippingProcess.addParam("drawablebands", drawableBands);

                clippingProcess.addParam("onelayerperband", new Boolean(false));

                clippingProcess.addParam("interpolationmethod", new Integer(Buffer.INTERPOLATION_NearestNeighbour));

                clippingProcess.addParam("affinetransform", mainRasterStore.getAffineTransform());

                /*if (params != null)

                        RasterToolsUtil.loadWriterParamsFromPropertiesPanel(panelProperty, params);

                clippingProcess.addParam("driverparams", params);*/

                clippingProcess.start();

        }

        public void loadLayer() {

                RasterDataStore mainRasterStore = panel.getResult();

                if(mainRasterStore == null) {

                        return;

                }

                saveMff(mainRasterStore);

                saveMetadata();

                try {

                        loadLayerInView(mainRasterStore);

                } catch (LoadLayerException e) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("error_loading_layer"), panel, e);

                }

        }

        private void saveMff(RasterDataStore mainRasterStore) {

                MultiFileDataParameters params = (MultiFileDataParameters)mainRasterStore.getParameters();

                ArrayList<File> uriList = new ArrayList<File>();

                List<RasterProvider> providers = params.getProviders();

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

                        uriList.add(new File(providers.get(i).getURI()));

                }

                try {

                        MultiFileFormat.saveMultiFileFormat(file, folder, uriList);

                } catch (IOException e) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("error_salvando_mff"), panel, e);

                }

        }

        private void saveMetadata() {

                RasterDataStore mainRasterStore = panel.getResult();

                int rBand = panel.getColorInterpretationByColorBandBand(RasterDataStore.RED_BAND);

                int gBand = panel.getColorInterpretationByColorBandBand(RasterDataStore.GREEN_BAND);

                int bBand = panel.getColorInterpretationByColorBandBand(RasterDataStore.BLUE_BAND);

                int aBand = panel.getColorInterpretationByColorBandBand(RasterDataStore.ALPHA_BAND);

                if (!isCorrectAssignedBand(rBand, gBand, bBand, aBand)) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("combinacion_no_asignable"), panel);

                        return;

                }

                try {

                        ColorInterpretation ci = RasterLocator.getManager().getDataStructFactory().createColorInterpretation(

                                        new String[mainRasterStore.getBandCount()]);

                        // Combinaci?n GRAY

                        if ((rBand == gBand) && (rBand == bBand) && (rBand >= 0)) {

                                for (int iBand = 0; iBand < panel.getARGBTable().getRowCount(); iBand++) {

                                        ci.setColorInterpValue(iBand, ColorInterpretation.UNDEF_BAND);

                                }

                                ci.setColorInterpValue(rBand, ColorInterpretation.GRAY_BAND);

                                ci.setColorInterpValue(aBand, ColorInterpretation.ALPHA_BAND);

                        } else {

                                // Combinaci?n RGB

                                ci.setColorInterpValue(aBand, ColorInterpretation.ALPHA_BAND);

                                ci.setColorInterpValue(rBand, ColorInterpretation.RED_BAND);

                                ci.setColorInterpValue(gBand, ColorInterpretation.GREEN_BAND);

                                ci.setColorInterpValue(bBand, ColorInterpretation.BLUE_BAND);

                        }

                        RasterLocator.getManager().getProviderServices().saveObjectToRmfFile(folder + File.separator + file, ci);

                } catch (RmfSerializerException exc) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("error_salvando_rmf"), panel, exc);

                } catch (NotInitializeException exc) {

                        RasterSwingLibrary.messageBoxError(Messages.getText("table_not_initialize"), panel, exc);

                }

        }

        private void loadLayerInView(RasterDataStore mainRasterStore) throws LoadLayerException {

                MapContextManager mcm = MapContextLocator.getMapContextManager();

                FLyrRaster lyr = (FLyrRaster) mcm.createLayer(file, mainRasterStore);

                lyr.reload();

                getView().getMapControl().getMapContext().beginAtomicEvent();

                getView().getMapControl().getMapContext().getLayers().addLayer(lyr);

                getView().getMapControl().getMapContext().invalidate();

                getView().getMapControl().getMapContext().endAtomicEvent();

        }

        private AbstractViewPanel getView() {

                if(selectedView == null) {

                        IWindow[] wList = PluginServices.getMDIManager().getAllWindows();

                        for (int i = 0; i < wList.length; i++) {

                                if(wList[i] instanceof AbstractViewPanel)

                                        selectedView = (AbstractViewPanel)wList[i];

                        }

                }

                return selectedView;

        }

        private boolean isCorrectAssignedBand(int r, int g, int b, int a) {

                // Si es gris es correcta la asignacion

                if ((r == g) && (r == b) && (r >= 0)) {

                        // Si el alpha esta asignado a la misma banda es incorrecto

                        if (r == a)

                                return false;

                        // En caso contrario es correcto

                        return true;

                }

                // Si dos bandas coinciden, se dice que no es correcta la asignacion

                int list[] = { r, g, b, a };

                for (int i = 0; i <= 3; i++)

                        for (int j = 0; j <= 3; j++)

                                if ((i != j) && (list[i] == list[j]) && (list[i] > -1))

                                        return false;

                return true;

        }

        public void interrupted() {

                // TODO Auto-generated method stub

        }

        public void end(Object param) {

        }

        public void updateProgress(int current, int total) {

                // TODO Auto-generated method stub

        }

}