RASTER_union_finalfn()

Datum RASTER_union_finalfn

(

PG_FUNCTION_ARGS

)

Definition at line 2795 of file rtpg_mapalgebra.c.

References rtpg_union_arg_t::bandarg, ES_NONE, ET_UNION, rt_iterator_t::nband, rtpg_union_arg_t::numband, rtpg_union_band_arg_t::numraster, POSTGIS_RT_DEBUG, POSTGIS_RT_DEBUGF, PT_END, rtpg_union_band_arg_t::raster, rt_iterator_t::raster, rt_band_get_hasnodata_flag(), rt_band_get_nodata(), rt_band_get_pixtype(), rt_pixtype_name(), rt_raster_copy_band(), rt_raster_destroy(), rt_raster_from_band(), rt_raster_get_band(), rt_raster_iterator(), rt_raster_serialize(), rtpg_union_arg_destroy(), rtpg_union_mean_callback(), rtpg_union_range_callback(), rt_raster_serialized_t::size, rtpg_union_band_arg_t::uniontype, UT_MEAN, and UT_RANGE.

Referenced by RASTER_union_transfn().

{
        rtpg_union_arg iwr;
        rt_raster _rtn = NULL;
        rt_raster _raster = NULL;
        rt_pgraster *pgraster = NULL;

        int i = 0;
        int j = 0;
        rt_iterator itrset = NULL;
        rt_band _band = NULL;
        int noerr = 1;
        int status = 0;
        rt_pixtype pixtype = PT_END;
        int hasnodata = 0;
        double nodataval = 0;

        POSTGIS_RT_DEBUG(3, "Starting...");

        /* cannot be called directly as this is exclusive aggregate function */
        if (!AggCheckCallContext(fcinfo, NULL)) {
                elog(ERROR, "RASTER_union_finalfn: Cannot be called in a non-aggregate context");
                PG_RETURN_NULL();
        }

        /* NULL, return null */
        if (PG_ARGISNULL(0))
                PG_RETURN_NULL();

        iwr = (rtpg_union_arg) PG_GETARG_POINTER(0);

        /* init itrset */
        itrset = palloc(sizeof(struct rt_iterator_t) * 2);
        if (itrset == NULL) {
                rtpg_union_arg_destroy(iwr);
                elog(ERROR, "RASTER_union_finalfn: Could not allocate memory for iterator arguments");
                PG_RETURN_NULL();
        }

        for (i = 0; i < iwr->numband; i++) {
                if (
                        iwr->bandarg[i].uniontype == UT_MEAN ||
                        iwr->bandarg[i].uniontype == UT_RANGE
                ) {
                        /* raster containing the SUM or MAX is at index 1 */
                        _band = rt_raster_get_band(iwr->bandarg[i].raster[1], 0);

                        pixtype = rt_band_get_pixtype(_band);
                        hasnodata = rt_band_get_hasnodata_flag(_band);
                        if (hasnodata)
                                rt_band_get_nodata(_band, &nodataval);
                        POSTGIS_RT_DEBUGF(4, "(pixtype, hasnodata, nodataval) = (%s, %d, %f)", rt_pixtype_name(pixtype), hasnodata, nodataval);

                        itrset[0].raster = iwr->bandarg[i].raster[0];
                        itrset[0].nband = 0;
                        itrset[1].raster = iwr->bandarg[i].raster[1];
                        itrset[1].nband = 0;

                        /* pass everything to iterator */
                        if (iwr->bandarg[i].uniontype == UT_MEAN) {
                                noerr = rt_raster_iterator(
                                        itrset, 2,
                                        ET_UNION, NULL,
                                        pixtype,
                                        hasnodata, nodataval,
                                        0, 0,
                                        NULL,
                                        NULL,
                                        rtpg_union_mean_callback,
                                        &_raster
                                );
                        }
                        else if (iwr->bandarg[i].uniontype == UT_RANGE) {
                                noerr = rt_raster_iterator(
                                        itrset, 2,
                                        ET_UNION, NULL,
                                        pixtype,
                                        hasnodata, nodataval,
                                        0, 0,
                                        NULL,
                                        NULL,
                                        rtpg_union_range_callback,
                                        &_raster
                                );
                        }

                        if (noerr != ES_NONE) {
                                pfree(itrset);
                                rtpg_union_arg_destroy(iwr);
                                if (_rtn != NULL)
                                        rt_raster_destroy(_rtn);
                                elog(ERROR, "RASTER_union_finalfn: Could not run raster iterator function");
                                PG_RETURN_NULL();
                        }
                }
                else
                        _raster = iwr->bandarg[i].raster[0];

                /* first band, _rtn doesn't exist */
                if (i < 1) {
                        uint32_t bandNums[1] = {0};
                        _rtn = rt_raster_from_band(_raster, bandNums, 1);
                        status = (_rtn == NULL) ? -1 : 0;
                }
                else
                        status = rt_raster_copy_band(_rtn, _raster, 0, i);

                POSTGIS_RT_DEBUG(4, "destroying source rasters");

                /* destroy source rasters */
                if (
                        iwr->bandarg[i].uniontype == UT_MEAN ||
                        iwr->bandarg[i].uniontype == UT_RANGE
                ) {
                        rt_raster_destroy(_raster);
                }

                for (j = 0; j < iwr->bandarg[i].numraster; j++) {
                        if (iwr->bandarg[i].raster[j] == NULL)
                                continue;
                        rt_raster_destroy(iwr->bandarg[i].raster[j]);
                        iwr->bandarg[i].raster[j] = NULL;
                }

                if (status < 0) {
                        rtpg_union_arg_destroy(iwr);
                        rt_raster_destroy(_rtn);
                        elog(ERROR, "RASTER_union_finalfn: Could not add band to final raster");
                        PG_RETURN_NULL();
                }
        }

        /* cleanup */
        pfree(itrset);
        rtpg_union_arg_destroy(iwr);

        if (!_rtn) PG_RETURN_NULL();

        pgraster = rt_raster_serialize(_rtn);
        rt_raster_destroy(_rtn);

        POSTGIS_RT_DEBUG(3, "Finished");

        if (!pgraster)
                PG_RETURN_NULL();

        SET_VARSIZE(pgraster, pgraster->size);
        PG_RETURN_POINTER(pgraster);
}

Here is the call graph for this function:

Here is the caller graph for this function: