qualityssim.hpp

// This file is part of OpenCV project.
// It is subject to the license terms in the LICENSE file found in the top-level directory
// of this distribution and at http://opencv.org/license.html.
 
#ifndef OPENCV_QUALITY_QUALITYSSIM_HPP
#define OPENCV_QUALITY_QUALITYSSIM_HPP
 
#include "qualitybase.hpp"
 
namespace cv
{
namespace quality
{
 
class CV_EXPORTS_W QualitySSIM
    : public QualityBase {
public:
 
    CV_WRAP cv::Scalar compute( InputArray cmp ) CV_OVERRIDE;
 
    CV_WRAP bool empty() const CV_OVERRIDE { return _refImgData.empty() && QualityBase::empty(); }
 
    CV_WRAP void clear() CV_OVERRIDE { _refImgData = _mat_data(); QualityBase::clear(); }
 
    CV_WRAP static Ptr<QualitySSIM> create( InputArray ref );
 
    CV_WRAP static cv::Scalar compute( InputArray ref, InputArray cmp, OutputArray qualityMap );
 
protected:
 
    // holds computed values for a mat
    struct _mat_data
    {
        // internal mat type
        using mat_type = QualityBase::_mat_type;
 
        mat_type
            I
            , I_2
            , mu
            , mu_2
            , sigma_2
            ;
 
        // allow default construction
        _mat_data() = default;
 
        // construct from mat_type
        _mat_data(const mat_type&);
 
        // construct from inputarray
        _mat_data(InputArray);
 
        // return flag if this is empty
        bool empty() const { return I.empty() && I_2.empty() && mu.empty() && mu_2.empty() && sigma_2.empty(); }
 
        // computes ssim and quality map for single frame
        static std::pair<cv::Scalar, mat_type> compute(const _mat_data& lhs, const _mat_data& rhs);
 
    };  // mat_data
 
    _mat_data _refImgData;
 
    QualitySSIM( _mat_data refImgData )
        : _refImgData( std::move(refImgData) )
    {}
 
};  // QualitySSIM
}   // quality
}   // cv
#endif