Datum LWGEOM_from_text(PG_FUNCTION_ARGS)
{
text *wkttext = PG_GETARG_TEXT_P(0);
char *wkt = text2cstring(wkttext);
LWGEOM_PARSER_RESULT lwg_parser_result;
GSERIALIZED *geom_result = NULL;
LWGEOM *lwgeom;
POSTGIS_DEBUG(2, "LWGEOM_from_text");
POSTGIS_DEBUGF(3, "wkt: [%s]", wkt);
if (lwgeom_parse_wkt(&lwg_parser_result, wkt, LW_PARSER_CHECK_ALL) == LW_FAILURE)
PG_PARSER_ERROR(lwg_parser_result);
lwgeom = lwg_parser_result.geom;
if ( lwgeom->srid != SRID_UNKNOWN )
{
elog(WARNING, "OGC WKT expected, EWKT provided - use GeomFromEWKT() for this");
}
/* read user-requested SRID if any */
if ( PG_NARGS() > 1 )
lwgeom_set_srid(lwgeom, PG_GETARG_INT32(1));
geom_result = geometry_serialize(lwgeom);
lwgeom_parser_result_free(&lwg_parser_result);
PG_RETURN_POINTER(geom_result);
}
static LWGEOM* lwgeom_from_text(const char *str)
{
LWGEOM_PARSER_RESULT r;
if( LW_FAILURE == lwgeom_parse_wkt(&r, (char*)str, LW_PARSER_CHECK_NONE) )
return NULL;
return r.geom;
}
/*
* Public function used for easy access to the parser.
*/
LWGEOM *lwgeom_from_wkt(const char *wkt, const char check)
{
LWGEOM_PARSER_RESULT r;
if( LW_FAILURE == lwgeom_parse_wkt(&r, (char*)wkt, check) )
lwerror(r.message);
return r.geom;
}
static void cu_wkb_malformed_in(char *hex)
{
LWGEOM_PARSER_RESULT p;
int rv = 0;
rv = lwgeom_parse_wkt(&p, hex, 0);
CU_ASSERT( LW_FAILURE == rv );
CU_ASSERT( p.errcode );
CU_ASSERT( ! p.geom );
lwgeom_parser_result_free(&p);
}
/*
* Return a char* that results from taking the input
* WKT, creating an LWGEOM, then writing it back out
* as an output WKT using the supplied variant.
* If there is an error, return that.
*/
static char* cu_wkt_in(char *wkt, uint8_t variant)
{
LWGEOM_PARSER_RESULT p;
int rv = 0;
char *s = 0;
rv = lwgeom_parse_wkt(&p, wkt, 0);
if( p.errcode ) {
return strdup(p.message);
}
s = lwgeom_to_wkt(p.geom, variant, 8, NULL);
lwgeom_parser_result_free(&p);
return s;
}
Datum geography_in(PG_FUNCTION_ARGS)
{
char *str = PG_GETARG_CSTRING(0);
/* Datum geog_oid = PG_GETARG_OID(1); Not needed. */
int32 geog_typmod = -1;
LWGEOM_PARSER_RESULT lwg_parser_result;
LWGEOM *lwgeom = NULL;
GSERIALIZED *g_ser = NULL;
if ( (PG_NARGS()>2) && (!PG_ARGISNULL(2)) ) {
geog_typmod = PG_GETARG_INT32(2);
}
lwgeom_parser_result_init(&lwg_parser_result);
/* Empty string. */
if ( str[0] == '\0' )
ereport(ERROR,(errmsg("parse error - invalid geometry")));
/* WKB? Let's find out. */
if ( str[0] == '0' )
{
/* TODO: 20101206: No parser checks! This is inline with current 1.5 behavior, but needs discussion */
lwgeom = lwgeom_from_hexwkb(str, LW_PARSER_CHECK_NONE);
/* Error out if something went sideways */
if ( ! lwgeom )
ereport(ERROR,(errmsg("parse error - invalid geometry")));
}
/* WKT then. */
else
{
if ( lwgeom_parse_wkt(&lwg_parser_result, str, LW_PARSER_CHECK_ALL) == LW_FAILURE )
PG_PARSER_ERROR(lwg_parser_result);
lwgeom = lwg_parser_result.geom;
}
/* Error on any SRID != default */
srid_is_latlong(fcinfo, lwgeom->srid);
/* Convert to gserialized */
g_ser = gserialized_geography_from_lwgeom(lwgeom, geog_typmod);
/* Clean up temporary object */
lwgeom_free(lwgeom);
PG_RETURN_POINTER(g_ser);
}
static void test_wkt_in_errlocation(void)
{
LWGEOM_PARSER_RESULT p;
int rv = 0;
char *wkt = 0;
wkt = "LINESTRING((0 0 0,1 1)";
lwgeom_parser_result_init(&p);
rv = lwgeom_parse_wkt(&p, wkt, LW_PARSER_CHECK_ALL);
CU_ASSERT_EQUAL(12,p.errlocation);
lwgeom_parser_result_free(&p);
// printf("errlocation %d\n", p.errlocation);
// printf("message %s\n", p.message);
}
static void cu_wkb_in(char *wkt)
{
LWGEOM_PARSER_RESULT pr;
LWGEOM *g_a, *g_b;
uint8_t *wkb_a, *wkb_b;
size_t wkb_size_a, wkb_size_b;
/* int i; char *hex; */
if ( hex_a ) free(hex_a);
if ( hex_b ) free(hex_b);
/* Turn WKT into geom */
lwgeom_parse_wkt(&pr, wkt, LW_PARSER_CHECK_NONE);
if ( pr.errcode )
{
printf("ERROR: %s\n", pr.message);
printf("POSITION: %d\n", pr.errlocation);
exit(0);
}
/* Get the geom */
g_a = pr.geom;
/* Turn geom into WKB */
wkb_a = lwgeom_to_wkb(g_a, WKB_NDR | WKB_EXTENDED, &wkb_size_a);
/* Turn WKB back into geom */
g_b = lwgeom_from_wkb(wkb_a, wkb_size_a, LW_PARSER_CHECK_NONE);
/* Turn geom to WKB again */
wkb_b = lwgeom_to_wkb(g_b, WKB_NDR | WKB_EXTENDED, &wkb_size_b);
/* Turn geoms into WKB for comparisons */
hex_a = hexbytes_from_bytes(wkb_a, wkb_size_a);
hex_b = hexbytes_from_bytes(wkb_b, wkb_size_b);
/* Clean up */
lwfree(wkb_a);
lwfree(wkb_b);
lwgeom_parser_result_free(&pr);
lwgeom_free(g_b);
}
Datum geography_from_text(PG_FUNCTION_ARGS)
{
LWGEOM_PARSER_RESULT lwg_parser_result;
GSERIALIZED *g_ser = NULL;
text *wkt_text = PG_GETARG_TEXT_P(0);
/* Extract the cstring from the varlena */
char *wkt = text2cstring(wkt_text);
/* Pass the cstring to the input parser, and magic occurs! */
if ( lwgeom_parse_wkt(&lwg_parser_result, wkt, LW_PARSER_CHECK_ALL) == LW_FAILURE )
PG_PARSER_ERROR(lwg_parser_result);
/* Clean up string */
pfree(wkt);
g_ser = gserialized_geography_from_lwgeom(lwg_parser_result.geom, -1);
/* Clean up temporary object */
lwgeom_free(lwg_parser_result.geom);
PG_RETURN_POINTER(g_ser);
}
Datum LWGEOM_in(PG_FUNCTION_ARGS)
{
char *input = PG_GETARG_CSTRING(0);
int32 geom_typmod = -1;
char *str = input;
LWGEOM_PARSER_RESULT lwg_parser_result;
LWGEOM *lwgeom;
GSERIALIZED *ret;
int srid = 0;
if ( (PG_NARGS()>2) && (!PG_ARGISNULL(2)) ) {
geom_typmod = PG_GETARG_INT32(2);
}
lwgeom_parser_result_init(&lwg_parser_result);
/* Empty string. */
if ( str[0] == '\0' )
ereport(ERROR,(errmsg("parse error - invalid geometry")));
/* Starts with "SRID=" */
if( strncasecmp(str,"SRID=",5) == 0 )
{
/* Roll forward to semi-colon */
char *tmp = str;
while ( tmp && *tmp != ';' )
tmp++;
/* Check next character to see if we have WKB */
if ( tmp && *(tmp+1) == '0' )
{
/* Null terminate the SRID= string */
*tmp = '\0';
/* Set str to the start of the real WKB */
str = tmp + 1;
/* Move tmp to the start of the numeric part */
tmp = input + 5;
/* Parse out the SRID number */
srid = atoi(tmp);
}
}
/* WKB? Let's find out. */
if ( str[0] == '0' )
{
size_t hexsize = strlen(str);
unsigned char *wkb = bytes_from_hexbytes(str, hexsize);
/* TODO: 20101206: No parser checks! This is inline with current 1.5 behavior, but needs discussion */
lwgeom = lwgeom_from_wkb(wkb, hexsize/2, LW_PARSER_CHECK_NONE);
/* If we picked up an SRID at the head of the WKB set it manually */
if ( srid ) lwgeom_set_srid(lwgeom, srid);
/* Add a bbox if necessary */
if ( lwgeom_needs_bbox(lwgeom) ) lwgeom_add_bbox(lwgeom);
pfree(wkb);
ret = geometry_serialize(lwgeom);
lwgeom_free(lwgeom);
}
/* WKT then. */
else
{
if ( lwgeom_parse_wkt(&lwg_parser_result, str, LW_PARSER_CHECK_ALL) == LW_FAILURE )
{
PG_PARSER_ERROR(lwg_parser_result);
}
lwgeom = lwg_parser_result.geom;
if ( lwgeom_needs_bbox(lwgeom) )
lwgeom_add_bbox(lwgeom);
ret = geometry_serialize(lwgeom);
lwgeom_parser_result_free(&lwg_parser_result);
}
if ( geom_typmod >= 0 )
{
postgis_valid_typmod(ret, geom_typmod);
POSTGIS_DEBUG(3, "typmod and geometry were consistent");
}
else
{
POSTGIS_DEBUG(3, "typmod was -1");
}
/* Don't free the parser result (and hence lwgeom) until we have done */
/* the typemod check with lwgeom */
PG_RETURN_POINTER(ret);
}
Datum geography_in(PG_FUNCTION_ARGS)
{
char *str = PG_GETARG_CSTRING(0);
/* Datum geog_oid = PG_GETARG_OID(1); Not needed. */
int32 geog_typmod = PG_GETARG_INT32(2);
LWGEOM_PARSER_RESULT lwg_parser_result;
LWGEOM *lwgeom = NULL;
GSERIALIZED *g_ser = NULL;
lwgeom_parser_result_init(&lwg_parser_result);
/* Empty string. */
if ( str[0] == '\0' )
ereport(ERROR,(errmsg("parse error - invalid geometry")));
/* WKB? Let's find out. */
if ( str[0] == '0' )
{
/* TODO: 20101206: No parser checks! This is inline with current 1.5 behavior, but needs discussion */
lwgeom = lwgeom_from_hexwkb(str, LW_PARSER_CHECK_NONE);
/* Error out if something went sideways */
if ( ! lwgeom )
ereport(ERROR,(errmsg("parse error - invalid geometry")));
}
/* WKT then. */
else
{
if ( lwgeom_parse_wkt(&lwg_parser_result, str, LW_PARSER_CHECK_ALL) == LW_FAILURE )
PG_PARSER_ERROR(lwg_parser_result);
lwgeom = lwg_parser_result.geom;
}
/* Set geodetic flag */
lwgeom_set_geodetic(lwgeom, true);
/* Check that this is a type we can handle */
geography_valid_type(lwgeom->type);
/* Check that the coordinates are in range */
if ( lwgeom_check_geodetic(lwgeom) == LW_FALSE )
{
ereport(ERROR, (
errcode(ERRCODE_INVALID_PARAMETER_VALUE),
errmsg("Coordinate values are out of range [-180 -90, 180 90] for GEOGRAPHY type" )));
}
/* Force default SRID to the default */
if ( (int)lwgeom->srid <= 0 )
lwgeom->srid = SRID_DEFAULT;
if ( geog_typmod >= 0 )
{
postgis_valid_typmod(lwgeom, geog_typmod);
POSTGIS_DEBUG(3, "typmod and geometry were consistent");
}
else
{
POSTGIS_DEBUG(3, "typmod was -1");
}
/*
** Serialize our lwgeom and set the geodetic flag so subsequent
** functions do the right thing.
*/
g_ser = geography_serialize(lwgeom);
/* Clean up temporary object */
lwgeom_free(lwgeom);
PG_RETURN_POINTER(g_ser);
}