Mercurial > mplayer.hg
annotate libass/ass.c @ 19492:c8daf3471201
SSA/ASS parser reworked, with 2 main results:
support for script embedded fonts (fonts, uuencoded directly into script) added;
matroska interface functions have got more sensible names.
author | eugeni |
---|---|
date | Tue, 22 Aug 2006 22:11:01 +0000 |
parents | 07209f48e527 |
children | 31ac2e1a5695 |
rev | line source |
---|---|
18937 | 1 #include "config.h" |
2 | |
3 #include <stdio.h> | |
4 #include <stdlib.h> | |
5 #include <string.h> | |
6 #include <assert.h> | |
7 #include <errno.h> | |
8 #include <sys/types.h> | |
9 #include <sys/stat.h> | |
10 #include <unistd.h> | |
19374 | 11 #include <inttypes.h> |
18937 | 12 |
13 #ifdef HAVE_ENCA | |
14 #include "subreader.h" // for guess_buffer_cp | |
15 #endif | |
16 | |
17 #ifdef USE_ICONV | |
18 #include <iconv.h> | |
19 extern char *sub_cp; | |
20 #endif | |
19492 | 21 extern int extract_embedded_fonts; |
18937 | 22 |
23 #include "mp_msg.h" | |
24 #include "ass.h" | |
25 #include "ass_utils.h" | |
26 #include "libvo/sub.h" // for utf8_get_char | |
27 | |
28 char *get_path(char *); | |
29 | |
19492 | 30 struct parser_priv_s { |
31 enum {PST_UNKNOWN = 0, PST_INFO, PST_STYLES, PST_EVENTS, PST_FONTS} state; | |
32 char* fontname; | |
33 char* fontdata; | |
34 int fontdata_size; | |
35 int fontdata_used; | |
36 }; | |
37 | |
18937 | 38 #define ASS_STYLES_ALLOC 20 |
39 #define ASS_EVENTS_ALLOC 200 | |
40 | |
41 void ass_free_track(ass_track_t* track) { | |
42 int i; | |
43 | |
19492 | 44 if (track->parser_priv) { |
45 if (track->parser_priv->fontname) | |
46 free(track->parser_priv->fontname); | |
47 if (track->parser_priv->fontdata) | |
48 free(track->parser_priv->fontdata); | |
49 free(track->parser_priv); | |
50 } | |
18937 | 51 if (track->style_format) |
52 free(track->style_format); | |
53 if (track->event_format) | |
54 free(track->event_format); | |
55 if (track->styles) { | |
19474
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
56 for (i = 0; i < track->n_styles; ++i) |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
57 ass_free_style(track, i); |
18937 | 58 free(track->styles); |
59 } | |
60 if (track->events) { | |
19474
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
61 for (i = 0; i < track->n_events; ++i) |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
62 ass_free_event(track, i); |
18937 | 63 free(track->events); |
64 } | |
65 } | |
66 | |
67 /// \brief Allocate a new style struct | |
68 /// \param track track | |
69 /// \return style id | |
70 int ass_alloc_style(ass_track_t* track) { | |
71 int sid; | |
72 | |
73 assert(track->n_styles <= track->max_styles); | |
74 | |
75 if (track->n_styles == track->max_styles) { | |
76 track->max_styles += ASS_STYLES_ALLOC; | |
77 track->styles = (ass_style_t*)realloc(track->styles, sizeof(ass_style_t)*track->max_styles); | |
78 } | |
79 | |
80 sid = track->n_styles++; | |
81 memset(track->styles + sid, 0, sizeof(ass_style_t)); | |
82 return sid; | |
83 } | |
84 | |
85 /// \brief Allocate a new event struct | |
86 /// \param track track | |
87 /// \return event id | |
88 int ass_alloc_event(ass_track_t* track) { | |
89 int eid; | |
90 | |
91 assert(track->n_events <= track->max_events); | |
92 | |
93 if (track->n_events == track->max_events) { | |
94 track->max_events += ASS_EVENTS_ALLOC; | |
95 track->events = (ass_event_t*)realloc(track->events, sizeof(ass_event_t)*track->max_events); | |
96 } | |
97 | |
98 eid = track->n_events++; | |
99 memset(track->events + eid, 0, sizeof(ass_event_t)); | |
100 return eid; | |
101 } | |
102 | |
19474
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
103 void ass_free_event(ass_track_t* track, int eid) { |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
104 ass_event_t* event = track->events + eid; |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
105 if (event->Name) |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
106 free(event->Name); |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
107 if (event->Effect) |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
108 free(event->Effect); |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
109 if (event->Text) |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
110 free(event->Text); |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
111 } |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
112 |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
113 void ass_free_style(ass_track_t* track, int sid) { |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
114 ass_style_t* style = track->styles + sid; |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
115 if (style->Name) |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
116 free(style->Name); |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
117 if (style->FontName) |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
118 free(style->FontName); |
18937 | 119 } |
120 | |
121 static int events_compare_f(const void* a_, const void* b_) { | |
122 ass_event_t* a = (ass_event_t*)a_; | |
123 ass_event_t* b = (ass_event_t*)b_; | |
124 if (a->Start < b->Start) | |
125 return -1; | |
126 else if (a->Start > b->Start) | |
127 return 1; | |
128 else | |
129 return 0; | |
130 } | |
131 | |
132 /// \brief Sort events by start time | |
133 /// \param tid track id | |
134 static void sort_events(ass_track_t* track) { | |
135 qsort(track->events, track->n_events, sizeof(ass_event_t), events_compare_f); | |
136 } | |
137 | |
138 // ============================================================================================== | |
139 | |
140 static void skip_spaces(char** str) { | |
141 char* p = *str; | |
142 while ((*p==' ') || (*p=='\t')) | |
143 ++p; | |
144 *str = p; | |
145 } | |
146 | |
147 static void rskip_spaces(char** str, char* limit) { | |
148 char* p = *str; | |
149 while ((p >= limit) && ((*p==' ') || (*p=='\t'))) | |
150 --p; | |
151 *str = p; | |
152 } | |
153 | |
154 /** | |
155 * \brief find style by name | |
156 * \param track track | |
157 * \param name style name | |
158 * \return index in track->styles | |
159 * Returnes 0 if no styles found => expects at least 1 style. | |
160 * Parsing code always adds "Default" style in the end. | |
161 */ | |
162 static int lookup_style(ass_track_t* track, char* name) { | |
163 int i; | |
164 for (i=0; i<track->n_styles; ++i) { | |
165 // FIXME: mb strcasecmp ? | |
166 if (strcmp(track->styles[i].Name, name) == 0) | |
167 return i; | |
168 } | |
169 i = track->default_style; | |
170 mp_msg(MSGT_GLOBAL, MSGL_WARN, "[%p] Warning: no style named '%s' found, using '%s'\n", track, name, track->styles[i].Name); | |
171 return i; // use the first style | |
172 } | |
173 | |
174 static uint32_t string2color(char* p) { | |
175 uint32_t tmp; | |
176 (void)strtocolor(&p, &tmp); | |
177 return tmp; | |
178 } | |
179 | |
180 static long long string2timecode(char* p) { | |
181 unsigned h, m, s, ms; | |
182 long long tm; | |
183 int res = sscanf(p, "%1d:%2d:%2d.%2d", &h, &m, &s, &ms); | |
184 if (res < 4) { | |
185 mp_msg(MSGT_GLOBAL, MSGL_WARN, "bad timestamp\n"); | |
186 return 0; | |
187 } | |
188 tm = ((h * 60 + m) * 60 + s) * 1000 + ms * 10; | |
189 return tm; | |
190 } | |
191 | |
192 /** | |
193 * \brief converts numpad-style align to align. | |
194 */ | |
195 static int numpad2align(int val) { | |
196 int res, v; | |
197 v = (val - 1) / 3; // 0, 1 or 2 for vertical alignment | |
198 if (v != 0) v = 3 - v; | |
199 res = ((val - 1) % 3) + 1; // horizontal alignment | |
200 res += v*4; | |
201 return res; | |
202 } | |
203 | |
204 #define NEXT(str,token) \ | |
205 token = next_token(&str); \ | |
206 if (!token) break; | |
207 | |
208 #define ANYVAL(name,func) \ | |
209 } else if (strcasecmp(tname, #name) == 0) { \ | |
210 target->name = func(token); \ | |
211 mp_msg(MSGT_GLOBAL, MSGL_DBG2, "%s = %s\n", #name, token); | |
212 #define STRVAL(name) ANYVAL(name,strdup) | |
213 #define COLORVAL(name) ANYVAL(name,string2color) | |
214 #define INTVAL(name) ANYVAL(name,atoi) | |
215 #define FPVAL(name) ANYVAL(name,atof) | |
216 #define TIMEVAL(name) ANYVAL(name,string2timecode) | |
217 #define STYLEVAL(name) \ | |
218 } else if (strcasecmp(tname, #name) == 0) { \ | |
219 target->name = lookup_style(track, token); \ | |
220 mp_msg(MSGT_GLOBAL, MSGL_DBG2, "%s = %s\n", #name, token); | |
221 | |
222 #define ALIAS(alias,name) \ | |
223 if (strcasecmp(tname, #alias) == 0) {tname = #name;} | |
224 | |
225 static char* next_token(char** str) { | |
226 char* p = *str; | |
227 char* start; | |
228 skip_spaces(&p); | |
229 if (*p == '\0') { | |
230 *str = p; | |
231 return 0; | |
232 } | |
233 start = p; // start of the token | |
234 for (; (*p != '\0') && (*p != ','); ++p) {} | |
235 if (*p == '\0') { | |
236 *str = p; // eos found, str will point to '\0' at exit | |
237 } else { | |
238 *p = '\0'; | |
239 *str = p + 1; // ',' found, str will point to the next char (beginning of the next token) | |
240 } | |
241 --p; // end of current token | |
242 rskip_spaces(&p, start); | |
243 if (p < start) | |
244 p = start; // empty token | |
245 else | |
246 ++p; // the first space character, or '\0' | |
247 *p = '\0'; | |
248 return start; | |
249 } | |
250 /** | |
251 * \brief Parse the tail of Dialogue line | |
252 * \param track track | |
253 * \param event parsed data goes here | |
254 * \param str string to parse, zero-terminated | |
255 * \param n_ignored number of format options to skip at the beginning | |
256 */ | |
257 static int process_event_tail(ass_track_t* track, ass_event_t* event, char* str, int n_ignored) | |
258 { | |
259 char* token; | |
260 char* tname; | |
261 char* p = str; | |
262 int i; | |
263 ass_event_t* target = event; | |
264 | |
265 char* format = strdup(track->event_format); | |
266 char* q = format; // format scanning pointer | |
267 | |
268 for (i = 0; i < n_ignored; ++i) { | |
269 NEXT(q, tname); | |
270 } | |
271 | |
272 while (1) { | |
273 NEXT(q, tname); | |
274 if (strcasecmp(tname, "Text") == 0) { | |
275 char* last; | |
276 event->Text = strdup(p); | |
277 last = event->Text + strlen(event->Text) - 1; | |
278 if (*last == '\r') | |
279 *last = 0; | |
280 mp_msg(MSGT_GLOBAL, MSGL_DBG2, "Text = %s\n", event->Text); | |
281 event->Duration -= event->Start; | |
282 free(format); | |
283 return 0; // "Text" is always the last | |
284 } | |
285 NEXT(p, token); | |
286 | |
287 ALIAS(End,Duration) // temporarily store end timecode in event->Duration | |
288 if (0) { // cool ;) | |
289 STYLEVAL(Style) | |
290 STRVAL(Name) | |
291 STRVAL(Effect) | |
292 INTVAL(MarginL) | |
293 INTVAL(MarginR) | |
294 INTVAL(MarginV) | |
295 TIMEVAL(Start) | |
296 TIMEVAL(Duration) | |
297 } | |
298 } | |
299 free(format); | |
300 return 1; | |
301 } | |
302 | |
303 /** | |
304 * \brief Parse the Style line | |
305 * \param track track | |
306 * \param str string to parse, zero-terminated | |
307 * Allocates a new style struct. | |
308 */ | |
309 static int process_style(ass_track_t* track, char *str) | |
310 { | |
311 | |
312 char* token; | |
313 char* tname; | |
314 char* p = str; | |
315 char* format; | |
316 char* q; // format scanning pointer | |
317 int sid; | |
318 ass_style_t* style; | |
319 ass_style_t* target; | |
320 | |
321 if (!track->style_format) { | |
322 // no style format header | |
323 // probably an ancient script version | |
324 if (track->track_type == TRACK_TYPE_SSA) | |
325 track->style_format = strdup("Name, Fontname, Fontsize, PrimaryColour, SecondaryColour," | |
326 "TertiaryColour, BackColour, Bold, Italic, BorderStyle, Outline," | |
327 "Shadow, Alignment, MarginL, MarginR, MarginV, AlphaLevel, Encoding"); | |
328 else | |
329 track->style_format = strdup("Name, Fontname, Fontsize, PrimaryColour, SecondaryColour," | |
330 "OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut," | |
331 "ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow," | |
332 "Alignment, MarginL, MarginR, MarginV, Encoding"); | |
333 } | |
334 | |
335 q = format = strdup(track->style_format); | |
336 | |
337 mp_msg(MSGT_GLOBAL, MSGL_V, "[%p] Style: %s\n", track, str); | |
338 | |
339 sid = ass_alloc_style(track); | |
340 | |
341 style = track->styles + sid; | |
342 target = style; | |
343 // fill style with some default values | |
344 style->ScaleX = 100.; | |
345 style->ScaleY = 100.; | |
346 | |
347 while (1) { | |
348 NEXT(q, tname); | |
349 NEXT(p, token); | |
350 | |
351 // ALIAS(TertiaryColour,OutlineColour) // ignore TertiaryColour; it appears only in SSA, and is overridden by BackColour | |
352 | |
353 if (0) { // cool ;) | |
354 STRVAL(Name) | |
355 if ((strcmp(target->Name, "Default")==0) || (strcmp(target->Name, "*Default")==0)) | |
356 track->default_style = sid; | |
357 STRVAL(FontName) | |
358 COLORVAL(PrimaryColour) | |
359 COLORVAL(SecondaryColour) | |
360 COLORVAL(OutlineColour) // TertiaryColor | |
361 COLORVAL(BackColour) | |
362 // SSA uses BackColour for both outline and shadow | |
363 // this will destroy SSA's TertiaryColour, but i'm not going to use it anyway | |
364 if (track->track_type == TRACK_TYPE_SSA) | |
365 target->OutlineColour = target->BackColour; | |
366 INTVAL(FontSize) | |
367 INTVAL(Bold) | |
368 INTVAL(Italic) | |
369 INTVAL(Underline) | |
370 INTVAL(StrikeOut) | |
371 INTVAL(Spacing) | |
372 INTVAL(Angle) | |
373 INTVAL(BorderStyle) | |
374 INTVAL(Alignment) | |
375 if (track->track_type == TRACK_TYPE_ASS) | |
376 target->Alignment = numpad2align(target->Alignment); | |
377 INTVAL(MarginL) | |
378 INTVAL(MarginR) | |
379 INTVAL(MarginV) | |
380 INTVAL(Encoding) | |
381 FPVAL(ScaleX) | |
382 FPVAL(ScaleY) | |
383 FPVAL(Outline) | |
384 FPVAL(Shadow) | |
385 } | |
386 } | |
387 style->ScaleX /= 100.; | |
388 style->ScaleY /= 100.; | |
389 if (!style->Name) | |
390 style->Name = strdup("Default"); | |
391 if (!style->FontName) | |
392 style->FontName = strdup("Arial"); | |
393 free(format); | |
394 return 0; | |
395 | |
396 } | |
397 | |
19492 | 398 static int process_styles_line(ass_track_t* track, char *str) |
399 { | |
400 if (!strncmp(str,"Format:", 7)) { | |
401 char* p = str + 7; | |
402 skip_spaces(&p); | |
403 track->style_format = strdup(p); | |
404 mp_msg(MSGT_GLOBAL, MSGL_DBG2, "Style format: %s\n", track->style_format); | |
405 } else if (!strncmp(str,"Style:", 6)) { | |
406 char* p = str + 6; | |
407 skip_spaces(&p); | |
408 process_style(track, p); | |
409 } | |
410 return 0; | |
411 } | |
412 | |
413 static int process_info_line(ass_track_t* track, char *str) | |
414 { | |
415 if (!strncmp(str, "PlayResX:", 9)) { | |
416 track->PlayResX = atoi(str + 9); | |
417 } else if (!strncmp(str,"PlayResY:", 9)) { | |
418 track->PlayResY = atoi(str + 9); | |
419 } else if (!strncmp(str,"Timer:", 6)) { | |
420 track->Timer = atof(str + 6); | |
421 } else if (!strncmp(str,"WrapStyle:", 10)) { | |
422 track->WrapStyle = atoi(str + 10); | |
423 } | |
424 return 0; | |
425 } | |
426 | |
427 static int process_events_line(ass_track_t* track, char *str) | |
428 { | |
429 if (!strncmp(str, "Format:", 7)) { | |
430 char* p = str + 7; | |
431 skip_spaces(&p); | |
432 track->event_format = strdup(p); | |
433 mp_msg(MSGT_GLOBAL, MSGL_DBG2, "Event format: %s\n", track->event_format); | |
434 } else if (!strncmp(str, "Dialogue:", 9)) { | |
435 // This should never be reached for embedded subtitles. | |
436 // They have slightly different format and are parsed in ass_process_chunk, | |
437 // called directly from demuxer | |
438 int eid; | |
439 ass_event_t* event; | |
440 | |
441 str += 9; | |
442 skip_spaces(&str); | |
443 | |
444 eid = ass_alloc_event(track); | |
445 event = track->events + eid; | |
446 | |
447 process_event_tail(track, event, str, 0); | |
448 } else { | |
449 mp_msg(MSGT_GLOBAL, MSGL_V, "Not understood: %s \n", str); | |
450 } | |
451 return 0; | |
452 } | |
453 | |
454 // Copied from mkvtoolnix | |
455 static unsigned char* decode_chars(unsigned char c1, unsigned char c2, | |
456 unsigned char c3, unsigned char c4, unsigned char* dst, int cnt) | |
457 { | |
458 uint32_t value; | |
459 unsigned char bytes[3]; | |
460 int i; | |
461 | |
462 value = ((c1 - 33) << 18) + ((c2 - 33) << 12) + ((c3 - 33) << 6) + (c4 - 33); | |
463 bytes[2] = value & 0xff; | |
464 bytes[1] = (value & 0xff00) >> 8; | |
465 bytes[0] = (value & 0xff0000) >> 16; | |
466 | |
467 for (i = 0; i < cnt; ++i) | |
468 *dst++ = bytes[i]; | |
469 return dst; | |
470 } | |
471 | |
472 static int decode_font(ass_track_t* track) | |
473 { | |
474 unsigned char* p; | |
475 unsigned char* q; | |
476 int i; | |
477 int size; // original size | |
478 int dsize; // decoded size | |
479 unsigned char* buf = 0; | |
480 | |
481 mp_msg(MSGT_GLOBAL, MSGL_V, "font: %d bytes encoded data \n", track->parser_priv->fontdata_used); | |
482 size = track->parser_priv->fontdata_used; | |
483 if (size % 4 == 1) { | |
484 mp_msg(MSGT_GLOBAL, MSGL_ERR, "bad encoded data size\n"); | |
485 goto error_decode_font; | |
486 } | |
487 buf = malloc(size / 4 * 3 + 2); | |
488 q = buf; | |
489 for (i = 0, p = (unsigned char*)track->parser_priv->fontdata; i < size / 4; i++, p+=4) { | |
490 q = decode_chars(p[0], p[1], p[2], p[3], q, 3); | |
491 } | |
492 if (size % 4 == 2) { | |
493 q = decode_chars(p[0], p[1], 0, 0, q, 1); | |
494 } else if (size % 4 == 3) { | |
495 q = decode_chars(p[0], p[1], p[2], 0, q, 2); | |
496 } | |
497 dsize = q - buf; | |
498 assert(dsize <= size / 4 * 3 + 2); | |
499 | |
500 if (extract_embedded_fonts) | |
501 ass_process_font(track->parser_priv->fontname, (char*)buf, dsize); | |
502 | |
503 error_decode_font: | |
504 if (buf) free(buf); | |
505 free(track->parser_priv->fontname); | |
506 free(track->parser_priv->fontdata); | |
507 track->parser_priv->fontname = 0; | |
508 track->parser_priv->fontdata = 0; | |
509 track->parser_priv->fontdata_size = 0; | |
510 track->parser_priv->fontdata_used = 0; | |
511 return 0; | |
512 } | |
513 | |
514 static char* validate_fname(char* name); | |
515 | |
516 static int process_fonts_line(ass_track_t* track, char *str) | |
517 { | |
518 int len; | |
519 | |
520 if (!strncmp(str, "fontname:", 9)) { | |
521 char* p = str + 9; | |
522 skip_spaces(&p); | |
523 if (track->parser_priv->fontname) { | |
524 decode_font(track); | |
525 } | |
526 track->parser_priv->fontname = validate_fname(p); | |
527 mp_msg(MSGT_GLOBAL, MSGL_V, "fontname: %s\n", track->parser_priv->fontname); | |
528 return 0; | |
529 } | |
530 | |
531 if (!track->parser_priv->fontname) { | |
532 mp_msg(MSGT_GLOBAL, MSGL_V, "Not understood: %s \n", str); | |
533 return 0; | |
534 } | |
535 | |
536 len = strlen(str); | |
537 if (len > 80) { | |
538 mp_msg(MSGT_GLOBAL, MSGL_WARN, "Font line too long: %d, %s\n", len, str); | |
539 return 0; | |
540 } | |
541 if (track->parser_priv->fontdata_used + len > track->parser_priv->fontdata_size) { | |
542 track->parser_priv->fontdata_size += 100 * 1024; | |
543 track->parser_priv->fontdata = realloc(track->parser_priv->fontdata, track->parser_priv->fontdata_size); | |
544 } | |
545 memcpy(track->parser_priv->fontdata + track->parser_priv->fontdata_used, str, len); | |
546 track->parser_priv->fontdata_used += len; | |
547 | |
548 return 0; | |
549 } | |
550 | |
18937 | 551 /** |
552 * \brief Parse a header line | |
553 * \param track track | |
554 * \param str string to parse, zero-terminated | |
555 */ | |
19492 | 556 static int process_line(ass_track_t* track, char *str) |
18937 | 557 { |
19492 | 558 if (strstr(str, "[Script Info]")) { // FIXME: strstr to skip possible BOM at the beginning of the script |
559 track->parser_priv->state = PST_INFO; | |
560 } else if (!strncmp(str, "[V4 Styles]", 11)) { | |
561 track->parser_priv->state = PST_STYLES; | |
562 track->track_type = TRACK_TYPE_SSA; | |
563 } else if (!strncmp(str, "[V4+ Styles]", 12)) { | |
564 track->parser_priv->state = PST_STYLES; | |
565 track->track_type = TRACK_TYPE_ASS; | |
566 } else if (!strncmp(str, "[Events]", 8)) { | |
567 track->parser_priv->state = PST_EVENTS; | |
568 } else if (!strncmp(str, "[Fonts]", 7)) { | |
569 track->parser_priv->state = PST_FONTS; | |
570 } else { | |
571 switch (track->parser_priv->state) { | |
572 case PST_INFO: | |
573 process_info_line(track, str); | |
574 break; | |
575 case PST_STYLES: | |
576 process_styles_line(track, str); | |
577 break; | |
578 case PST_EVENTS: | |
579 process_events_line(track, str); | |
580 break; | |
581 case PST_FONTS: | |
582 process_fonts_line(track, str); | |
583 break; | |
584 default: | |
585 break; | |
18937 | 586 } |
19492 | 587 } |
588 | |
589 // there is no explicit end-of-font marker in ssa/ass | |
590 if ((track->parser_priv->state != PST_FONTS) && (track->parser_priv->fontname)) | |
591 decode_font(track); | |
592 | |
593 return 0; | |
594 } | |
595 | |
596 static int process_text(ass_track_t* track, char* str) | |
597 { | |
598 char* p = str; | |
599 while(1) { | |
600 char* q; | |
601 for (;((*p=='\r')||(*p=='\n'));++p) {} | |
602 for (q=p; ((*q!='\0')&&(*q!='\r')&&(*q!='\n')); ++q) {}; | |
603 if (q==p) | |
604 break; | |
605 if (*q != '\0') | |
606 *(q++) = '\0'; | |
607 process_line(track, p); | |
608 if (*q == '\0') | |
609 break; | |
610 p = q; | |
18937 | 611 } |
612 return 0; | |
613 } | |
614 | |
615 /** | |
616 * \brief Process CodecPrivate section of subtitle stream | |
617 * \param track track | |
618 * \param data string to parse | |
619 * \param size length of data | |
19492 | 620 CodecPrivate section contains [Stream Info] and [V4+ Styles] ([V4 Styles] for SSA) sections |
18937 | 621 */ |
19492 | 622 void ass_process_codec_private(ass_track_t* track, char *data, int size) |
18937 | 623 { |
624 char* str = malloc(size + 1); | |
625 int sid; | |
626 | |
627 memcpy(str, data, size); | |
628 str[size] = '\0'; | |
629 | |
19492 | 630 process_text(track, str); |
18937 | 631 free(str); |
632 | |
633 // add "Default" style to the end | |
634 // will be used if track does not contain a default style (or even does not contain styles at all) | |
635 sid = ass_alloc_style(track); | |
636 track->styles[sid].Name = strdup("Default"); | |
637 track->styles[sid].FontName = strdup("Arial"); | |
638 | |
639 if (!track->event_format) { | |
640 // probably an mkv produced by ancient mkvtoolnix | |
641 // such files don't have [Events] and Format: headers | |
19492 | 642 track->parser_priv->state = PST_EVENTS; |
18937 | 643 if (track->track_type == TRACK_TYPE_SSA) |
644 track->event_format = strdup("Format: Marked, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text"); | |
645 else | |
646 track->event_format = strdup("Format: Layer, Start, End, Style, Actor, MarginL, MarginR, MarginV, Effect, Text"); | |
647 } | |
648 } | |
649 | |
650 static int check_duplicate_event(ass_track_t* track, int ReadOrder) | |
651 { | |
652 int i; | |
653 for (i = 0; i<track->n_events - 1; ++i) // ignoring last event, it is the one we are comparing with | |
654 if (track->events[i].ReadOrder == ReadOrder) | |
655 return 1; | |
656 return 0; | |
657 } | |
658 | |
659 /** | |
660 * \brief Process a chunk of subtitle stream data. In matroska, this containes exactly 1 event (or a commentary) | |
661 * \param track track | |
662 * \param data string to parse | |
663 * \param size length of data | |
664 * \param timecode starting time of the event (milliseconds) | |
665 * \param duration duration of the event (milliseconds) | |
666 */ | |
19492 | 667 void ass_process_chunk(ass_track_t* track, char *data, int size, long long timecode, long long duration) |
18937 | 668 { |
669 char* str; | |
670 int eid; | |
671 char* p; | |
672 char* token; | |
673 ass_event_t* event; | |
674 | |
675 if (!track->event_format) { | |
676 mp_msg(MSGT_GLOBAL, MSGL_WARN, "Event format header missing\n"); | |
677 return; | |
678 } | |
679 | |
680 str = malloc(size + 1); | |
681 memcpy(str, data, size); | |
682 str[size] = '\0'; | |
19378 | 683 mp_msg(MSGT_GLOBAL, MSGL_V, "event at %" PRId64 ", +%" PRId64 ": %s \n", (int64_t)timecode, (int64_t)duration, str); |
18937 | 684 |
685 eid = ass_alloc_event(track); | |
686 event = track->events + eid; | |
687 | |
688 p = str; | |
689 | |
690 do { | |
691 NEXT(p, token); | |
692 event->ReadOrder = atoi(token); | |
693 if (check_duplicate_event(track, event->ReadOrder)) | |
694 break; | |
695 | |
696 NEXT(p, token); | |
697 event->Layer = atoi(token); | |
698 | |
699 process_event_tail(track, event, p, 3); | |
700 | |
701 event->Start = timecode; | |
702 event->Duration = duration; | |
703 | |
704 free(str); | |
705 return; | |
706 // dump_events(tid); | |
707 } while (0); | |
708 // some error | |
19474
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
709 ass_free_event(track, eid); |
07209f48e527
Add public functions for removal of styles and events.
eugeni
parents:
19405
diff
changeset
|
710 track->n_events--; |
18937 | 711 free(str); |
712 } | |
713 | |
714 #ifdef USE_ICONV | |
715 /** \brief recode buffer to utf-8 | |
716 * constraint: sub_cp != 0 | |
717 * \param data pointer to text buffer | |
718 * \param size buffer size | |
719 * \return a pointer to recoded buffer, caller is responsible for freeing it | |
720 **/ | |
721 static char* sub_recode(char* data, size_t size) | |
722 { | |
723 static iconv_t icdsc = (iconv_t)(-1); | |
724 char* tocp = "UTF-8"; | |
725 char* outbuf; | |
726 assert(sub_cp); | |
727 | |
728 { | |
729 char* cp_tmp = sub_cp; | |
730 #ifdef HAVE_ENCA | |
731 char enca_lang[3], enca_fallback[100]; | |
732 if (sscanf(sub_cp, "enca:%2s:%99s", enca_lang, enca_fallback) == 2 | |
733 || sscanf(sub_cp, "ENCA:%2s:%99s", enca_lang, enca_fallback) == 2) { | |
734 cp_tmp = guess_buffer_cp((unsigned char*)data, size, enca_lang, enca_fallback); | |
735 } | |
736 #endif | |
737 if ((icdsc = iconv_open (tocp, cp_tmp)) != (iconv_t)(-1)){ | |
738 mp_msg(MSGT_SUBREADER,MSGL_V,"LIBSUB: opened iconv descriptor.\n"); | |
739 } else | |
740 mp_msg(MSGT_SUBREADER,MSGL_ERR,"LIBSUB: error opening iconv descriptor.\n"); | |
741 #ifdef HAVE_ENCA | |
742 if (cp_tmp) free(cp_tmp); | |
743 #endif | |
744 } | |
745 | |
746 { | |
747 size_t osize = size; | |
748 size_t ileft = size; | |
749 size_t oleft = size - 1; | |
750 char* ip; | |
751 char* op; | |
752 size_t rc; | |
753 | |
754 outbuf = malloc(size); | |
755 ip = data; | |
756 op = outbuf; | |
757 | |
758 while (ileft) { | |
759 rc = iconv(icdsc, &ip, &ileft, &op, &oleft); | |
760 if (rc == (size_t)(-1)) { | |
761 if (errno == E2BIG) { | |
762 int offset = op - outbuf; | |
763 outbuf = (char*)realloc(outbuf, osize + size); | |
764 op = outbuf + offset; | |
765 osize += size; | |
766 oleft += size; | |
767 } else { | |
768 mp_msg(MSGT_SUBREADER, MSGL_WARN, "LIBSUB: error recoding file.\n"); | |
769 return NULL; | |
770 } | |
771 } | |
772 } | |
773 outbuf[osize - oleft - 1] = 0; | |
774 } | |
775 | |
776 if (icdsc != (iconv_t)(-1)) { | |
777 (void)iconv_close(icdsc); | |
778 icdsc = (iconv_t)(-1); | |
779 mp_msg(MSGT_SUBREADER,MSGL_V,"LIBSUB: closed iconv descriptor.\n"); | |
780 } | |
781 | |
782 return outbuf; | |
783 } | |
784 #endif // ICONV | |
785 | |
786 /** | |
787 * \brief Read subtitles from file. | |
788 * \param fname file name | |
789 * \return newly allocated track | |
790 */ | |
791 ass_track_t* ass_read_file(char* fname) | |
792 { | |
793 int res; | |
794 long sz; | |
795 long bytes_read; | |
796 char* buf; | |
797 ass_track_t* track; | |
798 | |
799 FILE* fp = fopen(fname, "rb"); | |
800 if (!fp) { | |
801 mp_msg(MSGT_GLOBAL, MSGL_WARN, "ass_read_file(%s): fopen failed\n", fname); | |
802 return 0; | |
803 } | |
804 res = fseek(fp, 0, SEEK_END); | |
805 if (res == -1) { | |
806 mp_msg(MSGT_GLOBAL, MSGL_WARN, "ass_read_file(%s): fseek failed\n", fname); | |
807 fclose(fp); | |
808 return 0; | |
809 } | |
810 | |
811 sz = ftell(fp); | |
812 rewind(fp); | |
813 | |
814 if (sz > 10*1024*1024) { | |
815 mp_msg(MSGT_GLOBAL, MSGL_INFO, "ass_read_file(%s): Refusing to load subtitles larger than 10M\n", fname); | |
816 fclose(fp); | |
817 return 0; | |
818 } | |
819 | |
820 mp_msg(MSGT_GLOBAL, MSGL_V, "file size: %ld\n", sz); | |
821 | |
822 buf = malloc(sz + 1); | |
823 assert(buf); | |
824 bytes_read = 0; | |
825 do { | |
826 res = fread(buf + bytes_read, 1, sz - bytes_read, fp); | |
827 if (res <= 0) { | |
828 mp_msg(MSGT_GLOBAL, MSGL_INFO, "Read failed, %d: %s\n", errno, strerror(errno)); | |
829 fclose(fp); | |
830 free(buf); | |
831 return 0; | |
832 } | |
833 bytes_read += res; | |
834 } while (sz - bytes_read > 0); | |
835 buf[sz] = '\0'; | |
836 fclose(fp); | |
837 | |
838 #ifdef USE_ICONV | |
839 if (sub_cp) { | |
840 char* tmpbuf = sub_recode(buf, sz); | |
841 free(buf); | |
842 if (!tmpbuf) | |
843 return 0; | |
844 buf = tmpbuf; | |
845 } | |
846 #endif | |
847 | |
848 track = ass_new_track(); | |
849 track->name = strdup(fname); | |
850 | |
851 // process header | |
19492 | 852 process_text(track, buf); |
853 | |
854 // there is no explicit end-of-font marker in ssa/ass | |
855 if (track->parser_priv->fontname) | |
856 decode_font(track); | |
857 | |
18937 | 858 free(buf); |
859 | |
19492 | 860 if (track->track_type == TRACK_TYPE_UNKNOWN) { |
18937 | 861 ass_free_track(track); |
862 return 0; | |
863 } | |
864 | |
865 mp_msg(MSGT_GLOBAL, MSGL_INFO, "LIBASS: added subtitle file: %s (%d styles, %d events)\n", fname, track->n_styles, track->n_events); | |
866 | |
867 sort_events(track); | |
868 | |
869 // dump_events(forced_tid); | |
870 return track; | |
871 } | |
872 | |
873 static char* validate_fname(char* name) | |
874 { | |
875 char* fname; | |
876 char* p; | |
877 char* q; | |
878 unsigned code; | |
879 int sz = strlen(name); | |
880 | |
881 q = fname = malloc(sz + 1); | |
882 p = name; | |
883 while (*p) { | |
884 code = utf8_get_char(&p); | |
885 if (code == 0) | |
886 break; | |
887 if ( (code > 0x7F) || | |
888 (code == '\\') || | |
889 (code == '/') || | |
890 (code == ':') || | |
891 (code == '*') || | |
892 (code == '?') || | |
893 (code == '<') || | |
894 (code == '>') || | |
895 (code == '|') || | |
896 (code == 0)) | |
897 { | |
898 *q++ = '_'; | |
899 } else { | |
900 *q++ = code; | |
901 } | |
902 if (p - name > sz) | |
903 break; | |
904 } | |
905 *q = 0; | |
906 return fname; | |
907 } | |
908 | |
909 /** | |
910 * \brief Process embedded matroska font. Saves it to ~/.mplayer/fonts. | |
911 * \param name attachment name | |
912 * \param data binary font data | |
913 * \param data_size data size | |
914 */ | |
915 void ass_process_font(const char* name, char* data, int data_size) | |
916 { | |
917 char buf[1000]; | |
918 FILE* fp = 0; | |
919 int rc; | |
920 struct stat st; | |
921 char* fname; | |
922 | |
923 char* fonts_dir = get_path("fonts"); | |
924 rc = stat(fonts_dir, &st); | |
925 if (rc) { | |
926 int res; | |
927 #ifndef __MINGW32__ | |
928 res = mkdir(fonts_dir, 0700); | |
929 #else | |
930 res = mkdir(fonts_dir); | |
931 #endif | |
932 if (res) { | |
933 mp_msg(MSGT_GLOBAL, MSGL_WARN, "Failed to create: %s\n", fonts_dir); | |
934 } | |
935 } else if (!S_ISDIR(st.st_mode)) { | |
936 mp_msg(MSGT_GLOBAL, MSGL_WARN, "Not a directory: %s\n", fonts_dir); | |
937 } | |
938 | |
939 fname = validate_fname((char*)name); | |
940 | |
941 snprintf(buf, 1000, "%s/%s", fonts_dir, fname); | |
942 free(fname); | |
943 free(fonts_dir); | |
944 | |
945 fp = fopen(buf, "wb"); | |
946 if (!fp) return; | |
947 | |
948 fwrite(data, data_size, 1, fp); | |
949 fclose(fp); | |
950 } | |
951 | |
952 long long ass_step_sub(ass_track_t* track, long long now, int movement) { | |
953 int i; | |
954 | |
955 if (movement == 0) return 0; | |
956 if (track->n_events == 0) return 0; | |
957 | |
958 if (movement < 0) | |
959 for (i = 0; (i < track->n_events) && ((long long)(track->events[i].Start + track->events[i].Duration) <= now); ++i) {} | |
960 else | |
961 for (i = track->n_events - 1; (i >= 0) && ((long long)(track->events[i].Start) > now); --i) {} | |
962 | |
963 // -1 and n_events are ok | |
964 assert(i >= -1); assert(i <= track->n_events); | |
965 i += movement; | |
966 if (i < 0) i = 0; | |
967 if (i >= track->n_events) i = track->n_events - 1; | |
968 return ((long long)track->events[i].Start) - now; | |
969 } | |
970 | |
971 ass_track_t* ass_new_track(void) { | |
972 ass_track_t* track = calloc(1, sizeof(ass_track_t)); | |
19492 | 973 track->parser_priv = calloc(1, sizeof(parser_priv_t)); |
18937 | 974 return track; |
975 } | |
976 |