rtpg_union_unionarg_process()

static int rtpg_union_unionarg_process ( rtpg_union_arg  arg, ArrayType *  array  )

static

Definition at line 1950 of file rtpg_mapalgebra.c.

References rtpg_union_arg_t::bandarg, pixval::nband, rtpg_union_band_arg_t::nband, rtpg_union_arg_t::numband, rtpg_union_band_arg_t::numraster, POSTGIS_RT_DEBUGF, rtpg_union_band_arg_t::raster, rtpg_strtoupper(), rtpg_uniontype_index_from_name(), rtpg_union_band_arg_t::uniontype, UT_LAST, UT_MEAN, and UT_RANGE.

Referenced by RASTER_union_transfn().

                                                                             {
        Oid etype;
        Datum *e;
        bool *nulls;
        int16 typlen;
        bool typbyval;
        char typalign;
        int n = 0;

        HeapTupleHeader tup;
        bool isnull;
        Datum tupv;

        int i;
        int nband = 1;
        char *utypename = NULL;
        rtpg_union_type utype = UT_LAST;

        etype = ARR_ELEMTYPE(array);
        get_typlenbyvalalign(etype, &typlen, &typbyval, &typalign);

        deconstruct_array(
                array,
                etype,
                typlen, typbyval, typalign,
                &e, &nulls, &n
        );

        if (!n) {
                elog(ERROR, "rtpg_union_unionarg_process: Invalid argument for unionarg");
                return 0;
        }

        /* prep arg */
        arg->numband = n;
        arg->bandarg = palloc(sizeof(struct rtpg_union_band_arg_t) * arg->numband);
        if (arg->bandarg == NULL) {
                elog(ERROR, "rtpg_union_unionarg_process: Could not allocate memory for band information");
                return 0;
        }

        /* process each element */
        for (i = 0; i < n; i++) {
                if (nulls[i]) {
                        arg->numband--;
                        continue;
                }

                POSTGIS_RT_DEBUGF(4, "Processing unionarg at index %d", i);

                /* each element is a tuple */
                tup = (HeapTupleHeader) DatumGetPointer(e[i]);
                if (NULL == tup) {
                        elog(ERROR, "rtpg_union_unionarg_process: Invalid argument for unionarg");
                        return 0;
                }

                /* first element, bandnum */
                tupv = GetAttributeByName(tup, "nband", &isnull);
                if (isnull) {
                        nband = i + 1;
                        elog(NOTICE, "First argument (nband) of unionarg is NULL.  Assuming nband = %d", nband);
                }
                else
                        nband = DatumGetInt32(tupv);

                if (nband < 1) {
                        elog(ERROR, "rtpg_union_unionarg_process: Band number must be greater than zero (1-based)");
                        return 0;
                }

                /* second element, uniontype */
                tupv = GetAttributeByName(tup, "uniontype", &isnull);
                if (isnull) {
                        elog(NOTICE, "Second argument (uniontype) of unionarg is NULL.  Assuming uniontype = LAST");
                        utype = UT_LAST;
                }
                else {
                        utypename = text_to_cstring((text *) DatumGetPointer(tupv));
                        utype = rtpg_uniontype_index_from_name(rtpg_strtoupper(utypename));
                }

                arg->bandarg[i].uniontype = utype;
                arg->bandarg[i].nband = nband - 1;
                arg->bandarg[i].raster = NULL;

                if (
                        utype != UT_MEAN &&
                        utype != UT_RANGE
                ) {
                        arg->bandarg[i].numraster = 1;
                }
                else
                        arg->bandarg[i].numraster = 2;
        }

        if (arg->numband < n) {
                arg->bandarg = repalloc(arg->bandarg, sizeof(struct rtpg_union_band_arg_t) * arg->numband);
                if (arg->bandarg == NULL) {
                        elog(ERROR, "rtpg_union_unionarg_process: Could not reallocate memory for band information");
                        return 0;
                }
        }

        return 1;
}

Here is the call graph for this function:

Here is the caller graph for this function: