#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;
}