ImaqGrabber.h

Go to the documentation of this file.

#ifndef IMAQ_GRABBER_H
#define IMAQ_GRABBER_H

#include <interfaces/IImageStreamExtractor.h>
#include <ImageGrabber.h>

#include <zmq.hpp>

/* c++03 std lib */
#include <exception>
#include <string>
#include <vector>
#include <list>

/* c++11 std lib additions */
#include <thread>
#include <mutex>
#include <atomic>
#include <condition_variable>

/* 3rdparty libraries */
#define NI_STDCALL    __stdcall
#include <niimaq.h>

class ImaqException: public std::exception {
public:
    ImaqException(const std::string & _message);
    ~ImaqException() throw();
    virtual const char* what() const throw();
private:
    std::string message;
};

class _ImaqGrabberImpl : public _ImageGrabberImpl {
    friend ImageGrabber;
private:
    _ImaqGrabberImpl(const std::string & _deviceName, const std::string & _deviceSource, zmq::context_t & _ctx);
    ~_ImaqGrabberImpl();

    int getInteger(uInt32 attr);
    int getWidth();
    int getHeight();
    int getBytesPerPixel();
    void startServing();
    void stopServing();
    void acquisition();
    void acquireBuffer(int bufferId);
    void releaseBuffer(int bufferId);
    void addEndpoint(const std::string & _endpoint);

    std::string deviceName;
    std::string deviceSource;
    INTERFACE_ID ifid;
    SESSION_ID sid;
    bool interfaceOpen;
    bool sessionOpen;
    int totalLostFrames = 0;
    std::thread acquisitionThread;
    volatile bool enableAcquisition = false;
    std::vector<std::atomic<int>*> bufferUsers;
    std::atomic<int> openBuffers;
    std::atomic<int> totalBufferUsers;
    std::condition_variable bufferFreeNotifier;
    
    zmq::context_t & ctx;
    std::list<std::string> endpoints;
};

#endif /* IMAQ_GRABBER_H */