OpenCVGrabber.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2006-2021 Istituto Italiano di Tecnologia (IIT)
 * SPDX-FileCopyrightText: 2006-2010 RobotCub Consortium
 * SPDX-FileCopyrightText: 2006 Eric Mislivec
 * SPDX-License-Identifier: LGPL-2.1-or-later
 */
 
#ifndef YARP_OPENCV_GRABBER_DEVICE_OPENCVGRABBER_H
#define YARP_OPENCV_GRABBER_DEVICE_OPENCVGRABBER_H
 
/*
 * A YARP frame grabber device driver using OpenCV to implement
 * image capture from cameras and AVI files.
 */
 
 
#include <yarp/os/Property.h>
#include <yarp/dev/IFrameGrabberImage.h>
#include <yarp/dev/DeviceDriver.h>
#include <yarp/os/Stamp.h>
#include <yarp/dev/IPreciselyTimed.h>
 
#include <opencv2/highgui/highgui.hpp>
 
class OpenCVGrabber :
        public yarp::dev::IFrameGrabberImage,
        public yarp::dev::DeviceDriver,
        public yarp::dev::IPreciselyTimed
{
public:
 
    OpenCVGrabber() :
        m_w(0),
        m_h(0),
        m_loop(false),
        m_saidSize(false),
        m_saidResize(false),
        fromFile(false),
        m_cap(),
        m_transpose(false),
        m_flip_x(false),
        m_flip_y(false)
    {}
 
    virtual ~OpenCVGrabber() { ; }
 
 
 
    bool open(yarp::os::Searchable & config) override;
 
    bool close() override;
 
    bool getImage(yarp::sig::ImageOf<yarp::sig::PixelRgb> & image) override;
 
 
    inline int height() const override { return m_h; }
 
    inline int width() const override { return m_w; }
 
    inline yarp::os::Stamp getLastInputStamp() override { return m_laststamp; }
 
protected:
 
    int m_w;
    int m_h;
 
    bool m_loop;
 
    bool m_saidSize;
    bool m_saidResize;
 
    bool fromFile;
 
    cv::VideoCapture m_cap;
 
    /* optional image modifiers */
    bool m_transpose;
    bool m_flip_x;
    bool m_flip_y;
 
    yarp::os::Property m_config;
 
    yarp::os::Stamp m_laststamp;
};
 
#endif // YARP_OPENCV_GRABBER_DEVICE_OPENCVGRABBER_H