archive-gh-me/node-opencv
1
0
Fork 0
mirror of https://github.com/peterbraden/node-opencv.git synced 2025-12-08 19:45:55 +00:00
Code Issues Projects Releases Wiki Activity

hough lines

Browse Source
This commit is contained in:
Peter Braden 2012-09-19 16:35:04 -07:00
parent 48ab05f624
commit e8cc061187
3 changed files with 75 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
smoketest.js
View File

@ -18,11 +18,16 @@ new cv.VideoCapture(0).read(function(mat){
*/
cv.readImage("./examples/mona.png", function(err, im){
var features = im.goodFeaturesToTrack();
cv.readImage("./examples/stuff.png", function(err, im){
var i2 = im.copy()
i2.convertGrayscale()
i2.canny(5, 300)
var features = i2.houghLinesP();
for (var i=0;i<features.length; i++){
var x = features[i]
im.ellipse(x[0] - 5, x[1] -5, 10, 10);
im.line([x[0], x[1]], [x[2], x[3]]);
im.ellipse(x[0], x[1]);
im.ellipse(x[2], x[3]);
}
console.log(features)
im.save('./out.jpg');

64
src/Matrix.cc
View File

@ -40,6 +40,7 @@ Matrix::Init(Handle<Object> target) {
NODE_SET_PROTOTYPE_METHOD(constructor, "toBuffer", ToBuffer);
NODE_SET_PROTOTYPE_METHOD(constructor, "ellipse", Ellipse);
NODE_SET_PROTOTYPE_METHOD(constructor, "rectangle", Rectangle);
NODE_SET_PROTOTYPE_METHOD(constructor, "line", Line);
NODE_SET_PROTOTYPE_METHOD(constructor, "save", Save);
NODE_SET_PROTOTYPE_METHOD(constructor, "resize", Resize);
NODE_SET_PROTOTYPE_METHOD(constructor, "channels", Channels);
@ -58,6 +59,7 @@ Matrix::Init(Handle<Object> target) {
NODE_SET_PROTOTYPE_METHOD(constructor, "drawAllContours", DrawAllContours);
NODE_SET_PROTOTYPE_METHOD(constructor, "goodFeaturesToTrack", GoodFeaturesToTrack);
NODE_SET_PROTOTYPE_METHOD(constructor, "houghLinesP", HoughLinesP);
NODE_SET_METHOD(constructor, "Eye", Eye);
@ -346,6 +348,39 @@ Matrix::Rectangle(const Arguments& args) {
return scope.Close(v8::Null());
}
Handle<Value>
Matrix::Line(const Arguments& args) {
SETUP_FUNCTION(Matrix)
if(args[0]->IsArray() && args[1]->IsArray()) {
Local<Object> xy1 = args[0]->ToObject();
Local<Object> xy2 = args[1]->ToObject();
cv::Scalar color(0, 0, 255);
if(args[2]->IsArray()) {
Local<Object> objColor = args[2]->ToObject();
color = setColor(objColor);
}
int x1 = xy1->Get(0)->IntegerValue();
int y1 = xy1->Get(1)->IntegerValue();
int x2 = xy2->Get(0)->IntegerValue();
int y2 = xy2->Get(1)->IntegerValue();
int thickness = 1;
if(args[3]->IntegerValue())
thickness = args[3]->IntegerValue();
cv::line(self->mat, cv::Point(x1, y1), cv::Point(x2, y2), color, thickness);
}
return scope.Close(v8::Null());
}
Handle<Value>
Matrix::Save(const v8::Arguments& args){
HandleScope scope;
@ -585,6 +620,35 @@ Matrix::GoodFeaturesToTrack(const v8::Arguments& args) {
}
Handle<Value>
Matrix::HoughLinesP(const v8::Arguments& args) {
HandleScope scope;
Matrix *self = ObjectWrap::Unwrap<Matrix>(args.This());
std::vector<cv::Vec4i> lines;
cv::Mat gray;
equalizeHist(self->mat, gray);
// cv::Canny(gray, gray, 50, 200, 3);
cv::HoughLinesP(gray, lines, 1, CV_PI/180, 80, 30, 10);
v8::Local<v8::Array> arr = v8::Array::New(lines.size());
for (unsigned int i=0; i<lines.size(); i++){
v8::Local<v8::Array> pt = v8::Array::New(4);
pt->Set(0, Number::New((double) lines[i][0]));
pt->Set(1, Number::New((double) lines[i][1]));
pt->Set(2, Number::New((double) lines[i][2]));
pt->Set(3, Number::New((double) lines[i][3]));
arr->Set(i, pt);
}
return scope.Close(arr);
}
cv::Scalar setColor(Local<Object> objColor) {

3
src/Matrix.h
View File

@ -66,6 +66,7 @@ class Matrix: public node::ObjectWrap {
JSFUNC(ToBuffer)
JSFUNC(Ellipse)
JSFUNC(Rectangle)
JSFUNC(Line)
JSFUNC(Empty)
JSFUNC(Save)
@ -84,7 +85,9 @@ class Matrix: public node::ObjectWrap {
JSFUNC(DrawContour)
JSFUNC(DrawAllContours)
// Feature Detection
JSFUNC(GoodFeaturesToTrack)
JSFUNC(HoughLinesP)
};