Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.projecttemplate.algorithm;

import java.util.HashMap;

import java.util.List;

import javax.swing.SwingUtilities;

import java.awt.geom.Rectangle2D;

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

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

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

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

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

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

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

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

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

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

import org.gvsig.i18n.Messages;

import org.gvsig.raster.algorithm.RasterBaseAlgorithmLibrary;

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

import org.gvsig.raster.roi.ROI;

/**

 * Process 

 */

public class ProjectTemplateProcess extends DataProcess {

        public static String      RASTER_STORE      = "RasterStore";

        public static String      BUFFER            = "RasterStore";

        public static String      PATH              = "Path";

        public static String      FILENAME          = "FileName";

        public static String      EXPORT            = "Export";

        public static String      TEST_EXTENT       = "TestExtent";

        public static String      TEST_WIDTH        = "TestWidth";

        public static String      TEST_HEIGHT       = "TestHeight";

        private RasterDataStore   store             = null;

        private String            filename          = null;

        private boolean           export            = true;

        private List<ROI>         rois              = null;

        private Extent            extentResult      = null;

        private Extent            testExtent        = null;

        private int               testWidth         = 0;

        private int               testHeight        = 0;

        /**

         * This buffer is just to test

         */

        private Buffer            bufferForTest     = null;

        public static void registerParameters() {

                registerInputParameter(RASTER_STORE, RasterDataStore.class, ProjectTemplateAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(PATH, String.class, ProjectTemplateAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(EXPORT, Boolean.class, ProjectTemplateAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(TEST_EXTENT, Extent.class, ProjectTemplateAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(TEST_WIDTH, Integer.class, ProjectTemplateAlgorithmLibrary.PROCESS_LABEL);

                registerInputParameter(TEST_HEIGHT, Integer.class, ProjectTemplateAlgorithmLibrary.PROCESS_LABEL);

                registerOutputParameter(FILENAME, String.class, ProjectTemplateAlgorithmLibrary.PROCESS_LABEL);

        }

        public void init() {

                store = getParam(RASTER_STORE) != null ? (RasterDataStore)getParam(RASTER_STORE) : null;

                filename = getStringParam(PATH);

                export = getBooleanParam(EXPORT);

                testExtent = getParam(TEST_EXTENT) != null ? (Extent)getParam(TEST_EXTENT) : null;

                testWidth = getIntParam(TEST_WIDTH);

                testHeight = getIntParam(TEST_HEIGHT);

        }

        public void process() throws ProcessInterruptedException {

                insertLineLog(Messages.getText("..."));

                try {

                        if (store == null)

                                throw new ProjectTemplateException("...");

                        if(getROIEPSG() != null) {

                                try {

                                        rois = store.getRois(getROIEPSG());

                                } catch (ROIException e2) {

                                        logger.error(Messages.getText("error_getting_rois"), e2);

                                }

                        }

                        try {

                                store = ((RasterDataStore)store).cloneDataStore();

                        } catch (CloneException e) {

                                new ProjectTemplateException("Error cloning the input DataStore", e);

                        }

                        extentResult = getExtentResult(getOutputWindow(), rois, store);

                        Rectangle2D sourcePxBBox = getSourcePxBox(extentResult, store);

                        double cellSize = store.getCellSize();

                        RasterQuery query = RasterLocator.getManager().createQuery();

                        query.setAllDrawableBands();

                        query.setAreaOfInterest(testExtent, testWidth, testHeight);

                        Buffer sourceBuffer = null;

                        try {

                                sourceBuffer = store.query(query);

                                sourceBuffer.setDataExtent(testExtent.toRectangle2D());

                        } catch (RasterDriverException e) {

                                throw new ProjectTemplateException("");

                        } catch (InvalidSetViewException e) {

                                throw new ProjectTemplateException("");

                        }

                        bufferForTest = sourceBuffer;

                        //////////////////////////

                        //TODO:PROCESS!!

                        //int inDataType = store.getDataType()[0];

                        //Buffer buf = RasterLocator.getManager().createBuffer(

                        //                inDataType, 

                        //                (int)store.getWidth(), 

                        //                (int)store.getHeight(), 

                        //                store.getBandCount(), 

                        //                true);

                        //updatePercent(row, buf.getHeight());

                        //////////////////////////

                        /*if(export) {

                                super.exportRaster(filename, 

                                                buf, 

                                                cellSize, 

                                                windowExtent.getULX(), 

                                                windowExtent.getULY());

                        }*/

                        addOutputValue(FILENAME, filename);

                        addOutputValue(BUFFER, bufferForTest);

                } catch (ProjectTemplateException e) {

                        if (incrementableTask != null)

                                incrementableTask.processFinalize();

                        messageBoxError("...", this, e);

                }

        }

        public String getTitle() {

                return Messages.getText("...");

        }

}