Subversion Repositories seema-scanner

#ifndef CODEC_H

#define CODEC_H

#include <vector>

#include <opencv2/opencv.hpp>

#include "SMTypes.h"

enum CodecDir {CodecDirNone = 0,

               CodecDirHorizontal = 1 << 0,

               CodecDirVertical = 1 << 1,

               CodecDirBoth = CodecDirHorizontal | CodecDirVertical};

// Base class for all encoders

class Encoder {

    public:

        Encoder(unsigned int _screenCols, unsigned int _screenRows, CodecDir _dir = CodecDirHorizontal) : N(0),screenCols(_screenCols), screenRows(_screenRows),  dir(_dir){}

        unsigned int getNPatterns(){return N;}

        CodecDir getDir(){return dir;}

        // Encoding

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

        virtual ~Encoder(){}

    protected:

        unsigned int N;

        unsigned int screenCols, screenRows;

        CodecDir dir;

};

class Decoder {

    public:

        Decoder(CodecDir _dir = CodecDirHorizontal) : dir(_dir){}

        CodecDir getDir(){return dir;}

        // Decoding

        virtual void decodeFrames(const std::vector<cv::Mat> frames, cv::Mat &up, cv::Mat &vp, cv::Mat &mask, cv::Mat &shading) = 0;

        virtual ~Decoder(){}

    protected:

        CodecDir dir;

};

#endif // CODEC_H