RSpatialIndexNavel.h

Go to the documentation of this file.

#ifndef RSPATIALINDEXNAVEL_H
#define RSPATIALINDEXNAVEL_H

#include <QSet>
#include <QList>

#include "RSpatialIndex.h"

#include "spatialindex/include/SpatialIndex.h"

class RSiGetRegionVisitor: public RSpatialIndexVisitor {
public:
    RSiGetRegionVisitor(int id) :
        id(id), found(false) {
    }
    virtual ~RSiGetRegionVisitor() {
    }
    virtual void visitData(int id, int pos,
        double x1, double y1, double z1, 
        double x2, double y2, double z2) {

        Q_UNUSED(pos);

        if (this->id == id) {
            double p1[] = { x1, y1, z1 };
            double p2[] = { x2, y2, z2 };

            region = SpatialIndex::Region(p1, p2, 3);

            found = true;
        }
    }
    virtual void visitNode(
        double /*x1*/, double /*y1*/, double /*z1*/,
        double /*x2*/, double /*y2*/, double /*z2*/) {}

public:
    int id;
    SpatialIndex::Region region;
    bool found;
};



class RSpatialIndexNavel: public RSpatialIndex {
protected:
    class RSiRegion: public SpatialIndex::Region {
    public:
        RSiRegion(double x1, double y1, double z1, double x2, double y2,
                double z2);
    };

    class RSiPoint: public SpatialIndex::Point {
    public:
        RSiPoint(double x, double y, double z);
    };

    class Visitor: public SpatialIndex::IVisitor {
    public:
        Visitor(QMap<int, QSet<int> >& ids, RSpatialIndexVisitor* dataVisitor = NULL) :
            ids(ids), dataVisitor(dataVisitor) {
        }

        void visitNode(const SpatialIndex::INode& n) {
            //printf("Visitor::visitNode: %Ld\n", n.getIdentifier());
            if (dataVisitor != NULL) {
                SpatialIndex::IShape* shape;
                n.getShape(&shape);
                if (shape == NULL) {
                    fprintf(stderr, "error: node has no shape\n");
                    return;
                }
                SpatialIndex::Region* region =
                        dynamic_cast<SpatialIndex::Region*> (shape);
                if (region == NULL) {
                    fprintf(stderr,
                            "error: node shape in spacial index is not a SpatialIndex::Region\n");
                    return;
                }
                dataVisitor->visitNode(region->m_pLow[0],
                        region->m_pLow[1], region->m_pLow[2],
                        region->m_pHigh[0], region->m_pHigh[1],
                        region->m_pHigh[2]);
            }
        }

        void visitData(const SpatialIndex::IData& d) {
            qint64 siid = d.getIdentifier();
            int id = RSpatialIndex::getId(siid);
            int pos = RSpatialIndex::getPos(siid);

            //if (ids.contains(id)) {
                ids[id].insert(pos);
                /*
                qFatal("Visitor.visitData: double entry found in spatial index, identifier: %d", 
                    (int)d.getIdentifier());
                Q_ASSERT(false);
                */
//            }
//            else {
//                ids.insert(id, QSet<int>() << pos);
//            }

            //ids.insert(d.getIdentifier());

            if (dataVisitor != NULL) {
                SpatialIndex::IShape* shape;
                d.getShape(&shape);
                if (shape == NULL) {
                    fprintf(stderr, "error: data node has no shape\n");
                    return;
                }
                SpatialIndex::Region* region =
                        dynamic_cast<SpatialIndex::Region*> (shape);
                if (region == NULL) {
                    fprintf(stderr,
                            "error: shape in spacial index is not a SpatialIndex::Region\n");
                    return;
                }

                //size_t len;
                //uint8_t* data;
                //d.getData(len, &data);
                dataVisitor->visitData(id, pos, region->m_pLow[0],
                        region->m_pLow[1], region->m_pLow[2],
                        region->m_pHigh[0], region->m_pHigh[1],
                        region->m_pHigh[2]);
                        //region->getX1(), region->getY1(), region->getZ1(),
                        //region->getX2(), region->getY2(), region->getZ2(),
                        //len, data);
                //delete data;
                //delete region;
                delete shape;
            }
        }

        void visitData(std::vector<const SpatialIndex::IData*>& v) {
            std::vector<const SpatialIndex::IData*>::iterator it;
            for (it = v.begin(); it != v.end(); it++) {
                printf("Visitor::visitData[]: %Ld\n", (*it)->getIdentifier());
            }
        }

    private:
        QMap<int, QSet<int> >& ids;
        RSpatialIndexVisitor* dataVisitor;
    };

public:
    RSpatialIndexNavel();
    ~RSpatialIndexNavel();

    //static int dataToInt(const uint8_t* data);
    //static void intToData(int i, uint8_t* data);

    virtual void clear();

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

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

    //void removeFromIndex(int id);
    virtual bool removeFromIndex(int id, const QList<RBox>& bb);
    virtual bool removeFromIndex(int id, int pos, const RBox& bb);
    virtual bool removeFromIndex(
            int id,
            int pos,
            double x1, double y1, double z1,
            double x2, double y2, double z2);

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

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

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

protected:
    void addToIndex(int id, int pos, const RSiRegion& region, size_t dataLength = 0,
            const uint8_t* data = NULL);
    QMap<int, QSet<int> > queryIntersected(const RSiRegion& region,
            RSpatialIndexVisitor* dataVisitor = NULL);
    QMap<int, QSet<int> > queryContained(const RSiRegion& region,
            RSpatialIndexVisitor* dataVisitor = NULL);
    QMap<int, QSet<int> > queryNearestNeighbor(uint k, const RSiPoint& point,
            RSpatialIndexVisitor* dataVisitor = NULL);

//    QSet<int> sidsToIds(const QSet<int> sids);
//    QSet<int> idToSids(int id);

protected:
    void init();
    void uninit();

protected:
    SpatialIndex::ISpatialIndex* tree;
    SpatialIndex::IStorageManager* buff;

//    QMap<int, int> sidToId;
//    QMap<int, QList<int> > idToSid;
//    int sidCounter;
};

Q_DECLARE_METATYPE(RSpatialIndexNavel*)

#endif