gserialized_spgist_picksplit_3d()

PGDLLEXPORT Datum gserialized_spgist_picksplit_3d

(

PG_FUNCTION_ARGS

)

Definition at line 421 of file gserialized_spgist_3d.c.

{
        spgPickSplitIn *in = (spgPickSplitIn *)PG_GETARG_POINTER(0);
        spgPickSplitOut *out = (spgPickSplitOut *)PG_GETARG_POINTER(1);
        BOX3D *centroid;
        int median, i;
        double *lowXs = palloc(sizeof(double) * in->nTuples);
        double *highXs = palloc(sizeof(double) * in->nTuples);
        double *lowYs = palloc(sizeof(double) * in->nTuples);
        double *highYs = palloc(sizeof(double) * in->nTuples);
        double *lowZs = palloc(sizeof(double) * in->nTuples);
        double *highZs = palloc(sizeof(double) * in->nTuples);
        BOX3D *box = DatumGetBox3DP(in->datums[0]);
        int32_t srid = box->srid;
 
        /* Calculate median of all 6D coordinates */
        for (i = 0; i < in->nTuples; i++)
        {
                BOX3D *box = DatumGetBox3DP(in->datums[i]);
 
                lowXs[i] = box->xmin;
                highXs[i] = box->xmax;
                lowYs[i] = box->ymin;
                highYs[i] = box->ymax;
                lowZs[i] = box->zmin;
                highZs[i] = box->zmax;
        }
 
        qsort(lowXs, in->nTuples, sizeof(double), compareDoubles);
        qsort(highXs, in->nTuples, sizeof(double), compareDoubles);
        qsort(lowYs, in->nTuples, sizeof(double), compareDoubles);
        qsort(highYs, in->nTuples, sizeof(double), compareDoubles);
        qsort(lowZs, in->nTuples, sizeof(double), compareDoubles);
        qsort(highZs, in->nTuples, sizeof(double), compareDoubles);
 
        median = in->nTuples / 2;
 
        centroid = palloc(sizeof(BOX3D));
 
        centroid->xmin = lowXs[median];
        centroid->xmax = highXs[median];
        centroid->ymin = lowYs[median];
        centroid->ymax = highYs[median];
        centroid->zmin = lowZs[median];
        centroid->zmax = highZs[median];
        centroid->srid = srid;
 
        /* Fill the output */
        out->hasPrefix = true;
        out->prefixDatum = Box3DPGetDatum(centroid);
 
        out->nNodes = 64;
        out->nodeLabels = NULL; /* We don't need node labels. */
 
        out->mapTuplesToNodes = palloc(sizeof(int) * in->nTuples);
        out->leafTupleDatums = palloc(sizeof(Datum) * in->nTuples);
 
        /*
         * Assign ranges to corresponding nodes according to octants relative to
         * the "centroid" range
         */
        for (i = 0; i < in->nTuples; i++)
        {
                BOX3D *box = DatumGetBox3DP(in->datums[i]);
                uint8 octant = getOctant(centroid, box);
 
                out->leafTupleDatums[i] = Box3DPGetDatum(box);
                out->mapTuplesToNodes[i] = octant;
        }
 
        pfree(lowXs);
        pfree(highXs);
        pfree(lowYs);
        pfree(highYs);
        pfree(lowZs);
        pfree(highZs);
 
        PG_RETURN_VOID();
}

References centroid(), compareDoubles(), getOctant(), BOX3D::srid, BOX3D::xmax, BOX3D::xmin, BOX3D::ymax, BOX3D::ymin, BOX3D::zmax, and BOX3D::zmin.

Here is the call graph for this function: