Ask Your Question

Revision history [back]

the code below is from an older question maybe it will be a sample for use of std::queue<cv::Mat> ( i have no deep knowledge about the code )

#include "opencv2/imgproc.hpp"
#include "opencv2/highgui.hpp"
#include "opencv2/calib3d.hpp"

#include <map>
#include <queue>
#include <thread>
#include <mutex>
#include <atomic>
#include <string>
#include <iostream>

/** @brief Example Class that grabs from cv::VideoCapture into a queue.

It can be used as worker thread.

Even if you grab N frames and create N x Mat in the queue
only few blocks will be really allocated thanks to @c std::vector
and @c cv::Mat memory recycling. Look at @c matMemoryCounter to
know how many memory blocks has been allocated
@author PkLab answer to question:
@see http://answers.opencv.org/question/95193/capture-from-two-webcam/
@note It can be done much better but it's better than nothing :)
@note It requires C++11 or later
*/
class GrabberThread {
private:
    int device;
    std::mutex mtx;
    std::queue<cv::Mat > buffer;
    std::atomic<bool> grabOn;
    cv::VideoCapture cap;
    size_t bufSizeMax;
    std::map<unsigned char*, int> matMemoryCounter;
public:
    GrabberThread() : grabOn(false), device(0), bufSizeMax(0) {};
    ~GrabberThread()
    {
        grabOn.store(false);
        cap.release();
        while (!buffer.empty())
            buffer.pop();
        matMemoryCounter.clear();

    }
    /* @brief initialize members and open the capture device
    @return true if the device has successfully opened
    */
    bool Init(int dev = 0)
    {
        device = dev;
        cap.open(device);
        return cap.isOpened();
    }

    /* @brief Disable the grabbing loop control var.
    This will stop the grabbing thread
    */
    void StopGrabing()
    {
        grabOn.store(false);
    }

    /** @brief This is the thread that grabs into a queue
    @note In real application locks and multiple copyTo can be avoided
    using better queue/circular buffer!
    */
    void GrabThread()
    {
        matMemoryCounter.clear();
        uchar * frameMemoryAddr;

        if (!cap.isOpened()) cap.open(device);
        if (!cap.isOpened()) return;

        cv::Mat tmp;
        grabOn.store(true);
        while (grabOn.load() == true) //this is lock free
        {
            //read() will wait for cam FPS. 
            //keep grab out of lock so that
            //idle time can be used by other threads
            if (!cap.read(tmp))
                continue;
            //get lock only when we have a frame
            mtx.lock();
            /** @warning Avoid @c buffer.push_back(tmp) because it stores items by reference.
            We have to create a new cv::Mat for each frame
            STL and cv::Mat will recycle memory
            */
            buffer.push(cv::Mat(tmp.size(), tmp.type()));
            tmp.copyTo(buffer.back());
            //count how many times this memory block has been used
            frameMemoryAddr = buffer.front().data;
            matMemoryCounter[frameMemoryAddr]++;
            bufSizeMax = std::max(bufSizeMax, buffer.size());

            mtx.unlock();
            /* SHOW THE FRAME if do you want
            string winName = "Live Thread Dev: " + std::to_string(device);
            int font = CV_FONT_HERSHEY_PLAIN;
            cv::putText(tmp, winName, cv::Point(10, 10), font, 1, cv::Scalar(0, 255, 0));
            imshow(winName, tmp);
            cv::waitKey(1);    //just for imshow
            */
        } // while
    } // func

      /** @brief Get the oldest grabbed frame if available

      The frame is copied from the buffer than it can used freely

      @Param [out]frame the the oldest grabbed frame. if not available it will be empty
      @Param [out]pBufSize a pointer where returns current buffer size. If 0 it will be ignored
      @Param [out]pBufSizeMax a pointer where returns maximum buffer size. If 0 it will be ignored
      @return true if a frame is available
      */
    bool PopFrame(cv::Mat &frame, size_t *pBufSize = 0, size_t *pBufSizeMax = 0)
    {
        mtx.lock();
        size_t bufSize = buffer.size();
        bool res = bufSize > 0;
        if (res) {
            // get the oldest grabbed frame (queue=FIFO)
            buffer.front().copyTo(frame);
            // release the item
            buffer.pop();
        }
        else frame = cv::Mat();

        if (pBufSize) *pBufSize = bufSize;
        if (pBufSizeMax) *pBufSizeMax = bufSizeMax;
        mtx.unlock();
        return res;
    }

    // ----------------------------------------------
    // GET SET METHODS 

    /** @see VideoCapture::set */
    virtual bool capSet(int propId, double value) {
        std::lock_guard<std::mutex> lock(mtx);
        return cap.set(propId, value);
    }
    /** @see VideoCapture::get */
    virtual double capGet(int propId) {
        std::lock_guard<std::mutex> lock(mtx);
        return cap.get(propId);
    }
    size_t GetBufSize() {
        std::lock_guard<std::mutex> lock(mtx);
        return buffer.size();
    }
    size_t GetBufSizeMax() {
        std::lock_guard<std::mutex> lock(mtx);
        return bufSizeMax;
    }
    size_t GetMemoryCounter() {
        std::lock_guard<std::mutex> lock(mtx);
        return matMemoryCounter.size();
    }
    void PrintStats() {
        std::lock_guard<std::mutex> lock(mtx);
        std::cout << "DEVICE: " << std::to_string(device) << std::endl;
        std::cout << "\t MaxBufSize: " << bufSizeMax << std::endl;
        std::cout << "\t Mat Memory Counter: " << matMemoryCounter.size() << std::endl;
    }

};

void ProcessFrameUtil(cv::Mat &frame1, cv::Mat &frame2)
{
    if (!frame1.empty())
        cv::imshow("cap1", frame1);

    if (!frame2.empty())
        cv::imshow("cap2", frame2);
};

int main() {
    GrabberThread grab1, grab2;
    cv::Mat frame1, frame2;

    if (!grab1.Init(0 /*+ cv::CAP_DSHOW*/)) return -1;
    if (!grab2.Init(1 /*+ cv::CAP_DSHOW*/)) return -1;

    // start 2 grabbing thread from 2 cams
    std::thread t1(&GrabberThread::GrabThread, &grab1);
    std::thread t2(&GrabberThread::GrabThread, &grab2);
    size_t bufSize1, bufSize2;
    while (true) {

        grab1.PopFrame(frame1, &bufSize1);
        grab2.PopFrame(frame2, &bufSize2);
        // do some processing 
        ProcessFrameUtil(frame1, frame2);

        if (bufSize1 > 10)
            std::cout << "WARNING buf1 too large(" << bufSize1 << ")" << std::endl;
        if (bufSize2 > 10)
            std::cout << "WARNING buf2 too large(" << bufSize2 << ")" << std::endl;

        if (cv::waitKey(5) >= 0)    // press any key to terminate
        {
            grab1.StopGrabing();
            grab2.StopGrabing();
            t1.join();                // wait thread exits
            t2.join();                // wait thread exits

            std::cout << "Flushing buffer1 of:" << grab1.GetBufSize() << " frames...";
            std::cout << "Flushing buffer2 of:" << grab2.GetBufSize() << " frames...";
            while (!(frame1.empty() && frame2.empty()))
            {
                grab1.PopFrame(frame1);
                grab2.PopFrame(frame2);
                // do some processing 
                ProcessFrameUtil(frame1, frame2);
            }
            break; // exit from process loop
        }
    }
    std::cout << "done" << std::endl;
    grab1.PrintStats();
    grab2.PrintStats();

    std::cout << std::endl << "Press Enter to terminate ";
    std::cin.get();
    return 0;
}