fixed random failures in Core_DFT.complex_output2 test (the case of input_mat.cols == 1)

This commit is contained in:
Rostislav Vasilikhin 2016-09-05 22:33:31 +03:00
parent 27dd27400a
commit b8b7f155a5

View File

@ -1070,11 +1070,12 @@ RealDFT( const T* src, T* dst, int n, int nf, int* factors, const int* itab,
}
}
if( complex_output && (n & 1) == 0 )
if (complex_output && ((n & 1) == 0 || n == 1))
{
dst[-1] = dst[0];
dst[0] = 0;
dst[n] = 0;
if (n > 1)
dst[n] = 0;
}
}