1 /*M///////////////////////////////////////////////////////////////////////////////////////
2 //
3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
4 //
5 //  By downloading, copying, installing or using the software you agree to this license.
6 //  If you do not agree to this license, do not download, install,
7 //  copy or use the software.
8 //
9 //
10 //                           License Agreement
11 //                For Open Source Computer Vision Library
12 //
13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
14 // Copyright (C) 2009-2011, Willow Garage Inc., all rights reserved.
15 // Third party copyrights are property of their respective owners.
16 //
17 // Redistribution and use in source and binary forms, with or without modification,
18 // are permitted provided that the following conditions are met:
19 //
20 //   * Redistribution's of source code must retain the above copyright notice,
21 //     this list of conditions and the following disclaimer.
22 //
23 //   * Redistribution's in binary form must reproduce the above copyright notice,
24 //     this list of conditions and the following disclaimer in the documentation
25 //     and/or other materials provided with the distribution.
26 //
27 //   * The name of the copyright holders may not be used to endorse or promote products
28 //     derived from this software without specific prior written permission.
29 //
30 // This software is provided by the copyright holders and contributors "as is" and
31 // any express or implied warranties, including, but not limited to, the implied
32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
33 // In no event shall the Intel Corporation or contributors be liable for any direct,
34 // indirect, incidental, special, exemplary, or consequential damages
35 // (including, but not limited to, procurement of substitute goods or services;
36 // loss of use, data, or profits; or business interruption) however caused
37 // and on any theory of liability, whether in contract, strict liability,
38 // or tort (including negligence or otherwise) arising in any way out of
39 // the use of this software, even if advised of the possibility of such damage.
40 //
41 //M*/
42 
43 #ifndef __OPENCV_VIDEOSTAB_DEBLURRING_HPP__
44 #define __OPENCV_VIDEOSTAB_DEBLURRING_HPP__
45 
46 #include <vector>
47 #include "opencv2/core.hpp"
48 
49 namespace cv
50 {
51 namespace videostab
52 {
53 
54 //! @addtogroup videostab
55 //! @{
56 
57 CV_EXPORTS float calcBlurriness(const Mat &frame);
58 
59 class CV_EXPORTS DeblurerBase
60 {
61 public:
DeblurerBase()62     DeblurerBase() : radius_(0), frames_(0), motions_(0), blurrinessRates_(0) {}
63 
~DeblurerBase()64     virtual ~DeblurerBase() {}
65 
setRadius(int val)66     virtual void setRadius(int val) { radius_ = val; }
radius() const67     virtual int radius() const { return radius_; }
68 
69     virtual void deblur(int idx, Mat &frame) = 0;
70 
71 
72     // data from stabilizer
73 
setFrames(const std::vector<Mat> & val)74     virtual void setFrames(const std::vector<Mat> &val) { frames_ = &val; }
frames() const75     virtual const std::vector<Mat>& frames() const { return *frames_; }
76 
setMotions(const std::vector<Mat> & val)77     virtual void setMotions(const std::vector<Mat> &val) { motions_ = &val; }
motions() const78     virtual const std::vector<Mat>& motions() const { return *motions_; }
79 
setBlurrinessRates(const std::vector<float> & val)80     virtual void setBlurrinessRates(const std::vector<float> &val) { blurrinessRates_ = &val; }
blurrinessRates() const81     virtual const std::vector<float>& blurrinessRates() const { return *blurrinessRates_; }
82 
83 protected:
84     int radius_;
85     const std::vector<Mat> *frames_;
86     const std::vector<Mat> *motions_;
87     const std::vector<float> *blurrinessRates_;
88 };
89 
90 class CV_EXPORTS NullDeblurer : public DeblurerBase
91 {
92 public:
deblur(int,Mat &)93     virtual void deblur(int /*idx*/, Mat &/*frame*/) {}
94 };
95 
96 class CV_EXPORTS WeightingDeblurer : public DeblurerBase
97 {
98 public:
99     WeightingDeblurer();
100 
setSensitivity(float val)101     void setSensitivity(float val) { sensitivity_ = val; }
sensitivity() const102     float sensitivity() const { return sensitivity_; }
103 
104     virtual void deblur(int idx, Mat &frame);
105 
106 private:
107     float sensitivity_;
108     Mat_<float> bSum_, gSum_, rSum_, wSum_;
109 };
110 
111 //! @}
112 
113 } // namespace videostab
114 } // namespace cv
115 
116 #endif
117