DijkstraVectors.h

Go to the documentation of this file.

#ifndef DijkstraVectors_h
#define DijkstraVectors_h

#ifdef WIN32
#pragma warning(disable:4786)  //necessary because stl generates longer
                               // names than bill's compiler can handle!
#endif

#include "DistanceTransform.h"
#include <set>
#include <vector>

using namespace std;
//----------------------------------------------------------------------
class DijkstraVectors : public DistanceTransform {

public:

    DijkstraVectors ( const int xSize, const int ySize,
                      const bool unload=true )
        : DistanceTransform(xSize, ySize, unload)
    {
    }

    void doTransform ( const unsigned char* const I );

    virtual inline bool getP ( const int x, const int y, int& px, int& py )
    const { return getP(x, y, px, py, 0); }

    inline bool getP ( const int x, const int y, int& px, int& py,
                       const int which ) const
    {
        if (this->p == NULL) {
            px = py = -1;
            return false;
        }

        const int i = sub(x,y);
        if (which < (int)p[i].size()) {
            px = p[i][which]->x;
            py = p[i][which]->y;
            // for (int j=0; j<(int)p[i].size(); j++) {
            //     cout << "P(" << x << "," << y << ")=(" << p[i][j].x << ","
            //          << p[i][j].y << ")" << endl;
            // }
            return true;
        }

        px = py = -1;
        return false;
    }

private:
    class SetElement {
        public:
            double d;  
            int    x;  
            int    y;  
        SetElement ( const double d, const int x, const int y ) {
            this->d = d;
            this->x = x;
            this->y = y;
        }
    };

    struct ltSetElement {
        bool operator() ( const SetElement* s1, const SetElement* s2 ) const {
            if (s1->d < s2->d)  return true;
            if (s1->d > s2->d)  return false;
            //both must have the same 'd' value
            if (s1->x < s2->x)  return true;
            if (s1->x > s2->x)  return false;
            //both must have the same 'd' and 'x' values
            if (s1->y < s2->y)  return true;
            if (s1->y > s2->y)  return false;
            return false;  //d, x, and y are equal
        }
    };

    vector<SetElement*>* p;

    inline void check ( double* d, const int center, const int X, const int Y,
        set< SetElement*, ltSetElement > &s );

};

#endif
//----------------------------------------------------------------------

---