deblurring.hpp

00001 /*M///////////////////////////////////////////////////////////////////////////////////////
00002 //
00003 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
00004 //
00005 //  By downloading, copying, installing or using the software you agree to this license.
00006 //  If you do not agree to this license, do not download, install,
00007 //  copy or use the software.
00008 //
00009 //
00010 //                           License Agreement
00011 //                For Open Source Computer Vision Library
00012 //
00013 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
00014 // Copyright (C) 2009-2011, Willow Garage Inc., all rights reserved.
00015 // Third party copyrights are property of their respective owners.
00016 //
00017 // Redistribution and use in source and binary forms, with or without modification,
00018 // are permitted provided that the following conditions are met:
00019 //
00020 //   * Redistribution's of source code must retain the above copyright notice,
00021 //     this list of conditions and the following disclaimer.
00022 //
00023 //   * Redistribution's in binary form must reproduce the above copyright notice,
00024 //     this list of conditions and the following disclaimer in the documentation
00025 //     and/or other materials provided with the distribution.
00026 //
00027 //   * The name of the copyright holders may not be used to endorse or promote products
00028 //     derived from this software without specific prior written permission.
00029 //
00030 // This software is provided by the copyright holders and contributors "as is" and
00031 // any express or implied warranties, including, but not limited to, the implied
00032 // warranties of merchantability and fitness for a particular purpose are disclaimed.
00033 // In no event shall the Intel Corporation or contributors be liable for any direct,
00034 // indirect, incidental, special, exemplary, or consequential damages
00035 // (including, but not limited to, procurement of substitute goods or services;
00036 // loss of use, data, or profits; or business interruption) however caused
00037 // and on any theory of liability, whether in contract, strict liability,
00038 // or tort (including negligence or otherwise) arising in any way out of
00039 // the use of this software, even if advised of the possibility of such damage.
00040 //
00041 //M*/
00042 
00043 #ifndef __OPENCV_VIDEOSTAB_DEBLURRING_HPP__
00044 #define __OPENCV_VIDEOSTAB_DEBLURRING_HPP__
00045 
00046 #include <vector>
00047 #include "opencv2/core.hpp"
00048 
00049 namespace cv
00050 {
00051 namespace videostab
00052 {
00053 
00054 //! @addtogroup videostab
00055 //! @{
00056 
00057 CV_EXPORTS float calcBlurriness(const Mat &frame);
00058 
00059 class CV_EXPORTS DeblurerBase
00060 {
00061 public:
00062     DeblurerBase() : radius_(0), frames_(0), motions_(0), blurrinessRates_(0) {}
00063 
00064     virtual ~DeblurerBase() {}
00065 
00066     virtual void setRadius(int val) { radius_ = val; }
00067     virtual int radius() const { return radius_; }
00068 
00069     virtual void deblur(int idx, Mat &frame) = 0;
00070 
00071 
00072     // data from stabilizer
00073 
00074     virtual void setFrames(const std::vector<Mat> &val) { frames_ = &val; }
00075     virtual const std::vector<Mat>& frames() const { return *frames_; }
00076 
00077     virtual void setMotions(const std::vector<Mat> &val) { motions_ = &val; }
00078     virtual const std::vector<Mat>& motions() const { return *motions_; }
00079 
00080     virtual void setBlurrinessRates(const std::vector<float> &val) { blurrinessRates_ = &val; }
00081     virtual const std::vector<float>& blurrinessRates() const { return *blurrinessRates_; }
00082 
00083 protected:
00084     int radius_;
00085     const std::vector<Mat> *frames_;
00086     const std::vector<Mat> *motions_;
00087     const std::vector<float> *blurrinessRates_;
00088 };
00089 
00090 class CV_EXPORTS NullDeblurer : public DeblurerBase
00091 {
00092 public:
00093     virtual void deblur(int /*idx*/, Mat &/*frame*/) {}
00094 };
00095 
00096 class CV_EXPORTS WeightingDeblurer : public DeblurerBase
00097 {
00098 public:
00099     WeightingDeblurer();
00100 
00101     void setSensitivity(float val) { sensitivity_ = val; }
00102     float sensitivity() const { return sensitivity_; }
00103 
00104     virtual void deblur(int idx, Mat &frame);
00105 
00106 private:
00107     float sensitivity_;
00108     Mat_<float> bSum_, gSum_, rSum_, wSum_;
00109 };
00110 
00111 //! @}
00112 
00113 } // namespace videostab
00114 } // namespace cv
00115 
00116 #endif
00117