wobble_suppression.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_WOBBLE_SUPPRESSION_HPP__
00044 #define __OPENCV_VIDEOSTAB_WOBBLE_SUPPRESSION_HPP__
00045 
00046 #include <vector>
00047 #include "opencv2/core.hpp"
00048 #include "opencv2/core/cuda.hpp"
00049 #include "opencv2/videostab/global_motion.hpp"
00050 #include "opencv2/videostab/log.hpp"
00051 
00052 namespace cv
00053 {
00054 namespace videostab
00055 {
00056 
00057 //! @addtogroup videostab
00058 //! @{
00059 
00060 class CV_EXPORTS WobbleSuppressorBase
00061 {
00062 public:
00063     WobbleSuppressorBase();
00064 
00065     virtual ~WobbleSuppressorBase() {}
00066 
00067     void setMotionEstimator(Ptr<ImageMotionEstimatorBase> val) { motionEstimator_ = val; }
00068     Ptr<ImageMotionEstimatorBase> motionEstimator() const { return motionEstimator_; }
00069 
00070     virtual void suppress(int idx, const Mat &frame, Mat &result) = 0;
00071 
00072 
00073     // data from stabilizer
00074 
00075     virtual void setFrameCount(int val) { frameCount_ = val; }
00076     virtual int frameCount() const { return frameCount_; }
00077 
00078     virtual void setMotions(const std::vector<Mat> &val) { motions_ = &val; }
00079     virtual const std::vector<Mat>& motions() const { return *motions_; }
00080 
00081     virtual void setMotions2(const std::vector<Mat> &val) { motions2_ = &val; }
00082     virtual const std::vector<Mat>& motions2() const { return *motions2_; }
00083 
00084     virtual void setStabilizationMotions(const std::vector<Mat> &val) { stabilizationMotions_ = &val; }
00085     virtual const std::vector<Mat>& stabilizationMotions() const { return *stabilizationMotions_; }
00086 
00087 protected:
00088     Ptr<ImageMotionEstimatorBase> motionEstimator_;
00089     int frameCount_;
00090     const std::vector<Mat> *motions_;
00091     const std::vector<Mat> *motions2_;
00092     const std::vector<Mat> *stabilizationMotions_;
00093 };
00094 
00095 class CV_EXPORTS NullWobbleSuppressor : public WobbleSuppressorBase
00096 {
00097 public:
00098     virtual void suppress(int idx, const Mat &frame, Mat &result);
00099 };
00100 
00101 class CV_EXPORTS MoreAccurateMotionWobbleSuppressorBase : public WobbleSuppressorBase
00102 {
00103 public:
00104     virtual void setPeriod(int val) { period_ = val; }
00105     virtual int period() const { return period_; }
00106 
00107 protected:
00108     MoreAccurateMotionWobbleSuppressorBase() { setPeriod(30); }
00109 
00110     int period_;
00111 };
00112 
00113 class CV_EXPORTS MoreAccurateMotionWobbleSuppressor : public MoreAccurateMotionWobbleSuppressorBase
00114 {
00115 public:
00116     virtual void suppress(int idx, const Mat &frame, Mat &result);
00117 
00118 private:
00119     Mat_<float> mapx_, mapy_;
00120 };
00121 
00122 #if defined(HAVE_OPENCV_CUDAWARPING)
00123 class CV_EXPORTS MoreAccurateMotionWobbleSuppressorGpu : public MoreAccurateMotionWobbleSuppressorBase
00124 {
00125 public:
00126     void suppress(int idx, const cuda::GpuMat &frame, cuda::GpuMat &result);
00127     virtual void suppress(int idx, const Mat &frame, Mat &result);
00128 
00129 private:
00130     cuda::GpuMat frameDevice_, resultDevice_;
00131     cuda::GpuMat mapx_, mapy_;
00132 };
00133 #endif
00134 
00135 //! @}
00136 
00137 } // namespace videostab
00138 } // namespace cv
00139 
00140 #endif
00141