RASTER_addBand()

Datum RASTER_addBand

(

PG_FUNCTION_ARGS

)

Definition at line 121 of file rtpg_create.c.

References FALSE, PG_FUNCTION_INFO_V1(), POSTGIS_RT_DEBUG, POSTGIS_RT_DEBUGF, PT_END, rtrowdump::raster, RASTER_addBandRasterArray(), rt_pixtype_index_from_name(), rt_pixtype_name(), rt_raster_deserialize(), rt_raster_destroy(), rt_raster_generate_new_band(), rt_raster_get_num_bands(), rt_raster_serialize(), rt_raster_serialized_t::size, and TRUE.

Referenced by RASTER_makeEmpty().

{
        rt_pgraster *pgraster = NULL;
        rt_pgraster *pgrtn = NULL;
        rt_raster raster = NULL;
        int bandindex = 0;
        int maxbandindex = 0;
        int numbands = 0;
        int lastnumbands = 0;

        text *text_pixtype = NULL;
        char *char_pixtype = NULL;

        struct addbandarg {
                int index;
                bool append;
                rt_pixtype pixtype;
                double initialvalue;
                bool hasnodata;
                double nodatavalue;
        };
        struct addbandarg *arg = NULL;

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

        HeapTupleHeader tup;
        bool isnull;
        Datum tupv;

        int i = 0;

        /* pgraster is null, return null */
        if (PG_ARGISNULL(0))
                PG_RETURN_NULL();
        pgraster = (rt_pgraster *) PG_DETOAST_DATUM(PG_GETARG_DATUM(0));

        /* raster */
        raster = rt_raster_deserialize(pgraster, FALSE);
        if (!raster) {
                PG_FREE_IF_COPY(pgraster, 0);
                elog(ERROR, "RASTER_addBand: Could not deserialize raster");
                PG_RETURN_NULL();
        }

        /* process set of addbandarg */
        POSTGIS_RT_DEBUG(3, "Processing Arg 1 (addbandargset)");
        array = PG_GETARG_ARRAYTYPE_P(1);
        etype = ARR_ELEMTYPE(array);
        get_typlenbyvalalign(etype, &typlen, &typbyval, &typalign);

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

        if (!n) {
                PG_FREE_IF_COPY(pgraster, 0);
                elog(ERROR, "RASTER_addBand: Invalid argument for addbandargset");
                PG_RETURN_NULL();
        }

        /* allocate addbandarg */
        arg = (struct addbandarg *) palloc(sizeof(struct addbandarg) * n);
        if (arg == NULL) {
                rt_raster_destroy(raster);
                PG_FREE_IF_COPY(pgraster, 0);
                elog(ERROR, "RASTER_addBand: Could not allocate memory for addbandarg");
                PG_RETURN_NULL();
        }

        /*
                process each element of addbandargset
                each element is the index of where to add the new band,
                        new band's pixeltype, the new band's initial value and
                        the new band's NODATA value if NOT NULL
        */
        for (i = 0; i < n; i++) {
                if (nulls[i]) continue;

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

                /* each element is a tuple */
                tup = (HeapTupleHeader) DatumGetPointer(e[i]);
                if (NULL == tup) {
                        pfree(arg);
                        rt_raster_destroy(raster);
                        PG_FREE_IF_COPY(pgraster, 0);
                        elog(ERROR, "RASTER_addBand: Invalid argument for addbandargset");
                        PG_RETURN_NULL();
                }

                /* new band index, 1-based */
                arg[i].index = 0;
                arg[i].append = TRUE;
                tupv = GetAttributeByName(tup, "index", &isnull);
                if (!isnull) {
                        arg[i].index = DatumGetInt32(tupv);
                        arg[i].append = FALSE;
                }

                /* for now, only check that band index is 1-based */
                if (!arg[i].append && arg[i].index < 1) {
                        pfree(arg);
                        rt_raster_destroy(raster);
                        PG_FREE_IF_COPY(pgraster, 0);
                        elog(ERROR, "RASTER_addBand: Invalid argument for addbandargset. Invalid band index (must be 1-based) for addbandarg of index %d", i);
                        PG_RETURN_NULL();
                }

                /* new band pixeltype */
                arg[i].pixtype = PT_END;
                tupv = GetAttributeByName(tup, "pixeltype", &isnull);
                if (isnull) {
                        pfree(arg);
                        rt_raster_destroy(raster);
                        PG_FREE_IF_COPY(pgraster, 0);
                        elog(ERROR, "RASTER_addBand: Invalid argument for addbandargset. Pixel type cannot be NULL for addbandarg of index %d", i);
                        PG_RETURN_NULL();
                }
                text_pixtype = (text *) DatumGetPointer(tupv);
                if (text_pixtype == NULL) {
                        pfree(arg);
                        rt_raster_destroy(raster);
                        PG_FREE_IF_COPY(pgraster, 0);
                        elog(ERROR, "RASTER_addBand: Invalid argument for addbandargset. Pixel type cannot be NULL for addbandarg of index %d", i);
                        PG_RETURN_NULL();
                }
                char_pixtype = text_to_cstring(text_pixtype);

                arg[i].pixtype = rt_pixtype_index_from_name(char_pixtype);
                pfree(char_pixtype);
                if (arg[i].pixtype == PT_END) {
                        pfree(arg);
                        rt_raster_destroy(raster);
                        PG_FREE_IF_COPY(pgraster, 0);
                        elog(ERROR, "RASTER_addBand: Invalid argument for addbandargset. Invalid pixel type for addbandarg of index %d", i);
                        PG_RETURN_NULL();
                }

                /* new band initialvalue */
                arg[i].initialvalue = 0;
                tupv = GetAttributeByName(tup, "initialvalue", &isnull);
                if (!isnull)
                        arg[i].initialvalue = DatumGetFloat8(tupv);

                /* new band NODATA value */
                arg[i].hasnodata = FALSE;
                arg[i].nodatavalue = 0;
                tupv = GetAttributeByName(tup, "nodataval", &isnull);
                if (!isnull) {
                        arg[i].hasnodata = TRUE;
                        arg[i].nodatavalue = DatumGetFloat8(tupv);
                }
        }

        /* add new bands to raster */
        lastnumbands = rt_raster_get_num_bands(raster);
        for (i = 0; i < n; i++) {
                if (nulls[i]) continue;

                POSTGIS_RT_DEBUGF(3, "%d bands in old raster", lastnumbands);
                maxbandindex = lastnumbands + 1;

                /* check that new band's index doesn't exceed maxbandindex */
                if (!arg[i].append) {
                        if (arg[i].index > maxbandindex) {
                                elog(NOTICE, "Band index for addbandarg of index %d exceeds possible value. Adding band at index %d", i, maxbandindex);
                                arg[i].index = maxbandindex;
                        }
                }
                /* append, so use maxbandindex */
                else
                        arg[i].index = maxbandindex;

                POSTGIS_RT_DEBUGF(4, "new band (index, pixtype, initialvalue, hasnodata, nodatavalue) = (%d, %s, %f, %s, %f)",
                        arg[i].index,
                        rt_pixtype_name(arg[i].pixtype),
                        arg[i].initialvalue,
                        arg[i].hasnodata ? "TRUE" : "FALSE",
                        arg[i].nodatavalue
                );

                bandindex = rt_raster_generate_new_band(
                        raster,
                        arg[i].pixtype, arg[i].initialvalue,
                        arg[i].hasnodata, arg[i].nodatavalue,
                        arg[i].index - 1
                );

                numbands = rt_raster_get_num_bands(raster);
                if (numbands == lastnumbands || bandindex == -1) {
                        pfree(arg);
                        rt_raster_destroy(raster);
                        PG_FREE_IF_COPY(pgraster, 0);
                        elog(ERROR, "RASTER_addBand: Could not add band defined by addbandarg of index %d to raster", i);
                        PG_RETURN_NULL();
                }

                lastnumbands = numbands;
                POSTGIS_RT_DEBUGF(3, "%d bands in new raster", lastnumbands);
        }

        pfree(arg);

        pgrtn = rt_raster_serialize(raster);
        rt_raster_destroy(raster);
        PG_FREE_IF_COPY(pgraster, 0);
        if (!pgrtn)
                PG_RETURN_NULL();

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

Here is the call graph for this function:

Here is the caller graph for this function: