videoproducer.cpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2006-2021 Istituto Italiano di Tecnologia (IIT)
 * SPDX-License-Identifier: LGPL-2.1-or-later
 */
 
#include "videoproducer.h"
#include <QVideoFrame>
 
VideoProducer::VideoProducer(QObject *parent) :
    QObject(parent)
{
    m_format = nullptr;
    m_surface = nullptr;
    m_frame = nullptr;
}
 
VideoProducer::~VideoProducer()
{
    if(m_format)
    {
        delete m_format;
        m_format = nullptr;
    }
}
 
 
QAbstractVideoSurface* VideoProducer::videoSurface() const
{
    return m_surface;
}
 
void VideoProducer::setVideoSurface(QAbstractVideoSurface *surface)
{
    if (m_surface != surface && m_surface && m_surface->isActive()) {
        m_surface->stop();
    }
    m_surface = surface;
 
}
 
int VideoProducer::getWidth()
{
    if(!m_format){
        return 0;
    }
    int w = m_format->frameWidth();
    return w;
}
 
int VideoProducer::getHeight()
{
    if(!m_format){
        return 0;
    }
    int h = m_format->frameHeight();
    return h;
}
 
void VideoProducer::onNewVideoContentReceived(QVideoFrame *frame)
{
    if (m_surface){
        if(m_surface->isActive() && (m_format->frameWidth() != frame->size().width() ||
                                     m_format->frameHeight() != frame->size().height()))
        {
            m_surface->stop();
            delete m_format;
            m_format = nullptr;
        }
 
        if (!m_surface->isActive())
        {
            QSize s = frame->size();
            QVideoFrame::PixelFormat f = frame->pixelFormat();
            m_format = new QVideoSurfaceFormat(s, f);
 
            bool b = m_surface->start(*m_format);
            if (b)
            {
                qDebug("Surface STARTED! Dimensions: %dx%d -- PixelFormat: %d", s.width(), s.height(), (int)f);
            }
            else
            {
                qDebug("Surface START ERROR");
                delete m_format;
            }
            emit resizeWindowRequest();
        }
    }
 
    if (m_surface && m_format){
        bool b = m_surface->present(*frame);
        if (!b) {
            qWarning("Surface PRESENT ERROR");
        }
        else{
            mutex.lock();
            if(m_frame != nullptr){
                delete m_frame;
            }
            m_frame = new QVideoFrame(*frame);
            mutex.unlock();
        }
    }
 
}
 
QString VideoProducer::getPixelAsStr(int x, int y){
    mutex.lock();
    if (m_frame == nullptr || x>=m_frame->size().width() || y>=m_frame->size().height() || x<0 || y<0){
        mutex.unlock();
        return QString("Invalid");
    }
    m_frame->map( QAbstractVideoBuffer::ReadOnly );
    QImage image(m_frame->bits(), m_frame->width(),
                 m_frame->height(), m_frame->bytesPerLine(),
                 QImage::Format_RGB32);
    mutex.unlock();
    QRgb pixelVal = image.pixel(x,y);
    QString rgbHex("#");
    rgbHex += QString::number(pixelVal,16);
 
    return rgbHex;
}