00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00027 #include "avfilter.h"
00028 #include "libavutil/avassert.h"
00029 #include "libavutil/pixdesc.h"
00030 #include "libavutil/intreadwrite.h"
00031 #include "libavutil/imgutils.h"
00032
00033 #include "libmpcodecs/vf.h"
00034 #include "libmpcodecs/img_format.h"
00035 #include "libmpcodecs/cpudetect.h"
00036 #include "libmpcodecs/vd_ffmpeg.h"
00037 #include "libmpcodecs/vf_scale.h"
00038 #include "libmpcodecs/libvo/fastmemcpy.h"
00039
00040 #include "libswscale/swscale.h"
00041
00042
00043
00044 static const struct {
00045 int fmt;
00046 enum PixelFormat pix_fmt;
00047 } conversion_map[] = {
00048 {IMGFMT_ARGB, PIX_FMT_ARGB},
00049 {IMGFMT_BGRA, PIX_FMT_BGRA},
00050 {IMGFMT_BGR24, PIX_FMT_BGR24},
00051 {IMGFMT_BGR16BE, PIX_FMT_RGB565BE},
00052 {IMGFMT_BGR16LE, PIX_FMT_RGB565LE},
00053 {IMGFMT_BGR15BE, PIX_FMT_RGB555BE},
00054 {IMGFMT_BGR15LE, PIX_FMT_RGB555LE},
00055 {IMGFMT_BGR12BE, PIX_FMT_RGB444BE},
00056 {IMGFMT_BGR12LE, PIX_FMT_RGB444LE},
00057 {IMGFMT_BGR8, PIX_FMT_RGB8},
00058 {IMGFMT_BGR4, PIX_FMT_RGB4},
00059 {IMGFMT_BGR1, PIX_FMT_MONOBLACK},
00060 {IMGFMT_RGB1, PIX_FMT_MONOBLACK},
00061 {IMGFMT_RG4B, PIX_FMT_BGR4_BYTE},
00062 {IMGFMT_BG4B, PIX_FMT_RGB4_BYTE},
00063 {IMGFMT_RGB48LE, PIX_FMT_RGB48LE},
00064 {IMGFMT_RGB48BE, PIX_FMT_RGB48BE},
00065 {IMGFMT_ABGR, PIX_FMT_ABGR},
00066 {IMGFMT_RGBA, PIX_FMT_RGBA},
00067 {IMGFMT_RGB24, PIX_FMT_RGB24},
00068 {IMGFMT_RGB16BE, PIX_FMT_BGR565BE},
00069 {IMGFMT_RGB16LE, PIX_FMT_BGR565LE},
00070 {IMGFMT_RGB15BE, PIX_FMT_BGR555BE},
00071 {IMGFMT_RGB15LE, PIX_FMT_BGR555LE},
00072 {IMGFMT_RGB12BE, PIX_FMT_BGR444BE},
00073 {IMGFMT_RGB12LE, PIX_FMT_BGR444LE},
00074 {IMGFMT_RGB8, PIX_FMT_BGR8},
00075 {IMGFMT_RGB4, PIX_FMT_BGR4},
00076 {IMGFMT_BGR8, PIX_FMT_PAL8},
00077 {IMGFMT_YUY2, PIX_FMT_YUYV422},
00078 {IMGFMT_UYVY, PIX_FMT_UYVY422},
00079 {IMGFMT_NV12, PIX_FMT_NV12},
00080 {IMGFMT_NV21, PIX_FMT_NV21},
00081 {IMGFMT_Y800, PIX_FMT_GRAY8},
00082 {IMGFMT_Y8, PIX_FMT_GRAY8},
00083 {IMGFMT_YVU9, PIX_FMT_YUV410P},
00084 {IMGFMT_IF09, PIX_FMT_YUV410P},
00085 {IMGFMT_YV12, PIX_FMT_YUV420P},
00086 {IMGFMT_I420, PIX_FMT_YUV420P},
00087 {IMGFMT_IYUV, PIX_FMT_YUV420P},
00088 {IMGFMT_411P, PIX_FMT_YUV411P},
00089 {IMGFMT_422P, PIX_FMT_YUV422P},
00090 {IMGFMT_444P, PIX_FMT_YUV444P},
00091 {IMGFMT_440P, PIX_FMT_YUV440P},
00092
00093 {IMGFMT_420A, PIX_FMT_YUVA420P},
00094
00095 {IMGFMT_420P16_LE, PIX_FMT_YUV420P16LE},
00096 {IMGFMT_420P16_BE, PIX_FMT_YUV420P16BE},
00097 {IMGFMT_422P16_LE, PIX_FMT_YUV422P16LE},
00098 {IMGFMT_422P16_BE, PIX_FMT_YUV422P16BE},
00099 {IMGFMT_444P16_LE, PIX_FMT_YUV444P16LE},
00100 {IMGFMT_444P16_BE, PIX_FMT_YUV444P16BE},
00101
00102
00103
00104
00105 {IMGFMT_YV12, PIX_FMT_YUVJ420P},
00106 {IMGFMT_422P, PIX_FMT_YUVJ422P},
00107 {IMGFMT_444P, PIX_FMT_YUVJ444P},
00108 {IMGFMT_440P, PIX_FMT_YUVJ440P},
00109
00110 {IMGFMT_XVMC_MOCO_MPEG2, PIX_FMT_XVMC_MPEG2_MC},
00111 {IMGFMT_XVMC_IDCT_MPEG2, PIX_FMT_XVMC_MPEG2_IDCT},
00112 {IMGFMT_VDPAU_MPEG1, PIX_FMT_VDPAU_MPEG1},
00113 {IMGFMT_VDPAU_MPEG2, PIX_FMT_VDPAU_MPEG2},
00114 {IMGFMT_VDPAU_H264, PIX_FMT_VDPAU_H264},
00115 {IMGFMT_VDPAU_WMV3, PIX_FMT_VDPAU_WMV3},
00116 {IMGFMT_VDPAU_VC1, PIX_FMT_VDPAU_VC1},
00117 {IMGFMT_VDPAU_MPEG4, PIX_FMT_VDPAU_MPEG4},
00118 {0, PIX_FMT_NONE}
00119 };
00120
00121
00122 extern const vf_info_t vf_info_vo;
00123 extern const vf_info_t vf_info_rectangle;
00124 extern const vf_info_t vf_info_bmovl;
00125 extern const vf_info_t vf_info_crop;
00126 extern const vf_info_t vf_info_expand;
00127 extern const vf_info_t vf_info_pp;
00128 extern const vf_info_t vf_info_scale;
00129 extern const vf_info_t vf_info_format;
00130 extern const vf_info_t vf_info_noformat;
00131 extern const vf_info_t vf_info_flip;
00132 extern const vf_info_t vf_info_rotate;
00133 extern const vf_info_t vf_info_mirror;
00134 extern const vf_info_t vf_info_palette;
00135 extern const vf_info_t vf_info_lavc;
00136 extern const vf_info_t vf_info_zrmjpeg;
00137 extern const vf_info_t vf_info_dvbscale;
00138 extern const vf_info_t vf_info_cropdetect;
00139 extern const vf_info_t vf_info_test;
00140 extern const vf_info_t vf_info_noise;
00141 extern const vf_info_t vf_info_yvu9;
00142 extern const vf_info_t vf_info_lavcdeint;
00143 extern const vf_info_t vf_info_eq;
00144 extern const vf_info_t vf_info_eq2;
00145 extern const vf_info_t vf_info_gradfun;
00146 extern const vf_info_t vf_info_halfpack;
00147 extern const vf_info_t vf_info_dint;
00148 extern const vf_info_t vf_info_1bpp;
00149 extern const vf_info_t vf_info_2xsai;
00150 extern const vf_info_t vf_info_unsharp;
00151 extern const vf_info_t vf_info_swapuv;
00152 extern const vf_info_t vf_info_il;
00153 extern const vf_info_t vf_info_fil;
00154 extern const vf_info_t vf_info_boxblur;
00155 extern const vf_info_t vf_info_sab;
00156 extern const vf_info_t vf_info_smartblur;
00157 extern const vf_info_t vf_info_perspective;
00158 extern const vf_info_t vf_info_down3dright;
00159 extern const vf_info_t vf_info_field;
00160 extern const vf_info_t vf_info_denoise3d;
00161 extern const vf_info_t vf_info_hqdn3d;
00162 extern const vf_info_t vf_info_detc;
00163 extern const vf_info_t vf_info_telecine;
00164 extern const vf_info_t vf_info_tinterlace;
00165 extern const vf_info_t vf_info_tfields;
00166 extern const vf_info_t vf_info_ivtc;
00167 extern const vf_info_t vf_info_ilpack;
00168 extern const vf_info_t vf_info_dsize;
00169 extern const vf_info_t vf_info_decimate;
00170 extern const vf_info_t vf_info_softpulldown;
00171 extern const vf_info_t vf_info_pullup;
00172 extern const vf_info_t vf_info_filmdint;
00173 extern const vf_info_t vf_info_framestep;
00174 extern const vf_info_t vf_info_tile;
00175 extern const vf_info_t vf_info_delogo;
00176 extern const vf_info_t vf_info_remove_logo;
00177 extern const vf_info_t vf_info_hue;
00178 extern const vf_info_t vf_info_spp;
00179 extern const vf_info_t vf_info_uspp;
00180 extern const vf_info_t vf_info_fspp;
00181 extern const vf_info_t vf_info_pp7;
00182 extern const vf_info_t vf_info_yuvcsp;
00183 extern const vf_info_t vf_info_kerndeint;
00184 extern const vf_info_t vf_info_rgbtest;
00185 extern const vf_info_t vf_info_qp;
00186 extern const vf_info_t vf_info_phase;
00187 extern const vf_info_t vf_info_divtc;
00188 extern const vf_info_t vf_info_harddup;
00189 extern const vf_info_t vf_info_softskip;
00190 extern const vf_info_t vf_info_screenshot;
00191 extern const vf_info_t vf_info_ass;
00192 extern const vf_info_t vf_info_mcdeint;
00193 extern const vf_info_t vf_info_yadif;
00194 extern const vf_info_t vf_info_blackframe;
00195 extern const vf_info_t vf_info_geq;
00196 extern const vf_info_t vf_info_ow;
00197 extern const vf_info_t vf_info_fixpts;
00198 extern const vf_info_t vf_info_stereo3d;
00199
00200
00201 static const vf_info_t* const filters[]={
00202 &vf_info_2xsai,
00203 &vf_info_blackframe,
00204 &vf_info_boxblur,
00205 &vf_info_cropdetect,
00206 &vf_info_decimate,
00207 &vf_info_delogo,
00208 &vf_info_denoise3d,
00209 &vf_info_detc,
00210 &vf_info_dint,
00211 &vf_info_divtc,
00212 &vf_info_down3dright,
00213 &vf_info_dsize,
00214 &vf_info_eq2,
00215 &vf_info_eq,
00216 &vf_info_field,
00217 &vf_info_fil,
00218
00219 &vf_info_fixpts,
00220 &vf_info_framestep,
00221 &vf_info_fspp,
00222 &vf_info_geq,
00223 &vf_info_gradfun,
00224 &vf_info_harddup,
00225 &vf_info_hqdn3d,
00226 &vf_info_hue,
00227 &vf_info_il,
00228 &vf_info_ilpack,
00229 &vf_info_ivtc,
00230 &vf_info_kerndeint,
00231 &vf_info_mcdeint,
00232 &vf_info_mirror,
00233 &vf_info_noise,
00234 &vf_info_ow,
00235 &vf_info_palette,
00236 &vf_info_perspective,
00237 &vf_info_phase,
00238 &vf_info_pp7,
00239 &vf_info_pullup,
00240 &vf_info_qp,
00241 &vf_info_rectangle,
00242 &vf_info_remove_logo,
00243 &vf_info_rgbtest,
00244 &vf_info_rotate,
00245 &vf_info_sab,
00246 &vf_info_screenshot,
00247 &vf_info_smartblur,
00248 &vf_info_softpulldown,
00249 &vf_info_softskip,
00250 &vf_info_spp,
00251 &vf_info_swapuv,
00252 &vf_info_telecine,
00253 &vf_info_test,
00254 &vf_info_tile,
00255 &vf_info_tinterlace,
00256 &vf_info_unsharp,
00257 &vf_info_uspp,
00258 &vf_info_yuvcsp,
00259 &vf_info_yvu9,
00260
00261 NULL
00262 };
00263
00264
00265
00266
00267
00268
00269
00270
00271
00272
00273
00274
00275
00276
00277
00278
00279
00280
00281
00282
00283
00284
00285
00286
00287 CpuCaps gCpuCaps;
00288
00289
00290
00291 int get_sws_cpuflags(void){
00292 return
00293 (gCpuCaps.hasMMX ? SWS_CPU_CAPS_MMX : 0)
00294 | (gCpuCaps.hasMMX2 ? SWS_CPU_CAPS_MMX2 : 0)
00295 | (gCpuCaps.has3DNow ? SWS_CPU_CAPS_3DNOW : 0)
00296 | (gCpuCaps.hasAltiVec ? SWS_CPU_CAPS_ALTIVEC : 0);
00297 }
00298
00299 static void sws_getFlagsAndFilterFromCmdLine(int *flags, SwsFilter **srcFilterParam, SwsFilter **dstFilterParam)
00300 {
00301 static int firstTime=1;
00302 *flags=0;
00303
00304 #if ARCH_X86
00305 if(gCpuCaps.hasMMX)
00306 __asm__ volatile("emms\n\t"::: "memory");
00307 #endif
00308 if(firstTime)
00309 {
00310 firstTime=0;
00311 *flags= SWS_PRINT_INFO;
00312 }
00313 else if( mp_msg_test(MSGT_VFILTER,MSGL_DBG2) ) *flags= SWS_PRINT_INFO;
00314
00315 switch(SWS_BILINEAR)
00316 {
00317 case 0: *flags|= SWS_FAST_BILINEAR; break;
00318 case 1: *flags|= SWS_BILINEAR; break;
00319 case 2: *flags|= SWS_BICUBIC; break;
00320 case 3: *flags|= SWS_X; break;
00321 case 4: *flags|= SWS_POINT; break;
00322 case 5: *flags|= SWS_AREA; break;
00323 case 6: *flags|= SWS_BICUBLIN; break;
00324 case 7: *flags|= SWS_GAUSS; break;
00325 case 8: *flags|= SWS_SINC; break;
00326 case 9: *flags|= SWS_LANCZOS; break;
00327 case 10:*flags|= SWS_SPLINE; break;
00328 default:*flags|= SWS_BILINEAR; break;
00329 }
00330
00331 *srcFilterParam= NULL;
00332 *dstFilterParam= NULL;
00333 }
00334
00335
00336
00337 struct SwsContext *sws_getContextFromCmdLine(int srcW, int srcH, int srcFormat, int dstW, int dstH, int dstFormat)
00338 {
00339 int flags, i;
00340 SwsFilter *dstFilterParam, *srcFilterParam;
00341 enum PixelFormat dfmt, sfmt;
00342
00343 for(i=0; conversion_map[i].fmt && dstFormat != conversion_map[i].fmt; i++);
00344 dfmt= conversion_map[i].pix_fmt;
00345 for(i=0; conversion_map[i].fmt && srcFormat != conversion_map[i].fmt; i++);
00346 sfmt= conversion_map[i].pix_fmt;
00347
00348 if (srcFormat == IMGFMT_RGB8 || srcFormat == IMGFMT_BGR8) sfmt = PIX_FMT_PAL8;
00349 sws_getFlagsAndFilterFromCmdLine(&flags, &srcFilterParam, &dstFilterParam);
00350
00351 return sws_getContext(srcW, srcH, sfmt, dstW, dstH, dfmt, flags | get_sws_cpuflags(), srcFilterParam, dstFilterParam, NULL);
00352 }
00353
00354 typedef struct {
00355 vf_instance_t vf;
00356 vf_instance_t next_vf;
00357 AVFilterContext *avfctx;
00358 int frame_returned;
00359 } MPContext;
00360
00361 void mp_msg(int mod, int lev, const char *format, ... ){
00362 va_list va;
00363 va_start(va, format);
00364
00365 av_vlog(NULL, AV_LOG_DEBUG, format, va);
00366 va_end(va);
00367 }
00368
00369 int mp_msg_test(int mod, int lev){
00370 return 123;
00371 }
00372
00373 void init_avcodec(void)
00374 {
00375
00376 }
00377
00378
00379 void vf_clone_mpi_attributes(mp_image_t* dst, mp_image_t* src){
00380 dst->pict_type= src->pict_type;
00381 dst->fields = src->fields;
00382 dst->qscale_type= src->qscale_type;
00383 if(dst->width == src->width && dst->height == src->height){
00384 dst->qstride= src->qstride;
00385 dst->qscale= src->qscale;
00386 }
00387 }
00388
00389
00390 void vf_next_draw_slice(struct vf_instance *vf,unsigned char** src, int * stride,int w, int h, int x, int y){
00391 if (vf->next->draw_slice) {
00392 vf->next->draw_slice(vf->next,src,stride,w,h,x,y);
00393 return;
00394 }
00395 if (!vf->dmpi) {
00396 mp_msg(MSGT_VFILTER,MSGL_ERR,"draw_slice: dmpi not stored by vf_%s\n", vf->info->name);
00397 return;
00398 }
00399 if (!(vf->dmpi->flags & MP_IMGFLAG_PLANAR)) {
00400 memcpy_pic(vf->dmpi->planes[0]+y*vf->dmpi->stride[0]+vf->dmpi->bpp/8*x,
00401 src[0], vf->dmpi->bpp/8*w, h, vf->dmpi->stride[0], stride[0]);
00402 return;
00403 }
00404 memcpy_pic(vf->dmpi->planes[0]+y*vf->dmpi->stride[0]+x, src[0],
00405 w, h, vf->dmpi->stride[0], stride[0]);
00406 memcpy_pic(vf->dmpi->planes[1]+(y>>vf->dmpi->chroma_y_shift)*vf->dmpi->stride[1]+(x>>vf->dmpi->chroma_x_shift),
00407 src[1], w>>vf->dmpi->chroma_x_shift, h>>vf->dmpi->chroma_y_shift, vf->dmpi->stride[1], stride[1]);
00408 memcpy_pic(vf->dmpi->planes[2]+(y>>vf->dmpi->chroma_y_shift)*vf->dmpi->stride[2]+(x>>vf->dmpi->chroma_x_shift),
00409 src[2], w>>vf->dmpi->chroma_x_shift, h>>vf->dmpi->chroma_y_shift, vf->dmpi->stride[2], stride[2]);
00410 }
00411
00412
00413 void vf_mpi_clear(mp_image_t* mpi,int x0,int y0,int w,int h){
00414 int y;
00415 if(mpi->flags&MP_IMGFLAG_PLANAR){
00416 y0&=~1;h+=h&1;
00417 if(x0==0 && w==mpi->width){
00418
00419 memset(mpi->planes[0]+mpi->stride[0]*y0,0,mpi->stride[0]*h);
00420 memset(mpi->planes[1]+mpi->stride[1]*(y0>>mpi->chroma_y_shift),128,mpi->stride[1]*(h>>mpi->chroma_y_shift));
00421 memset(mpi->planes[2]+mpi->stride[2]*(y0>>mpi->chroma_y_shift),128,mpi->stride[2]*(h>>mpi->chroma_y_shift));
00422 } else
00423 for(y=y0;y<y0+h;y+=2){
00424 memset(mpi->planes[0]+x0+mpi->stride[0]*y,0,w);
00425 memset(mpi->planes[0]+x0+mpi->stride[0]*(y+1),0,w);
00426 memset(mpi->planes[1]+(x0>>mpi->chroma_x_shift)+mpi->stride[1]*(y>>mpi->chroma_y_shift),128,(w>>mpi->chroma_x_shift));
00427 memset(mpi->planes[2]+(x0>>mpi->chroma_x_shift)+mpi->stride[2]*(y>>mpi->chroma_y_shift),128,(w>>mpi->chroma_x_shift));
00428 }
00429 return;
00430 }
00431
00432 for(y=y0;y<y0+h;y++){
00433 unsigned char* dst=mpi->planes[0]+mpi->stride[0]*y+(mpi->bpp>>3)*x0;
00434 if(mpi->flags&MP_IMGFLAG_YUV){
00435 unsigned int* p=(unsigned int*) dst;
00436 int size=(mpi->bpp>>3)*w/4;
00437 int i;
00438 #if HAVE_BIGENDIAN
00439 #define CLEAR_PACKEDYUV_PATTERN 0x00800080
00440 #define CLEAR_PACKEDYUV_PATTERN_SWAPPED 0x80008000
00441 #else
00442 #define CLEAR_PACKEDYUV_PATTERN 0x80008000
00443 #define CLEAR_PACKEDYUV_PATTERN_SWAPPED 0x00800080
00444 #endif
00445 if(mpi->flags&MP_IMGFLAG_SWAPPED){
00446 for(i=0;i<size-3;i+=4) p[i]=p[i+1]=p[i+2]=p[i+3]=CLEAR_PACKEDYUV_PATTERN_SWAPPED;
00447 for(;i<size;i++) p[i]=CLEAR_PACKEDYUV_PATTERN_SWAPPED;
00448 } else {
00449 for(i=0;i<size-3;i+=4) p[i]=p[i+1]=p[i+2]=p[i+3]=CLEAR_PACKEDYUV_PATTERN;
00450 for(;i<size;i++) p[i]=CLEAR_PACKEDYUV_PATTERN;
00451 }
00452 } else
00453 memset(dst,0,(mpi->bpp>>3)*w);
00454 }
00455 }
00456
00457 int vf_next_query_format(struct vf_instance *vf, unsigned int fmt){
00458 return 1;
00459 }
00460
00461
00462 unsigned int vf_match_csp(vf_instance_t** vfp,const unsigned int* list,unsigned int preferred){
00463 return preferred;
00464 }
00465
00466 mp_image_t* vf_get_image(vf_instance_t* vf, unsigned int outfmt, int mp_imgtype, int mp_imgflag, int w, int h){
00467 MPContext *m= ((uint8_t*)vf) - offsetof(MPContext, next_vf);
00468 mp_image_t* mpi=NULL;
00469 int w2;
00470 int number = mp_imgtype >> 16;
00471
00472 av_assert0(vf->next == NULL);
00473
00474
00475 if(vf->w==0 && w>0) vf->w=w;
00476 if(vf->h==0 && h>0) vf->h=h;
00477
00478 av_assert0(w == -1 || w >= vf->w);
00479 av_assert0(h == -1 || h >= vf->h);
00480 av_assert0(vf->w > 0);
00481 av_assert0(vf->h > 0);
00482
00483 av_log(m->avfctx, AV_LOG_DEBUG, "get_image: %d:%d, vf: %d:%d\n", w,h,vf->w,vf->h);
00484
00485 if (w == -1) w = vf->w;
00486 if (h == -1) h = vf->h;
00487
00488 w2=(mp_imgflag&MP_IMGFLAG_ACCEPT_ALIGNED_STRIDE)?((w+15)&(~15)):w;
00489
00490
00491
00492 switch(mp_imgtype & 0xff){
00493 case MP_IMGTYPE_EXPORT:
00494 if(!vf->imgctx.export_images[0]) vf->imgctx.export_images[0]=new_mp_image(w2,h);
00495 mpi=vf->imgctx.export_images[0];
00496 break;
00497 case MP_IMGTYPE_STATIC:
00498 if(!vf->imgctx.static_images[0]) vf->imgctx.static_images[0]=new_mp_image(w2,h);
00499 mpi=vf->imgctx.static_images[0];
00500 break;
00501 case MP_IMGTYPE_TEMP:
00502 if(!vf->imgctx.temp_images[0]) vf->imgctx.temp_images[0]=new_mp_image(w2,h);
00503 mpi=vf->imgctx.temp_images[0];
00504 break;
00505 case MP_IMGTYPE_IPB:
00506 if(!(mp_imgflag&MP_IMGFLAG_READABLE)){
00507 if(!vf->imgctx.temp_images[0]) vf->imgctx.temp_images[0]=new_mp_image(w2,h);
00508 mpi=vf->imgctx.temp_images[0];
00509 break;
00510 }
00511 case MP_IMGTYPE_IP:
00512 if(!vf->imgctx.static_images[vf->imgctx.static_idx]) vf->imgctx.static_images[vf->imgctx.static_idx]=new_mp_image(w2,h);
00513 mpi=vf->imgctx.static_images[vf->imgctx.static_idx];
00514 vf->imgctx.static_idx^=1;
00515 break;
00516 case MP_IMGTYPE_NUMBERED:
00517 if (number == -1) {
00518 int i;
00519 for (i = 0; i < NUM_NUMBERED_MPI; i++)
00520 if (!vf->imgctx.numbered_images[i] || !vf->imgctx.numbered_images[i]->usage_count)
00521 break;
00522 number = i;
00523 }
00524 if (number < 0 || number >= NUM_NUMBERED_MPI) return NULL;
00525 if (!vf->imgctx.numbered_images[number]) vf->imgctx.numbered_images[number] = new_mp_image(w2,h);
00526 mpi = vf->imgctx.numbered_images[number];
00527 mpi->number = number;
00528 break;
00529 }
00530 if(mpi){
00531 mpi->type=mp_imgtype;
00532 mpi->w=vf->w; mpi->h=vf->h;
00533
00534
00535 mpi->flags&=MP_IMGFLAG_ALLOCATED|MP_IMGFLAG_TYPE_DISPLAYED|MP_IMGFLAGMASK_COLORS;
00536
00537 mpi->flags|=mp_imgflag&(MP_IMGFLAGMASK_RESTRICTIONS|MP_IMGFLAG_DRAW_CALLBACK|MP_IMGFLAG_RGB_PALETTE);
00538 if(!vf->draw_slice) mpi->flags&=~MP_IMGFLAG_DRAW_CALLBACK;
00539 if(mpi->width!=w2 || mpi->height!=h){
00540
00541 if(mpi->flags&MP_IMGFLAG_ALLOCATED){
00542 if(mpi->width<w2 || mpi->height<h){
00543
00544 av_free(mpi->planes[0]);
00545 mpi->flags&=~MP_IMGFLAG_ALLOCATED;
00546 mp_msg(MSGT_VFILTER,MSGL_V,"vf.c: have to REALLOCATE buffer memory :(\n");
00547 }
00548
00549 } {
00550 mpi->width=w2; mpi->chroma_width=(w2 + (1<<mpi->chroma_x_shift) - 1)>>mpi->chroma_x_shift;
00551 mpi->height=h; mpi->chroma_height=(h + (1<<mpi->chroma_y_shift) - 1)>>mpi->chroma_y_shift;
00552 }
00553 }
00554 if(!mpi->bpp) mp_image_setfmt(mpi,outfmt);
00555 if(!(mpi->flags&MP_IMGFLAG_ALLOCATED) && mpi->type>MP_IMGTYPE_EXPORT){
00556
00557 av_assert0(!vf->get_image);
00558
00559 if(vf->get_image) vf->get_image(vf,mpi);
00560
00561 if(!(mpi->flags&MP_IMGFLAG_DIRECT)){
00562
00563 if (!mpi->bpp) {
00564 mp_msg(MSGT_DECVIDEO, MSGL_FATAL,
00565 "vf_get_image: Tried to allocate a format that can not be allocated!\n");
00566 return NULL;
00567 }
00568
00569
00570 if(mp_imgflag&MP_IMGFLAG_PREFER_ALIGNED_STRIDE){
00571 int align=(mpi->flags&MP_IMGFLAG_PLANAR &&
00572 mpi->flags&MP_IMGFLAG_YUV) ?
00573 (8<<mpi->chroma_x_shift)-1 : 15;
00574 w2=((w+align)&(~align));
00575 if(mpi->width!=w2){
00576 #if 0
00577
00578 int flags=vf->query_format(vf,outfmt);
00579 if(!(flags&3)) mp_msg(MSGT_DECVIDEO,MSGL_WARN,"??? vf_get_image{vf->query_format(outfmt)} failed!\n");
00580
00581 if(flags&VFCAP_ACCEPT_STRIDE){
00582 #endif
00583 mpi->width=w2;
00584 mpi->chroma_width=(w2 + (1<<mpi->chroma_x_shift) - 1)>>mpi->chroma_x_shift;
00585
00586 }
00587 }
00588
00589 mp_image_alloc_planes(mpi);
00590
00591 vf_mpi_clear(mpi,0,0,mpi->width,mpi->height);
00592 }
00593 }
00594 av_assert0(!vf->start_slice);
00595 if(mpi->flags&MP_IMGFLAG_DRAW_CALLBACK)
00596 if(vf->start_slice) vf->start_slice(vf,mpi);
00597 if(!(mpi->flags&MP_IMGFLAG_TYPE_DISPLAYED)){
00598 mp_msg(MSGT_DECVIDEO,MSGL_V,"*** [%s] %s%s mp_image_t, %dx%dx%dbpp %s %s, %d bytes\n",
00599 "NULL",
00600 (mpi->type==MP_IMGTYPE_EXPORT)?"Exporting":
00601 ((mpi->flags&MP_IMGFLAG_DIRECT)?"Direct Rendering":"Allocating"),
00602 (mpi->flags&MP_IMGFLAG_DRAW_CALLBACK)?" (slices)":"",
00603 mpi->width,mpi->height,mpi->bpp,
00604 (mpi->flags&MP_IMGFLAG_YUV)?"YUV":((mpi->flags&MP_IMGFLAG_SWAPPED)?"BGR":"RGB"),
00605 (mpi->flags&MP_IMGFLAG_PLANAR)?"planar":"packed",
00606 mpi->bpp*mpi->width*mpi->height/8);
00607 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",
00608 mpi->imgfmt, mpi->planes[0], mpi->planes[1], mpi->planes[2],
00609 mpi->stride[0], mpi->stride[1], mpi->stride[2],
00610 mpi->chroma_width, mpi->chroma_height, mpi->chroma_x_shift, mpi->chroma_y_shift);
00611 mpi->flags|=MP_IMGFLAG_TYPE_DISPLAYED;
00612 }
00613
00614 mpi->qscale = NULL;
00615 }
00616 mpi->usage_count++;
00617
00618
00619
00620 return mpi;
00621 }
00622
00623
00624 int vf_next_put_image(struct vf_instance *vf,mp_image_t *mpi, double pts){
00625 MPContext *m= (void*)vf;
00626 AVFilterLink *outlink = m->avfctx->outputs[0];
00627 AVFilterBuffer *pic = av_mallocz(sizeof(AVFilterBuffer));
00628 AVFilterBufferRef *picref = av_mallocz(sizeof(AVFilterBufferRef));
00629 int i;
00630
00631 av_assert0(vf->next);
00632
00633 av_log(m->avfctx, AV_LOG_DEBUG, "vf_next_put_image\n");
00634
00635 if (!pic || !picref)
00636 goto fail;
00637
00638 picref->buf = pic;
00639 picref->buf->please_use_av_free= av_free;
00640 if (!(picref->video = av_mallocz(sizeof(AVFilterBufferRefVideoProps))))
00641 goto fail;
00642
00643 pic->w = picref->video->w = mpi->w;
00644 pic->h = picref->video->h = mpi->h;
00645
00646
00647 picref->perms = AV_PERM_READ | AV_PERM_REUSE2;
00648
00649 if(!(mpi->flags&MP_IMGFLAG_PRESERVE))
00650 picref->perms |= AV_PERM_WRITE;
00651
00652 pic->refcount = 1;
00653 picref->type = AVMEDIA_TYPE_VIDEO;
00654
00655 for(i=0; conversion_map[i].fmt && mpi->imgfmt != conversion_map[i].fmt; i++);
00656 pic->format = picref->format = conversion_map[i].pix_fmt;
00657
00658 memcpy(pic->data, mpi->planes, FFMIN(sizeof(pic->data) , sizeof(mpi->planes)));
00659 memcpy(pic->linesize, mpi->stride, FFMIN(sizeof(pic->linesize), sizeof(mpi->stride)));
00660 memcpy(picref->data, pic->data, sizeof(picref->data));
00661 memcpy(picref->linesize, pic->linesize, sizeof(picref->linesize));
00662
00663 if(pts != MP_NOPTS_VALUE)
00664 picref->pts= pts * av_q2d(outlink->time_base);
00665
00666 avfilter_start_frame(outlink, avfilter_ref_buffer(picref, ~0));
00667 avfilter_draw_slice(outlink, 0, picref->video->h, 1);
00668 avfilter_end_frame(outlink);
00669 avfilter_unref_buffer(picref);
00670 m->frame_returned++;
00671
00672 return 1;
00673 fail:
00674 if (picref && picref->video)
00675 av_free(picref->video);
00676 av_free(picref);
00677 av_free(pic);
00678 return 0;
00679 }
00680
00681 int vf_next_config(struct vf_instance *vf,
00682 int width, int height, int d_width, int d_height,
00683 unsigned int voflags, unsigned int outfmt){
00684
00685 av_assert0(width>0 && height>0);
00686 vf->next->w = width; vf->next->h = height;
00687
00688 return 1;
00689 #if 0
00690 int flags=vf->next->query_format(vf->next,outfmt);
00691 if(!flags){
00692
00693
00694 return 0;
00695 }
00696 mp_msg(MSGT_VFILTER,MSGL_V,"REQ: flags=0x%X req=0x%X \n",flags,vf->default_reqs);
00697 miss=vf->default_reqs - (flags&vf->default_reqs);
00698 if(miss&VFCAP_ACCEPT_STRIDE){
00699
00700
00701 vf_instance_t* vf2=vf_open_filter(vf->next,"expand",NULL);
00702 if(!vf2) return 0;
00703 vf->next=vf2;
00704 }
00705 vf->next->w = width; vf->next->h = height;
00706 #endif
00707 return 1;
00708 }
00709
00710 int vf_next_control(struct vf_instance *vf, int request, void* data){
00711 MPContext *m= (void*)vf;
00712 av_log(m->avfctx, AV_LOG_DEBUG, "Received control %d\n", request);
00713 return 0;
00714 }
00715
00716 static int vf_default_query_format(struct vf_instance *vf, unsigned int fmt){
00717 MPContext *m= (void*)vf;
00718 int i;
00719 av_log(m->avfctx, AV_LOG_DEBUG, "query %X\n", fmt);
00720
00721 for(i=0; conversion_map[i].fmt; i++){
00722 if(fmt==conversion_map[i].fmt)
00723 return 1;
00724 }
00725 return 0;
00726 }
00727
00728
00729 static av_cold int init(AVFilterContext *ctx, const char *args, void *opaque)
00730 {
00731 MPContext *m = ctx->priv;
00732 char name[256];
00733 int i;
00734
00735 av_log(ctx, AV_LOG_WARNING,
00736 "This is a unholy filter, it will be purified by the ffmpeg exorcist team\n"
00737 "which will change its syntax from dark -vf mp to light -vf.\n"
00738 "Thou shalst not make spells or scripts that depend on it\n");
00739
00740 m->avfctx= ctx;
00741
00742 if(!args || 1!=sscanf(args, "%255[^:]", name)){
00743 av_log(ctx, AV_LOG_ERROR, "Invalid parameter.\n");
00744 return AVERROR(EINVAL);
00745 }
00746 args+= strlen(name)+1;
00747
00748 for(i=0; ;i++){
00749 if(!filters[i] || !strcmp(name, filters[i]->name))
00750 break;
00751 }
00752
00753 if(!filters[i]){
00754 av_log(ctx, AV_LOG_ERROR, "Unknown filter %s\n", name);
00755 return AVERROR(EINVAL);
00756 }
00757
00758 memset(&m->vf,0,sizeof(m->vf));
00759 m->vf.info= filters[i];
00760
00761 m->vf.next = &m->next_vf;
00762 m->vf.put_image = vf_next_put_image;
00763 m->vf.config = vf_next_config;
00764 m->vf.query_format= vf_default_query_format;
00765 m->vf.control = vf_next_control;
00766 m->vf.default_caps=VFCAP_ACCEPT_STRIDE;
00767 m->vf.default_reqs=0;
00768 if(m->vf.info->opts)
00769 av_log(ctx, AV_LOG_ERROR, "opts / m_struct_set is unsupported\n");
00770 #if 0
00771 if(vf->info->opts) {
00772 const m_struct_t* st = vf->info->opts;
00773 void* vf_priv = m_struct_alloc(st);
00774 int n;
00775 for(n = 0 ; args && args[2*n] ; n++)
00776 m_struct_set(st,vf_priv,args[2*n],args[2*n+1]);
00777 vf->priv = vf_priv;
00778 args = NULL;
00779 } else
00780 if(args && !strcmp(args[0],"_oldargs_"))
00781 args = (char**)args[1];
00782 else
00783 args = NULL;
00784 #endif
00785 if(m->vf.info->vf_open(&m->vf, args)<=0){
00786 av_log(ctx, AV_LOG_ERROR, "vf_open() of %s with arg=%s failed\n", name, args);
00787 return -1;
00788 }
00789
00790 return 0;
00791 }
00792
00793 static int query_formats(AVFilterContext *ctx)
00794 {
00795 AVFilterFormats *avfmts=NULL;
00796 MPContext *m = ctx->priv;
00797 int i;
00798
00799 for(i=0; conversion_map[i].fmt; i++){
00800 av_log(ctx, AV_LOG_DEBUG, "query: %X\n", conversion_map[i].fmt);
00801 if(m->vf.query_format(&m->vf, conversion_map[i].fmt)){
00802 av_log(ctx, AV_LOG_DEBUG, "supported,adding\n");
00803 avfilter_add_format(&avfmts, conversion_map[i].pix_fmt);
00804 }
00805 }
00806
00807
00808 avfilter_set_common_formats(ctx, avfmts);
00809 return 0;
00810 }
00811
00812 static int config_inprops(AVFilterLink *inlink)
00813 {
00814 MPContext *m = inlink->dst->priv;
00815 int i;
00816 for(i=0; conversion_map[i].fmt && conversion_map[i].pix_fmt != inlink->format; i++);
00817
00818 av_assert0(conversion_map[i].fmt && inlink->w && inlink->h);
00819
00820 m->vf.fmt.have_configured = 1;
00821 m->vf.fmt.orig_height = inlink->h;
00822 m->vf.fmt.orig_width = inlink->w;
00823 m->vf.fmt.orig_fmt = conversion_map[i].fmt;
00824
00825 if(m->vf.config(&m->vf, inlink->w, inlink->h, inlink->w, inlink->h, 0, conversion_map[i].fmt)<=0)
00826 return -1;
00827
00828 return 0;
00829 }
00830
00831 static int config_outprops(AVFilterLink *outlink)
00832 {
00833 MPContext *m = outlink->src->priv;
00834
00835 outlink->w = m->next_vf.w;
00836 outlink->h = m->next_vf.h;
00837
00838 return 0;
00839 }
00840
00841 static int request_frame(AVFilterLink *outlink)
00842 {
00843 MPContext *m = outlink->src->priv;
00844 int ret;
00845
00846 av_log(m->avfctx, AV_LOG_DEBUG, "mp request_frame\n");
00847
00848 for(m->frame_returned=0; !m->frame_returned;){
00849 ret=avfilter_request_frame(outlink->src->inputs[0]);
00850 if(ret<0)
00851 break;
00852 }
00853
00854 av_log(m->avfctx, AV_LOG_DEBUG, "mp request_frame ret=%d\n", ret);
00855 return ret;
00856 }
00857
00858 static void start_frame(AVFilterLink *inlink, AVFilterBufferRef *picref)
00859 {
00860 }
00861
00862 static void null_draw_slice(AVFilterLink *link, int y, int h, int slice_dir)
00863 {
00864 }
00865
00866 static void end_frame(AVFilterLink *inlink)
00867 {
00868 MPContext *m = inlink->dst->priv;
00869 AVFilterBufferRef *inpic = inlink->cur_buf;
00870 AVFilterLink *outlink = inlink->dst->outputs[0];
00871 int i;
00872 double pts= MP_NOPTS_VALUE;
00873 mp_image_t* mpi = new_mp_image(inpic->video->w, inpic->video->h);
00874
00875 if(inpic->pts != AV_NOPTS_VALUE)
00876 pts= inpic->pts / av_q2d(inlink->time_base);
00877
00878 for(i=0; conversion_map[i].fmt && conversion_map[i].pix_fmt != inlink->format; i++);
00879 mp_image_setfmt(mpi,conversion_map[i].fmt);
00880
00881 memcpy(mpi->planes, inpic->data, FFMIN(sizeof(inpic->data) , sizeof(mpi->planes)));
00882 memcpy(mpi->stride, inpic->linesize, FFMIN(sizeof(inpic->linesize), sizeof(mpi->stride)));
00883
00884
00885
00886
00887 mpi->flags |= MP_IMGFLAG_READABLE;
00888 if(!(inpic->perms & AV_PERM_WRITE))
00889 mpi->flags |= MP_IMGFLAG_PRESERVE;
00890 if(m->vf.put_image(&m->vf, mpi, pts) == 0){
00891 av_log(m->avfctx, AV_LOG_DEBUG, "put_image() says skip\n");
00892 }
00893 free_mp_image(mpi);
00894
00895
00896 }
00897
00898 AVFilter avfilter_vf_mp = {
00899 .name = "mp",
00900 .description = NULL_IF_CONFIG_SMALL("libmpcodecs wrapper."),
00901 .init = init,
00902 .priv_size = sizeof(MPContext),
00903 .query_formats = query_formats,
00904
00905 .inputs = (AVFilterPad[]) {{ .name = "default",
00906 .type = AVMEDIA_TYPE_VIDEO,
00907 .start_frame = start_frame,
00908 .draw_slice = null_draw_slice,
00909 .end_frame = end_frame,
00910 .config_props = config_inprops,
00911 .min_perms = AV_PERM_READ, },
00912 { .name = NULL}},
00913 .outputs = (AVFilterPad[]) {{ .name = "default",
00914 .type = AVMEDIA_TYPE_VIDEO,
00915 .request_frame = request_frame,
00916 .config_props = config_outprops, },
00917 { .name = NULL}},
00918 };