vo_dxr3: Fix validity checks before closing a file descriptor
[mplayer/glamo.git] / subreader.c
blobdcadc3083ee1cffda22c27a90d965b823ad0e50f
1 /*
2 * Subtitle reader with format autodetection
4 * Copyright (c) 2001 laaz
5 * Some code cleanup & realloc() by A'rpi/ESP-team
7 * This file is part of MPlayer.
9 * MPlayer is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * MPlayer is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License along
20 * with MPlayer; if not, write to the Free Software Foundation, Inc.,
21 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <string.h>
27 #include <ctype.h>
29 #include <sys/types.h>
30 #include <dirent.h>
32 #include "config.h"
33 #include "mp_msg.h"
34 #include "subreader.h"
35 #include "stream/stream.h"
36 #include "libavutil/common.h"
37 #include "libavutil/avstring.h"
39 #ifdef CONFIG_ENCA
40 #include <enca.h>
41 #endif
43 #define ERR ((void *) -1)
45 #ifdef CONFIG_ICONV
46 #include <iconv.h>
47 char *sub_cp=NULL;
48 #endif
49 #ifdef CONFIG_FRIBIDI
50 #include <fribidi/fribidi.h>
51 char *fribidi_charset = NULL; ///character set that will be passed to FriBiDi
52 int flip_hebrew = 1; ///flip subtitles using fribidi
53 int fribidi_flip_commas = 0; ///flip comma when fribidi is used
54 #endif
56 /* Maximal length of line of a subtitle */
57 #define LINE_LEN 1000
58 static float mpsub_position=0;
59 static float mpsub_multiplier=1.;
60 static int sub_slacktime = 20000; //20 sec
62 int sub_no_text_pp=0; // 1 => do not apply text post-processing
63 // like {\...} elimination in SSA format.
65 int sub_match_fuzziness=0; // level of sub name matching fuzziness
67 /* Use the SUB_* constant defined in the header file */
68 int sub_format=SUB_INVALID;
69 #ifdef CONFIG_SORTSUB
71 Some subtitling formats, namely AQT and Subrip09, define the end of a
72 subtitle as the beginning of the following. Since currently we read one
73 subtitle at time, for these format we keep two global *subtitle,
74 previous_aqt_sub and previous_subrip09_sub, pointing to previous subtitle,
75 so we can change its end when we read current subtitle starting time.
76 When CONFIG_SORTSUB is defined, we use a single global unsigned long,
77 previous_sub_end, for both (and even future) formats, to store the end of
78 the previous sub: it is initialized to 0 in sub_read_file and eventually
79 modified by sub_read_aqt_line or sub_read_subrip09_line.
81 unsigned long previous_sub_end;
82 #endif
84 static int eol(char p) {
85 return p=='\r' || p=='\n' || p=='\0';
88 /* Remove leading and trailing space */
89 static void trail_space(char *s) {
90 int i = 0;
91 while (isspace(s[i])) ++i;
92 if (i) strcpy(s, s + i);
93 i = strlen(s) - 1;
94 while (i > 0 && isspace(s[i])) s[i--] = '\0';
97 static char *stristr(const char *haystack, const char *needle) {
98 int len = 0;
99 const char *p = haystack;
101 if (!(haystack && needle)) return NULL;
103 len=strlen(needle);
104 while (*p != '\0') {
105 if (strncasecmp(p, needle, len) == 0) return (char*)p;
106 p++;
109 return NULL;
112 static void sami_add_line(subtitle *current, char *buffer, char **pos) {
113 char *p = *pos;
114 *p = 0;
115 trail_space(buffer);
116 if (*buffer && current->lines < SUB_MAX_TEXT)
117 current->text[current->lines++] = strdup(buffer);
118 *pos = buffer;
121 static subtitle *sub_read_line_sami(stream_t* st, subtitle *current, int utf16) {
122 static char line[LINE_LEN+1];
123 static char *s = NULL, *slacktime_s;
124 char text[LINE_LEN+1], *p=NULL, *q;
125 int state;
127 current->lines = current->start = current->end = 0;
128 current->alignment = SUB_ALIGNMENT_BOTTOMCENTER;
129 state = 0;
131 /* read the first line */
132 if (!s)
133 if (!(s = stream_read_line(st, line, LINE_LEN, utf16))) return 0;
135 do {
136 switch (state) {
138 case 0: /* find "START=" or "Slacktime:" */
139 slacktime_s = stristr (s, "Slacktime:");
140 if (slacktime_s)
141 sub_slacktime = strtol (slacktime_s+10, NULL, 0) / 10;
143 s = stristr (s, "Start=");
144 if (s) {
145 current->start = strtol (s + 6, &s, 0) / 10;
146 /* eat '>' */
147 for (; *s != '>' && *s != '\0'; s++);
148 s++;
149 state = 1; continue;
151 break;
153 case 1: /* find (optional) "<P", skip other TAGs */
154 for (; *s == ' ' || *s == '\t'; s++); /* strip blanks, if any */
155 if (*s == '\0') break;
156 if (*s != '<') { state = 3; p = text; continue; } /* not a TAG */
157 s++;
158 if (*s == 'P' || *s == 'p') { s++; state = 2; continue; } /* found '<P' */
159 for (; *s != '>' && *s != '\0'; s++); /* skip remains of non-<P> TAG */
160 if (s == '\0')
161 break;
162 s++;
163 continue;
165 case 2: /* find ">" */
166 if ((s = strchr (s, '>'))) { s++; state = 3; p = text; continue; }
167 break;
169 case 3: /* get all text until '<' appears */
170 if (*s == '\0') break;
171 else if (!strncasecmp (s, "<br>", 4)) {
172 sami_add_line(current, text, &p);
173 s += 4;
175 else if ((*s == '{') && !sub_no_text_pp) { state = 5; ++s; continue; }
176 else if (*s == '<') { state = 4; }
177 else if (!strncasecmp (s, "&nbsp;", 6)) { *p++ = ' '; s += 6; }
178 else if (*s == '\t') { *p++ = ' '; s++; }
179 else if (*s == '\r' || *s == '\n') { s++; }
180 else *p++ = *s++;
182 /* skip duplicated space */
183 if (p > text + 2) if (*(p-1) == ' ' && *(p-2) == ' ') p--;
185 continue;
187 case 4: /* get current->end or skip <TAG> */
188 q = stristr (s, "Start=");
189 if (q) {
190 current->end = strtol (q + 6, &q, 0) / 10 - 1;
191 *p = '\0'; trail_space (text);
192 if (text[0] != '\0')
193 current->text[current->lines++] = strdup (text);
194 if (current->lines > 0) { state = 99; break; }
195 state = 0; continue;
197 s = strchr (s, '>');
198 if (s) { s++; state = 3; continue; }
199 break;
200 case 5: /* get rid of {...} text, but read the alignment code */
201 if ((*s == '\\') && (*(s + 1) == 'a') && !sub_no_text_pp) {
202 if (stristr(s, "\\a1") != NULL) {
203 current->alignment = SUB_ALIGNMENT_BOTTOMLEFT;
204 s = s + 3;
206 if (stristr(s, "\\a2") != NULL) {
207 current->alignment = SUB_ALIGNMENT_BOTTOMCENTER;
208 s = s + 3;
209 } else if (stristr(s, "\\a3") != NULL) {
210 current->alignment = SUB_ALIGNMENT_BOTTOMRIGHT;
211 s = s + 3;
212 } else if ((stristr(s, "\\a4") != NULL) || (stristr(s, "\\a5") != NULL) || (stristr(s, "\\a8") != NULL)) {
213 current->alignment = SUB_ALIGNMENT_TOPLEFT;
214 s = s + 3;
215 } else if (stristr(s, "\\a6") != NULL) {
216 current->alignment = SUB_ALIGNMENT_TOPCENTER;
217 s = s + 3;
218 } else if (stristr(s, "\\a7") != NULL) {
219 current->alignment = SUB_ALIGNMENT_TOPRIGHT;
220 s = s + 3;
221 } else if (stristr(s, "\\a9") != NULL) {
222 current->alignment = SUB_ALIGNMENT_MIDDLELEFT;
223 s = s + 3;
224 } else if (stristr(s, "\\a10") != NULL) {
225 current->alignment = SUB_ALIGNMENT_MIDDLECENTER;
226 s = s + 4;
227 } else if (stristr(s, "\\a11") != NULL) {
228 current->alignment = SUB_ALIGNMENT_MIDDLERIGHT;
229 s = s + 4;
232 if (*s == '}') state = 3;
233 ++s;
234 continue;
237 /* read next line */
238 if (state != 99 && !(s = stream_read_line (st, line, LINE_LEN, utf16))) {
239 if (current->start > 0) {
240 break; // if it is the last subtitle
241 } else {
242 return 0;
246 } while (state != 99);
248 // For the last subtitle
249 if (current->end <= 0) {
250 current->end = current->start + sub_slacktime;
251 sami_add_line(current, text, &p);
254 return current;
258 static char *sub_readtext(char *source, char **dest) {
259 int len=0;
260 char *p=source;
262 // printf("src=%p dest=%p \n",source,dest);
264 while ( !eol(*p) && *p!= '|' ) {
265 p++,len++;
268 *dest= malloc (len+1);
269 if (!dest) {return ERR;}
271 strncpy(*dest, source, len);
272 (*dest)[len]=0;
274 while (*p=='\r' || *p=='\n' || *p=='|') p++;
276 if (*p) return p; // not-last text field
277 else return NULL; // last text field
280 static subtitle *sub_read_line_microdvd(stream_t *st,subtitle *current, int utf16) {
281 char line[LINE_LEN+1];
282 char line2[LINE_LEN+1];
283 char *p, *next;
284 int i;
286 do {
287 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
288 } while ((sscanf (line,
289 "{%ld}{}%[^\r\n]",
290 &(current->start), line2) < 2) &&
291 (sscanf (line,
292 "{%ld}{%ld}%[^\r\n]",
293 &(current->start), &(current->end), line2) < 3));
295 p=line2;
297 next=p, i=0;
298 while ((next =sub_readtext (next, &(current->text[i])))) {
299 if (current->text[i]==ERR) {return ERR;}
300 i++;
301 if (i>=SUB_MAX_TEXT) { mp_msg(MSGT_SUBREADER,MSGL_WARN,"Too many lines in a subtitle\n");current->lines=i;return current;}
303 current->lines= ++i;
305 return current;
308 static subtitle *sub_read_line_mpl2(stream_t *st,subtitle *current, int utf16) {
309 char line[LINE_LEN+1];
310 char line2[LINE_LEN+1];
311 char *p, *next;
312 int i;
314 do {
315 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
316 } while ((sscanf (line,
317 "[%ld][%ld]%[^\r\n]",
318 &(current->start), &(current->end), line2) < 3));
319 current->start *= 10;
320 current->end *= 10;
321 p=line2;
323 next=p, i=0;
324 while ((next =sub_readtext (next, &(current->text[i])))) {
325 if (current->text[i]==ERR) {return ERR;}
326 i++;
327 if (i>=SUB_MAX_TEXT) { mp_msg(MSGT_SUBREADER,MSGL_WARN,"Too many lines in a subtitle\n");current->lines=i;return current;}
329 current->lines= ++i;
331 return current;
334 static subtitle *sub_read_line_subrip(stream_t* st, subtitle *current, int utf16) {
335 char line[LINE_LEN+1];
336 int a1,a2,a3,a4,b1,b2,b3,b4;
337 char *p=NULL, *q=NULL;
338 int len;
340 while (1) {
341 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
342 if (sscanf (line, "%d:%d:%d.%d,%d:%d:%d.%d",&a1,&a2,&a3,&a4,&b1,&b2,&b3,&b4) < 8) continue;
343 current->start = a1*360000+a2*6000+a3*100+a4;
344 current->end = b1*360000+b2*6000+b3*100+b4;
346 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
348 p=q=line;
349 for (current->lines=1; current->lines < SUB_MAX_TEXT; current->lines++) {
350 for (q=p,len=0; *p && *p!='\r' && *p!='\n' && *p!='|' && strncmp(p,"[br]",4); p++,len++);
351 current->text[current->lines-1]=malloc (len+1);
352 if (!current->text[current->lines-1]) return ERR;
353 strncpy (current->text[current->lines-1], q, len);
354 current->text[current->lines-1][len]='\0';
355 if (!*p || *p=='\r' || *p=='\n') break;
356 if (*p=='|') p++;
357 else while (*p++!=']');
359 break;
361 return current;
364 static subtitle *sub_read_line_subviewer(stream_t *st,subtitle *current, int utf16) {
365 char line[LINE_LEN+1];
366 int a1,a2,a3,a4,b1,b2,b3,b4;
367 char *p=NULL;
368 int i,len;
370 while (!current->text[0]) {
371 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
372 if ((len=sscanf (line, "%d:%d:%d%[,.:]%d --> %d:%d:%d%[,.:]%d",&a1,&a2,&a3,(char *)&i,&a4,&b1,&b2,&b3,(char *)&i,&b4)) < 10)
373 continue;
374 current->start = a1*360000+a2*6000+a3*100+a4/10;
375 current->end = b1*360000+b2*6000+b3*100+b4/10;
376 for (i=0; i<SUB_MAX_TEXT;) {
377 int blank = 1;
378 if (!stream_read_line (st, line, LINE_LEN, utf16)) break;
379 len=0;
380 for (p=line; *p!='\n' && *p!='\r' && *p; p++,len++)
381 if (*p != ' ' && *p != '\t')
382 blank = 0;
383 if (len && !blank) {
384 int j=0,skip=0;
385 char *curptr=current->text[i]=malloc (len+1);
386 if (!current->text[i]) return ERR;
387 //strncpy (current->text[i], line, len); current->text[i][len]='\0';
388 for(; j<len; j++) {
389 /* let's filter html tags ::atmos */
390 if(line[j]=='>') {
391 skip=0;
392 continue;
394 if(line[j]=='<') {
395 skip=1;
396 continue;
398 if(skip) {
399 continue;
401 *curptr=line[j];
402 curptr++;
404 *curptr='\0';
406 i++;
407 } else {
408 break;
411 current->lines=i;
413 return current;
416 static subtitle *sub_read_line_subviewer2(stream_t *st,subtitle *current, int utf16) {
417 char line[LINE_LEN+1];
418 int a1,a2,a3,a4;
419 char *p=NULL;
420 int i,len;
422 while (!current->text[0]) {
423 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
424 if (line[0]!='{')
425 continue;
426 if ((len=sscanf (line, "{T %d:%d:%d:%d",&a1,&a2,&a3,&a4)) < 4)
427 continue;
428 current->start = a1*360000+a2*6000+a3*100+a4/10;
429 for (i=0; i<SUB_MAX_TEXT;) {
430 if (!stream_read_line (st, line, LINE_LEN, utf16)) break;
431 if (line[0]=='}') break;
432 len=0;
433 for (p=line; *p!='\n' && *p!='\r' && *p; ++p,++len);
434 if (len) {
435 current->text[i]=malloc (len+1);
436 if (!current->text[i]) return ERR;
437 strncpy (current->text[i], line, len); current->text[i][len]='\0';
438 ++i;
439 } else {
440 break;
443 current->lines=i;
445 return current;
449 static subtitle *sub_read_line_vplayer(stream_t *st,subtitle *current, int utf16) {
450 char line[LINE_LEN+1];
451 int a1,a2,a3;
452 char *p=NULL, *next,separator;
453 int i,len,plen;
455 while (!current->text[0]) {
456 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
457 if ((len=sscanf (line, "%d:%d:%d%c%n",&a1,&a2,&a3,&separator,&plen)) < 4)
458 continue;
460 if (!(current->start = a1*360000+a2*6000+a3*100))
461 continue;
462 /* removed by wodzu
463 p=line;
464 // finds the body of the subtitle
465 for (i=0; i<3; i++){
466 p=strchr(p,':');
467 if (p==NULL) break;
468 ++p;
470 if (p==NULL) {
471 printf("SUB: Skipping incorrect subtitle line!\n");
472 continue;
475 // by wodzu: hey! this time we know what length it has! what is
476 // that magic for? it can't deal with space instead of third
477 // colon! look, what simple it can be:
478 p = &line[ plen ];
480 i=0;
481 if (*p!='|') {
483 next = p,i=0;
484 while ((next =sub_readtext (next, &(current->text[i])))) {
485 if (current->text[i]==ERR) {return ERR;}
486 i++;
487 if (i>=SUB_MAX_TEXT) { mp_msg(MSGT_SUBREADER,MSGL_WARN,"Too many lines in a subtitle\n");current->lines=i;return current;}
489 current->lines=i+1;
492 return current;
495 static subtitle *sub_read_line_rt(stream_t *st,subtitle *current, int utf16) {
496 //TODO: This format uses quite rich (sub/super)set of xhtml
497 // I couldn't check it since DTD is not included.
498 // WARNING: full XML parses can be required for proper parsing
499 char line[LINE_LEN+1];
500 int a1,a2,a3,a4,b1,b2,b3,b4;
501 char *p=NULL,*next=NULL;
502 int i,len,plen;
504 while (!current->text[0]) {
505 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
506 //TODO: it seems that format of time is not easily determined, it may be 1:12, 1:12.0 or 0:1:12.0
507 //to describe the same moment in time. Maybe there are even more formats in use.
508 //if ((len=sscanf (line, "<Time Begin=\"%d:%d:%d.%d\" End=\"%d:%d:%d.%d\"",&a1,&a2,&a3,&a4,&b1,&b2,&b3,&b4)) < 8)
509 plen=a1=a2=a3=a4=b1=b2=b3=b4=0;
510 if (
511 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d.%d\" %*[Ee]nd=\"%d.%d\"%*[^<]<clear/>%n",&a3,&a4,&b3,&b4,&plen)) < 4) &&
512 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d.%d\" %*[Ee]nd=\"%d:%d.%d\"%*[^<]<clear/>%n",&a3,&a4,&b2,&b3,&b4,&plen)) < 5) &&
513 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d:%d\" %*[Ee]nd=\"%d:%d\"%*[^<]<clear/>%n",&a2,&a3,&b2,&b3,&plen)) < 4) &&
514 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d:%d\" %*[Ee]nd=\"%d:%d.%d\"%*[^<]<clear/>%n",&a2,&a3,&b2,&b3,&b4,&plen)) < 5) &&
515 // ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d:%d.%d\" %*[Ee]nd=\"%d:%d\"%*[^<]<clear/>%n",&a2,&a3,&a4,&b2,&b3,&plen)) < 5) &&
516 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d:%d.%d\" %*[Ee]nd=\"%d:%d.%d\"%*[^<]<clear/>%n",&a2,&a3,&a4,&b2,&b3,&b4,&plen)) < 6) &&
517 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d:%d:%d.%d\" %*[Ee]nd=\"%d:%d:%d.%d\"%*[^<]<clear/>%n",&a1,&a2,&a3,&a4,&b1,&b2,&b3,&b4,&plen)) < 8) &&
518 //now try it without end time
519 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d.%d\"%*[^<]<clear/>%n",&a3,&a4,&plen)) < 2) &&
520 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d:%d\"%*[^<]<clear/>%n",&a2,&a3,&plen)) < 2) &&
521 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d:%d.%d\"%*[^<]<clear/>%n",&a2,&a3,&a4,&plen)) < 3) &&
522 ((len=sscanf (line, "<%*[tT]ime %*[bB]egin=\"%d:%d:%d.%d\"%*[^<]<clear/>%n",&a1,&a2,&a3,&a4,&plen)) < 4)
524 continue;
525 current->start = a1*360000+a2*6000+a3*100+a4/10;
526 current->end = b1*360000+b2*6000+b3*100+b4/10;
527 if (b1 == 0 && b2 == 0 && b3 == 0 && b4 == 0)
528 current->end = current->start+200;
529 p=line; p+=plen;i=0;
530 // TODO: I don't know what kind of convention is here for marking multiline subs, maybe <br/> like in xml?
531 next = strstr(line,"<clear/>");
532 if(next && strlen(next)>8){
533 next+=8;i=0;
534 while ((next =sub_readtext (next, &(current->text[i])))) {
535 if (current->text[i]==ERR) {return ERR;}
536 i++;
537 if (i>=SUB_MAX_TEXT) { mp_msg(MSGT_SUBREADER,MSGL_WARN,"Too many lines in a subtitle\n");current->lines=i;return current;}
540 current->lines=i+1;
542 return current;
545 static subtitle *sub_read_line_ssa(stream_t *st,subtitle *current, int utf16) {
547 * Sub Station Alpha v4 (and v2?) scripts have 9 commas before subtitle
548 * other Sub Station Alpha scripts have only 8 commas before subtitle
549 * Reading the "ScriptType:" field is not reliable since many scripts appear
550 * w/o it
552 * http://www.scriptclub.org is a good place to find more examples
553 * http://www.eswat.demon.co.uk is where the SSA specs can be found
555 int comma;
556 static int max_comma = 32; /* let's use 32 for the case that the */
557 /* amount of commas increase with newer SSA versions */
559 int hour1, min1, sec1, hunsec1,
560 hour2, min2, sec2, hunsec2, nothing;
561 int num;
563 char line[LINE_LEN+1],
564 line3[LINE_LEN+1],
565 *line2;
566 char *tmp;
568 do {
569 if (!stream_read_line (st, line, LINE_LEN, utf16)) return NULL;
570 } while (sscanf (line, "Dialogue: Marked=%d,%d:%d:%d.%d,%d:%d:%d.%d"
571 "%[^\n\r]", &nothing,
572 &hour1, &min1, &sec1, &hunsec1,
573 &hour2, &min2, &sec2, &hunsec2,
574 line3) < 9
576 sscanf (line, "Dialogue: %d,%d:%d:%d.%d,%d:%d:%d.%d"
577 "%[^\n\r]", &nothing,
578 &hour1, &min1, &sec1, &hunsec1,
579 &hour2, &min2, &sec2, &hunsec2,
580 line3) < 9 );
582 line2=strchr(line3, ',');
583 if (!line2) return NULL;
585 for (comma = 4; comma < max_comma; comma ++)
587 tmp = line2;
588 if(!(tmp=strchr(++tmp, ','))) break;
589 if(*(++tmp) == ' ') break;
590 /* a space after a comma means we're already in a sentence */
591 line2 = tmp;
594 if(comma < max_comma)max_comma = comma;
595 /* eliminate the trailing comma */
596 if(*line2 == ',') line2++;
598 current->lines=0;num=0;
599 current->start = 360000*hour1 + 6000*min1 + 100*sec1 + hunsec1;
600 current->end = 360000*hour2 + 6000*min2 + 100*sec2 + hunsec2;
602 while (((tmp=strstr(line2, "\\n")) != NULL) || ((tmp=strstr(line2, "\\N")) != NULL) ){
603 current->text[num]=malloc(tmp-line2+1);
604 strncpy (current->text[num], line2, tmp-line2);
605 current->text[num][tmp-line2]='\0';
606 line2=tmp+2;
607 num++;
608 current->lines++;
609 if (current->lines >= SUB_MAX_TEXT) return current;
612 current->text[num]=strdup(line2);
613 current->lines++;
615 return current;
618 static void sub_pp_ssa(subtitle *sub) {
619 int l=sub->lines;
620 char *so,*de,*start;
622 while (l){
623 /* eliminate any text enclosed with {}, they are font and color settings */
624 so=de=sub->text[--l];
625 while (*so) {
626 if(*so == '{' && so[1]=='\\') {
627 for (start=so; *so && *so!='}'; so++);
628 if(*so) so++; else so=start;
630 if(*so) {
631 *de=*so;
632 so++; de++;
635 *de=*so;
640 * PJS subtitles reader.
641 * That's the "Phoenix Japanimation Society" format.
642 * I found some of them in http://www.scriptsclub.org/ (used for anime).
643 * The time is in tenths of second.
645 * by set, based on code by szabi (dunnowhat sub format ;-)
647 static subtitle *sub_read_line_pjs(stream_t *st,subtitle *current, int utf16) {
648 char line[LINE_LEN+1];
649 char text[LINE_LEN+1], *s, *d;
651 if (!stream_read_line (st, line, LINE_LEN, utf16))
652 return NULL;
653 /* skip spaces */
654 for (s=line; *s && isspace(*s); s++);
655 /* allow empty lines at the end of the file */
656 if (*s==0)
657 return NULL;
658 /* get the time */
659 if (sscanf (s, "%ld,%ld,", &(current->start),
660 &(current->end)) <2) {
661 return ERR;
663 /* the files I have are in tenths of second */
664 current->start *= 10;
665 current->end *= 10;
666 /* walk to the beggining of the string */
667 for (; *s; s++) if (*s==',') break;
668 if (*s) {
669 for (s++; *s; s++) if (*s==',') break;
670 if (*s) s++;
672 if (*s!='"') {
673 return ERR;
675 /* copy the string to the text buffer */
676 for (s++, d=text; *s && *s!='"'; s++, d++)
677 *d=*s;
678 *d=0;
679 current->text[0] = strdup(text);
680 current->lines = 1;
682 return current;
685 static subtitle *sub_read_line_mpsub(stream_t *st, subtitle *current, int utf16) {
686 char line[LINE_LEN+1];
687 float a,b;
688 int num=0;
689 char *p, *q;
693 if (!stream_read_line(st, line, LINE_LEN, utf16)) return NULL;
694 } while (sscanf (line, "%f %f", &a, &b) !=2);
696 mpsub_position += a*mpsub_multiplier;
697 current->start=(int) mpsub_position;
698 mpsub_position += b*mpsub_multiplier;
699 current->end=(int) mpsub_position;
701 while (num < SUB_MAX_TEXT) {
702 if (!stream_read_line (st, line, LINE_LEN, utf16)) {
703 if (num == 0) return NULL;
704 else return current;
706 p=line;
707 while (isspace(*p)) p++;
708 if (eol(*p) && num > 0) return current;
709 if (eol(*p)) return NULL;
711 for (q=p; !eol(*q); q++);
712 *q='\0';
713 if (strlen(p)) {
714 current->text[num]=strdup(p);
715 // printf (">%s<\n",p);
716 current->lines = ++num;
717 } else {
718 if (num) return current;
719 else return NULL;
722 return NULL; // we should have returned before if it's OK
725 #ifndef CONFIG_SORTSUB
726 //we don't need this if we use previous_sub_end
727 subtitle *previous_aqt_sub = NULL;
728 #endif
730 static subtitle *sub_read_line_aqt(stream_t *st,subtitle *current, int utf16) {
731 char line[LINE_LEN+1];
732 char *next;
733 int i;
735 while (1) {
736 // try to locate next subtitle
737 if (!stream_read_line (st, line, LINE_LEN, utf16))
738 return NULL;
739 if (!(sscanf (line, "-->> %ld", &(current->start)) <1))
740 break;
743 #ifdef CONFIG_SORTSUB
744 previous_sub_end = (current->start) ? current->start - 1 : 0;
745 #else
746 if (previous_aqt_sub != NULL)
747 previous_aqt_sub->end = current->start-1;
749 previous_aqt_sub = current;
750 #endif
752 if (!stream_read_line (st, line, LINE_LEN, utf16))
753 return NULL;
755 sub_readtext((char *) &line,&current->text[0]);
756 current->lines = 1;
757 current->end = current->start; // will be corrected by next subtitle
759 if (!stream_read_line (st, line, LINE_LEN, utf16))
760 return current;
762 next = line,i=1;
763 while ((next =sub_readtext (next, &(current->text[i])))) {
764 if (current->text[i]==ERR) {return ERR;}
765 i++;
766 if (i>=SUB_MAX_TEXT) { mp_msg(MSGT_SUBREADER,MSGL_WARN,"Too many lines in a subtitle\n");current->lines=i;return current;}
768 current->lines=i+1;
770 if (!strlen(current->text[0]) && !strlen(current->text[1])) {
771 #ifdef CONFIG_SORTSUB
772 previous_sub_end = 0;
773 #else
774 // void subtitle -> end of previous marked and exit
775 previous_aqt_sub = NULL;
776 #endif
777 return NULL;
780 return current;
783 #ifndef CONFIG_SORTSUB
784 subtitle *previous_subrip09_sub = NULL;
785 #endif
787 static subtitle *sub_read_line_subrip09(stream_t *st,subtitle *current, int utf16) {
788 char line[LINE_LEN+1];
789 int a1,a2,a3;
790 char * next=NULL;
791 int i,len;
793 while (1) {
794 // try to locate next subtitle
795 if (!stream_read_line (st, line, LINE_LEN, utf16))
796 return NULL;
797 if (!((len=sscanf (line, "[%d:%d:%d]",&a1,&a2,&a3)) < 3))
798 break;
801 current->start = a1*360000+a2*6000+a3*100;
803 #ifdef CONFIG_SORTSUB
804 previous_sub_end = (current->start) ? current->start - 1 : 0;
805 #else
806 if (previous_subrip09_sub != NULL)
807 previous_subrip09_sub->end = current->start-1;
809 previous_subrip09_sub = current;
810 #endif
812 if (!stream_read_line (st, line, LINE_LEN, utf16))
813 return NULL;
815 next = line,i=0;
817 current->text[0]=""; // just to be sure that string is clear
819 while ((next =sub_readtext (next, &(current->text[i])))) {
820 if (current->text[i]==ERR) {return ERR;}
821 i++;
822 if (i>=SUB_MAX_TEXT) { mp_msg(MSGT_SUBREADER,MSGL_WARN,"Too many lines in a subtitle\n");current->lines=i;return current;}
824 current->lines=i+1;
826 if (!strlen(current->text[0]) && (i==0)) {
827 #ifdef CONFIG_SORTSUB
828 previous_sub_end = 0;
829 #else
830 // void subtitle -> end of previous marked and exit
831 previous_subrip09_sub = NULL;
832 #endif
833 return NULL;
836 return current;
839 static subtitle *sub_read_line_jacosub(stream_t* st, subtitle * current, int utf16)
841 char line1[LINE_LEN], line2[LINE_LEN], directive[LINE_LEN], *p, *q;
842 unsigned a1, a2, a3, a4, b1, b2, b3, b4, comment = 0;
843 static unsigned jacoTimeres = 30;
844 static int jacoShift = 0;
846 memset(current, 0, sizeof(subtitle));
847 memset(line1, 0, LINE_LEN);
848 memset(line2, 0, LINE_LEN);
849 memset(directive, 0, LINE_LEN);
850 while (!current->text[0]) {
851 if (!stream_read_line(st, line1, LINE_LEN, utf16)) {
852 return NULL;
854 if (sscanf
855 (line1, "%u:%u:%u.%u %u:%u:%u.%u %[^\n\r]", &a1, &a2, &a3, &a4,
856 &b1, &b2, &b3, &b4, line2) < 9) {
857 if (sscanf(line1, "@%u @%u %[^\n\r]", &a4, &b4, line2) < 3) {
858 if (line1[0] == '#') {
859 int hours = 0, minutes = 0, seconds, delta, inverter =
861 unsigned units = jacoShift;
862 switch (toupper(line1[1])) {
863 case 'S':
864 if (isalpha(line1[2])) {
865 delta = 6;
866 } else {
867 delta = 2;
869 if (sscanf(&line1[delta], "%d", &hours)) {
870 if (hours < 0) {
871 hours *= -1;
872 inverter = -1;
874 if (sscanf(&line1[delta], "%*d:%d", &minutes)) {
875 if (sscanf
876 (&line1[delta], "%*d:%*d:%d",
877 &seconds)) {
878 sscanf(&line1[delta], "%*d:%*d:%*d.%d",
879 &units);
880 } else {
881 hours = 0;
882 sscanf(&line1[delta], "%d:%d.%d",
883 &minutes, &seconds, &units);
884 minutes *= inverter;
886 } else {
887 hours = minutes = 0;
888 sscanf(&line1[delta], "%d.%d", &seconds,
889 &units);
890 seconds *= inverter;
892 jacoShift =
893 ((hours * 3600 + minutes * 60 +
894 seconds) * jacoTimeres +
895 units) * inverter;
897 break;
898 case 'T':
899 if (isalpha(line1[2])) {
900 delta = 8;
901 } else {
902 delta = 2;
904 sscanf(&line1[delta], "%u", &jacoTimeres);
905 break;
908 continue;
909 } else {
910 current->start =
911 (unsigned long) ((a4 + jacoShift) * 100.0 /
912 jacoTimeres);
913 current->end =
914 (unsigned long) ((b4 + jacoShift) * 100.0 /
915 jacoTimeres);
917 } else {
918 current->start =
919 (unsigned
920 long) (((a1 * 3600 + a2 * 60 + a3) * jacoTimeres + a4 +
921 jacoShift) * 100.0 / jacoTimeres);
922 current->end =
923 (unsigned
924 long) (((b1 * 3600 + b2 * 60 + b3) * jacoTimeres + b4 +
925 jacoShift) * 100.0 / jacoTimeres);
927 current->lines = 0;
928 p = line2;
929 while ((*p == ' ') || (*p == '\t')) {
930 ++p;
932 if (isalpha(*p)||*p == '[') {
933 int cont, jLength;
935 if (sscanf(p, "%s %[^\n\r]", directive, line1) < 2)
936 return (subtitle *) ERR;
937 jLength = strlen(directive);
938 for (cont = 0; cont < jLength; ++cont) {
939 if (isalpha(*(directive + cont)))
940 *(directive + cont) = toupper(*(directive + cont));
942 if ((strstr(directive, "RDB") != NULL)
943 || (strstr(directive, "RDC") != NULL)
944 || (strstr(directive, "RLB") != NULL)
945 || (strstr(directive, "RLG") != NULL)) {
946 continue;
948 if (strstr(directive, "JL") != NULL) {
949 current->alignment = SUB_ALIGNMENT_BOTTOMLEFT;
950 } else if (strstr(directive, "JR") != NULL) {
951 current->alignment = SUB_ALIGNMENT_BOTTOMRIGHT;
952 } else {
953 current->alignment = SUB_ALIGNMENT_BOTTOMCENTER;
955 strcpy(line2, line1);
956 p = line2;
958 for (q = line1; (!eol(*p)) && (current->lines < SUB_MAX_TEXT); ++p) {
959 switch (*p) {
960 case '{':
961 comment++;
962 break;
963 case '}':
964 if (comment) {
965 --comment;
966 //the next line to get rid of a blank after the comment
967 if ((*(p + 1)) == ' ')
968 p++;
970 break;
971 case '~':
972 if (!comment) {
973 *q = ' ';
974 ++q;
976 break;
977 case ' ':
978 case '\t':
979 if ((*(p + 1) == ' ') || (*(p + 1) == '\t'))
980 break;
981 if (!comment) {
982 *q = ' ';
983 ++q;
985 break;
986 case '\\':
987 if (*(p + 1) == 'n') {
988 *q = '\0';
989 q = line1;
990 current->text[current->lines++] = strdup(line1);
991 ++p;
992 break;
994 if ((toupper(*(p + 1)) == 'C')
995 || (toupper(*(p + 1)) == 'F')) {
996 ++p,++p;
997 break;
999 if ((*(p + 1) == 'B') || (*(p + 1) == 'b') || (*(p + 1) == 'D') || //actually this means "insert current date here"
1000 (*(p + 1) == 'I') || (*(p + 1) == 'i') || (*(p + 1) == 'N') || (*(p + 1) == 'T') || //actually this means "insert current time here"
1001 (*(p + 1) == 'U') || (*(p + 1) == 'u')) {
1002 ++p;
1003 break;
1005 if ((*(p + 1) == '\\') ||
1006 (*(p + 1) == '~') || (*(p + 1) == '{')) {
1007 ++p;
1008 } else if (eol(*(p + 1))) {
1009 if (!stream_read_line(st, directive, LINE_LEN, utf16))
1010 return NULL;
1011 trail_space(directive);
1012 av_strlcat(line2, directive, LINE_LEN);
1013 break;
1015 default:
1016 if (!comment) {
1017 *q = *p;
1018 ++q;
1020 } //-- switch
1021 } //-- for
1022 *q = '\0';
1023 current->text[current->lines] = strdup(line1);
1024 } //-- while
1025 current->lines++;
1026 return current;
1029 static int sub_autodetect (stream_t* st, int *uses_time, int utf16) {
1030 char line[LINE_LEN+1];
1031 int i,j=0;
1033 while (j < 100) {
1034 j++;
1035 if (!stream_read_line (st, line, LINE_LEN, utf16))
1036 return SUB_INVALID;
1038 if (sscanf (line, "{%d}{%d}", &i, &i)==2)
1039 {*uses_time=0;return SUB_MICRODVD;}
1040 if (sscanf (line, "{%d}{}", &i)==1)
1041 {*uses_time=0;return SUB_MICRODVD;}
1042 if (sscanf (line, "[%d][%d]", &i, &i)==2)
1043 {*uses_time=1;return SUB_MPL2;}
1044 if (sscanf (line, "%d:%d:%d.%d,%d:%d:%d.%d", &i, &i, &i, &i, &i, &i, &i, &i)==8)
1045 {*uses_time=1;return SUB_SUBRIP;}
1046 if (sscanf (line, "%d:%d:%d%[,.:]%d --> %d:%d:%d%[,.:]%d", &i, &i, &i, (char *)&i, &i, &i, &i, &i, (char *)&i, &i)==10)
1047 {*uses_time=1;return SUB_SUBVIEWER;}
1048 if (sscanf (line, "{T %d:%d:%d:%d",&i, &i, &i, &i)==4)
1049 {*uses_time=1;return SUB_SUBVIEWER2;}
1050 if (strstr (line, "<SAMI>"))
1051 {*uses_time=1; return SUB_SAMI;}
1052 if (sscanf(line, "%d:%d:%d.%d %d:%d:%d.%d", &i, &i, &i, &i, &i, &i, &i, &i) == 8)
1053 {*uses_time = 1; return SUB_JACOSUB;}
1054 if (sscanf(line, "@%d @%d", &i, &i) == 2)
1055 {*uses_time = 1; return SUB_JACOSUB;}
1056 if (sscanf (line, "%d:%d:%d:", &i, &i, &i )==3)
1057 {*uses_time=1;return SUB_VPLAYER;}
1058 if (sscanf (line, "%d:%d:%d ", &i, &i, &i )==3)
1059 {*uses_time=1;return SUB_VPLAYER;}
1060 //TODO: just checking if first line of sub starts with "<" is WAY
1061 // too weak test for RT
1062 // Please someone who knows the format of RT... FIX IT!!!
1063 // It may conflict with other sub formats in the future (actually it doesn't)
1064 if ( *line == '<' )
1065 {*uses_time=1;return SUB_RT;}
1067 if (!memcmp(line, "Dialogue: Marked", 16))
1068 {*uses_time=1; return SUB_SSA;}
1069 if (!memcmp(line, "Dialogue: ", 10))
1070 {*uses_time=1; return SUB_SSA;}
1071 if (sscanf (line, "%d,%d,\"%c", &i, &i, (char *) &i) == 3)
1072 {*uses_time=1;return SUB_PJS;}
1073 if (sscanf (line, "FORMAT=%d", &i) == 1)
1074 {*uses_time=0; return SUB_MPSUB;}
1075 if (!memcmp(line, "FORMAT=TIME", 11))
1076 {*uses_time=1; return SUB_MPSUB;}
1077 if (strstr (line, "-->>"))
1078 {*uses_time=0; return SUB_AQTITLE;}
1079 if (sscanf (line, "[%d:%d:%d]", &i, &i, &i)==3)
1080 {*uses_time=1;return SUB_SUBRIP09;}
1083 return SUB_INVALID; // too many bad lines
1086 extern int sub_utf8;
1087 int sub_utf8_prev=0;
1089 extern float sub_delay;
1090 extern float sub_fps;
1092 #ifdef CONFIG_ICONV
1093 static iconv_t icdsc = (iconv_t)(-1);
1095 void subcp_open (stream_t *st)
1097 char *tocp = "UTF-8";
1099 if (sub_cp){
1100 const char *cp_tmp = sub_cp;
1101 #ifdef CONFIG_ENCA
1102 char enca_lang[3], enca_fallback[100];
1103 if (sscanf(sub_cp, "enca:%2s:%99s", enca_lang, enca_fallback) == 2
1104 || sscanf(sub_cp, "ENCA:%2s:%99s", enca_lang, enca_fallback) == 2) {
1105 if (st && st->flags & MP_STREAM_SEEK ) {
1106 cp_tmp = guess_cp(st, enca_lang, enca_fallback);
1107 } else {
1108 cp_tmp = enca_fallback;
1109 if (st)
1110 mp_msg(MSGT_SUBREADER,MSGL_WARN,"SUB: enca failed, stream must be seekable.\n");
1113 #endif
1114 if ((icdsc = iconv_open (tocp, cp_tmp)) != (iconv_t)(-1)){
1115 mp_msg(MSGT_SUBREADER,MSGL_V,"SUB: opened iconv descriptor.\n");
1116 sub_utf8 = 2;
1117 } else
1118 mp_msg(MSGT_SUBREADER,MSGL_ERR,"SUB: error opening iconv descriptor.\n");
1122 void subcp_close (void)
1124 if (icdsc != (iconv_t)(-1)){
1125 (void) iconv_close (icdsc);
1126 icdsc = (iconv_t)(-1);
1127 mp_msg(MSGT_SUBREADER,MSGL_V,"SUB: closed iconv descriptor.\n");
1131 subtitle* subcp_recode (subtitle *sub)
1133 int l=sub->lines;
1134 size_t ileft, oleft;
1135 char *op, *ip, *ot;
1136 if(icdsc == (iconv_t)(-1)) return sub;
1138 while (l){
1139 ip = sub->text[--l];
1140 ileft = strlen(ip);
1141 oleft = 4 * ileft;
1143 if (!(ot = malloc(oleft + 1))){
1144 mp_msg(MSGT_SUBREADER,MSGL_WARN,"SUB: error allocating mem.\n");
1145 continue;
1147 op = ot;
1148 if (iconv(icdsc, &ip, &ileft,
1149 &op, &oleft) == (size_t)(-1)) {
1150 mp_msg(MSGT_SUBREADER,MSGL_WARN,"SUB: error recoding line.\n");
1151 free(ot);
1152 continue;
1154 // In some stateful encodings, we must clear the state to handle the last character
1155 if (iconv(icdsc, NULL, NULL,
1156 &op, &oleft) == (size_t)(-1)) {
1157 mp_msg(MSGT_SUBREADER,MSGL_WARN,"SUB: error recoding line, can't clear encoding state.\n");
1159 *op='\0' ;
1160 free (sub->text[l]);
1161 sub->text[l] = ot;
1163 return sub;
1165 #endif
1167 #ifdef CONFIG_FRIBIDI
1169 * Do conversion necessary for right-to-left language support via fribidi.
1170 * @param sub subtitle to convert
1171 * @param sub_utf8 whether the subtitle is encoded in UTF-8
1172 * @param from first new subtitle, all lines before this are assumed to be already converted
1174 static subtitle* sub_fribidi (subtitle *sub, int sub_utf8, int from)
1176 FriBidiChar logical[LINE_LEN+1], visual[LINE_LEN+1]; // Hopefully these two won't smash the stack
1177 char *ip = NULL, *op = NULL;
1178 size_t len,orig_len;
1179 int l=sub->lines;
1180 int char_set_num;
1181 fribidi_boolean log2vis;
1182 if (!flip_hebrew)
1183 return sub;
1184 fribidi_set_mirroring(1);
1185 fribidi_set_reorder_nsm(0);
1187 if( sub_utf8 == 0 ) {
1188 char_set_num = fribidi_parse_charset (fribidi_charset?fribidi_charset:"ISO8859-8");
1189 }else {
1190 char_set_num = fribidi_parse_charset ("UTF-8");
1192 while (l > from) {
1193 ip = sub->text[--l];
1194 orig_len = len = strlen( ip ); // We assume that we don't use full unicode, only UTF-8 or ISO8859-x
1195 if(len > LINE_LEN) {
1196 mp_msg(MSGT_SUBREADER,MSGL_WARN,"SUB: sub->text is longer than LINE_LEN.\n");
1197 l++;
1198 break;
1200 len = fribidi_charset_to_unicode (char_set_num, ip, len, logical);
1201 #if FRIBIDI_INTERFACE_VERSION < 3
1202 FriBidiCharType base = fribidi_flip_commas?FRIBIDI_TYPE_ON:FRIBIDI_TYPE_L;
1203 #else
1204 FriBidiParType base = fribidi_flip_commas?FRIBIDI_TYPE_ON:FRIBIDI_TYPE_L;
1205 #endif
1206 log2vis = fribidi_log2vis (logical, len, &base,
1207 /* output */
1208 visual, NULL, NULL, NULL);
1209 if(log2vis) {
1210 len = fribidi_remove_bidi_marks (visual, len, NULL, NULL,
1211 NULL);
1212 if((op = malloc((FFMAX(2*orig_len,2*len) + 1))) == NULL) {
1213 mp_msg(MSGT_SUBREADER,MSGL_WARN,"SUB: error allocating mem.\n");
1214 l++;
1215 break;
1217 fribidi_unicode_to_charset ( char_set_num, visual, len,op);
1218 free (ip);
1219 sub->text[l] = op;
1222 if (!from && l){
1223 for (l = sub->lines; l;)
1224 free (sub->text[--l]);
1225 return ERR;
1227 return sub;
1230 #endif
1232 static void adjust_subs_time(subtitle* sub, float subtime, float fps, int block,
1233 int sub_num, int sub_uses_time) {
1234 int n,m;
1235 subtitle* nextsub;
1236 int i = sub_num;
1237 unsigned long subfms = (sub_uses_time ? 100 : fps) * subtime;
1238 unsigned long overlap = (sub_uses_time ? 100 : fps) / 5; // 0.2s
1240 n=m=0;
1241 if (i) for (;;){
1242 if (sub->end <= sub->start){
1243 sub->end = sub->start + subfms;
1244 m++;
1245 n++;
1247 if (!--i) break;
1248 nextsub = sub + 1;
1249 if(block){
1250 if ((sub->end > nextsub->start) && (sub->end <= nextsub->start + overlap)) {
1251 // these subtitles overlap for less than 0.2 seconds
1252 // and would result in very short overlapping subtitle
1253 // so let's fix the problem here, before overlapping code
1254 // get its hands on them
1255 unsigned delta = sub->end - nextsub->start, half = delta / 2;
1256 sub->end -= half + 1;
1257 nextsub->start += delta - half;
1259 if (sub->end >= nextsub->start){
1260 sub->end = nextsub->start - 1;
1261 if (sub->end - sub->start > subfms)
1262 sub->end = sub->start + subfms;
1263 if (!m)
1264 n++;
1268 /* Theory:
1269 * Movies are often converted from FILM (24 fps)
1270 * to PAL (25) by simply speeding it up, so we
1271 * to multiply the original timestmaps by
1272 * (Movie's FPS / Subtitle's (guessed) FPS)
1273 * so eg. for 23.98 fps movie and PAL time based
1274 * subtitles we say -subfps 25 and we're fine!
1277 /* timed sub fps correction ::atmos */
1278 /* the frame-based case is handled in mpcommon.c
1279 * where find_sub is called */
1280 if(sub_uses_time && sub_fps) {
1281 sub->start *= sub_fps/fps;
1282 sub->end *= sub_fps/fps;
1285 sub = nextsub;
1286 m = 0;
1288 if (n) mp_msg(MSGT_SUBREADER,MSGL_V,"SUB: Adjusted %d subtitle(s).\n", n);
1291 struct subreader {
1292 subtitle * (*read)(stream_t *st,subtitle *dest,int utf16);
1293 void (*post)(subtitle *dest);
1294 const char *name;
1297 #ifdef CONFIG_ENCA
1298 const char* guess_buffer_cp(unsigned char* buffer, int buflen, const char *preferred_language, const char *fallback)
1300 const char **languages;
1301 size_t langcnt;
1302 EncaAnalyser analyser;
1303 EncaEncoding encoding;
1304 const char *detected_sub_cp = NULL;
1305 int i;
1307 languages = enca_get_languages(&langcnt);
1308 mp_msg(MSGT_SUBREADER, MSGL_V, "ENCA supported languages: ");
1309 for (i = 0; i < langcnt; i++) {
1310 mp_msg(MSGT_SUBREADER, MSGL_V, "%s ", languages[i]);
1312 mp_msg(MSGT_SUBREADER, MSGL_V, "\n");
1314 for (i = 0; i < langcnt; i++) {
1315 if (strcasecmp(languages[i], preferred_language) != 0) continue;
1316 analyser = enca_analyser_alloc(languages[i]);
1317 encoding = enca_analyse_const(analyser, buffer, buflen);
1318 enca_analyser_free(analyser);
1319 if (encoding.charset != ENCA_CS_UNKNOWN) {
1320 detected_sub_cp = enca_charset_name(encoding.charset, ENCA_NAME_STYLE_ICONV);
1321 break;
1325 free(languages);
1327 if (!detected_sub_cp) {
1328 detected_sub_cp = fallback;
1329 mp_msg(MSGT_SUBREADER, MSGL_INFO, "ENCA detection failed: fallback to %s\n", fallback);
1330 }else{
1331 mp_msg(MSGT_SUBREADER, MSGL_INFO, "ENCA detected charset: %s\n", detected_sub_cp);
1334 return detected_sub_cp;
1337 #define MAX_GUESS_BUFFER_SIZE (256*1024)
1338 const char* guess_cp(stream_t *st, const char *preferred_language, const char *fallback)
1340 size_t buflen;
1341 unsigned char *buffer;
1342 const char *detected_sub_cp = NULL;
1344 buffer = malloc(MAX_GUESS_BUFFER_SIZE);
1345 buflen = stream_read(st,buffer, MAX_GUESS_BUFFER_SIZE);
1347 detected_sub_cp = guess_buffer_cp(buffer, buflen, preferred_language, fallback);
1349 free(buffer);
1350 stream_reset(st);
1351 stream_seek(st,0);
1353 return detected_sub_cp;
1355 #undef MAX_GUESS_BUFFER_SIZE
1356 #endif
1358 sub_data* sub_read_file (char *filename, float fps) {
1359 int utf16;
1360 stream_t* fd;
1361 int n_max, n_first, i, j, sub_first, sub_orig;
1362 subtitle *first, *second, *sub, *return_sub, *alloced_sub = NULL;
1363 sub_data *subt_data;
1364 int uses_time = 0, sub_num = 0, sub_errs = 0;
1365 static const struct subreader sr[]=
1367 { sub_read_line_microdvd, NULL, "microdvd" },
1368 { sub_read_line_subrip, NULL, "subrip" },
1369 { sub_read_line_subviewer, NULL, "subviewer" },
1370 { sub_read_line_sami, NULL, "sami" },
1371 { sub_read_line_vplayer, NULL, "vplayer" },
1372 { sub_read_line_rt, NULL, "rt" },
1373 { sub_read_line_ssa, sub_pp_ssa, "ssa" },
1374 { sub_read_line_pjs, NULL, "pjs" },
1375 { sub_read_line_mpsub, NULL, "mpsub" },
1376 { sub_read_line_aqt, NULL, "aqt" },
1377 { sub_read_line_subviewer2, NULL, "subviewer 2.0" },
1378 { sub_read_line_subrip09, NULL, "subrip 0.9" },
1379 { sub_read_line_jacosub, NULL, "jacosub" },
1380 { sub_read_line_mpl2, NULL, "mpl2" }
1382 const struct subreader *srp;
1384 if(filename==NULL) return NULL; //qnx segfault
1385 fd=open_stream (filename, NULL, NULL); if (!fd) return NULL;
1387 sub_format = SUB_INVALID;
1388 for (utf16 = 0; sub_format == SUB_INVALID && utf16 < 3; utf16++) {
1389 sub_format=sub_autodetect (fd, &uses_time, utf16);
1390 stream_reset(fd);
1391 stream_seek(fd,0);
1393 utf16--;
1395 mpsub_multiplier = (uses_time ? 100.0 : 1.0);
1396 if (sub_format==SUB_INVALID) {mp_msg(MSGT_SUBREADER,MSGL_WARN,"SUB: Could not determine file format\n");return NULL;}
1397 srp=sr+sub_format;
1398 mp_msg(MSGT_SUBREADER, MSGL_V, "SUB: Detected subtitle file format: %s\n", srp->name);
1400 #ifdef CONFIG_ICONV
1401 sub_utf8_prev=sub_utf8;
1403 int l,k;
1404 k = -1;
1405 if ((l=strlen(filename))>4){
1406 char *exts[] = {".utf", ".utf8", ".utf-8" };
1407 for (k=3;--k>=0;)
1408 if (l >= strlen(exts[k]) && !strcasecmp(filename+(l - strlen(exts[k])), exts[k])){
1409 sub_utf8 = 1;
1410 break;
1413 if (k<0) subcp_open(fd);
1415 #endif
1417 sub_num=0;n_max=32;
1418 first=malloc(n_max*sizeof(subtitle));
1419 if(!first){
1420 #ifdef CONFIG_ICONV
1421 subcp_close();
1422 sub_utf8=sub_utf8_prev;
1423 #endif
1424 return NULL;
1427 #ifdef CONFIG_SORTSUB
1428 alloced_sub =
1429 sub = malloc(sizeof(subtitle));
1430 //This is to deal with those formats (AQT & Subrip) which define the end of a subtitle
1431 //as the beginning of the following
1432 previous_sub_end = 0;
1433 #endif
1434 while(1){
1435 if(sub_num>=n_max){
1436 n_max+=16;
1437 first=realloc(first,n_max*sizeof(subtitle));
1439 #ifndef CONFIG_SORTSUB
1440 sub = &first[sub_num];
1441 #endif
1442 memset(sub, '\0', sizeof(subtitle));
1443 sub=srp->read(fd,sub,utf16);
1444 if(!sub) break; // EOF
1445 #ifdef CONFIG_ICONV
1446 if ((sub!=ERR) && sub_utf8 == 2) sub=subcp_recode(sub);
1447 #endif
1448 #ifdef CONFIG_FRIBIDI
1449 if (sub!=ERR) sub=sub_fribidi(sub,sub_utf8,0);
1450 #endif
1451 if ( sub == ERR )
1453 #ifdef CONFIG_ICONV
1454 subcp_close();
1455 #endif
1456 if ( first ) free(first);
1457 free(alloced_sub);
1458 return NULL;
1460 // Apply any post processing that needs recoding first
1461 if ((sub!=ERR) && !sub_no_text_pp && srp->post) srp->post(sub);
1462 #ifdef CONFIG_SORTSUB
1463 if(!sub_num || (first[sub_num - 1].start <= sub->start)){
1464 first[sub_num].start = sub->start;
1465 first[sub_num].end = sub->end;
1466 first[sub_num].lines = sub->lines;
1467 first[sub_num].alignment = sub->alignment;
1468 for(i = 0; i < sub->lines; ++i){
1469 first[sub_num].text[i] = sub->text[i];
1471 if (previous_sub_end){
1472 first[sub_num - 1].end = previous_sub_end;
1473 previous_sub_end = 0;
1475 } else {
1476 for(j = sub_num - 1; j >= 0; --j){
1477 first[j + 1].start = first[j].start;
1478 first[j + 1].end = first[j].end;
1479 first[j + 1].lines = first[j].lines;
1480 first[j + 1].alignment = first[j].alignment;
1481 for(i = 0; i < first[j].lines; ++i){
1482 first[j + 1].text[i] = first[j].text[i];
1484 if(!j || (first[j - 1].start <= sub->start)){
1485 first[j].start = sub->start;
1486 first[j].end = sub->end;
1487 first[j].lines = sub->lines;
1488 first[j].alignment = sub->alignment;
1489 for(i = 0; i < SUB_MAX_TEXT; ++i){
1490 first[j].text[i] = sub->text[i];
1492 if (previous_sub_end){
1493 first[j].end = first[j - 1].end;
1494 first[j - 1].end = previous_sub_end;
1495 previous_sub_end = 0;
1497 break;
1501 #endif
1502 if(sub==ERR) ++sub_errs; else ++sub_num; // Error vs. Valid
1505 free_stream(fd);
1507 #ifdef CONFIG_ICONV
1508 subcp_close();
1509 #endif
1510 free(alloced_sub);
1512 // printf ("SUB: Subtitle format %s time.\n", uses_time?"uses":"doesn't use");
1513 mp_msg(MSGT_SUBREADER, MSGL_V,"SUB: Read %i subtitles, %i bad line(s).\n",
1514 sub_num, sub_errs);
1516 if(sub_num<=0){
1517 free(first);
1518 return NULL;
1521 // we do overlap if the user forced it (suboverlap_enable == 2) or
1522 // the user didn't forced no-overlapsub and the format is Jacosub or Ssa.
1523 // this is because usually overlapping subtitles are found in these formats,
1524 // while in others they are probably result of bad timing
1525 if ((suboverlap_enabled == 2) ||
1526 ((suboverlap_enabled) && ((sub_format == SUB_JACOSUB) || (sub_format == SUB_SSA)))) {
1527 adjust_subs_time(first, 6.0, fps, 0, sub_num, uses_time);/*~6 secs AST*/
1528 // here we manage overlapping subtitles
1529 sub_orig = sub_num;
1530 n_first = sub_num;
1531 sub_num = 0;
1532 second = NULL;
1533 // for each subtitle in first[] we deal with its 'block' of
1534 // bonded subtitles
1535 for (sub_first = 0; sub_first < n_first; ++sub_first) {
1536 unsigned long global_start = first[sub_first].start,
1537 global_end = first[sub_first].end, local_start, local_end;
1538 int lines_to_add = first[sub_first].lines, sub_to_add = 0,
1539 **placeholder = NULL, higher_line = 0, counter, start_block_sub = sub_num;
1540 char real_block = 1;
1542 // here we find the number of subtitles inside the 'block'
1543 // and its span interval. this works well only with sorted
1544 // subtitles
1545 while ((sub_first + sub_to_add + 1 < n_first) && (first[sub_first + sub_to_add + 1].start < global_end)) {
1546 ++sub_to_add;
1547 lines_to_add += first[sub_first + sub_to_add].lines;
1548 if (first[sub_first + sub_to_add].start < global_start) {
1549 global_start = first[sub_first + sub_to_add].start;
1551 if (first[sub_first + sub_to_add].end > global_end) {
1552 global_end = first[sub_first + sub_to_add].end;
1556 /* Avoid n^2 memory use for the "placeholder" data structure
1557 * below with subtitles that have a huge number of
1558 * consecutive overlapping lines. */
1559 lines_to_add = FFMIN(lines_to_add, SUB_MAX_TEXT);
1561 // we need a structure to keep trace of the screen lines
1562 // used by the subs, a 'placeholder'
1563 counter = 2 * sub_to_add + 1; // the maximum number of subs derived
1564 // from a block of sub_to_add+1 subs
1565 placeholder = malloc(sizeof(int *) * counter);
1566 for (i = 0; i < counter; ++i) {
1567 placeholder[i] = malloc(sizeof(int) * lines_to_add);
1568 for (j = 0; j < lines_to_add; ++j) {
1569 placeholder[i][j] = -1;
1573 counter = 0;
1574 local_end = global_start - 1;
1575 do {
1576 int ls;
1578 // here we find the beginning and the end of a new
1579 // subtitle in the block
1580 local_start = local_end + 1;
1581 local_end = global_end;
1582 for (j = 0; j <= sub_to_add; ++j) {
1583 if ((first[sub_first + j].start - 1 > local_start) && (first[sub_first + j].start - 1 < local_end)) {
1584 local_end = first[sub_first + j].start - 1;
1585 } else if ((first[sub_first + j].end > local_start) && (first[sub_first + j].end < local_end)) {
1586 local_end = first[sub_first + j].end;
1589 // here we allocate the screen lines to subs we must
1590 // display in current local_start-local_end interval.
1591 // if the subs were yet presents in the previous interval
1592 // they keep the same lines, otherside they get unused lines
1593 for (j = 0; j <= sub_to_add; ++j) {
1594 if ((first[sub_first + j].start <= local_end) && (first[sub_first + j].end > local_start)) {
1595 unsigned long sub_lines = first[sub_first + j].lines, fragment_length = lines_to_add + 1,
1596 tmp = 0;
1597 char boolean = 0;
1598 int fragment_position = -1;
1600 // if this is not the first new sub of the block
1601 // we find if this sub was present in the previous
1602 // new sub
1603 if (counter)
1604 for (i = 0; i < lines_to_add; ++i) {
1605 if (placeholder[counter - 1][i] == sub_first + j) {
1606 placeholder[counter][i] = sub_first + j;
1607 boolean = 1;
1610 if (boolean)
1611 continue;
1613 // we are looking for the shortest among all groups of
1614 // sequential blank lines whose length is greater than or
1615 // equal to sub_lines. we store in fragment_position the
1616 // position of the shortest group, in fragment_length its
1617 // length, and in tmp the length of the group currently
1618 // examinated
1619 for (i = 0; i < lines_to_add; ++i) {
1620 if (placeholder[counter][i] == -1) {
1621 // placeholder[counter][i] is part of the current group
1622 // of blank lines
1623 ++tmp;
1624 } else {
1625 if (tmp == sub_lines) {
1626 // current group's size fits exactly the one we
1627 // need, so we stop looking
1628 fragment_position = i - tmp;
1629 tmp = 0;
1630 break;
1632 if ((tmp) && (tmp > sub_lines) && (tmp < fragment_length)) {
1633 // current group is the best we found till here,
1634 // but is still bigger than the one we are looking
1635 // for, so we keep on looking
1636 fragment_length = tmp;
1637 fragment_position = i - tmp;
1638 tmp = 0;
1639 } else {
1640 // current group doesn't fit at all, so we forget it
1641 tmp = 0;
1645 if (tmp) {
1646 // last screen line is blank, a group ends with it
1647 if ((tmp >= sub_lines) && (tmp < fragment_length)) {
1648 fragment_position = i - tmp;
1651 if (fragment_position == -1) {
1652 // it was not possible to find free screen line(s) for a subtitle,
1653 // usually this means a bug in the code; however we do not overlap
1654 mp_msg(MSGT_SUBREADER, MSGL_WARN, "SUB: we could not find a suitable position for an overlapping subtitle\n");
1655 higher_line = SUB_MAX_TEXT + 1;
1656 break;
1657 } else {
1658 for (tmp = 0; tmp < sub_lines; ++tmp) {
1659 placeholder[counter][fragment_position + tmp] = sub_first + j;
1664 for (j = higher_line + 1; j < lines_to_add; ++j) {
1665 if (placeholder[counter][j] != -1)
1666 higher_line = j;
1667 else
1668 break;
1670 if (higher_line >= SUB_MAX_TEXT) {
1671 // the 'block' has too much lines, so we don't overlap the
1672 // subtitles
1673 second = realloc(second, (sub_num + sub_to_add + 1) * sizeof(subtitle));
1674 for (j = 0; j <= sub_to_add; ++j) {
1675 int ls;
1676 memset(&second[sub_num + j], '\0', sizeof(subtitle));
1677 second[sub_num + j].start = first[sub_first + j].start;
1678 second[sub_num + j].end = first[sub_first + j].end;
1679 second[sub_num + j].lines = first[sub_first + j].lines;
1680 second[sub_num + j].alignment = first[sub_first + j].alignment;
1681 for (ls = 0; ls < second[sub_num + j].lines; ls++) {
1682 second[sub_num + j].text[ls] = strdup(first[sub_first + j].text[ls]);
1685 sub_num += sub_to_add + 1;
1686 sub_first += sub_to_add;
1687 real_block = 0;
1688 break;
1691 // we read the placeholder structure and create the new
1692 // subs.
1693 second = realloc(second, (sub_num + 1) * sizeof(subtitle));
1694 memset(&second[sub_num], '\0', sizeof(subtitle));
1695 second[sub_num].start = local_start;
1696 second[sub_num].end = local_end;
1697 second[sub_num].alignment = first[sub_first].alignment;
1698 n_max = (lines_to_add < SUB_MAX_TEXT) ? lines_to_add : SUB_MAX_TEXT;
1699 for (i = 0, j = 0; j < n_max; ++j) {
1700 if (placeholder[counter][j] != -1) {
1701 int lines = first[placeholder[counter][j]].lines;
1702 for (ls = 0; ls < lines; ++ls) {
1703 second[sub_num].text[i++] = strdup(first[placeholder[counter][j]].text[ls]);
1705 j += lines - 1;
1706 } else {
1707 second[sub_num].text[i++] = strdup(" ");
1710 ++sub_num;
1711 ++counter;
1712 } while (local_end < global_end);
1713 if (real_block)
1714 for (i = 0; i < counter; ++i)
1715 second[start_block_sub + i].lines = higher_line + 1;
1717 counter = 2 * sub_to_add + 1;
1718 for (i = 0; i < counter; ++i) {
1719 free(placeholder[i]);
1721 free(placeholder);
1722 sub_first += sub_to_add;
1725 for (j = sub_orig - 1; j >= 0; --j) {
1726 for (i = first[j].lines - 1; i >= 0; --i) {
1727 free(first[j].text[i]);
1730 free(first);
1732 return_sub = second;
1733 } else { //if(suboverlap_enabled)
1734 adjust_subs_time(first, 6.0, fps, 1, sub_num, uses_time);/*~6 secs AST*/
1735 return_sub = first;
1737 if (return_sub == NULL) return NULL;
1738 subt_data = malloc(sizeof(sub_data));
1739 subt_data->filename = strdup(filename);
1740 subt_data->sub_uses_time = uses_time;
1741 subt_data->sub_num = sub_num;
1742 subt_data->sub_errs = sub_errs;
1743 subt_data->subtitles = return_sub;
1744 return subt_data;
1747 #if 0
1748 char * strreplace( char * in,char * what,char * whereof )
1750 int i;
1751 char * tmp;
1753 if ( ( in == NULL )||( what == NULL )||( whereof == NULL )||( ( tmp=strstr( in,what ) ) == NULL ) ) return NULL;
1754 for( i=0;i<strlen( whereof );i++ ) tmp[i]=whereof[i];
1755 if ( strlen( what ) > strlen( whereof ) ) tmp[i]=0;
1756 return in;
1758 #endif
1761 static void strcpy_trim(char *d, char *s)
1763 // skip leading whitespace
1764 while (*s && isspace(*s)) {
1765 s++;
1767 for (;;) {
1768 // copy word
1769 while (*s && !isspace(*s)) {
1770 *d = tolower(*s);
1771 s++; d++;
1773 if (*s == 0) break;
1774 // trim excess whitespace
1775 while (*s && isspace(*s)) {
1776 s++;
1778 if (*s == 0) break;
1779 *d++ = ' ';
1781 *d = 0;
1784 static void strcpy_strip_ext(char *d, char *s)
1786 char *tmp = strrchr(s,'.');
1787 if (!tmp) {
1788 strcpy(d, s);
1789 return;
1790 } else {
1791 strncpy(d, s, tmp-s);
1792 d[tmp-s] = 0;
1794 while (*d) {
1795 *d = tolower(*d);
1796 d++;
1800 static void strcpy_get_ext(char *d, char *s)
1802 char *tmp = strrchr(s,'.');
1803 if (!tmp) {
1804 strcpy(d, "");
1805 return;
1806 } else {
1807 strcpy(d, tmp+1);
1811 static int whiteonly(char *s)
1813 while (*s) {
1814 if (!isspace(*s)) return 0;
1815 s++;
1817 return 1;
1820 typedef struct subfn
1822 int priority;
1823 char *fname;
1824 } subfn;
1826 static int compare_sub_priority(const void *a, const void *b)
1828 if (((const subfn*)a)->priority > ((const subfn*)b)->priority) {
1829 return -1;
1830 } else if (((const subfn*)a)->priority < ((const subfn*)b)->priority) {
1831 return 1;
1832 } else {
1833 return strcoll(((const subfn*)a)->fname, ((const subfn*)b)->fname);
1837 char** sub_filenames(const char* path, char *fname)
1839 char *f_dir, *f_fname, *f_fname_noext, *f_fname_trim, *tmp, *tmp_sub_id;
1840 char *tmp_fname_noext, *tmp_fname_trim, *tmp_fname_ext, *tmpresult;
1842 int len, pos, found, i, j;
1843 char * sub_exts[] = { "utf", "utf8", "utf-8", "sub", "srt", "smi", "rt", "txt", "ssa", "aqt", "jss", "js", "ass", NULL};
1844 subfn *result;
1845 char **result2;
1847 int subcnt;
1849 FILE *f;
1851 DIR *d;
1852 struct dirent *de;
1854 len = (strlen(fname) > 256 ? strlen(fname) : 256)
1855 +(strlen(path) > 256 ? strlen(path) : 256)+2;
1857 f_dir = malloc(len);
1858 f_fname = malloc(len);
1859 f_fname_noext = malloc(len);
1860 f_fname_trim = malloc(len);
1862 tmp_fname_noext = malloc(len);
1863 tmp_fname_trim = malloc(len);
1864 tmp_fname_ext = malloc(len);
1866 tmpresult = malloc(len);
1868 result = calloc(MAX_SUBTITLE_FILES, sizeof(*result));
1870 subcnt = 0;
1872 tmp = strrchr(fname,'/');
1873 #if HAVE_DOS_PATHS
1874 if(!tmp)tmp = strrchr(fname,'\\');
1875 if(!tmp)tmp = strrchr(fname,':');
1876 #endif
1878 // extract filename & dirname from fname
1879 if (tmp) {
1880 strcpy(f_fname, tmp+1);
1881 pos = tmp - fname;
1882 strncpy(f_dir, fname, pos+1);
1883 f_dir[pos+1] = 0;
1884 } else {
1885 strcpy(f_fname, fname);
1886 strcpy(f_dir, "./");
1889 strcpy_strip_ext(f_fname_noext, f_fname);
1890 strcpy_trim(f_fname_trim, f_fname_noext);
1892 /* The code using sub language here is broken - it assumes strict
1893 * "videoname languagename" syntax for the subtitle file, which is
1894 * very unlikely to match especially if language name uses "en,de"
1895 * syntax... */
1896 tmp_sub_id = NULL;
1897 #if 0
1898 if (dvdsub_lang && !whiteonly(dvdsub_lang)) {
1899 tmp_sub_id = malloc(strlen(dvdsub_lang)+1);
1900 strcpy_trim(tmp_sub_id, dvdsub_lang);
1902 #endif
1904 // 0 = nothing
1905 // 1 = any subtitle file
1906 // 2 = any sub file containing movie name
1907 // 3 = sub file containing movie name and the lang extension
1908 for (j = 0; j <= 1; j++) {
1909 d = opendir(j == 0 ? f_dir : path);
1910 if (d) {
1911 while ((de = readdir(d))) {
1912 // retrieve various parts of the filename
1913 strcpy_strip_ext(tmp_fname_noext, de->d_name);
1914 strcpy_get_ext(tmp_fname_ext, de->d_name);
1915 strcpy_trim(tmp_fname_trim, tmp_fname_noext);
1917 // does it end with a subtitle extension?
1918 found = 0;
1919 #ifdef CONFIG_ICONV
1920 #ifdef CONFIG_ENCA
1921 for (i = ((sub_cp && strncasecmp(sub_cp, "enca", 4) != 0) ? 3 : 0); sub_exts[i]; i++) {
1922 #else
1923 for (i = (sub_cp ? 3 : 0); sub_exts[i]; i++) {
1924 #endif
1925 #else
1926 for (i = 0; sub_exts[i]; i++) {
1927 #endif
1928 if (strcasecmp(sub_exts[i], tmp_fname_ext) == 0) {
1929 found = 1;
1930 break;
1934 // we have a (likely) subtitle file
1935 if (found) {
1936 int prio = 0;
1937 if (!prio && tmp_sub_id)
1939 sprintf(tmpresult, "%s %s", f_fname_trim, tmp_sub_id);
1940 if (strcmp(tmp_fname_trim, tmpresult) == 0 && sub_match_fuzziness >= 1) {
1941 // matches the movie name + lang extension
1942 prio = 5;
1945 if (!prio && strcmp(tmp_fname_trim, f_fname_trim) == 0) {
1946 // matches the movie name
1947 prio = 4;
1949 if (!prio && (tmp = strstr(tmp_fname_trim, f_fname_trim)) && (sub_match_fuzziness >= 1)) {
1950 // contains the movie name
1951 tmp += strlen(f_fname_trim);
1952 if (tmp_sub_id && strstr(tmp, tmp_sub_id)) {
1953 // with sub_id specified prefer localized subtitles
1954 prio = 3;
1955 } else if ((tmp_sub_id == NULL) && whiteonly(tmp)) {
1956 // without sub_id prefer "plain" name
1957 prio = 3;
1958 } else {
1959 // with no localized subs found, try any else instead
1960 prio = 2;
1963 if (!prio) {
1964 // doesn't contain the movie name
1965 // don't try in the mplayer subtitle directory
1966 if ((j == 0) && (sub_match_fuzziness >= 2)) {
1967 prio = 1;
1971 mp_msg(MSGT_SUBREADER, MSGL_DBG2, "Potential sub file: "
1972 "\"%s\" Priority: %d\n", de->d_name, prio);
1973 if (prio) {
1974 prio += prio;
1975 #ifdef CONFIG_ICONV
1976 if (i<3){ // prefer UTF-8 coded
1977 prio++;
1979 #endif
1980 sprintf(tmpresult, "%s%s", j == 0 ? f_dir : path, de->d_name);
1981 // fprintf(stderr, "%s priority %d\n", tmpresult, prio);
1982 if ((f = fopen(tmpresult, "rt"))) {
1983 fclose(f);
1984 result[subcnt].priority = prio;
1985 result[subcnt].fname = strdup(tmpresult);
1986 subcnt++;
1991 if (subcnt >= MAX_SUBTITLE_FILES) break;
1993 closedir(d);
1998 if (tmp_sub_id) free(tmp_sub_id);
2000 free(f_dir);
2001 free(f_fname);
2002 free(f_fname_noext);
2003 free(f_fname_trim);
2005 free(tmp_fname_noext);
2006 free(tmp_fname_trim);
2007 free(tmp_fname_ext);
2009 free(tmpresult);
2011 qsort(result, subcnt, sizeof(subfn), compare_sub_priority);
2013 result2 = calloc(subcnt + 1, sizeof(*result2));
2015 for (i = 0; i < subcnt; i++) {
2016 result2[i] = result[i].fname;
2018 result2[subcnt] = NULL;
2020 free(result);
2022 return result2;
2025 void list_sub_file(sub_data* subd){
2026 int i,j;
2027 subtitle *subs = subd->subtitles;
2029 for(j=0; j < subd->sub_num; j++){
2030 subtitle* egysub=&subs[j];
2031 mp_msg(MSGT_SUBREADER,MSGL_INFO,"%i line%c (%li-%li)\n",
2032 egysub->lines,
2033 (1==egysub->lines)?' ':'s',
2034 egysub->start,
2035 egysub->end);
2036 for (i=0; i<egysub->lines; i++) {
2037 mp_msg(MSGT_SUBREADER,MSGL_INFO,"\t\t%d: %s%s", i,egysub->text[i], i==egysub->lines-1?"":" \n ");
2039 mp_msg(MSGT_SUBREADER,MSGL_INFO,"\n");
2042 mp_msg(MSGT_SUBREADER,MSGL_INFO,"Subtitle format %s time.\n",
2043 subd->sub_uses_time ? "uses":"doesn't use");
2044 mp_msg(MSGT_SUBREADER,MSGL_INFO,"Read %i subtitles, %i errors.\n", subd->sub_num, subd->sub_errs);
2047 void dump_srt(sub_data* subd, float fps){
2048 int i,j;
2049 int h,m,s,ms;
2050 FILE * fd;
2051 subtitle * onesub;
2052 unsigned long temp;
2053 subtitle *subs = subd->subtitles;
2055 if (!subd->sub_uses_time && sub_fps == 0)
2056 sub_fps = fps;
2057 fd=fopen("dumpsub.srt","w");
2058 if(!fd)
2060 perror("dump_srt: fopen");
2061 return;
2063 for(i=0; i < subd->sub_num; i++)
2065 onesub=subs+i; //=&subs[i];
2066 fprintf(fd,"%d\n",i+1);//line number
2068 temp=onesub->start;
2069 if (!subd->sub_uses_time)
2070 temp = temp * 100 / sub_fps;
2071 temp -= sub_delay * 100;
2072 h=temp/360000;temp%=360000; //h =1*100*60*60
2073 m=temp/6000; temp%=6000; //m =1*100*60
2074 s=temp/100; temp%=100; //s =1*100
2075 ms=temp*10; //ms=1*10
2076 fprintf(fd,"%02d:%02d:%02d,%03d --> ",h,m,s,ms);
2078 temp=onesub->end;
2079 if (!subd->sub_uses_time)
2080 temp = temp * 100 / sub_fps;
2081 temp -= sub_delay * 100;
2082 h=temp/360000;temp%=360000;
2083 m=temp/6000; temp%=6000;
2084 s=temp/100; temp%=100;
2085 ms=temp*10;
2086 fprintf(fd,"%02d:%02d:%02d,%03d\n",h,m,s,ms);
2088 for(j=0;j<onesub->lines;j++)
2089 fprintf(fd,"%s\n",onesub->text[j]);
2091 fprintf(fd,"\n");
2093 fclose(fd);
2094 mp_msg(MSGT_SUBREADER,MSGL_INFO,"SUB: Subtitles dumped in \'dumpsub.srt\'.\n");
2097 void dump_mpsub(sub_data* subd, float fps){
2098 int i,j;
2099 FILE *fd;
2100 float a,b;
2101 subtitle *subs = subd->subtitles;
2103 mpsub_position = subd->sub_uses_time? (sub_delay*100) : (sub_delay*fps);
2104 if (sub_fps==0) sub_fps=fps;
2106 fd=fopen ("dump.mpsub", "w");
2107 if (!fd) {
2108 perror ("dump_mpsub: fopen");
2109 return;
2113 if (subd->sub_uses_time) fprintf (fd,"FORMAT=TIME\n\n");
2114 else fprintf (fd, "FORMAT=%5.2f\n\n", fps);
2116 for(j=0; j < subd->sub_num; j++){
2117 subtitle* egysub=&subs[j];
2118 if (subd->sub_uses_time) {
2119 a=((egysub->start-mpsub_position)/100.0);
2120 b=((egysub->end-egysub->start)/100.0);
2121 if ( (float)((int)a) == a)
2122 fprintf (fd, "%.0f",a);
2123 else
2124 fprintf (fd, "%.2f",a);
2126 if ( (float)((int)b) == b)
2127 fprintf (fd, " %.0f\n",b);
2128 else
2129 fprintf (fd, " %.2f\n",b);
2130 } else {
2131 fprintf (fd, "%ld %ld\n", (long)((egysub->start*(fps/sub_fps))-((mpsub_position*(fps/sub_fps)))),
2132 (long)(((egysub->end)-(egysub->start))*(fps/sub_fps)));
2135 mpsub_position = egysub->end;
2136 for (i=0; i<egysub->lines; i++) {
2137 fprintf (fd, "%s\n",egysub->text[i]);
2139 fprintf (fd, "\n");
2141 fclose (fd);
2142 mp_msg(MSGT_SUBREADER,MSGL_INFO,"SUB: Subtitles dumped in \'dump.mpsub\'.\n");
2145 void dump_microdvd(sub_data* subd, float fps) {
2146 int i, delay;
2147 FILE *fd;
2148 subtitle *subs = subd->subtitles;
2149 if (sub_fps == 0)
2150 sub_fps = fps;
2151 fd = fopen("dumpsub.sub", "w");
2152 if (!fd) {
2153 perror("dumpsub.sub: fopen");
2154 return;
2156 delay = sub_delay * sub_fps;
2157 for (i = 0; i < subd->sub_num; ++i) {
2158 int j, start, end;
2159 start = subs[i].start;
2160 end = subs[i].end;
2161 if (subd->sub_uses_time) {
2162 start = start * sub_fps / 100 ;
2163 end = end * sub_fps / 100;
2165 else {
2166 start = start * sub_fps / fps;
2167 end = end * sub_fps / fps;
2169 start -= delay;
2170 end -= delay;
2171 fprintf(fd, "{%d}{%d}", start, end);
2172 for (j = 0; j < subs[i].lines; ++j)
2173 fprintf(fd, "%s%s", j ? "|" : "", subs[i].text[j]);
2174 fprintf(fd, "\n");
2176 fclose(fd);
2177 mp_msg(MSGT_SUBREADER,MSGL_INFO,"SUB: Subtitles dumped in \'dumpsub.sub\'.\n");
2180 void dump_jacosub(sub_data* subd, float fps) {
2181 int i,j;
2182 int h,m,s,cs;
2183 FILE * fd;
2184 subtitle * onesub;
2185 unsigned long temp;
2186 subtitle *subs = subd->subtitles;
2188 if (!subd->sub_uses_time && sub_fps == 0)
2189 sub_fps = fps;
2190 fd=fopen("dumpsub.jss","w");
2191 if(!fd)
2193 perror("dump_jacosub: fopen");
2194 return;
2196 fprintf(fd, "#TIMERES %d\n", (subd->sub_uses_time) ? 100 : (int)sub_fps);
2197 for(i=0; i < subd->sub_num; i++)
2199 onesub=subs+i; //=&subs[i];
2201 temp=onesub->start;
2202 if (!subd->sub_uses_time)
2203 temp = temp * 100 / sub_fps;
2204 temp -= sub_delay * 100;
2205 h=temp/360000;temp%=360000; //h =1*100*60*60
2206 m=temp/6000; temp%=6000; //m =1*100*60
2207 s=temp/100; temp%=100; //s =1*100
2208 cs=temp; //cs=1*10
2209 fprintf(fd,"%02d:%02d:%02d.%02d ",h,m,s,cs);
2211 temp=onesub->end;
2212 if (!subd->sub_uses_time)
2213 temp = temp * 100 / sub_fps;
2214 temp -= sub_delay * 100;
2215 h=temp/360000;temp%=360000;
2216 m=temp/6000; temp%=6000;
2217 s=temp/100; temp%=100;
2218 cs=temp;
2219 fprintf(fd,"%02d:%02d:%02d.%02d {~} ",h,m,s,cs);
2221 for(j=0;j<onesub->lines;j++)
2222 fprintf(fd,"%s%s",j ? "\\n" : "", onesub->text[j]);
2224 fprintf(fd,"\n");
2226 fclose(fd);
2227 mp_msg(MSGT_SUBREADER,MSGL_INFO,"SUB: Subtitles dumped in \'dumpsub.js\'.\n");
2230 void dump_sami(sub_data* subd, float fps) {
2231 int i,j;
2232 FILE * fd;
2233 subtitle * onesub;
2234 unsigned long temp;
2235 subtitle *subs = subd->subtitles;
2237 if (!subd->sub_uses_time && sub_fps == 0)
2238 sub_fps = fps;
2239 fd=fopen("dumpsub.smi","w");
2240 if(!fd)
2242 perror("dump_jacosub: fopen");
2243 return;
2245 fprintf(fd, "<SAMI>\n"
2246 "<HEAD>\n"
2247 " <STYLE TYPE=\"Text/css\">\n"
2248 " <!--\n"
2249 " P {margin-left: 29pt; margin-right: 29pt; font-size: 24pt; text-align: center; font-family: Tahoma; font-weight: bold; color: #FCDD03; background-color: #000000;}\n"
2250 " .SUBTTL {Name: 'Subtitles'; Lang: en-US; SAMIType: CC;}\n"
2251 " -->\n"
2252 " </STYLE>\n"
2253 "</HEAD>\n"
2254 "<BODY>\n");
2255 for(i=0; i < subd->sub_num; i++)
2257 onesub=subs+i; //=&subs[i];
2259 temp=onesub->start;
2260 if (!subd->sub_uses_time)
2261 temp = temp * 100 / sub_fps;
2262 temp -= sub_delay * 100;
2263 fprintf(fd,"\t<SYNC Start=%lu>\n"
2264 "\t <P>", temp * 10);
2266 for(j=0;j<onesub->lines;j++)
2267 fprintf(fd,"%s%s",j ? "<br>" : "", onesub->text[j]);
2269 fprintf(fd,"\n");
2271 temp=onesub->end;
2272 if (!subd->sub_uses_time)
2273 temp = temp * 100 / sub_fps;
2274 temp -= sub_delay * 100;
2275 fprintf(fd,"\t<SYNC Start=%lu>\n"
2276 "\t <P>&nbsp;\n", temp * 10);
2278 fprintf(fd, "</BODY>\n"
2279 "</SAMI>\n");
2280 fclose(fd);
2281 mp_msg(MSGT_SUBREADER,MSGL_INFO,"SUB: Subtitles dumped in \'dumpsub.smi\'.\n");
2284 void sub_free( sub_data * subd )
2286 int i, j;
2288 if ( !subd ) return;
2290 for (i = 0; i < subd->sub_num; i++)
2291 for (j = 0; j < subd->subtitles[i].lines; j++)
2292 free( subd->subtitles[i].text[j] );
2293 free( subd->subtitles );
2294 free( subd->filename );
2295 free( subd );
2298 #define MAX_SUBLINE 512
2300 * \brief parse text and append it to subtitle in sub
2301 * \param sub subtitle struct to add text to
2302 * \param txt text to parse
2303 * \param len length of text in txt
2304 * \param endpts pts at which this subtitle text should be removed again
2306 * <> and {} are interpreted as comment delimiters, "\n", "\N", '\n', '\r'
2307 * and '\0' are interpreted as newlines, duplicate, leading and trailing
2308 * newlines are ignored.
2310 void sub_add_text(subtitle *sub, const char *txt, int len, double endpts) {
2311 int comment = 0;
2312 int double_newline = 1; // ignore newlines at the beginning
2313 int i, pos;
2314 char *buf;
2315 #ifdef CONFIG_FRIBIDI
2316 int orig_lines = sub->lines;
2317 #endif
2318 if (sub->lines >= SUB_MAX_TEXT) return;
2319 pos = 0;
2320 buf = malloc(MAX_SUBLINE + 1);
2321 sub->text[sub->lines] = buf;
2322 sub->endpts[sub->lines] = endpts;
2323 for (i = 0; i < len && pos < MAX_SUBLINE; i++) {
2324 char c = txt[i];
2325 if (c == '<') comment |= 1;
2326 if (c == '{') comment |= 2;
2327 if (comment) {
2328 if (c == '}') comment &= ~2;
2329 if (c == '>') comment &= ~1;
2330 continue;
2332 if (pos == MAX_SUBLINE - 1) {
2333 i--;
2334 c = 0;
2336 if (c == '\\' && i + 1 < len) {
2337 c = txt[++i];
2338 if (c == 'n' || c == 'N') c = 0;
2340 if (c == '\n' || c == '\r') c = 0;
2341 if (c) {
2342 double_newline = 0;
2343 buf[pos++] = c;
2344 } else if (!double_newline) {
2345 if (sub->lines >= SUB_MAX_TEXT - 1) {
2346 mp_msg(MSGT_VO, MSGL_WARN, "Too many subtitle lines\n");
2347 break;
2349 double_newline = 1;
2350 buf[pos] = 0;
2351 sub->lines++;
2352 pos = 0;
2353 buf = malloc(MAX_SUBLINE + 1);
2354 sub->text[sub->lines] = buf;
2355 sub->endpts[sub->lines] = endpts;
2358 buf[pos] = 0;
2359 if (sub->lines < SUB_MAX_TEXT &&
2360 strlen(sub->text[sub->lines]))
2361 sub->lines++;
2362 #ifdef CONFIG_FRIBIDI
2363 sub = sub_fribidi(sub, sub_utf8, orig_lines);
2364 #endif
2367 #define MP_NOPTS_VALUE (-1LL<<63)
2369 * \brief remove outdated subtitle lines.
2370 * \param sub subtitle struct to modify
2371 * \param pts current pts. All lines with endpts <= this will be removed.
2372 * Use MP_NOPTS_VALUE to remove all lines
2373 * \return 1 if sub was modified, 0 otherwise.
2375 int sub_clear_text(subtitle *sub, double pts) {
2376 int i = 0;
2377 int changed = 0;
2378 while (i < sub->lines) {
2379 double endpts = sub->endpts[i];
2380 if (pts == MP_NOPTS_VALUE || (endpts != MP_NOPTS_VALUE && pts >= endpts)) {
2381 int j;
2382 free(sub->text[i]);
2383 for (j = i + 1; j < sub->lines; j++) {
2384 sub->text[j - 1] = sub->text[j];
2385 sub->endpts[j - 1] = sub->endpts[j];
2387 sub->lines--;
2388 changed = 1;
2389 } else
2390 i++;
2392 return changed;