Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.sqlite.dal.geopackage.index;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

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

import org.gvsig.fmap.dal.store.jdbc2.JDBCConnection;

import org.gvsig.fmap.dal.store.jdbc2.JDBCUtils;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.GeometryLocator;

import org.gvsig.fmap.geom.GeometryManager;

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

import org.gvsig.sqlite.dal.utils.TemplateUtils;

import org.sqlite.SQLiteException;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public abstract class AbstractGeopackageIndex implements GeopackageIndex {

    public AbstractGeopackageIndex() {

    }

    protected abstract String getGroup();

    protected abstract String getAttributeIndexFlagName();

    @Override

    public boolean hasIndex(JDBCConnection conn, String tablename, String fieldnamegeom) {

        String sql = TemplateUtils.getSqlTemplate(this.getGroup(), "has_index", tablename, fieldnamegeom);

        Statement st = null;

        ResultSet rs = null;

        try {

            st = conn.createStatement();

            rs = JDBCUtils.executeQuery(st, sql);

            if( !rs.next() ) {

                return false;

            }

            int n = rs.getInt(1);

            return n==1;

        } catch(SQLiteException ex) {

            if( ex.getMessage().contains("no such table") ) {

                return false;

            }

            throw new RuntimeException("Can't get information about "+this.getGroup()+" index",ex);

        } catch(SQLException ex) {

            throw new RuntimeException("Can't get information about "+this.getGroup()+" index",ex);

        } finally {

            JDBCUtils.closeQuietly(rs);

            JDBCUtils.closeQuietly(st);

        }

    }

    @Override

    public void createIndex(JDBCConnection conn, String tablename, String fieldnamegeom, String fieldnameid) throws SQLException {

        for (String sql : this.getCreateIndexSQL(tablename, fieldnamegeom, fieldnameid)) {

            JDBCUtils.execute(conn.get(), sql);

        }

    }

    @Override

    public String getBBoxIntersectsFilter(String tablename, String fieldnameid, String fieldnamegeom, double minX, double minY, double maxX, double maxY) {

        String sql = TemplateUtils.getSqlTemplate(getGroup(), "bboxintersects", tablename, fieldnameid, fieldnamegeom, minX, minY, maxX, maxY);

        return sql;

    }

    @Override

    public String getBBoxIntersectsFilter(String tablename, String fieldnameid, String fieldnamegeom, String minX, String minY, String maxX, String maxY) {

        String sql = TemplateUtils.getSqlTemplate(getGroup(), "bboxintersects", tablename, fieldnameid, minX, minY, maxX, maxY);

        return sql;

    }

    @Override

    public String getBBoxIntersectsFilter(String tablename, String fieldnameid, String fieldnamegeom, Envelope env) {

        String sql = this.getBBoxIntersectsFilter(

                tablename, 

                fieldnameid, 

                fieldnamegeom,

                env.getMinimum(Geometry.DIMENSIONS.X), 

                env.getMinimum(Geometry.DIMENSIONS.Y), 

                env.getMaximum(Geometry.DIMENSIONS.X), 

                env.getMaximum(Geometry.DIMENSIONS.Y)

        );

        return sql;

    }

    @Override

    public String getBBoxIntersectsFilter(String tablename, String fieldnameid, String fieldnamegeom, Geometry geom) {

        return this.getBBoxIntersectsFilter(tablename, fieldnameid, fieldnamegeom, geom.getEnvelope());

    }

    @Override

    public void setHasIndex(FeatureAttributeDescriptor attr, boolean b) {

        attr.getTags().set(this.getAttributeIndexFlagName(), b?1:0);

    }

    @Override

    public boolean hasIndex(FeatureAttributeDescriptor attr) {

        return attr.getTags().getInt(this.getAttributeIndexFlagName(), 0)==1;

    }

    @Override

    public Envelope getEnvelope(JDBCConnection conn, String tablename, String fieldnamegeom) throws SQLException {

        String sql = this.getEnvelopeSQL(tablename, fieldnamegeom);

        Statement st = null;

        ResultSet rs = null;

        try {

            st = conn.createStatement();

            rs = JDBCUtils.executeQuery(st, sql);

            if( !rs.next() ) {

                return null;

            }

            int minx = rs.getInt("minx");

            int miny = rs.getInt("miny");

            int maxx = rs.getInt("maxx");

            int maxy = rs.getInt("maxy");

            GeometryManager geommanager = GeometryLocator.getGeometryManager();

            Envelope env = geommanager.createEnvelope(minx, miny, maxx, maxy, Geometry.SUBTYPES.GEOM2D);

            return env;

        } catch(SQLiteException ex) {

            if( ex.getMessage().contains("no such table") ) {

                return null;

            }

            throw new RuntimeException("Can't get envelop of index for group "+this.getGroup(),ex);

        } catch(Throwable ex) {

            throw new RuntimeException("Can't get envelop of index for group "+this.getGroup(),ex);

        } finally {

            JDBCUtils.closeQuietly(rs);

            JDBCUtils.closeQuietly(st);

        }

    }

}