RSpatialIndex.h

Go to the documentation of this file.

#ifndef RSPATIALINDEX_H
#define RSPATIALINDEX_H

#include <stdint.h>

#include <QSet>
#include <QList>

#include "RBox.h"
#include "RDebug.h"
#include "RMath.h"
#include "RRequireHeap.h"
#include "RSpatialIndexVisitor.h"

class RSpatialIndexDebugVisitor : public RSpatialIndexVisitor {
public:
    RSpatialIndexDebugVisitor(QDebug dbg) : dbg(dbg) {}
    virtual ~RSpatialIndexDebugVisitor() {}
    virtual void visitData(
        int id,
        int pos,
        double x1, double y1, double z1,
        double x2, double y2, double z2);

    virtual void visitNode(
        double x1, double y1, double z1,
        double x2, double y2, double z2) {
        Q_UNUSED(x1)
        Q_UNUSED(y1)
        Q_UNUSED(z1)
        Q_UNUSED(x2)
        Q_UNUSED(y2)
        Q_UNUSED(z2)
    }

    QDebug dbg;
    QList<QPair<int, int> > matches;
};


class RSpatialIndex: public RRequireHeap {
public:
    RSpatialIndex() {
    }

    virtual ~RSpatialIndex() {
    }

    static qint64 getSIId(int id, int pos);
    static int getId(qint64 siid);
    static int getPos(qint64 siid);

    virtual void clear() = 0;

    virtual void addToIndex(
        int id, int pos,
        double x1, double y1, double z1,
        double x2, double y2, double z2
    ) = 0;

    virtual void addToIndex(
        int id, int pos,
        const RBox& bb
    );

    virtual void addToIndex(
        int id,
        const QList<RBox>& bbs
    );

    //virtual void removeFromIndex(int id) = 0;

    virtual bool removeFromIndex(
            int id, int pos,
            double x1, double y1, double z1,
            double x2, double y2, double z2) = 0;

    virtual bool removeFromIndex(int id, int pos, const RBox& bb);

    virtual bool removeFromIndex(int id, const QList<RBox>& bb);

    virtual QMap<int, QSet<int> > queryIntersected(
        double x1, double y1, double z1,
        double x2, double y2, double z2,
        RSpatialIndexVisitor* dataVisitor=NULL
    ) = 0;

    virtual QMap<int, QSet<int> > queryIntersected(
        const RBox& b,
        RSpatialIndexVisitor* dataVisitor=NULL
    );

    virtual QMap<int, QSet<int> > queryContained(
        double x1, double y1, double z1,
        double x2, double y2, double z2,
        RSpatialIndexVisitor* dataVisitor=NULL
    ) = 0;

    virtual QMap<int, QSet<int> > queryContained(
        const RBox& b,
        RSpatialIndexVisitor* dataVisitor=NULL
    );

    virtual QMap<int, QSet<int> > queryNearestNeighbor(
        uint k,
        double x, double y, double z,
        RSpatialIndexVisitor* dataVisitor=NULL
    ) = 0;

    virtual QPair<int, int> queryNearestNeighbor(double x, double y, double z);

protected:
    int idCounter;
};

QDebug operator<<(QDebug dbg, RSpatialIndex& si);

Q_DECLARE_METATYPE(RSpatialIndex*)

#endif