pgis_union_geometry_array()

Datum pgis_union_geometry_array

(

PG_FUNCTION_ARGS

)

Definition at line 527 of file postgis/lwgeom_geos.c.

{
        ArrayType *array;
 
        ArrayIterator iterator;
        Datum value;
        bool isnull;
 
        int is3d = LW_FALSE, gotsrid = LW_FALSE;
        int nelems = 0, geoms_size = 0, curgeom = 0, count = 0;
 
        GSERIALIZED *gser_out = NULL;
 
        GEOSGeometry *g = NULL;
        GEOSGeometry *g_union = NULL;
        GEOSGeometry **geoms = NULL;
 
        int32_t srid = SRID_UNKNOWN;
 
        int empty_type = 0;
 
        /* Null array, null geometry (should be empty?) */
        if ( PG_ARGISNULL(0) )
                PG_RETURN_NULL();
 
        array = PG_GETARG_ARRAYTYPE_P(0);
        nelems = ArrayGetNItems(ARR_NDIM(array), ARR_DIMS(array));
 
        /* Empty array? Null return */
        if ( nelems == 0 ) PG_RETURN_NULL();
 
        /* Quick scan for nulls */
        iterator = array_create_iterator(array, 0, NULL);
        while (array_iterate(iterator, &value, &isnull))
        {
                /* Skip null array items */
                if (isnull) continue;
                count++;
        }
        array_free_iterator(iterator);
 
 
        /* All-nulls? Return null */
        if ( count == 0 )
                PG_RETURN_NULL();
 
        /* Ok, we really need GEOS now ;) */
        initGEOS(lwpgnotice, lwgeom_geos_error);
 
        /* One geom, good geom? Return it */
        if ( count == 1 && nelems == 1 )
        {
#if __GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 6)
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wsign-compare"
#endif
                g = POSTGIS2GEOS((GSERIALIZED *)(ARR_DATA_PTR(array)));
#if __GNUC__ > 4 || (__GNUC__ == 4 && __GNUC_MINOR__ >= 6)
#pragma GCC diagnostic pop
#endif
                srid = GEOSGetSRID(g);
                g_union = GEOSUnaryUnion(g);
                GEOSGeom_destroy(g);
                if (!g_union) HANDLE_GEOS_ERROR("GEOSUnaryUnion");
                GEOSSetSRID(g_union, srid);
                gser_out = GEOS2POSTGIS(g_union, is3d);
                GEOSGeom_destroy(g_union);
                PG_RETURN_POINTER(gser_out);
        }
 
        /*
        ** Collect the non-empty inputs and stuff them into a GEOS collection
        */
        geoms_size = nelems;
        geoms = palloc(sizeof(GEOSGeometry*) * geoms_size);
 
        /*
        ** We need to convert the array of GSERIALIZED into a GEOS collection.
        ** First make an array of GEOS geometries.
        */
        iterator = array_create_iterator(array, 0, NULL);
        while (array_iterate(iterator, &value, &isnull))
        {
                GSERIALIZED *gser_in;
 
                /* Skip null array items */
                if (isnull) continue;
                gser_in = (GSERIALIZED *)DatumGetPointer(value);
 
                /* Check for SRID mismatch in array elements */
                if ( gotsrid )
                        gserialized_error_if_srid_mismatch_reference(gser_in, srid, __func__);
                else
                {
                        /* Initialize SRID/dimensions info */
                        srid = gserialized_get_srid(gser_in);
                        is3d = gserialized_has_z(gser_in);
                        gotsrid = 1;
                }
 
                /* Don't include empties in the union */
                if ( gserialized_is_empty(gser_in) )
                {
                        int gser_type = gserialized_get_type(gser_in);
                        if (gser_type > empty_type)
                        {
                                empty_type = gser_type;
                                POSTGIS_DEBUGF(4, "empty_type = %d  gser_type = %d", empty_type, gser_type);
                        }
                }
                else
                {
                        g = POSTGIS2GEOS(gser_in);
 
                        /* Uh oh! Exception thrown at construction... */
                        if ( ! g )
                        {
                                HANDLE_GEOS_ERROR(
                                    "One of the geometries in the set "
                                    "could not be converted to GEOS");
                        }
 
                        /* Ensure we have enough space in our storage array */
                        if ( curgeom == geoms_size )
                        {
                                geoms_size *= 2;
                                geoms = repalloc( geoms, sizeof(GEOSGeometry*) * geoms_size );
                        }
 
                        geoms[curgeom] = g;
                        curgeom++;
                }
 
        }
        array_free_iterator(iterator);
 
        /*
        ** Take our GEOS geometries and turn them into a GEOS collection,
        ** then pass that into cascaded union.
        */
        if (curgeom > 0)
        {
                g = GEOSGeom_createCollection(GEOS_GEOMETRYCOLLECTION, geoms, curgeom);
                if (!g) HANDLE_GEOS_ERROR("Could not create GEOS COLLECTION from geometry array");
 
                g_union = GEOSUnaryUnion(g);
                GEOSGeom_destroy(g);
                if (!g_union) HANDLE_GEOS_ERROR("GEOSUnaryUnion");
 
                GEOSSetSRID(g_union, srid);
                gser_out = GEOS2POSTGIS(g_union, is3d);
                GEOSGeom_destroy(g_union);
        }
        /* No real geometries in our array, any empties? */
        else
        {
                /* If it was only empties, we'll return the largest type number */
                if ( empty_type > 0 )
                {
                        PG_RETURN_POINTER(geometry_serialize(lwgeom_construct_empty(empty_type, srid, is3d, 0)));
                }
                /* Nothing but NULL, returns NULL */
                else
                {
                        PG_RETURN_NULL();
                }
        }
 
        if ( ! gser_out )
        {
                /* Union returned a NULL geometry */
                PG_RETURN_NULL();
        }
 
        PG_RETURN_POINTER(gser_out);
}

References GEOS2POSTGIS(), gserialized_error_if_srid_mismatch_reference(), gserialized_get_srid(), gserialized_get_type(), gserialized_has_z(), gserialized_is_empty(), HANDLE_GEOS_ERROR, LW_FALSE, lwgeom_construct_empty(), lwgeom_geos_error(), POSTGIS2GEOS(), and SRID_UNKNOWN.

Here is the call graph for this function: