projFileCreate()

static int projFileCreate ( SHPDUMPERSTATE *  state )

static

Creates ESRI .prj file for this shp output It looks in the spatial_ref_sys table and outputs the srtext field for this data If data is a table will use geometry_columns, if a query or view will read SRID from query output.

Warning

Will give warning and not output a .prj file if SRID is -1, Unknown, mixed SRIDS or not found in spatial_ref_sys. The dbf and shp will still be output.

make our address space large enough to hold query with table/schema

Definition at line 717 of file pgsql2shp-core.c.

References _, shp_dumper_state::conn, free(), shp_dumper_state::geo_col_name, LWDEBUGF, malloc(), shp_dumper_state::message, window::res, shp_dumper_state::schema, shp_dumper_state::shp_file, SHPDUMPERMSGLEN, SHPDUMPEROK, SHPDUMPERWARN, and shp_dumper_state::table.

Referenced by ShpDumperCloseTable().

{
        FILE    *fp;
        char    *pszFullname, *pszBasename;
        int     i;

        char *pszFilename = state->shp_file;
        char *schema = state->schema;
        char *table = state->table;
        char *geo_col_name = state->geo_col_name;

        char *srtext;
        char *query;
        char *esc_schema;
        char *esc_table;
        char *esc_geo_col_name;

        int error, result;
        PGresult *res;
        int size;

        /***********
        *** I'm multiplying by 2 instead of 3 because I am too lazy to figure out how many characters to add
        *** after escaping if any **/
        size = 1000;
        if ( schema )
        {
                size += 3 * strlen(schema);
        }
        size += 1000;
        esc_table = (char *) malloc(3 * strlen(table) + 1);
        esc_geo_col_name = (char *) malloc(3 * strlen(geo_col_name) + 1);
        PQescapeStringConn(state->conn, esc_table, table, strlen(table), &error);
        PQescapeStringConn(state->conn, esc_geo_col_name, geo_col_name, strlen(geo_col_name), &error);

        query = (char *) malloc(size);
        if ( ! query ) return 0; /* out of virtual memory */

        /**************************************************
         * Get what kind of spatial ref is the selected geometry field
         * We first check the geometry_columns table for a match and then if no match do a distinct against the table
         * NOTE: COALESCE does a short-circuit check returning the faster query result and skipping the second if first returns something
         *      Escaping quotes in the schema and table in query may not be necessary except to prevent malicious attacks
         *      or should someone be crazy enough to have quotes or other weird character in their table, column or schema names
         **************************************************/
        if ( schema )
        {
                esc_schema = (char *) malloc(2 * strlen(schema) + 1);
                PQescapeStringConn(state->conn, esc_schema, schema, strlen(schema), &error);
                sprintf(query, "SELECT COALESCE((SELECT sr.srtext "
                        " FROM  geometry_columns As gc INNER JOIN spatial_ref_sys sr ON sr.srid = gc.srid "
                        " WHERE gc.f_table_schema = '%s' AND gc.f_table_name = '%s' AND gc.f_geometry_column = '%s' LIMIT 1),  "
                        " (SELECT CASE WHEN COUNT(DISTINCT sr.srid) > 1 THEN 'm' ELSE MAX(sr.srtext) END As srtext "
                        " FROM \"%s\".\"%s\" As g INNER JOIN spatial_ref_sys sr ON sr.srid = ST_SRID((g.\"%s\")::geometry)) , ' ') As srtext ",
                        esc_schema, esc_table,esc_geo_col_name, schema, table, geo_col_name);
                free(esc_schema);
        }
        else
        {
                sprintf(query, "SELECT COALESCE((SELECT sr.srtext "
                        " FROM  geometry_columns As gc INNER JOIN spatial_ref_sys sr ON sr.srid = gc.srid "
                        " WHERE gc.f_table_name = '%s' AND gc.f_geometry_column = '%s' AND pg_table_is_visible((gc.f_table_schema || '.' || gc.f_table_name)::regclass) LIMIT 1),  "
                        " (SELECT CASE WHEN COUNT(DISTINCT sr.srid) > 1 THEN 'm' ELSE MAX(sr.srtext) END as srtext "
                        " FROM \"%s\" As g INNER JOIN spatial_ref_sys sr ON sr.srid = ST_SRID((g.\"%s\")::geometry)), ' ') As srtext ",
                        esc_table, esc_geo_col_name, table, geo_col_name);
        }

        LWDEBUGF(3,"%s\n",query);
        free(esc_table);
        free(esc_geo_col_name);

        res = PQexec(state->conn, query);

        if ( ! res || PQresultStatus(res) != PGRES_TUPLES_OK )
        {
                snprintf(state->message, SHPDUMPERMSGLEN, _("WARNING: Could not execute prj query: %s"), PQresultErrorMessage(res));
                PQclear(res);
                free(query);
                return SHPDUMPERWARN;
        }

        for (i=0; i < PQntuples(res); i++)
        {
                srtext = PQgetvalue(res, i, 0);
                if (strcmp(srtext,"m") == 0)
                {
                        snprintf(state->message, SHPDUMPERMSGLEN, _("WARNING: Mixed set of spatial references. No prj file will be generated"));
                        PQclear(res);
                        free(query);
                        return SHPDUMPERWARN;
                }
                else
                {
                        if (srtext[0] == ' ')
                        {
                                snprintf(state->message, SHPDUMPERMSGLEN, _("WARNING: Cannot determine spatial reference (empty table or unknown spatial ref). No prj file will be generated."));
                                PQclear(res);
                                free(query);
                                return SHPDUMPERWARN;
                        }
                        else
                        {
                                /* -------------------------------------------------------------------- */
                                /*      Compute the base (layer) name.  If there is any extension       */
                                /*      on the passed in filename we will strip it off.                 */
                                /* -------------------------------------------------------------------- */
                                pszBasename = (char *) malloc(strlen(pszFilename)+5);
                                strcpy( pszBasename, pszFilename );
                                for ( i = strlen(pszBasename)-1;
                                        i > 0 && pszBasename[i] != '.' && pszBasename[i] != '/'
                                        && pszBasename[i] != '\\';
                                        i-- ) {}

                                if ( pszBasename[i] == '.' )
                                        pszBasename[i] = '\0';

                                pszFullname = (char *) malloc(strlen(pszBasename) + 5);
                                sprintf( pszFullname, "%s.prj", pszBasename );
                                free( pszBasename );


                                /* -------------------------------------------------------------------- */
                                /*      Create the file.                                                */
                                /* -------------------------------------------------------------------- */
                                fp = fopen( pszFullname, "wb" );
                                if ( fp == NULL )
                                {
                                        free(pszFullname);
                                        free(query);
                                        return 0;
                                }
                                else
                                {
                                    result = fputs (srtext,fp);
                    LWDEBUGF(3, "\n result %d proj SRText is %s .\n", result, srtext);
                    if (result == EOF)
                    {
                        fclose( fp );
                        free( pszFullname );
                        PQclear(res);
                        free(query);
                        return 0;
                    }
                                }
                                fclose( fp );
                                free( pszFullname );
                        }
                }
        }
        PQclear(res);
        free(query);
        return SHPDUMPEROK;
}

Here is the call graph for this function:

Here is the caller graph for this function: