FakeFrameGrabber.h

Go to the documentation of this file.

/*
 * Copyright (C) 2006-2020 Istituto Italiano di Tecnologia (IIT)
 * Copyright (C) 2006-2010 RobotCub Consortium
 * All rights reserved.
 *
 * This software may be modified and distributed under the terms of the
 * BSD-3-Clause license. See the accompanying LICENSE file for details.
 */
 
#ifndef YARP_FAKEFRAMEGRABBER_FAKEFRAMEGRABBER_H
#define YARP_FAKEFRAMEGRABBER_FAKEFRAMEGRABBER_H
 
#include <cstdio>
 
#include <yarp/sig/ImageFile.h>
#include <yarp/dev/DeviceDriver.h>
#include <yarp/dev/FrameGrabberInterfaces.h>
#include <yarp/dev/AudioVisualInterfaces.h>
#include <yarp/dev/IPreciselyTimed.h>
#include <yarp/os/Searchable.h>
#include <yarp/os/Time.h>
#include <yarp/os/Vocab.h>
#include <yarp/os/Log.h>
#include <yarp/os/Value.h>
#include <yarp/dev/IVisualParams.h>
 
constexpr yarp::conf::vocab32_t VOCAB_LINE = yarp::os::createVocab('l','i','n','e');
 
class FakeFrameGrabber :
        public yarp::dev::DeviceDriver,
        public yarp::dev::IFrameGrabberImage,
        public yarp::dev::IFrameGrabberImageRaw,
        public yarp::dev::IFrameGrabberControls,
        public yarp::dev::IPreciselyTimed,
        public yarp::dev::IAudioVisualStream,
        public yarp::dev::IRgbVisualParams
{
private:
    size_t ct;
    size_t bx, by;
    size_t w, h;
    unsigned long rnd;
    double freq;
    double period;
    double first;
    double horizontalFov,verticalFov;
    double prev;
    int mode;
    bool use_bayer;
    bool use_mono;
    bool mirror;
    yarp::os::Property intrinsic;
    yarp::sig::VectorOf<yarp::dev::CameraConfig> configurations;
 
public:
    FakeFrameGrabber();
 
    bool close() override;
 
    bool open(yarp::os::Searchable& config) override;
 
    void timing();
 
    int height() const override;
 
    int width() const override;
    //IRgbVisualParams
    int getRgbHeight() override;
 
    int getRgbWidth() override;
 
    bool getRgbSupportedConfigurations(yarp::sig::VectorOf<yarp::dev::CameraConfig>& configurations) override;
 
    bool getRgbResolution(int& width, int& height) override;
 
    bool setRgbResolution(int width, int height) override;
 
    bool getRgbFOV(double& horizontalFov, double& verticalFov) override;
 
    bool setRgbFOV(double horizontalFov, double verticalFov) override;
 
    bool getRgbIntrinsicParam(yarp::os::Property& intrinsic) override;
 
    bool getRgbMirroring(bool& mirror) override;
 
    bool setRgbMirroring(bool mirror) override;
    //
    bool getImage(yarp::sig::ImageOf<yarp::sig::PixelRgb>& image) override;
 
    bool getImage(yarp::sig::ImageOf<yarp::sig::PixelMono>& image) override;
 
    yarp::os::Stamp getLastInputStamp() override;
 
    bool hasAudio() override;
 
    bool hasVideo() override;
 
    bool hasRawVideo() override;
 
    bool getCameraDescription(CameraDescriptor *camera) override;
 
    bool hasFeature(int feature, bool *hasFeature) override;
 
    bool setFeature(int feature, double value) override;
 
    bool getFeature(int feature, double *value) override;
 
    bool setFeature(int feature, double  value1, double  value2) override;
 
    bool getFeature(int feature, double *value1, double *value2) override;
 
    bool hasOnOff(int feature, bool *HasOnOff) override;
 
    bool setActive(int feature, bool onoff) override;
 
    bool getActive(int feature, bool *isActive) override;
 
    bool hasAuto(int feature, bool *hasAuto) override;
 
    bool hasManual(int feature, bool *hasManual) override;
 
    bool hasOnePush(int feature, bool *hasOnePush) override;
 
    bool setMode(int feature, FeatureMode mode) override;
 
    bool getMode(int feature, FeatureMode *mode) override;
 
    bool setOnePush(int feature) override;
 
private:
    yarp::sig::ImageOf<yarp::sig::PixelRgb> background, rgb_image;
    yarp::os::Stamp stamp;
    void createTestImage(yarp::sig::ImageOf<yarp::sig::PixelRgb>& image);
 
    bool makeSimpleBayer(yarp::sig::ImageOf<yarp::sig::PixelRgb>& src,
                         yarp::sig::ImageOf<yarp::sig::PixelMono>& bayer);
 
    void printTime(unsigned char* pixbuf, int pixbuf_w, int pixbuf_h, int x, int y, char* s, int size);
    struct txtnum_type
    {
        char data[16];
    };
    txtnum_type num[12];
    double start_time;
};
 
 
class TestFrameGrabber : public FakeFrameGrabber
{
public:
    bool open(yarp::os::Searchable& config) override;
};
 
#endif // YARP_FAKEFRAMEGRABBER_FAKEFRAMEGRABBER_H