ImgProc.h

#ifndef IMGPROC_H
#define IMGPROC_H
#include <opencv2/imgproc.hpp>


class ImgProc {
    public:
        virtual ~ImgProc() {}
        virtual void gaussianFilter(const cv::Mat& src, cv::Mat& dst) const = 0;
        virtual void medianFilter(const cv::Mat& src, cv::Mat& dst) const = 0;
        virtual void bilateralFilter(const cv::Mat& src, cv::Mat& dst) const = 0;
        virtual void histEqualize(const cv::Mat& src, cv::Mat& dst) const = 0;
        virtual void lowPassFilter(const cv::Mat& src, cv::Mat& dst) const = 0;
        virtual void lowPassFilterFreq(const cv::Mat& src, cv::Mat& dst) const = 0;
        virtual void highPassFilter(const cv::Mat& src, cv::Mat& dst) const = 0;
        virtual void highPassFilterFreq(const cv::Mat& src, cv::Mat& dst) const = 0;
        virtual void rgbFilter(const cv::Mat& src, cv::Mat& dst, std::function<void(const cv::Mat&, cv::Mat&)> filter) const = 0;
        virtual void freqFilter(const cv::Mat& src, cv::Mat& dst, cv::Mat mask) const = 0;
        virtual cv::Mat highPassMask(cv::Size maskSize) const = 0;
        virtual cv::Mat lowPassMask(cv::Size maskSize) const = 0;
};


/* This is not used for now (It should be used if we needed to support runtime implementation change)*/
class ImgProcImplementation {
    private:
        ImgProc *_imgProc;

    public:
        ImgProcImplementation(ImgProc *imgProc = nullptr) : _imgProc(imgProc) { }
        ~ImgProcImplementation() { delete this->_imgProc; }

        void setImplementation(ImgProc *imgProc) {
            delete this->_imgProc;
            this->_imgProc = imgProc;
        }

        void gaussianFilter(const cv::Mat& src, cv::Mat& dst) const;
        void medianFilter(const cv::Mat& src, cv::Mat& dst) const;
        void averageFilter(const cv::Mat& src, cv::Mat& dst) const;
        void bilateralFilter(const cv::Mat& src, cv::Mat& dst) const;
};


/* This is not used for now (This is a skeleton implementation that do nothing)*/
class FromScratchImgProc : public ImgProc {
    public:
        void gaussianFilter(const cv::Mat& src, cv::Mat& dst) const override { dst = src; }
        void medianFilter(const cv::Mat& src, cv::Mat& dst) const override { dst = src; }
        void lowPassFilter(const cv::Mat& src, cv::Mat& dst) const override { dst = src; }
        void bilateralFilter(const cv::Mat& src, cv::Mat& dst) const override { dst = src; }
};


class OpenCvImgProc : public ImgProc {
    public:
        void gaussianFilter(const cv::Mat& src, cv::Mat& dst) const override;
        void medianFilter(const cv::Mat& src, cv::Mat& dst) const override;
        void bilateralFilter(const cv::Mat& src, cv::Mat& dst) const override;
        void lowPassFilter(const cv::Mat& src, cv::Mat& dst) const override;
        void lowPassFilterFreq(const cv::Mat& src, cv::Mat& dst) const override;
        void histEqualize(const cv::Mat& src, cv::Mat& dst) const override;
        void highPassFilter(const cv::Mat& src, cv::Mat& dst) const override;
        void highPassFilterFreq(const cv::Mat& src, cv::Mat& dst) const override;
        void rgbFilter(const cv::Mat& src, cv::Mat& dst, std::function<void(const cv::Mat&, cv::Mat&)> filter) const override;
        void freqFilter(const cv::Mat& src, cv::Mat& dst, cv::Mat mask) const override;
        cv::Mat highPassMask(cv::Size maskSize) const override;
        cv::Mat lowPassMask(cv::Size maskSize) const override;
};

#endif // IMGPROC_H