496 lines
14 KiB
C++
Executable File
496 lines
14 KiB
C++
Executable File
/* This file is a part of plaYUVer project
|
|
* Copyright (C) 2014 by Luis Lucas (luisfrlucas@gmail.com)
|
|
* Joao Carreira (jfmcarreira@gmail.com)
|
|
*
|
|
* This program is free 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 2 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
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
/**
|
|
* \file PlaYUVerFrame.cpp
|
|
* \brief Video Frame handling
|
|
*/
|
|
#include <cstdio>
|
|
|
|
#include "PlaYUVerFrame.h"
|
|
#include "LibMemAlloc.h"
|
|
#include "LibMemory.h"
|
|
|
|
namespace plaYUVer
|
|
{
|
|
|
|
PlaYUVerFrame::PlaYUVerFrame( UInt width, UInt height, Int pel_format )
|
|
{
|
|
m_pppcInputPel = NULL;
|
|
m_uiWidth = width;
|
|
m_uiHeight = height;
|
|
m_iPixelFormat = pel_format;
|
|
|
|
m_bHasRGBPel = false;
|
|
|
|
switch( m_iPixelFormat )
|
|
{
|
|
case YUV420p:
|
|
getMem3ImageComponents<Pel>( &m_pppcInputPel, m_uiHeight, m_uiWidth, 2, 2 );
|
|
break;
|
|
case YUV444p:
|
|
getMem3ImageComponents<Pel>( &m_pppcInputPel, m_uiHeight, m_uiWidth, 1, 1 );
|
|
break;
|
|
case YUV422p:
|
|
getMem3ImageComponents<Pel>( &m_pppcInputPel, m_uiHeight, m_uiWidth, 1, 2 );
|
|
break;
|
|
case YUV410p:
|
|
case YUV411p:
|
|
break;
|
|
case GRAY:
|
|
getMem3ImageComponents<Pel>( &m_pppcInputPel, m_uiHeight, m_uiWidth, 4, 4 );
|
|
break;
|
|
case RGB8:
|
|
getMem3ImageComponents<Pel>( &m_pppcInputPel, m_uiHeight, m_uiWidth, 1, 1 );
|
|
break;
|
|
}
|
|
getMem1D<Pel>( &m_pcRGBPelInterlaced, m_uiHeight * m_uiWidth * 3 );
|
|
openPixfc();
|
|
}
|
|
|
|
PlaYUVerFrame::~PlaYUVerFrame()
|
|
{
|
|
if( m_pppcInputPel )
|
|
freeMem3ImageComponents<Pel>( m_pppcInputPel );
|
|
|
|
if( m_pcRGBPelInterlaced )
|
|
freeMem1D<UChar>( m_pcRGBPelInterlaced );
|
|
|
|
closePixfc();
|
|
}
|
|
|
|
Void PlaYUVerFrame::openPixfc()
|
|
{
|
|
#ifdef USE_PIXFC
|
|
m_pcPixfc = NULL;
|
|
PixFcPixelFormat input_format = PixFcYUV420P;
|
|
PixFcFlag flags = (PixFcFlag)(PixFcFlag_Default + PixFcFlag_NNbResamplingOnly);
|
|
UInt input_row_size = m_uiWidth;
|
|
if (create_pixfc(&m_pcPixfc, input_format, PixFcRGB24, m_uiWidth, m_uiHeight, input_row_size, m_uiWidth * 3, flags ) != 0)
|
|
{
|
|
fprintf(stderr, "Error creating struct pixfc\n");
|
|
m_pcPixfc = NULL;
|
|
return;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
Void PlaYUVerFrame::closePixfc()
|
|
{
|
|
#ifdef USE_PIXFC
|
|
if( m_pcPixfc )
|
|
destroy_pixfc(m_pcPixfc);
|
|
#endif
|
|
}
|
|
|
|
#ifdef USE_PIXFC
|
|
inline Void PlaYUVerFrame::FrametoRGB8Pixfc()
|
|
{
|
|
switch( m_iPixelFormat )
|
|
{
|
|
case YUV420p:
|
|
m_pcPixfc->convert(m_pcPixfc, m_pppcInputPel, m_pcRGBPelInterlaced);
|
|
break;
|
|
case YUV422p:
|
|
break;
|
|
case YUV444p:
|
|
m_pcPixfc->convert(m_pcPixfc, m_pppcInputPel, m_pcRGBPelInterlaced);
|
|
break;
|
|
case YUV410p:
|
|
case YUV411p:
|
|
break;
|
|
case GRAY:
|
|
break;
|
|
case RGB8:
|
|
break;
|
|
default:
|
|
// No action.
|
|
break;
|
|
}
|
|
m_bHasRGBPel = true;
|
|
}
|
|
|
|
#endif
|
|
|
|
template<typename T>
|
|
Void yuvToRgb( T iY, T iU, T iV, T &iR, T &iG, T &iB )
|
|
{
|
|
iR = iY + 1402 * iV / 1000;
|
|
iG = iY - ( 101004 * iU + 209599 * iV ) / 293500;
|
|
iB = iY + 1772 * iU / 1000;
|
|
|
|
if( iR < 0 )
|
|
iR = 0;
|
|
if( iG < 0 )
|
|
iG = 0;
|
|
if( iB < 0 )
|
|
iB = 0;
|
|
|
|
if( iR > 255 )
|
|
iR = 255;
|
|
if( iG > 255 )
|
|
iG = 255;
|
|
if( iB > 255 )
|
|
iB = 255;
|
|
}
|
|
|
|
static inline Void yuvToRgb( Int iY, Int iU, Int iV, Int &iR, Int &iG, Int &iB )
|
|
{
|
|
iR = iY + 1402 * iV / 1000;
|
|
iG = iY - ( 101004 * iU + 209599 * iV ) / 293500;
|
|
iB = iY + 1772 * iU / 1000;
|
|
|
|
if( iR < 0 )
|
|
iR = 0;
|
|
if( iG < 0 )
|
|
iG = 0;
|
|
if( iB < 0 )
|
|
iB = 0;
|
|
|
|
if( iR > 255 )
|
|
iR = 255;
|
|
if( iG > 255 )
|
|
iG = 255;
|
|
if( iB > 255 )
|
|
iB = 255;
|
|
}
|
|
|
|
template<typename T>
|
|
Void rgbToyuv( T iR, T iG, T iB, T &iY, T &iU, T &iV )
|
|
{
|
|
iY = 0;
|
|
iU = 0;
|
|
iV = 0;
|
|
}
|
|
|
|
Void PlaYUVerFrame::FrametoRGB8()
|
|
{
|
|
if( m_bHasRGBPel )
|
|
return;
|
|
#ifdef USE_PIXFC
|
|
if( m_pcPixfc )
|
|
{
|
|
FrametoRGB8Pixfc();
|
|
}
|
|
#endif
|
|
{
|
|
Int iY, iU, iV, iR, iG, iB;
|
|
Pel** ppInputPelY = m_pppcInputPel[LUMA];
|
|
Pel* pInputPelY = &( m_pppcInputPel[LUMA][0][0] );
|
|
|
|
Pel *pInputPelU = NULL, *pInputPelV = NULL;
|
|
if( m_pppcInputPel[CHROMA_U] )
|
|
{
|
|
pInputPelU = m_pppcInputPel[CHROMA_U][0];
|
|
pInputPelV = m_pppcInputPel[CHROMA_V][0];
|
|
}
|
|
|
|
Pel* pcRGBPelInterlaced = m_pcRGBPelInterlaced;
|
|
Pel* pcRGBPelInterlacedl1 = m_pcRGBPelInterlaced + m_uiWidth * 3;
|
|
|
|
switch( m_iPixelFormat )
|
|
{
|
|
case YUV420p:
|
|
for( UInt y = 0; y < m_uiHeight; y += 2 )
|
|
{
|
|
for( UInt x = 0; x < m_uiWidth; x += 2 )
|
|
{
|
|
// Pixel (x, y).
|
|
iY = ppInputPelY[y][x];
|
|
iU = *pInputPelU++ - 128;
|
|
iV = *pInputPelV++ - 128;
|
|
yuvToRgb<Int>( iY, iU, iV, iR, iG, iB );
|
|
*pcRGBPelInterlaced++ = iR;
|
|
*pcRGBPelInterlaced++ = iG;
|
|
*pcRGBPelInterlaced++ = iB;
|
|
// Pixel (x+1, y)
|
|
iY = ppInputPelY[y][x + 1];
|
|
yuvToRgb<Int>( iY, iU, iV, iR, iG, iB );
|
|
*pcRGBPelInterlaced++ = iR;
|
|
*pcRGBPelInterlaced++ = iG;
|
|
*pcRGBPelInterlaced++ = iB;
|
|
// Pixel (x, y+1)
|
|
iY = ppInputPelY[y + 1][x];
|
|
yuvToRgb<Int>( iY, iU, iV, iR, iG, iB );
|
|
*pcRGBPelInterlacedl1++ = iR;
|
|
*pcRGBPelInterlacedl1++ = iG;
|
|
*pcRGBPelInterlacedl1++ = iB;
|
|
// Pixel (x+1, y+1)
|
|
iY = ppInputPelY[y + 1][x + 1];
|
|
yuvToRgb<Int>( iY, iU, iV, iR, iG, iB );
|
|
*pcRGBPelInterlacedl1++ = iR;
|
|
*pcRGBPelInterlacedl1++ = iG;
|
|
*pcRGBPelInterlacedl1++ = iB;
|
|
}
|
|
pcRGBPelInterlaced += m_uiWidth * 3;
|
|
pcRGBPelInterlacedl1 += m_uiWidth * 3;
|
|
}
|
|
break;
|
|
case YUV422p:
|
|
for( UInt y = 0; y < m_uiHeight; y += 1 )
|
|
for( UInt x = 0; x < m_uiWidth; x += 2 )
|
|
{
|
|
// Pixel (x, y).
|
|
iY = ppInputPelY[y][x];
|
|
iU = *pInputPelU++ - 128;
|
|
iV = *pInputPelV++ - 128;
|
|
yuvToRgb<Int>( iY, iU, iV, iR, iG, iB );
|
|
*pcRGBPelInterlaced++ = iR;
|
|
*pcRGBPelInterlaced++ = iG;
|
|
*pcRGBPelInterlaced++ = iB;
|
|
// Pixel (x+1, y)
|
|
iY = ppInputPelY[y][x + 1];
|
|
yuvToRgb<Int>( iY, iU, iV, iR, iG, iB );
|
|
*pcRGBPelInterlaced++ = iR;
|
|
*pcRGBPelInterlaced++ = iG;
|
|
*pcRGBPelInterlaced++ = iB;
|
|
}
|
|
break;
|
|
case YUV444p:
|
|
for( UInt y = 0; y < m_uiHeight; y++ )
|
|
for( UInt x = 0; x < m_uiWidth; x++ )
|
|
{
|
|
iY = *pInputPelY++;
|
|
iU = *pInputPelU++ - 128;
|
|
iV = *pInputPelV++ - 128;
|
|
yuvToRgb<Int>( iY, iU, iV, iR, iG, iB );
|
|
*pcRGBPelInterlaced++ = iR;
|
|
*pcRGBPelInterlaced++ = iG;
|
|
*pcRGBPelInterlaced++ = iB;
|
|
}
|
|
break;
|
|
case YUV410p:
|
|
case YUV411p:
|
|
break;
|
|
case GRAY:
|
|
for( UInt y = 0; y < m_uiHeight; y++ )
|
|
for( UInt x = 0; x < m_uiWidth; x++ )
|
|
{
|
|
iY = *pInputPelY++;
|
|
//yuvToRgb( iY, iY, iY, iR, iG, iB );
|
|
iR = iG = iB = iY;
|
|
*pcRGBPelInterlaced++ = iR;
|
|
*pcRGBPelInterlaced++ = iG;
|
|
*pcRGBPelInterlaced++ = iB;
|
|
}
|
|
break;
|
|
case RGB8:
|
|
break;
|
|
default:
|
|
// No action.
|
|
break;
|
|
}
|
|
m_bHasRGBPel = true;
|
|
}
|
|
}
|
|
|
|
Void PlaYUVerFrame::FrameFromBuffer( Pel *input_buffer, Int pel_format )
|
|
{
|
|
Pel* pcRGBPelInterlaced = m_pcRGBPelInterlaced;
|
|
if( m_iPixelFormat != pel_format )
|
|
{
|
|
return;
|
|
}
|
|
m_bHasRGBPel = false;
|
|
UInt64 frame_bytes_input = m_uiWidth * m_uiHeight;
|
|
switch( m_iPixelFormat )
|
|
{
|
|
case YUV420p:
|
|
memcpy( &m_pppcInputPel[LUMA][0][0], input_buffer + frame_bytes_input * 0, frame_bytes_input * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_U][0][0], input_buffer + frame_bytes_input * 1, frame_bytes_input / 4 * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_V][0][0], input_buffer + frame_bytes_input * 5 / 4, frame_bytes_input / 4 * sizeof(Pel) );
|
|
break;
|
|
case YUV444p:
|
|
memcpy( &m_pppcInputPel[LUMA][0][0], input_buffer + frame_bytes_input * 0, frame_bytes_input * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_U][0][0], input_buffer + frame_bytes_input * 1, frame_bytes_input * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_V][0][0], input_buffer + frame_bytes_input * 2, frame_bytes_input * sizeof(Pel) );
|
|
break;
|
|
case YUV422p:
|
|
memcpy( &m_pppcInputPel[LUMA][0][0], input_buffer + frame_bytes_input * 0, frame_bytes_input * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_U][0][0], input_buffer + frame_bytes_input * 1, frame_bytes_input / 2 * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_V][0][0], input_buffer + frame_bytes_input * 3 / 2, frame_bytes_input / 2 * sizeof(Pel) );
|
|
break;
|
|
case YUV410p:
|
|
case YUV411p:
|
|
break;
|
|
case GRAY:
|
|
memcpy( &m_pppcInputPel[LUMA][0][0], input_buffer + frame_bytes_input * 0, frame_bytes_input * sizeof(Pel) );
|
|
break;
|
|
case RGB8:
|
|
memcpy( pcRGBPelInterlaced, input_buffer, frame_bytes_input * sizeof(Pel) * 3 );
|
|
for( UInt y = 0; y < m_uiHeight; y++ )
|
|
for( UInt x = 0; x < m_uiWidth; x++ )
|
|
{
|
|
m_pppcInputPel[COLOR_R][y][x] = *pcRGBPelInterlaced++;
|
|
m_pppcInputPel[COLOR_G][y][x] = *pcRGBPelInterlaced++;
|
|
m_pppcInputPel[COLOR_B][y][x] = *pcRGBPelInterlaced++;
|
|
}
|
|
m_bHasRGBPel = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
Void PlaYUVerFrame::CopyFrom( PlaYUVerFrame* input_frame )
|
|
{
|
|
if( m_iPixelFormat != input_frame->getPelFormat() )
|
|
{
|
|
return;
|
|
}
|
|
m_bHasRGBPel = false;
|
|
UInt64 num_bytes = m_uiWidth * m_uiHeight;
|
|
switch( m_iPixelFormat )
|
|
{
|
|
case YUV420p:
|
|
memcpy( &m_pppcInputPel[LUMA][0][0], &( input_frame->getPelBufferYUV()[LUMA][0][0] ), num_bytes * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_U][0][0], &( input_frame->getPelBufferYUV()[CHROMA_U][0][0] ), num_bytes / 4 * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_V][0][0], &( input_frame->getPelBufferYUV()[CHROMA_V][0][0] ), num_bytes / 4 * sizeof(Pel) );
|
|
break;
|
|
case YUV444p:
|
|
memcpy( &m_pppcInputPel[LUMA][0][0], &( input_frame->getPelBufferYUV()[LUMA][0][0] ), num_bytes * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_U][0][0], &( input_frame->getPelBufferYUV()[CHROMA_U][0][0] ), num_bytes * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_V][0][0], &( input_frame->getPelBufferYUV()[CHROMA_V][0][0] ), num_bytes * sizeof(Pel) );
|
|
break;
|
|
case YUV422p:
|
|
memcpy( &m_pppcInputPel[LUMA][0][0], &( input_frame->getPelBufferYUV()[LUMA][0][0] ), num_bytes * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_U][0][0], &( input_frame->getPelBufferYUV()[CHROMA_U][0][0] ), num_bytes / 2 * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[CHROMA_V][0][0], &( input_frame->getPelBufferYUV()[CHROMA_V][0][0] ), num_bytes / 2 * sizeof(Pel) );
|
|
case YUV410p:
|
|
case YUV411p:
|
|
break;
|
|
case GRAY:
|
|
memcpy( &m_pppcInputPel[LUMA][0][0], &( input_frame->getPelBufferYUV()[LUMA][0][0] ), num_bytes * sizeof(Pel) );
|
|
break;
|
|
case RGB8:
|
|
memcpy( &m_pppcInputPel[COLOR_R][0][0], &( input_frame->getPelBufferYUV()[COLOR_R][0][0] ), num_bytes * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[COLOR_G][0][0], &( input_frame->getPelBufferYUV()[COLOR_G][0][0] ), num_bytes * sizeof(Pel) );
|
|
memcpy( &m_pppcInputPel[COLOR_B][0][0], &( input_frame->getPelBufferYUV()[COLOR_B][0][0] ), num_bytes * sizeof(Pel) );
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
UInt64 PlaYUVerFrame::getBytesPerFrame()
|
|
{
|
|
switch( m_iPixelFormat )
|
|
{
|
|
case YUV420p:
|
|
return m_uiWidth * m_uiHeight * 1.5;
|
|
break;
|
|
case YUV444p:
|
|
return m_uiWidth * m_uiHeight * 3;
|
|
break;
|
|
case YUV422p:
|
|
return m_uiWidth * m_uiHeight * 2;
|
|
break;
|
|
case YUV410p:
|
|
return m_uiWidth * m_uiHeight * 1.25;
|
|
break;
|
|
case GRAY:
|
|
return m_uiWidth * m_uiHeight;
|
|
break;
|
|
case RGB8:
|
|
return m_uiWidth * m_uiHeight * 3;
|
|
break;
|
|
default:
|
|
return 0;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
Pixel PlaYUVerFrame::ConvertPixel( Pixel sInputPixel, ColorSpace eOutputSpace )
|
|
{
|
|
Pixel sOutputPixel =
|
|
{
|
|
COLOR_INVALID,
|
|
0,
|
|
0,
|
|
0 };
|
|
|
|
if( sInputPixel.color_space == eOutputSpace )
|
|
return sInputPixel;
|
|
|
|
if( eOutputSpace == COLOR_RGB )
|
|
{
|
|
yuvToRgb( sInputPixel.Luma, sInputPixel.ChromaU, sInputPixel.ChromaU, sOutputPixel.ColorR, sOutputPixel.ColorG, sOutputPixel.ColorB );
|
|
}
|
|
if( eOutputSpace == COLOR_YUV )
|
|
{
|
|
rgbToyuv( sInputPixel.ColorR, sInputPixel.ColorG, sInputPixel.ColorB, sOutputPixel.Luma, sOutputPixel.ChromaU, sOutputPixel.ChromaU );
|
|
}
|
|
return sOutputPixel;
|
|
}
|
|
|
|
Pixel PlaYUVerFrame::getPixelValue( const QPoint &pos, ColorSpace space )
|
|
{
|
|
Pixel pixel_value;
|
|
switch( m_iPixelFormat )
|
|
{
|
|
case YUV420p:
|
|
pixel_value.color_space = COLOR_YUV;
|
|
pixel_value.Luma = m_pppcInputPel[LUMA][pos.y()][pos.x()];
|
|
pixel_value.ChromaU = m_pppcInputPel[CHROMA_U][pos.y() >> 1][pos.x() >> 1];
|
|
pixel_value.ChromaV = m_pppcInputPel[CHROMA_V][pos.y() >> 1][pos.x() >> 1];
|
|
break;
|
|
case YUV444p:
|
|
pixel_value.Luma = m_pppcInputPel[LUMA][pos.y()][pos.x()];
|
|
pixel_value.ChromaU = m_pppcInputPel[CHROMA_U][pos.y()][pos.x()];
|
|
pixel_value.ChromaV = m_pppcInputPel[CHROMA_V][pos.y()][pos.x()];
|
|
break;
|
|
case YUV422p:
|
|
pixel_value.color_space = COLOR_YUV;
|
|
pixel_value.Luma = m_pppcInputPel[LUMA][pos.y()][pos.x()];
|
|
pixel_value.ChromaU = m_pppcInputPel[CHROMA_U][pos.y()][pos.x() >> 1];
|
|
pixel_value.ChromaV = m_pppcInputPel[CHROMA_V][pos.y()][pos.x() >> 1];
|
|
break;
|
|
case YUV410p:
|
|
pixel_value.color_space = COLOR_YUV;
|
|
pixel_value.Luma = m_pppcInputPel[LUMA][pos.y()][pos.x()];
|
|
pixel_value.ChromaU = m_pppcInputPel[CHROMA_U][pos.y() >> 2][pos.x() >> 2];
|
|
pixel_value.ChromaV = m_pppcInputPel[CHROMA_V][pos.y() >> 2][pos.x() >> 2];
|
|
break;
|
|
case GRAY:
|
|
pixel_value.color_space = COLOR_YUV;
|
|
pixel_value.Luma = m_pppcInputPel[LUMA][pos.y()][pos.x()];
|
|
pixel_value.ChromaU = 0;
|
|
pixel_value.ChromaV = 0;
|
|
break;
|
|
case RGB8:
|
|
pixel_value.color_space = COLOR_RGB;
|
|
pixel_value.ColorR = m_pppcInputPel[COLOR_R][pos.y()][pos.x()];
|
|
pixel_value.ColorG = m_pppcInputPel[COLOR_G][pos.y()][pos.x()];
|
|
pixel_value.ColorB = m_pppcInputPel[COLOR_B][pos.y()][pos.x()];
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
ConvertPixel( pixel_value, space );
|
|
return pixel_value;
|
|
}
|
|
|
|
// Format conversion
|
|
|
|
#ifdef USE_OPENCV
|
|
cv::Mat PlaYUVerFrame::getCvMat()
|
|
{
|
|
return cv::Mat();
|
|
}
|
|
#endif
|
|
}
|
|
// NAMESPACE
|