RASTER_dwithin()

Datum RASTER_dwithin

(

PG_FUNCTION_ARGS

)

Definition at line 907 of file rtpg_spatial_relationship.c.

{
        const uint32_t set_count = 2;
        rt_pgraster *pgrast[2];
        int pgrastpos[2] = {-1, -1};
        rt_raster rast[2] = {NULL};
        uint32_t bandindex[2] = {0};
        uint32_t hasbandindex[2] = {0};
        double distance = 0;
 
        uint32_t i;
        uint32_t j;
        uint32_t k;
        uint32_t numBands;
        int rtn;
        int result;
 
        for (i = 0, j = 0; i < set_count; i++) {
                /* pgrast is null, return null */
                if (PG_ARGISNULL(j)) {
                        for (k = 0; k < i; k++) {
                                rt_raster_destroy(rast[k]);
                                PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
                        }
                        PG_RETURN_NULL();
                }
                pgrast[i] = (rt_pgraster *) PG_DETOAST_DATUM(PG_GETARG_DATUM(j));
                pgrastpos[i] = j;
                j++;
 
                /* raster */
                rast[i] = rt_raster_deserialize(pgrast[i], FALSE);
                if (!rast[i]) {
                        for (k = 0; k <= i; k++) {
                                if (k < i)
                                        rt_raster_destroy(rast[k]);
                                PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
                        }
                        elog(ERROR, "RASTER_dwithin: Could not deserialize the %s raster", i < 1 ? "first" : "second");
                        PG_RETURN_NULL();
                }
 
                /* numbands */
                numBands = rt_raster_get_num_bands(rast[i]);
                if (numBands < 1) {
                        elog(NOTICE, "The %s raster provided has no bands", i < 1 ? "first" : "second");
                        if (i > 0) i++;
                        for (k = 0; k < i; k++) {
                                rt_raster_destroy(rast[k]);
                                PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
                        }
                        PG_RETURN_NULL();
                }
 
                /* band index */
                if (!PG_ARGISNULL(j)) {
                        bandindex[i] = PG_GETARG_INT32(j);
                        if (bandindex[i] < 1 || bandindex[i] > numBands) {
                                elog(NOTICE, "Invalid band index (must use 1-based) for the %s raster. Returning NULL", i < 1 ? "first" : "second");
                                if (i > 0) i++;
                                for (k = 0; k < i; k++) {
                                        rt_raster_destroy(rast[k]);
                                        PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
                                }
                                PG_RETURN_NULL();
                        }
                        hasbandindex[i] = 1;
                }
                else
                        hasbandindex[i] = 0;
                POSTGIS_RT_DEBUGF(4, "hasbandindex[%d] = %d", i, hasbandindex[i]);
                POSTGIS_RT_DEBUGF(4, "bandindex[%d] = %d", i, bandindex[i]);
                j++;
        }
 
        /* distance */
        if (PG_ARGISNULL(4)) {
                elog(NOTICE, "Distance cannot be NULL.  Returning NULL");
                for (k = 0; k < set_count; k++) {
                        rt_raster_destroy(rast[k]);
                        PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
                }
                PG_RETURN_NULL();
        }
 
        distance = PG_GETARG_FLOAT8(4);
        if (distance < 0) {
                elog(NOTICE, "Distance cannot be less than zero.  Returning NULL");
                for (k = 0; k < set_count; k++) {
                        rt_raster_destroy(rast[k]);
                        PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
                }
                PG_RETURN_NULL();
        }
 
        /* hasbandindex must be balanced */
        if (
                (hasbandindex[0] && !hasbandindex[1]) ||
                (!hasbandindex[0] && hasbandindex[1])
        ) {
                elog(NOTICE, "Missing band index.  Band indices must be provided for both rasters if any one is provided");
                for (k = 0; k < set_count; k++) {
                        rt_raster_destroy(rast[k]);
                        PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
                }
                PG_RETURN_NULL();
        }
 
        /* SRID must match */
        if (rt_raster_get_srid(rast[0]) != rt_raster_get_srid(rast[1])) {
                for (k = 0; k < set_count; k++) {
                        rt_raster_destroy(rast[k]);
                        PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
                }
                elog(ERROR, "The two rasters provided have different SRIDs");
                PG_RETURN_NULL();
        }
 
        rtn = rt_raster_within_distance(
                rast[0], (hasbandindex[0] ? (int)bandindex[0] - 1 : -1),
                rast[1], (hasbandindex[1] ? (int)bandindex[1] - 1 : -1),
                distance,
                &result
        );
        for (k = 0; k < set_count; k++) {
                rt_raster_destroy(rast[k]);
                PG_FREE_IF_COPY(pgrast[k], pgrastpos[k]);
        }
 
        if (rtn != ES_NONE) {
                elog(ERROR, "RASTER_dwithin: Could not test that the two rasters are within the specified distance of each other");
                PG_RETURN_NULL();
        }
 
        PG_RETURN_BOOL(result);
}

References distance(), ES_NONE, FALSE, POSTGIS_RT_DEBUGF, rtpixdump::rast, rt_raster_deserialize(), rt_raster_destroy(), rt_raster_get_num_bands(), rt_raster_get_srid(), and rt_raster_within_distance().

Here is the call graph for this function: