core: fix inplace support in mulSpectrums

21f3531b · Alexander Alekhin · 7be4a0ee · 21f3531b
Commit 21f3531b authored Dec 07, 2016 by Alexander Alekhin
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

dxt.cpp modules/core/src/dxt.cpp +6 -0

No files found.
--- a/modules/core/src/dxt.cpp
+++ b/modules/core/src/dxt.cpp
@@ -1905,6 +1905,12 @@ void cv::mulSpectrums( InputArray _srcA, InputArray _srcB,
    _dst.create( srcA.rows, srcA.cols, type );
    Mat dst = _dst.getMat();

+    // correct inplace support
+    if (dst.data == srcA.data)
+        srcA = srcA.clone();
+    if (dst.data == srcB.data)
+        srcB = srcB.clone();
+
    bool is_1d = (flags & DFT_ROWS) || (rows == 1 || (cols == 1 &&
             srcA.isContinuous() && srcB.isContinuous() && dst.isContinuous()));