Application: gvSIG desktop » gvSIG raster

package org.gvsig.fmap.mapcontext.raster.impl;

import java.awt.Dimension;

import java.awt.Graphics2D;

import java.awt.image.BufferedImage;

import java.io.File;

import java.io.IOException;

import java.net.URISyntaxException;

import java.net.URL;

import javax.imageio.ImageIO;

import org.cresques.cts.IProjection;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.exception.InitializeException;

import org.gvsig.fmap.dal.exception.ProviderNotRegisteredException;

import org.gvsig.fmap.dal.exception.ReadException;

import org.gvsig.fmap.dal.exception.ValidateDataParametersException;

import org.gvsig.fmap.dal.file.jimi.JimiRasterStoreProvider;

import org.gvsig.fmap.dal.file.jimi.JimiRasterStoreProviderFactory;

import org.gvsig.fmap.dal.file.jimi.JimiRasterStoreProviderParameters;

import org.gvsig.fmap.dal.raster.api.RasterStore;

import org.gvsig.fmap.geom.jts.primitive.Envelope2D;

import org.gvsig.fmap.geom.primitive.Envelope;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.raster.api.RasterLayer;

import org.gvsig.raster.lib.legend.api.ColorInterpretation;

import org.gvsig.raster.lib.legend.api.RasterLegend;

import org.gvsig.raster.lib.legend.api.RasterLegendLocator;

import org.gvsig.raster.lib.legend.api.RasterLegendManager;

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

import org.gvsig.tools.task.Cancellable;

/**

 * @author <a href="mailto:lmarques@disid.com">Lluis Marques</a>

 *

 */

public class DefaultRasterLayerTest extends AbstractLibraryAutoInitTestCase {

    @Override

    protected void doSetUp() throws Exception {

    }

    /**

     * Tests to create a rasterLayer

     */

    public void testCreateLayer() {

        IProjection projection = CRSFactory.getCRS("EPSG:25830");

        final URL url = this.getClass().getResource("/pathcv.png");

        //The store is created

        DataManager dataManager = DALLocator.getDataManager();

        try {

            File file=new File(url.toURI());

            JimiRasterStoreProviderFactory jimiFactory=

                (JimiRasterStoreProviderFactory)dataManager.getStoreProviderFactory(JimiRasterStoreProvider.NAME);

            JimiRasterStoreProviderParameters params = jimiFactory.createJimiParameters();

            params.setFile(file);

            params.setCRS(projection);

            RasterStore store=null;

            store = (RasterStore) dataManager.openStore(JimiRasterStoreProvider.NAME,params);

            RasterLegendManager legendManager=RasterLegendLocator.getRasterLegendManager();

            ColorInterpretation colorInterpretation =

                legendManager.createColorInterpretation(ColorInterpretation.RGB);

            RasterLegend legend= legendManager.createLegend(colorInterpretation);

            RasterLayer layer=new DefaultRasterLayer(store, legend);

            assertNotNull(layer);

            //589 x 467

            //1178 x 934

            //294 x 233

            BufferedImage image = new BufferedImage(589, 467, BufferedImage.TYPE_INT_ARGB);

            ViewPort viewPort = new ViewPort(projection);

            Envelope envelope = new Envelope2D(-100, -100, 1000, 1000);

            viewPort.setEnvelope(envelope);

//            viewPort.setEnvelope(layer.getFullEnvelope());

            viewPort.setImageSize(new Dimension(589, 467));

            viewPort.refreshExtent();

            layer.draw(image, (Graphics2D)image.getGraphics(), viewPort, new CancelDraw(), 1.0);

            try {

                File output = File.createTempFile("test", ".png");

                ImageIO.write(image, "png", output);

            } catch (IOException e) {

                fail("Exception happened: "+e.getMessage());

            }

        } catch (InitializeException | URISyntaxException | ReadException | ValidateDataParametersException | ProviderNotRegisteredException e) {

            fail("Exception happened: "+e.getMessage());

        }

    }

    public class CancelDraw implements Cancellable {

        /**

         * <p>

         * Determines if the drawing task must be canceled or not.

         * </p>

         *

         * @see #isCanceled()

         * @see #setCanceled(boolean)

         */

        private boolean cancel = false;

        /**

         * Creates a new <code>CancelDraw</code> object.

         */

        public CancelDraw() {

        }

        /*

         * (non-Javadoc)

         *

         * @see com.iver.utiles.swing.threads.Cancellable#setCanceled(boolean)

         */

        public void setCanceled(boolean b) {

            cancel = b;

        }

        /*

         * (non-Javadoc)

         *

         * @see com.iver.utiles.swing.threads.Cancellable#isCanceled()

         */

        public boolean isCanceled() {

            return cancel;

        }

    }

}