Home | History | Annotate | Download | only in detail
      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:
     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:
     90     void feed(const std::vector<Point> &/*corners*/, const std::vector<UMat> &/*images*/,
     91               const std::vector<std::pair<UMat,uchar> > &/*masks*/) { }
     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:
    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