Subversion Repositories seema-scanner

#ifndef CODEC_H

#define CODEC_H

#include <vector>

#include <opencv2/opencv.hpp>

#include "SMTypes.h"

enum CodingDir {CodingDirNone = 0,

               CodingDirHorizontal = 1 << 0,

               CodingDirVertical = 1 << 1,

               CodingDirBoth = CodingDirHorizontal | CodingDirVertical};

// Base class for all Algorithms

class Algorithm {

    public:

        Algorithm(unsigned int _screenCols, unsigned int _screenRows, CodingDir _dir = CodingDirHorizontal) : N(0),screenCols(_screenCols), screenRows(_screenRows),  dir(_dir){}

        unsigned int getNPatterns(){return N;}

        CodingDir getDir(){return dir;}

        // Encoding

        virtual cv::Mat getEncodingPattern(unsigned int depth) = 0;

        virtual ~Algorithm(){}

    protected:

        unsigned int N;

        unsigned int screenCols, screenRows;

        CodingDir dir;

};

class Algorithm {

    public:

        Algorithm(CodingDir _dir, int _screenCols, int _screenRows) : dir(_dir), screenCols(_screenCols), screenRows(_screenRows){}

        CodingDir getDir(){return dir;}

        int getscreenCols(){return screenCols;}

        int getscreenRows(){return screenRows;}

        // Decoding

        virtual void getCorrespondences(const std::vector<cv::Mat>& frames0, const std::vector<cv::Mat>& frames1, std::vector<cv::Point2f>& q0, std::vector<cv::Point2f>& q1, std::vector<cv::Point3f>& color) = 0;

        virtual ~Algorithm(){}

    protected:

        CodingDir dir;

        int screenCols, screenRows;

};

#endif // CODEC_H