Merge pull request #366 from sirotenko/master

Added getFrameCount method to VideoCapture

Author: peterbraden

src/VideoCaptureWrap.cc

@@ -33,6 +33,7 @@ void VideoCaptureWrap::Init(Local<Object> target) {
   Nan::SetPrototypeMethod(ctor, "setHeight", SetHeight);
   Nan::SetPrototypeMethod(ctor, "setPosition", SetPosition);
   Nan::SetPrototypeMethod(ctor, "getFrameAt", GetFrameAt);
+  Nan::SetPrototypeMethod(ctor, "getFrameCount", GetFrameCount);
   Nan::SetPrototypeMethod(ctor, "close", Close);
   Nan::SetPrototypeMethod(ctor, "ReadSync", ReadSync);
   Nan::SetPrototypeMethod(ctor, "grab", Grab);
@@ -94,6 +95,15 @@ NAN_METHOD(VideoCaptureWrap::SetWidth) {
   return;
 }
 
+NAN_METHOD(VideoCaptureWrap::GetFrameCount) {
+  Nan::HandleScope scope;
+  VideoCaptureWrap *v = Nan::ObjectWrap::Unwrap<VideoCaptureWrap>(info.This());
+
+  int cnt = int(v->cap.get(CV_CAP_PROP_FRAME_COUNT));
+
+  info.GetReturnValue().Set(Nan::New<Number>(cnt));
+}
+
 NAN_METHOD(VideoCaptureWrap::SetHeight) {
   Nan::HandleScope scope;
   VideoCaptureWrap *v = Nan::ObjectWrap::Unwrap<VideoCaptureWrap>(info.This());

src/VideoCaptureWrap.h

@@ -23,6 +23,7 @@ class VideoCaptureWrap: public Nan::ObjectWrap {
 
   // to set frame position
   static NAN_METHOD(SetPosition);
+  static NAN_METHOD(GetFrameCount);
 
   static NAN_METHOD(GetFrameAt);