PostGIS  2.3.7dev-r@@SVN_REVISION@@
Datum TWKBFromLWGEOMArray ( PG_FUNCTION_ARGS  )

Definition at line 524 of file lwgeom_inout.c.

References COLLECTIONTYPE, lwcollection_add_lwgeom(), lwcollection_as_lwgeom(), lwcollection_construct_empty(), lwcollection_free(), lwgeom_from_gserialized(), lwgeom_get_srid(), lwgeom_get_type(), lwgeom_has_m(), lwgeom_has_z(), lwgeom_to_twkb_with_idlist(), lwtype_get_collectiontype(), TWKB_BBOX, TWKB_DEFAULT_PRECISION, TWKB_ID, TWKB_SIZE, LWCOLLECTION::type, and variant.

525 {
526  ArrayType *arr_geoms = NULL;
527  ArrayType *arr_ids = NULL;
528  int num_geoms, num_ids, i = 0;
529 
530  ArrayIterator iter_geoms, iter_ids;
531  bool null_geom, null_id;
532  Datum val_geom, val_id;
533 
534  int is_homogeneous = true;
535  int subtype = 0;
536  int has_z = 0;
537  int has_m = 0;
538  LWCOLLECTION *col = NULL;
539  int64_t *idlist = NULL;
540  uint8_t variant = 0;
541 
542  srs_precision sp;
543  uint8_t *twkb;
544  size_t twkb_size;
545  bytea *result;
546 
547  /* The first two arguments are required */
548  if ( PG_NARGS() < 2 || PG_ARGISNULL(0) || PG_ARGISNULL(1) )
549  PG_RETURN_NULL();
550 
551  arr_geoms = PG_GETARG_ARRAYTYPE_P(0);
552  arr_ids = PG_GETARG_ARRAYTYPE_P(1);
553 
554  num_geoms = ArrayGetNItems(ARR_NDIM(arr_geoms), ARR_DIMS(arr_geoms));
555  num_ids = ArrayGetNItems(ARR_NDIM(arr_ids), ARR_DIMS(arr_ids));
556 
557  if ( num_geoms != num_ids )
558  {
559  elog(ERROR, "size of geometry[] and integer[] arrays must match");
560  PG_RETURN_NULL();
561  }
562 
563  /* Loop through array and build a collection of geometry and */
564  /* a simple array of ids. If either side is NULL, skip it */
565 
566 #if POSTGIS_PGSQL_VERSION >= 95
567  iter_geoms = array_create_iterator(arr_geoms, 0, NULL);
568  iter_ids = array_create_iterator(arr_ids, 0, NULL);
569 #else
570  iter_geoms = array_create_iterator(arr_geoms, 0);
571  iter_ids = array_create_iterator(arr_ids, 0);
572 #endif
573 
574  while( array_iterate(iter_geoms, &val_geom, &null_geom) &&
575  array_iterate(iter_ids, &val_id, &null_id) )
576  {
577  LWGEOM *geom;
578  int32_t uid;
579 
580  if ( null_geom || null_id )
581  {
582  elog(NOTICE, "ST_AsTWKB skipping NULL entry at position %d", i);
583  continue;
584  }
585 
586  geom = lwgeom_from_gserialized((GSERIALIZED*)DatumGetPointer(val_geom));
587  uid = DatumGetInt64(val_id);
588 
589  /* Construct collection/idlist first time through */
590  if ( ! col )
591  {
592  has_z = lwgeom_has_z(geom);
593  has_m = lwgeom_has_m(geom);
594  col = lwcollection_construct_empty(COLLECTIONTYPE, lwgeom_get_srid(geom), has_z, has_m);
595  }
596  if ( ! idlist )
597  idlist = palloc0(num_geoms * sizeof(int64_t));
598 
599 
600  /*Check if there is differences in dimmenstionality*/
601  if( lwgeom_has_z(geom)!=has_z || lwgeom_has_m(geom)!=has_m)
602  {
603  elog(ERROR, "Geometries have differenct dimensionality");
604  PG_FREE_IF_COPY(arr_geoms, 0);
605  PG_FREE_IF_COPY(arr_ids, 1);
606  PG_RETURN_NULL();
607  }
608  /* Store the values */
609  lwcollection_add_lwgeom(col, geom);
610  idlist[i++] = uid;
611 
612  /* Grab the geometry type and note if all geometries share it */
613  /* If so, we can make this a homogeneous collection and save some space */
614  if ( lwgeom_get_type(geom) != subtype && subtype )
615  {
616  is_homogeneous = false;
617  }
618  else
619  {
620  subtype = lwgeom_get_type(geom);
621  }
622 
623  }
624  array_free_iterator(iter_geoms);
625  array_free_iterator(iter_ids);
626 
627  if(i==0)
628  {
629  elog(NOTICE, "No valid geometry - id pairs found");
630  PG_FREE_IF_COPY(arr_geoms, 0);
631  PG_FREE_IF_COPY(arr_ids, 1);
632  PG_RETURN_NULL();
633  }
634  if ( is_homogeneous )
635  {
636  col->type = lwtype_get_collectiontype(subtype);
637  }
638 
639  /* Read sensible precision defaults (about one meter) given the srs */
640  sp = srid_axis_precision(fcinfo, lwgeom_get_srid(lwcollection_as_lwgeom(col)), TWKB_DEFAULT_PRECISION);
641 
642  /* If user specified XY precision, use it */
643  if ( PG_NARGS() > 2 && ! PG_ARGISNULL(2) )
644  sp.precision_xy = PG_GETARG_INT32(2);
645 
646  /* If user specified Z precision, use it */
647  if ( PG_NARGS() > 3 && ! PG_ARGISNULL(3) )
648  sp.precision_z = PG_GETARG_INT32(3);
649 
650  /* If user specified M precision, use it */
651  if ( PG_NARGS() > 4 && ! PG_ARGISNULL(4) )
652  sp.precision_m = PG_GETARG_INT32(4);
653 
654  /* We are building an ID'ed output */
655  variant = TWKB_ID;
656 
657  /* If user wants registered twkb sizes */
658  if ( PG_NARGS() > 5 && ! PG_ARGISNULL(5) && PG_GETARG_BOOL(5) )
659  variant |= TWKB_SIZE;
660 
661  /* If user wants bounding boxes */
662  if ( PG_NARGS() > 6 && ! PG_ARGISNULL(6) && PG_GETARG_BOOL(6) )
663  variant |= TWKB_BBOX;
664 
665  /* Write out the TWKB */
667  idlist, variant,
668  sp.precision_xy, sp.precision_z, sp.precision_m,
669  &twkb_size);
670 
671  /* Convert to a bytea return type */
672  result = palloc(twkb_size + VARHDRSZ);
673  memcpy(VARDATA(result), twkb, twkb_size);
674  SET_VARSIZE(result, twkb_size + VARHDRSZ);
675 
676  /* Clean up */
677  pfree(twkb);
678  pfree(idlist);
679  lwcollection_free(col);
680  PG_FREE_IF_COPY(arr_geoms, 0);
681  PG_FREE_IF_COPY(arr_ids, 1);
682 
683  PG_RETURN_BYTEA_P(result);
684 }
int lwtype_get_collectiontype(uint8_t type)
Given an lwtype number, what homogeneous collection can hold it?
Definition: lwgeom.c:1039
uint8_t * lwgeom_to_twkb_with_idlist(const LWGEOM *geom, int64_t *idlist, uint8_t variant, int8_t precision_xy, int8_t precision_z, int8_t precision_m, size_t *twkb_size)
Convert LWGEOM to a char* in TWKB format.
Definition: lwout_twkb.c:560
uint8_t variant
Definition: cu_in_twkb.c:26
LWGEOM * lwgeom_from_gserialized(const GSERIALIZED *g)
Allocate a new LWGEOM from a GSERIALIZED.
uint8_t type
Definition: liblwgeom.h:502
uint32_t lwgeom_get_type(const LWGEOM *geom)
Return LWTYPE number.
Definition: lwgeom.c:842
int32_t lwgeom_get_srid(const LWGEOM *geom)
Return SRID number.
Definition: lwgeom.c:835
int lwgeom_has_z(const LWGEOM *geom)
Return LW_TRUE if geometry has Z ordinates.
Definition: lwgeom.c:849
#define TWKB_ID
Definition: liblwgeom.h:2065
#define TWKB_DEFAULT_PRECISION
Definition: liblwgeom.h:2068
#define TWKB_BBOX
Definition: liblwgeom.h:2063
#define TWKB_SIZE
Definition: liblwgeom.h:2064
void lwcollection_free(LWCOLLECTION *col)
Definition: lwcollection.c:339
LWCOLLECTION * lwcollection_construct_empty(uint8_t type, int srid, char hasz, char hasm)
Definition: lwcollection.c:94
int lwgeom_has_m(const LWGEOM *geom)
Return LW_TRUE if geometry has M ordinates.
Definition: lwgeom.c:856
LWCOLLECTION * lwcollection_add_lwgeom(LWCOLLECTION *col, const LWGEOM *geom)
Appends geom to the collection managed by col.
Definition: lwcollection.c:187
#define COLLECTIONTYPE
Definition: liblwgeom.h:90
LWGEOM * lwcollection_as_lwgeom(const LWCOLLECTION *obj)
Definition: lwgeom.c:232

Here is the call graph for this function: