Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.tools;

import java.util.HashMap;

import java.util.List;

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

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

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

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

import org.gvsig.geoprocess.lib.sextante.AbstractSextanteGeoProcess;

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

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 es.unex.sextante.core.AnalysisExtent;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.IRasterLayer;

import es.unex.sextante.exceptions.GeoAlgorithmExecutionException;

import es.unex.sextante.exceptions.NullParameterAdditionalInfoException;

import es.unex.sextante.exceptions.NullParameterValueException;

import es.unex.sextante.exceptions.RepeatedParameterNameException;

import es.unex.sextante.exceptions.WrongParameterIDException;

import es.unex.sextante.exceptions.WrongParameterTypeException;

/**

 * Process to change a layer of data type

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class LayerDatatypeSextanteAlgorithm extends AbstractSextanteGeoProcess implements IProcessActions {

    public static final String RESULT            = "RESULT";

    public static final String LAYER             = "RasterStore1";

        public static final String DATATYPE          = "Datatype";

        public static final String ADJUST_DEC2INT    = "AdjustDec2Int";

        public static final String ADJUST_BIG2SMALL  = "AdjustBig2Small";

        public static String[]    DEC2INT_OPTIONS    = new String[]{"Trunk", "Round", "Ceil", "Floor"};

        public static String[]    BIG2SMALL_OPTIONS  = new String[]{"Trunk", "Maxvalue", "NoData"};

        public static String[]    DATATYPES          = new String[]{"Byte", "UShort", "Short", "Integer", "Float", "Double"};

    private DataProcess        task            = null;

    public void defineCharacteristics() {

        setName(getTranslation("layer_datatype"));

        setGroup(getTranslation("raster_layer"));

        try {

            m_Parameters.addInputRasterLayer(LAYER, getTranslation("Input_layer"), true);

            m_Parameters.addSelection(DATATYPE, getTranslation("dst_datatype"), DATATYPES);

            m_Parameters.addSelection(ADJUST_DEC2INT, getTranslation("dec_2_int"), DEC2INT_OPTIONS);

            m_Parameters.addSelection(ADJUST_BIG2SMALL, getTranslation("big_2_small"), BIG2SMALL_OPTIONS);

        } catch (RepeatedParameterNameException e) {

            Sextante.addErrorToLog(e);

        }

        addOutputRasterLayer(RESULT, getTranslation("principalcomponents"));

    }

    public boolean processAlgorithm() throws GeoAlgorithmExecutionException {

            if(existsOutPutFile(LayerDatatypeSextanteAlgorithm.RESULT, 0)) {

                    throw new GeoAlgorithmExecutionException(getTranslation("file_exists"));

            }

            IRasterLayer input = m_Parameters.getParameterValueAsRasterLayer(LAYER);

            FLyrRaster lyrRaster = ((FLyrRaster)input.getBaseDataObject());

            IRasterLayer output = null;

            output = getNewRORasterLayer(

                            RESULT, 

                            Sextante.getText("layerdatatype_description"), 

                            input.getDataType(), 

                            input.getBandsCount());

            String fileName = ((FLyrRasterIRasterLayer)output).getFileName(); //getOutPutFile(RESULT);

            try {

                    setProgressText(getTranslation("convert_datatype"));

                        task = createLayerDatatypeProcess(lyrRaster.getDataStore(), fileName);

                        task.execute();

                        HashMap<String, Object> params = task.getResult();

                        fileName = (String)params.get("FileName");

                        ((FLyrRasterIRasterLayer)output).setBaseDataObject(fileName);

                } catch (ProcessInterruptedException e) {

                        Sextante.addErrorToLog(e);

                } catch (ProcessException e) {

                        Sextante.addErrorToLog(e);

                } 

                if(getTaskMonitor().isCanceled())

                        return false;

        return true;

    }

    /**

     * Creates a process to calculate statistics

     * @param inputStore

     * @return

     * @throws ProcessException

     * @throws NullParameterAdditionalInfoException 

     * @throws NullParameterValueException 

     * @throws WrongParameterIDException 

     * @throws WrongParameterTypeException 

     */

    private DataProcess createLayerDatatypeProcess(RasterDataStore inputStore, String fileName) throws ProcessException, WrongParameterTypeException, WrongParameterIDException, NullParameterValueException, NullParameterAdditionalInfoException {

            DataProcess taskStats = RasterBaseAlgorithmLibrary.getManager().createRasterTask("LayerDatatypeProcess");

            taskStats.setActions(this);

            List<String> params = taskStats.getRasterTaskInputParameters("LayerDatatypeProcess");

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

                    String paramName = params.get(i);

                    Class<?> paramType = taskStats.getParameterTypeByProcess("LayerDatatypeProcess", paramName);

                    if(paramType == RasterDataStore.class) {

                            taskStats.addParam(paramName, (RasterDataStore)inputStore);

                    }

                    if(paramName.equals("Path")) {

                            taskStats.addParam(paramName, fileName);

                    }

                    if(paramName.equals(DATATYPE)) {

                            String value = m_Parameters.getParameterValueAsString(DATATYPE);

                            int position = 0;

                            for (int j = 0; j < DATATYPES.length; j++) {

                                        if(DATATYPES[j].equals(value))

                                                position = j;

                                }

                            taskStats.addParam(paramName, position);

                    }

                    if(paramName.equals(ADJUST_DEC2INT)) {

                            String value = m_Parameters.getParameterValueAsString(ADJUST_DEC2INT);

                            int position = 0;

                            for (int j = 0; j < DEC2INT_OPTIONS.length; j++) {

                                        if(DEC2INT_OPTIONS[j].equals(value))

                                                position = j;

                                }

                            taskStats.addParam(paramName, position);

                    }

                    if(paramName.equals(ADJUST_BIG2SMALL)) {

                            String value = m_Parameters.getParameterValueAsString(ADJUST_BIG2SMALL);

                            int position = 0;

                            for (int j = 0; j < BIG2SMALL_OPTIONS.length; j++) {

                                        if(BIG2SMALL_OPTIONS[j].equals(value))

                                                position = j;

                                }

                            taskStats.addParam(paramName, position);

                    }

                    if(paramName.equals("WINDOW")) {

                            AnalysisExtent ext = getAnalysisExtent();

                            Extent bbox = RasterLocator.getManager().getDataStructFactory().createExtent(

                                            ext.getXMin(), ext.getYMax(), ext.getXMax(), ext.getYMin());

                            Extent inputBbox = inputStore.getExtent();

                            if(bbox.getULX() != inputBbox.getULX() || 

                                    bbox.getULY() != inputBbox.getULY() || 

                                    bbox.getLRX() != inputBbox.getLRX() || 

                                    bbox.getLRY() != inputBbox.getLRY()) {

                                    taskStats.addParam(paramName, bbox);

                            }

                    }

            }

            return taskStats;

    }

        public void interrupted() {

        }

        public void end(Object param) {

        }

        public void updateProgress(int current, int total) {

                boolean cancelled = setProgress(current, total);

                if(!cancelled) {

                        if(task != null)

                                task.actionCanceled(null);

                }

        }

    /*

     * TODO: Customized panels

    @Override

    public Class<? extends GeoAlgorithmParametersPanel> getCustomParametersPanelClass() {

        return PrincipalComponentsParametersPanel.class;

    }*/

}