Mercurial > mplayer.hg
annotate libmpcodecs/vf_ass.c @ 19137:d9656bb2ae0e
new vo driver for ivtv cards tv-out through internal h/w mpeg decoder
author | ben |
---|---|
date | Wed, 19 Jul 2006 22:02:09 +0000 |
parents | 26ed3fcd5cd4 |
children | f6661e33aa60 |
rev | line source |
---|---|
18937 | 1 #include "config.h" |
2 | |
3 #include <stdio.h> | |
4 #include <stdlib.h> | |
5 #include <string.h> | |
6 #include <stdint.h> | |
7 #include <assert.h> | |
8 | |
9 #include "config.h" | |
10 #include "mp_msg.h" | |
11 #include "help_mp.h" | |
12 | |
13 #include "img_format.h" | |
14 #include "mp_image.h" | |
15 #include "vf.h" | |
16 | |
17 #include "libvo/fastmemcpy.h" | |
18 | |
19 #include "m_option.h" | |
20 #include "m_struct.h" | |
21 | |
22 #include "libass/ass.h" | |
23 #include "libass/ass_mp.h" | |
24 | |
25 #define _r(c) ((c)>>24) | |
26 #define _g(c) (((c)>>16)&0xFF) | |
27 #define _b(c) (((c)>>8)&0xFF) | |
28 #define _a(c) ((c)&0xFF) | |
29 #define rgba2y(c) ( (( 263*_r(c) + 516*_g(c) + 100*_b(c)) >> 10) + 16 ) | |
30 #define rgba2u(c) ( ((-152*_r(c) - 298*_g(c) + 450*_b(c)) >> 10) + 128 ) | |
31 #define rgba2v(c) ( (( 450*_r(c) - 376*_g(c) - 73*_b(c)) >> 10) + 128 ) | |
32 | |
33 | |
34 static struct vf_priv_s { | |
35 int outh, outw; | |
36 | |
37 unsigned int outfmt; | |
38 | |
39 // 1 = auto-added filter: insert only if chain does not support EOSD already | |
40 // 0 = insert always | |
41 int auto_insert; | |
42 | |
43 ass_instance_t* ass_priv; | |
44 | |
45 unsigned char* planes[3]; | |
46 unsigned char* dirty_rows; | |
47 } vf_priv_dflt; | |
48 | |
49 extern int opt_screen_size_x; | |
50 extern int opt_screen_size_y; | |
51 | |
52 extern ass_track_t* ass_track; | |
53 extern float sub_delay; | |
54 extern int sub_visibility; | |
55 | |
56 static int config(struct vf_instance_s* vf, | |
57 int width, int height, int d_width, int d_height, | |
58 unsigned int flags, unsigned int outfmt) | |
59 { | |
60 ass_settings_t settings; | |
61 | |
62 if (outfmt == IMGFMT_IF09) return 0; | |
63 | |
64 vf->priv->outh = height + ass_top_margin + ass_bottom_margin; | |
65 vf->priv->outw = width; | |
66 | |
67 if(!opt_screen_size_x && !opt_screen_size_y){ | |
68 d_width = d_width * vf->priv->outw / width; | |
69 d_height = d_height * vf->priv->outh / height; | |
70 } | |
71 | |
19065
26ed3fcd5cd4
drops casts from void * on malloc/calloc, leftover on libmpcodecs
reynaldo
parents:
18937
diff
changeset
|
72 vf->priv->planes[1] = malloc(vf->priv->outw * vf->priv->outh); |
26ed3fcd5cd4
drops casts from void * on malloc/calloc, leftover on libmpcodecs
reynaldo
parents:
18937
diff
changeset
|
73 vf->priv->planes[2] = malloc(vf->priv->outw * vf->priv->outh); |
26ed3fcd5cd4
drops casts from void * on malloc/calloc, leftover on libmpcodecs
reynaldo
parents:
18937
diff
changeset
|
74 vf->priv->dirty_rows = malloc(vf->priv->outh); |
18937 | 75 |
76 if (vf->priv->ass_priv) { | |
77 settings.frame_width = vf->priv->outw; | |
78 settings.frame_height = vf->priv->outh; | |
79 settings.font_size_coeff = ass_font_scale; | |
80 settings.line_spacing = ass_line_spacing; | |
81 settings.top_margin = ass_top_margin; | |
82 settings.bottom_margin = ass_bottom_margin; | |
83 settings.aspect = ((double)d_width) / d_height; | |
84 | |
85 ass_configure(vf->priv->ass_priv, &settings); | |
86 } | |
87 | |
88 return vf_next_config(vf, vf->priv->outw, vf->priv->outh, d_width, d_height, flags, outfmt); | |
89 } | |
90 | |
91 static void get_image(struct vf_instance_s* vf, mp_image_t *mpi) | |
92 { | |
93 if(mpi->type == MP_IMGTYPE_IPB) return; | |
94 if(mpi->flags & MP_IMGFLAG_PRESERVE) return; | |
95 if(mpi->imgfmt != vf->priv->outfmt) return; // colorspace differ | |
96 | |
97 // width never changes, always try full DR | |
98 mpi->priv = vf->dmpi = vf_get_image(vf->next, mpi->imgfmt, | |
99 mpi->type, mpi->flags, | |
100 vf->priv->outw, | |
101 vf->priv->outh); | |
102 | |
103 if((vf->dmpi->flags & MP_IMGFLAG_DRAW_CALLBACK) && | |
104 !(vf->dmpi->flags & MP_IMGFLAG_DIRECT)){ | |
105 mp_msg(MSGT_VFILTER, MSGL_INFO, MSGTR_MPCODECS_FullDRNotPossible); | |
106 return; | |
107 } | |
108 | |
109 // set up mpi as a cropped-down image of dmpi: | |
110 if(mpi->flags&MP_IMGFLAG_PLANAR){ | |
111 mpi->planes[0]=vf->dmpi->planes[0] + ass_top_margin * vf->dmpi->stride[0]; | |
112 mpi->planes[1]=vf->dmpi->planes[1] + (ass_top_margin >> mpi->chroma_y_shift) * vf->dmpi->stride[1]; | |
113 mpi->planes[2]=vf->dmpi->planes[2] + (ass_top_margin >> mpi->chroma_y_shift) * vf->dmpi->stride[2]; | |
114 mpi->stride[1]=vf->dmpi->stride[1]; | |
115 mpi->stride[2]=vf->dmpi->stride[2]; | |
116 } else { | |
117 mpi->planes[0]=vf->dmpi->planes[0] + ass_top_margin * vf->dmpi->stride[0]; | |
118 } | |
119 mpi->stride[0]=vf->dmpi->stride[0]; | |
120 mpi->width=vf->dmpi->width; | |
121 mpi->flags|=MP_IMGFLAG_DIRECT; | |
122 mpi->flags&=~MP_IMGFLAG_DRAW_CALLBACK; | |
123 // vf->dmpi->flags&=~MP_IMGFLAG_DRAW_CALLBACK; | |
124 } | |
125 | |
126 static void blank(mp_image_t *mpi, int y1, int y2) | |
127 { | |
128 int color[3] = {16, 128, 128}; // black (YUV) | |
129 int y; | |
130 unsigned char* dst; | |
131 int chroma_rows = (y2 - y1) >> mpi->chroma_y_shift; | |
132 | |
133 dst = mpi->planes[0] + y1 * mpi->stride[0]; | |
134 for (y = 0; y < y2 - y1; ++y) { | |
135 memset(dst, color[0], mpi->w); | |
136 dst += mpi->stride[0]; | |
137 } | |
138 dst = mpi->planes[1] + (y1 >> mpi->chroma_y_shift) * mpi->stride[1]; | |
139 for (y = 0; y < chroma_rows ; ++y) { | |
140 memset(dst, color[1], mpi->chroma_width); | |
141 dst += mpi->stride[1]; | |
142 } | |
143 dst = mpi->planes[2] + (y1 >> mpi->chroma_y_shift) * mpi->stride[2]; | |
144 for (y = 0; y < chroma_rows ; ++y) { | |
145 memset(dst, color[2], mpi->chroma_width); | |
146 dst += mpi->stride[2]; | |
147 } | |
148 } | |
149 | |
150 static int prepare_image(struct vf_instance_s* vf, mp_image_t *mpi) | |
151 { | |
152 if(mpi->flags&MP_IMGFLAG_DIRECT || mpi->flags&MP_IMGFLAG_DRAW_CALLBACK){ | |
153 vf->dmpi = mpi->priv; | |
154 if (!vf->dmpi) { mp_msg(MSGT_VFILTER, MSGL_WARN, MSGTR_MPCODECS_FunWhydowegetNULL); return 0; } | |
155 mpi->priv = NULL; | |
156 // we've used DR, so we're ready... | |
157 if (ass_top_margin) | |
158 blank(vf->dmpi, 0, ass_top_margin); | |
159 if (ass_bottom_margin) | |
160 blank(vf->dmpi, vf->priv->outh - ass_bottom_margin, vf->priv->outh); | |
161 if(!(mpi->flags&MP_IMGFLAG_PLANAR)) | |
162 vf->dmpi->planes[1] = mpi->planes[1]; // passthrough rgb8 palette | |
163 return 0; | |
164 } | |
165 | |
166 // hope we'll get DR buffer: | |
167 vf->dmpi = vf_get_image(vf->next, vf->priv->outfmt, | |
168 MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE, | |
169 vf->priv->outw, vf->priv->outh); | |
170 | |
171 // copy mpi->dmpi... | |
172 if(mpi->flags&MP_IMGFLAG_PLANAR){ | |
173 memcpy_pic(vf->dmpi->planes[0] + ass_top_margin * vf->dmpi->stride[0], | |
174 mpi->planes[0], mpi->w, mpi->h, | |
175 vf->dmpi->stride[0], mpi->stride[0]); | |
176 memcpy_pic(vf->dmpi->planes[1] + (ass_top_margin >> mpi->chroma_y_shift) * vf->dmpi->stride[1], | |
177 mpi->planes[1], mpi->w >> mpi->chroma_x_shift, mpi->h >> mpi->chroma_y_shift, | |
178 vf->dmpi->stride[1], mpi->stride[1]); | |
179 memcpy_pic(vf->dmpi->planes[2] + (ass_top_margin >> mpi->chroma_y_shift) * vf->dmpi->stride[2], | |
180 mpi->planes[2], mpi->w >> mpi->chroma_x_shift, mpi->h >> mpi->chroma_y_shift, | |
181 vf->dmpi->stride[2], mpi->stride[2]); | |
182 } else { | |
183 memcpy_pic(vf->dmpi->planes[0] + ass_top_margin * vf->dmpi->stride[0], | |
184 mpi->planes[0], mpi->w*(vf->dmpi->bpp/8), mpi->h, | |
185 vf->dmpi->stride[0], mpi->stride[0]); | |
186 vf->dmpi->planes[1] = mpi->planes[1]; // passthrough rgb8 palette | |
187 } | |
188 if (ass_top_margin) | |
189 blank(vf->dmpi, 0, ass_top_margin); | |
190 if (ass_bottom_margin) | |
191 blank(vf->dmpi, vf->priv->outh - ass_bottom_margin, vf->priv->outh); | |
192 return 0; | |
193 } | |
194 | |
195 /** | |
196 * \brief Copy specified rows from render_context.dmpi to render_context.planes, upsampling to 4:4:4 | |
197 */ | |
198 static void copy_from_image(struct vf_instance_s* vf, int first_row, int last_row) | |
199 { | |
200 int pl; | |
201 int i, j, k; | |
202 unsigned char val; | |
203 int chroma_rows; | |
204 | |
205 first_row -= (first_row % 2); | |
206 last_row += (last_row % 2); | |
207 chroma_rows = (last_row - first_row) / 2; | |
208 | |
209 for (pl = 1; pl < 3; ++pl) { | |
210 int dst_stride = vf->dmpi->stride[pl] * 2; | |
211 int src_stride = vf->dmpi->stride[pl]; | |
212 | |
213 unsigned char* src = vf->dmpi->planes[pl] + (first_row/2) * src_stride; | |
214 unsigned char* dst = vf->priv->planes[pl] + first_row * dst_stride; | |
215 unsigned char* dst_next = dst + dst_stride; | |
216 for(i = 0; i < chroma_rows; ++i) | |
217 { | |
218 if ((vf->priv->dirty_rows[first_row + i*2] == 0) || | |
219 (vf->priv->dirty_rows[first_row + i*2 + 1] == 0)) { | |
220 for (j = 0, k = 0; j < vf->dmpi->chroma_width; ++j, k+=2) { | |
221 val = *(src + j); | |
222 *(dst + k) = val; | |
223 *(dst + k + 1) = val; | |
224 *(dst_next + k) = val; | |
225 *(dst_next + k + 1) = val; | |
226 } | |
227 } | |
228 src += src_stride; | |
229 dst = dst_next + dst_stride; | |
230 dst_next = dst + dst_stride; | |
231 } | |
232 } | |
233 for (i = first_row; i < last_row; ++i) | |
234 vf->priv->dirty_rows[i] = 1; | |
235 } | |
236 | |
237 /** | |
238 * \brief Copy all previously copied rows back to render_context.dmpi | |
239 */ | |
240 static void copy_to_image(struct vf_instance_s* vf) | |
241 { | |
242 int pl; | |
243 int i, j, k; | |
244 for (pl = 1; pl < 3; ++pl) { | |
245 int dst_stride = vf->dmpi->stride[pl]; | |
246 int src_stride = vf->dmpi->stride[pl] * 2; | |
247 | |
248 unsigned char* dst = vf->dmpi->planes[pl]; | |
249 unsigned char* src = vf->priv->planes[pl]; | |
250 unsigned char* src_next = vf->priv->planes[pl] + src_stride; | |
251 for(i = 0; i < vf->dmpi->chroma_height; ++i) | |
252 { | |
253 if ((vf->priv->dirty_rows[i*2] == 1)) { | |
254 assert(vf->priv->dirty_rows[i*2 + 1] == 1); | |
255 for (j = 0, k = 0; j < vf->dmpi->chroma_width; ++j, k+=2) { | |
256 unsigned val = 0; | |
257 val += *(src + k); | |
258 val += *(src + k + 1); | |
259 val += *(src_next + k); | |
260 val += *(src_next + k + 1); | |
261 *(dst + j) = val >> 2; | |
262 } | |
263 } | |
264 dst += dst_stride; | |
265 src = src_next + src_stride; | |
266 src_next = src + src_stride; | |
267 } | |
268 } | |
269 } | |
270 | |
271 static void my_draw_bitmap(struct vf_instance_s* vf, unsigned char* bitmap, int bitmap_w, int bitmap_h, int stride, int dst_x, int dst_y, unsigned color) | |
272 { | |
273 unsigned char y = rgba2y(color); | |
274 unsigned char u = rgba2u(color); | |
275 unsigned char v = rgba2v(color); | |
276 unsigned char opacity = 255 - _a(color); | |
277 unsigned char* src; | |
278 unsigned char* dst; | |
279 int i, j; | |
280 mp_image_t* dmpi = vf->dmpi; | |
281 | |
282 | |
283 src = bitmap; | |
284 dst = dmpi->planes[0] + dst_x + dst_y * dmpi->stride[0]; | |
285 for(i = 0; i < bitmap_h; ++i) | |
286 { | |
287 for (j = 0; j < bitmap_w; ++j) { | |
288 // unsigned k = *(src+j); | |
289 unsigned k = ((unsigned)*(bitmap + stride * i + j)) * opacity / 255; | |
290 unsigned char orig_color = *(dst+j); | |
291 *(dst+j) = (k*y + (255-k)*orig_color) / 255; | |
292 } | |
293 src += stride; | |
294 dst += dmpi->stride[0]; | |
295 } | |
296 | |
297 for (i = 0; i < bitmap_h; ++i) { | |
298 for (j = 0; j < bitmap_w; ++j) { | |
299 int x = dst_x + j; | |
300 int y = dst_y + i; | |
301 unsigned k; | |
302 unsigned char orig_u, orig_v; | |
303 unsigned char new_u, new_v; | |
304 | |
305 k = ((unsigned)*(bitmap + stride * i + j)) * opacity / 255; | |
306 | |
307 orig_u = *(vf->priv->planes[1] + x + y * 2 * dmpi->chroma_width); | |
308 new_u = (k*u + (255-k)*orig_u) / 255; | |
309 *(vf->priv->planes[1] + x + y * 2 * dmpi->chroma_width) = new_u; | |
310 | |
311 orig_v = *(vf->priv->planes[2] + x + y * 2 * dmpi->chroma_width); | |
312 new_v = (k*v + (255-k)*orig_v) / 255; | |
313 *(vf->priv->planes[2] + x + y * 2 * dmpi->chroma_width) = new_v; | |
314 } | |
315 } | |
316 } | |
317 | |
318 static int render_frame(struct vf_instance_s* vf, mp_image_t *mpi, const ass_image_t* img) | |
319 { | |
320 if (img) { | |
321 memset(vf->priv->dirty_rows, 0, vf->priv->outh); // reset dirty rows | |
322 while (img) { | |
323 copy_from_image(vf, img->dst_y, img->dst_y + img->h); | |
324 my_draw_bitmap(vf, img->bitmap, img->w, img->h, img->stride, | |
325 img->dst_x, img->dst_y, img->color); | |
326 img = img->next; | |
327 } | |
328 copy_to_image(vf); | |
329 } | |
330 return 0; | |
331 } | |
332 | |
333 static int put_image(struct vf_instance_s* vf, mp_image_t *mpi, double pts) | |
334 { | |
335 ass_image_t* images = 0; | |
336 if (sub_visibility && vf->priv->ass_priv && ass_track && (pts != MP_NOPTS_VALUE)) | |
337 images = ass_render_frame(vf->priv->ass_priv, ass_track, (pts+sub_delay) * 1000 + .5); | |
338 | |
339 prepare_image(vf, mpi); | |
340 if (images) render_frame(vf, mpi, images); | |
341 | |
342 return vf_next_put_image(vf, vf->dmpi, pts); | |
343 } | |
344 | |
345 static int query_format(struct vf_instance_s* vf, unsigned int fmt) | |
346 { | |
347 switch(fmt){ | |
348 case IMGFMT_YV12: | |
349 case IMGFMT_I420: | |
350 case IMGFMT_IYUV: | |
351 return vf_next_query_format(vf, vf->priv->outfmt); | |
352 } | |
353 return 0; | |
354 } | |
355 | |
356 static int control(vf_instance_t *vf, int request, void *data) | |
357 { | |
358 if (request == VFCTRL_EOSD) { | |
359 vf->priv->ass_priv = ass_init(); | |
360 if (!vf->priv->ass_priv) { | |
361 mp_msg(MSGT_VFILTER, MSGL_ERR, "[ass] init failed\n"); | |
362 return 0; | |
363 } else | |
364 mp_msg(MSGT_VFILTER, MSGL_INFO, "[ass] init\n"); | |
365 return CONTROL_TRUE; | |
366 } | |
367 return vf_next_control(vf, request, data); | |
368 } | |
369 | |
370 static void uninit(struct vf_instance_s* vf) | |
371 { | |
372 if (vf->priv->ass_priv) | |
373 ass_done(vf->priv->ass_priv); | |
374 if (vf->priv->planes[1]) | |
375 free(vf->priv->planes[1]); | |
376 if (vf->priv->planes[2]) | |
377 free(vf->priv->planes[2]); | |
378 if (vf->priv->dirty_rows) | |
379 free(vf->priv->dirty_rows); | |
380 } | |
381 | |
382 static unsigned int fmt_list[]={ | |
383 IMGFMT_YV12, | |
384 IMGFMT_I420, | |
385 IMGFMT_IYUV, | |
386 0 | |
387 }; | |
388 | |
389 static int open(vf_instance_t *vf, char* args) | |
390 { | |
391 int flags; | |
392 vf->priv->outfmt = vf_match_csp(&vf->next,fmt_list,IMGFMT_YV12); | |
393 if (vf->priv->outfmt) | |
394 flags = vf_next_query_format(vf, vf->priv->outfmt); | |
395 if (!vf->priv->outfmt || (vf->priv->auto_insert && flags&VFCAP_EOSD)) | |
396 { | |
397 uninit(vf); | |
398 return 0; | |
399 } | |
400 | |
401 if (vf->priv->auto_insert) | |
402 mp_msg(MSGT_VFILTER, MSGL_INFO, "[ass] auto-open\n"); | |
403 | |
404 vf->config = config; | |
405 vf->query_format = query_format; | |
406 vf->uninit = uninit; | |
407 vf->control = control; | |
408 vf->get_image = get_image; | |
409 vf->put_image = put_image; | |
410 vf->default_caps=VFCAP_EOSD; | |
411 return 1; | |
412 } | |
413 | |
414 #define ST_OFF(f) M_ST_OFF(struct vf_priv_s,f) | |
415 static m_option_t vf_opts_fields[] = { | |
416 {"auto", ST_OFF(auto_insert), CONF_TYPE_FLAG, 0 , 0, 1, NULL}, | |
417 { NULL, NULL, 0, 0, 0, 0, NULL } | |
418 }; | |
419 | |
420 static m_struct_t vf_opts = { | |
421 "ass", | |
422 sizeof(struct vf_priv_s), | |
423 &vf_priv_dflt, | |
424 vf_opts_fields | |
425 }; | |
426 | |
427 vf_info_t vf_info_ass = { | |
428 "Render ASS/SSA subtitles", | |
429 "ass", | |
430 "Evgeniy Stepanov", | |
431 "", | |
432 open, | |
433 &vf_opts | |
434 }; | |
435 |