Application: gvSIG desktop » gvSIG data providers based on gdal

package org.gvsig.raster.gdal.provider;

import java.io.File;

import java.net.URL;

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

import org.gvsig.fmap.dal.DataServerExplorerParameters;

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

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

import org.gvsig.fmap.dal.serverexplorer.filesystem.FilesystemStoreParameters;

import org.gvsig.fmap.dal.spi.DALSPILocator;

import org.gvsig.fmap.dal.spi.DataManagerProviderServices;

import org.gvsig.raster.lib.buffer.api.Buffer;

import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;

public class RasterGdalStoreProviderTest extends AbstractLibraryAutoInitTestCase {

    @Override

    protected void doSetUp() throws Exception {

    }

    /**

     * Tries to copy an store into a new one using gdal provider

     * @throws Exception

     */

    public void testGdalCopyStore() throws Exception{

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

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

        //The store is created using the PNG provider

        String openingProviderName="PNG";

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

        //The store is copied using the GTiff provider

        String storingProviderName="GTiff";

        String extension="tif";

        File destFile=new File(openedFile.getParentFile(), "copiedFile."+extension);

        DataManager dataManager = DALLocator.getDataManager();

        try {

            //Opening the store

            RasterGdalStoreProviderFactory gdalFactory=

                (RasterGdalStoreProviderFactory)dataManager.getStoreProviderFactory(openingProviderName);

            RasterGdalFileStoreParameters openParams = (RasterGdalFileStoreParameters)gdalFactory.createParameters();

            openParams.setFile(openedFile);

            openParams.setCRS(projection);

            RasterStore store=null;

            store = (RasterStore) dataManager.openStore(openingProviderName,openParams);

            assertNotNull(store);

            //The store is opened and must be copied

            DataManagerProviderServices manager=DALSPILocator.getDataManagerProviderServices();

            DataServerExplorerParameters eparams = manager.createServerExplorerParameters("FilesystemExplorer");

            eparams.setDynValue("initialpath","/");

            DataServerExplorer serverExplorer= manager.createServerExplorer(eparams);

            NewRasterStoreParameters newStoreParams =(NewRasterStoreParameters) serverExplorer.getAddParameters(storingProviderName);

            if (newStoreParams instanceof FilesystemStoreParameters) {

                ((FilesystemStoreParameters) newStoreParams).setFile(destFile);

            }

            if (newStoreParams instanceof NewRasterStoreParameters) {

                Buffer buffer = store.getRasterSet().clip(store.getEnvelope());

                ((NewRasterStoreParameters) newStoreParams).setBuffer(buffer);

            }

            if (newStoreParams.hasDynValue("compress")) {

                newStoreParams.setDynValue("compress", "NONE");

            }

            if (newStoreParams.hasDynValue("tfw")) {

                newStoreParams.setDynValue("tfw", true);

            }

            if (newStoreParams.hasDynValue("photometric")) {

                newStoreParams.setDynValue("photometric", "RGB");

            }

            if (newStoreParams.isValid() ) {

                serverExplorer.add(storingProviderName, newStoreParams, true);

            }

            assert(destFile.exists());

        }catch (Exception e){

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

        }finally{

            if (destFile!=null && destFile.exists()){

                destFile.delete();

            }

        }

    }

}