Mercurial Hosting > traffic-intelligence
view c/InputVideoFileModule.cpp @ 400:7ef1071e3cc3
clean up of input classes for list of images and video files
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Mon, 29 Jul 2013 18:06:55 -0400 |
parents | c389fae9689a |
children | b829ebdc18e6 |
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 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; }