annotate ffmpeg/libavfilter/vf_mp.c @ 13:844d341cf643 tip

Back up before ISMIR
author Yading Song <yading.song@eecs.qmul.ac.uk>
date Thu, 31 Oct 2013 13:17:06 +0000
parents 6840f77b83aa
children
rev   line source
yading@10 1 /*
yading@10 2 * Copyright (c) 2011 Michael Niedermayer
yading@10 3 *
yading@10 4 * This file is part of FFmpeg.
yading@10 5 *
yading@10 6 * FFmpeg is free software; you can redistribute it and/or modify
yading@10 7 * it under the terms of the GNU General Public License as published by
yading@10 8 * the Free Software Foundation; either version 2 of the License, or
yading@10 9 * (at your option) any later version.
yading@10 10 *
yading@10 11 * FFmpeg is distributed in the hope that it will be useful,
yading@10 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
yading@10 14 * GNU General Public License for more details.
yading@10 15 *
yading@10 16 * You should have received a copy of the GNU General Public License
yading@10 17 * along with FFmpeg; if not, write to the Free Software
yading@10 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10 19 *
yading@10 20 * Parts of this file have been stolen from mplayer
yading@10 21 */
yading@10 22
yading@10 23 /**
yading@10 24 * @file
yading@10 25 */
yading@10 26
yading@10 27 #include "avfilter.h"
yading@10 28 #include "video.h"
yading@10 29 #include "formats.h"
yading@10 30 #include "internal.h"
yading@10 31 #include "libavutil/avassert.h"
yading@10 32 #include "libavutil/pixdesc.h"
yading@10 33 #include "libavutil/intreadwrite.h"
yading@10 34 #include "libavutil/imgutils.h"
yading@10 35 #include "libavutil/opt.h"
yading@10 36
yading@10 37 #include "libmpcodecs/vf.h"
yading@10 38 #include "libmpcodecs/img_format.h"
yading@10 39 #include "libmpcodecs/cpudetect.h"
yading@10 40 #include "libmpcodecs/av_helpers.h"
yading@10 41 #include "libmpcodecs/vf_scale.h"
yading@10 42 #include "libmpcodecs/libvo/fastmemcpy.h"
yading@10 43
yading@10 44 #include "libswscale/swscale.h"
yading@10 45
yading@10 46
yading@10 47 //FIXME maybe link the orig in
yading@10 48 //XXX: identical pix_fmt must be following with each others
yading@10 49 static const struct {
yading@10 50 int fmt;
yading@10 51 enum AVPixelFormat pix_fmt;
yading@10 52 } conversion_map[] = {
yading@10 53 {IMGFMT_ARGB, AV_PIX_FMT_ARGB},
yading@10 54 {IMGFMT_BGRA, AV_PIX_FMT_BGRA},
yading@10 55 {IMGFMT_BGR24, AV_PIX_FMT_BGR24},
yading@10 56 {IMGFMT_BGR16BE, AV_PIX_FMT_RGB565BE},
yading@10 57 {IMGFMT_BGR16LE, AV_PIX_FMT_RGB565LE},
yading@10 58 {IMGFMT_BGR15BE, AV_PIX_FMT_RGB555BE},
yading@10 59 {IMGFMT_BGR15LE, AV_PIX_FMT_RGB555LE},
yading@10 60 {IMGFMT_BGR12BE, AV_PIX_FMT_RGB444BE},
yading@10 61 {IMGFMT_BGR12LE, AV_PIX_FMT_RGB444LE},
yading@10 62 {IMGFMT_BGR8, AV_PIX_FMT_RGB8},
yading@10 63 {IMGFMT_BGR4, AV_PIX_FMT_RGB4},
yading@10 64 {IMGFMT_BGR1, AV_PIX_FMT_MONOBLACK},
yading@10 65 {IMGFMT_RGB1, AV_PIX_FMT_MONOBLACK},
yading@10 66 {IMGFMT_RG4B, AV_PIX_FMT_BGR4_BYTE},
yading@10 67 {IMGFMT_BG4B, AV_PIX_FMT_RGB4_BYTE},
yading@10 68 {IMGFMT_RGB48LE, AV_PIX_FMT_RGB48LE},
yading@10 69 {IMGFMT_RGB48BE, AV_PIX_FMT_RGB48BE},
yading@10 70 {IMGFMT_ABGR, AV_PIX_FMT_ABGR},
yading@10 71 {IMGFMT_RGBA, AV_PIX_FMT_RGBA},
yading@10 72 {IMGFMT_RGB24, AV_PIX_FMT_RGB24},
yading@10 73 {IMGFMT_RGB16BE, AV_PIX_FMT_BGR565BE},
yading@10 74 {IMGFMT_RGB16LE, AV_PIX_FMT_BGR565LE},
yading@10 75 {IMGFMT_RGB15BE, AV_PIX_FMT_BGR555BE},
yading@10 76 {IMGFMT_RGB15LE, AV_PIX_FMT_BGR555LE},
yading@10 77 {IMGFMT_RGB12BE, AV_PIX_FMT_BGR444BE},
yading@10 78 {IMGFMT_RGB12LE, AV_PIX_FMT_BGR444LE},
yading@10 79 {IMGFMT_RGB8, AV_PIX_FMT_BGR8},
yading@10 80 {IMGFMT_RGB4, AV_PIX_FMT_BGR4},
yading@10 81 {IMGFMT_BGR8, AV_PIX_FMT_PAL8},
yading@10 82 {IMGFMT_YUY2, AV_PIX_FMT_YUYV422},
yading@10 83 {IMGFMT_UYVY, AV_PIX_FMT_UYVY422},
yading@10 84 {IMGFMT_NV12, AV_PIX_FMT_NV12},
yading@10 85 {IMGFMT_NV21, AV_PIX_FMT_NV21},
yading@10 86 {IMGFMT_Y800, AV_PIX_FMT_GRAY8},
yading@10 87 {IMGFMT_Y8, AV_PIX_FMT_GRAY8},
yading@10 88 {IMGFMT_YVU9, AV_PIX_FMT_YUV410P},
yading@10 89 {IMGFMT_IF09, AV_PIX_FMT_YUV410P},
yading@10 90 {IMGFMT_YV12, AV_PIX_FMT_YUV420P},
yading@10 91 {IMGFMT_I420, AV_PIX_FMT_YUV420P},
yading@10 92 {IMGFMT_IYUV, AV_PIX_FMT_YUV420P},
yading@10 93 {IMGFMT_411P, AV_PIX_FMT_YUV411P},
yading@10 94 {IMGFMT_422P, AV_PIX_FMT_YUV422P},
yading@10 95 {IMGFMT_444P, AV_PIX_FMT_YUV444P},
yading@10 96 {IMGFMT_440P, AV_PIX_FMT_YUV440P},
yading@10 97
yading@10 98 {IMGFMT_420A, AV_PIX_FMT_YUVA420P},
yading@10 99
yading@10 100 {IMGFMT_420P16_LE, AV_PIX_FMT_YUV420P16LE},
yading@10 101 {IMGFMT_420P16_BE, AV_PIX_FMT_YUV420P16BE},
yading@10 102 {IMGFMT_422P16_LE, AV_PIX_FMT_YUV422P16LE},
yading@10 103 {IMGFMT_422P16_BE, AV_PIX_FMT_YUV422P16BE},
yading@10 104 {IMGFMT_444P16_LE, AV_PIX_FMT_YUV444P16LE},
yading@10 105 {IMGFMT_444P16_BE, AV_PIX_FMT_YUV444P16BE},
yading@10 106
yading@10 107 // YUVJ are YUV formats that use the full Y range and not just
yading@10 108 // 16 - 235 (see colorspaces.txt).
yading@10 109 // Currently they are all treated the same way.
yading@10 110 {IMGFMT_YV12, AV_PIX_FMT_YUVJ420P},
yading@10 111 {IMGFMT_422P, AV_PIX_FMT_YUVJ422P},
yading@10 112 {IMGFMT_444P, AV_PIX_FMT_YUVJ444P},
yading@10 113 {IMGFMT_440P, AV_PIX_FMT_YUVJ440P},
yading@10 114
yading@10 115 {IMGFMT_XVMC_MOCO_MPEG2, AV_PIX_FMT_XVMC_MPEG2_MC},
yading@10 116 {IMGFMT_XVMC_IDCT_MPEG2, AV_PIX_FMT_XVMC_MPEG2_IDCT},
yading@10 117 {IMGFMT_VDPAU_MPEG1, AV_PIX_FMT_VDPAU_MPEG1},
yading@10 118 {IMGFMT_VDPAU_MPEG2, AV_PIX_FMT_VDPAU_MPEG2},
yading@10 119 {IMGFMT_VDPAU_H264, AV_PIX_FMT_VDPAU_H264},
yading@10 120 {IMGFMT_VDPAU_WMV3, AV_PIX_FMT_VDPAU_WMV3},
yading@10 121 {IMGFMT_VDPAU_VC1, AV_PIX_FMT_VDPAU_VC1},
yading@10 122 {IMGFMT_VDPAU_MPEG4, AV_PIX_FMT_VDPAU_MPEG4},
yading@10 123 {0, AV_PIX_FMT_NONE}
yading@10 124 };
yading@10 125
yading@10 126 extern const vf_info_t ff_vf_info_dint;
yading@10 127 extern const vf_info_t ff_vf_info_down3dright;
yading@10 128 extern const vf_info_t ff_vf_info_eq2;
yading@10 129 extern const vf_info_t ff_vf_info_eq;
yading@10 130 extern const vf_info_t ff_vf_info_fil;
yading@10 131 extern const vf_info_t ff_vf_info_fspp;
yading@10 132 extern const vf_info_t ff_vf_info_ilpack;
yading@10 133 extern const vf_info_t ff_vf_info_mcdeint;
yading@10 134 extern const vf_info_t ff_vf_info_ow;
yading@10 135 extern const vf_info_t ff_vf_info_perspective;
yading@10 136 extern const vf_info_t ff_vf_info_phase;
yading@10 137 extern const vf_info_t ff_vf_info_pp7;
yading@10 138 extern const vf_info_t ff_vf_info_pullup;
yading@10 139 extern const vf_info_t ff_vf_info_qp;
yading@10 140 extern const vf_info_t ff_vf_info_sab;
yading@10 141 extern const vf_info_t ff_vf_info_softpulldown;
yading@10 142 extern const vf_info_t ff_vf_info_spp;
yading@10 143 extern const vf_info_t ff_vf_info_tinterlace;
yading@10 144 extern const vf_info_t ff_vf_info_uspp;
yading@10 145
yading@10 146
yading@10 147 static const vf_info_t* const filters[]={
yading@10 148 &ff_vf_info_dint,
yading@10 149 &ff_vf_info_down3dright,
yading@10 150 &ff_vf_info_eq2,
yading@10 151 &ff_vf_info_eq,
yading@10 152 &ff_vf_info_fil,
yading@10 153 &ff_vf_info_fspp,
yading@10 154 &ff_vf_info_ilpack,
yading@10 155 &ff_vf_info_mcdeint,
yading@10 156 &ff_vf_info_ow,
yading@10 157 &ff_vf_info_perspective,
yading@10 158 &ff_vf_info_phase,
yading@10 159 &ff_vf_info_pp7,
yading@10 160 &ff_vf_info_pullup,
yading@10 161 &ff_vf_info_qp,
yading@10 162 &ff_vf_info_sab,
yading@10 163 &ff_vf_info_softpulldown,
yading@10 164 &ff_vf_info_spp,
yading@10 165 &ff_vf_info_tinterlace,
yading@10 166 &ff_vf_info_uspp,
yading@10 167
yading@10 168 NULL
yading@10 169 };
yading@10 170
yading@10 171 /*
yading@10 172 Unsupported filters
yading@10 173 1bpp
yading@10 174 ass
yading@10 175 bmovl
yading@10 176 crop
yading@10 177 dvbscale
yading@10 178 flip
yading@10 179 expand
yading@10 180 format
yading@10 181 halfpack
yading@10 182 lavc
yading@10 183 lavcdeint
yading@10 184 noformat
yading@10 185 pp
yading@10 186 scale
yading@10 187 tfields
yading@10 188 vo
yading@10 189 yadif
yading@10 190 zrmjpeg
yading@10 191 */
yading@10 192
yading@10 193 CpuCaps ff_gCpuCaps; //FIXME initialize this so optims work
yading@10 194
yading@10 195 enum AVPixelFormat ff_mp2ff_pix_fmt(int mp){
yading@10 196 int i;
yading@10 197 for(i=0; conversion_map[i].fmt && mp != conversion_map[i].fmt; i++)
yading@10 198 ;
yading@10 199 return mp == conversion_map[i].fmt ? conversion_map[i].pix_fmt : AV_PIX_FMT_NONE;
yading@10 200 }
yading@10 201
yading@10 202 static void ff_sws_getFlagsAndFilterFromCmdLine(int *flags, SwsFilter **srcFilterParam, SwsFilter **dstFilterParam)
yading@10 203 {
yading@10 204 static int firstTime=1;
yading@10 205 *flags=0;
yading@10 206
yading@10 207 #if ARCH_X86
yading@10 208 if(ff_gCpuCaps.hasMMX)
yading@10 209 __asm__ volatile("emms\n\t"::: "memory"); //FIXME this should not be required but it IS (even for non-MMX versions)
yading@10 210 #endif
yading@10 211 if(firstTime)
yading@10 212 {
yading@10 213 firstTime=0;
yading@10 214 *flags= SWS_PRINT_INFO;
yading@10 215 }
yading@10 216 else if( ff_mp_msg_test(MSGT_VFILTER,MSGL_DBG2) ) *flags= SWS_PRINT_INFO;
yading@10 217
yading@10 218 switch(SWS_BILINEAR)
yading@10 219 {
yading@10 220 case 0: *flags|= SWS_FAST_BILINEAR; break;
yading@10 221 case 1: *flags|= SWS_BILINEAR; break;
yading@10 222 case 2: *flags|= SWS_BICUBIC; break;
yading@10 223 case 3: *flags|= SWS_X; break;
yading@10 224 case 4: *flags|= SWS_POINT; break;
yading@10 225 case 5: *flags|= SWS_AREA; break;
yading@10 226 case 6: *flags|= SWS_BICUBLIN; break;
yading@10 227 case 7: *flags|= SWS_GAUSS; break;
yading@10 228 case 8: *flags|= SWS_SINC; break;
yading@10 229 case 9: *flags|= SWS_LANCZOS; break;
yading@10 230 case 10:*flags|= SWS_SPLINE; break;
yading@10 231 default:*flags|= SWS_BILINEAR; break;
yading@10 232 }
yading@10 233
yading@10 234 *srcFilterParam= NULL;
yading@10 235 *dstFilterParam= NULL;
yading@10 236 }
yading@10 237
yading@10 238 //exact copy from vf_scale.c
yading@10 239 // will use sws_flags & src_filter (from cmd line)
yading@10 240 struct SwsContext *ff_sws_getContextFromCmdLine(int srcW, int srcH, int srcFormat, int dstW, int dstH, int dstFormat)
yading@10 241 {
yading@10 242 int flags, i;
yading@10 243 SwsFilter *dstFilterParam, *srcFilterParam;
yading@10 244 enum AVPixelFormat dfmt, sfmt;
yading@10 245
yading@10 246 for(i=0; conversion_map[i].fmt && dstFormat != conversion_map[i].fmt; i++);
yading@10 247 dfmt= conversion_map[i].pix_fmt;
yading@10 248 for(i=0; conversion_map[i].fmt && srcFormat != conversion_map[i].fmt; i++);
yading@10 249 sfmt= conversion_map[i].pix_fmt;
yading@10 250
yading@10 251 if (srcFormat == IMGFMT_RGB8 || srcFormat == IMGFMT_BGR8) sfmt = AV_PIX_FMT_PAL8;
yading@10 252 ff_sws_getFlagsAndFilterFromCmdLine(&flags, &srcFilterParam, &dstFilterParam);
yading@10 253
yading@10 254 return sws_getContext(srcW, srcH, sfmt, dstW, dstH, dfmt, flags , srcFilterParam, dstFilterParam, NULL);
yading@10 255 }
yading@10 256
yading@10 257 typedef struct {
yading@10 258 const AVClass *class;
yading@10 259 vf_instance_t vf;
yading@10 260 vf_instance_t next_vf;
yading@10 261 AVFilterContext *avfctx;
yading@10 262 int frame_returned;
yading@10 263 char *filter;
yading@10 264 } MPContext;
yading@10 265
yading@10 266 #define OFFSET(x) offsetof(MPContext, x)
yading@10 267 #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
yading@10 268 static const AVOption mp_options[] = {
yading@10 269 { "filter", "set MPlayer filter name and parameters", OFFSET(filter), AV_OPT_TYPE_STRING, {.str=NULL}, .flags = FLAGS },
yading@10 270 { NULL }
yading@10 271 };
yading@10 272
yading@10 273 AVFILTER_DEFINE_CLASS(mp);
yading@10 274
yading@10 275 void ff_mp_msg(int mod, int lev, const char *format, ... ){
yading@10 276 va_list va;
yading@10 277 va_start(va, format);
yading@10 278 //FIXME convert lev/mod
yading@10 279 av_vlog(NULL, AV_LOG_DEBUG, format, va);
yading@10 280 va_end(va);
yading@10 281 }
yading@10 282
yading@10 283 int ff_mp_msg_test(int mod, int lev){
yading@10 284 return 123;
yading@10 285 }
yading@10 286
yading@10 287 void ff_init_avcodec(void)
yading@10 288 {
yading@10 289 //we maybe should init but its kinda 1. unneeded 2. a bit inpolite from here
yading@10 290 }
yading@10 291
yading@10 292 //Exact copy of vf.c
yading@10 293 void ff_vf_clone_mpi_attributes(mp_image_t* dst, mp_image_t* src){
yading@10 294 dst->pict_type= src->pict_type;
yading@10 295 dst->fields = src->fields;
yading@10 296 dst->qscale_type= src->qscale_type;
yading@10 297 if(dst->width == src->width && dst->height == src->height){
yading@10 298 dst->qstride= src->qstride;
yading@10 299 dst->qscale= src->qscale;
yading@10 300 }
yading@10 301 }
yading@10 302
yading@10 303 //Exact copy of vf.c
yading@10 304 void ff_vf_next_draw_slice(struct vf_instance *vf,unsigned char** src, int * stride,int w, int h, int x, int y){
yading@10 305 if (vf->next->draw_slice) {
yading@10 306 vf->next->draw_slice(vf->next,src,stride,w,h,x,y);
yading@10 307 return;
yading@10 308 }
yading@10 309 if (!vf->dmpi) {
yading@10 310 ff_mp_msg(MSGT_VFILTER,MSGL_ERR,"draw_slice: dmpi not stored by vf_%s\n", vf->info->name);
yading@10 311 return;
yading@10 312 }
yading@10 313 if (!(vf->dmpi->flags & MP_IMGFLAG_PLANAR)) {
yading@10 314 memcpy_pic(vf->dmpi->planes[0]+y*vf->dmpi->stride[0]+vf->dmpi->bpp/8*x,
yading@10 315 src[0], vf->dmpi->bpp/8*w, h, vf->dmpi->stride[0], stride[0]);
yading@10 316 return;
yading@10 317 }
yading@10 318 memcpy_pic(vf->dmpi->planes[0]+y*vf->dmpi->stride[0]+x, src[0],
yading@10 319 w, h, vf->dmpi->stride[0], stride[0]);
yading@10 320 memcpy_pic(vf->dmpi->planes[1]+(y>>vf->dmpi->chroma_y_shift)*vf->dmpi->stride[1]+(x>>vf->dmpi->chroma_x_shift),
yading@10 321 src[1], w>>vf->dmpi->chroma_x_shift, h>>vf->dmpi->chroma_y_shift, vf->dmpi->stride[1], stride[1]);
yading@10 322 memcpy_pic(vf->dmpi->planes[2]+(y>>vf->dmpi->chroma_y_shift)*vf->dmpi->stride[2]+(x>>vf->dmpi->chroma_x_shift),
yading@10 323 src[2], w>>vf->dmpi->chroma_x_shift, h>>vf->dmpi->chroma_y_shift, vf->dmpi->stride[2], stride[2]);
yading@10 324 }
yading@10 325
yading@10 326 //Exact copy of vf.c
yading@10 327 void ff_vf_mpi_clear(mp_image_t* mpi,int x0,int y0,int w,int h){
yading@10 328 int y;
yading@10 329 if(mpi->flags&MP_IMGFLAG_PLANAR){
yading@10 330 y0&=~1;h+=h&1;
yading@10 331 if(x0==0 && w==mpi->width){
yading@10 332 // full width clear:
yading@10 333 memset(mpi->planes[0]+mpi->stride[0]*y0,0,mpi->stride[0]*h);
yading@10 334 memset(mpi->planes[1]+mpi->stride[1]*(y0>>mpi->chroma_y_shift),128,mpi->stride[1]*(h>>mpi->chroma_y_shift));
yading@10 335 memset(mpi->planes[2]+mpi->stride[2]*(y0>>mpi->chroma_y_shift),128,mpi->stride[2]*(h>>mpi->chroma_y_shift));
yading@10 336 } else
yading@10 337 for(y=y0;y<y0+h;y+=2){
yading@10 338 memset(mpi->planes[0]+x0+mpi->stride[0]*y,0,w);
yading@10 339 memset(mpi->planes[0]+x0+mpi->stride[0]*(y+1),0,w);
yading@10 340 memset(mpi->planes[1]+(x0>>mpi->chroma_x_shift)+mpi->stride[1]*(y>>mpi->chroma_y_shift),128,(w>>mpi->chroma_x_shift));
yading@10 341 memset(mpi->planes[2]+(x0>>mpi->chroma_x_shift)+mpi->stride[2]*(y>>mpi->chroma_y_shift),128,(w>>mpi->chroma_x_shift));
yading@10 342 }
yading@10 343 return;
yading@10 344 }
yading@10 345 // packed:
yading@10 346 for(y=y0;y<y0+h;y++){
yading@10 347 unsigned char* dst=mpi->planes[0]+mpi->stride[0]*y+(mpi->bpp>>3)*x0;
yading@10 348 if(mpi->flags&MP_IMGFLAG_YUV){
yading@10 349 unsigned int* p=(unsigned int*) dst;
yading@10 350 int size=(mpi->bpp>>3)*w/4;
yading@10 351 int i;
yading@10 352 #if HAVE_BIGENDIAN
yading@10 353 #define CLEAR_PACKEDYUV_PATTERN 0x00800080
yading@10 354 #define CLEAR_PACKEDYUV_PATTERN_SWAPPED 0x80008000
yading@10 355 #else
yading@10 356 #define CLEAR_PACKEDYUV_PATTERN 0x80008000
yading@10 357 #define CLEAR_PACKEDYUV_PATTERN_SWAPPED 0x00800080
yading@10 358 #endif
yading@10 359 if(mpi->flags&MP_IMGFLAG_SWAPPED){
yading@10 360 for(i=0;i<size-3;i+=4) p[i]=p[i+1]=p[i+2]=p[i+3]=CLEAR_PACKEDYUV_PATTERN_SWAPPED;
yading@10 361 for(;i<size;i++) p[i]=CLEAR_PACKEDYUV_PATTERN_SWAPPED;
yading@10 362 } else {
yading@10 363 for(i=0;i<size-3;i+=4) p[i]=p[i+1]=p[i+2]=p[i+3]=CLEAR_PACKEDYUV_PATTERN;
yading@10 364 for(;i<size;i++) p[i]=CLEAR_PACKEDYUV_PATTERN;
yading@10 365 }
yading@10 366 } else
yading@10 367 memset(dst,0,(mpi->bpp>>3)*w);
yading@10 368 }
yading@10 369 }
yading@10 370
yading@10 371 int ff_vf_next_query_format(struct vf_instance *vf, unsigned int fmt){
yading@10 372 return 1;
yading@10 373 }
yading@10 374
yading@10 375 //used by delogo
yading@10 376 unsigned int ff_vf_match_csp(vf_instance_t** vfp,const unsigned int* list,unsigned int preferred){
yading@10 377 return preferred;
yading@10 378 }
yading@10 379
yading@10 380 mp_image_t* ff_vf_get_image(vf_instance_t* vf, unsigned int outfmt, int mp_imgtype, int mp_imgflag, int w, int h){
yading@10 381 MPContext *m= (MPContext*)(((uint8_t*)vf) - offsetof(MPContext, next_vf));
yading@10 382 mp_image_t* mpi=NULL;
yading@10 383 int w2;
yading@10 384 int number = mp_imgtype >> 16;
yading@10 385
yading@10 386 av_assert0(vf->next == NULL); // all existing filters call this just on next
yading@10 387
yading@10 388 //vf_dint needs these as it calls ff_vf_get_image() before configuring the output
yading@10 389 if(vf->w==0 && w>0) vf->w=w;
yading@10 390 if(vf->h==0 && h>0) vf->h=h;
yading@10 391
yading@10 392 av_assert0(w == -1 || w >= vf->w);
yading@10 393 av_assert0(h == -1 || h >= vf->h);
yading@10 394 av_assert0(vf->w > 0);
yading@10 395 av_assert0(vf->h > 0);
yading@10 396
yading@10 397 av_log(m->avfctx, AV_LOG_DEBUG, "get_image: %d:%d, vf: %d:%d\n", w,h,vf->w,vf->h);
yading@10 398
yading@10 399 if (w == -1) w = vf->w;
yading@10 400 if (h == -1) h = vf->h;
yading@10 401
yading@10 402 w2=(mp_imgflag&MP_IMGFLAG_ACCEPT_ALIGNED_STRIDE)?((w+15)&(~15)):w;
yading@10 403
yading@10 404 // Note: we should call libvo first to check if it supports direct rendering
yading@10 405 // and if not, then fallback to software buffers:
yading@10 406 switch(mp_imgtype & 0xff){
yading@10 407 case MP_IMGTYPE_EXPORT:
yading@10 408 if(!vf->imgctx.export_images[0]) vf->imgctx.export_images[0]=ff_new_mp_image(w2,h);
yading@10 409 mpi=vf->imgctx.export_images[0];
yading@10 410 break;
yading@10 411 case MP_IMGTYPE_STATIC:
yading@10 412 if(!vf->imgctx.static_images[0]) vf->imgctx.static_images[0]=ff_new_mp_image(w2,h);
yading@10 413 mpi=vf->imgctx.static_images[0];
yading@10 414 break;
yading@10 415 case MP_IMGTYPE_TEMP:
yading@10 416 if(!vf->imgctx.temp_images[0]) vf->imgctx.temp_images[0]=ff_new_mp_image(w2,h);
yading@10 417 mpi=vf->imgctx.temp_images[0];
yading@10 418 break;
yading@10 419 case MP_IMGTYPE_IPB:
yading@10 420 if(!(mp_imgflag&MP_IMGFLAG_READABLE)){ // B frame:
yading@10 421 if(!vf->imgctx.temp_images[0]) vf->imgctx.temp_images[0]=ff_new_mp_image(w2,h);
yading@10 422 mpi=vf->imgctx.temp_images[0];
yading@10 423 break;
yading@10 424 }
yading@10 425 case MP_IMGTYPE_IP:
yading@10 426 if(!vf->imgctx.static_images[vf->imgctx.static_idx]) vf->imgctx.static_images[vf->imgctx.static_idx]=ff_new_mp_image(w2,h);
yading@10 427 mpi=vf->imgctx.static_images[vf->imgctx.static_idx];
yading@10 428 vf->imgctx.static_idx^=1;
yading@10 429 break;
yading@10 430 case MP_IMGTYPE_NUMBERED:
yading@10 431 if (number == -1) {
yading@10 432 int i;
yading@10 433 for (i = 0; i < NUM_NUMBERED_MPI; i++)
yading@10 434 if (!vf->imgctx.numbered_images[i] || !vf->imgctx.numbered_images[i]->usage_count)
yading@10 435 break;
yading@10 436 number = i;
yading@10 437 }
yading@10 438 if (number < 0 || number >= NUM_NUMBERED_MPI) return NULL;
yading@10 439 if (!vf->imgctx.numbered_images[number]) vf->imgctx.numbered_images[number] = ff_new_mp_image(w2,h);
yading@10 440 mpi = vf->imgctx.numbered_images[number];
yading@10 441 mpi->number = number;
yading@10 442 break;
yading@10 443 }
yading@10 444 if(mpi){
yading@10 445 mpi->type=mp_imgtype;
yading@10 446 mpi->w=vf->w; mpi->h=vf->h;
yading@10 447 // keep buffer allocation status & color flags only:
yading@10 448 // mpi->flags&=~(MP_IMGFLAG_PRESERVE|MP_IMGFLAG_READABLE|MP_IMGFLAG_DIRECT);
yading@10 449 mpi->flags&=MP_IMGFLAG_ALLOCATED|MP_IMGFLAG_TYPE_DISPLAYED|MP_IMGFLAGMASK_COLORS;
yading@10 450 // accept restrictions, draw_slice and palette flags only:
yading@10 451 mpi->flags|=mp_imgflag&(MP_IMGFLAGMASK_RESTRICTIONS|MP_IMGFLAG_DRAW_CALLBACK|MP_IMGFLAG_RGB_PALETTE);
yading@10 452 if(!vf->draw_slice) mpi->flags&=~MP_IMGFLAG_DRAW_CALLBACK;
yading@10 453 if(mpi->width!=w2 || mpi->height!=h){
yading@10 454 // printf("vf.c: MPI parameters changed! %dx%d -> %dx%d \n", mpi->width,mpi->height,w2,h);
yading@10 455 if(mpi->flags&MP_IMGFLAG_ALLOCATED){
yading@10 456 if(mpi->width<w2 || mpi->height<h){
yading@10 457 // need to re-allocate buffer memory:
yading@10 458 av_free(mpi->planes[0]);
yading@10 459 mpi->flags&=~MP_IMGFLAG_ALLOCATED;
yading@10 460 ff_mp_msg(MSGT_VFILTER,MSGL_V,"vf.c: have to REALLOCATE buffer memory :(\n");
yading@10 461 }
yading@10 462 // } else {
yading@10 463 } {
yading@10 464 mpi->width=w2; mpi->chroma_width=(w2 + (1<<mpi->chroma_x_shift) - 1)>>mpi->chroma_x_shift;
yading@10 465 mpi->height=h; mpi->chroma_height=(h + (1<<mpi->chroma_y_shift) - 1)>>mpi->chroma_y_shift;
yading@10 466 }
yading@10 467 }
yading@10 468 if(!mpi->bpp) ff_mp_image_setfmt(mpi,outfmt);
yading@10 469 if(!(mpi->flags&MP_IMGFLAG_ALLOCATED) && mpi->type>MP_IMGTYPE_EXPORT){
yading@10 470
yading@10 471 av_assert0(!vf->get_image);
yading@10 472 // check libvo first!
yading@10 473 if(vf->get_image) vf->get_image(vf,mpi);
yading@10 474
yading@10 475 if(!(mpi->flags&MP_IMGFLAG_DIRECT)){
yading@10 476 // non-direct and not yet allocated image. allocate it!
yading@10 477 if (!mpi->bpp) { // no way we can allocate this
yading@10 478 ff_mp_msg(MSGT_DECVIDEO, MSGL_FATAL,
yading@10 479 "ff_vf_get_image: Tried to allocate a format that can not be allocated!\n");
yading@10 480 return NULL;
yading@10 481 }
yading@10 482
yading@10 483 // check if codec prefer aligned stride:
yading@10 484 if(mp_imgflag&MP_IMGFLAG_PREFER_ALIGNED_STRIDE){
yading@10 485 int align=(mpi->flags&MP_IMGFLAG_PLANAR &&
yading@10 486 mpi->flags&MP_IMGFLAG_YUV) ?
yading@10 487 (8<<mpi->chroma_x_shift)-1 : 15; // -- maybe FIXME
yading@10 488 w2=((w+align)&(~align));
yading@10 489 if(mpi->width!=w2){
yading@10 490 #if 0
yading@10 491 // we have to change width... check if we CAN co it:
yading@10 492 int flags=vf->query_format(vf,outfmt); // should not fail
yading@10 493 if(!(flags&3)) ff_mp_msg(MSGT_DECVIDEO,MSGL_WARN,"??? ff_vf_get_image{vf->query_format(outfmt)} failed!\n");
yading@10 494 // printf("query -> 0x%X \n",flags);
yading@10 495 if(flags&VFCAP_ACCEPT_STRIDE){
yading@10 496 #endif
yading@10 497 mpi->width=w2;
yading@10 498 mpi->chroma_width=(w2 + (1<<mpi->chroma_x_shift) - 1)>>mpi->chroma_x_shift;
yading@10 499 // }
yading@10 500 }
yading@10 501 }
yading@10 502
yading@10 503 ff_mp_image_alloc_planes(mpi);
yading@10 504 // printf("clearing img!\n");
yading@10 505 ff_vf_mpi_clear(mpi,0,0,mpi->width,mpi->height);
yading@10 506 }
yading@10 507 }
yading@10 508 av_assert0(!vf->start_slice);
yading@10 509 if(mpi->flags&MP_IMGFLAG_DRAW_CALLBACK)
yading@10 510 if(vf->start_slice) vf->start_slice(vf,mpi);
yading@10 511 if(!(mpi->flags&MP_IMGFLAG_TYPE_DISPLAYED)){
yading@10 512 ff_mp_msg(MSGT_DECVIDEO,MSGL_V,"*** [%s] %s%s mp_image_t, %dx%dx%dbpp %s %s, %d bytes\n",
yading@10 513 "NULL"/*vf->info->name*/,
yading@10 514 (mpi->type==MP_IMGTYPE_EXPORT)?"Exporting":
yading@10 515 ((mpi->flags&MP_IMGFLAG_DIRECT)?"Direct Rendering":"Allocating"),
yading@10 516 (mpi->flags&MP_IMGFLAG_DRAW_CALLBACK)?" (slices)":"",
yading@10 517 mpi->width,mpi->height,mpi->bpp,
yading@10 518 (mpi->flags&MP_IMGFLAG_YUV)?"YUV":((mpi->flags&MP_IMGFLAG_SWAPPED)?"BGR":"RGB"),
yading@10 519 (mpi->flags&MP_IMGFLAG_PLANAR)?"planar":"packed",
yading@10 520 mpi->bpp*mpi->width*mpi->height/8);
yading@10 521 ff_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",
yading@10 522 mpi->imgfmt, mpi->planes[0], mpi->planes[1], mpi->planes[2],
yading@10 523 mpi->stride[0], mpi->stride[1], mpi->stride[2],
yading@10 524 mpi->chroma_width, mpi->chroma_height, mpi->chroma_x_shift, mpi->chroma_y_shift);
yading@10 525 mpi->flags|=MP_IMGFLAG_TYPE_DISPLAYED;
yading@10 526 }
yading@10 527
yading@10 528 mpi->qscale = NULL;
yading@10 529 mpi->usage_count++;
yading@10 530 }
yading@10 531 // printf("\rVF_MPI: %p %p %p %d %d %d \n",
yading@10 532 // mpi->planes[0],mpi->planes[1],mpi->planes[2],
yading@10 533 // mpi->stride[0],mpi->stride[1],mpi->stride[2]);
yading@10 534 return mpi;
yading@10 535 }
yading@10 536
yading@10 537 static void dummy_free(void *opaque, uint8_t *data){}
yading@10 538
yading@10 539 int ff_vf_next_put_image(struct vf_instance *vf,mp_image_t *mpi, double pts){
yading@10 540 MPContext *m= (MPContext*)(((uint8_t*)vf) - offsetof(MPContext, vf));
yading@10 541 AVFilterLink *outlink = m->avfctx->outputs[0];
yading@10 542 AVFrame *picref = av_frame_alloc();
yading@10 543 int i;
yading@10 544
yading@10 545 av_assert0(vf->next);
yading@10 546
yading@10 547 av_log(m->avfctx, AV_LOG_DEBUG, "ff_vf_next_put_image\n");
yading@10 548
yading@10 549 if (!picref)
yading@10 550 goto fail;
yading@10 551
yading@10 552 picref->width = mpi->w;
yading@10 553 picref->height = mpi->h;
yading@10 554
yading@10 555 picref->type = AVMEDIA_TYPE_VIDEO;
yading@10 556
yading@10 557 for(i=0; conversion_map[i].fmt && mpi->imgfmt != conversion_map[i].fmt; i++);
yading@10 558 picref->format = conversion_map[i].pix_fmt;
yading@10 559
yading@10 560 memcpy(picref->linesize, mpi->stride, FFMIN(sizeof(picref->linesize), sizeof(mpi->stride)));
yading@10 561
yading@10 562 for(i=0; i<4 && mpi->stride[i]; i++){
yading@10 563 picref->data[i] = mpi->planes[i];
yading@10 564 }
yading@10 565
yading@10 566 if(pts != MP_NOPTS_VALUE)
yading@10 567 picref->pts= pts * av_q2d(outlink->time_base);
yading@10 568
yading@10 569 if(1) { // mp buffers are currently unsupported in libavfilter, we thus must copy
yading@10 570 AVFrame *tofree = picref;
yading@10 571 picref = av_frame_clone(picref);
yading@10 572 av_frame_free(&tofree);
yading@10 573 }
yading@10 574
yading@10 575 ff_filter_frame(outlink, picref);
yading@10 576 m->frame_returned++;
yading@10 577
yading@10 578 return 1;
yading@10 579 fail:
yading@10 580 av_frame_free(&picref);
yading@10 581 return 0;
yading@10 582 }
yading@10 583
yading@10 584 int ff_vf_next_config(struct vf_instance *vf,
yading@10 585 int width, int height, int d_width, int d_height,
yading@10 586 unsigned int voflags, unsigned int outfmt){
yading@10 587
yading@10 588 av_assert0(width>0 && height>0);
yading@10 589 vf->next->w = width; vf->next->h = height;
yading@10 590
yading@10 591 return 1;
yading@10 592 #if 0
yading@10 593 int flags=vf->next->query_format(vf->next,outfmt);
yading@10 594 if(!flags){
yading@10 595 // hmm. colorspace mismatch!!!
yading@10 596 //this is fatal for us ATM
yading@10 597 return 0;
yading@10 598 }
yading@10 599 ff_mp_msg(MSGT_VFILTER,MSGL_V,"REQ: flags=0x%X req=0x%X \n",flags,vf->default_reqs);
yading@10 600 miss=vf->default_reqs - (flags&vf->default_reqs);
yading@10 601 if(miss&VFCAP_ACCEPT_STRIDE){
yading@10 602 // vf requires stride support but vf->next doesn't support it!
yading@10 603 // let's insert the 'expand' filter, it does the job for us:
yading@10 604 vf_instance_t* vf2=vf_open_filter(vf->next,"expand",NULL);
yading@10 605 if(!vf2) return 0; // shouldn't happen!
yading@10 606 vf->next=vf2;
yading@10 607 }
yading@10 608 vf->next->w = width; vf->next->h = height;
yading@10 609 return 1;
yading@10 610 #endif
yading@10 611 }
yading@10 612
yading@10 613 int ff_vf_next_control(struct vf_instance *vf, int request, void* data){
yading@10 614 MPContext *m= (MPContext*)(((uint8_t*)vf) - offsetof(MPContext, vf));
yading@10 615 av_log(m->avfctx, AV_LOG_DEBUG, "Received control %d\n", request);
yading@10 616 return 0;
yading@10 617 }
yading@10 618
yading@10 619 static int vf_default_query_format(struct vf_instance *vf, unsigned int fmt){
yading@10 620 MPContext *m= (MPContext*)(((uint8_t*)vf) - offsetof(MPContext, vf));
yading@10 621 int i;
yading@10 622 av_log(m->avfctx, AV_LOG_DEBUG, "query %X\n", fmt);
yading@10 623
yading@10 624 for(i=0; conversion_map[i].fmt; i++){
yading@10 625 if(fmt==conversion_map[i].fmt)
yading@10 626 return 1; //we suport all
yading@10 627 }
yading@10 628 return 0;
yading@10 629 }
yading@10 630
yading@10 631
yading@10 632 static av_cold int init(AVFilterContext *ctx)
yading@10 633 {
yading@10 634 MPContext *m = ctx->priv;
yading@10 635 int cpu_flags = av_get_cpu_flags();
yading@10 636 char name[256];
yading@10 637 const char *args;
yading@10 638 int i;
yading@10 639
yading@10 640 ff_gCpuCaps.hasMMX = cpu_flags & AV_CPU_FLAG_MMX;
yading@10 641 ff_gCpuCaps.hasMMX2 = cpu_flags & AV_CPU_FLAG_MMX2;
yading@10 642 ff_gCpuCaps.hasSSE = cpu_flags & AV_CPU_FLAG_SSE;
yading@10 643 ff_gCpuCaps.hasSSE2 = cpu_flags & AV_CPU_FLAG_SSE2;
yading@10 644 ff_gCpuCaps.hasSSE3 = cpu_flags & AV_CPU_FLAG_SSE3;
yading@10 645 ff_gCpuCaps.hasSSSE3 = cpu_flags & AV_CPU_FLAG_SSSE3;
yading@10 646 ff_gCpuCaps.hasSSE4 = cpu_flags & AV_CPU_FLAG_SSE4;
yading@10 647 ff_gCpuCaps.hasSSE42 = cpu_flags & AV_CPU_FLAG_SSE42;
yading@10 648 ff_gCpuCaps.hasAVX = cpu_flags & AV_CPU_FLAG_AVX;
yading@10 649 ff_gCpuCaps.has3DNow = cpu_flags & AV_CPU_FLAG_3DNOW;
yading@10 650 ff_gCpuCaps.has3DNowExt = cpu_flags & AV_CPU_FLAG_3DNOWEXT;
yading@10 651
yading@10 652 m->avfctx= ctx;
yading@10 653
yading@10 654 args = m->filter;
yading@10 655 if(!args || 1!=sscanf(args, "%255[^:=]", name)){
yading@10 656 av_log(ctx, AV_LOG_ERROR, "Invalid parameter.\n");
yading@10 657 return AVERROR(EINVAL);
yading@10 658 }
yading@10 659 args += strlen(name);
yading@10 660 if (args[0] == '=')
yading@10 661 args++;
yading@10 662
yading@10 663 for(i=0; ;i++){
yading@10 664 if(!filters[i] || !strcmp(name, filters[i]->name))
yading@10 665 break;
yading@10 666 }
yading@10 667
yading@10 668 if(!filters[i]){
yading@10 669 av_log(ctx, AV_LOG_ERROR, "Unknown filter %s\n", name);
yading@10 670 return AVERROR(EINVAL);
yading@10 671 }
yading@10 672
yading@10 673 av_log(ctx, AV_LOG_WARNING,
yading@10 674 "'%s' is a wrapped MPlayer filter (libmpcodecs). This filter may be removed\n"
yading@10 675 "once it has been ported to a native libavfilter.\n", name);
yading@10 676
yading@10 677 memset(&m->vf,0,sizeof(m->vf));
yading@10 678 m->vf.info= filters[i];
yading@10 679
yading@10 680 m->vf.next = &m->next_vf;
yading@10 681 m->vf.put_image = ff_vf_next_put_image;
yading@10 682 m->vf.config = ff_vf_next_config;
yading@10 683 m->vf.query_format= vf_default_query_format;
yading@10 684 m->vf.control = ff_vf_next_control;
yading@10 685 m->vf.default_caps=VFCAP_ACCEPT_STRIDE;
yading@10 686 m->vf.default_reqs=0;
yading@10 687 if(m->vf.info->opts)
yading@10 688 av_log(ctx, AV_LOG_ERROR, "opts / m_struct_set is unsupported\n");
yading@10 689 #if 0
yading@10 690 if(vf->info->opts) { // vf_vo get some special argument
yading@10 691 const m_struct_t* st = vf->info->opts;
yading@10 692 void* vf_priv = m_struct_alloc(st);
yading@10 693 int n;
yading@10 694 for(n = 0 ; args && args[2*n] ; n++)
yading@10 695 m_struct_set(st,vf_priv,args[2*n],args[2*n+1]);
yading@10 696 vf->priv = vf_priv;
yading@10 697 args = NULL;
yading@10 698 } else // Otherwise we should have the '_oldargs_'
yading@10 699 if(args && !strcmp(args[0],"_oldargs_"))
yading@10 700 args = (char**)args[1];
yading@10 701 else
yading@10 702 args = NULL;
yading@10 703 #endif
yading@10 704 if(m->vf.info->vf_open(&m->vf, (char*)args)<=0){
yading@10 705 av_log(ctx, AV_LOG_ERROR, "vf_open() of %s with arg=%s failed\n", name, args);
yading@10 706 return -1;
yading@10 707 }
yading@10 708
yading@10 709 return 0;
yading@10 710 }
yading@10 711
yading@10 712 static av_cold void uninit(AVFilterContext *ctx)
yading@10 713 {
yading@10 714 MPContext *m = ctx->priv;
yading@10 715 vf_instance_t *vf = &m->vf;
yading@10 716
yading@10 717 while(vf){
yading@10 718 vf_instance_t *next = vf->next;
yading@10 719 if(vf->uninit)
yading@10 720 vf->uninit(vf);
yading@10 721 ff_free_mp_image(vf->imgctx.static_images[0]);
yading@10 722 ff_free_mp_image(vf->imgctx.static_images[1]);
yading@10 723 ff_free_mp_image(vf->imgctx.temp_images[0]);
yading@10 724 ff_free_mp_image(vf->imgctx.export_images[0]);
yading@10 725 vf = next;
yading@10 726 }
yading@10 727 }
yading@10 728
yading@10 729 static int query_formats(AVFilterContext *ctx)
yading@10 730 {
yading@10 731 AVFilterFormats *avfmts=NULL;
yading@10 732 MPContext *m = ctx->priv;
yading@10 733 enum AVPixelFormat lastpixfmt = AV_PIX_FMT_NONE;
yading@10 734 int i;
yading@10 735
yading@10 736 for(i=0; conversion_map[i].fmt; i++){
yading@10 737 av_log(ctx, AV_LOG_DEBUG, "query: %X\n", conversion_map[i].fmt);
yading@10 738 if(m->vf.query_format(&m->vf, conversion_map[i].fmt)){
yading@10 739 av_log(ctx, AV_LOG_DEBUG, "supported,adding\n");
yading@10 740 if (conversion_map[i].pix_fmt != lastpixfmt) {
yading@10 741 ff_add_format(&avfmts, conversion_map[i].pix_fmt);
yading@10 742 lastpixfmt = conversion_map[i].pix_fmt;
yading@10 743 }
yading@10 744 }
yading@10 745 }
yading@10 746
yading@10 747 if (!avfmts)
yading@10 748 return -1;
yading@10 749
yading@10 750 //We assume all allowed input formats are also allowed output formats
yading@10 751 ff_set_common_formats(ctx, avfmts);
yading@10 752 return 0;
yading@10 753 }
yading@10 754
yading@10 755 static int config_inprops(AVFilterLink *inlink)
yading@10 756 {
yading@10 757 MPContext *m = inlink->dst->priv;
yading@10 758 int i;
yading@10 759 for(i=0; conversion_map[i].fmt && conversion_map[i].pix_fmt != inlink->format; i++);
yading@10 760
yading@10 761 av_assert0(conversion_map[i].fmt && inlink->w && inlink->h);
yading@10 762
yading@10 763 m->vf.fmt.have_configured = 1;
yading@10 764 m->vf.fmt.orig_height = inlink->h;
yading@10 765 m->vf.fmt.orig_width = inlink->w;
yading@10 766 m->vf.fmt.orig_fmt = conversion_map[i].fmt;
yading@10 767
yading@10 768 if(m->vf.config(&m->vf, inlink->w, inlink->h, inlink->w, inlink->h, 0, conversion_map[i].fmt)<=0)
yading@10 769 return -1;
yading@10 770
yading@10 771 return 0;
yading@10 772 }
yading@10 773
yading@10 774 static int config_outprops(AVFilterLink *outlink)
yading@10 775 {
yading@10 776 MPContext *m = outlink->src->priv;
yading@10 777
yading@10 778 outlink->w = m->next_vf.w;
yading@10 779 outlink->h = m->next_vf.h;
yading@10 780
yading@10 781 return 0;
yading@10 782 }
yading@10 783
yading@10 784 static int request_frame(AVFilterLink *outlink)
yading@10 785 {
yading@10 786 MPContext *m = outlink->src->priv;
yading@10 787 int ret;
yading@10 788
yading@10 789 av_log(m->avfctx, AV_LOG_DEBUG, "mp request_frame\n");
yading@10 790
yading@10 791 for(m->frame_returned=0; !m->frame_returned;){
yading@10 792 ret=ff_request_frame(outlink->src->inputs[0]);
yading@10 793 if(ret<0)
yading@10 794 break;
yading@10 795 }
yading@10 796
yading@10 797 av_log(m->avfctx, AV_LOG_DEBUG, "mp request_frame ret=%d\n", ret);
yading@10 798 return ret;
yading@10 799 }
yading@10 800
yading@10 801 static int filter_frame(AVFilterLink *inlink, AVFrame *inpic)
yading@10 802 {
yading@10 803 MPContext *m = inlink->dst->priv;
yading@10 804 int i;
yading@10 805 double pts= MP_NOPTS_VALUE;
yading@10 806 mp_image_t* mpi = ff_new_mp_image(inpic->width, inpic->height);
yading@10 807
yading@10 808 if(inpic->pts != AV_NOPTS_VALUE)
yading@10 809 pts= inpic->pts / av_q2d(inlink->time_base);
yading@10 810
yading@10 811 for(i=0; conversion_map[i].fmt && conversion_map[i].pix_fmt != inlink->format; i++);
yading@10 812 ff_mp_image_setfmt(mpi,conversion_map[i].fmt);
yading@10 813
yading@10 814 memcpy(mpi->planes, inpic->data, FFMIN(sizeof(inpic->data) , sizeof(mpi->planes)));
yading@10 815 memcpy(mpi->stride, inpic->linesize, FFMIN(sizeof(inpic->linesize), sizeof(mpi->stride)));
yading@10 816
yading@10 817 //FIXME pass interleced & tff flags around
yading@10 818
yading@10 819 // mpi->flags|=MP_IMGFLAG_ALLOCATED; ?
yading@10 820 mpi->flags |= MP_IMGFLAG_READABLE;
yading@10 821 if(!av_frame_is_writable(inpic))
yading@10 822 mpi->flags |= MP_IMGFLAG_PRESERVE;
yading@10 823 if(m->vf.put_image(&m->vf, mpi, pts) == 0){
yading@10 824 av_log(m->avfctx, AV_LOG_DEBUG, "put_image() says skip\n");
yading@10 825 }else{
yading@10 826 av_frame_free(&inpic);
yading@10 827 }
yading@10 828 ff_free_mp_image(mpi);
yading@10 829 return 0;
yading@10 830 }
yading@10 831
yading@10 832 static const AVFilterPad mp_inputs[] = {
yading@10 833 {
yading@10 834 .name = "default",
yading@10 835 .type = AVMEDIA_TYPE_VIDEO,
yading@10 836 .filter_frame = filter_frame,
yading@10 837 .config_props = config_inprops,
yading@10 838 },
yading@10 839 { NULL }
yading@10 840 };
yading@10 841
yading@10 842 static const AVFilterPad mp_outputs[] = {
yading@10 843 {
yading@10 844 .name = "default",
yading@10 845 .type = AVMEDIA_TYPE_VIDEO,
yading@10 846 .request_frame = request_frame,
yading@10 847 .config_props = config_outprops,
yading@10 848 },
yading@10 849 { NULL }
yading@10 850 };
yading@10 851
yading@10 852 AVFilter avfilter_vf_mp = {
yading@10 853 .name = "mp",
yading@10 854 .description = NULL_IF_CONFIG_SMALL("Apply a libmpcodecs filter to the input video."),
yading@10 855 .init = init,
yading@10 856 .uninit = uninit,
yading@10 857 .priv_size = sizeof(MPContext),
yading@10 858 .query_formats = query_formats,
yading@10 859 .inputs = mp_inputs,
yading@10 860 .outputs = mp_outputs,
yading@10 861 .priv_class = &mp_class,
yading@10 862 };