RASTER_band()

Datum RASTER_band

(

PG_FUNCTION_ARGS

)

Definition at line 1457 of file rtpg_create.c.

{
        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;
        }
        if (!skip) {
                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;
                }
        }
 
        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);
}

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.

Here is the call graph for this function: