PostGIS  2.5.7dev-r@@SVN_REVISION@@

◆ ShpLoaderGenerateShapeRow()

int ShpLoaderGenerateShapeRow ( SHPDUMPERSTATE state)

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

1985 {
1986  char *hexewkb = NULL;
1987  unsigned char *hexewkb_binary = NULL;
1988  size_t hexewkb_len;
1989  char *val;
1990  SHPObject *obj = NULL;
1991  LWGEOM *lwgeom;
1992 
1993  int i, geocolnum = 0;
1994 
1995  /* If we try to go pass the end of the table, fail immediately */
1996  if (state->currow > state->rowcount)
1997  {
1998  snprintf(state->message, SHPDUMPERMSGLEN, _("Tried to read past end of table!"));
1999  PQclear(state->fetchres);
2000  return SHPDUMPERERR;
2001  }
2002 
2003  /* If we have reached the end of the current batch, fetch a new one */
2004  if (state->curresrow == state->currescount && state->currow < state->rowcount)
2005  {
2006  /* Clear the previous batch results */
2007  if (state->fetchres)
2008  PQclear(state->fetchres);
2009 
2010  state->fetchres = PQexec(state->conn, state->fetch_query);
2011  if (PQresultStatus(state->fetchres) != PGRES_TUPLES_OK)
2012  {
2013  snprintf(state->message, SHPDUMPERMSGLEN, _("Error executing fetch query: %s"), PQresultErrorMessage(state->fetchres));
2014  PQclear(state->fetchres);
2015  return SHPDUMPERERR;
2016  }
2017 
2018  state->curresrow = 0;
2019  state->currescount = PQntuples(state->fetchres);
2020  }
2021 
2022  /* Grab the id of the geo column if we have one */
2023  if (state->geo_col_name)
2024  geocolnum = PQfnumber(state->fetchres, "_geoX");
2025 
2026  /* Process the next record within the batch. First write out all of
2027  the non-geo fields */
2028  for (i = 0; i < state->fieldcount; i++)
2029  {
2030  /*
2031  * Transform NULL numbers to '0'
2032  * This is because the shapelib
2033  * won't easly take care of setting
2034  * nulls unless paying the acquisition
2035  * of a bug in long integer values
2036  */
2037  if (PQgetisnull(state->fetchres, state->curresrow, i))
2038  {
2039  val = nullDBFValue(state->dbffieldtypes[i]);
2040  }
2041  else
2042  {
2043  val = PQgetvalue(state->fetchres, state->curresrow, i);
2044  val = goodDBFValue(val, state->dbffieldtypes[i]);
2045  }
2046 
2047  /* Write it to the DBF file */
2048  if (!DBFWriteAttributeDirectly(state->dbf, state->currow, i, val))
2049  {
2050  snprintf(state->message, SHPDUMPERMSGLEN, _("Error: record %d could not be created"), state->currow);
2051  PQclear(state->fetchres);
2052  return SHPDUMPERERR;
2053  }
2054  }
2055 
2056  /* Now process the geo field, if present */
2057  if (state->geo_col_name)
2058  {
2059  /* Handle NULL shapes */
2060  if (PQgetisnull(state->fetchres, state->curresrow, geocolnum))
2061  {
2062  obj = SHPCreateSimpleObject(SHPT_NULL, 0, NULL, NULL, NULL);
2063  if (SHPWriteObject(state->shp, -1, obj) == -1)
2064  {
2065  snprintf(state->message, SHPDUMPERMSGLEN, _("Error writing NULL shape for record %d"), state->currow);
2066  PQclear(state->fetchres);
2067  SHPDestroyObject(obj);
2068  return SHPDUMPERERR;
2069  }
2070  SHPDestroyObject(obj);
2071  }
2072  else
2073  {
2074  /* Get the value from the result set */
2075  val = PQgetvalue(state->fetchres, state->curresrow, geocolnum);
2076 
2077  if (!state->config->binary)
2078  {
2079  if (state->pgis_major_version > 0)
2080  {
2081  LWDEBUG(4, "PostGIS >= 1.0, non-binary cursor");
2082 
2083  /* Input is bytea encoded text field, so it must be unescaped and
2084  then converted to hexewkb string */
2085  hexewkb_binary = PQunescapeBytea((unsigned char *)val, &hexewkb_len);
2086  hexewkb = convert_bytes_to_hex(hexewkb_binary, hexewkb_len);
2087  }
2088  else
2089  {
2090  LWDEBUG(4, "PostGIS < 1.0, non-binary cursor");
2091 
2092  /* Input is already hexewkb string, so we can just
2093  copy directly to hexewkb */
2094  hexewkb_len = PQgetlength(state->fetchres, state->curresrow, geocolnum);
2095  hexewkb = malloc(hexewkb_len + 1);
2096  strncpy(hexewkb, val, hexewkb_len + 1);
2097  }
2098  }
2099  else /* binary */
2100  {
2101  LWDEBUG(4, "PostGIS (any version) using binary cursor");
2102 
2103  /* Input is binary field - must convert to hexewkb string */
2104  hexewkb_len = PQgetlength(state->fetchres, state->curresrow, geocolnum);
2105  hexewkb = convert_bytes_to_hex((unsigned char *)val, hexewkb_len);
2106  }
2107 
2108  LWDEBUGF(4, "HexEWKB - length: %d value: %s", strlen(hexewkb), hexewkb);
2109 
2110  /* Deserialize the LWGEOM */
2111  lwgeom = lwgeom_from_hexwkb(hexewkb, LW_PARSER_CHECK_NONE);
2112  if (!lwgeom)
2113  {
2114  snprintf(state->message, SHPDUMPERMSGLEN, _("Error parsing HEXEWKB for record %d"), state->currow);
2115  PQclear(state->fetchres);
2116  free(hexewkb);
2117  return SHPDUMPERERR;
2118  }
2119 
2120  /* Call the relevant method depending upon the geometry type */
2121  LWDEBUGF(4, "geomtype: %s\n", lwtype_name(lwgeom->type));
2122 
2123  switch (lwgeom->type)
2124  {
2125  case POINTTYPE:
2126  if (lwgeom_is_empty(lwgeom))
2127  {
2128  obj = create_point_empty(state, lwgeom_as_lwpoint(lwgeom));
2129  }
2130  else
2131  {
2132  obj = create_point(state, lwgeom_as_lwpoint(lwgeom));
2133  }
2134  break;
2135 
2136  case MULTIPOINTTYPE:
2137  obj = create_multipoint(state, lwgeom_as_lwmpoint(lwgeom));
2138  break;
2139 
2140  case POLYGONTYPE:
2141  obj = create_polygon(state, lwgeom_as_lwpoly(lwgeom));
2142  break;
2143 
2144  case MULTIPOLYGONTYPE:
2145  obj = create_multipolygon(state, lwgeom_as_lwmpoly(lwgeom));
2146  break;
2147 
2148  case LINETYPE:
2149  obj = create_linestring(state, lwgeom_as_lwline(lwgeom));
2150  break;
2151 
2152  case MULTILINETYPE:
2153  obj = create_multilinestring(state, lwgeom_as_lwmline(lwgeom));
2154  break;
2155 
2156  default:
2157  snprintf(state->message, SHPDUMPERMSGLEN, _("Unknown WKB type (%d) for record %d"), lwgeom->type, state->currow);
2158  PQclear(state->fetchres);
2159  SHPDestroyObject(obj);
2160  return SHPDUMPERERR;
2161  }
2162 
2163  /* Free both the original and geometries */
2164  lwgeom_free(lwgeom);
2165 
2166  /* Write the shape out to the file */
2167  if (SHPWriteObject(state->shp, -1, obj) == -1)
2168  {
2169  snprintf(state->message, SHPDUMPERMSGLEN, _("Error writing shape %d"), state->currow);
2170  PQclear(state->fetchres);
2171  SHPDestroyObject(obj);
2172  return SHPDUMPERERR;
2173  }
2174 
2175  SHPDestroyObject(obj);
2176 
2177  /* Free the hexewkb (and temporary bytea unescaped string if used) */
2178  if (hexewkb) free(hexewkb);
2179  if (hexewkb_binary) PQfreemem(hexewkb_binary);
2180  }
2181  }
2182 
2183  /* Increment ready for next time */
2184  state->curresrow++;
2185  state->currow++;
2186 
2187  return SHPDUMPEROK;
2188 }
int SHPAPI_CALL DBFWriteAttributeDirectly(DBFHandle psDBF, int hEntity, int iField, void *pValue)
Definition: dbfopen.c:1410
LWLINE * lwgeom_as_lwline(const LWGEOM *lwgeom)
Definition: lwgeom.c:170
LWGEOM * lwgeom_from_hexwkb(const char *hexwkb, const char check)
Definition: lwin_wkb.c:809
LWMPOINT * lwgeom_as_lwmpoint(const LWGEOM *lwgeom)
Definition: lwgeom.c:233
void lwgeom_free(LWGEOM *geom)
Definition: lwgeom.c:1144
#define LW_PARSER_CHECK_NONE
Definition: liblwgeom.h:2005
#define MULTILINETYPE
Definition: liblwgeom.h:89
#define LINETYPE
Definition: liblwgeom.h:86
#define MULTIPOINTTYPE
Definition: liblwgeom.h:88
LWMPOLY * lwgeom_as_lwmpoly(const LWGEOM *lwgeom)
Definition: lwgeom.c:251
#define POINTTYPE
LWTYPE numbers, used internally by PostGIS.
Definition: liblwgeom.h:85
LWPOINT * lwgeom_as_lwpoint(const LWGEOM *lwgeom)
Definition: lwgeom.c:161
LWMLINE * lwgeom_as_lwmline(const LWGEOM *lwgeom)
Definition: lwgeom.c:242
#define MULTIPOLYGONTYPE
Definition: liblwgeom.h:90
#define POLYGONTYPE
Definition: liblwgeom.h:87
const char * lwtype_name(uint8_t type)
Return the type name string associated with a type number (e.g.
Definition: lwutil.c:218
int lwgeom_is_empty(const LWGEOM *geom)
Return true or false depending on whether a geometry is an "empty" geometry (no vertices members)
Definition: lwgeom.c:1393
LWPOLY * lwgeom_as_lwpoly(const LWGEOM *lwgeom)
Definition: lwgeom.c:206
#define LWDEBUG(level, msg)
Definition: lwgeom_log.h:83
#define LWDEBUGF(level, msg,...)
Definition: lwgeom_log.h:88
void * malloc(YYSIZE_T)
void free(void *)
static SHPObject * create_polygon(SHPDUMPERSTATE *state, LWPOLY *lwpolygon)
static SHPObject * create_multipolygon(SHPDUMPERSTATE *state, LWMPOLY *lwmultipolygon)
char * convert_bytes_to_hex(uint8_t *ewkb, size_t size)
Binary to hexewkb conversion function.
static SHPObject * create_linestring(SHPDUMPERSTATE *state, LWLINE *lwlinestring)
static char * goodDBFValue(char *in, char fieldType)
Make appropriate formatting of a DBF value based on type.
static SHPObject * create_point(SHPDUMPERSTATE *state, LWPOINT *lwpoint)
static SHPObject * create_point_empty(SHPDUMPERSTATE *state, LWPOINT *lwpoint)
static char * nullDBFValue(char fieldType)
static SHPObject * create_multipoint(SHPDUMPERSTATE *state, LWMPOINT *lwmultipoint)
static SHPObject * create_multilinestring(SHPDUMPERSTATE *state, LWMLINE *lwmultilinestring)
#define SHPDUMPERMSGLEN
#define SHPDUMPEROK
#define SHPDUMPERERR
#define SHPT_NULL
Definition: shapefil.h:306
void SHPAPI_CALL SHPDestroyObject(SHPObject *psObject)
Definition: shpopen.c:2182
#define _(String)
Definition: shpcommon.h:24
SHPObject SHPAPI_CALL1 * SHPCreateSimpleObject(int nSHPType, int nVertices, const double *padfX, const double *padfY, const double *padfZ){ return(SHPCreateObject(nSHPType, -1, 0, NULL, NULL, nVertices, padfX, padfY, padfZ, NULL)
int SHPAPI_CALL SHPWriteObject(SHPHandle psSHP, int nShapeId, SHPObject *psObject)
Definition: shpopen.c:1170
uint8_t type
Definition: liblwgeom.h:399
SHPDUMPERCONFIG * config
PGresult * fetchres
char message[SHPDUMPERMSGLEN]

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().

Here is the call graph for this function:
Here is the caller graph for this function: