Mercurial Hosting > traffic-intelligence
view c/InputVideoFileModule.cpp @ 473:178c69ba2970
bug corrected
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 12 Mar 2014 17:47:49 -0400 |
parents | 2be846d36dec |
children |
line wrap: on
line source
#include "InputVideoFileModule.h" InputVideoFileModule::InputVideoFileModule(const std::string& videoPath) : mInit(false) , mNumberOfFrame(0) { mInit = mVideoCapture.open(videoPath.c_str()); double frameCount; frameCount = mVideoCapture.get(CV_CAP_PROP_FRAME_COUNT); mSize = cv::Size(mVideoCapture.get(CV_CAP_PROP_FRAME_WIDTH), mVideoCapture.get(CV_CAP_PROP_FRAME_HEIGHT)); mNumberOfFrame = (unsigned int)frameCount; } InputVideoFileModule::~InputVideoFileModule(void) { } void InputVideoFileModule::setFrameNumber(const unsigned int& frameNumber) { mVideoCapture.set(CV_CAP_PROP_POS_FRAMES, frameNumber); } bool InputVideoFileModule::getNextFrame(cv::Mat& outputPicture) { bool success = false; if(mInit) { mVideoCapture >> outputPicture; success = !outputPicture.empty(); } return success; }