LWGEOM_dumpsegments()

Datum LWGEOM_dumpsegments

(

PG_FUNCTION_ARGS

)

Definition at line 301 of file lwgeom_dumppoints.c.

{
        FuncCallContext *funcctx;
        MemoryContext oldcontext, newcontext;
 
        GSERIALIZED *pglwgeom;
        LWCOLLECTION *lwcoll;
        LWGEOM *lwgeom;
        struct dumpstate *state;
        struct dumpnode *node;
 
        HeapTuple tuple;
        Datum pathpt[2];         /* used to construct the composite return value */
        bool isnull[2] = {0, 0}; /* needed to say neither value is null */
        Datum result;            /* the actual composite return value */
 
        if (SRF_IS_FIRSTCALL())
        {
                funcctx = SRF_FIRSTCALL_INIT();
 
                newcontext = funcctx->multi_call_memory_ctx;
                oldcontext = MemoryContextSwitchTo(newcontext);
 
                pglwgeom = PG_GETARG_GSERIALIZED_P_COPY(0);
                lwgeom = lwgeom_from_gserialized(pglwgeom);
 
                /* return early if nothing to do */
                if (!lwgeom || lwgeom_is_empty(lwgeom))
                {
                        MemoryContextSwitchTo(oldcontext);
                        funcctx = SRF_PERCALL_SETUP();
                        SRF_RETURN_DONE(funcctx);
                }
 
                /* Create function state */
                state = lwalloc(sizeof *state);
                state->root = lwgeom;
                state->stacklen = 0;
                state->pathlen = 0;
                state->pt = 0;
                state->ring = 0;
 
                funcctx->user_fctx = state;
 
                /*
                 * Push a struct dumpnode on the state stack
                 */
                state->stack[0].idx = 0;
                state->stack[0].geom = lwgeom;
                state->stacklen++;
 
                /*
                 * get tuple description for return type
                 */
                if (get_call_result_type(fcinfo, 0, &funcctx->tuple_desc) != TYPEFUNC_COMPOSITE)
                {
                        ereport(ERROR,
                                (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                                 errmsg("set-valued function called in context that cannot accept a set")));
                }
 
                BlessTupleDesc(funcctx->tuple_desc);
 
                /* get and cache data for constructing int4 arrays */
                get_typlenbyvalalign(INT4OID, &state->typlen, &state->byval, &state->align);
 
                MemoryContextSwitchTo(oldcontext);
        }
 
        /* stuff done on every call of the function */
        funcctx = SRF_PERCALL_SETUP();
        newcontext = funcctx->multi_call_memory_ctx;
 
        /* get state */
        state = funcctx->user_fctx;
 
        while (1)
        {
                POINTARRAY *points = NULL;
                LWLINE *line;
                LWTRIANGLE *tri;
                LWPOLY *poly;
                POINT4D pts_out[3];
                POINTARRAY *segment_pa;
                LWGEOM *segment;
                uint8_t stride = 1;
 
                node = &state->stack[state->stacklen - 1];
                lwgeom = node->geom;
 
                if (lwgeom->type == CIRCSTRINGTYPE)
                        stride =2 ;
 
                if ( !lwgeom_is_empty(lwgeom) && (lwgeom->type == LINETYPE || lwgeom->type == TRIANGLETYPE || lwgeom->type == POLYGONTYPE || lwgeom->type == CIRCSTRINGTYPE) )
                {
                        if (lwgeom->type == LINETYPE || lwgeom->type == CIRCSTRINGTYPE)
                        {
                                line = (LWLINE *)lwgeom;
 
                                if (state->pt < (line->points->npoints - stride))
                                {
                                        points = line->points;
                                }
                        }
                        else if (lwgeom->type == TRIANGLETYPE)
                        {
                                tri = (LWTRIANGLE *)lwgeom;
 
                                if (state->pt == 0)
                                {
                                        state->path[state->pathlen++] = Int32GetDatum(state->ring + 1);
                                }
 
                                if (state->pt < 3)
                                {
                                        points = tri->points;
                                }
                                else
                                {
                                        state->pathlen--;
                                }
                        }
                        else if (lwgeom->type == POLYGONTYPE)
                        {
                                poly = (LWPOLY *)lwgeom;
 
                                if (state->pt == poly->rings[state->ring]->npoints)
                                {
                                        state->pt = 0;
                                        state->ring++;
                                        state->pathlen--;
                                }
 
                                if (state->ring < poly->nrings)
                                {
                                        if (state->pt == 0)
                                        {
                                                state->path[state->pathlen] = Int32GetDatum(state->ring + 1);
                                                state->pathlen++;
                                        }
 
                                        if (state->pt < poly->rings[state->ring]->npoints - 1)
                                        {
                                                points = poly->rings[state->ring];
                                        }
                                        else
                                        {
                                                state->pt += stride;
                                                continue;
                                        }
                                }
                        }
 
                        if (points && ((state->pt % stride) == 0))
                        {
                                segment_pa = ptarray_construct(lwgeom_has_z(lwgeom), lwgeom_has_m(lwgeom), stride+1);
                                for (uint8_t i = 0; i < stride+1; i++)
                                {
                                        getPoint4d_p(points, state->pt+i, &pts_out[i]);
                                        ptarray_set_point4d(segment_pa, i, &pts_out[i]);
                                }
 
                                if (stride == 1)
                                        segment = lwline_as_lwgeom(lwline_construct(lwgeom->srid, NULL, segment_pa));
                                else if (stride == 2)
                                        segment = lwcircstring_as_lwgeom(lwcircstring_construct(lwgeom->srid, NULL, segment_pa));
                                else
                                        elog(ERROR, "%s got unexpected strid", __func__);
 
                                state->pt += stride;
 
                                state->path[state->pathlen] = Int32GetDatum(state->pt / stride);
                                pathpt[0] = PointerGetDatum(construct_array(state->path,
                                                                            state->pathlen + 1,
                                                                            INT4OID,
                                                                            state->typlen,
                                                                            state->byval,
                                                                            state->align));
                                pathpt[1] = PointerGetDatum(geometry_serialize((LWGEOM *)segment));
 
                                tuple = heap_form_tuple(funcctx->tuple_desc, pathpt, isnull);
                                result = HeapTupleGetDatum(tuple);
                                SRF_RETURN_NEXT(funcctx, result);
                        }
                        else
                        {
                                if (--state->stacklen == 0)
                                        SRF_RETURN_DONE(funcctx);
                                state->pathlen--;
                                continue;
                        }
                }
 
                if (lwgeom_is_collection(lwgeom))
                {
                        lwcoll = (LWCOLLECTION *)node->geom;
 
                        /* if a collection and we have more geoms */
                        if (node->idx < lwcoll->ngeoms)
                        {
                                if (state->stacklen > MAXDEPTH)
                                        elog(ERROR, "Unable to dump overly nested collection");
 
                                /* push the next geom on the path and the stack */
                                lwgeom = lwcoll->geoms[node->idx++];
                                state->path[state->pathlen++] = Int32GetDatum(node->idx);
 
                                node = &state->stack[state->stacklen++];
                                node->idx = 0;
                                node->geom = lwgeom;
 
                                state->pt = 0;
                                state->ring = 0;
 
                                /* loop back to beginning, which will then check whatever node we just pushed */
                                continue;
                        }
                }
 
                /* no more geometries in the current collection */
                if (--state->stacklen == 0)
                        SRF_RETURN_DONE(funcctx);
                state->pathlen--;
        }
}

References dumpstate::align, dumpstate::byval, CIRCSTRINGTYPE, dumpnode::geom, LWCOLLECTION::geoms, getPoint4d_p(), dumpnode::idx, if(), LINETYPE, lwalloc(), lwcircstring_as_lwgeom(), lwcircstring_construct(), lwgeom_from_gserialized(), lwgeom_has_m(), lwgeom_has_z(), lwgeom_is_collection(), lwgeom_is_empty(), lwline_as_lwgeom(), lwline_construct(), MAXDEPTH, LWCOLLECTION::ngeoms, POINTARRAY::npoints, LWPOLY::nrings, dumpstate::path, dumpstate::pathlen, LWLINE::points, LWTRIANGLE::points, POLYGONTYPE, dumpstate::pt, ptarray_construct(), ptarray_set_point4d(), result, dumpstate::ring, LWPOLY::rings, dumpstate::root, LWGEOM::srid, dumpstate::stack, dumpstate::stacklen, TRIANGLETYPE, LWGEOM::type, and dumpstate::typlen.

Here is the call graph for this function: