Renesas / opencv-lib

openCV library for Renesas RZ/A

Dependents:   RZ_A2M_Mbed_samples

include/opencv2/videostab/fast_marching.hpp@0:0e0631af0305, 2021-01-29 (annotated)

Committer:
RyoheiHagimoto
Date:
Fri Jan 29 04:53:38 2021 +0000
Revision:
0:0e0631af0305
copied from https://github.com/d-kato/opencv-lib.

Who changed what in which revision?

UserRevisionLine numberNew contents of line
RyoheiHagimoto 0:0e0631af0305 1 /*M///////////////////////////////////////////////////////////////////////////////////////
RyoheiHagimoto 0:0e0631af0305 2 //
RyoheiHagimoto 0:0e0631af0305 3 // IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
RyoheiHagimoto 0:0e0631af0305 4 //
RyoheiHagimoto 0:0e0631af0305 5 // By downloading, copying, installing or using the software you agree to this license.
RyoheiHagimoto 0:0e0631af0305 6 // If you do not agree to this license, do not download, install,
RyoheiHagimoto 0:0e0631af0305 7 // copy or use the software.
RyoheiHagimoto 0:0e0631af0305 8 //
RyoheiHagimoto 0:0e0631af0305 9 //
RyoheiHagimoto 0:0e0631af0305 10 // License Agreement
RyoheiHagimoto 0:0e0631af0305 11 // For Open Source Computer Vision Library
RyoheiHagimoto 0:0e0631af0305 12 //
RyoheiHagimoto 0:0e0631af0305 13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
RyoheiHagimoto 0:0e0631af0305 14 // Copyright (C) 2009-2011, Willow Garage Inc., all rights reserved.
RyoheiHagimoto 0:0e0631af0305 15 // Third party copyrights are property of their respective owners.
RyoheiHagimoto 0:0e0631af0305 16 //
RyoheiHagimoto 0:0e0631af0305 17 // Redistribution and use in source and binary forms, with or without modification,
RyoheiHagimoto 0:0e0631af0305 18 // are permitted provided that the following conditions are met:
RyoheiHagimoto 0:0e0631af0305 19 //
RyoheiHagimoto 0:0e0631af0305 20 // * Redistribution's of source code must retain the above copyright notice,
RyoheiHagimoto 0:0e0631af0305 21 // this list of conditions and the following disclaimer.
RyoheiHagimoto 0:0e0631af0305 22 //
RyoheiHagimoto 0:0e0631af0305 23 // * Redistribution's in binary form must reproduce the above copyright notice,
RyoheiHagimoto 0:0e0631af0305 24 // this list of conditions and the following disclaimer in the documentation
RyoheiHagimoto 0:0e0631af0305 25 // and/or other materials provided with the distribution.
RyoheiHagimoto 0:0e0631af0305 26 //
RyoheiHagimoto 0:0e0631af0305 27 // * The name of the copyright holders may not be used to endorse or promote products
RyoheiHagimoto 0:0e0631af0305 28 // derived from this software without specific prior written permission.
RyoheiHagimoto 0:0e0631af0305 29 //
RyoheiHagimoto 0:0e0631af0305 30 // This software is provided by the copyright holders and contributors "as is" and
RyoheiHagimoto 0:0e0631af0305 31 // any express or implied warranties, including, but not limited to, the implied
RyoheiHagimoto 0:0e0631af0305 32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
RyoheiHagimoto 0:0e0631af0305 33 // In no event shall the Intel Corporation or contributors be liable for any direct,
RyoheiHagimoto 0:0e0631af0305 34 // indirect, incidental, special, exemplary, or consequential damages
RyoheiHagimoto 0:0e0631af0305 35 // (including, but not limited to, procurement of substitute goods or services;
RyoheiHagimoto 0:0e0631af0305 36 // loss of use, data, or profits; or business interruption) however caused
RyoheiHagimoto 0:0e0631af0305 37 // and on any theory of liability, whether in contract, strict liability,
RyoheiHagimoto 0:0e0631af0305 38 // or tort (including negligence or otherwise) arising in any way out of
RyoheiHagimoto 0:0e0631af0305 39 // the use of this software, even if advised of the possibility of such damage.
RyoheiHagimoto 0:0e0631af0305 40 //
RyoheiHagimoto 0:0e0631af0305 41 //M*/
RyoheiHagimoto 0:0e0631af0305 42
RyoheiHagimoto 0:0e0631af0305 43 #ifndef OPENCV_VIDEOSTAB_FAST_MARCHING_HPP
RyoheiHagimoto 0:0e0631af0305 44 #define OPENCV_VIDEOSTAB_FAST_MARCHING_HPP
RyoheiHagimoto 0:0e0631af0305 45
RyoheiHagimoto 0:0e0631af0305 46 #include <cmath>
RyoheiHagimoto 0:0e0631af0305 47 #include <queue>
RyoheiHagimoto 0:0e0631af0305 48 #include <algorithm>
RyoheiHagimoto 0:0e0631af0305 49 #include "opencv2/core.hpp"
RyoheiHagimoto 0:0e0631af0305 50
RyoheiHagimoto 0:0e0631af0305 51 namespace cv
RyoheiHagimoto 0:0e0631af0305 52 {
RyoheiHagimoto 0:0e0631af0305 53 namespace videostab
RyoheiHagimoto 0:0e0631af0305 54 {
RyoheiHagimoto 0:0e0631af0305 55
RyoheiHagimoto 0:0e0631af0305 56 //! @addtogroup videostab_marching
RyoheiHagimoto 0:0e0631af0305 57 //! @{
RyoheiHagimoto 0:0e0631af0305 58
RyoheiHagimoto 0:0e0631af0305 59 /** @brief Describes the Fast Marching Method implementation.
RyoheiHagimoto 0:0e0631af0305 60
RyoheiHagimoto 0:0e0631af0305 61 See http://iwi.eldoc.ub.rug.nl/FILES/root/2004/JGraphToolsTelea/2004JGraphToolsTelea.pdf
RyoheiHagimoto 0:0e0631af0305 62 */
RyoheiHagimoto 0:0e0631af0305 63 class CV_EXPORTS FastMarchingMethod
RyoheiHagimoto 0:0e0631af0305 64 {
RyoheiHagimoto 0:0e0631af0305 65 public:
RyoheiHagimoto 0:0e0631af0305 66 FastMarchingMethod() : inf_(1e6f) {}
RyoheiHagimoto 0:0e0631af0305 67
RyoheiHagimoto 0:0e0631af0305 68 /** @brief Template method that runs the Fast Marching Method.
RyoheiHagimoto 0:0e0631af0305 69
RyoheiHagimoto 0:0e0631af0305 70 @param mask Image mask. 0 value indicates that the pixel value must be inpainted, 255 indicates
RyoheiHagimoto 0:0e0631af0305 71 that the pixel value is known, other values aren't acceptable.
RyoheiHagimoto 0:0e0631af0305 72 @param inpaint Inpainting functor that overloads void operator ()(int x, int y).
RyoheiHagimoto 0:0e0631af0305 73 @return Inpainting functor.
RyoheiHagimoto 0:0e0631af0305 74 */
RyoheiHagimoto 0:0e0631af0305 75 template <typename Inpaint>
RyoheiHagimoto 0:0e0631af0305 76 Inpaint run(const Mat &mask, Inpaint inpaint);
RyoheiHagimoto 0:0e0631af0305 77
RyoheiHagimoto 0:0e0631af0305 78 /**
RyoheiHagimoto 0:0e0631af0305 79 @return Distance map that's created during working of the method.
RyoheiHagimoto 0:0e0631af0305 80 */
RyoheiHagimoto 0:0e0631af0305 81 Mat distanceMap() const { return dist_; }
RyoheiHagimoto 0:0e0631af0305 82
RyoheiHagimoto 0:0e0631af0305 83 private:
RyoheiHagimoto 0:0e0631af0305 84 enum { INSIDE = 0, BAND = 1, KNOWN = 255 };
RyoheiHagimoto 0:0e0631af0305 85
RyoheiHagimoto 0:0e0631af0305 86 struct DXY
RyoheiHagimoto 0:0e0631af0305 87 {
RyoheiHagimoto 0:0e0631af0305 88 float dist;
RyoheiHagimoto 0:0e0631af0305 89 int x, y;
RyoheiHagimoto 0:0e0631af0305 90
RyoheiHagimoto 0:0e0631af0305 91 DXY() : dist(0), x(0), y(0) {}
RyoheiHagimoto 0:0e0631af0305 92 DXY(float _dist, int _x, int _y) : dist(_dist), x(_x), y(_y) {}
RyoheiHagimoto 0:0e0631af0305 93 bool operator <(const DXY &dxy) const { return dist < dxy.dist; }
RyoheiHagimoto 0:0e0631af0305 94 };
RyoheiHagimoto 0:0e0631af0305 95
RyoheiHagimoto 0:0e0631af0305 96 float solve(int x1, int y1, int x2, int y2) const;
RyoheiHagimoto 0:0e0631af0305 97 int& indexOf(const DXY &dxy) { return index_(dxy.y, dxy.x); }
RyoheiHagimoto 0:0e0631af0305 98
RyoheiHagimoto 0:0e0631af0305 99 void heapUp(int idx);
RyoheiHagimoto 0:0e0631af0305 100 void heapDown(int idx);
RyoheiHagimoto 0:0e0631af0305 101 void heapAdd(const DXY &dxy);
RyoheiHagimoto 0:0e0631af0305 102 void heapRemoveMin();
RyoheiHagimoto 0:0e0631af0305 103
RyoheiHagimoto 0:0e0631af0305 104 float inf_;
RyoheiHagimoto 0:0e0631af0305 105
RyoheiHagimoto 0:0e0631af0305 106 cv::Mat_<uchar> flag_; // flag map
RyoheiHagimoto 0:0e0631af0305 107 cv::Mat_<float> dist_; // distance map
RyoheiHagimoto 0:0e0631af0305 108
RyoheiHagimoto 0:0e0631af0305 109 cv::Mat_<int> index_; // index of point in the narrow band
RyoheiHagimoto 0:0e0631af0305 110 std::vector<DXY> narrowBand_; // narrow band heap
RyoheiHagimoto 0:0e0631af0305 111 int size_; // narrow band size
RyoheiHagimoto 0:0e0631af0305 112 };
RyoheiHagimoto 0:0e0631af0305 113
RyoheiHagimoto 0:0e0631af0305 114 //! @}
RyoheiHagimoto 0:0e0631af0305 115
RyoheiHagimoto 0:0e0631af0305 116 } // namespace videostab
RyoheiHagimoto 0:0e0631af0305 117 } // namespace cv
RyoheiHagimoto 0:0e0631af0305 118
RyoheiHagimoto 0:0e0631af0305 119 #include "fast_marching_inl.hpp"
RyoheiHagimoto 0:0e0631af0305 120
RyoheiHagimoto 0:0e0631af0305 121 #endif