make Mat::at<>(i) work with 2d matrices (to retrieve all elements in row-major order) (reported by Kurt) and fixed bug #1804
This commit is contained in:
@@ -561,18 +561,28 @@ template<typename _Tp> inline const _Tp& Mat::at(Point pt) const
|
||||
|
||||
template<typename _Tp> inline _Tp& Mat::at(int i0)
|
||||
{
|
||||
CV_DbgAssert( dims <= 2 && data && (size.p[0] == 1 || size.p[1] == 1) &&
|
||||
(unsigned)i0 < (unsigned)(size.p[0] + size.p[1] - 1) &&
|
||||
CV_DbgAssert( dims <= 2 && data &&
|
||||
(unsigned)i0 < (unsigned)(size.p[0]*size.p[1]) &&
|
||||
elemSize() == CV_ELEM_SIZE(DataType<_Tp>::type) );
|
||||
return *(_Tp*)(data + step.p[size.p[0]==1]*i0);
|
||||
if( isContinuous() || size.p[0] == 1 )
|
||||
return ((_Tp*)data)[i0];
|
||||
if( size.p[1] == 1 )
|
||||
return *(_Tp*)(data + step.p[0]*i0);
|
||||
int i = i0/cols, j = i0 - i*cols;
|
||||
return ((_Tp*)(data + step.p[0]*i))[j];
|
||||
}
|
||||
|
||||
template<typename _Tp> inline const _Tp& Mat::at(int i0) const
|
||||
{
|
||||
CV_DbgAssert( dims <= 2 && data && (size.p[0] == 1 || size.p[1] == 1) &&
|
||||
(unsigned)i0 < (unsigned)(size.p[0] + size.p[1] - 1) &&
|
||||
CV_DbgAssert( dims <= 2 && data &&
|
||||
(unsigned)i0 < (unsigned)(size.p[0]*size.p[1]) &&
|
||||
elemSize() == CV_ELEM_SIZE(DataType<_Tp>::type) );
|
||||
return *(_Tp*)(data + step.p[size.p[0]==1]*i0);
|
||||
if( isContinuous() || size.p[0] == 1 )
|
||||
return ((const _Tp*)data)[i0];
|
||||
if( size.p[1] == 1 )
|
||||
return *(const _Tp*)(data + step.p[0]*i0);
|
||||
int i = i0/cols, j = i0 - i*cols;
|
||||
return ((const _Tp*)(data + step.p[0]*i))[j];
|
||||
}
|
||||
|
||||
template<typename _Tp> inline _Tp& Mat::at(int i0, int i1, int i2)
|
||||
|
Reference in New Issue
Block a user