ModifiedDijkstra.h

Go to the documentation of this file.

#ifndef ModifiedDijkstra_h
#define ModifiedDijkstra_h

#include "DistanceTransform.h"
#ifdef WIN32
#pragma warning(disable:4786)  //necessary because stl generates longer
                               // names than bill's compiler can handle!
#endif
#include <set>

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

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

    void doTransform ( const unsigned char* const I ) {
        modifiedDijkstra(I);
    }

    void modifiedDijkstra ( const unsigned char* const I );
    void modifiedDijkstra_DeadReckoning ( const unsigned char* const I );
    void modifiedDijkstra_ModifiedDeadReckoning ( const unsigned char* const I );

    virtual inline bool getP ( const int x, const int y, int& px, int& py ) const {
        if (p==NULL)    return false;
        const int s = sub(x,y);
        px = p[s].x;
        py = p[s].y;
        return true;
    }

protected:
    P *p;

private:
    typedef struct {
        double d;  
        int    x;  
        int    y;  
    } SetElement;

    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
        }
    };

    void init ( const unsigned char* const I, double* &d, SetElement* &se,
        set< const SetElement *, ltSetElement > &s );

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

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

---