rtpg_nmapalgebra_rastbandarg_process()

static int rtpg_nmapalgebra_rastbandarg_process ( rtpg_nmapalgebra_arg  arg, ArrayType *  array, int *  allnull, int *  allempty, int *  noband  )

static

Definition at line 184 of file rtpg_mapalgebra.c.

References FALSE, rtpg_nmapalgebra_arg_t::hasband, rtpg_nmapalgebra_arg_t::isempty, rtpg_nmapalgebra_arg_t::mask, pixval::nband, rtpg_nmapalgebra_arg_t::nband, rtpg_nmapalgebra_arg_t::numraster, rtpg_nmapalgebra_arg_t::ownsdata, rtpg_nmapalgebra_arg_t::pgraster, POSTGIS_RT_DEBUG, POSTGIS_RT_DEBUGF, rtpg_nmapalgebra_arg_t::raster, rt_raster_deserialize(), rt_raster_has_band(), and rt_raster_is_empty().

Referenced by RASTER_nMapAlgebra(), and RASTER_nMapAlgebraExpr().

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

        HeapTupleHeader tup;
        bool isnull;
        Datum tupv;

        int i;
        int j;
        int nband;

        if (arg == NULL || array == NULL) {
                elog(ERROR, "rtpg_nmapalgebra_rastbandarg_process: NULL values not permitted for parameters");
                return 0;
        }

        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_nmapalgebra_rastbandarg_process: Invalid argument for rastbandarg");
                return 0;
        }

        /* prep arg */
        arg->numraster = n;
        arg->pgraster = palloc(sizeof(rt_pgraster *) * arg->numraster);
        arg->raster = palloc(sizeof(rt_raster) * arg->numraster);
        arg->isempty = palloc(sizeof(uint8_t) * arg->numraster);
        arg->ownsdata = palloc(sizeof(uint8_t) * arg->numraster);
        arg->nband = palloc(sizeof(int) * arg->numraster);
        arg->hasband = palloc(sizeof(uint8_t) * arg->numraster);
        arg->mask = palloc(sizeof(struct rt_mask_t));
        if (
                arg->pgraster == NULL ||
                arg->raster == NULL ||
                arg->isempty == NULL ||
                arg->ownsdata == NULL ||
                arg->nband == NULL ||
                arg->hasband == NULL ||
                arg->mask == NULL
        ) {
                elog(ERROR, "rtpg_nmapalgebra_rastbandarg_process: Could not allocate memory for processing rastbandarg");
                return 0;
        }

        *allnull = 0;
        *allempty = 0;
        *noband = 0;

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

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

                arg->raster[i] = NULL;
                arg->isempty[i] = 0;
                arg->ownsdata[i] = 1;
                arg->nband[i] = 0;
                arg->hasband[i] = 0;

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

                /* first element, raster */
                POSTGIS_RT_DEBUG(4, "Processing first element (raster)");
                tupv = GetAttributeByName(tup, "rast", &isnull);
                if (isnull) {
                        elog(NOTICE, "First argument (nband) of rastbandarg at index %d is NULL. Assuming NULL raster", i);
                        arg->isempty[i] = 1;
                        arg->ownsdata[i] = 0;

                        (*allnull)++;
                        (*allempty)++;
                        (*noband)++;

                        continue;
                }

                arg->pgraster[i] = (rt_pgraster *) PG_DETOAST_DATUM(tupv);

                /* see if this is a copy of an existing pgraster */
                for (j = 0; j < i; j++) {
                        if (!arg->isempty[j] && (arg->pgraster[i] == arg->pgraster[j])) {
                                POSTGIS_RT_DEBUG(4, "raster matching existing same raster found");
                                arg->raster[i] = arg->raster[j];
                                arg->ownsdata[i] = 0;
                                break;
                        }
                }

                if (arg->ownsdata[i]) {
                        POSTGIS_RT_DEBUG(4, "deserializing raster");
                        arg->raster[i] = rt_raster_deserialize(arg->pgraster[i], FALSE);
                        if (arg->raster[i] == NULL) {
                                elog(ERROR, "rtpg_nmapalgebra_rastbandarg_process: Could not deserialize raster at index %d", i);
                                return 0;
                        }
                }

                /* is raster empty? */
                arg->isempty[i] = rt_raster_is_empty(arg->raster[i]);
                if (arg->isempty[i]) {
                        (*allempty)++;
                        (*noband)++;

                        continue;
                }

                /* second element, nband */
                POSTGIS_RT_DEBUG(4, "Processing second element (nband)");
                tupv = GetAttributeByName(tup, "nband", &isnull);
                if (isnull) {
                        nband = 1;
                        elog(NOTICE, "First argument (nband) of rastbandarg at index %d is NULL. Assuming nband = %d", i, nband);
                }
                else
                        nband = DatumGetInt32(tupv);

                if (nband < 1) {
                        elog(ERROR, "rtpg_nmapalgebra_rastbandarg_process: Band number provided for rastbandarg at index %d must be greater than zero (1-based)", i);
                        return 0;
                }

                arg->nband[i] = nband - 1;
                arg->hasband[i] = rt_raster_has_band(arg->raster[i], arg->nband[i]);
                if (!arg->hasband[i]) {
                        (*noband)++;
                        POSTGIS_RT_DEBUGF(4, "Band at index %d not found in raster", nband);
                }
        }

        if (arg->numraster < n) {
                arg->pgraster = repalloc(arg->pgraster, sizeof(rt_pgraster *) * arg->numraster);
                arg->raster = repalloc(arg->raster, sizeof(rt_raster) * arg->numraster);
                arg->isempty = repalloc(arg->isempty, sizeof(uint8_t) * arg->numraster);
                arg->ownsdata = repalloc(arg->ownsdata, sizeof(uint8_t) * arg->numraster);
                arg->nband = repalloc(arg->nband, sizeof(int) * arg->numraster);
                arg->hasband = repalloc(arg->hasband, sizeof(uint8_t) * arg->numraster);
                if (
                        arg->pgraster == NULL ||
                        arg->raster == NULL ||
                        arg->isempty == NULL ||
                        arg->ownsdata == NULL ||
                        arg->nband == NULL ||
                        arg->hasband == NULL
                ) {
                        elog(ERROR, "rtpg_nmapalgebra_rastbandarg_process: Could not reallocate memory for processed rastbandarg");
                        return 0;
                }
        }

        POSTGIS_RT_DEBUGF(4, "arg->numraster = %d", arg->numraster);

        return 1;
}

Here is the call graph for this function:

Here is the caller graph for this function: