qTox/src/video/netvideosource.cpp

/*
    This file is part of qTox, a Qt-based graphical interface for Tox.

    This program is libre software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

    See the COPYING file for more details.
*/

#include "netvideosource.h"

#include <QDebug>
#include <vpx/vpx_image.h>

NetVideoSource::NetVideoSource()
{
}

void NetVideoSource::pushFrame(VideoFrame frame)
{
    emit frameAvailable(frame);
}

void NetVideoSource::pushVPXFrame(const vpx_image *image)
{
    const int dw = image->d_w;
    const int dh = image->d_h;

    const int bpl = image->stride[VPX_PLANE_Y];
    const int cxbpl = image->stride[VPX_PLANE_V];

    VideoFrame frame;
    frame.frameData.resize(dw * dh * 3); //YUV 24bit
    frame.resolution = QSize(dw, dh);
    frame.format = VideoFrame::YUV;

    const uint8_t* yData = image->planes[VPX_PLANE_Y];
    const uint8_t* uData = image->planes[VPX_PLANE_V];
    const uint8_t* vData = image->planes[VPX_PLANE_U];

    // convert from planar to packed
    for (int y = 0; y < dh; ++y)
    {
        for (int x = 0; x < dw; ++x)
        {
            uint8_t Y = yData[x + y * bpl];
            uint8_t U = uData[x/(1 << image->x_chroma_shift) + y/(1 << image->y_chroma_shift)*cxbpl];
            uint8_t V = vData[x/(1 << image->x_chroma_shift) + y/(1 << image->y_chroma_shift)*cxbpl];

            frame.frameData.data()[dw * 3 * y + x * 3 + 0] = Y;
            frame.frameData.data()[dw * 3 * y + x * 3 + 1] = U;
            frame.frameData.data()[dw * 3 * y + x * 3 + 2] = V;
        }
    }

    pushFrame(frame);
}