| gvSIG desktop 1

package org.gvsig.fmap.dal.db.h2spatial.impl;

import java.sql.Connection;

import java.sql.ResultSetMetaData;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import net.sourceforge.hatbox.MetaNode;

import net.sourceforge.hatbox.RTreeDml;

import net.sourceforge.hatbox.RTreeSessionDb;

import net.sourceforge.hatbox.jts.Proc;

import org.gvsig.fmap.crs.CRSFactory;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.dal.NewDataStoreParameters;

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

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

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

import org.gvsig.fmap.dal.feature.EditableFeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.EditableFeatureType;

import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.feature.exception.UnsupportedDataTypeException;

import org.gvsig.fmap.dal.feature.exception.UnsupportedGeometryException;

import org.gvsig.fmap.dal.resource.spi.ResourceManagerProviderServices;

import org.gvsig.fmap.dal.store.jdbc.JDBCConnectionParameters;

import org.gvsig.fmap.dal.store.jdbc.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc.JDBCHelperUser;

import org.gvsig.fmap.dal.store.jdbc.JDBCStoreParameters;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCException;

import org.gvsig.fmap.dal.store.jdbc.exception.JDBCSQLException;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.Geometry.SUBTYPES;

import org.gvsig.fmap.geom.exception.CreateEnvelopeException;

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

/**

 * 

 * @author Vicente Caballero Navarro

 */

public class H2SpatialHelper extends JDBCHelper {

        private String rsid=null;

        private GeometryManager gm;

        public H2SpatialHelper(JDBCHelperUser consumer,

                        JDBCConnectionParameters params) throws InitializeException {

                super(consumer, params);

                gm = GeometryLocator.getGeometryManager();

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#initializeResource()

         */

        protected void initializeResource() throws InitializeException {

                ResourceManagerProviderServices manager = (ResourceManagerProviderServices) DALLocator

                                .getResourceManager();

                H2SpatialResource resource = (H2SpatialResource) manager

                                .createAddResource(H2SpatialResource.NAME, new Object[] {

                                                params.getUrl(), params.getDBName(),

                                                params.getJDBCDriverClassName(),

                                                ((H2SpatialConnectionParameters) params).getUseSSL() });

                this.setResource(resource);

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#loadFeatureType(org.gvsig.fmap.dal.feature.EditableFeatureType, org.gvsig.fmap.dal.store.jdbc.JDBCStoreParameters)

         */

        public void loadFeatureType(EditableFeatureType featureType,

                        JDBCStoreParameters storeParams) throws DataException {

                if (storeParams.getSQL() != null

                                && storeParams.getSQL().trim().length() == 0) {

                        loadFeatureType(featureType, storeParams, storeParams.getSQL(),

                                        null, null);

                } else {

                        String sql = "Select * from " + storeParams.tableID();

                        loadFeatureType(featureType, storeParams, sql, storeParams

                                        .getSchema(), storeParams.getTable());

                }

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#getFullEnvelopeOfField(org.gvsig.fmap.dal.store.jdbc.JDBCStoreParameters, java.lang.String, org.gvsig.fmap.geom.primitive.Envelope)

         */

        public org.gvsig.fmap.geom.primitive.Envelope getFullEnvelopeOfField(JDBCStoreParameters storeParams,

                        String geometryAttrName, Envelope limit) throws DataException {

                Connection con = getConnection();

                RTreeDml dml=null;

                RTreeSessionDb session=null;

                net.sourceforge.hatbox.Envelope hbEnvelope=null;

                Envelope gvSigEnvelope = null;

                try {

                        dml = RTreeDml.createDml(con, storeParams.getSchema(), storeParams.getTable().toUpperCase());

                        session = new RTreeSessionDb(con, dml, false);

                        hbEnvelope = session.getNode(session.getRootId()).getBounds();

                        gvSigEnvelope = 

                        gm.createEnvelope(hbEnvelope.getMinX(), hbEnvelope.getMinY(), hbEnvelope.getMaxX(), hbEnvelope.getMaxY(), SUBTYPES.GEOM2D);

                } catch (SQLException e2) {

                        e2.printStackTrace();

                } catch (CreateEnvelopeException e) {

                        e.printStackTrace();

                }

                session.closeAll();

                try {

                        con.close();

                } catch (SQLException e) {

                        e.printStackTrace();

                }

                return gvSigEnvelope;

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#getSqlColumnTypeDescription(org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor)

         */

        public String getSqlColumnTypeDescription(FeatureAttributeDescriptor attr) {

                switch (attr.getDataType()) {

                case DataTypes.BOOLEAN:

                        return "BOOLEAN";

                case DataTypes.BYTE:

                        return "SMALLINT";

                case DataTypes.BYTEARRAY:

                        return "BYTEA";

                case DataTypes.CHAR:

                        return "CHAR";

                case DataTypes.DATE:

                        return "DATE";

                case DataTypes.DOUBLE:

                        return "DOUBLE";

                case DataTypes.FLOAT:

                        return "FLOAT";

                case DataTypes.GEOMETRY:

                        return "BLOB";

                case DataTypes.INT:

                        return "INTEGER";

                case DataTypes.LONG:

                        return "BIGINT";

                case DataTypes.STRING:

                        return "VARCHAR("+attr.getSize()+")";

                case DataTypes.TIME:

                        return "TIME";

                case DataTypes.TIMESTAMP:

                        return "TIMESTAMP";

                default:

                        String typeName = (String) attr.getAdditionalInfo("SQLTypeName");

                        if (typeName != null) {

                                return typeName;

                        }

                        throw new UnsupportedDataTypeException(attr.getDataTypeName(), attr

                                        .getDataType());

                }

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#getSqlFieldName(org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor)

         */

        public String getSqlFieldName(FeatureAttributeDescriptor attribute) {

//                if (attribute.getDataType() == DataTypes.GEOMETRY) {

//                        return "asBinary(" + super.getSqlFieldName(attribute) + ")";

//                }

                return super.getSqlFieldName(attribute);

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#createAttributeFromJDBC(org.gvsig.fmap.dal.feature.EditableFeatureType, java.sql.Connection, java.sql.ResultSetMetaData, int)

         */

        protected EditableFeatureAttributeDescriptor createAttributeFromJDBC(

                        EditableFeatureType type, Connection conn,

                        ResultSetMetaData rsMetadata, int colIndex) throws SQLException {

                // if (rsMetadata.getColumnType(colIndex) == java.sql.Types.OTHER) {

                if (rsMetadata.getColumnName(colIndex).equalsIgnoreCase("the_geom")) {

                        EditableFeatureAttributeDescriptor efad = type.add(rsMetadata

                                        .getColumnName(colIndex), DataTypes.GEOMETRY);

                        efad.setGeometrySubType(Geometry.SUBTYPES.GEOM2D);

                        efad.setGeometryType(Geometry.TYPES.GEOMETRY);

                        if (rsid!=null)

                                efad.setSRS(CRSFactory.getCRS(rsid));

                        return efad;

                }

                // }

                return super.createAttributeFromJDBC(type, conn, rsMetadata, colIndex);

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#allowAutomaticValues()

         */

        public boolean allowAutomaticValues() {

                return Boolean.TRUE;

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#getSqlFieldDescription(org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor)

         */

        public String getSqlFieldDescription(FeatureAttributeDescriptor attr)

                        throws DataException {

//                return getSqlFieldName(attr)+ " "+ getSqlColumnTypeDescription(attr);

                if (attr.getDataType() == DataTypes.GEOMETRY) {

                        return null;

                }

                return super.getSqlFieldDescription(attr);

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#supportOffset()

         */

        public boolean supportOffset() {

                return true;

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#supportsUnion()

         */

        public boolean supportsUnion() {

                return true;

        }

//        public String getH2SpatialGeomType(int geometryType, int geometrySubType) {

//                String slGeomType = new String();

//                switch (geometryType) {

//                case Geometry.TYPES.GEOMETRY:

//                        slGeomType = "GEOMETRY";

//                        break;

//                case Geometry.TYPES.POINT:

//                        slGeomType = "POINT";

//                        break;

//                case Geometry.TYPES.CURVE:

//                        slGeomType = "LINESTRING";

//                        break;

//                case Geometry.TYPES.SURFACE:

//                        slGeomType = "POLYGON";

//                        break;

//                case Geometry.TYPES.MULTIPOINT:

//                        slGeomType = "MULTIPOINT";

//                        break;

//                case Geometry.TYPES.MULTICURVE:

//                        slGeomType = "MULTILINESTRING";

//                        break;

//                case Geometry.TYPES.MULTISURFACE:

//                        slGeomType = "MULTIPOLYGON";

//                        break;

//                default:

//                        throw new UnsupportedGeometryException(geometryType,

//                                        geometrySubType);

//                }

//                return slGeomType;

//        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#getProviderSRID(java.lang.String)

         */

        public int getProviderSRID(String srs) {

//                if (srs != null) {

//                        Integer slSRID = (Integer) srsID2slSR.get(srs);

//                        if (slSRID != null) {

//                                return slSRID.intValue();

//                        }

//

//                        return searchslSRID(srs);

//

//                }

                return -1;

        }

//        public int getH2SpatialGeomDimensions(int geometrySubType) {

//                switch (geometrySubType) {

//                case Geometry.SUBTYPES.GEOM2D:

//                        return 2;

//                case Geometry.SUBTYPES.GEOM2DM:

//                case Geometry.SUBTYPES.GEOM3D:

//                        return 3;

//

//                case Geometry.SUBTYPES.GEOM3DM:

//                        return 4;

//                default:

//                        throw new UnsupportedDataTypeException(

//                                        DataTypes.TYPE_NAMES[DataTypes.GEOMETRY],

//                                        DataTypes.GEOMETRY);

//                }

//        }

        public List getSqlGeometyFieldAdd(FeatureAttributeDescriptor attr,

                        String table, String schema) {

                List sqls = new ArrayList();

        return sqls;

        }

        public List getAdditionalSqlToCreate(NewDataStoreParameters ndsp,

                        FeatureType fType) {

                FeatureAttributeDescriptor attr;

                Iterator iter = fType.iterator();

                List result = new ArrayList();

                H2SpatialNewStoreParameters slNdsp = (H2SpatialNewStoreParameters) ndsp;

                while (iter.hasNext()) {

                        attr = (FeatureAttributeDescriptor) iter.next();

                        if (attr.getDataType() == DataTypes.GEOMETRY) {

                                result.addAll(getSqlGeometyFieldAdd(attr, slNdsp.getTable(),

                                                slNdsp.getSchema()));

                        }

                }

                return result;

        }

        /* (non-Javadoc)

         * @see org.gvsig.fmap.dal.store.jdbc.JDBCHelper#loadSRS_and_shapeType(java.sql.Connection, java.sql.ResultSetMetaData, org.gvsig.fmap.dal.feature.EditableFeatureType, java.lang.String, java.lang.String)

         */

        protected void loadSRS_and_shapeType(Connection conn,

                        ResultSetMetaData rsMetadata, EditableFeatureType featureType,

                        String baseSchema, String baseTable)

                        throws JDBCException {

                try {

                        // Sacamos la lista de los attributos geometricos

                        EditableFeatureAttributeDescriptor attr;

                        Iterator iter = featureType.iterator();

                        while (iter.hasNext()) {

                                attr = (EditableFeatureAttributeDescriptor) iter.next();

                                if (attr.getDataType() == DataTypes.GEOMETRY) {

                                        MetaNode metaNode = Proc.spatialMetaData(conn, baseSchema, baseTable.toUpperCase());

                                        String epsg = "EPSG:"+metaNode.getSrid();

                                        attr.setSRS(CRSFactory.getCRS(epsg));

                                        rsid=epsg;

                                }

                        }

                } catch (java.sql.SQLException e) {

                        throw new JDBCSQLException(e);

                } finally {

                }

        }

}