postgis/liblwgeom/cunit/cu_in_twkb.c
Raúl Marín 591d0c69ba Revert incorrect "Optimize varlena returning functions"
This reverts commit 0696fa4669.

I'm not sure what I did there, it seems like I pushed smaller commits
instead of a squashed one, and then did a revert that undid everything.

References #4623
2020-04-28 19:13:38 +02:00

243 lines
6.6 KiB
C

/**********************************************************************
*
* PostGIS - Spatial Types for PostgreSQL
* http://postgis.net
* Copyright 2010 Paul Ramsey <pramsey@cleverelephant.ca>
*
* This is free software; you can redistribute and/or modify it under
* the terms of the GNU General Public Licence. See the COPYING file.
*
**********************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "CUnit/Basic.h"
#include "liblwgeom_internal.h"
#include "cu_tester.h"
/*
** Global variable to hold TWKB strings
*/
static char *hex_a;
static char *hex_b;
static uint8_t precision = 0;
static uint8_t variant = 0;
/*
** The suite initialization function.
** Create any re-used objects.
*/
static int init_twkb_in_suite(void)
{
hex_a = NULL;
hex_b = NULL;
return 0;
}
/*
** The suite cleanup function.
** Frees any global objects.
*/
static int clean_twkb_in_suite(void)
{
if (hex_a) free(hex_a);
if (hex_b) free(hex_b);
hex_a = NULL;
hex_b = NULL;
return 0;
}
static void cu_twkb_in(char *wkt)
{
LWGEOM_PARSER_RESULT pr;
LWGEOM *g_a, *g_b;
/* int i; char *hex; */
/* 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 TWKB */
lwvarlena_t *twkb_a = lwgeom_to_twkb(g_a, variant, precision, precision, precision);
// printf("\n Size: %ld\n", twkb_size_a);
/* Turn TWKB back into geom */
g_b = lwgeom_from_twkb((uint8_t *)twkb_a->data, LWSIZE_GET(twkb_a->size) - LWVARHDRSZ, LW_PARSER_CHECK_NONE);
// printf("\n Org: %s\n 1st: %s\n 2nd: %s\n", wkt, lwgeom_to_ewkt(g_a), lwgeom_to_ewkt(g_b));
/* Turn geom to TWKB again */
lwvarlena_t *twkb_b = lwgeom_to_twkb(g_b, variant, precision, precision, precision);
/* Turn TWKB into hex for comparisons */
if ( hex_a ) lwfree(hex_a);
if ( hex_b ) lwfree(hex_b);
hex_a = hexbytes_from_bytes((uint8_t *)twkb_a->data, LWSIZE_GET(twkb_a->size) - LWVARHDRSZ);
hex_b = hexbytes_from_bytes((uint8_t *)twkb_b->data, LWSIZE_GET(twkb_b->size) - LWVARHDRSZ);
/* Clean up */
lwfree(twkb_a);
lwfree(twkb_b);
lwgeom_parser_result_free(&pr);
lwgeom_free(g_b);
}
static void test_twkb_in_point(void)
{
cu_twkb_in("POINT(0 0 0 0)");
// printf("old: %s\nnew: %s\n",hex_a, hex_b);
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("POINT(1 1)");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("POINT EMPTY");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
}
static void test_twkb_in_linestring(void)
{
cu_twkb_in("LINESTRING(0 0,1 1)");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("LINESTRING(0 0 1,1 1 2,2 2 3)");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("LINESTRING EMPTY");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
}
static void test_twkb_in_polygon(void)
{
cu_twkb_in("POLYGON((0 0 0,0 1 0,1 1 0,1 0 0,0 0 0))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("POLYGON((0 0 0 1,0 1 0 2,1 1 0 3,1 0 0 4,0 0 0 5))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("POLYGON((0 0 0 1,0 1 0 2,1 1 0 3,1 0 0 4,0 0 0 5))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("POLYGON EMPTY");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
}
static void test_twkb_in_multipoint(void)
{
cu_twkb_in("MULTIPOINT Z EMPTY");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("MULTIPOINT(1 2, EMPTY, 5 6)");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
// printf(" 1st: %s\n 2nd: %s\n", hex_a, hex_b);
cu_twkb_in("MULTIPOINT(0 0 0,0 1 0,1 1 0,1 0 0,0 0 1)");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("MULTIPOINT(1 2 3, 1 2 3, 4 5 6, -3 -4 -5, -10 -5 -1)");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
}
static void test_twkb_in_multilinestring(void)
{
cu_twkb_in("MULTILINESTRING((0 0,0 1),(1 1, 10 10))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("MULTILINESTRING((0 0,0 1),EMPTY,(1 1, 10 10))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("MULTILINESTRING((0 0 200000,0 1 10),(1 100000000 23, 10 10 45))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
// printf(" 1st: %s\n 2nd: %s\n", hex_a, hex_b);
cu_twkb_in("MULTILINESTRING EMPTY");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
}
static void test_twkb_in_multipolygon(void)
{
cu_twkb_in("MULTIPOLYGON(((0 0 0,0 1 0,1 1 0,1 0 0,0 0 0)),((-1 -1 0,-1 2 0,2 2 0,2 -1 0,-1 -1 0),(0 0 0,0 1 0,1 1 0,1 0 0,0 0 0)))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
//printf("old: %s\nnew: %s\n",hex_a, hex_b);
cu_twkb_in("MULTIPOLYGON EMPTY");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
//printf("old: %s\nnew: %s\n",hex_a, hex_b);
}
static void test_twkb_in_collection(void)
{
cu_twkb_in("GEOMETRYCOLLECTION(MULTIPOLYGON(((0 0 0,0 1 0,1 1 0,1 0 0,0 0 0))),POLYGON((0 0 0,0 1 0,1 1 0,1 0 0,0 0 0)),POINT(1 1 1),LINESTRING(0 0 0, 1 1 1))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("GEOMETRYCOLLECTION(POLYGON((0 0 0,0 1 0,1 1 0,1 0 0,0 0 0)),POINT(1 1 1))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("GEOMETRYCOLLECTION EMPTY");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("GEOMETRYCOLLECTION(POINT(1 2 3), LINESTRING EMPTY, POINT(4 5 6))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("GEOMETRYCOLLECTION(POINT(1 2 3), POINT EMPTY, POINT(4 5 6))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
}
/*
** PRECISION TESTS HERE: We ALTER THE 'precision' GLOBAL
*/
static void test_twkb_in_precision(void)
{
/* Try these cases at several different precisions */
for ( precision = 1; precision <= 6; precision++ )
{
cu_twkb_in("MULTILINESTRING((0 0,0 1),EMPTY,(1 1, 10 10))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("MULTIPOLYGON(((0 0 0,0 1 0,1 1 0,1 0 0,0 0 0)),((-1 -1 0,-1 2 0,2 2 0,2 -1 0,-1 -1 0),(0 0 0,0 1 0,1 1 0,1 0 0,0 0 0)))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("GEOMETRYCOLLECTION(POLYGON((0 0 0,0 1 0,1 1 0,1 0 0,0 0 0)),POINT(1 1 1))");
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
cu_twkb_in("MULTILINESTRING((0 0 200000,0 1 10),(1 100000000 23, 10 10 45))");
// printf("old: %s\nnew: %s\n",hex_a, hex_b);
CU_ASSERT_STRING_EQUAL(hex_a, hex_b);
}
/* Go back to default precision */
precision = 0;
}
/*
** Used by test harness to register the tests in this file.
*/
void twkb_in_suite_setup(void);
void twkb_in_suite_setup(void)
{
CU_pSuite suite = CU_add_suite("twkb_input", init_twkb_in_suite, clean_twkb_in_suite);
PG_ADD_TEST(suite, test_twkb_in_point);
PG_ADD_TEST(suite, test_twkb_in_linestring);
PG_ADD_TEST(suite, test_twkb_in_polygon);
PG_ADD_TEST(suite, test_twkb_in_multipoint);
PG_ADD_TEST(suite, test_twkb_in_multilinestring);
PG_ADD_TEST(suite, test_twkb_in_multipolygon);
PG_ADD_TEST(suite, test_twkb_in_collection);
PG_ADD_TEST(suite, test_twkb_in_precision);
}