123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899 |
- /*
- * Copyright (c) 2011 Michael Niedermayer
- *
- * This file is part of FFmpeg.
- *
- * FFmpeg 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.
- *
- * FFmpeg 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 FFmpeg; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
- *
- * Parts of this file have been stolen from mplayer
- */
- /**
- * @file
- */
- #include "avfilter.h"
- #include "libavutil/avassert.h"
- #include "libavutil/pixdesc.h"
- #include "libavutil/intreadwrite.h"
- #include "libavutil/imgutils.h"
- #include "libmpcodecs/vf.h"
- #include "libmpcodecs/img_format.h"
- #include "libmpcodecs/cpudetect.h"
- #include "libmpcodecs/vd_ffmpeg.h"
- #include "libmpcodecs/vf_scale.h"
- #include "libmpcodecs/libvo/fastmemcpy.h"
- #include "libswscale/swscale.h"
- //FIXME maybe link the orig in
- //XXX: identical pix_fmt must be following with each others
- static const struct {
- int fmt;
- enum PixelFormat pix_fmt;
- } conversion_map[] = {
- {IMGFMT_ARGB, PIX_FMT_ARGB},
- {IMGFMT_BGRA, PIX_FMT_BGRA},
- {IMGFMT_BGR24, PIX_FMT_BGR24},
- {IMGFMT_BGR16BE, PIX_FMT_RGB565BE},
- {IMGFMT_BGR16LE, PIX_FMT_RGB565LE},
- {IMGFMT_BGR15BE, PIX_FMT_RGB555BE},
- {IMGFMT_BGR15LE, PIX_FMT_RGB555LE},
- {IMGFMT_BGR12BE, PIX_FMT_RGB444BE},
- {IMGFMT_BGR12LE, PIX_FMT_RGB444LE},
- {IMGFMT_BGR8, PIX_FMT_RGB8},
- {IMGFMT_BGR4, PIX_FMT_RGB4},
- {IMGFMT_BGR1, PIX_FMT_MONOBLACK},
- {IMGFMT_RGB1, PIX_FMT_MONOBLACK},
- {IMGFMT_RG4B, PIX_FMT_BGR4_BYTE},
- {IMGFMT_BG4B, PIX_FMT_RGB4_BYTE},
- {IMGFMT_RGB48LE, PIX_FMT_RGB48LE},
- {IMGFMT_RGB48BE, PIX_FMT_RGB48BE},
- {IMGFMT_ABGR, PIX_FMT_ABGR},
- {IMGFMT_RGBA, PIX_FMT_RGBA},
- {IMGFMT_RGB24, PIX_FMT_RGB24},
- {IMGFMT_RGB16BE, PIX_FMT_BGR565BE},
- {IMGFMT_RGB16LE, PIX_FMT_BGR565LE},
- {IMGFMT_RGB15BE, PIX_FMT_BGR555BE},
- {IMGFMT_RGB15LE, PIX_FMT_BGR555LE},
- {IMGFMT_RGB12BE, PIX_FMT_BGR444BE},
- {IMGFMT_RGB12LE, PIX_FMT_BGR444LE},
- {IMGFMT_RGB8, PIX_FMT_BGR8},
- {IMGFMT_RGB4, PIX_FMT_BGR4},
- {IMGFMT_BGR8, PIX_FMT_PAL8},
- {IMGFMT_YUY2, PIX_FMT_YUYV422},
- {IMGFMT_UYVY, PIX_FMT_UYVY422},
- {IMGFMT_NV12, PIX_FMT_NV12},
- {IMGFMT_NV21, PIX_FMT_NV21},
- {IMGFMT_Y800, PIX_FMT_GRAY8},
- {IMGFMT_Y8, PIX_FMT_GRAY8},
- {IMGFMT_YVU9, PIX_FMT_YUV410P},
- {IMGFMT_IF09, PIX_FMT_YUV410P},
- {IMGFMT_YV12, PIX_FMT_YUV420P},
- {IMGFMT_I420, PIX_FMT_YUV420P},
- {IMGFMT_IYUV, PIX_FMT_YUV420P},
- {IMGFMT_411P, PIX_FMT_YUV411P},
- {IMGFMT_422P, PIX_FMT_YUV422P},
- {IMGFMT_444P, PIX_FMT_YUV444P},
- {IMGFMT_440P, PIX_FMT_YUV440P},
- {IMGFMT_420A, PIX_FMT_YUVA420P},
- {IMGFMT_420P16_LE, PIX_FMT_YUV420P16LE},
- {IMGFMT_420P16_BE, PIX_FMT_YUV420P16BE},
- {IMGFMT_422P16_LE, PIX_FMT_YUV422P16LE},
- {IMGFMT_422P16_BE, PIX_FMT_YUV422P16BE},
- {IMGFMT_444P16_LE, PIX_FMT_YUV444P16LE},
- {IMGFMT_444P16_BE, PIX_FMT_YUV444P16BE},
- // YUVJ are YUV formats that use the full Y range and not just
- // 16 - 235 (see colorspaces.txt).
- // Currently they are all treated the same way.
- {IMGFMT_YV12, PIX_FMT_YUVJ420P},
- {IMGFMT_422P, PIX_FMT_YUVJ422P},
- {IMGFMT_444P, PIX_FMT_YUVJ444P},
- {IMGFMT_440P, PIX_FMT_YUVJ440P},
- {IMGFMT_XVMC_MOCO_MPEG2, PIX_FMT_XVMC_MPEG2_MC},
- {IMGFMT_XVMC_IDCT_MPEG2, PIX_FMT_XVMC_MPEG2_IDCT},
- {IMGFMT_VDPAU_MPEG1, PIX_FMT_VDPAU_MPEG1},
- {IMGFMT_VDPAU_MPEG2, PIX_FMT_VDPAU_MPEG2},
- {IMGFMT_VDPAU_H264, PIX_FMT_VDPAU_H264},
- {IMGFMT_VDPAU_WMV3, PIX_FMT_VDPAU_WMV3},
- {IMGFMT_VDPAU_VC1, PIX_FMT_VDPAU_VC1},
- {IMGFMT_VDPAU_MPEG4, PIX_FMT_VDPAU_MPEG4},
- {0, PIX_FMT_NONE}
- };
- //copied from vf.c
- extern const vf_info_t vf_info_1bpp;
- extern const vf_info_t vf_info_2xsai;
- extern const vf_info_t vf_info_ass;
- extern const vf_info_t vf_info_bmovl;
- extern const vf_info_t vf_info_crop;
- extern const vf_info_t vf_info_decimate;
- extern const vf_info_t vf_info_denoise3d;
- extern const vf_info_t vf_info_detc;
- extern const vf_info_t vf_info_dint;
- extern const vf_info_t vf_info_divtc;
- extern const vf_info_t vf_info_down3dright;
- extern const vf_info_t vf_info_dsize;
- extern const vf_info_t vf_info_dvbscale;
- extern const vf_info_t vf_info_eq2;
- extern const vf_info_t vf_info_eq;
- extern const vf_info_t vf_info_expand;
- extern const vf_info_t vf_info_field;
- extern const vf_info_t vf_info_fil;
- extern const vf_info_t vf_info_filmdint;
- extern const vf_info_t vf_info_fixpts;
- extern const vf_info_t vf_info_flip;
- extern const vf_info_t vf_info_format;
- extern const vf_info_t vf_info_framestep;
- extern const vf_info_t vf_info_fspp;
- extern const vf_info_t vf_info_geq;
- extern const vf_info_t vf_info_halfpack;
- extern const vf_info_t vf_info_harddup;
- extern const vf_info_t vf_info_hqdn3d;
- extern const vf_info_t vf_info_hue;
- extern const vf_info_t vf_info_il;
- extern const vf_info_t vf_info_ilpack;
- extern const vf_info_t vf_info_ivtc;
- extern const vf_info_t vf_info_kerndeint;
- extern const vf_info_t vf_info_lavc;
- extern const vf_info_t vf_info_lavcdeint;
- extern const vf_info_t vf_info_mcdeint;
- extern const vf_info_t vf_info_mirror;
- extern const vf_info_t vf_info_noformat;
- extern const vf_info_t vf_info_noise;
- extern const vf_info_t vf_info_ow;
- extern const vf_info_t vf_info_palette;
- extern const vf_info_t vf_info_perspective;
- extern const vf_info_t vf_info_phase;
- extern const vf_info_t vf_info_pp7;
- extern const vf_info_t vf_info_pp;
- extern const vf_info_t vf_info_pullup;
- extern const vf_info_t vf_info_qp;
- extern const vf_info_t vf_info_rectangle;
- extern const vf_info_t vf_info_remove_logo;
- extern const vf_info_t vf_info_rotate;
- extern const vf_info_t vf_info_sab;
- extern const vf_info_t vf_info_scale;
- extern const vf_info_t vf_info_screenshot;
- extern const vf_info_t vf_info_smartblur;
- extern const vf_info_t vf_info_softpulldown;
- extern const vf_info_t vf_info_softskip;
- extern const vf_info_t vf_info_spp;
- extern const vf_info_t vf_info_stereo3d;
- extern const vf_info_t vf_info_swapuv;
- extern const vf_info_t vf_info_telecine;
- extern const vf_info_t vf_info_test;
- extern const vf_info_t vf_info_tfields;
- extern const vf_info_t vf_info_tile;
- extern const vf_info_t vf_info_tinterlace;
- extern const vf_info_t vf_info_unsharp;
- extern const vf_info_t vf_info_uspp;
- extern const vf_info_t vf_info_vo;
- extern const vf_info_t vf_info_yadif;
- extern const vf_info_t vf_info_yuvcsp;
- extern const vf_info_t vf_info_yvu9;
- extern const vf_info_t vf_info_zrmjpeg;
- static const vf_info_t* const filters[]={
- &vf_info_2xsai,
- &vf_info_decimate,
- &vf_info_denoise3d,
- &vf_info_detc,
- &vf_info_dint,
- &vf_info_divtc,
- &vf_info_down3dright,
- &vf_info_dsize,
- &vf_info_eq2,
- &vf_info_eq,
- &vf_info_field,
- &vf_info_fil,
- // &vf_info_filmdint, cmmx.h vd.h ‘opt_screen_size_x’
- &vf_info_fixpts,
- &vf_info_framestep,
- &vf_info_fspp,
- &vf_info_geq,
- &vf_info_harddup,
- &vf_info_hqdn3d,
- &vf_info_hue,
- &vf_info_il,
- &vf_info_ilpack,
- &vf_info_ivtc,
- &vf_info_kerndeint,
- &vf_info_mcdeint,
- &vf_info_mirror,
- &vf_info_noise,
- &vf_info_ow,
- &vf_info_palette,
- &vf_info_perspective,
- &vf_info_phase,
- &vf_info_pp7,
- &vf_info_pullup,
- &vf_info_qp,
- &vf_info_rectangle,
- &vf_info_remove_logo,
- &vf_info_rotate,
- &vf_info_sab,
- &vf_info_screenshot,
- &vf_info_smartblur,
- &vf_info_softpulldown,
- &vf_info_softskip,
- &vf_info_spp,
- &vf_info_stereo3d,
- &vf_info_swapuv,
- &vf_info_telecine,
- &vf_info_tile,
- &vf_info_tinterlace,
- &vf_info_unsharp,
- &vf_info_uspp,
- &vf_info_yuvcsp,
- &vf_info_yvu9,
- NULL
- };
- /*
- Unsupported filters
- 1bpp
- ass
- bmovl
- crop
- dvbscale
- flip
- expand
- format
- halfpack
- lavc
- lavcdeint
- noformat
- pp
- scale
- tfields
- vo
- yadif
- zrmjpeg
- */
- CpuCaps gCpuCaps; //FIXME initialize this so optims work
- static void sws_getFlagsAndFilterFromCmdLine(int *flags, SwsFilter **srcFilterParam, SwsFilter **dstFilterParam)
- {
- static int firstTime=1;
- *flags=0;
- #if ARCH_X86
- if(gCpuCaps.hasMMX)
- __asm__ volatile("emms\n\t"::: "memory"); //FIXME this should not be required but it IS (even for non-MMX versions)
- #endif
- if(firstTime)
- {
- firstTime=0;
- *flags= SWS_PRINT_INFO;
- }
- else if( mp_msg_test(MSGT_VFILTER,MSGL_DBG2) ) *flags= SWS_PRINT_INFO;
- switch(SWS_BILINEAR)
- {
- case 0: *flags|= SWS_FAST_BILINEAR; break;
- case 1: *flags|= SWS_BILINEAR; break;
- case 2: *flags|= SWS_BICUBIC; break;
- case 3: *flags|= SWS_X; break;
- case 4: *flags|= SWS_POINT; break;
- case 5: *flags|= SWS_AREA; break;
- case 6: *flags|= SWS_BICUBLIN; break;
- case 7: *flags|= SWS_GAUSS; break;
- case 8: *flags|= SWS_SINC; break;
- case 9: *flags|= SWS_LANCZOS; break;
- case 10:*flags|= SWS_SPLINE; break;
- default:*flags|= SWS_BILINEAR; break;
- }
- *srcFilterParam= NULL;
- *dstFilterParam= NULL;
- }
- //exact copy from vf_scale.c
- // will use sws_flags & src_filter (from cmd line)
- struct SwsContext *sws_getContextFromCmdLine(int srcW, int srcH, int srcFormat, int dstW, int dstH, int dstFormat)
- {
- int flags, i;
- SwsFilter *dstFilterParam, *srcFilterParam;
- enum PixelFormat dfmt, sfmt;
- for(i=0; conversion_map[i].fmt && dstFormat != conversion_map[i].fmt; i++);
- dfmt= conversion_map[i].pix_fmt;
- for(i=0; conversion_map[i].fmt && srcFormat != conversion_map[i].fmt; i++);
- sfmt= conversion_map[i].pix_fmt;
- if (srcFormat == IMGFMT_RGB8 || srcFormat == IMGFMT_BGR8) sfmt = PIX_FMT_PAL8;
- sws_getFlagsAndFilterFromCmdLine(&flags, &srcFilterParam, &dstFilterParam);
- return sws_getContext(srcW, srcH, sfmt, dstW, dstH, dfmt, flags , srcFilterParam, dstFilterParam, NULL);
- }
- typedef struct {
- vf_instance_t vf;
- vf_instance_t next_vf;
- AVFilterContext *avfctx;
- int frame_returned;
- } MPContext;
- void mp_msg(int mod, int lev, const char *format, ... ){
- va_list va;
- va_start(va, format);
- //FIXME convert lev/mod
- av_vlog(NULL, AV_LOG_DEBUG, format, va);
- va_end(va);
- }
- int mp_msg_test(int mod, int lev){
- return 123;
- }
- void init_avcodec(void)
- {
- //we maybe should init but its kinda 1. unneeded 2. a bit inpolite from here
- }
- //Exact copy of vf.c
- void vf_clone_mpi_attributes(mp_image_t* dst, mp_image_t* src){
- dst->pict_type= src->pict_type;
- dst->fields = src->fields;
- dst->qscale_type= src->qscale_type;
- if(dst->width == src->width && dst->height == src->height){
- dst->qstride= src->qstride;
- dst->qscale= src->qscale;
- }
- }
- //Exact copy of vf.c
- void vf_next_draw_slice(struct vf_instance *vf,unsigned char** src, int * stride,int w, int h, int x, int y){
- if (vf->next->draw_slice) {
- vf->next->draw_slice(vf->next,src,stride,w,h,x,y);
- return;
- }
- if (!vf->dmpi) {
- mp_msg(MSGT_VFILTER,MSGL_ERR,"draw_slice: dmpi not stored by vf_%s\n", vf->info->name);
- return;
- }
- if (!(vf->dmpi->flags & MP_IMGFLAG_PLANAR)) {
- memcpy_pic(vf->dmpi->planes[0]+y*vf->dmpi->stride[0]+vf->dmpi->bpp/8*x,
- src[0], vf->dmpi->bpp/8*w, h, vf->dmpi->stride[0], stride[0]);
- return;
- }
- memcpy_pic(vf->dmpi->planes[0]+y*vf->dmpi->stride[0]+x, src[0],
- w, h, vf->dmpi->stride[0], stride[0]);
- memcpy_pic(vf->dmpi->planes[1]+(y>>vf->dmpi->chroma_y_shift)*vf->dmpi->stride[1]+(x>>vf->dmpi->chroma_x_shift),
- src[1], w>>vf->dmpi->chroma_x_shift, h>>vf->dmpi->chroma_y_shift, vf->dmpi->stride[1], stride[1]);
- memcpy_pic(vf->dmpi->planes[2]+(y>>vf->dmpi->chroma_y_shift)*vf->dmpi->stride[2]+(x>>vf->dmpi->chroma_x_shift),
- src[2], w>>vf->dmpi->chroma_x_shift, h>>vf->dmpi->chroma_y_shift, vf->dmpi->stride[2], stride[2]);
- }
- //Exact copy of vf.c
- void vf_mpi_clear(mp_image_t* mpi,int x0,int y0,int w,int h){
- int y;
- if(mpi->flags&MP_IMGFLAG_PLANAR){
- y0&=~1;h+=h&1;
- if(x0==0 && w==mpi->width){
- // full width clear:
- memset(mpi->planes[0]+mpi->stride[0]*y0,0,mpi->stride[0]*h);
- memset(mpi->planes[1]+mpi->stride[1]*(y0>>mpi->chroma_y_shift),128,mpi->stride[1]*(h>>mpi->chroma_y_shift));
- memset(mpi->planes[2]+mpi->stride[2]*(y0>>mpi->chroma_y_shift),128,mpi->stride[2]*(h>>mpi->chroma_y_shift));
- } else
- for(y=y0;y<y0+h;y+=2){
- memset(mpi->planes[0]+x0+mpi->stride[0]*y,0,w);
- memset(mpi->planes[0]+x0+mpi->stride[0]*(y+1),0,w);
- memset(mpi->planes[1]+(x0>>mpi->chroma_x_shift)+mpi->stride[1]*(y>>mpi->chroma_y_shift),128,(w>>mpi->chroma_x_shift));
- memset(mpi->planes[2]+(x0>>mpi->chroma_x_shift)+mpi->stride[2]*(y>>mpi->chroma_y_shift),128,(w>>mpi->chroma_x_shift));
- }
- return;
- }
- // packed:
- for(y=y0;y<y0+h;y++){
- unsigned char* dst=mpi->planes[0]+mpi->stride[0]*y+(mpi->bpp>>3)*x0;
- if(mpi->flags&MP_IMGFLAG_YUV){
- unsigned int* p=(unsigned int*) dst;
- int size=(mpi->bpp>>3)*w/4;
- int i;
- #if HAVE_BIGENDIAN
- #define CLEAR_PACKEDYUV_PATTERN 0x00800080
- #define CLEAR_PACKEDYUV_PATTERN_SWAPPED 0x80008000
- #else
- #define CLEAR_PACKEDYUV_PATTERN 0x80008000
- #define CLEAR_PACKEDYUV_PATTERN_SWAPPED 0x00800080
- #endif
- if(mpi->flags&MP_IMGFLAG_SWAPPED){
- for(i=0;i<size-3;i+=4) p[i]=p[i+1]=p[i+2]=p[i+3]=CLEAR_PACKEDYUV_PATTERN_SWAPPED;
- for(;i<size;i++) p[i]=CLEAR_PACKEDYUV_PATTERN_SWAPPED;
- } else {
- for(i=0;i<size-3;i+=4) p[i]=p[i+1]=p[i+2]=p[i+3]=CLEAR_PACKEDYUV_PATTERN;
- for(;i<size;i++) p[i]=CLEAR_PACKEDYUV_PATTERN;
- }
- } else
- memset(dst,0,(mpi->bpp>>3)*w);
- }
- }
- int vf_next_query_format(struct vf_instance *vf, unsigned int fmt){
- return 1;
- }
- //used by delogo
- unsigned int vf_match_csp(vf_instance_t** vfp,const unsigned int* list,unsigned int preferred){
- return preferred;
- }
- mp_image_t* vf_get_image(vf_instance_t* vf, unsigned int outfmt, int mp_imgtype, int mp_imgflag, int w, int h){
- MPContext *m= (MPContext*)(((uint8_t*)vf) - offsetof(MPContext, next_vf));
- mp_image_t* mpi=NULL;
- int w2;
- int number = mp_imgtype >> 16;
- av_assert0(vf->next == NULL); // all existing filters call this just on next
- //vf_dint needs these as it calls vf_get_image() before configuring the output
- if(vf->w==0 && w>0) vf->w=w;
- if(vf->h==0 && h>0) vf->h=h;
- av_assert0(w == -1 || w >= vf->w);
- av_assert0(h == -1 || h >= vf->h);
- av_assert0(vf->w > 0);
- av_assert0(vf->h > 0);
- av_log(m->avfctx, AV_LOG_DEBUG, "get_image: %d:%d, vf: %d:%d\n", w,h,vf->w,vf->h);
- if (w == -1) w = vf->w;
- if (h == -1) h = vf->h;
- w2=(mp_imgflag&MP_IMGFLAG_ACCEPT_ALIGNED_STRIDE)?((w+15)&(~15)):w;
- // Note: we should call libvo first to check if it supports direct rendering
- // and if not, then fallback to software buffers:
- switch(mp_imgtype & 0xff){
- case MP_IMGTYPE_EXPORT:
- if(!vf->imgctx.export_images[0]) vf->imgctx.export_images[0]=new_mp_image(w2,h);
- mpi=vf->imgctx.export_images[0];
- break;
- case MP_IMGTYPE_STATIC:
- if(!vf->imgctx.static_images[0]) vf->imgctx.static_images[0]=new_mp_image(w2,h);
- mpi=vf->imgctx.static_images[0];
- break;
- case MP_IMGTYPE_TEMP:
- if(!vf->imgctx.temp_images[0]) vf->imgctx.temp_images[0]=new_mp_image(w2,h);
- mpi=vf->imgctx.temp_images[0];
- break;
- case MP_IMGTYPE_IPB:
- if(!(mp_imgflag&MP_IMGFLAG_READABLE)){ // B frame:
- if(!vf->imgctx.temp_images[0]) vf->imgctx.temp_images[0]=new_mp_image(w2,h);
- mpi=vf->imgctx.temp_images[0];
- break;
- }
- case MP_IMGTYPE_IP:
- if(!vf->imgctx.static_images[vf->imgctx.static_idx]) vf->imgctx.static_images[vf->imgctx.static_idx]=new_mp_image(w2,h);
- mpi=vf->imgctx.static_images[vf->imgctx.static_idx];
- vf->imgctx.static_idx^=1;
- break;
- case MP_IMGTYPE_NUMBERED:
- if (number == -1) {
- int i;
- for (i = 0; i < NUM_NUMBERED_MPI; i++)
- if (!vf->imgctx.numbered_images[i] || !vf->imgctx.numbered_images[i]->usage_count)
- break;
- number = i;
- }
- if (number < 0 || number >= NUM_NUMBERED_MPI) return NULL;
- if (!vf->imgctx.numbered_images[number]) vf->imgctx.numbered_images[number] = new_mp_image(w2,h);
- mpi = vf->imgctx.numbered_images[number];
- mpi->number = number;
- break;
- }
- if(mpi){
- mpi->type=mp_imgtype;
- mpi->w=vf->w; mpi->h=vf->h;
- // keep buffer allocation status & color flags only:
- // mpi->flags&=~(MP_IMGFLAG_PRESERVE|MP_IMGFLAG_READABLE|MP_IMGFLAG_DIRECT);
- mpi->flags&=MP_IMGFLAG_ALLOCATED|MP_IMGFLAG_TYPE_DISPLAYED|MP_IMGFLAGMASK_COLORS;
- // accept restrictions, draw_slice and palette flags only:
- mpi->flags|=mp_imgflag&(MP_IMGFLAGMASK_RESTRICTIONS|MP_IMGFLAG_DRAW_CALLBACK|MP_IMGFLAG_RGB_PALETTE);
- if(!vf->draw_slice) mpi->flags&=~MP_IMGFLAG_DRAW_CALLBACK;
- if(mpi->width!=w2 || mpi->height!=h){
- // printf("vf.c: MPI parameters changed! %dx%d -> %dx%d \n", mpi->width,mpi->height,w2,h);
- if(mpi->flags&MP_IMGFLAG_ALLOCATED){
- if(mpi->width<w2 || mpi->height<h){
- // need to re-allocate buffer memory:
- av_free(mpi->planes[0]);
- mpi->flags&=~MP_IMGFLAG_ALLOCATED;
- mp_msg(MSGT_VFILTER,MSGL_V,"vf.c: have to REALLOCATE buffer memory :(\n");
- }
- // } else {
- } {
- mpi->width=w2; mpi->chroma_width=(w2 + (1<<mpi->chroma_x_shift) - 1)>>mpi->chroma_x_shift;
- mpi->height=h; mpi->chroma_height=(h + (1<<mpi->chroma_y_shift) - 1)>>mpi->chroma_y_shift;
- }
- }
- if(!mpi->bpp) mp_image_setfmt(mpi,outfmt);
- if(!(mpi->flags&MP_IMGFLAG_ALLOCATED) && mpi->type>MP_IMGTYPE_EXPORT){
- av_assert0(!vf->get_image);
- // check libvo first!
- if(vf->get_image) vf->get_image(vf,mpi);
- if(!(mpi->flags&MP_IMGFLAG_DIRECT)){
- // non-direct and not yet allocated image. allocate it!
- if (!mpi->bpp) { // no way we can allocate this
- mp_msg(MSGT_DECVIDEO, MSGL_FATAL,
- "vf_get_image: Tried to allocate a format that can not be allocated!\n");
- return NULL;
- }
- // check if codec prefer aligned stride:
- if(mp_imgflag&MP_IMGFLAG_PREFER_ALIGNED_STRIDE){
- int align=(mpi->flags&MP_IMGFLAG_PLANAR &&
- mpi->flags&MP_IMGFLAG_YUV) ?
- (8<<mpi->chroma_x_shift)-1 : 15; // -- maybe FIXME
- w2=((w+align)&(~align));
- if(mpi->width!=w2){
- #if 0
- // we have to change width... check if we CAN co it:
- int flags=vf->query_format(vf,outfmt); // should not fail
- if(!(flags&3)) mp_msg(MSGT_DECVIDEO,MSGL_WARN,"??? vf_get_image{vf->query_format(outfmt)} failed!\n");
- // printf("query -> 0x%X \n",flags);
- if(flags&VFCAP_ACCEPT_STRIDE){
- #endif
- mpi->width=w2;
- mpi->chroma_width=(w2 + (1<<mpi->chroma_x_shift) - 1)>>mpi->chroma_x_shift;
- // }
- }
- }
- mp_image_alloc_planes(mpi);
- // printf("clearing img!\n");
- vf_mpi_clear(mpi,0,0,mpi->width,mpi->height);
- }
- }
- av_assert0(!vf->start_slice);
- if(mpi->flags&MP_IMGFLAG_DRAW_CALLBACK)
- if(vf->start_slice) vf->start_slice(vf,mpi);
- if(!(mpi->flags&MP_IMGFLAG_TYPE_DISPLAYED)){
- mp_msg(MSGT_DECVIDEO,MSGL_V,"*** [%s] %s%s mp_image_t, %dx%dx%dbpp %s %s, %d bytes\n",
- "NULL"/*vf->info->name*/,
- (mpi->type==MP_IMGTYPE_EXPORT)?"Exporting":
- ((mpi->flags&MP_IMGFLAG_DIRECT)?"Direct Rendering":"Allocating"),
- (mpi->flags&MP_IMGFLAG_DRAW_CALLBACK)?" (slices)":"",
- mpi->width,mpi->height,mpi->bpp,
- (mpi->flags&MP_IMGFLAG_YUV)?"YUV":((mpi->flags&MP_IMGFLAG_SWAPPED)?"BGR":"RGB"),
- (mpi->flags&MP_IMGFLAG_PLANAR)?"planar":"packed",
- mpi->bpp*mpi->width*mpi->height/8);
- mp_msg(MSGT_DECVIDEO,MSGL_DBG2,"(imgfmt: %x, planes: %p,%p,%p strides: %d,%d,%d, chroma: %dx%d, shift: h:%d,v:%d)\n",
- mpi->imgfmt, mpi->planes[0], mpi->planes[1], mpi->planes[2],
- mpi->stride[0], mpi->stride[1], mpi->stride[2],
- mpi->chroma_width, mpi->chroma_height, mpi->chroma_x_shift, mpi->chroma_y_shift);
- mpi->flags|=MP_IMGFLAG_TYPE_DISPLAYED;
- }
- mpi->qscale = NULL;
- }
- mpi->usage_count++;
- // printf("\rVF_MPI: %p %p %p %d %d %d \n",
- // mpi->planes[0],mpi->planes[1],mpi->planes[2],
- // mpi->stride[0],mpi->stride[1],mpi->stride[2]);
- return mpi;
- }
- int vf_next_put_image(struct vf_instance *vf,mp_image_t *mpi, double pts){
- MPContext *m= (void*)vf;
- AVFilterLink *outlink = m->avfctx->outputs[0];
- AVFilterBuffer *pic = av_mallocz(sizeof(AVFilterBuffer));
- AVFilterBufferRef *picref = av_mallocz(sizeof(AVFilterBufferRef));
- int i;
- av_assert0(vf->next);
- av_log(m->avfctx, AV_LOG_DEBUG, "vf_next_put_image\n");
- if (!pic || !picref)
- goto fail;
- picref->buf = pic;
- picref->buf->please_use_av_free= (void*)av_free;
- if (!(picref->video = av_mallocz(sizeof(AVFilterBufferRefVideoProps))))
- goto fail;
- pic->w = picref->video->w = mpi->w;
- pic->h = picref->video->h = mpi->h;
- /* make sure the buffer gets read permission or it's useless for output */
- picref->perms = AV_PERM_READ | AV_PERM_REUSE2;
- // av_assert0(mpi->flags&MP_IMGFLAG_READABLE);
- if(!(mpi->flags&MP_IMGFLAG_PRESERVE))
- picref->perms |= AV_PERM_WRITE;
- pic->refcount = 1;
- picref->type = AVMEDIA_TYPE_VIDEO;
- for(i=0; conversion_map[i].fmt && mpi->imgfmt != conversion_map[i].fmt; i++);
- pic->format = picref->format = conversion_map[i].pix_fmt;
- memcpy(pic->data, mpi->planes, FFMIN(sizeof(pic->data) , sizeof(mpi->planes)));
- memcpy(pic->linesize, mpi->stride, FFMIN(sizeof(pic->linesize), sizeof(mpi->stride)));
- memcpy(picref->data, pic->data, sizeof(picref->data));
- memcpy(picref->linesize, pic->linesize, sizeof(picref->linesize));
- if(pts != MP_NOPTS_VALUE)
- picref->pts= pts * av_q2d(outlink->time_base);
- avfilter_start_frame(outlink, avfilter_ref_buffer(picref, ~0));
- avfilter_draw_slice(outlink, 0, picref->video->h, 1);
- avfilter_end_frame(outlink);
- avfilter_unref_buffer(picref);
- m->frame_returned++;
- return 1;
- fail:
- if (picref && picref->video)
- av_free(picref->video);
- av_free(picref);
- av_free(pic);
- return 0;
- }
- int vf_next_config(struct vf_instance *vf,
- int width, int height, int d_width, int d_height,
- unsigned int voflags, unsigned int outfmt){
- av_assert0(width>0 && height>0);
- vf->next->w = width; vf->next->h = height;
- return 1;
- #if 0
- int flags=vf->next->query_format(vf->next,outfmt);
- if(!flags){
- // hmm. colorspace mismatch!!!
- //this is fatal for us ATM
- return 0;
- }
- mp_msg(MSGT_VFILTER,MSGL_V,"REQ: flags=0x%X req=0x%X \n",flags,vf->default_reqs);
- miss=vf->default_reqs - (flags&vf->default_reqs);
- if(miss&VFCAP_ACCEPT_STRIDE){
- // vf requires stride support but vf->next doesn't support it!
- // let's insert the 'expand' filter, it does the job for us:
- vf_instance_t* vf2=vf_open_filter(vf->next,"expand",NULL);
- if(!vf2) return 0; // shouldn't happen!
- vf->next=vf2;
- }
- vf->next->w = width; vf->next->h = height;
- #endif
- return 1;
- }
- int vf_next_control(struct vf_instance *vf, int request, void* data){
- MPContext *m= (void*)vf;
- av_log(m->avfctx, AV_LOG_DEBUG, "Received control %d\n", request);
- return 0;
- }
- static int vf_default_query_format(struct vf_instance *vf, unsigned int fmt){
- MPContext *m= (void*)vf;
- int i;
- av_log(m->avfctx, AV_LOG_DEBUG, "query %X\n", fmt);
- for(i=0; conversion_map[i].fmt; i++){
- if(fmt==conversion_map[i].fmt)
- return 1; //we suport all
- }
- return 0;
- }
- static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
- {
- MPContext *m = ctx->priv;
- char name[256];
- int i;
- m->avfctx= ctx;
- if(!args || 1!=sscanf(args, "%255[^:=]", name)){
- av_log(ctx, AV_LOG_ERROR, "Invalid parameter.\n");
- return AVERROR(EINVAL);
- }
- args+= strlen(name)+1;
- for(i=0; ;i++){
- if(!filters[i] || !strcmp(name, filters[i]->name))
- break;
- }
- if(!filters[i]){
- av_log(ctx, AV_LOG_ERROR, "Unknown filter %s\n", name);
- return AVERROR(EINVAL);
- }
- av_log(ctx, AV_LOG_WARNING,
- "'%s' is a wrapped MPlayer filter (libmpcodecs). This filter may be removed\n"
- "once it has been ported to a native libavfilter.\n", name);
- memset(&m->vf,0,sizeof(m->vf));
- m->vf.info= filters[i];
- m->vf.next = &m->next_vf;
- m->vf.put_image = vf_next_put_image;
- m->vf.config = vf_next_config;
- m->vf.query_format= vf_default_query_format;
- m->vf.control = vf_next_control;
- m->vf.default_caps=VFCAP_ACCEPT_STRIDE;
- m->vf.default_reqs=0;
- if(m->vf.info->opts)
- av_log(ctx, AV_LOG_ERROR, "opts / m_struct_set is unsupported\n");
- #if 0
- if(vf->info->opts) { // vf_vo get some special argument
- const m_struct_t* st = vf->info->opts;
- void* vf_priv = m_struct_alloc(st);
- int n;
- for(n = 0 ; args && args[2*n] ; n++)
- m_struct_set(st,vf_priv,args[2*n],args[2*n+1]);
- vf->priv = vf_priv;
- args = NULL;
- } else // Otherwise we should have the '_oldargs_'
- if(args && !strcmp(args[0],"_oldargs_"))
- args = (char**)args[1];
- else
- args = NULL;
- #endif
- if(m->vf.info->vf_open(&m->vf, args)<=0){
- av_log(ctx, AV_LOG_ERROR, "vf_open() of %s with arg=%s failed\n", name, args);
- return -1;
- }
- return 0;
- }
- static int query_formats(AVFilterContext *ctx)
- {
- AVFilterFormats *avfmts=NULL;
- MPContext *m = ctx->priv;
- enum PixelFormat lastpixfmt = PIX_FMT_NONE;
- int i;
- for(i=0; conversion_map[i].fmt; i++){
- av_log(ctx, AV_LOG_DEBUG, "query: %X\n", conversion_map[i].fmt);
- if(m->vf.query_format(&m->vf, conversion_map[i].fmt)){
- av_log(ctx, AV_LOG_DEBUG, "supported,adding\n");
- if (conversion_map[i].pix_fmt != lastpixfmt) {
- avfilter_add_format(&avfmts, conversion_map[i].pix_fmt);
- lastpixfmt = conversion_map[i].pix_fmt;
- }
- }
- }
- //We assume all allowed input formats are also allowed output formats
- avfilter_set_common_pixel_formats(ctx, avfmts);
- return 0;
- }
- static int config_inprops(AVFilterLink *inlink)
- {
- MPContext *m = inlink->dst->priv;
- int i;
- for(i=0; conversion_map[i].fmt && conversion_map[i].pix_fmt != inlink->format; i++);
- av_assert0(conversion_map[i].fmt && inlink->w && inlink->h);
- m->vf.fmt.have_configured = 1;
- m->vf.fmt.orig_height = inlink->h;
- m->vf.fmt.orig_width = inlink->w;
- m->vf.fmt.orig_fmt = conversion_map[i].fmt;
- if(m->vf.config(&m->vf, inlink->w, inlink->h, inlink->w, inlink->h, 0, conversion_map[i].fmt)<=0)
- return -1;
- return 0;
- }
- static int config_outprops(AVFilterLink *outlink)
- {
- MPContext *m = outlink->src->priv;
- outlink->w = m->next_vf.w;
- outlink->h = m->next_vf.h;
- return 0;
- }
- static int request_frame(AVFilterLink *outlink)
- {
- MPContext *m = outlink->src->priv;
- int ret;
- av_log(m->avfctx, AV_LOG_DEBUG, "mp request_frame\n");
- for(m->frame_returned=0; !m->frame_returned;){
- ret=avfilter_request_frame(outlink->src->inputs[0]);
- if(ret<0)
- break;
- }
- av_log(m->avfctx, AV_LOG_DEBUG, "mp request_frame ret=%d\n", ret);
- return ret;
- }
- static void start_frame(AVFilterLink *inlink, AVFilterBufferRef *picref)
- {
- }
- static void null_draw_slice(AVFilterLink *link, int y, int h, int slice_dir)
- {
- }
- static void end_frame(AVFilterLink *inlink)
- {
- MPContext *m = inlink->dst->priv;
- AVFilterBufferRef *inpic = inlink->cur_buf;
- int i;
- double pts= MP_NOPTS_VALUE;
- mp_image_t* mpi = new_mp_image(inpic->video->w, inpic->video->h);
- if(inpic->pts != AV_NOPTS_VALUE)
- pts= inpic->pts / av_q2d(inlink->time_base);
- for(i=0; conversion_map[i].fmt && conversion_map[i].pix_fmt != inlink->format; i++);
- mp_image_setfmt(mpi,conversion_map[i].fmt);
- memcpy(mpi->planes, inpic->data, FFMIN(sizeof(inpic->data) , sizeof(mpi->planes)));
- memcpy(mpi->stride, inpic->linesize, FFMIN(sizeof(inpic->linesize), sizeof(mpi->stride)));
- //FIXME pass interleced & tff flags around
- // mpi->flags|=MP_IMGFLAG_ALLOCATED; ?
- mpi->flags |= MP_IMGFLAG_READABLE;
- if(!(inpic->perms & AV_PERM_WRITE))
- mpi->flags |= MP_IMGFLAG_PRESERVE;
- if(m->vf.put_image(&m->vf, mpi, pts) == 0){
- av_log(m->avfctx, AV_LOG_DEBUG, "put_image() says skip\n");
- }
- free_mp_image(mpi);
- avfilter_unref_buffer(inpic);
- }
- AVFilter avfilter_vf_mp = {
- .name = "mp",
- .description = NULL_IF_CONFIG_SMALL("Apply a libmpcodecs filter to the input video."),
- .init = init,
- .priv_size = sizeof(MPContext),
- .query_formats = query_formats,
- .inputs = (const AVFilterPad[]) {{ .name = "default",
- .type = AVMEDIA_TYPE_VIDEO,
- .start_frame = start_frame,
- .draw_slice = null_draw_slice,
- .end_frame = end_frame,
- .config_props = config_inprops,
- .min_perms = AV_PERM_READ, },
- { .name = NULL}},
- .outputs = (const AVFilterPad[]) {{ .name = "default",
- .type = AVMEDIA_TYPE_VIDEO,
- .request_frame = request_frame,
- .config_props = config_outprops, },
- { .name = NULL}},
- };
|