relaxrate new contour synthesis
added a rficv folder where we can change ofxcv as needed
This commit is contained in:
dviid
Jamie Allen
@@ -0,0 +1,211 @@
|
||||
|
||||
#include "rfiCvContourFinder.h"
|
||||
|
||||
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
static bool sort_carea_compare( const CvSeq* a, const CvSeq* b) {
|
||||
// use opencv to calc size, then sort based on size
|
||||
float areaa = fabs(cvContourArea(a, CV_WHOLE_SEQ));
|
||||
float areab = fabs(cvContourArea(b, CV_WHOLE_SEQ));
|
||||
|
||||
//return 0;
|
||||
return (areaa > areab);
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
rfiCvContourFinder::rfiCvContourFinder() {
|
||||
_width = 0;
|
||||
_height = 0;
|
||||
myMoments = (CvMoments*)malloc( sizeof(CvMoments) );
|
||||
reset();
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
rfiCvContourFinder::~rfiCvContourFinder() {
|
||||
free( myMoments );
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
void rfiCvContourFinder::reset() {
|
||||
cvSeqBlobs.clear();
|
||||
blobs.clear();
|
||||
nBlobs = 0;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
int rfiCvContourFinder::findContours( ofxCvGrayscaleImage& input,
|
||||
int minArea,
|
||||
int maxArea,
|
||||
int nConsidered,
|
||||
bool bFindHoles,
|
||||
bool bUseApproximation) {
|
||||
|
||||
// get width/height disregarding ROI
|
||||
IplImage* ipltemp = input.getCvImage();
|
||||
_width = ipltemp->width;
|
||||
_height = ipltemp->height;
|
||||
|
||||
reset();
|
||||
|
||||
// opencv will clober the image it detects contours on, so we want to
|
||||
// copy it into a copy before we detect contours. That copy is allocated
|
||||
// if necessary (necessary = (a) not allocated or (b) wrong size)
|
||||
// so be careful if you pass in different sized images to "findContours"
|
||||
// there is a performance penalty, but we think there is not a memory leak
|
||||
// to worry about better to create mutiple contour finders for different
|
||||
// sizes, ie, if you are finding contours in a 640x480 image but also a
|
||||
// 320x240 image better to make two rfiCvContourFinder objects then to use
|
||||
// one, because you will get penalized less.
|
||||
|
||||
if( inputCopy.getWidth() == 0 ) {
|
||||
inputCopy.setUseTexture(false);
|
||||
inputCopy.allocate( _width, _height );
|
||||
} else if( inputCopy.getWidth() != _width || inputCopy.getHeight() != _height ) {
|
||||
// reallocate to new size
|
||||
inputCopy.clear();
|
||||
inputCopy.setUseTexture(false);
|
||||
inputCopy.allocate( _width, _height );
|
||||
}
|
||||
|
||||
inputCopy.setROI( input.getROI() );
|
||||
inputCopy = input;
|
||||
|
||||
CvSeq* contour_list = NULL;
|
||||
contour_storage = cvCreateMemStorage( 1000 );
|
||||
storage = cvCreateMemStorage( 1000 );
|
||||
|
||||
CvContourRetrievalMode retrieve_mode
|
||||
= (bFindHoles) ? CV_RETR_LIST : CV_RETR_EXTERNAL;
|
||||
cvFindContours( inputCopy.getCvImage(), contour_storage, &contour_list,
|
||||
sizeof(CvContour), retrieve_mode, bUseApproximation ? CV_CHAIN_APPROX_SIMPLE : CV_CHAIN_APPROX_NONE );
|
||||
CvSeq* contour_ptr = contour_list;
|
||||
|
||||
// put the contours from the linked list, into an array for sorting
|
||||
while( (contour_ptr != NULL) ) {
|
||||
float area = fabs( cvContourArea(contour_ptr, CV_WHOLE_SEQ) );
|
||||
if( (area > minArea) && (area < maxArea) ) {
|
||||
cvSeqBlobs.push_back(contour_ptr);
|
||||
}
|
||||
contour_ptr = contour_ptr->h_next;
|
||||
}
|
||||
|
||||
|
||||
// sort the pointers based on size
|
||||
if( cvSeqBlobs.size() > 1 ) {
|
||||
sort( cvSeqBlobs.begin(), cvSeqBlobs.end(), sort_carea_compare );
|
||||
}
|
||||
|
||||
|
||||
// now, we have cvSeqBlobs.size() contours, sorted by size in the array
|
||||
// cvSeqBlobs let's get the data out and into our structures that we like
|
||||
for( int i = 0; i < MIN(nConsidered, (int)cvSeqBlobs.size()); i++ ) {
|
||||
blobs.push_back( ofxCvBlob() );
|
||||
float area = cvContourArea( cvSeqBlobs[i], CV_WHOLE_SEQ );
|
||||
CvRect rect = cvBoundingRect( cvSeqBlobs[i], 0 );
|
||||
cvMoments( cvSeqBlobs[i], myMoments );
|
||||
|
||||
blobs[i].area = fabs(area);
|
||||
blobs[i].hole = area < 0 ? true : false;
|
||||
blobs[i].length = cvArcLength(cvSeqBlobs[i]);
|
||||
blobs[i].boundingRect.x = rect.x;
|
||||
blobs[i].boundingRect.y = rect.y;
|
||||
blobs[i].boundingRect.width = rect.width;
|
||||
blobs[i].boundingRect.height = rect.height;
|
||||
blobs[i].centroid.x = (myMoments->m10 / myMoments->m00);
|
||||
blobs[i].centroid.y = (myMoments->m01 / myMoments->m00);
|
||||
|
||||
// get the points for the blob:
|
||||
CvPoint pt;
|
||||
CvSeqReader reader;
|
||||
cvStartReadSeq( cvSeqBlobs[i], &reader, 0 );
|
||||
|
||||
for( int j=0; j < cvSeqBlobs[i]->total; j++ ) {
|
||||
CV_READ_SEQ_ELEM( pt, reader );
|
||||
blobs[i].pts.push_back( ofPoint((float)pt.x, (float)pt.y) );
|
||||
}
|
||||
blobs[i].nPts = blobs[i].pts.size();
|
||||
|
||||
}
|
||||
|
||||
nBlobs = blobs.size();
|
||||
|
||||
// Free the storage memory.
|
||||
// Warning: do this inside this function otherwise a strange memory leak
|
||||
if( contour_storage != NULL ) { cvReleaseMemStorage(&contour_storage); }
|
||||
if( storage != NULL ) { cvReleaseMemStorage(&storage); }
|
||||
|
||||
return nBlobs;
|
||||
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
void rfiCvContourFinder::draw( float x, float y, float w, float h ) {
|
||||
|
||||
float scalex = 0.0f;
|
||||
float scaley = 0.0f;
|
||||
if( _width != 0 ) { scalex = w/_width; } else { scalex = 1.0f; }
|
||||
if( _height != 0 ) { scaley = h/_height; } else { scaley = 1.0f; }
|
||||
|
||||
if(bAnchorIsPct){
|
||||
x -= anchor.x * w;
|
||||
y -= anchor.y * h;
|
||||
}else{
|
||||
x -= anchor.x;
|
||||
y -= anchor.y;
|
||||
}
|
||||
|
||||
ofPushStyle();
|
||||
glPushMatrix();
|
||||
glTranslatef( x, y, 0.0 );
|
||||
glScalef( scalex, scaley, 0.0 );
|
||||
ofSetHexColor(0xFFFFFF);
|
||||
|
||||
for( int i=0; i<(int)blobs.size(); i++ ) {
|
||||
ofNoFill();
|
||||
ofBeginShape();
|
||||
for( int j=0; j<blobs[i].nPts; j++ ) {
|
||||
ofVertex( blobs[i].pts[j].x, blobs[i].pts[j].y );
|
||||
}
|
||||
ofEndShape();
|
||||
|
||||
}
|
||||
glPopMatrix();
|
||||
ofPopStyle();
|
||||
}
|
||||
|
||||
|
||||
//----------------------------------------------------------
|
||||
void rfiCvContourFinder::draw(const ofPoint & point){
|
||||
draw(point.x, point.y);
|
||||
}
|
||||
|
||||
//----------------------------------------------------------
|
||||
void rfiCvContourFinder::draw(const ofRectangle & rect){
|
||||
draw(rect.x, rect.y, rect.width, rect.height);
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
void rfiCvContourFinder::setAnchorPercent( float xPct, float yPct ){
|
||||
anchor.x = xPct;
|
||||
anchor.y = yPct;
|
||||
bAnchorIsPct = true;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
void rfiCvContourFinder::setAnchorPoint( int x, int y ){
|
||||
anchor.x = x;
|
||||
anchor.y = y;
|
||||
bAnchorIsPct = false;
|
||||
}
|
||||
|
||||
//--------------------------------------------------------------------------------
|
||||
void rfiCvContourFinder::resetAnchor(){
|
||||
anchor.set(0,0);
|
||||
bAnchorIsPct = false;
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,73 @@
|
||||
/*
|
||||
* rfiCvContourFinder.h
|
||||
*
|
||||
* Finds white blobs in binary images and identifies
|
||||
* centroid, bounding box, area, length and polygonal contour
|
||||
* The result is placed in a vector of ofxCvBlob objects.
|
||||
*
|
||||
*/
|
||||
|
||||
#ifndef RFI_CV_CONTOUR_FINDER
|
||||
#define RFI_CV_CONTOUR_FINDER
|
||||
|
||||
|
||||
|
||||
#include "ofxCvConstants.h"
|
||||
#include "ofxCvBlob.h"
|
||||
#include "ofxCvGrayscaleImage.h"
|
||||
#include <algorithm>
|
||||
|
||||
class rfiCvContourFinder : public ofBaseDraws {
|
||||
|
||||
public:
|
||||
|
||||
vector<ofxCvBlob> blobs;
|
||||
int nBlobs; // DEPRECATED: use blobs.size() instead
|
||||
|
||||
|
||||
rfiCvContourFinder();
|
||||
virtual ~rfiCvContourFinder();
|
||||
|
||||
virtual float getWidth() { return _width; }; //set after first findContours call
|
||||
virtual float getHeight() { return _height; }; //set after first findContours call
|
||||
|
||||
virtual int findContours( ofxCvGrayscaleImage& input,
|
||||
int minArea, int maxArea,
|
||||
int nConsidered, bool bFindHoles,
|
||||
bool bUseApproximation = true);
|
||||
// approximation = don't do points for all points
|
||||
// of the contour, if the contour runs
|
||||
// along a straight line, for example...
|
||||
|
||||
virtual void draw() { draw(0,0, _width, _height); };
|
||||
virtual void draw( float x, float y ) { draw(x,y, _width, _height); };
|
||||
virtual void draw( float x, float y, float w, float h );
|
||||
virtual void draw(const ofPoint & point);
|
||||
virtual void draw(const ofRectangle & rect);
|
||||
virtual void setAnchorPercent(float xPct, float yPct);
|
||||
virtual void setAnchorPoint(int x, int y);
|
||||
virtual void resetAnchor();
|
||||
//virtual ofxCvBlob getBlob(int num);
|
||||
|
||||
|
||||
|
||||
protected:
|
||||
|
||||
int _width;
|
||||
int _height;
|
||||
ofxCvGrayscaleImage inputCopy;
|
||||
CvMemStorage* contour_storage;
|
||||
CvMemStorage* storage;
|
||||
CvMoments* myMoments;
|
||||
vector<CvSeq*> cvSeqBlobs; //these will become blobs
|
||||
|
||||
ofPoint anchor;
|
||||
bool bAnchorIsPct;
|
||||
|
||||
virtual void reset();
|
||||
|
||||
};
|
||||
|
||||
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user