Mercurial Hosting > traffic-intelligence
view c/InputVideoFileModule.cpp @ 590:0fa73cbe9fdb
annotation centroid are computed explicitly and projected at that step if required
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 05 Dec 2014 16:41:11 -0500 |
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; }