247 lines
6.2 KiB
C++
247 lines
6.2 KiB
C++
/* 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 "config.h"
|
|
#include <cstdio>
|
|
#include "LibMemory.h"
|
|
#include "PlaYUVerFrame.h"
|
|
#ifdef USE_FFMPEG
|
|
#include "LibAvContextHandle.h"
|
|
#endif
|
|
|
|
namespace plaYUVer
|
|
{
|
|
|
|
#ifdef USE_FFMPEG
|
|
#define ADD_FFMPEG_PEL_FMT( fmt ) fmt
|
|
#else
|
|
#define ADD_FFMPEG_PEL_FMT( fmt ) 0
|
|
#endif
|
|
|
|
|
|
Void bufferFromFrame_yuv420p( Pel ***in, Pel* out, UInt width, UInt height )
|
|
{
|
|
|
|
}
|
|
Void fillRGBbuffer_yuv420p( Pel*** in, UChar* out, UInt width, UInt height )
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Pixel getPixelValueYUV420p( Pel ***Img, Int xPos, Int yPos )
|
|
{
|
|
Pixel pixel_value;
|
|
pixel_value.color_space = PlaYUVerFrame::COLOR_YUV;
|
|
pixel_value.Luma = Img[LUMA][yPos][xPos];
|
|
pixel_value.ChromaU = Img[CHROMA_U][yPos >> 1][xPos >> 1];
|
|
pixel_value.ChromaV = Img[CHROMA_V][yPos >> 1][xPos >> 1];
|
|
return pixel_value;
|
|
}
|
|
Void frameFromBufferYUV420p( Pel *in, Pel*** out, UInt width, UInt height )
|
|
{
|
|
memcpy( &out[LUMA][0][0], in + width * height * 0, width * height * sizeof(Pel) );
|
|
memcpy( &out[CHROMA_U][0][0], in + width * height * 1, width * height / 4 * sizeof(Pel) );
|
|
memcpy( &out[CHROMA_V][0][0], in + width * height * 5 / 4, width * height / 4 * sizeof(Pel) );
|
|
}
|
|
PlaYUVerFramePelFormat yuv420p =
|
|
{
|
|
"YUV420p",
|
|
PlaYUVerFrame::COLOR_YUV,
|
|
3,
|
|
2,
|
|
2,
|
|
getPixelValueYUV420p,
|
|
frameFromBufferYUV420p,
|
|
bufferFromFrame_yuv420p,
|
|
fillRGBbuffer_yuv420p,
|
|
ADD_FFMPEG_PEL_FMT( AV_PIX_FMT_YUV420P ),
|
|
};
|
|
|
|
|
|
Void frameFromBufferYUV444p( Pel *in, Pel*** out, UInt width, UInt height )
|
|
{
|
|
memcpy( &out[LUMA][0][0], in + width * height * 0, width * height * sizeof(Pel) );
|
|
memcpy( &out[CHROMA_U][0][0], in + width * height * 1, width * height * sizeof(Pel) );
|
|
memcpy( &out[CHROMA_V][0][0], in + width * height * 2, width * height * sizeof(Pel) );
|
|
}
|
|
Pixel getPixelValueYUV444p( Pel ***Img, Int xPos, Int yPos )
|
|
{
|
|
Pixel pixel_value;
|
|
pixel_value.color_space = PlaYUVerFrame::COLOR_YUV;
|
|
pixel_value.Luma = Img[LUMA][yPos][xPos];
|
|
pixel_value.ChromaU = Img[CHROMA_U][yPos][xPos];
|
|
pixel_value.ChromaV = Img[CHROMA_V][yPos][xPos];
|
|
return pixel_value;
|
|
}
|
|
PlaYUVerFramePelFormat yuv444p =
|
|
{
|
|
"YUV444p",
|
|
PlaYUVerFrame::COLOR_YUV,
|
|
3,
|
|
1,
|
|
1,
|
|
getPixelValueYUV444p,
|
|
frameFromBufferYUV444p,
|
|
bufferFromFrame_yuv420p,
|
|
fillRGBbuffer_yuv420p,
|
|
ADD_FFMPEG_PEL_FMT( AV_PIX_FMT_YUV444P ),
|
|
};
|
|
|
|
|
|
Void frameFromBufferYUV422p( Pel *in, Pel*** out, UInt width, UInt height )
|
|
{
|
|
memcpy( &out[LUMA][0][0], in + width * height * 0, width * height * sizeof(Pel) );
|
|
memcpy( &out[CHROMA_U][0][0], in + width * height * 1, width * height / 2 * sizeof(Pel) );
|
|
memcpy( &out[CHROMA_V][0][0], in + width * height * 3 / 2, width * height / 2 * sizeof(Pel) );
|
|
}
|
|
Pixel getPixelValueYUV422( Pel ***Img, Int xPos, Int yPos )
|
|
{
|
|
Pixel pixel_value;
|
|
pixel_value.color_space = PlaYUVerFrame::COLOR_YUV;
|
|
pixel_value.Luma = Img[LUMA][yPos][xPos];
|
|
pixel_value.ChromaU = Img[CHROMA_U][yPos][xPos >> 1];
|
|
pixel_value.ChromaV = Img[CHROMA_V][yPos][xPos >> 1];
|
|
return pixel_value;
|
|
}
|
|
PlaYUVerFramePelFormat yuv422p =
|
|
{
|
|
"YUV422p",
|
|
PlaYUVerFrame::COLOR_YUV,
|
|
3,
|
|
2,
|
|
1,
|
|
getPixelValueYUV422,
|
|
frameFromBufferYUV422p,
|
|
bufferFromFrame_yuv420p,
|
|
fillRGBbuffer_yuv420p,
|
|
ADD_FFMPEG_PEL_FMT( AV_PIX_FMT_YUV422P ),
|
|
};
|
|
|
|
|
|
Void frameFromBufferYUV422( Pel *in, Pel*** out, UInt width, UInt height )
|
|
{
|
|
Pel* pY = out[LUMA][0];
|
|
Pel* pU = out[CHROMA_U][0];
|
|
Pel* pV = out[CHROMA_V][0];
|
|
for( UInt i = 0; i < width * height / 2; i++ )
|
|
{
|
|
*pY++ = *in++;
|
|
*pU++ = *in++;
|
|
*pY++ = *in++;
|
|
*pV++ = *in++;
|
|
}
|
|
}
|
|
PlaYUVerFramePelFormat yuv422 =
|
|
{
|
|
"YUV422",
|
|
PlaYUVerFrame::COLOR_YUV,
|
|
3,
|
|
2,
|
|
1,
|
|
getPixelValueYUV422,
|
|
frameFromBufferYUV422,
|
|
bufferFromFrame_yuv420p,
|
|
fillRGBbuffer_yuv420p,
|
|
ADD_FFMPEG_PEL_FMT( AV_PIX_FMT_YUYV422 ),
|
|
};
|
|
|
|
|
|
Pixel getPixelValueGray( Pel ***Img, Int xPos, Int yPos )
|
|
{
|
|
Pixel pixel_value;
|
|
pixel_value.color_space = PlaYUVerFrame::COLOR_YUV;
|
|
pixel_value.Luma = Img[LUMA][yPos][xPos];
|
|
pixel_value.ChromaU = 0;
|
|
pixel_value.ChromaV = 0;
|
|
return pixel_value;
|
|
}
|
|
Void frameFromBufferGray( Pel *in, Pel*** out, UInt width, UInt height )
|
|
{
|
|
memcpy( &out[LUMA][0][0], in, width * height * sizeof(Pel) );
|
|
}
|
|
PlaYUVerFramePelFormat gray =
|
|
{
|
|
"GRAY",
|
|
PlaYUVerFrame::COLOR_GRAY,
|
|
1,
|
|
0,
|
|
0,
|
|
getPixelValueGray,
|
|
frameFromBufferGray,
|
|
bufferFromFrame_yuv420p,
|
|
fillRGBbuffer_yuv420p,
|
|
ADD_FFMPEG_PEL_FMT( AV_PIX_FMT_GRAY8 ),
|
|
};
|
|
|
|
|
|
Pixel getPixelValueRGB8( Pel ***Img, Int xPos, Int yPos )
|
|
{
|
|
Pixel pixel_value;
|
|
pixel_value.color_space = PlaYUVerFrame::COLOR_RGB;
|
|
pixel_value.ColorR = Img[COLOR_R][yPos][xPos];
|
|
pixel_value.ColorG = Img[COLOR_G][yPos][xPos];
|
|
pixel_value.ColorB = Img[COLOR_B][yPos][xPos];
|
|
return pixel_value;
|
|
}
|
|
Void frameFromBufferRGB8( Pel *in, Pel*** out, UInt width, UInt height )
|
|
{
|
|
Pel* pR = out[COLOR_R][0];
|
|
Pel* pG = out[COLOR_G][0];
|
|
Pel* pB = out[COLOR_B][0];
|
|
for( UInt i = 0; i < width * height; i++ )
|
|
{
|
|
*pR++ = *in++;
|
|
*pG++ = *in++;
|
|
*pB++ = *in++;
|
|
}
|
|
}
|
|
PlaYUVerFramePelFormat rgb8 =
|
|
{
|
|
"RGB8",
|
|
PlaYUVerFrame::COLOR_RGB,
|
|
3,
|
|
1,
|
|
1,
|
|
getPixelValueRGB8,
|
|
frameFromBufferRGB8,
|
|
bufferFromFrame_yuv420p,
|
|
fillRGBbuffer_yuv420p,
|
|
ADD_FFMPEG_PEL_FMT( AV_PIX_FMT_RGB24 ),
|
|
};
|
|
|
|
|
|
extern PlaYUVerFramePelFormat g_PlaYUVerFramePelFormatsList[PLAYUVER_NUMBER_FORMATS] =
|
|
{
|
|
yuv420p,
|
|
yuv444p,
|
|
yuv422p,
|
|
yuv422,
|
|
gray,
|
|
rgb8, };
|
|
|
|
} // NAMESPACE
|