added size check to static Mat::diag()

13 years ago · 645bb0bfbc
parent 25261e6ebc
commit 645bb0bfbc
1 changed files with 1 additions and 0 deletions
--- a/modules/core/include/opencv2/core/mat.hpp
+++ b/modules/core/include/opencv2/core/mat.hpp
@ -338,6 +338,7 @@ inline Mat Mat::colRange(const Range& r) const

 inline Mat Mat::diag(const Mat& d)
 {
+    CV_Assert( d.cols == 1 );
    Mat m(d.rows, d.rows, d.type(), Scalar(0)), md = m.diag();
    d.copyTo(md);
    return m;