pull/13383/head
Vadim Pisarevsky
14 years ago
parent
72541721a1
commit
1badec0b2d
15 changed files with 323 additions and 287 deletions
@ -1,219 +0,0 @@ |
|||||||
#ifndef _OPENCV_API_EXTRA_HPP_ |
|
||||||
#define _OPENCV_API_EXTRA_HPP_ |
|
||||||
|
|
||||||
#include "opencv2/core/core.hpp" |
|
||||||
#include "opencv2/imgproc/imgproc.hpp" |
|
||||||
#include "opencv2/imgproc/imgproc_c.h" |
|
||||||
#include "opencv2/calib3d/calib3d.hpp" |
|
||||||
|
|
||||||
namespace cv |
|
||||||
{ |
|
||||||
|
|
||||||
template<typename _Tp> |
|
||||||
static inline void mv2vv(const vector<Mat>& src, vector<vector<_Tp> >& dst) |
|
||||||
{ |
|
||||||
size_t i, n = src.size(); |
|
||||||
dst.resize(src.size()); |
|
||||||
for( i = 0; i < n; i++ ) |
|
||||||
src[i].copyTo(dst[i]); |
|
||||||
} |
|
||||||
|
|
||||||
///////////////////////////// core /////////////////////////////
|
|
||||||
|
|
||||||
CV_WRAP_AS(getTickCount) static inline double getTickCount_() |
|
||||||
{ |
|
||||||
return (double)getTickCount(); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP_AS(getCPUTickCount) static inline double getCPUTickCount_() |
|
||||||
{ |
|
||||||
return (double)getCPUTickCount(); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP void randShuffle(const Mat& src, CV_OUT Mat& dst, double iterFactor=1.) |
|
||||||
{ |
|
||||||
src.copyTo(dst); |
|
||||||
randShuffle(dst, iterFactor, 0); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void SVDecomp(const Mat& src, CV_OUT Mat& w, CV_OUT Mat& u, CV_OUT Mat& vt, int flags=0 ) |
|
||||||
{ |
|
||||||
SVD::compute(src, w, u, vt, flags); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void SVBackSubst( const Mat& w, const Mat& u, const Mat& vt, |
|
||||||
const Mat& rhs, CV_OUT Mat& dst ) |
|
||||||
{ |
|
||||||
SVD::backSubst(w, u, vt, rhs, dst); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void mixChannels(const vector<Mat>& src, vector<Mat>& dst, |
|
||||||
const vector<int>& fromTo) |
|
||||||
{ |
|
||||||
if(fromTo.empty()) |
|
||||||
return; |
|
||||||
CV_Assert(fromTo.size()%2 == 0); |
|
||||||
mixChannels(&src[0], (int)src.size(), &dst[0], (int)dst.size(), &fromTo[0], (int)(fromTo.size()/2)); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline bool eigen(const Mat& src, bool computeEigenvectors, |
|
||||||
CV_OUT Mat& eigenvalues, CV_OUT Mat& eigenvectors, |
|
||||||
int lowindex=-1, int highindex=-1) |
|
||||||
{ |
|
||||||
return computeEigenvectors ? eigen(src, eigenvalues, eigenvectors, lowindex, highindex) : |
|
||||||
eigen(src, eigenvalues, lowindex, highindex); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void fillConvexPoly(Mat& img, const Mat& points, |
|
||||||
const Scalar& color, int lineType=8, |
|
||||||
int shift=0) |
|
||||||
{ |
|
||||||
CV_Assert(points.checkVector(2, CV_32S) >= 0); |
|
||||||
fillConvexPoly(img, (const Point*)points.data, points.rows*points.cols*points.channels()/2, color, lineType, shift); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void fillPoly(Mat& img, const vector<Mat>& pts, |
|
||||||
const Scalar& color, int lineType=8, int shift=0, |
|
||||||
Point offset=Point() ) |
|
||||||
{ |
|
||||||
if( pts.empty() ) |
|
||||||
return; |
|
||||||
AutoBuffer<Point*> _ptsptr(pts.size()); |
|
||||||
AutoBuffer<int> _npts(pts.size()); |
|
||||||
Point** ptsptr = _ptsptr; |
|
||||||
int* npts = _npts; |
|
||||||
|
|
||||||
for( size_t i = 0; i < pts.size(); i++ ) |
|
||||||
{ |
|
||||||
const Mat& p = pts[i]; |
|
||||||
CV_Assert(p.checkVector(2, CV_32S) >= 0); |
|
||||||
ptsptr[i] = (Point*)p.data; |
|
||||||
npts[i] = p.rows*p.cols*p.channels()/2; |
|
||||||
} |
|
||||||
fillPoly(img, (const Point**)ptsptr, npts, (int)pts.size(), color, lineType, shift, offset); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void polylines(Mat& img, const vector<Mat>& pts, |
|
||||||
bool isClosed, const Scalar& color, |
|
||||||
int thickness=1, int lineType=8, int shift=0 ) |
|
||||||
{ |
|
||||||
if( pts.empty() ) |
|
||||||
return; |
|
||||||
AutoBuffer<Point*> _ptsptr(pts.size()); |
|
||||||
AutoBuffer<int> _npts(pts.size()); |
|
||||||
Point** ptsptr = _ptsptr; |
|
||||||
int* npts = _npts; |
|
||||||
|
|
||||||
for( size_t i = 0; i < pts.size(); i++ ) |
|
||||||
{ |
|
||||||
const Mat& p = pts[i]; |
|
||||||
CV_Assert(p.checkVector(2, CV_32S) >= 0); |
|
||||||
ptsptr[i] = (Point*)p.data; |
|
||||||
npts[i] = p.rows*p.cols*p.channels()/2; |
|
||||||
} |
|
||||||
polylines(img, (const Point**)ptsptr, npts, (int)pts.size(), isClosed, color, thickness, lineType, shift); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void PCACompute(const Mat& data, CV_OUT Mat& mean, |
|
||||||
CV_OUT Mat& eigenvectors, int maxComponents=0) |
|
||||||
{ |
|
||||||
PCA pca; |
|
||||||
pca.mean = mean; |
|
||||||
pca.eigenvectors = eigenvectors; |
|
||||||
pca(data, Mat(), 0, maxComponents); |
|
||||||
pca.mean.copyTo(mean); |
|
||||||
pca.eigenvectors.copyTo(eigenvectors); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void PCAProject(const Mat& data, const Mat& mean, |
|
||||||
const Mat& eigenvectors, CV_OUT Mat& result) |
|
||||||
{ |
|
||||||
PCA pca; |
|
||||||
pca.mean = mean; |
|
||||||
pca.eigenvectors = eigenvectors; |
|
||||||
pca.project(data, result); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void PCABackProject(const Mat& data, const Mat& mean, |
|
||||||
const Mat& eigenvectors, CV_OUT Mat& result) |
|
||||||
{ |
|
||||||
PCA pca; |
|
||||||
pca.mean = mean; |
|
||||||
pca.eigenvectors = eigenvectors; |
|
||||||
pca.backProject(data, result); |
|
||||||
}
|
|
||||||
|
|
||||||
/////////////////////////// imgproc /////////////////////////////////
|
|
||||||
|
|
||||||
CV_WRAP static inline void HuMoments(const Moments& m, CV_OUT vector<double>& hu) |
|
||||||
{ |
|
||||||
hu.resize(7); |
|
||||||
HuMoments(m, &hu[0]); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline Mat getPerspectiveTransform(const vector<Point2f>& src, const vector<Point2f>& dst) |
|
||||||
{ |
|
||||||
CV_Assert(src.size() == 4 && dst.size() == 4); |
|
||||||
return getPerspectiveTransform(&src[0], &dst[0]); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline Mat getAffineTransform(const vector<Point2f>& src, const vector<Point2f>& dst) |
|
||||||
{ |
|
||||||
CV_Assert(src.size() == 3 && dst.size() == 3); |
|
||||||
return getAffineTransform(&src[0], &dst[0]); |
|
||||||
} |
|
||||||
|
|
||||||
CV_WRAP static inline void calcHist( const vector<Mat>& images, const vector<int>& channels, |
|
||||||
const Mat& mask, CV_OUT Mat& hist, |
|
||||||
const vector<int>& histSize, |
|
||||||
const vector<float>& ranges, |
|
||||||
bool accumulate=false) |
|
||||||
{ |
|
||||||
int i, dims = (int)histSize.size(), rsz = (int)ranges.size(), csz = (int)channels.size(); |
|
||||||
CV_Assert(images.size() > 0 && dims > 0); |
|
||||||
CV_Assert(rsz == dims*2 || (rsz == 0 && images[0].depth() == CV_8U)); |
|
||||||
CV_Assert(csz == 0 || csz == dims); |
|
||||||
float* _ranges[CV_MAX_DIM]; |
|
||||||
if( rsz > 0 ) |
|
||||||
{ |
|
||||||
for( i = 0; i < rsz/2; i++ ) |
|
||||||
_ranges[i] = (float*)&ranges[i*2]; |
|
||||||
} |
|
||||||
calcHist(&images[0], (int)images.size(), csz ? &channels[0] : 0, |
|
||||||
mask, hist, dims, &histSize[0], rsz ? (const float**)_ranges : 0, |
|
||||||
true, accumulate); |
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
CV_WRAP void calcBackProject( const vector<Mat>& images, const vector<int>& channels, |
|
||||||
const Mat& hist, CV_OUT Mat& dst, |
|
||||||
const vector<float>& ranges, |
|
||||||
double scale=1 ) |
|
||||||
{ |
|
||||||
int i, dims = hist.dims, rsz = (int)ranges.size(), csz = (int)channels.size(); |
|
||||||
CV_Assert(images.size() > 0); |
|
||||||
CV_Assert(rsz == dims*2 || (rsz == 0 && images[0].depth() == CV_8U)); |
|
||||||
CV_Assert(csz == 0 || csz == dims); |
|
||||||
float* _ranges[CV_MAX_DIM]; |
|
||||||
if( rsz > 0 ) |
|
||||||
{ |
|
||||||
for( i = 0; i < rsz/2; i++ ) |
|
||||||
_ranges[i] = (float*)&ranges[i*2]; |
|
||||||
} |
|
||||||
calcBackProject(&images[0], (int)images.size(), csz ? &channels[0] : 0, |
|
||||||
hist, dst, rsz ? (const float**)_ranges : 0, scale, true); |
|
||||||
} |
|
||||||
|
|
||||||
|
|
||||||
/////////////////////////////// calib3d ///////////////////////////////////////////
|
|
||||||
|
|
||||||
//! finds circles' grid pattern of the specified size in the image
|
|
||||||
CV_WRAP static inline void findCirclesGridDefault( InputArray image, Size patternSize, |
|
||||||
OutputArray centers, int flags=CALIB_CB_SYMMETRIC_GRID ) |
|
||||||
{ |
|
||||||
findCirclesGrid(image, patternSize, centers, flags); |
|
||||||
} |
|
||||||
|
|
||||||
} |
|
||||||
|
|
||||||
#endif |
|
||||||
Loading…
Reference in new issue