mpv/TOOLS/subreader.c

240 lines
6.0 KiB
C

/*
* Subtitle reader with format autodetection
* Mier nem muxik realloccal!?!?! - nekem muxik :)
*
* Written by laaz
* Some code cleanup & realloc() by A'rpi/ESP-team
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define ERR (void *)-1
#define MAX_TEXT 5
int sub_uses_time=0;
int sub_errs=0;
int sub_num=0; // number of subtitle structs
int sub_format=-1; // 0 for microdvd, 1 for SubRip, 2 for the third format
struct subtitle {
int lines;
unsigned long start;
unsigned long end;
char *text[MAX_TEXT];
};
char *sub_readtext(char *source, char **dest) {
int len=0;
char *p;
for (p=source;*p!='\r' && *p!='\n' && *p!='|'; p++,len++);
*dest= (char *)malloc (len+1);
if (!dest) {return ERR;}
strncpy(*dest, source, len);
(*dest)[len]=0;
while (*p=='\r' || *p=='\n' || *p=='|') p++;
if (*p) return p; // not-last text field
else return NULL; // last text field
}
struct subtitle *sub_read_line_microdvd(FILE *fd,struct subtitle *current) {
char line[1001];
char line2[1001];
char *p, *next;
int i;
bzero (current, sizeof(current));
do {
if (!fgets (line, 1000, fd)) return NULL;
} while (*line=='\n' || *line == '\r' || !*line);
if (sscanf (line, "{%i}{%i}%s", &(current->start), &(current->end),line2) <2) {return ERR;}
p=line;
while (*p++!='}');
while (*p++!='}');
next=p, i=0;
while ((next =sub_readtext (next, &(current->text[i])))) {
if (current->text[i]==ERR || current->text[i]==ERR) {return ERR;}
i++;
if (i>MAX_TEXT) { printf ("Too many lines in a subtitle\n");return ERR;}
}
current->lines=i+1;
return current;
}
struct subtitle *sub_read_line_subrip(FILE *fd, struct subtitle *current) {
char line[1001];
int a1,a2,a3,a4,b1,b2,b3,b4;
char *p=NULL, *q=NULL;
int len;
bzero (current, sizeof(current));
while (!current->text[0]) {
if (!fgets (line, 1000, fd)) return NULL;
if (sscanf (line, "%i:%i:%i.%i,%i:%i:%i.%i",&a1,&a2,&a3,&a4,&b1,&b2,&b3,&b4) < 8) continue;
current->start = a1*360000+a2*6000+a3*100+a4;
current->end = b1*360000+b2*6000+b3*100+b4;
if (!fgets (line, 1000, fd)) return NULL;
p=q=line;
for (current->lines=1; current->lines < MAX_TEXT; current->lines++) {
for (q=p,len=0; *p && *p!='\r' && *p!='\n' && strncmp(p,"[br]",4); p++,len++);
current->text[current->lines-1]=(char *)malloc (len+1);
if (!current->text[current->lines-1]) return ERR;
strncpy (current->text[current->lines-1], q, len);
if (!*p || *p=='\r' || *p=='\n') break;
while (*p++!=']');
}
}
return current;
}
struct subtitle *sub_read_line_third(FILE *fd,struct subtitle *current) {
char line[1001];
int a1,a2,a3,a4,b1,b2,b3,b4;
char *p=NULL;
int i,len;
bzero (current, sizeof(current));
while (!current->text[0]) {
if (!fgets (line, 1000, fd)) return NULL;
if ((len=sscanf (line, "%i:%i:%i,%i --> %i:%i:%i,%i",&a1,&a2,&a3,&a4,&b1,&b2,&b3,&b4)) < 8)
continue;
current->start = a1*360000+a2*6000+a3*100+a4/10;
current->end = b1*360000+b2*6000+b3*100+b4/10;
for (i=0; i<MAX_TEXT;) {
if (!fgets (line, 1000, fd)) return NULL;
len=0;
for (p=line; *p!='\n' && *p!='\r' && *p; p++,len++);
if (len) {
current->text[i]=(char *)malloc (len+1);
if (!current->text[i]) return ERR;
strncpy (current->text[i], line, len);
i++;
} else {
break;
}
}
current->lines=i;
}
return current;
}
int sub_autodetect (FILE *fd) {
char line[1001];
int i,j=0;
// char *p;
while (1) {
j++;
if (!fgets (line, 1000, fd))
return -1;
// if (sscanf (line, "{%i}{%i}", &i, &i, p)==2) // ha valaki tudja miert 2, mondja mar el nekem ;)
if (sscanf (line, "{%i}{%i}", &i, &i)==2) // ha valaki tudja miert 2, mondja mar el nekem ;)
{sub_uses_time=0;return 0;}
if (sscanf (line, "%i:%i:%i.%i,%i:%i:%i.%i", &i, &i, &i, &i, &i, &i, &i, &i)==8)
{sub_uses_time=1;return 1;}
if (sscanf (line, "%i:%i:%i,%i --> %i:%i:%i,%i", &i, &i, &i, &i, &i, &i, &i, &i)==8)
{sub_uses_time=1;return 2;}
if (j>100) return -1; // too many bad lines or bad coder
}
}
struct subtitle * sub_get_subtitles (char *filename) {
FILE *fd;
int n_max;
struct subtitle *first;
struct subtitle * (*func[3])(FILE *fd,struct subtitle *dest)=
{
sub_read_line_microdvd,
sub_read_line_subrip,
sub_read_line_third
};
fd=fopen (filename, "r"); if (!fd) return NULL;
sub_format=sub_autodetect (fd);
if (sub_format==-1) {printf ("SUB: Could not determine file format\n");return NULL;}
printf ("SUB: Detected subtitle file format: %i\n",sub_format);
rewind (fd);
sub_num=0;n_max=32;
first=(struct subtitle *)malloc(n_max*sizeof(struct subtitle));
if(!first) return NULL;
while(1){
struct subtitle *sub;
if(sub_num>=n_max){
n_max+=16;
first=realloc(first,n_max*sizeof(struct subtitle));
}
sub=func[sub_format](fd,&first[sub_num]);
if(!sub) break; // EOF
if(sub==ERR) ++sub_errs; else ++sub_num; // Error vs. Valid
}
fclose(fd);
return first;
}
int main(int argc, char **argv) { // for testing
int i,j;
struct subtitle *subs;
struct subtitle *egysub;
if(argc<2){
printf("\nUsage: subreader filename.sub\n\n");
exit(1);
}
subs=sub_get_subtitles(argv[1]);
if(!subs){
printf("Couldn't load file... let's write a bugreport :)\n");
exit(1);
}
for(j=0;j<sub_num;j++){
egysub=&subs[j];
printf ("%i line%c (%i-%i) ",
egysub->lines,
(1==egysub->lines)?' ':'s',
egysub->start,
egysub->end);
for (i=0; i<egysub->lines; i++) {
printf ("%s%s",egysub->text[i], i==egysub->lines-1?"":" <BREAK> ");
}
printf ("\n");
}
printf ("Subtitle format %s time.\n", sub_uses_time?"uses":"doesn't use");
printf ("Read %i subtitles, %i errors.\n", sub_num, sub_errs);
return 0;
}