mirror of
https://github.com/mpv-player/mpv
synced 2024-12-24 07:42:17 +00:00
c082240c62
Based on a patch by qyot27. Add the missing parts in mp_get_chroma_shift(), which allow allocation of such images, and which make vo_opengl automatically accept the new formats. Change the IMGFMT_IS_YUVP16_LE/BE macros to properly report IMGFMT_444P14 as supported: this pixel format has the highest numerical bit width identifier (0x55), which is not covered by the mask ~0xfc. Remove 1 bit from the mask (makes it 0xf8) so that IMGFMT_IS_YUVP16(IMGFMT_444P14) is 1. This is slightly risky, as the organization of the image format IDs (actually FourCCs + mplayer internal IDs) is messy at best, but it should be ok.
285 lines
8.6 KiB
C
285 lines
8.6 KiB
C
/*
|
|
* This file is part of MPlayer.
|
|
*
|
|
* MPlayer 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.
|
|
*
|
|
* MPlayer 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 MPlayer; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "talloc.h"
|
|
|
|
#include "video/img_format.h"
|
|
#include "video/mp_image.h"
|
|
#include "video/sws_utils.h"
|
|
|
|
#include "video/memcpy_pic.h"
|
|
#include "libavutil/mem.h"
|
|
#include "libavutil/common.h"
|
|
|
|
void mp_image_alloc_planes(mp_image_t *mpi) {
|
|
// IF09 - allocate space for 4. plane delta info - unused
|
|
if (mpi->imgfmt == IMGFMT_IF09) {
|
|
mpi->planes[0]=av_malloc(mpi->bpp*mpi->width*(mpi->height+2)/8+
|
|
mpi->chroma_width*mpi->chroma_height);
|
|
} else
|
|
mpi->planes[0]=av_malloc(mpi->bpp*mpi->width*(mpi->height+2)/8);
|
|
if (!mpi->planes[0])
|
|
abort(); //out of memory
|
|
if (mpi->flags&MP_IMGFLAG_PLANAR) {
|
|
// FIXME this code only supports same bpp for all planes, and bpp divisible
|
|
// by 8. Currently the case for all planar formats.
|
|
int bpp = MP_IMAGE_PLANAR_BITS_PER_PIXEL_ON_PLANE(mpi, 0) / 8;
|
|
// YV12/I420/YVU9/IF09. feel free to add other planar formats here...
|
|
mpi->stride[0]=mpi->stride[3]=bpp*mpi->width;
|
|
if(mpi->num_planes > 2){
|
|
mpi->stride[1]=mpi->stride[2]=bpp*mpi->chroma_width;
|
|
if(mpi->flags&MP_IMGFLAG_SWAPPED){
|
|
// I420/IYUV (Y,U,V)
|
|
mpi->planes[1]=mpi->planes[0]+mpi->stride[0]*mpi->height;
|
|
mpi->planes[2]=mpi->planes[1]+mpi->stride[1]*mpi->chroma_height;
|
|
if (mpi->num_planes > 3)
|
|
mpi->planes[3]=mpi->planes[2]+mpi->stride[2]*mpi->chroma_height;
|
|
} else {
|
|
// YV12,YVU9,IF09 (Y,V,U)
|
|
mpi->planes[2]=mpi->planes[0]+mpi->stride[0]*mpi->height;
|
|
mpi->planes[1]=mpi->planes[2]+mpi->stride[1]*mpi->chroma_height;
|
|
if (mpi->num_planes > 3)
|
|
mpi->planes[3]=mpi->planes[1]+mpi->stride[1]*mpi->chroma_height;
|
|
}
|
|
} else {
|
|
// NV12/NV21
|
|
mpi->stride[1]=mpi->chroma_width;
|
|
mpi->planes[1]=mpi->planes[0]+mpi->stride[0]*mpi->height;
|
|
}
|
|
} else {
|
|
mpi->stride[0]=mpi->width*mpi->bpp/8;
|
|
if (mpi->flags & MP_IMGFLAG_RGB_PALETTE)
|
|
mpi->planes[1] = av_malloc(1024);
|
|
}
|
|
mpi->flags|=MP_IMGFLAG_ALLOCATED;
|
|
}
|
|
|
|
mp_image_t* alloc_mpi(int w, int h, unsigned long int fmt) {
|
|
mp_image_t* mpi = new_mp_image(w,h);
|
|
|
|
mpi->width=FFALIGN(w, MP_STRIDE_ALIGNMENT);
|
|
mp_image_setfmt(mpi,fmt);
|
|
mp_image_alloc_planes(mpi);
|
|
mpi->width=w;
|
|
mp_image_setfmt(mpi,fmt); // reset chroma size
|
|
|
|
return mpi;
|
|
}
|
|
|
|
void copy_mpi(mp_image_t *dmpi, mp_image_t *mpi) {
|
|
if(mpi->flags&MP_IMGFLAG_PLANAR){
|
|
memcpy_pic(dmpi->planes[0],mpi->planes[0], MP_IMAGE_BYTES_PER_ROW_ON_PLANE(mpi, 0), mpi->h,
|
|
dmpi->stride[0],mpi->stride[0]);
|
|
memcpy_pic(dmpi->planes[1],mpi->planes[1], MP_IMAGE_BYTES_PER_ROW_ON_PLANE(mpi, 1), mpi->chroma_height,
|
|
dmpi->stride[1],mpi->stride[1]);
|
|
memcpy_pic(dmpi->planes[2], mpi->planes[2], MP_IMAGE_BYTES_PER_ROW_ON_PLANE(mpi, 2), mpi->chroma_height,
|
|
dmpi->stride[2],mpi->stride[2]);
|
|
} else {
|
|
memcpy_pic(dmpi->planes[0],mpi->planes[0],
|
|
MP_IMAGE_BYTES_PER_ROW_ON_PLANE(mpi, 0), mpi->h,
|
|
dmpi->stride[0],mpi->stride[0]);
|
|
}
|
|
}
|
|
|
|
void mp_image_setfmt(mp_image_t* mpi,unsigned int out_fmt){
|
|
mpi->flags&=~(MP_IMGFLAG_PLANAR|MP_IMGFLAG_YUV|MP_IMGFLAG_SWAPPED);
|
|
mpi->imgfmt=out_fmt;
|
|
// compressed formats
|
|
if(IMGFMT_IS_HWACCEL(out_fmt)){
|
|
mpi->bpp=0;
|
|
return;
|
|
}
|
|
mpi->num_planes=1;
|
|
if (IMGFMT_IS_RGB(out_fmt)) {
|
|
if (IMGFMT_RGB_DEPTH(out_fmt) < 8 && !(out_fmt&128))
|
|
mpi->bpp = IMGFMT_RGB_DEPTH(out_fmt);
|
|
else
|
|
mpi->bpp=(IMGFMT_RGB_DEPTH(out_fmt)+7)&(~7);
|
|
return;
|
|
}
|
|
if (IMGFMT_IS_BGR(out_fmt)) {
|
|
if (IMGFMT_BGR_DEPTH(out_fmt) < 8 && !(out_fmt&128))
|
|
mpi->bpp = IMGFMT_BGR_DEPTH(out_fmt);
|
|
else
|
|
mpi->bpp=(IMGFMT_BGR_DEPTH(out_fmt)+7)&(~7);
|
|
mpi->flags|=MP_IMGFLAG_SWAPPED;
|
|
return;
|
|
}
|
|
switch (out_fmt) {
|
|
case IMGFMT_BGR0:
|
|
mpi->bpp = 32;
|
|
return;
|
|
}
|
|
mpi->num_planes=3;
|
|
if (out_fmt == IMGFMT_GBRP) {
|
|
mpi->bpp=24;
|
|
mpi->flags|=MP_IMGFLAG_PLANAR;
|
|
mpi->chroma_x_shift = 0;
|
|
mpi->chroma_y_shift = 0;
|
|
mpi->chroma_width=mpi->width;
|
|
mpi->chroma_height=mpi->height;
|
|
return;
|
|
}
|
|
mpi->flags|=MP_IMGFLAG_YUV;
|
|
if (mp_get_chroma_shift(out_fmt, NULL, NULL, NULL)) {
|
|
mpi->flags|=MP_IMGFLAG_PLANAR;
|
|
mpi->bpp = mp_get_chroma_shift(out_fmt, &mpi->chroma_x_shift, &mpi->chroma_y_shift, NULL);
|
|
mpi->chroma_width = mpi->width >> mpi->chroma_x_shift;
|
|
mpi->chroma_height = mpi->height >> mpi->chroma_y_shift;
|
|
}
|
|
switch(out_fmt){
|
|
case IMGFMT_I420:
|
|
case IMGFMT_IYUV:
|
|
mpi->flags|=MP_IMGFLAG_SWAPPED;
|
|
case IMGFMT_YV12:
|
|
return;
|
|
case IMGFMT_420A:
|
|
case IMGFMT_IF09:
|
|
mpi->num_planes=4;
|
|
case IMGFMT_YVU9:
|
|
case IMGFMT_444P:
|
|
case IMGFMT_422P:
|
|
case IMGFMT_411P:
|
|
case IMGFMT_440P:
|
|
case IMGFMT_444P16_LE:
|
|
case IMGFMT_444P16_BE:
|
|
case IMGFMT_444P14_LE:
|
|
case IMGFMT_444P14_BE:
|
|
case IMGFMT_444P12_LE:
|
|
case IMGFMT_444P12_BE:
|
|
case IMGFMT_444P10_LE:
|
|
case IMGFMT_444P10_BE:
|
|
case IMGFMT_444P9_LE:
|
|
case IMGFMT_444P9_BE:
|
|
case IMGFMT_422P16_LE:
|
|
case IMGFMT_422P16_BE:
|
|
case IMGFMT_422P14_LE:
|
|
case IMGFMT_422P14_BE:
|
|
case IMGFMT_422P12_LE:
|
|
case IMGFMT_422P12_BE:
|
|
case IMGFMT_422P10_LE:
|
|
case IMGFMT_422P10_BE:
|
|
case IMGFMT_422P9_LE:
|
|
case IMGFMT_422P9_BE:
|
|
case IMGFMT_420P16_LE:
|
|
case IMGFMT_420P16_BE:
|
|
case IMGFMT_420P14_LE:
|
|
case IMGFMT_420P14_BE:
|
|
case IMGFMT_420P12_LE:
|
|
case IMGFMT_420P12_BE:
|
|
case IMGFMT_420P10_LE:
|
|
case IMGFMT_420P10_BE:
|
|
case IMGFMT_420P9_LE:
|
|
case IMGFMT_420P9_BE:
|
|
return;
|
|
case IMGFMT_Y800:
|
|
case IMGFMT_Y8:
|
|
case IMGFMT_Y16LE:
|
|
case IMGFMT_Y16BE:
|
|
/* they're planar ones, but for easier handling use them as packed */
|
|
mpi->flags&=~MP_IMGFLAG_PLANAR;
|
|
mpi->num_planes=1;
|
|
return;
|
|
case IMGFMT_UYVY:
|
|
mpi->flags|=MP_IMGFLAG_SWAPPED;
|
|
case IMGFMT_YUY2:
|
|
mpi->chroma_x_shift = 1;
|
|
mpi->chroma_y_shift = 1;
|
|
mpi->chroma_width=(mpi->width>>1);
|
|
mpi->chroma_height=(mpi->height>>1);
|
|
mpi->bpp=16;
|
|
mpi->num_planes=1;
|
|
return;
|
|
case IMGFMT_NV12:
|
|
mpi->flags|=MP_IMGFLAG_SWAPPED;
|
|
case IMGFMT_NV21:
|
|
mpi->flags|=MP_IMGFLAG_PLANAR;
|
|
mpi->bpp=12;
|
|
mpi->num_planes=2;
|
|
mpi->chroma_width=(mpi->width>>0);
|
|
mpi->chroma_height=(mpi->height>>1);
|
|
mpi->chroma_x_shift=0;
|
|
mpi->chroma_y_shift=1;
|
|
return;
|
|
}
|
|
mp_msg(MSGT_DECVIDEO,MSGL_WARN,"mp_image: unknown out_fmt: 0x%X\n",out_fmt);
|
|
mpi->bpp=0;
|
|
}
|
|
|
|
static int mp_image_destructor(void *ptr)
|
|
{
|
|
mp_image_t *mpi = ptr;
|
|
|
|
if(mpi->flags&MP_IMGFLAG_ALLOCATED){
|
|
/* because we allocate the whole image at once */
|
|
av_free(mpi->planes[0]);
|
|
if (mpi->flags & MP_IMGFLAG_RGB_PALETTE)
|
|
av_free(mpi->planes[1]);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
mp_image_t* new_mp_image(int w,int h){
|
|
mp_image_t* mpi = talloc_zero(NULL, mp_image_t);
|
|
talloc_set_destructor(mpi, mp_image_destructor);
|
|
mpi->width=mpi->w=w;
|
|
mpi->height=mpi->h=h;
|
|
return mpi;
|
|
}
|
|
|
|
void free_mp_image(mp_image_t* mpi){
|
|
talloc_free(mpi);
|
|
}
|
|
|
|
enum mp_csp mp_image_csp(struct mp_image *img)
|
|
{
|
|
if (img->colorspace != MP_CSP_AUTO)
|
|
return img->colorspace;
|
|
return (img->flags & MP_IMGFLAG_YUV) ? MP_CSP_BT_601 : MP_CSP_RGB;
|
|
}
|
|
|
|
enum mp_csp_levels mp_image_levels(struct mp_image *img)
|
|
{
|
|
if (img->levels != MP_CSP_LEVELS_AUTO)
|
|
return img->levels;
|
|
return (img->flags & MP_IMGFLAG_YUV) ? MP_CSP_LEVELS_TV : MP_CSP_LEVELS_PC;
|
|
}
|
|
|
|
void mp_image_set_colorspace_details(struct mp_image *image,
|
|
struct mp_csp_details *csp)
|
|
{
|
|
if (image->flags & MP_IMGFLAG_YUV) {
|
|
image->colorspace = csp->format;
|
|
if (image->colorspace == MP_CSP_AUTO)
|
|
image->colorspace = MP_CSP_BT_601;
|
|
image->levels = csp->levels_in;
|
|
if (image->levels == MP_CSP_LEVELS_AUTO)
|
|
image->levels = MP_CSP_LEVELS_TV;
|
|
} else {
|
|
image->colorspace = MP_CSP_RGB;
|
|
image->levels = MP_CSP_LEVELS_PC;
|
|
}
|
|
}
|