Mercurial > mplayer.hg
annotate sub/subassconvert.c @ 34918:6dd0dd63179b
add yuv8 / YUV8 to other decoders as well
author | compn |
---|---|
date | Wed, 04 Jul 2012 00:32:26 +0000 |
parents | 69fe59abe044 |
children | 482418e9e922 |
rev | line source |
---|---|
32462 | 1 /* |
2 * Subtitles converter to SSA/ASS in order to allow special formatting | |
3 * | |
4 * This file is part of MPlayer. | |
5 * | |
6 * MPlayer is free software; you can redistribute it and/or modify | |
7 * it under the terms of the GNU General Public License as published by | |
8 * the Free Software Foundation; either version 2 of the License, or | |
9 * (at your option) any later version. | |
10 * | |
11 * MPlayer is distributed in the hope that it will be useful, | |
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 * GNU General Public License for more details. | |
15 * | |
16 * You should have received a copy of the GNU General Public License along | |
17 * with MPlayer; if not, write to the Free Software Foundation, Inc., | |
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | |
19 */ | |
20 | |
21 #include <string.h> | |
22 #include <stdint.h> | |
23 #include <stdlib.h> | |
24 #include <stdio.h> | |
25 #include <stdarg.h> | |
26 | |
27 #include "mp_msg.h" | |
28 #include "help_mp.h" | |
29 #include "bstr.h" | |
32464
22888a8cb312
Do not use a path for including files in the same directory.
reimar
parents:
32462
diff
changeset
|
30 #include "subassconvert.h" |
32462 | 31 #include "libavutil/common.h" |
32 | |
33 struct line { | |
34 char *buf; | |
35 size_t bufsize; | |
36 size_t len; | |
37 }; | |
38 | |
39 #ifdef __GNUC__ | |
40 static void append_text(struct line *dst, char *fmt, ...) __attribute__ ((format(printf, 2, 3))); | |
41 #endif | |
42 | |
43 static void append_text(struct line *dst, char *fmt, ...) | |
44 { | |
45 va_list va; | |
46 int ret; | |
47 | |
48 va_start(va, fmt); | |
49 ret = vsnprintf(dst->buf + dst->len, dst->bufsize - dst->len, fmt, va); | |
50 if (ret >= 0) { | |
51 dst->len += ret; | |
52 if (dst->len > dst->bufsize) | |
53 dst->len = dst->bufsize; | |
54 } | |
55 va_end(va); | |
56 } | |
57 | |
58 static int indexof(const char *s, int c) | |
59 { | |
60 char *f = strchr(s, c); | |
61 return f ? (f - s) : -1; | |
62 } | |
63 | |
64 | |
65 | |
66 /* | |
67 * SubRip | |
68 * | |
69 * Support basic tags (italic, bold, underline, strike-through) | |
70 * and font tag with size, color and face attributes. | |
71 * | |
72 */ | |
73 | |
74 struct font_tag { | |
75 struct bstr face; | |
76 int size; | |
77 uint32_t color; | |
78 }; | |
79 | |
80 static const struct tag_conv { | |
81 const char *from; | |
82 const char *to; | |
83 } subrip_basic_tags[] = { | |
84 {"<i>", "{\\i1}"}, {"</i>", "{\\i0}"}, | |
85 {"<b>", "{\\b1}"}, {"</b>", "{\\b0}"}, | |
86 {"<u>", "{\\u1}"}, {"</u>", "{\\u0}"}, | |
87 {"<s>", "{\\s1}"}, {"</s>", "{\\s0}"}, | |
88 {"{", "\\{"}, {"}", "\\}"}, | |
34757 | 89 {"\r\n", "\\N"}, {"\n", "\\N"}, {"\r", "\\N"}, |
32462 | 90 }; |
91 | |
92 static const struct { | |
93 const char *s; | |
94 uint32_t v; | |
95 } subrip_web_colors[] = { | |
34912
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
96 /* Named CSS3 colors in BGR format; a subset of those |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
97 at http://www.w3.org/TR/css3-color/#svg-color */ |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
98 {"aqua", 0xffff00}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
99 {"black", 0x000000}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
100 {"blue", 0xff0000}, |
34913 | 101 {"cyan", 0xffff00}, |
34912
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
102 {"fuchsia", 0xff00ff}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
103 {"gray", 0x808080}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
104 {"green", 0x008000}, |
34913 | 105 {"grey", 0x808080}, |
34912
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
106 {"lime", 0x00ff00}, |
34913 | 107 {"magenta", 0xff00ff}, |
34912
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
108 {"maroon", 0x000080}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
109 {"navy", 0x800000}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
110 {"olive", 0x008080}, |
34913 | 111 {"orange", 0x00a5ff}, |
112 {"pink", 0xcbc0ff}, | |
34912
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
113 {"purple", 0x800080}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
114 {"red", 0x0000ff}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
115 {"silver", 0xc0c0c0}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
116 {"teal", 0x808000}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
117 {"white", 0xffffff}, |
3f8886645ebd
Order colour names alphabetically / better match W3 list.
cehoyos
parents:
34758
diff
changeset
|
118 {"yellow", 0x00ffff}, |
32462 | 119 }; |
120 | |
121 #define SUBRIP_MAX_STACKED_FONT_TAGS 16 | |
122 #define SUBRIP_FLAG_COLOR 0x01000000 | |
123 | |
124 /** | |
125 * \brief Convert SubRip lines into ASS markup | |
126 * \param orig original SubRip lines. The content will remain untouched. | |
127 * \param dest ASS markup destination buffer. | |
128 * \param dest_buffer_size maximum size for the destination buffer. | |
129 */ | |
130 void subassconvert_subrip(const char *orig, char *dest, size_t dest_buffer_size) | |
131 { | |
132 /* line is not const to avoid warnings with strtol, etc. | |
133 * orig content won't be changed */ | |
134 char *line = (char *)orig; | |
135 struct line new_line = { | |
136 .buf = dest, | |
137 .bufsize = dest_buffer_size, | |
138 }; | |
139 struct font_tag font_stack[SUBRIP_MAX_STACKED_FONT_TAGS]; | |
140 int sp = 0; | |
141 | |
142 font_stack[0] = (struct font_tag){}; // type with all defaults | |
143 while (*line && new_line.len < new_line.bufsize - 1) { | |
144 char *orig_line = line; | |
145 int i; | |
146 | |
147 for (i = 0; i < FF_ARRAY_ELEMS(subrip_basic_tags); i++) { | |
148 const struct tag_conv *tag = &subrip_basic_tags[i]; | |
149 int from_len = strlen(tag->from); | |
150 if (strncmp(line, tag->from, from_len) == 0) { | |
151 append_text(&new_line, "%s", tag->to); | |
152 line += from_len; | |
153 } | |
154 } | |
155 | |
156 if (strncmp(line, "</font>", 7) == 0) { | |
157 /* Closing font tag */ | |
158 line += 7; | |
159 | |
160 if (sp > 0) { | |
161 struct font_tag *tag = &font_stack[sp]; | |
162 struct font_tag *last_tag = &tag[-1]; | |
163 sp--; | |
164 | |
165 if (tag->size) { | |
166 if (!last_tag->size) | |
167 append_text(&new_line, "{\\fs}"); | |
168 else if (last_tag->size != tag->size) | |
169 append_text(&new_line, "{\\fs%d}", last_tag->size); | |
170 } | |
171 | |
172 if (tag->color & SUBRIP_FLAG_COLOR) { | |
173 if (!(last_tag->color & SUBRIP_FLAG_COLOR)) | |
174 append_text(&new_line, "{\\c}"); | |
175 else if (last_tag->color != tag->color) | |
176 append_text(&new_line, "{\\c&H%06X&}", | |
177 last_tag->color & 0xffffff); | |
178 } | |
179 | |
180 if (tag->face.len) { | |
181 if (!last_tag->face.len) | |
182 append_text(&new_line, "{\\fn}"); | |
183 else if (bstrcmp(last_tag->face, tag->face) != 0) | |
184 append_text(&new_line, "{\\fn%.*s}", | |
185 BSTR_P(last_tag->face)); | |
186 } | |
187 } | |
188 } else if (strncmp(line, "<font ", 6) == 0 | |
189 && sp + 1 < FF_ARRAY_ELEMS(font_stack)) { | |
190 /* Opening font tag */ | |
191 char *potential_font_tag_start = line; | |
192 int len_backup = new_line.len; | |
193 struct font_tag *tag = &font_stack[sp + 1]; | |
194 int has_valid_attr = 0; | |
195 | |
196 *tag = tag[-1]; // keep values from previous tag | |
197 line += 6; | |
198 | |
199 while (*line && *line != '>') { | |
200 if (strncmp(line, "size=\"", 6) == 0) { | |
201 line += 6; | |
202 tag->size = strtol(line, &line, 10); | |
203 if (*line != '"' || !tag->size) | |
204 break; | |
205 append_text(&new_line, "{\\fs%d}", tag->size); | |
206 has_valid_attr = 1; | |
207 } else if (strncmp(line, "color=\"", 7) == 0) { | |
208 line += 7; | |
209 if (*line == '#') { | |
210 // #RRGGBB format | |
211 line++; | |
212 tag->color = strtol(line, &line, 16) & 0x00ffffff; | |
213 if (*line != '"') | |
214 break; | |
215 tag->color = ((tag->color & 0xff) << 16) | | |
216 (tag->color & 0xff00) | | |
217 ((tag->color & 0xff0000) >> 16) | | |
218 SUBRIP_FLAG_COLOR; | |
219 } else { | |
220 // Standard web colors | |
221 int i, len = indexof(line, '"'); | |
222 if (len <= 0) | |
223 break; | |
224 for (i = 0; i < FF_ARRAY_ELEMS(subrip_web_colors); i++) { | |
225 const char *color = subrip_web_colors[i].s; | |
226 if (strlen(color) == len | |
227 && strncasecmp(line, color, len) == 0) { | |
228 tag->color = SUBRIP_FLAG_COLOR | subrip_web_colors[i].v; | |
229 break; | |
230 } | |
231 } | |
232 | |
233 if (i == FF_ARRAY_ELEMS(subrip_web_colors)) { | |
234 /* We didn't find any matching color */ | |
235 line = strchr(line, '"'); // can't be NULL, see above | |
236 mp_msg(MSGT_SUBREADER, MSGL_WARN, | |
237 MSGTR_SUBTITLES_SubRip_UnknownFontColor, orig); | |
238 append_text(&new_line, "{\\c}"); | |
239 line += 2; | |
240 continue; | |
241 } | |
242 | |
243 line += len; | |
244 } | |
245 append_text(&new_line, "{\\c&H%06X&}", tag->color & 0xffffff); | |
246 has_valid_attr = 1; | |
247 } else if (strncmp(line, "face=\"", 6) == 0) { | |
248 /* Font face attribute */ | |
249 int len; | |
250 line += 6; | |
251 len = indexof(line, '"'); | |
252 if (len <= 0) | |
253 break; | |
254 tag->face.start = line; | |
255 tag->face.len = len; | |
256 line += len; | |
257 append_text(&new_line, "{\\fn%.*s}", BSTR_P(tag->face)); | |
258 has_valid_attr = 1; | |
259 } | |
260 line++; | |
261 } | |
262 | |
263 if (!has_valid_attr || *line != '>') { /* Not valid font tag */ | |
264 line = potential_font_tag_start; | |
265 new_line.len = len_backup; | |
266 } else { | |
267 sp++; | |
268 line++; | |
269 } | |
270 } | |
271 | |
272 /* Tag conversion code didn't match */ | |
273 if (line == orig_line) | |
274 new_line.buf[new_line.len++] = *line++; | |
275 } | |
276 new_line.buf[new_line.len] = 0; | |
277 } | |
278 | |
279 | |
280 /* | |
281 * MicroDVD | |
282 * | |
283 * Based on the specifications found here: | |
284 * https://trac.videolan.org/vlc/ticket/1825#comment:6 | |
285 */ | |
286 | |
287 struct microdvd_tag { | |
288 char key; | |
289 int persistent; | |
290 uint32_t data1; | |
291 uint32_t data2; | |
292 struct bstr data_string; | |
293 }; | |
294 | |
295 #define MICRODVD_PERSISTENT_OFF 0 | |
296 #define MICRODVD_PERSISTENT_ON 1 | |
297 #define MICRODVD_PERSISTENT_OPENED 2 | |
298 | |
299 // Color, Font, Size, cHarset, stYle, Position, cOordinate | |
300 #define MICRODVD_TAGS "cfshyYpo" | |
301 | |
302 static void microdvd_set_tag(struct microdvd_tag *tags, struct microdvd_tag tag) | |
303 { | |
304 int tag_index = indexof(MICRODVD_TAGS, tag.key); | |
305 | |
306 if (tag_index < 0) | |
307 return; | |
308 memcpy(&tags[tag_index], &tag, sizeof(tag)); | |
309 } | |
310 | |
311 // italic, bold, underline, strike-through | |
312 #define MICRODVD_STYLES "ibus" | |
313 | |
314 static char *microdvd_load_tags(struct microdvd_tag *tags, char *s) | |
315 { | |
316 while (*s == '{') { | |
317 char *start = s; | |
318 char tag_char = *(s + 1); | |
34758 | 319 struct microdvd_tag tag = {0}; |
32462 | 320 |
321 if (!tag_char || *(s + 2) != ':') | |
322 break; | |
323 s += 3; | |
324 | |
325 switch (tag_char) { | |
326 | |
327 /* Style */ | |
328 case 'Y': | |
329 tag.persistent = MICRODVD_PERSISTENT_ON; | |
330 case 'y': | |
331 while (*s && *s != '}') { | |
332 int style_index = indexof(MICRODVD_STYLES, *s); | |
333 | |
334 if (style_index >= 0) | |
335 tag.data1 |= (1 << style_index); | |
336 s++; | |
337 } | |
338 if (*s != '}') | |
339 break; | |
340 /* We must distinguish persistent and non-persistent styles | |
341 * to handle this kind of style tags: {y:ib}{Y:us} */ | |
342 tag.key = tag_char; | |
343 break; | |
344 | |
345 /* Color */ | |
346 case 'C': | |
347 tag.persistent = MICRODVD_PERSISTENT_ON; | |
348 case 'c': | |
349 tag.data1 = strtol(s, &s, 16) & 0x00ffffff; | |
350 if (*s != '}') | |
351 break; | |
352 tag.key = 'c'; | |
353 break; | |
354 | |
355 /* Font name */ | |
356 case 'F': | |
357 tag.persistent = MICRODVD_PERSISTENT_ON; | |
358 case 'f': | |
359 { | |
360 int len = indexof(s, '}'); | |
361 if (len < 0) | |
362 break; | |
363 tag.data_string.start = s; | |
364 tag.data_string.len = len; | |
365 s += len; | |
366 tag.key = 'f'; | |
367 break; | |
368 } | |
369 | |
370 /* Font size */ | |
371 case 'S': | |
372 tag.persistent = MICRODVD_PERSISTENT_ON; | |
373 case 's': | |
374 tag.data1 = strtol(s, &s, 10); | |
375 if (*s != '}') | |
376 break; | |
377 tag.key = 's'; | |
378 break; | |
379 | |
380 /* Charset */ | |
381 case 'H': | |
382 { | |
383 //TODO: not yet handled, just parsed. | |
384 int len = indexof(s, '}'); | |
385 if (len < 0) | |
386 break; | |
387 tag.data_string.start = s; | |
388 tag.data_string.len = len; | |
389 s += len; | |
390 tag.key = 'h'; | |
391 break; | |
392 } | |
393 | |
394 /* Position */ | |
395 case 'P': | |
396 tag.persistent = MICRODVD_PERSISTENT_ON; | |
397 tag.data1 = (*s++ == '1'); | |
398 if (*s != '}') | |
399 break; | |
400 tag.key = 'p'; | |
401 break; | |
402 | |
403 /* Coordinates */ | |
404 case 'o': | |
405 tag.persistent = MICRODVD_PERSISTENT_ON; | |
406 tag.data1 = strtol(s, &s, 10); | |
407 if (*s != ',') | |
408 break; | |
409 s++; | |
410 tag.data2 = strtol(s, &s, 10); | |
411 if (*s != '}') | |
412 break; | |
413 tag.key = 'o'; | |
414 break; | |
415 | |
416 default: /* Unknown tag, we consider it's text */ | |
417 break; | |
418 } | |
419 | |
420 if (tag.key == 0) | |
421 return start; | |
422 | |
423 microdvd_set_tag(tags, tag); | |
424 s++; | |
425 } | |
426 return s; | |
427 } | |
428 | |
429 static void microdvd_open_tags(struct line *new_line, struct microdvd_tag *tags) | |
430 { | |
431 int i, sidx; | |
432 for (i = 0; i < sizeof(MICRODVD_TAGS) - 1; i++) { | |
433 if (tags[i].persistent == MICRODVD_PERSISTENT_OPENED) | |
434 continue; | |
435 switch (tags[i].key) { | |
436 case 'Y': | |
437 case 'y': | |
438 for (sidx = 0; sidx < sizeof(MICRODVD_STYLES) - 1; sidx++) | |
439 if (tags[i].data1 & (1 << sidx)) | |
440 append_text(new_line, "{\\%c1}", MICRODVD_STYLES[sidx]); | |
441 break; | |
442 | |
443 case 'c': | |
444 append_text(new_line, "{\\c&H%06X&}", tags[i].data1); | |
445 break; | |
446 | |
447 case 'f': | |
448 append_text(new_line, "{\\fn%.*s}", BSTR_P(tags[i].data_string)); | |
449 break; | |
450 | |
451 case 's': | |
452 append_text(new_line, "{\\fs%d}", tags[i].data1); | |
453 break; | |
454 | |
455 case 'p': | |
456 if (tags[i].data1 == 0) | |
457 append_text(new_line, "{\\an8}"); | |
458 break; | |
459 | |
460 case 'o': | |
461 append_text(new_line, "{\\pos(%d,%d)}", | |
462 tags[i].data1, tags[i].data2); | |
463 break; | |
464 } | |
465 if (tags[i].persistent == MICRODVD_PERSISTENT_ON) | |
466 tags[i].persistent = MICRODVD_PERSISTENT_OPENED; | |
467 } | |
468 } | |
469 | |
470 static void microdvd_close_no_persistent_tags(struct line *new_line, | |
471 struct microdvd_tag *tags) | |
472 { | |
473 int i, sidx; | |
474 | |
475 for (i = sizeof(MICRODVD_TAGS) - 2; i; i--) { | |
476 if (tags[i].persistent != MICRODVD_PERSISTENT_OFF) | |
477 continue; | |
478 switch (tags[i].key) { | |
479 | |
480 case 'y': | |
481 for (sidx = sizeof(MICRODVD_STYLES) - 2; sidx >= 0; sidx--) | |
482 if (tags[i].data1 & (1 << sidx)) | |
483 append_text(new_line, "{\\%c0}", MICRODVD_STYLES[sidx]); | |
484 break; | |
485 | |
486 case 'c': | |
487 append_text(new_line, "{\\c}"); | |
488 break; | |
489 | |
490 case 'f': | |
491 append_text(new_line, "{\\fn}"); | |
492 break; | |
493 | |
494 case 's': | |
495 append_text(new_line, "{\\fs}"); | |
496 break; | |
497 } | |
498 tags[i].key = 0; | |
499 } | |
500 } | |
501 | |
502 /** | |
503 * \brief Convert MicroDVD lines into ASS markup | |
504 * \param orig original MicroDVD line. The content will remain untouched. | |
505 * \param dest ASS markup destination buffer. | |
506 * \param dest_buffer_size maximum size for the destination buffer. | |
507 */ | |
508 void subassconvert_microdvd(const char *orig, char *dest, size_t dest_buffer_size) | |
509 { | |
510 /* line is not const to avoid warnings with strtol, etc. | |
511 * orig content won't be changed */ | |
512 char *line = (char *)orig; | |
513 struct line new_line = { | |
514 .buf = dest, | |
515 .bufsize = dest_buffer_size, | |
516 }; | |
34758 | 517 struct microdvd_tag tags[sizeof(MICRODVD_TAGS) - 1] = {{0}}; |
32462 | 518 |
519 while (*line) { | |
520 line = microdvd_load_tags(tags, line); | |
521 microdvd_open_tags(&new_line, tags); | |
522 | |
523 while (*line && *line != '|') | |
524 new_line.buf[new_line.len++] = *line++; | |
525 | |
526 if (*line == '|') { | |
527 microdvd_close_no_persistent_tags(&new_line, tags); | |
528 append_text(&new_line, "\\N"); | |
529 line++; | |
530 } | |
531 } | |
532 new_line.buf[new_line.len] = 0; | |
533 } |