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, 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_STITCHING_EXPOSURE_COMPENSATE_HPP__
44 #define __OPENCV_STITCHING_EXPOSURE_COMPENSATE_HPP__
45 
46 #include "opencv2/core.hpp"
47 
48 namespace cv {
49 namespace detail {
50 
51 //! @addtogroup stitching_exposure
52 //! @{
53 
54 /** @brief Base class for all exposure compensators.
55  */
56 class CV_EXPORTS ExposureCompensator
57 {
58 public:
~ExposureCompensator()59     virtual ~ExposureCompensator() {}
60 
61     enum { NO, GAIN, GAIN_BLOCKS };
62     static Ptr<ExposureCompensator> createDefault(int type);
63 
64     /**
65     @param corners Source image top-left corners
66     @param images Source images
67     @param masks Image masks to update (second value in pair specifies the value which should be used
68     to detect where image is)
69      */
70     void feed(const std::vector<Point> &corners, const std::vector<UMat> &images,
71               const std::vector<UMat> &masks);
72     /** @overload */
73     virtual void feed(const std::vector<Point> &corners, const std::vector<UMat> &images,
74                       const std::vector<std::pair<UMat,uchar> > &masks) = 0;
75     /** @brief Compensate exposure in the specified image.
76 
77     @param index Image index
78     @param corner Image top-left corner
79     @param image Image to process
80     @param mask Image mask
81      */
82     virtual void apply(int index, Point corner, InputOutputArray image, InputArray mask) = 0;
83 };
84 
85 /** @brief Stub exposure compensator which does nothing.
86  */
87 class CV_EXPORTS NoExposureCompensator : public ExposureCompensator
88 {
89 public:
feed(const std::vector<Point> &,const std::vector<UMat> &,const std::vector<std::pair<UMat,uchar>> &)90     void feed(const std::vector<Point> &/*corners*/, const std::vector<UMat> &/*images*/,
91               const std::vector<std::pair<UMat,uchar> > &/*masks*/) { }
apply(int,Point,InputOutputArray,InputArray)92     void apply(int /*index*/, Point /*corner*/, InputOutputArray /*image*/, InputArray /*mask*/) { }
93 };
94 
95 /** @brief Exposure compensator which tries to remove exposure related artifacts by adjusting image
96 intensities, see @cite BL07 and @cite WJ10 for details.
97  */
98 class CV_EXPORTS GainCompensator : public ExposureCompensator
99 {
100 public:
101     void feed(const std::vector<Point> &corners, const std::vector<UMat> &images,
102               const std::vector<std::pair<UMat,uchar> > &masks);
103     void apply(int index, Point corner, InputOutputArray image, InputArray mask);
104     std::vector<double> gains() const;
105 
106 private:
107     Mat_<double> gains_;
108 };
109 
110 /** @brief Exposure compensator which tries to remove exposure related artifacts by adjusting image block
111 intensities, see @cite UES01 for details.
112  */
113 class CV_EXPORTS BlocksGainCompensator : public ExposureCompensator
114 {
115 public:
BlocksGainCompensator(int bl_width=32,int bl_height=32)116     BlocksGainCompensator(int bl_width = 32, int bl_height = 32)
117             : bl_width_(bl_width), bl_height_(bl_height) {}
118     void feed(const std::vector<Point> &corners, const std::vector<UMat> &images,
119               const std::vector<std::pair<UMat,uchar> > &masks);
120     void apply(int index, Point corner, InputOutputArray image, InputArray mask);
121 
122 private:
123     int bl_width_, bl_height_;
124     std::vector<UMat> gain_maps_;
125 };
126 
127 //! @}
128 
129 } // namespace detail
130 } // namespace cv
131 
132 #endif // __OPENCV_STITCHING_EXPOSURE_COMPENSATE_HPP__
133