test_kmeans()

static void test_kmeans ( void  )

static

Definition at line 1162 of file cu_algorithm.c.

References algorithms_suite_setup(), lwalloc(), lwfree(), lwgeom_cluster_2d_kmeans(), lwgeom_free(), lwpoint_as_lwgeom(), lwpoint_make2d(), r, and SRID_UNKNOWN.

Referenced by algorithms_suite_setup().

{
        static int cluster_size = 100;
        static int num_clusters = 4;
        static double spread = 1.5;
        int N = cluster_size * num_clusters;
        LWGEOM **geoms;
        int i, j, k=0;
        int *r;

        geoms = lwalloc(sizeof(LWGEOM*) * N);

        for (j = 0; j < num_clusters; j++) {
                for (i = 0; i < cluster_size; i++)
                {
                        double u1 = 1.0 * rand() / RAND_MAX;
                        double u2 = 1.0 * rand() / RAND_MAX;
                        double z1 = spread * j + sqrt(-2*log2(u1))*cos(2*M_PI*u2);
                        double z2 = spread * j + sqrt(-2*log2(u1))*sin(2*M_PI*u2);

                        LWPOINT *lwp = lwpoint_make2d(SRID_UNKNOWN, z1, z2);
                        geoms[k++] = lwpoint_as_lwgeom(lwp);
                }
        }

        r = lwgeom_cluster_2d_kmeans((const LWGEOM **)geoms, N, num_clusters);

        // for (i = 0; i < k; i++)
        // {
        //      printf("[%d] %s\n", r[i], lwgeom_to_ewkt(geoms[i]));
        // }

        /* Clean up */
        lwfree(r);
        for (i = 0; i < k; i++)
                lwgeom_free(geoms[i]);
        lwfree(geoms);

        return;
}

Here is the call graph for this function:

Here is the caller graph for this function: