OpenCVFrameSource.cpp

Go to the documentation of this file.

// -*- mode: c++; indent-tabs-mode: nil; c-basic-offset: 4; coding: iso-8859-1; -*-

/*
ACT-CV - Machine Vision for Cognitive Modeling
Copyright (c) 2008 Marc Halbruegge

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/


#include <iostream>
#if HAVE_CONFIG_H
#include <config.h>
#endif

#include <mhexcptn.h>
#include "OpenCVFrameSource.h"

using namespace std;

OpenCVFrameSource::OpenCVFrameSource(CvCapture* c)
  : capture(c), n(0) {
    MHDBGMSG(__FILE__,__LINE__,__PRETTY_FUNCTION__);
    cout << "source: " << cvGetCaptureProperty(capture, CV_CAP_PROP_FPS) << " fps" << endl;
    int fcc = static_cast<int>(cvGetCaptureProperty(capture, CV_CAP_PROP_FOURCC));
    char fourcc[] = {fcc & 0XFF , (fcc & 0XFF00) >> 8,(fcc & 0XFF0000) >> 16,(fcc & 0XFF000000) >> 24, 0};
    cout << "codec: " << fourcc << endl;
}

OpenCVFrameSource::~OpenCVFrameSource() {
    cvReleaseCapture(&capture);
}

IplImage* OpenCVFrameSource::GetNextFrame() {
    if (capture==nullptr) return nullptr;
    MHDBGMSG(__FILE__,__LINE__,__PRETTY_FUNCTION__);
    if (!cvGrabFrame(capture)) return nullptr;
    MHDBGMSG(__FILE__,__LINE__,__PRETTY_FUNCTION__);
    
    ++n;

    return cvRetrieveFrame(capture);
}

OpenCVFrameSource*
OpenCVFrameSource::CameraSourceFactory(int i) {
    return new OpenCVFrameSource(cvCaptureFromCAM(i));
}

OpenCVFrameSource*
OpenCVFrameSource::FileSourceFactory(const char *fn) {
    return new OpenCVFrameSource(cvCaptureFromFile(fn));
}