ShpLoaderGenerateShapeRow()

int ShpLoaderGenerateShapeRow

(

SHPDUMPERSTATE *

state

)

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

References _, shp_dumper_config::binary, shp_dumper_state::config, shp_dumper_state::conn, convert_bytes_to_hex(), create_linestring(), create_multilinestring(), create_multipoint(), create_multipolygon(), create_point(), create_point_empty(), create_polygon(), shp_dumper_state::currescount, shp_dumper_state::curresrow, shp_dumper_state::currow, shp_dumper_state::dbf, shp_dumper_state::dbffieldtypes, DBFWriteAttributeDirectly(), shp_dumper_state::fetch_query, shp_dumper_state::fetchres, shp_dumper_state::fieldcount, free(), shp_dumper_state::geo_col_name, goodDBFValue(), LINETYPE, LW_PARSER_CHECK_NONE, LWDEBUG, LWDEBUGF, lwgeom_as_lwline(), lwgeom_as_lwmline(), lwgeom_as_lwmpoint(), lwgeom_as_lwmpoly(), lwgeom_as_lwpoint(), lwgeom_as_lwpoly(), lwgeom_free(), lwgeom_from_hexwkb(), lwgeom_is_empty(), lwtype_name(), malloc(), shp_dumper_state::message, MULTILINETYPE, MULTIPOINTTYPE, MULTIPOLYGONTYPE, nullDBFValue(), shp_dumper_state::pgis_major_version, POINTTYPE, POLYGONTYPE, shp_dumper_state::rowcount, shp_dumper_state::shp, SHPCreateSimpleObject(), SHPDestroyObject(), SHPDUMPERERR, SHPDUMPERMSGLEN, SHPDUMPEROK, SHPT_NULL, SHPWriteObject(), and LWGEOM::type.

Referenced by main(), and pgui_action_export().

{
        char *hexewkb = NULL;
        unsigned char *hexewkb_binary = NULL;
        size_t hexewkb_len;
        char *val;
        SHPObject *obj = NULL;
        LWGEOM *lwgeom;

        int i, geocolnum = 0;

        /* If we try to go pass the end of the table, fail immediately */
        if (state->currow > state->rowcount)
        {
                snprintf(state->message, SHPDUMPERMSGLEN, _("Tried to read past end of table!"));
                PQclear(state->fetchres);
                return SHPDUMPERERR;
        }

        /* If we have reached the end of the current batch, fetch a new one */
        if (state->curresrow == state->currescount && state->currow < state->rowcount)
        {
                /* Clear the previous batch results */
                if (state->fetchres)
                        PQclear(state->fetchres);

                state->fetchres = PQexec(state->conn, state->fetch_query);
                if (PQresultStatus(state->fetchres) != PGRES_TUPLES_OK)
                {
                        snprintf(state->message, SHPDUMPERMSGLEN, _("Error executing fetch query: %s"), PQresultErrorMessage(state->fetchres));
                        PQclear(state->fetchres);
                        return SHPDUMPERERR;
                }

                state->curresrow = 0;
                state->currescount = PQntuples(state->fetchres);
        }

        /* Grab the id of the geo column if we have one */
        if (state->geo_col_name)
                geocolnum = PQfnumber(state->fetchres, "_geoX");

        /* Process the next record within the batch. First write out all of
        the non-geo fields */
        for (i = 0; i < state->fieldcount; i++)
        {
                /*
                * Transform NULL numbers to '0'
                * This is because the shapelib
                * won't easly take care of setting
                * nulls unless paying the acquisition
                * of a bug in long integer values
                */
                if (PQgetisnull(state->fetchres, state->curresrow, i))
                {
                        val = nullDBFValue(state->dbffieldtypes[i]);
                }
                else
                {
                        val = PQgetvalue(state->fetchres, state->curresrow, i);
                        val = goodDBFValue(val, state->dbffieldtypes[i]);
                }

                /* Write it to the DBF file */
                if (!DBFWriteAttributeDirectly(state->dbf, state->currow, i, val))
                {
                        snprintf(state->message, SHPDUMPERMSGLEN, _("Error: record %d could not be created"), state->currow);
                        PQclear(state->fetchres);
                        return SHPDUMPERERR;
                }
        }

        /* Now process the geo field, if present */
        if (state->geo_col_name)
        {
                /* Handle NULL shapes */
                if (PQgetisnull(state->fetchres, state->curresrow, geocolnum))
                {
                        obj = SHPCreateSimpleObject(SHPT_NULL, 0, NULL, NULL, NULL);
                        if (SHPWriteObject(state->shp, -1, obj) == -1)
                        {
                                snprintf(state->message, SHPDUMPERMSGLEN, _("Error writing NULL shape for record %d"), state->currow);
                                PQclear(state->fetchres);
                                SHPDestroyObject(obj);
                                return SHPDUMPERERR;
                        }
                        SHPDestroyObject(obj);
                }
                else
                {
                        /* Get the value from the result set */
                        val = PQgetvalue(state->fetchres, state->curresrow, geocolnum);

                        if (!state->config->binary)
                        {
                                if (state->pgis_major_version > 0)
                                {
                                        LWDEBUG(4, "PostGIS >= 1.0, non-binary cursor");

                                        /* Input is bytea encoded text field, so it must be unescaped and
                                        then converted to hexewkb string */
                                        hexewkb_binary = PQunescapeBytea((unsigned char *)val, &hexewkb_len);
                                        hexewkb = convert_bytes_to_hex(hexewkb_binary, hexewkb_len);
                                }
                                else
                                {
                                        LWDEBUG(4, "PostGIS < 1.0, non-binary cursor");

                                        /* Input is already hexewkb string, so we can just
                                        copy directly to hexewkb */
                                        hexewkb_len = PQgetlength(state->fetchres, state->curresrow, geocolnum);
                                        hexewkb = malloc(hexewkb_len + 1);
                                        strncpy(hexewkb, val, hexewkb_len + 1);
                                }
                        }
                        else /* binary */
                        {
                                LWDEBUG(4, "PostGIS (any version) using binary cursor");

                                /* Input is binary field - must convert to hexewkb string */
                                hexewkb_len = PQgetlength(state->fetchres, state->curresrow, geocolnum);
                                hexewkb = convert_bytes_to_hex((unsigned char *)val, hexewkb_len);
                        }

                        LWDEBUGF(4, "HexEWKB - length: %d  value: %s", strlen(hexewkb), hexewkb);

                        /* Deserialize the LWGEOM */
                        lwgeom = lwgeom_from_hexwkb(hexewkb, LW_PARSER_CHECK_NONE);
                        if (!lwgeom)
                        {
                                snprintf(state->message, SHPDUMPERMSGLEN, _("Error parsing HEXEWKB for record %d"), state->currow);
                                PQclear(state->fetchres);
                                free(hexewkb);
                                return SHPDUMPERERR;
                        }

                        /* Call the relevant method depending upon the geometry type */
                        LWDEBUGF(4, "geomtype: %s\n", lwtype_name(lwgeom->type));

                        switch (lwgeom->type)
                        {
                        case POINTTYPE:
                                if (lwgeom_is_empty(lwgeom))
                                {
                                        obj = create_point_empty(state, lwgeom_as_lwpoint(lwgeom));
                                }
                                else
                                {
                                        obj = create_point(state, lwgeom_as_lwpoint(lwgeom));
                                }
                                break;

                        case MULTIPOINTTYPE:
                                obj = create_multipoint(state, lwgeom_as_lwmpoint(lwgeom));
                                break;

                        case POLYGONTYPE:
                                obj = create_polygon(state, lwgeom_as_lwpoly(lwgeom));
                                break;

                        case MULTIPOLYGONTYPE:
                                obj = create_multipolygon(state, lwgeom_as_lwmpoly(lwgeom));
                                break;

                        case LINETYPE:
                                obj = create_linestring(state, lwgeom_as_lwline(lwgeom));
                                break;

                        case MULTILINETYPE:
                                obj = create_multilinestring(state, lwgeom_as_lwmline(lwgeom));
                                break;

                        default:
                                snprintf(state->message, SHPDUMPERMSGLEN, _("Unknown WKB type (%d) for record %d"), lwgeom->type, state->currow);
                                PQclear(state->fetchres);
                                SHPDestroyObject(obj);
                                return SHPDUMPERERR;
                        }

                        /* Free both the original and geometries */
                        lwgeom_free(lwgeom);

                        /* Write the shape out to the file */
                        if (SHPWriteObject(state->shp, -1, obj) == -1)
                        {
                                snprintf(state->message, SHPDUMPERMSGLEN, _("Error writing shape %d"), state->currow);
                                PQclear(state->fetchres);
                                SHPDestroyObject(obj);
                                return SHPDUMPERERR;
                        }

                        SHPDestroyObject(obj);

                        /* Free the hexewkb (and temporary bytea unescaped string if used) */
                        if (hexewkb) free(hexewkb);
                        if (hexewkb_binary) PQfreemem(hexewkb_binary);
                }
        }

        /* Increment ready for next time */
        state->curresrow++;
        state->currow++;

        return SHPDUMPEROK;
}

Here is the call graph for this function:

Here is the caller graph for this function: