Application: gvSIG desktop » gvSIG raster

package org.gvsig.raster.projecttemplate.algorithm;

import java.util.HashMap;

import javax.swing.SwingUtilities;

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.InvalidSetViewException;

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

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.RasterProcess;

/**

 * Process

 */

public class ProjectTemplateProcess extends RasterProcess {

        public static String      RASTER_STORE1     = "RasterStore1";

        public static String      BUFFER            = "RasterStore1";

        public static String      PATH              = "Path";

        public static String      FILENAME          = "FileName";

        public static String      TIME              = "Time";

        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 long              millis            = 0;

        private boolean           export            = true;

        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() {

                RASTER_STORE1 = RasterBaseAlgorithmLibrary.registerInputParameter(RASTER_STORE1, RasterDataStore.class);

                PATH = RasterBaseAlgorithmLibrary.registerInputParameter(PATH, String.class);

                EXPORT = RasterBaseAlgorithmLibrary.registerInputParameter(EXPORT, Boolean.class);

                TEST_EXTENT = RasterBaseAlgorithmLibrary.registerInputParameter(TEST_EXTENT, Extent.class);

                TEST_WIDTH = RasterBaseAlgorithmLibrary.registerInputParameter(TEST_WIDTH, Integer.class);

                TEST_HEIGHT = RasterBaseAlgorithmLibrary.registerInputParameter(TEST_HEIGHT, Integer.class);

                FILENAME = RasterBaseAlgorithmLibrary.registerOutputParameter(FILENAME, String.class);

                TIME = RasterBaseAlgorithmLibrary.registerOutputParameter(TIME, Long.class);

        }

        public void init() {

                store = getParam(RASTER_STORE1) != null ? (RasterDataStore)getParam(RASTER_STORE1) : 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 {

                long t1 = new java.util.Date().getTime();

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

                try {

                        if (store == null)

                                throw new ProjectTemplateException("...");

                        Extent windowExtent = store.getExtent();

                        int w = (int)store.getWidth();

                        int h = (int)store.getHeight();

                        double cellSize = store.getCellSize();

                        if(testExtent != null) {

                                windowExtent = testExtent;

                                w = testWidth;

                                h = testHeight;

                                cellSize = testExtent.width() / w;

                        }

                        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());

                        }*/

                        long t2 = new java.util.Date().getTime();

                        millis = t2 - t1;

                        SwingUtilities.invokeLater(new Runnable() {

                                public void run() {

                                        if (externalActions != null) {

                                                externalActions.end(getResult());

                                        }

                                }

                        });

                } catch (ProjectTemplateException e) {

                        if (incrementableTask != null)

                                incrementableTask.processFinalize();

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

                }

        }

        public Object getResult() {

                HashMap<String, Object> map = new HashMap<String, Object>();

                map.put(FILENAME, filename);

                map.put(TIME, new Long(millis));

                map.put(BUFFER, bufferForTest);

                return map;

        }

        public int getPercent() {

                return percent;

        }

        public String getTitle() {

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

        }

}