vector/rtree/index.c

Go to the documentation of this file.

/*!
   \file lib/vector/rtree/index.c
 
   \brief R-Tree library - Multidimensional index
 
   Higher level functions for managing R*-Trees.
 
   (C) 2010-2012 by the GRASS Development Team
 
   This program is free software under the
   GNU General Public License (>=v2).
   Read the file COPYING that comes with GRASS
   for details.
 
   \author Antonin Guttman - original code
   \author Daniel Green (green@superliminal.com) - major clean-up
   and implementation of bounding spheres
   \author Markus Metz - file-based and memory-based R*-tree
 */
 
/* Read these articles first before attempting to modify the code
 *
 * R-Tree reference:
 * Guttman, A. (1984). "R-Trees: A Dynamic Index Structure for Spatial
 * Searching". Proceedings of the 1984 ACM SIGMOD international
 * conference on Management of data - SIGMOD '84. pp. 47.
 * DOI:10.1145/602259.602266
 * ISBN 0897911288
 *
 * R*-Tree reference:
 * Beckmann, N.; Kriegel, H. P.; Schneider, R.; Seeger, B. (1990).
 * "The R*-tree: an efficient and robust access method for points and
 * rectangles". Proceedings of the 1990 ACM SIGMOD international
 * conference on Management of data - SIGMOD '90. pp. 322.
 * DOI:10.1145/93597.98741
 * ISBN 0897913655
 */
 
#include <stdlib.h>
#include <sys/types.h>
#include <unistd.h>
#include <assert.h>
#include <grass/gis.h>
#include "index.h"
 
/*!
   \brief Create new empty R*-Tree
 
   This method creates a new RTree, either in memory (fd < 0) or in file.
   If the file descriptor is positive, the corresponding file must have
   been opened for reading and writing.
   This method must also be called if an existing tree previously saved
   to file is going to be accessed.
 
   \param fd file descriptor to hold data, negative toggles memory mode
   \param rootpos offset in file to root node (past any header info)
   \param ndims number of dimensions for the new tree: min 2, max 20
 
   \return pointer to new RTree structure
 */
struct RTree *RTreeCreateTree(int fd, off_t rootpos, int ndims)
{
    struct RTree *new_rtree;
    struct RTree_Node *n;
    int i, j, k;
 
    new_rtree = (struct RTree *)malloc(sizeof(struct RTree));
 
    new_rtree->fd = fd;
    new_rtree->rootpos = rootpos;
    new_rtree->ndims = ndims;
    new_rtree->nsides = 2 * ndims;
    /* hack to keep compatibility */
    if (ndims < 3)
        new_rtree->ndims_alloc = 3;
    else
        new_rtree->ndims_alloc = ndims;
 
    new_rtree->nsides_alloc = 2 * new_rtree->ndims_alloc;
 
    /* init free node positions */
    new_rtree->free_nodes.avail = 0;
    new_rtree->free_nodes.alloc = 0;
    new_rtree->free_nodes.pos = NULL;
 
    new_rtree->rectsize = new_rtree->nsides_alloc * sizeof(RectReal);
    new_rtree->branchsize = sizeof(struct RTree_Branch) -
                            sizeof(struct RTree_Rect) + new_rtree->rectsize;
    new_rtree->nodesize = sizeof(struct RTree_Node) -
                          sizeof(struct RTree_Branch *) +
                          MAXCARD * new_rtree->branchsize;
 
    /* create empty root node */
    n = RTreeAllocNode(new_rtree, 0);
    new_rtree->rootlevel = n->level = 0; /* leaf */
 
    /* use overflow by default */
    new_rtree->overflow = 1;
 
    if (fd > -1) { /* file based */
        /* nodecard and leafcard can be adjusted, must NOT be larger than
         * MAXCARD */
        new_rtree->nodecard = MAXCARD;
        new_rtree->leafcard = MAXCARD;
 
        /* initialize node buffer */
        new_rtree->nb = calloc(MAXLEVEL, sizeof(struct NodeBuffer *));
        new_rtree->nb[0] =
            calloc(MAXLEVEL * NODE_BUFFER_SIZE, sizeof(struct NodeBuffer));
        for (i = 1; i < MAXLEVEL; i++) {
            new_rtree->nb[i] = new_rtree->nb[i - 1] + NODE_BUFFER_SIZE;
        }
 
        new_rtree->used = malloc(MAXLEVEL * sizeof(int *));
        new_rtree->used[0] = malloc(MAXLEVEL * NODE_BUFFER_SIZE * sizeof(int));
        for (i = 0; i < MAXLEVEL; i++) {
            if (i)
                new_rtree->used[i] = new_rtree->used[i - 1] + NODE_BUFFER_SIZE;
            for (j = 0; j < NODE_BUFFER_SIZE; j++) {
                new_rtree->nb[i][j].dirty = 0;
                new_rtree->nb[i][j].pos = -1;
                /* usage order */
                new_rtree->used[i][j] = j;
 
                new_rtree->nb[i][j].n.branch =
                    malloc(MAXCARD * sizeof(struct RTree_Branch));
 
                /* alloc memory for rectangles */
                for (k = 0; k < MAXCARD; k++) {
                    new_rtree->nb[i][j].n.branch[k].rect.boundary =
                        RTreeAllocBoundary(new_rtree);
                }
            }
        }
 
        /* write empty root node */
        lseek(new_rtree->fd, rootpos, SEEK_SET);
        RTreeWriteNode(n, new_rtree);
        RTreeFreeNode(n);
        new_rtree->root = NULL;
 
        new_rtree->insert_rect = RTreeInsertRectF;
        new_rtree->delete_rect = RTreeDeleteRectF;
        new_rtree->search_rect = RTreeSearchF;
        new_rtree->valid_child = RTreeValidChildF;
    }
    else { /* memory based */
        new_rtree->nodecard = MAXCARD;
        new_rtree->leafcard = MAXCARD;
 
        new_rtree->insert_rect = RTreeInsertRectM;
        new_rtree->delete_rect = RTreeDeleteRectM;
        new_rtree->search_rect = RTreeSearchM;
        new_rtree->valid_child = RTreeValidChildM;
 
        new_rtree->root = n;
    }
 
    /* minimum number of remaining children for RTreeDeleteRect */
    /* NOTE: min fill can be changed if needed, must be < nodecard and leafcard.
     */
    new_rtree->min_node_fill = (new_rtree->nodecard - 2) / 2;
    new_rtree->min_leaf_fill = (new_rtree->leafcard - 2) / 2;
 
    /* balance criteria for node splitting */
    new_rtree->minfill_node_split = (new_rtree->nodecard - 1) / 2;
    new_rtree->minfill_leaf_split = (new_rtree->leafcard - 1) / 2;
 
    new_rtree->n_nodes = 1;
    new_rtree->n_leafs = 0;
 
    /* initialize temp variables */
    new_rtree->ns = malloc(MAXLEVEL * sizeof(struct nstack));
 
    new_rtree->p.cover[0].boundary = RTreeAllocBoundary(new_rtree);
    new_rtree->p.cover[1].boundary = RTreeAllocBoundary(new_rtree);
 
    new_rtree->tmpb1.rect.boundary = RTreeAllocBoundary(new_rtree);
    new_rtree->tmpb2.rect.boundary = RTreeAllocBoundary(new_rtree);
    new_rtree->c.rect.boundary = RTreeAllocBoundary(new_rtree);
 
    new_rtree->BranchBuf = malloc((MAXCARD + 1) * sizeof(struct RTree_Branch));
    for (i = 0; i <= MAXCARD; i++) {
        new_rtree->BranchBuf[i].rect.boundary = RTreeAllocBoundary(new_rtree);
    }
    new_rtree->rect_0.boundary = RTreeAllocBoundary(new_rtree);
    new_rtree->rect_1.boundary = RTreeAllocBoundary(new_rtree);
    new_rtree->upperrect.boundary = RTreeAllocBoundary(new_rtree);
    new_rtree->orect.boundary = RTreeAllocBoundary(new_rtree);
    new_rtree->center_n =
        (RectReal *)malloc(new_rtree->ndims_alloc * sizeof(RectReal));
 
    return new_rtree;
}
 
/*!
   \brief Enable/disable R*-tree forced reinsertion (overflow)
 
   For dynamic R*-trees with runtime insertion and deletion,
   forced reinsertion results in a more compact tree, searches are a bit
   faster. For static R*-trees (no insertion/deletion after creation)
   forced reinsertion can be disabled at the cost of slower searches.
 
   \param t pointer to RTree structure
   \param overflow binary flag
 
   \return nothing
 */
void RTreeSetOverflow(struct RTree *t, char overflow)
{
    t->overflow = overflow != 0;
}
 
/*!
   \brief Destroy an R*-Tree
 
   This method releases all memory allocated to a RTree. It deletes all
   rectangles and all memory allocated for internal support data.
   Note that for a file-based RTree, the file is not deleted and not
   closed. The file can thus be used to permanently store an RTree.
 
   \param t pointer to RTree structure
 
   \return nothing
 */
 
void RTreeDestroyTree(struct RTree *t)
{
    int i;
 
    assert(t);
 
    if (t->fd > -1) {
        int j, k;
 
        for (i = 0; i < MAXLEVEL; i++) {
            for (j = 0; j < NODE_BUFFER_SIZE; j++) {
                for (k = 0; k < MAXCARD; k++) {
                    RTreeFreeBoundary(&t->nb[i][j].n.branch[k].rect);
                }
                free(t->nb[i][j].n.branch);
            }
        }
 
        if (t->free_nodes.alloc)
            free(t->free_nodes.pos);
        free(t->nb[0]);
        free(t->nb);
        free(t->used[0]);
        free(t->used);
    }
    else if (t->root)
        RTreeDestroyNode(t->root, t->root->level ? t->nodecard : t->leafcard);
 
    /* free temp variables */
    free(t->ns);
 
    RTreeFreeBoundary(&(t->p.cover[0]));
    RTreeFreeBoundary(&(t->p.cover[1]));
 
    RTreeFreeBoundary(&(t->tmpb1.rect));
    RTreeFreeBoundary(&(t->tmpb2.rect));
    RTreeFreeBoundary(&(t->c.rect));
    for (i = 0; i <= MAXCARD; i++) {
        RTreeFreeBoundary(&(t->BranchBuf[i].rect));
    }
    free(t->BranchBuf);
    RTreeFreeBoundary(&(t->rect_0));
    RTreeFreeBoundary(&(t->rect_1));
    RTreeFreeBoundary(&(t->upperrect));
    RTreeFreeBoundary(&(t->orect));
    free(t->center_n);
 
    free(t);
 
    return;
}
 
/*!
   \brief Search an R*-Tree
 
   Search in an RTree for all data rectangles that overlap or touch the
   argument rectangle.
   Return the number of qualifying data rectangles.
   The search stops if the SearchHitCallBack function returns 0 (zero)
   or if there are no more qualifying data rectangles.
 
   \param t pointer to RTree structure
   \param r pointer to rectangle to use for searching
   \param shcb Search Hit CallBack function
   \param cbarg custom pointer used as argument for the shcb fn
 
   \return number of qualifying data rectangles
 */
/*
 *
 * add option to select operator to select rectangles ?
 * current: overlap
 * possible alternatives:
 *  - select all rectangles that are fully contained in r
 *  - select all rectangles that fully contain r
 */
int RTreeSearch(struct RTree *t, struct RTree_Rect *r, SearchHitCallback *shcb,
                void *cbarg)
{
    assert(r && t);
 
    return t->search_rect(t, r, shcb, cbarg);
}
 
/*!
   \brief Insert an item into a R*-Tree
 
   \param r pointer to rectangle to use for searching
   \param tid data id stored with rectangle, must be > 0
   \param t pointer to RTree structure
 
   \return number of qualifying data rectangles
 */
int RTreeInsertRect(struct RTree_Rect *r, int tid, struct RTree *t)
{
    union RTree_Child newchild;
 
    assert(r && t && tid > 0);
 
    t->n_leafs++;
    newchild.id = tid;
 
    return t->insert_rect(r, newchild, 0, t);
}
 
/*!
   \brief Delete an item from a R*-Tree
 
   This method deletes an item from the RTree. The rectangle passed to
   this method does not need to be the exact rectangle, the only
   requirement is that this rectangle overlaps with the rectangle to
   be deleted. The rectangle to be deleted is identified by its id.
 
   \param r pointer to rectangle to use for searching
   \param tid id of the data to be deleted, must be > 0
   \param t pointer to RTree structure
 
   \return 0 on success
   \return 1 if data item not found
 */
int RTreeDeleteRect(struct RTree_Rect *r, int tid, struct RTree *t)
{
    union RTree_Child child;
 
    assert(r && t && tid > 0);
 
    child.id = tid;
 
    return t->delete_rect(r, child, t);
}
 
/***********************************
 *    internally used functions    *
 ***********************************/
 
/*
 * Allocate space for a node in the list used in DeleteRect to
 * store Nodes that are too empty.
 */
struct RTree_ListNode *RTreeNewListNode(void)
{
    return (struct RTree_ListNode *)malloc(sizeof(struct RTree_ListNode));
}
 
void RTreeFreeListNode(struct RTree_ListNode *p)
{
    free(p);
}
 
/*
 * Add a node to the reinsertion list.  All its branches will later
 * be reinserted into the index structure.
 */
void RTreeReInsertNode(struct RTree_Node *n, struct RTree_ListNode **ee)
{
    struct RTree_ListNode *l = RTreeNewListNode();
 
    l->node = n;
    l->next = *ee;
    *ee = l;
}
 
/*
 * Free ListBranch, used by R*-type forced reinsertion
 */
void RTreeFreeListBranch(struct RTree_ListBranch *p)
{
    RTreeFreeBoundary(&(p->b.rect));
    free(p);
}