Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 */

package org.gvsig.raster.postgis.io;

import java.awt.geom.AffineTransform;

import java.awt.geom.Point2D;

import java.awt.geom.Rectangle2D;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Iterator;

import java.util.List;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataStore;

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

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

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

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

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

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

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

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

import org.gvsig.fmap.dal.serverexplorer.db.DBServerExplorer;

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

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

import org.gvsig.fmap.dal.store.db.DBStoreParameters;

import org.gvsig.jgdal.GdalException;

import org.gvsig.metadata.MetadataLocator;

import org.gvsig.raster.cache.tile.provider.TileListener;

import org.gvsig.raster.gdal.io.GdalNative;

import org.gvsig.raster.gdal.io.GdalProvider;

import org.gvsig.raster.impl.datastruct.ExtentImpl;

import org.gvsig.raster.impl.store.AbstractRasterDataParameters;

import org.gvsig.raster.impl.store.DefaultStoreFactory;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.extensionpoint.ExtensionPoint;

import org.gvsig.tools.extensionpoint.ExtensionPointManager;

/**

 * This class represents the data access just for PostGIS raster databases.

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

 */

public class PostGISRasterProvider extends GdalProvider {

        public static String         NAME                     = "PostGIS Raster Store";

        public static String         DESCRIPTION              = "PostGIS Raster file";

        public static final String   METADATA_DEFINITION_NAME = "PostGISRasterStore";

        private static final String  OVERVIEW_PREFIX          = "o_";

        private DBServerExplorer     dbServerExplorer         = null;

        private DBStoreParameters    dbParameters             = null;

        private ArrayList<Overview>  overviews                = new ArrayList<Overview>();

        class Overview implements Comparable<Overview> {

                public int               width               = 0;

                public int               height              = 0;

                public int               factor              = 0;

                public double            pxSizeOverview      = 0;

                public DBStoreParameters param             = null;

                public Overview() {

                }

                public Overview(int f, int w, int h, DBStoreParameters p) {

                        this.width = w;

                        this.height = h;

                        this.factor = f;

                        this.param = p;

                        this.pxSizeOverview = getExtent().width() / (double)w;

                }

                public int compareTo(Overview o) {

                        if(factor < o.factor)

                                return -1;

                        if(factor > o.factor)

                                return 1;

                        return 0;

                }

        }

        public static void register() {

                ExtensionPointManager extensionPoints = ToolsLocator.getExtensionPointManager();

                ExtensionPoint point = extensionPoints.get("DefaultDriver");

                point.append("reader", PostGISRasterProvider.NAME, PostGISRasterProvider.class);

                RasterLocator.getManager().registerFileProvidersTiled(PostGISRasterProvider.class);

                DataManagerProviderServices dataman = (DataManagerProviderServices) DALLocator.getDataManager();

                if (dataman != null && !dataman.getStoreProviders().contains(NAME)) {

                        dataman.registerStoreProvider(NAME,

                                        PostGISRasterProvider.class, PostGISRasterDataParameters.class);

                }

                if (!dataman.getExplorerProviders().contains(PostGISRasterProvider.NAME)) {

                        dataman.registerExplorerProvider(PostGISRasterServerExplorer.NAME, PostGISRasterServerExplorer.class, PostGISRasterServerExplorerParameters.class);

                }

                dataman.registerStoreFactory(NAME, DefaultStoreFactory.class);

        }

        public PostGISRasterProvider() {

        }

        public PostGISRasterProvider (PostGISRasterDataParameters params,

                        DataStoreProviderServices storeServices) throws NotSupportedExtensionException {

                super(params, storeServices, ToolsLocator.getDynObjectManager()

                                .createDynObject(

                                                MetadataLocator.getMetadataManager().getDefinition(

                                                                DataStore.METADATA_DEFINITION_NAME)));

                init(params, storeServices);

        }

        /**

         * Creates data base references and loads structures with the information and metadata

         * @param params load parameters

         * @throws NotSupportedExtensionException

         */

        public void init (AbstractRasterDataParameters params,

                        DataStoreProviderServices storeServices) throws NotSupportedExtensionException {

                try {

                        setParam(storeServices, params);

                        file = new GdalNative(translateFileName(params.getURI()));

                        setColorInterpretation(file.getColorInterpretation());

                        setColorTable(file.getColorTable());

                        noData = file.getNoDataValue();

                        wktProjection = file.getProjectionRef();

                        //CrsWkt crs = new CrsWkt(wktProjection);

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

                        ownTransformation = file.getOwnTransformation();

                        externalTransformation = (AffineTransform)ownTransformation.clone();

                        bandCount = file.getRasterCount();

                        load();

                } catch (Exception e) {

                        throw new NotSupportedExtensionException("Gdal library can't be initialized with this PostGIS raster parameters", e);

                } 

                //Obtenemos el tipo de dato de gdal y lo convertimos el de RasterBuf

                int[] dt = new int[file.getDataType().length];

                for (int i = 0; i < dt.length; i++)

                        dt[i] = rasterUtil.getRasterBufTypeFromGdalType(file.getDataType()[i]);

                setDataType(dt);

                PostGISRasterDataParameters p = (PostGISRasterDataParameters)params;

                //Object obj = p.getDynValue(PostGISRasterDataParameters.FIELD_DBPARAMS);

                dbParameters = (DBStoreParameters)p.getDynValue(PostGISRasterDataParameters.FIELD_DBPARAMS);

                dbServerExplorer = (DBServerExplorer)p.getDynValue(PostGISRasterDataParameters.FIELD_DBEXPLORER);

                try {

                        readOverviews();

                } catch (DataException e) {

                        throw new NotSupportedExtensionException("Overviews can't be read", e);

                }

        }

        /**

         * Reads overviews from database. The overviews in PostGIS raster are loaded 

         * in separated tables. The name of those tables is the same that the original table

         * with a overview prefix. Now this prefix is "o_X_" where the X is a factor scale.

         * 

         * For that reason the overview object contains the DataParameter which correspond

         * to each table.

         *   

         * @throws DataException

         */

        private void readOverviews() throws DataException {

                overviews.clear();

                overviews.add(new Overview(1, (int)getWidth(), (int)getHeight(), (DBStoreParameters)param));

                String table = dbParameters.getTable();

                List parameters = dbServerExplorer.list();

                Iterator iter = parameters.iterator();

                DBStoreParameters param = null;

                while (iter.hasNext()) {

                        DBStoreParameters p = (DBStoreParameters) iter.next();

                        String tname = p.getTable();

                        if(tname.endsWith(table) && tname.startsWith(OVERVIEW_PREFIX)) {

                                String factor = tname.substring(tname.indexOf('_') + 1);

                                factor = factor.substring(0, factor.indexOf('_'));

                                try {

                                        Overview o = new Overview();

                                        o.factor = new Integer(factor);

                                        o.width = (int)(getWidth() / o.factor);

                                        o.height = (int)(getHeight() / o.factor);

                                        o.pxSizeOverview = getExtent().width() / (double)o.width;

                                        o.param = p;

                                        overviews.add(o);

                                } catch (NumberFormatException e) {

                                }

                        }

                }

                Collections.sort(overviews);

        }

        /**

         * Selects the right overview. The right overview means the first 

         * one with a pixel size lesser or equals than the original. 

         * @param pixels

         * @param mts

         */

        private void overviewSelector(int pixels, double mts) {

                this.param = overviews.get(0).param;

                double pxSizeRequest = mts / (double)pixels;

                for (int i = overviews.size() - 1; i > 0; i--) {

                        Overview ovb = overviews.get(i);

                        if(pxSizeRequest <= ovb.pxSizeOverview) {

                                this.param = ovb.param;

                        }

                }

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.impl.provider.DefaultRasterProvider#getWindowRaster(double, double, double, double, int, int, org.gvsig.fmap.dal.coverage.datastruct.BandList, org.gvsig.raster.cache.tile.provider.TileListener)

         */

        public void getWindow(Extent ex, int bufWidth, int bufHeight, 

                        BandList bandList, TileListener listener) throws ProcessInterruptedException, RasterDriverException {

                 super.getWindow(ex, bufWidth, bufHeight, bandList, listener);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.impl.provider.DefaultRasterProvider#getWindowRaster(org.gvsig.fmap.dal.coverage.datastruct.Extent, org.gvsig.fmap.dal.coverage.datastruct.BandList, org.gvsig.fmap.dal.coverage.dataset.Buffer)

         */

        public Buffer getWindow(Extent ex, BandList bandList, Buffer rasterBuf) 

                throws ProcessInterruptedException, RasterDriverException {

                return super.getWindow(ex, bandList, rasterBuf);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.impl.provider.DefaultRasterProvider#getWindowRaster(double, double, double, double, org.gvsig.fmap.dal.coverage.datastruct.BandList, org.gvsig.fmap.dal.coverage.dataset.Buffer, boolean)

         */

        public Buffer getWindow(double ulx, double uly, double w, double h, 

                        BandList bandList, Buffer rasterBuf, boolean adjustToExtent) throws ProcessInterruptedException, RasterDriverException {

                return super.getWindow(ulx, uly, w, h, bandList, rasterBuf, adjustToExtent);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.impl.provider.DefaultRasterProvider#getWindowRaster(double, double, double, double, int, int, org.gvsig.fmap.dal.coverage.datastruct.BandList, org.gvsig.fmap.dal.coverage.dataset.Buffer, boolean)

         */

        public Buffer getWindow(Extent extent, 

                        int bufWidth, int bufHeight, BandList bandList, Buffer rasterBuf, boolean adjustToExtent) throws ProcessInterruptedException, RasterDriverException {

                return super.getWindow(extent, bufWidth, bufHeight, bandList, rasterBuf, adjustToExtent);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.impl.provider.DefaultRasterProvider#getWindowRaster(int, int, org.gvsig.fmap.dal.coverage.datastruct.BandList, org.gvsig.fmap.dal.coverage.dataset.Buffer)

         */

        public Buffer getWindow(int x, int y, 

                        BandList bandList, Buffer rasterBuf) throws ProcessInterruptedException, RasterDriverException {

                return super.getWindow(x, y, bandList, rasterBuf);

        }

        /*

         * (non-Javadoc)

         * @see org.gvsig.raster.impl.provider.DefaultRasterProvider#getWindowRaster(int, int, int, int, int, int, org.gvsig.fmap.dal.coverage.datastruct.BandList, org.gvsig.fmap.dal.coverage.dataset.Buffer)

         */

        public Buffer getWindow(int x, int y, int w, int h, 

                        int bufWidth, int bufHeight, BandList bandList, Buffer rasterBuf) throws ProcessInterruptedException, RasterDriverException {

                return super.getWindow(x, y, w, h, bufWidth, bufHeight, bandList, rasterBuf);

        }

}