ChessSquareMesh.hpp

Go to the documentation of this file.

#ifndef LOPER_CHESSSQUAREMESH
#define LOPER_CHESSSQUAREMESH

#include "Contour.hpp"
#include "ImageCalibrator.hpp"
#include <set>
using std::set;

class ChessCorner;
class ChessSquare;
class ChessSquareMesh;

class ChessCorner  {
public:
    ChessCorner()
    {
        link = NULL;
        pos.resize(2);
    }
    FloatVec pos;
    void *link;
};



class ChessSquare
{
public:
    ChessSquare (Contour &c);
    bool isFullyConnected() const;
    float getSummedDistanceFromPoint(FloatVec point);
    PointCorrespondence getPointCorrespondence(int whichCorner, int orientation, int curX, int curY);
    FloatVec getCenter();

    ChessCorner corners[4];
};



class ChessSquareMesh
{
public:
    void addPotentialSquare (const Contour &c);
    std::vector< ChessSquare *> getConnectedSquares();
    bool hasFullyConnectedNeighbors(unsigned int index);
    int findClosestSquareIndex(const FloatVec & center);
    void connectPotentialSquares();

    void surfNeighbors (int curX, int curY, int orientation, set< ChessSquare *> & squaresDone, std::vector< PointCorrespondence > & points, int indexToSurf);
    std::vector< ChessSquare > squares;
private:
    void findClosest (int curSquare1, int curCorner1, int *curSquare2, int *curCorner2);
};



#endif // LOPER_CHESSSQUAREMESH

---