Mercurial > mplayer.hg
annotate libmpcodecs/vd_ffmpeg.c @ 8411:5630321c8026
ffmpeg cleanup/messup support
fixing a typo in vstats (found by Markus Liebl <lieblm at web dot de>)
printing qscale statistics
author | michael |
---|---|
date | Mon, 09 Dec 2002 09:12:23 +0000 |
parents | 03ee1d7deba5 |
children | 2670aec42989 |
rev | line source |
---|---|
4952 | 1 #include <stdio.h> |
2 #include <stdlib.h> | |
3 #include <assert.h> | |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
4 #include <time.h> |
4952 | 5 |
6 #include "config.h" | |
7 #include "mp_msg.h" | |
8 #include "help_mp.h" | |
9 | |
10 #ifdef USE_LIBAVCODEC | |
11 | |
12 #include "bswap.h" | |
13 | |
14 #include "vd_internal.h" | |
15 | |
16 static vd_info_t info = { | |
17 "FFmpeg's libavcodec codec family", | |
18 "ffmpeg", | |
19 "A'rpi", | |
7388 | 20 "A'rpi, Michael, Alex", |
21 "native codecs (http://ffmpeg.sf.net/)" | |
4952 | 22 }; |
23 | |
24 LIBVD_EXTERN(ffmpeg) | |
25 | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
26 #include "../postproc/rgb2rgb.h" |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
27 |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
28 |
4952 | 29 #ifdef USE_LIBAVCODEC_SO |
7004 | 30 #include <ffmpeg/avcodec.h> |
4952 | 31 #else |
32 #include "libavcodec/avcodec.h" | |
33 #endif | |
34 | |
8339 | 35 #if LIBAVCODEC_BUILD < 4641 |
8340 | 36 #error we dont support libavcodec prior to build 4641, get the latest libavcodec CVS |
8339 | 37 #endif |
38 | |
4952 | 39 int avcodec_inited=0; |
40 | |
8130 | 41 #if defined(FF_POSTPROCESS) && defined(MBR) |
5517 | 42 int quant_store[MBR+1][MBC+1]; |
43 #endif | |
44 | |
5280 | 45 typedef struct { |
46 AVCodecContext *avctx; | |
8339 | 47 AVVideoFrame *pic; |
8237 | 48 float last_aspect; |
5482 | 49 int do_slices; |
6734 | 50 int do_dr1; |
5482 | 51 int vo_inited; |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
52 int convert; |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
53 int best_csp; |
8339 | 54 int b_age; |
55 int ip_age[2]; | |
8411 | 56 int qp_stat[32]; |
57 double qp_sum; | |
58 double inv_qp_sum; | |
5280 | 59 } vd_ffmpeg_ctx; |
60 | |
4952 | 61 //#ifdef FF_POSTPROCESS |
62 //unsigned int lavc_pp=0; | |
63 //#endif | |
64 | |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
65 #include "cfgparser.h" |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
66 |
8339 | 67 static int get_buffer(AVCodecContext *avctx, AVVideoFrame *pic); |
68 static void release_buffer(AVCodecContext *avctx, AVVideoFrame *pic); | |
6739 | 69 |
7722 | 70 static int lavc_param_workaround_bugs= FF_BUG_AUTODETECT; |
71 static int lavc_param_error_resilience=2; | |
72 static int lavc_param_error_concealment=3; | |
6355 | 73 static int lavc_param_gray=0; |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
74 static int lavc_param_vstats=0; |
7564 | 75 static int lavc_param_idct_algo=0; |
8341 | 76 static int lavc_param_debug=0; |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
77 |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
78 struct config lavc_decode_opts_conf[]={ |
7564 | 79 {"bug", &lavc_param_workaround_bugs, CONF_TYPE_INT, CONF_RANGE, -1, 99, NULL}, |
7722 | 80 {"er", &lavc_param_error_resilience, CONF_TYPE_INT, CONF_RANGE, 0, 99, NULL}, |
6355 | 81 {"gray", &lavc_param_gray, CONF_TYPE_FLAG, 0, 0, CODEC_FLAG_PART, NULL}, |
7564 | 82 {"idct", &lavc_param_idct_algo, CONF_TYPE_INT, CONF_RANGE, 0, 99, NULL}, |
7722 | 83 {"ec", &lavc_param_error_concealment, CONF_TYPE_INT, CONF_RANGE, 0, 99, NULL}, |
6869 | 84 {"vstats", &lavc_param_vstats, CONF_TYPE_FLAG, 0, 0, 1, NULL}, |
8341 | 85 #if LIBAVCODEC_BUILD >= 4642 |
86 {"debug", &lavc_param_debug, CONF_TYPE_INT, CONF_RANGE, 0, 9999999, NULL}, | |
87 #endif | |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
88 {NULL, NULL, 0, 0, 0, 0, NULL} |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
89 }; |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
90 |
4952 | 91 // to set/get/query special features/parameters |
92 static int control(sh_video_t *sh,int cmd,void* arg,...){ | |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
93 vd_ffmpeg_ctx *ctx = sh->context; |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
94 AVCodecContext *avctx = ctx->avctx; |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
95 switch(cmd){ |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
96 case VDCTRL_QUERY_FORMAT: |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
97 if( (*((int*)arg)) == ctx->best_csp ) return CONTROL_TRUE;//supported |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
98 // possible conversions: |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
99 switch( (*((int*)arg)) ){ |
6739 | 100 case IMGFMT_YV12: |
101 case IMGFMT_IYUV: | |
102 case IMGFMT_I420: | |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
103 // "converted" using pointer/stride modification |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
104 if(avctx->pix_fmt==PIX_FMT_YUV420P) return CONTROL_TRUE;// u/v swap |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
105 if(avctx->pix_fmt==PIX_FMT_YUV422P) return CONTROL_TRUE;// half stride |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
106 break; |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
107 #if 1 |
6739 | 108 case IMGFMT_YUY2: |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
109 // converted using yuv422ptoyuy2() |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
110 if(avctx->pix_fmt==PIX_FMT_YUV422P) return CONTROL_TRUE; |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
111 break; |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
112 } |
6681 | 113 #endif |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
114 return CONTROL_FALSE; |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
115 } |
4952 | 116 return CONTROL_UNKNOWN; |
117 } | |
118 | |
119 // init driver | |
120 static int init(sh_video_t *sh){ | |
5280 | 121 AVCodecContext *avctx; |
122 vd_ffmpeg_ctx *ctx; | |
5457
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
123 AVCodec *lavc_codec; |
4952 | 124 |
125 if(!avcodec_inited){ | |
126 avcodec_init(); | |
127 avcodec_register_all(); | |
128 avcodec_inited=1; | |
129 } | |
5280 | 130 |
131 ctx = sh->context = malloc(sizeof(vd_ffmpeg_ctx)); | |
132 if (!ctx) | |
133 return(0); | |
134 memset(ctx, 0, sizeof(vd_ffmpeg_ctx)); | |
4952 | 135 |
5457
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
136 lavc_codec = (AVCodec *)avcodec_find_decoder_by_name(sh->codec->dll); |
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
137 if(!lavc_codec){ |
4952 | 138 mp_msg(MSGT_DECVIDEO,MSGL_ERR,MSGTR_MissingLAVCcodec,sh->codec->dll); |
139 return 0; | |
140 } | |
5482 | 141 |
5494
1c45b1484ffb
i just fixed b-frames & slices in libavcodec :) ... iam too tired for benchmarks now ...
michael
parents:
5482
diff
changeset
|
142 if(vd_use_slices && lavc_codec->capabilities&CODEC_CAP_DRAW_HORIZ_BAND) |
1c45b1484ffb
i just fixed b-frames & slices in libavcodec :) ... iam too tired for benchmarks now ...
michael
parents:
5482
diff
changeset
|
143 ctx->do_slices=1; |
6734 | 144 |
6735 | 145 if(lavc_codec->capabilities&CODEC_CAP_DR1) |
6734 | 146 ctx->do_dr1=1; |
8200
3ae71e4f2858
disable dr1 for huffyuv (422P & stride hack & dr1 -> bad )
michael
parents:
8190
diff
changeset
|
147 //XXX:FIXME:HACK:UGLY 422P with direct rendering is buggy cuz of that chroma stride trick ... |
3ae71e4f2858
disable dr1 for huffyuv (422P & stride hack & dr1 -> bad )
michael
parents:
8190
diff
changeset
|
148 if(sh->format == mmioFOURCC('H','F','Y','U')) |
3ae71e4f2858
disable dr1 for huffyuv (422P & stride hack & dr1 -> bad )
michael
parents:
8190
diff
changeset
|
149 ctx->do_dr1=0; |
8339 | 150 ctx->b_age= ctx->ip_age[0]= ctx->ip_age[1]= 256*256*256*64; |
7444 | 151 |
8339 | 152 ctx->pic = avcodec_alloc_picture(); |
7444 | 153 ctx->avctx = avcodec_alloc_context(); |
5280 | 154 avctx = ctx->avctx; |
6739 | 155 |
156 if(ctx->do_dr1){ | |
8339 | 157 avctx->flags|= CODEC_FLAG_EMU_EDGE; |
158 avctx->get_buffer= get_buffer; | |
159 avctx->release_buffer= release_buffer; | |
6739 | 160 } |
7303 | 161 |
162 #ifdef CODEC_FLAG_NOT_TRUNCATED | |
163 avctx->flags|= CODEC_FLAG_NOT_TRUNCATED; | |
164 #endif | |
4952 | 165 |
5280 | 166 avctx->width = sh->disp_w; |
167 avctx->height= sh->disp_h; | |
6265
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
168 avctx->workaround_bugs= lavc_param_workaround_bugs; |
f49ec39ab0c6
workaround bugs & error resilience ffmpeg decoder options
michael
parents:
5940
diff
changeset
|
169 avctx->error_resilience= lavc_param_error_resilience; |
6355 | 170 if(lavc_param_gray) avctx->flags|= CODEC_FLAG_GRAY; |
7512 | 171 avctx->fourcc= sh->format; |
7564 | 172 avctx->idct_algo= lavc_param_idct_algo; |
7722 | 173 avctx->error_concealment= lavc_param_error_concealment; |
8341 | 174 #if LIBAVCODEC_BUILD >= 4642 |
175 avctx->debug= lavc_param_debug; | |
176 #endif | |
5280 | 177 mp_dbg(MSGT_DECVIDEO,MSGL_DBG2,"libavcodec.size: %d x %d\n",avctx->width,avctx->height); |
5939 | 178 /* AVRn stores huffman table in AVI header */ |
179 /* Pegasus MJPEG stores it also in AVI header, but it uses the common | |
180 MJPG fourcc :( */ | |
5940 | 181 if (sh->bih && (sh->bih->biSize != sizeof(BITMAPINFOHEADER)) && |
5939 | 182 (sh->format == mmioFOURCC('A','V','R','n') || |
183 sh->format == mmioFOURCC('M','J','P','G'))) | |
184 { | |
185 avctx->flags |= CODEC_FLAG_EXTERN_HUFF; | |
186 avctx->extradata_size = sh->bih->biSize-sizeof(BITMAPINFOHEADER); | |
187 avctx->extradata = malloc(avctx->extradata_size); | |
188 memcpy(avctx->extradata, sh->bih+sizeof(BITMAPINFOHEADER), | |
189 avctx->extradata_size); | |
190 | |
191 #if 0 | |
192 { | |
193 int x; | |
194 uint8_t *p = avctx->extradata; | |
195 | |
196 for (x=0; x<avctx->extradata_size; x++) | |
197 printf("[%x] ", p[x]); | |
198 printf("\n"); | |
199 } | |
200 #endif | |
201 } | |
7126 | 202 if( sh->format == mmioFOURCC('R', 'V', '1', '0') |
203 || sh->format == mmioFOURCC('R', 'V', '1', '3')){ | |
204 avctx->extradata_size= 8; | |
7573 | 205 avctx->extradata = malloc(avctx->extradata_size); |
7574
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
206 if(sh->bih->biSize!=sizeof(*sh->bih)+8){ |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
207 /* only 1 packet per frame & sub_id from fourcc */ |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
208 ((uint32_t*)avctx->extradata)[0] = 0; |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
209 avctx->sub_id= |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
210 ((uint32_t*)avctx->extradata)[1] = |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
211 (sh->format == mmioFOURCC('R', 'V', '1', '3')) ? 0x10003001 : 0x10000000; |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
212 } else { |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
213 /* has extra slice header (demux_rm or rm->avi streamcopy) */ |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
214 unsigned int* extrahdr=(unsigned int*)(sh->bih+1); |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
215 ((uint32_t*)avctx->extradata)[0] = extrahdr[0]; |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
216 avctx->sub_id= |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
217 ((uint32_t*)avctx->extradata)[1] = extrahdr[1]; |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
218 } |
7573 | 219 |
7126 | 220 // printf("%X %X %d %d\n", extrahdr[0], extrahdr[1]); |
221 } | |
7736
b81b0ab0aa40
put M4S2 & MP4S headers in avctx->extradata (in the unlikely case that they arent missing completly)
michael
parents:
7722
diff
changeset
|
222 if (sh->bih && (sh->bih->biSize != sizeof(BITMAPINFOHEADER)) && |
b81b0ab0aa40
put M4S2 & MP4S headers in avctx->extradata (in the unlikely case that they arent missing completly)
michael
parents:
7722
diff
changeset
|
223 (sh->format == mmioFOURCC('M','4','S','2') || |
7928 | 224 sh->format == mmioFOURCC('M','P','4','S') || |
8190 | 225 sh->format == mmioFOURCC('H','F','Y','U') || |
7928 | 226 sh->format == mmioFOURCC('W','M','V','2') |
227 )) | |
7736
b81b0ab0aa40
put M4S2 & MP4S headers in avctx->extradata (in the unlikely case that they arent missing completly)
michael
parents:
7722
diff
changeset
|
228 { |
b81b0ab0aa40
put M4S2 & MP4S headers in avctx->extradata (in the unlikely case that they arent missing completly)
michael
parents:
7722
diff
changeset
|
229 avctx->extradata_size = sh->bih->biSize-sizeof(BITMAPINFOHEADER); |
b81b0ab0aa40
put M4S2 & MP4S headers in avctx->extradata (in the unlikely case that they arent missing completly)
michael
parents:
7722
diff
changeset
|
230 avctx->extradata = malloc(avctx->extradata_size); |
b81b0ab0aa40
put M4S2 & MP4S headers in avctx->extradata (in the unlikely case that they arent missing completly)
michael
parents:
7722
diff
changeset
|
231 memcpy(avctx->extradata, sh->bih+1, avctx->extradata_size); |
b81b0ab0aa40
put M4S2 & MP4S headers in avctx->extradata (in the unlikely case that they arent missing completly)
michael
parents:
7722
diff
changeset
|
232 } |
8190 | 233 |
8264 | 234 if(sh->bih) |
235 avctx->bits_per_sample= sh->bih->biBitCount; | |
7126 | 236 |
4952 | 237 /* open it */ |
5457
f248c9e86423
config vo only if aspect really changed and width&&height isn't changed (if w||h changes, we set it later)
alex
parents:
5331
diff
changeset
|
238 if (avcodec_open(avctx, lavc_codec) < 0) { |
4952 | 239 mp_msg(MSGT_DECVIDEO,MSGL_ERR, MSGTR_CantOpenCodec); |
240 return 0; | |
241 } | |
242 mp_msg(MSGT_DECVIDEO,MSGL_V,"INFO: libavcodec init OK!\n"); | |
5482 | 243 ctx->last_aspect=-3; |
5510 | 244 return 1; //mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h,IMGFMT_YV12); |
4952 | 245 } |
246 | |
247 // uninit driver | |
248 static void uninit(sh_video_t *sh){ | |
5280 | 249 vd_ffmpeg_ctx *ctx = sh->context; |
250 AVCodecContext *avctx = ctx->avctx; | |
8411 | 251 |
252 if(lavc_param_vstats){ | |
253 int i; | |
254 for(i=1; i<32; i++){ | |
255 printf("QP: %d, count: %d\n", i, ctx->qp_stat[i]); | |
256 } | |
257 printf("Arithmetic mean of QP: %2.4f, Harmonic mean of QP: %2.4f\n", | |
258 ctx->qp_sum / avctx->coded_picture->coded_picture_number, | |
259 1.0/(ctx->inv_qp_sum / avctx->coded_picture->coded_picture_number) | |
260 ); | |
261 } | |
5280 | 262 |
263 if (avcodec_close(avctx) < 0) | |
4952 | 264 mp_msg(MSGT_DECVIDEO,MSGL_ERR, MSGTR_CantCloseCodec); |
5939 | 265 |
266 if (avctx->extradata_size) | |
267 free(avctx->extradata); | |
7573 | 268 avctx->extradata=NULL; |
269 if(avctx->slice_offset!=NULL) | |
270 free(avctx->slice_offset); | |
271 avctx->slice_offset=NULL; | |
5939 | 272 |
5280 | 273 if (avctx) |
274 free(avctx); | |
8339 | 275 if (ctx->pic) |
276 free(ctx->pic); | |
5280 | 277 if (ctx) |
278 free(ctx); | |
4952 | 279 } |
280 | |
5482 | 281 static void draw_slice(struct AVCodecContext *s, |
282 UINT8 **src, int linesize, | |
283 int y, int width, int height){ | |
6710 | 284 sh_video_t * sh = s->opaque; |
5482 | 285 int stride[3]; |
6740 | 286 int start=0, i; |
287 int skip_stride= (s->width+15)>>4; | |
8339 | 288 UINT8 *skip= &s->coded_picture->mbskip_table[(y>>4)*skip_stride]; |
289 int threshold= s->coded_picture->age; | |
5482 | 290 |
291 stride[0]=linesize; | |
8339 | 292 if(s->coded_picture->linesize[1]){ |
293 stride[1]= s->coded_picture->linesize[1]; | |
294 stride[2]= s->coded_picture->linesize[2]; | |
295 }else | |
6740 | 296 stride[1]=stride[2]=stride[0]/2; |
297 #if 0 | |
298 if(s->pict_type!=B_TYPE){ | |
299 for(i=0; i*16<width+16; i++){ | |
300 if(i*16>=width || skip[i]>=threshold){ | |
301 if(start==i) start++; | |
302 else{ | |
303 UINT8 *src2[3]= {src[0] + start*16, | |
304 src[1] + start*8, | |
305 src[2] + start*8}; | |
306 //printf("%2d-%2d x %d\n", start, i, y); | |
307 mpcodecs_draw_slice (sh,src2, stride, (i-start)*16, height, start*16, y); | |
308 start= i+1; | |
309 } | |
310 } | |
311 } | |
312 }else | |
313 #endif | |
7220
e3ecccc7e505
warning fixes by Dominik Mierzejewski <dominik@rangers.eu.org>
arpi
parents:
7180
diff
changeset
|
314 mpcodecs_draw_slice (sh,src, stride, width, height, 0, y); |
5482 | 315 } |
4952 | 316 |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
317 static int init_vo(sh_video_t *sh){ |
5280 | 318 vd_ffmpeg_ctx *ctx = sh->context; |
319 AVCodecContext *avctx = ctx->avctx; | |
4952 | 320 |
8237 | 321 if (avctx->aspect_ratio != ctx->last_aspect || |
5482 | 322 avctx->width != sh->disp_w || |
323 avctx->height != sh->disp_h || | |
324 !ctx->vo_inited) | |
5280 | 325 { |
8237 | 326 mp_dbg(MSGT_DECVIDEO, MSGL_DBG2, "aspect_ratio: %d\n", avctx->aspect_ratio); |
327 sh->aspect = | |
328 ctx->last_aspect = avctx->aspect_ratio; | |
5482 | 329 sh->disp_w = avctx->width; |
330 sh->disp_h = avctx->height; | |
331 ctx->vo_inited=1; | |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
332 switch(avctx->pix_fmt){ |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
333 case PIX_FMT_YUV410P: ctx->best_csp=IMGFMT_YVU9;break; //svq1 |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
334 case PIX_FMT_YUV420P: ctx->best_csp=IMGFMT_YV12;break; //mpegs |
8190 | 335 case PIX_FMT_YUV422P: ctx->best_csp=IMGFMT_422P;break; //mjpeg / huffyuv |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
336 case PIX_FMT_YUV444P: ctx->best_csp=IMGFMT_444P;break; //??? |
7662 | 337 case PIX_FMT_YUV411P: ctx->best_csp=IMGFMT_411P;break; //dv ntsc |
8190 | 338 case PIX_FMT_YUV422: ctx->best_csp=IMGFMT_YUY2;break; //huffyuv perhaps in the future |
339 case PIX_FMT_BGR24 : ctx->best_csp=IMGFMT_BGR24;break; //huffyuv | |
340 case PIX_FMT_BGRA32: ctx->best_csp=IMGFMT_BGR32;break; //huffyuv | |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
341 default: |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
342 ctx->best_csp=0; |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
343 } |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
344 if (!mpcodecs_config_vo(sh,sh->disp_w,sh->disp_h, ctx->best_csp)) |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
345 return -1; |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
346 ctx->convert=(sh->codec->outfmt[sh->outfmtidx]==IMGFMT_YUY2 |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
347 && ctx->best_csp!=IMGFMT_YUY2); // yuv422p->yuy2 conversion |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
348 } |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
349 return 0; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
350 } |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
351 |
8339 | 352 static int get_buffer(AVCodecContext *avctx, AVVideoFrame *pic){ |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
353 sh_video_t * sh = avctx->opaque; |
6734 | 354 vd_ffmpeg_ctx *ctx = sh->context; |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
355 mp_image_t* mpi=NULL; |
6875 | 356 int flags= MP_IMGFLAG_ACCEPT_STRIDE | MP_IMGFLAG_PREFER_ALIGNED_STRIDE; |
6737 | 357 int type= MP_IMGTYPE_IPB; |
8339 | 358 int width= avctx->width; |
359 int height= avctx->height; | |
6738 | 360 int align=15; |
361 | |
362 if(avctx->pix_fmt == PIX_FMT_YUV410P) | |
363 align=63; //yes seriously, its really needed (16x16 chroma blocks in SVQ1 -> 64x64) | |
364 | |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
365 if(init_vo(sh)<0){ |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
366 printf("init_vo failed\n"); |
8339 | 367 |
368 avctx->get_buffer= avcodec_default_get_buffer; | |
369 avctx->release_buffer= avcodec_default_release_buffer; | |
370 return avctx->get_buffer(avctx, pic); | |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
371 } |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
372 |
8339 | 373 if(!pic->reference) |
6734 | 374 flags|=(!avctx->hurry_up && ctx->do_slices) ? |
375 MP_IMGFLAG_DRAW_CALLBACK:0; | |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
376 else |
6742
93bce3460e2a
fallback to slices, if dr1 fails (bug found by kabi)
michael
parents:
6740
diff
changeset
|
377 flags|= MP_IMGFLAG_PRESERVE|MP_IMGFLAG_READABLE |
93bce3460e2a
fallback to slices, if dr1 fails (bug found by kabi)
michael
parents:
6740
diff
changeset
|
378 | (ctx->do_slices ? MP_IMGFLAG_DRAW_CALLBACK : 0); |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
379 |
6737 | 380 if(avctx->has_b_frames){ |
381 type= MP_IMGTYPE_IPB; | |
382 }else{ | |
383 type= MP_IMGTYPE_IP; | |
384 } | |
385 mp_msg(MSGT_DECVIDEO,MSGL_DBG2, type== MP_IMGTYPE_IPB ? "using IPB\n" : "using IP\n"); | |
386 | |
387 mpi= mpcodecs_get_image(sh,type, flags, | |
6738 | 388 (width+align)&(~align), (height+align)&(~align)); |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
389 |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
390 // ok, lets see what did we get: |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
391 if( mpi->flags&MP_IMGFLAG_DRAW_CALLBACK && |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
392 !(mpi->flags&MP_IMGFLAG_DIRECT)){ |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
393 // nice, filter/vo likes draw_callback :) |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
394 avctx->draw_horiz_band= draw_slice; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
395 } else |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
396 avctx->draw_horiz_band= NULL; |
8339 | 397 pic->data[0]= mpi->planes[0]; |
398 pic->data[1]= mpi->planes[1]; | |
399 pic->data[2]= mpi->planes[2]; | |
7051
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
400 |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
401 assert(mpi->width >= ((width +align)&(~align))); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
402 assert(mpi->height >= ((height+align)&(~align))); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
403 assert(mpi->stride[0] >= mpi->width); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
404 if(mpi->imgfmt==IMGFMT_I420 || mpi->imgfmt==IMGFMT_YV12 || mpi->imgfmt==IMGFMT_IYUV){ |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
405 const int y_size= mpi->stride[0] * mpi->height; |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
406 const int c_size= mpi->stride[1] * mpi->chroma_height; |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
407 |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
408 assert(mpi->planes[0] > mpi->planes[1] || mpi->planes[0] + y_size <= mpi->planes[1]); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
409 assert(mpi->planes[0] > mpi->planes[2] || mpi->planes[0] + y_size <= mpi->planes[2]); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
410 assert(mpi->planes[1] > mpi->planes[0] || mpi->planes[1] + c_size <= mpi->planes[0]); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
411 assert(mpi->planes[1] > mpi->planes[2] || mpi->planes[1] + c_size <= mpi->planes[2]); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
412 assert(mpi->planes[2] > mpi->planes[0] || mpi->planes[2] + c_size <= mpi->planes[0]); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
413 assert(mpi->planes[2] > mpi->planes[1] || mpi->planes[2] + c_size <= mpi->planes[1]); |
d03fad6123a3
asserts to check buffer size and non overlapingness
michael
parents:
7004
diff
changeset
|
414 } |
6734 | 415 |
8339 | 416 /* Note, some (many) codecs in libavcodec must have stride1==stride2 && no changes between frames |
417 * lavc will check that and die with an error message, if its not true | |
418 */ | |
419 pic->linesize[0]= mpi->stride[0]; | |
420 pic->linesize[1]= mpi->stride[1]; | |
421 pic->linesize[2]= mpi->stride[2]; | |
6734 | 422 |
8339 | 423 pic->opaque = mpi; |
6742
93bce3460e2a
fallback to slices, if dr1 fails (bug found by kabi)
michael
parents:
6740
diff
changeset
|
424 //printf("%X\n", (int)mpi->planes[0]); |
6869 | 425 #if 0 |
426 if(mpi->flags&MP_IMGFLAG_DIRECT) | |
427 printf("D"); | |
428 else if(mpi->flags&MP_IMGFLAG_DRAW_CALLBACK) | |
429 printf("S"); | |
430 else | |
431 printf("."); | |
432 #endif | |
8339 | 433 if(pic->reference){ |
434 pic->age= ctx->ip_age[0]; | |
435 | |
436 ctx->ip_age[0]= ctx->ip_age[1]+1; | |
437 ctx->ip_age[1]= 1; | |
438 ctx->b_age++; | |
439 }else{ | |
440 pic->age= ctx->b_age; | |
441 | |
442 ctx->ip_age[0]++; | |
443 ctx->ip_age[1]++; | |
444 ctx->b_age=1; | |
445 } | |
8411 | 446 #if LIBAVCODEC_BUILD >= 4644 |
447 pic->type= FF_BUFFER_TYPE_USER; | |
448 #endif | |
7928 | 449 return 0; |
8339 | 450 } |
451 | |
452 static void release_buffer(struct AVCodecContext *avctx, AVVideoFrame *pic){ | |
453 int i; | |
8411 | 454 |
455 #if LIBAVCODEC_BUILD >= 4644 | |
456 assert(pic->type == FF_BUFFER_TYPE_USER); | |
457 #endif | |
458 | |
8339 | 459 for(i=0; i<4; i++){ |
460 pic->data[i]= NULL; | |
461 } | |
462 //printf("R%X %X\n", pic->linesize[0], pic->data[0]); | |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
463 } |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
464 |
7573 | 465 // copypaste from demux_real.c - it should match to get it working! |
466 //FIXME put into some header | |
467 typedef struct dp_hdr_s { | |
468 uint32_t chunks; // number of chunks | |
469 uint32_t timestamp; // timestamp from packet header | |
470 uint32_t len; // length of actual data | |
471 uint32_t chunktab; // offset to chunk offset array | |
472 } dp_hdr_t; | |
473 | |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
474 // decode a frame |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
475 static mp_image_t* decode(sh_video_t *sh,void* data,int len,int flags){ |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
476 int got_picture=0; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
477 int ret; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
478 vd_ffmpeg_ctx *ctx = sh->context; |
8339 | 479 AVVideoFrame *pic= ctx->pic; |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
480 AVCodecContext *avctx = ctx->avctx; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
481 mp_image_t* mpi=NULL; |
6737 | 482 int dr1= ctx->do_dr1; |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
483 |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
484 if(len<=0) return NULL; // skipped frame |
6734 | 485 |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
486 avctx->draw_horiz_band=NULL; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
487 avctx->opaque=sh; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
488 if(ctx->vo_inited && !ctx->convert && !(flags&3) && !dr1){ |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
489 mpi=mpcodecs_get_image(sh, MP_IMGTYPE_EXPORT, MP_IMGFLAG_PRESERVE | |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
490 (ctx->do_slices?MP_IMGFLAG_DRAW_CALLBACK:0), |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
491 sh->disp_w, sh->disp_h); |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
492 if(mpi && mpi->flags&MP_IMGFLAG_DRAW_CALLBACK){ |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
493 // vd core likes slices! |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
494 avctx->draw_horiz_band=draw_slice; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
495 } |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
496 } |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
497 |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
498 avctx->hurry_up=(flags&3)?((flags&2)?2:1):0; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
499 |
7574
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
500 // if(sh->ds->demuxer->type == DEMUXER_TYPE_REAL){ |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
501 if( sh->format == mmioFOURCC('R', 'V', '1', '0') |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
502 || sh->format == mmioFOURCC('R', 'V', '1', '3')) |
87f57e23e301
fixing RV10 streamcopy - detect packet format (simple single frame vs. slices
arpi
parents:
7573
diff
changeset
|
503 if(sh->bih->biSize==sizeof(*sh->bih)+8){ |
7573 | 504 int i; |
505 dp_hdr_t *hdr= (dp_hdr_t*)data; | |
506 | |
507 if(avctx->slice_offset==NULL) | |
508 avctx->slice_offset= malloc(sizeof(int)*1000); | |
509 | |
510 // for(i=0; i<25; i++) printf("%02X ", ((uint8_t*)data)[i]); | |
511 | |
512 avctx->slice_count= hdr->chunks+1; | |
513 for(i=0; i<avctx->slice_count; i++) | |
514 avctx->slice_offset[i]= ((uint32_t*)(data+hdr->chunktab))[2*i+1]; | |
7578 | 515 len=hdr->len; |
7573 | 516 data+= sizeof(dp_hdr_t); |
517 } | |
518 | |
8339 | 519 ret = avcodec_decode_video(avctx, pic, |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
520 &got_picture, data, len); |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
521 if(ret<0) mp_msg(MSGT_DECVIDEO,MSGL_WARN, "Error while decoding frame!\n"); |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
522 |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
523 //-- vstats generation |
8347 | 524 #if LIBAVCODEC_BUILD >= 4643 |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
525 while(lavc_param_vstats){ // always one time loop |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
526 static FILE *fvstats=NULL; |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
527 char filename[20]; |
6834
2d7dfcc79651
Fix overall frametime overflow, hopefully long long int is portable. (untested, will test tomorrow)
atmos4
parents:
6833
diff
changeset
|
528 static long long int all_len=0; |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
529 static int frame_number=0; |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
530 static double all_frametime=0.0; |
8339 | 531 AVVideoFrame *pic= avctx->coded_picture; |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
532 |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
533 if(!fvstats) { |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
534 time_t today2; |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
535 struct tm *today; |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
536 today2 = time(NULL); |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
537 today = localtime(&today2); |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
538 sprintf(filename, "vstats_%02d%02d%02d.log", today->tm_hour, |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
539 today->tm_min, today->tm_sec); |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
540 fvstats = fopen(filename,"w"); |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
541 if(!fvstats) { |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
542 perror("fopen"); |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
543 lavc_param_vstats=0; // disable block |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
544 break; |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
545 /*exit(1);*/ |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
546 } |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
547 } |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
548 |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
549 all_len+=len; |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
550 all_frametime+=sh->frametime; |
8411 | 551 fprintf(fvstats, "frame= %5d q= %2.2f f_size= %6d s_size= %8.0fkB ", |
8339 | 552 ++frame_number, pic->quality, len, (double)all_len/1024); |
6833
a709a7662cd1
Add type= and fix a minor typing difference from ffmpeg
atmos4
parents:
6828
diff
changeset
|
553 fprintf(fvstats, "time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ", |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
554 all_frametime, (double)(len*8)/sh->frametime/1000.0, |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
555 (double)(all_len*8)/all_frametime/1000.0); |
8339 | 556 switch(pic->pict_type){ |
8347 | 557 case FF_I_TYPE: |
8065
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
558 fprintf(fvstats, "type= I\n"); |
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
559 break; |
8347 | 560 case FF_P_TYPE: |
8065
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
561 fprintf(fvstats, "type= P\n"); |
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
562 break; |
8347 | 563 case FF_S_TYPE: |
8065
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
564 fprintf(fvstats, "type= S\n"); |
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
565 break; |
8347 | 566 case FF_B_TYPE: |
8065
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
567 fprintf(fvstats, "type= B\n"); |
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
568 break; |
a3e7c0e16d5b
fixing vstats so B frames are shown as B and not P
michael
parents:
7984
diff
changeset
|
569 } |
8411 | 570 |
571 ctx->qp_stat[(int)(pic->quality+0.5)]++; | |
572 ctx->qp_sum += pic->quality; | |
573 ctx->inv_qp_sum += 1.0/pic->quality; | |
574 | |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
575 break; |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
576 } |
8347 | 577 #endif |
6828
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
578 //-- |
010be15e48ad
Generate ffmpeg compatible vstats_<time>.log, when -lavdopts vstats is specified.
atmos4
parents:
6771
diff
changeset
|
579 |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
580 if(!got_picture) return NULL; // skipped image |
6738 | 581 |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
582 if(init_vo(sh)<0) return NULL; |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
583 |
8339 | 584 if(dr1 && pic->opaque){ |
585 mpi= (mp_image_t*)pic->opaque; | |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
586 } |
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
587 |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
588 if(!mpi && ctx->convert){ |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
589 // do yuv422p -> yuy2 conversion: |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
590 mpi=mpcodecs_get_image(sh, MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE, |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
591 avctx->width, avctx->height); |
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
592 if(!mpi) return NULL; |
8339 | 593 yuv422ptoyuy2(pic->data[0],pic->data[1],pic->data[2], |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
594 mpi->planes[0],avctx->width,avctx->height, |
8339 | 595 pic->linesize[0],pic->linesize[1],mpi->stride[0]); |
5592
b545d56314d2
yuy2 support, fixed bug of dropping frame after config_vo calls
arpi
parents:
5517
diff
changeset
|
596 return mpi; |
5280 | 597 } |
4952 | 598 |
5482 | 599 if(!mpi) |
4952 | 600 mpi=mpcodecs_get_image(sh, MP_IMGTYPE_EXPORT, MP_IMGFLAG_PRESERVE, |
5280 | 601 avctx->width, avctx->height); |
4952 | 602 if(!mpi){ // temporary! |
5207
d337cc4ab0ee
config vo if resolution changed (after decoded image read the dimensions out of lavc context)
alex
parents:
5124
diff
changeset
|
603 printf("couldn't allocate image for codec\n"); |
4952 | 604 return NULL; |
605 } | |
606 | |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
607 if(!dr1){ |
8339 | 608 mpi->planes[0]=pic->data[0]; |
609 mpi->planes[1]=pic->data[1]; | |
610 mpi->planes[2]=pic->data[2]; | |
611 mpi->stride[0]=pic->linesize[0]; | |
612 mpi->stride[1]=pic->linesize[1]; | |
613 mpi->stride[2]=pic->linesize[2]; | |
6733
3b1f37fc0693
direct rendering method 1 (disabled currently as its not bugfree / finished yet, just set dr1=1 if u want to try it)
michael
parents:
6710
diff
changeset
|
614 } |
4952 | 615 |
6873
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
616 if(avctx->pix_fmt==PIX_FMT_YUV422P && mpi->chroma_y_shift==1){ |
1206fa765697
colorspace part cleanup and support for IMGFMT_422P, _444P
arpi
parents:
6869
diff
changeset
|
617 // we have 422p but user wants 420p |
4952 | 618 mpi->stride[1]*=2; |
619 mpi->stride[2]*=2; | |
620 } | |
621 | |
6665 | 622 /* to comfirm with newer lavc style */ |
8339 | 623 mpi->qscale =pic->qscale_table; |
624 mpi->qstride=pic->qstride; | |
625 mpi->pict_type=pic->pict_type; | |
5517 | 626 |
4952 | 627 return mpi; |
628 } | |
629 | |
630 #endif | |
631 |