From 1d84f2820243564bb376f22c6a1ff8640027a0fa Mon Sep 17 00:00:00 2001
From: unknown <admin@.(none)>
Date: Tue, 17 Nov 2020 10:58:58 +0100
Subject: [PATCH] phase_unwrapping: add asserts, documentation for input types

---
 .../include/opencv2/phase_unwrapping/phase_unwrapping.hpp     | 4 ++--
 modules/phase_unwrapping/src/histogramphaseunwrapping.cpp     | 3 +++
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/modules/phase_unwrapping/include/opencv2/phase_unwrapping/phase_unwrapping.hpp b/modules/phase_unwrapping/include/opencv2/phase_unwrapping/phase_unwrapping.hpp
index 5b5cb5100..1091ff380 100644
--- a/modules/phase_unwrapping/include/opencv2/phase_unwrapping/phase_unwrapping.hpp
+++ b/modules/phase_unwrapping/include/opencv2/phase_unwrapping/phase_unwrapping.hpp
@@ -58,9 +58,9 @@ public:
     /**
      * @brief Unwraps a 2D phase map.
 
-     * @param wrappedPhaseMap The wrapped phase map that needs to be unwrapped.
+     * @param wrappedPhaseMap The wrapped phase map of type CV_32FC1 that needs to be unwrapped.
      * @param unwrappedPhaseMap The unwrapped phase map.
-     * @param shadowMask Optional parameter used when some pixels do not hold any phase information in the wrapped phase map.
+     * @param shadowMask Optional CV_8UC1 mask image used when some pixels do not hold any phase information in the wrapped phase map.
      */
     CV_WRAP
     virtual void unwrapPhaseMap( InputArray wrappedPhaseMap, OutputArray unwrappedPhaseMap,
diff --git a/modules/phase_unwrapping/src/histogramphaseunwrapping.cpp b/modules/phase_unwrapping/src/histogramphaseunwrapping.cpp
index 591deaf8d..c2b3030ac 100644
--- a/modules/phase_unwrapping/src/histogramphaseunwrapping.cpp
+++ b/modules/phase_unwrapping/src/histogramphaseunwrapping.cpp
@@ -402,6 +402,9 @@ void HistogramPhaseUnwrapping_Impl::unwrapPhaseMap( InputArray wrappedPhaseMap,
         temp.copyTo(mask);
     }
 
+    CV_CheckTypeEQ(wPhaseMap.type(), CV_32FC1, "");
+    CV_CheckTypeEQ(mask.type(), CV_8UC1, "");
+
     computePixelsReliability(wPhaseMap, mask);
     computeEdgesReliabilityAndCreateHistogram();