Merge pull request #16751 from alalek:core_coverity_issues

pull/16753/head^2
Alexander Alekhin 5 years ago
commit 6d113bd03f
  1. 1
      modules/core/src/minmax.cpp
  2. 90
      modules/core/src/norm.cpp

@ -1089,6 +1089,7 @@ bool ocl_minMaxIdx( InputArray _src, double* minVal, double* maxVal, int* minLoc
getMinMaxRes<double> getMinMaxRes<double>
}; };
CV_Assert(ddepth <= CV_64F);
getMinMaxResFunc func = functab[ddepth]; getMinMaxResFunc func = functab[ddepth];
int locTemp[2]; int locTemp[2];

@ -710,29 +710,27 @@ double cv::norm( InputArray _src, int normType, InputArray _mask )
result; result;
result.d = 0; result.d = 0;
NAryMatIterator it(arrays, ptrs); NAryMatIterator it(arrays, ptrs);
int j, total = (int)it.size, blockSize = total, intSumBlockSize = 0, count = 0; CV_CheckLT((size_t)it.size, (size_t)INT_MAX, "");
bool blockSum = (normType == NORM_L1 && depth <= CV_16S) ||
((normType == NORM_L2 || normType == NORM_L2SQR) && depth <= CV_8S);
int isum = 0;
int *ibuf = &result.i;
size_t esz = 0;
if( blockSum ) if ((normType == NORM_L1 && depth <= CV_16S) ||
((normType == NORM_L2 || normType == NORM_L2SQR) && depth <= CV_8S))
{ {
intSumBlockSize = (normType == NORM_L1 && depth <= CV_8S ? (1 << 23) : (1 << 15))/cn; // special case to handle "integer" overflow in accumulator
blockSize = std::min(blockSize, intSumBlockSize); const size_t esz = src.elemSize();
ibuf = &isum; const int total = (int)it.size;
esz = src.elemSize(); const int intSumBlockSize = (normType == NORM_L1 && depth <= CV_8S ? (1 << 23) : (1 << 15))/cn;
} const int blockSize = std::min(total, intSumBlockSize);
int isum = 0;
int count = 0;
for (size_t i = 0; i < it.nplanes; i++, ++it) for (size_t i = 0; i < it.nplanes; i++, ++it)
{ {
for( j = 0; j < total; j += blockSize ) for (int j = 0; j < total; j += blockSize)
{ {
int bsz = std::min(total - j, blockSize); int bsz = std::min(total - j, blockSize);
func( ptrs[0], ptrs[1], (uchar*)ibuf, bsz, cn ); func(ptrs[0], ptrs[1], (uchar*)&isum, bsz, cn);
count += bsz; count += bsz;
if( blockSum && (count + blockSize >= intSumBlockSize || (i+1 >= it.nplanes && j+bsz >= total)) ) if (count + blockSize >= intSumBlockSize || (i+1 >= it.nplanes && j+bsz >= total))
{ {
result.d += isum; result.d += isum;
isum = 0; isum = 0;
@ -743,18 +741,27 @@ double cv::norm( InputArray _src, int normType, InputArray _mask )
ptrs[1] += bsz; ptrs[1] += bsz;
} }
} }
}
else
{
// generic implementation
for (size_t i = 0; i < it.nplanes; i++, ++it)
{
func(ptrs[0], ptrs[1], (uchar*)&result, (int)it.size, cn);
}
}
if( normType == NORM_INF ) if( normType == NORM_INF )
{ {
if( depth == CV_64F ) if( depth == CV_64F )
; return result.d;
else if( depth == CV_32F ) else if( depth == CV_32F )
result.d = result.f; return result.f;
else else
result.d = result.i; return result.i;
} }
else if( normType == NORM_L2 ) else if( normType == NORM_L2 )
result.d = std::sqrt(result.d); return std::sqrt(result.d);
return result.d; return result.d;
} }
@ -1170,29 +1177,27 @@ double cv::norm( InputArray _src1, InputArray _src2, int normType, InputArray _m
result; result;
result.d = 0; result.d = 0;
NAryMatIterator it(arrays, ptrs); NAryMatIterator it(arrays, ptrs);
int j, total = (int)it.size, blockSize = total, intSumBlockSize = 0, count = 0; CV_CheckLT((size_t)it.size, (size_t)INT_MAX, "");
bool blockSum = (normType == NORM_L1 && depth <= CV_16S) ||
((normType == NORM_L2 || normType == NORM_L2SQR) && depth <= CV_8S);
unsigned isum = 0;
unsigned *ibuf = &result.u;
size_t esz = 0;
if( blockSum ) if ((normType == NORM_L1 && depth <= CV_16S) ||
((normType == NORM_L2 || normType == NORM_L2SQR) && depth <= CV_8S))
{ {
intSumBlockSize = normType == NORM_L1 && depth <= CV_8S ? (1 << 23) : (1 << 15); // special case to handle "integer" overflow in accumulator
blockSize = std::min(blockSize, intSumBlockSize); const size_t esz = src1.elemSize();
ibuf = &isum; const int total = (int)it.size;
esz = src1.elemSize(); const int intSumBlockSize = normType == NORM_L1 && depth <= CV_8S ? (1 << 23) : (1 << 15);
} const int blockSize = std::min(total, intSumBlockSize);
int isum = 0;
int count = 0;
for (size_t i = 0; i < it.nplanes; i++, ++it) for (size_t i = 0; i < it.nplanes; i++, ++it)
{ {
for( j = 0; j < total; j += blockSize ) for (int j = 0; j < total; j += blockSize)
{ {
int bsz = std::min(total - j, blockSize); int bsz = std::min(total - j, blockSize);
func( ptrs[0], ptrs[1], ptrs[2], (uchar*)ibuf, bsz, cn ); func(ptrs[0], ptrs[1], ptrs[2], (uchar*)&isum, bsz, cn);
count += bsz; count += bsz;
if( blockSum && (count + blockSize >= intSumBlockSize || (i+1 >= it.nplanes && j+bsz >= total)) ) if (count + blockSize >= intSumBlockSize || (i+1 >= it.nplanes && j+bsz >= total))
{ {
result.d += isum; result.d += isum;
isum = 0; isum = 0;
@ -1204,18 +1209,27 @@ double cv::norm( InputArray _src1, InputArray _src2, int normType, InputArray _m
ptrs[2] += bsz; ptrs[2] += bsz;
} }
} }
}
else
{
// generic implementation
for (size_t i = 0; i < it.nplanes; i++, ++it)
{
func(ptrs[0], ptrs[1], ptrs[2], (uchar*)&result, (int)it.size, cn);
}
}
if( normType == NORM_INF ) if( normType == NORM_INF )
{ {
if( depth == CV_64F ) if( depth == CV_64F )
; return result.d;
else if( depth == CV_32F ) else if( depth == CV_32F )
result.d = result.f; return result.f;
else else
result.d = result.u; return result.u;
} }
else if( normType == NORM_L2 ) else if( normType == NORM_L2 )
result.d = std::sqrt(result.d); return std::sqrt(result.d);
return result.d; return result.d;
} }

Loading…
Cancel
Save