RASTER_band()

Datum RASTER_band

(

PG_FUNCTION_ARGS

)

Definition at line 1493 of file rtpg_create.c.

References FALSE, POSTGIS_RT_DEBUGF, rtpixdump::rast, rtrowdump::raster, rt_raster_deserialize(), rt_raster_destroy(), rt_raster_from_band(), rt_raster_get_num_bands(), rt_raster_serialize(), rt_raster_serialized_t::size, and TRUE.

Referenced by RASTER_tile().

{
        rt_pgraster *pgraster;
        rt_pgraster *pgrast;
        rt_raster raster;
        rt_raster rast;

        bool skip = FALSE;
        ArrayType *array;
        Oid etype;
        Datum *e;
        bool *nulls;
        int16 typlen;
        bool typbyval;
        char typalign;

        uint32_t numBands;
        uint32_t *bandNums;
        uint32 idx = 0;
        int n;
        int i = 0;
        int j = 0;

        if (PG_ARGISNULL(0))
                PG_RETURN_NULL();
        pgraster = (rt_pgraster *) PG_DETOAST_DATUM(PG_GETARG_DATUM(0));

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

        /* process bandNums */
        if (PG_ARGISNULL(1)) {
                elog(NOTICE, "Band number(s) not provided.  Returning original raster");
                skip = TRUE;
        }
        do {
                if (skip) break;

                numBands = rt_raster_get_num_bands(raster);

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

                switch (etype) {
                        case INT2OID:
                        case INT4OID:
                                break;
                        default:
                                rt_raster_destroy(raster);
                                PG_FREE_IF_COPY(pgraster, 0);
                                elog(ERROR, "RASTER_band: Invalid data type for band number(s)");
                                PG_RETURN_NULL();
                                break;
                }

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

                bandNums = palloc(sizeof(uint32_t) * n);
                for (i = 0, j = 0; i < n; i++) {
                        if (nulls[i]) continue;

                        switch (etype) {
                                case INT2OID:
                                        idx = (uint32_t) DatumGetInt16(e[i]);
                                        break;
                                case INT4OID:
                                        idx = (uint32_t) DatumGetInt32(e[i]);
                                        break;
                        }

                        POSTGIS_RT_DEBUGF(3, "band idx (before): %d", idx);
                        if (idx > numBands || idx < 1) {
        elog(NOTICE, "Invalid band index (must use 1-based). Returning original raster");
                                skip = TRUE;
                                break;
                        }

                        bandNums[j] = idx - 1;
                        POSTGIS_RT_DEBUGF(3, "bandNums[%d] = %d", j, bandNums[j]);
                        j++;
                }

                if (skip || j < 1) {
                        pfree(bandNums);
                        skip = TRUE;
                }
        }
        while (0);

        if (!skip) {
                rast = rt_raster_from_band(raster, bandNums, j);
                pfree(bandNums);
                rt_raster_destroy(raster);
                PG_FREE_IF_COPY(pgraster, 0);
                if (!rast) {
                        elog(ERROR, "RASTER_band: Could not create new raster");
                        PG_RETURN_NULL();
                }

                pgrast = rt_raster_serialize(rast);
                rt_raster_destroy(rast);

                if (!pgrast)
                        PG_RETURN_NULL();

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

        PG_RETURN_POINTER(pgraster);
}

Here is the call graph for this function:

Here is the caller graph for this function: