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;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.Map;

import org.gvsig.expressionevaluator.ExpressionBuilder;

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

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

import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase;

import org.gvsig.sqlite.dal.operations.SQLiteFetchFeatureTypeOperation;

import org.gvsig.sqlite.dal.operations.SQLiteFetchFeatureTypeOperation.SQLiteColumnInfo;

import org.sqlite.SQLiteException;

/**

 *

 * @author jjdelcerro

 */

public class GeopackageGeometryColumns {

    public static class GeopackageGeometryColumn extends SQLiteFetchFeatureTypeOperation.SQLiteColumnInfo {

        private String geometry_type_name;

        private int srs_id;

        private int z;

        private int m;

        /*friend*/ GeopackageGeometryColumn() {

            super();

        }

        public void set(

                String table_name,

                String column_column,

                String geometry_type_name,

                int srs_id,

                int z,

                int m

        ) {

            this.table_name = table_name;

            this.column_column = column_column;

            this.geometry_type_name = geometry_type_name;

            this.srs_id = srs_id;

            this.z = z;

            this.m = m;

        }

        /**

         * @return the geometry_type_name

         */

        public String getGeometry_type_name() {

            return geometry_type_name;

        }

        /**

         * @return the srs_id

         */

        public int getSrs_id() {

            return srs_id;

        }

        /**

         * @return the z

         */

        public int getZ() {

            return z;

        }

        /**

         * @return the m

         */

        public int getM() {

            return m;

        }

    }

    /*friend*/ static void load_gpkg_geometry_columns(Map<String,SQLiteColumnInfo> columns_info, JDBCConnection conn, String tableName, JDBCSQLBuilderBase sqlbuilder) {

        ExpressionBuilder expbuilder = sqlbuilder.expression();

        sqlbuilder.select().column().name("table_name");

        sqlbuilder.select().column().name("column_name");

        sqlbuilder.select().column().name("geometry_type_name");

        sqlbuilder.select().column().name("srs_id");

        sqlbuilder.select().column().name("z");

        sqlbuilder.select().column().name("m");

        sqlbuilder.select().where().set(

                expbuilder.eq(

                        expbuilder.column("table_name"),

                        expbuilder.constant(tableName)

                )

        );                

        sqlbuilder.select().from().table().name("gpkg_geometry_columns");

        Statement st = null;

        ResultSet rs = null;

        try {

            st = conn.createStatement();

            rs = JDBCUtils.executeQuery(st, sqlbuilder.toString());

            while( rs.next() ) {

                String columnname = rs.getString("column_name");

                GeopackageGeometryColumn geominfo = new GeopackageGeometryColumn();

                geominfo.copyfrom(columns_info.get(columnname));

                geominfo.set(

                    rs.getString("table_name"),

                    columnname,

                    rs.getString("geometry_type_name"),

                    rs.getInt("srs_id"),

                    rs.getInt("z"),

                    rs.getInt("m")

                );

                columns_info.put(columnname, geominfo);

            }

        } catch(SQLiteException ex) {

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

                throw new RuntimeException("Can't load information from table gpkg_geometry_columns",ex);

            }

        } catch (Throwable ex) {

            throw new RuntimeException("Can't load information from table gpkg_geometry_columns",ex);

        } finally {

            JDBCUtils.closeQuietly(rs);

            JDBCUtils.closeQuietly(st);

        }

    }    

}