ST_ClusterDBSCAN()

Datum ST_ClusterDBSCAN

(

PG_FUNCTION_ARGS

)

Definition at line 83 of file lwgeom_window.c.

References dbscan_context::cluster_assignments, dbscan_cluster_result::cluster_id, dbscan_context::is_error, dbscan_cluster_result::is_null, LW_FALSE, LW_SUCCESS, LW_TRUE, lwalloc(), lwfree(), lwgeom_free(), lwgeom_geos_error(), lwnotice(), PG_FUNCTION_INFO_V1(), read_lwgeom_from_partition(), ST_ClusterKMeans(), UF_create(), UF_destroy(), UF_get_collapsed_cluster_ids(), and union_dbscan().

Referenced by read_lwgeom_from_partition().

{
        WindowObject win_obj = PG_WINDOW_OBJECT();
        uint32_t row = WinGetCurrentPosition(win_obj);
        uint32_t ngeoms = WinGetPartitionRowCount(win_obj);
        dbscan_context* context = WinGetPartitionLocalMemory(win_obj, sizeof(dbscan_context) + ngeoms * sizeof(dbscan_cluster_result));

        if (row == 0) /* beginning of the partition; do all of the work now */
        {
                uint32_t i;
                uint32_t* result_ids;
                LWGEOM** geoms;
                char* is_in_cluster = NULL;
                UNIONFIND* uf;
                bool tolerance_is_null;
                bool minpoints_is_null;
                Datum tolerance_datum = WinGetFuncArgCurrent(win_obj, 1, &tolerance_is_null);
                Datum minpoints_datum = WinGetFuncArgCurrent(win_obj, 2, &minpoints_is_null);
                double tolerance = DatumGetFloat8(tolerance_datum);
                int minpoints = DatumGetInt32(minpoints_datum);

                context->is_error = LW_TRUE; /* until proven otherwise */

                /* Validate input parameters */
                if (tolerance_is_null || tolerance < 0)
                {
                        lwpgerror("Tolerance must be a positive number", tolerance);
                        PG_RETURN_NULL();
                }
                if (minpoints_is_null || minpoints < 0)
                {
                        lwpgerror("Minpoints must be a positive number", minpoints);
                }

                initGEOS(lwnotice, lwgeom_geos_error);
                geoms = lwalloc(ngeoms * sizeof(LWGEOM*));
                uf = UF_create(ngeoms);
                for (i = 0; i < ngeoms; i++)
                {
                        geoms[i] = read_lwgeom_from_partition(win_obj, i, &(context->cluster_assignments[i].is_null));

                        if (!geoms[i]) {
                                /* TODO release memory ? */
                                lwpgerror("Error reading geometry.");
                                PG_RETURN_NULL();
                        }
                }

                if (union_dbscan(geoms, ngeoms, uf, tolerance, minpoints, minpoints > 1 ? &is_in_cluster : NULL) == LW_SUCCESS)
                        context->is_error = LW_FALSE;

                for (i = 0; i < ngeoms; i++)
                {
                        lwgeom_free(geoms[i]);
                }
                lwfree(geoms);

                if (context->is_error)
                {
                        UF_destroy(uf);
                        if (is_in_cluster)
                                lwfree(is_in_cluster);
                        lwpgerror("Error during clustering");
                        PG_RETURN_NULL();
                }

                result_ids = UF_get_collapsed_cluster_ids(uf, is_in_cluster);
                for (i = 0; i < ngeoms; i++)
                {
                        if (minpoints > 1 && !is_in_cluster[i])
                        {
                                context->cluster_assignments[i].is_null = LW_TRUE;
                        }
                        else
                        {
                                context->cluster_assignments[i].cluster_id = result_ids[i];
                        }
                }

                lwfree(result_ids);
                UF_destroy(uf);
        }

        if (context->cluster_assignments[row].is_null)
                PG_RETURN_NULL();

        PG_RETURN_INT32(context->cluster_assignments[row].cluster_id);
}

Here is the call graph for this function:

Here is the caller graph for this function: