Commit 8e857dc4 authored by ivo's avatar ivo

I keep forgetting I have to manually add files. Sorry about that.

svn path=/trunk/vorbis-tools/; revision=15396
parent 0d6a0f00
/* OggEnc
**
** This program is distributed under the GNU General Public License, version 2.
** A copy of this license is included with this source.
**
** This particular file may also be distributed under (at your option) any
** later version of the GNU General Public License.
**
** Copyright 2008, ogg.k.ogg.k <ogg.k.ogg.k@googlemail.com>
**
** Portions from ffmpeg2theora, (c) j <j@v2v.cc>
**/
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdio.h>
#include <malloc.h>
#include <string.h>
#include <errno.h>
#ifdef HAVE_KATE
#include <kate/oggkate.h>
#endif
#include "lyrics.h"
#include "utf8.h"
#include "i18n.h"
typedef enum {
lf_unknown,
lf_srt,
lf_lrc,
} lyrics_format;
#ifdef HAVE_KATE
static char *fgets2(char *s,size_t sz,FILE *f)
{
char *ret = fgets(s, sz, f);
if (ret) {
/* fixup DOS newline character */
char *ptr=strchr(ret, '\r');
if (ptr) {
*ptr='\n';
*(ptr+1)=0;
}
}
return ret;
}
static double hmsms2s(int h,int m,int s,int ms)
{
return h*3600+m*60+s+ms/1000.0;
}
static int add_lyrics(oe_lyrics *lyrics, char *text, kate_motion *km, double t0,double t1)
{
size_t len;
int ret;
char *utf8;
ret=utf8_encode(text,&utf8);
if (ret<0) {
fprintf(stderr,_("Failed to convert to UTF-8: %s\n"),text);
return ret;
}
len = strlen(utf8);
lyrics->lyrics = (oe_lyrics_item*)realloc(lyrics->lyrics, (lyrics->count+1)*sizeof(oe_lyrics_item));
if (!lyrics->lyrics) {
fprintf(stderr, _("Out of memory\n"));
return -1;
}
ret=kate_text_validate(kate_utf8,utf8,len+1);
if (ret<0) {
fprintf(stderr,_("WARNING: subtitle %s is not valid UTF-8\n"),utf8);
}
else {
/* kill off trailing \n characters */
while (len>0) {
if (utf8[len-1]=='\n') utf8[--len]=0; else break;
}
lyrics->lyrics[lyrics->count].text = utf8;
lyrics->lyrics[lyrics->count].len = len;
lyrics->lyrics[lyrics->count].t0 = t0;
lyrics->lyrics[lyrics->count].t1 = t1;
lyrics->lyrics[lyrics->count].km = km;
lyrics->count++;
}
return 0;
}
static int is_line_empty(const char *s)
{
/* will work fine with UTF-8 despite the appearance */
if (s) while (*s) {
if (!strchr(" \t\r\n",*s)) return 0;
++s;
}
return 1;
}
static oe_lyrics *load_srt_lyrics(FILE *f)
{
enum { need_id, need_timing, need_text };
int need = need_id;
int last_seen_id=0;
int ret;
int id;
static char text[4096];
static char str[4096];
int h0,m0,s0,ms0,h1,m1,s1,ms1;
double t0=0.0;
double t1=0.0;
oe_lyrics *lyrics;
unsigned int line=0;
if (!f) return NULL;
lyrics=(oe_lyrics*)malloc(sizeof(oe_lyrics));
if (!lyrics) return NULL;
lyrics->count = 0;
lyrics->lyrics = NULL;
lyrics->karaoke = 0;
fgets2(str,sizeof(str),f);
++line;
while (!feof(f)) {
switch (need) {
case need_id:
if (is_line_empty(str)) {
/* be nice and ignore extra empty lines between records */
}
else {
ret=sscanf(str,"%d\n",&id);
if (ret!=1 || id<0) {
fprintf(stderr,_("ERROR - line %u: Syntax error: %s\n"),line,str);
free_lyrics(lyrics);
return NULL;
}
if (id!=last_seen_id+1) {
fprintf(stderr,_("WARNING - line %u: non consecutive ids: %s - pretending not to have noticed\n"),line,str);
}
last_seen_id=id;
need=need_timing;
strcpy(text,"");
}
break;
case need_timing:
/* we could use %u, but glibc accepts minus signs for %u for some reason */
ret=sscanf(str,"%d:%d:%d%*[.,]%d --> %d:%d:%d%*[.,]%d\n",&h0,&m0,&s0,&ms0,&h1,&m1,&s1,&ms1);
if (ret!=8 || (h0|m0|s0|ms0)<0 || (h1|m1|s1|ms1)<0) {
fprintf(stderr,_("ERROR - line %u: Syntax error: %s\n"),line,str);
free_lyrics(lyrics);
return NULL;
}
else if (t1<t0) {
fprintf(stderr,_("ERROR - line %u: end time must not be less than start time: %s\n"),line,str);
free_lyrics(lyrics);
return NULL;
}
else {
t0=hmsms2s(h0,m0,s0,ms0);
t1=hmsms2s(h1,m1,s1,ms1);
}
need=need_text;
break;
case need_text:
if (str[0]=='\n') {
if (add_lyrics(lyrics,text,NULL,t0,t1) < 0) {
free_lyrics(lyrics);
return NULL;
}
need=need_id;
}
else {
/* in case of very long lines */
size_t len=strlen(text);
if (len+strlen(str) >= sizeof(text)) {
fprintf(stderr, _("WARNING - line %u: text is too long - truncated\n"),line);
}
strncpy(text+len,str,sizeof(text)-len);
text[sizeof(text)-1]=0;
}
break;
}
fgets2(str,sizeof(str),f);
++line;
}
if (need!=need_id) {
/* shouldn't be a problem though, but warn */
fprintf(stderr, _("WARNING - line %u: missing data - truncated file?\n"),line);
}
return lyrics;
}
static void add_kate_karaoke_tag(kate_motion *km,kate_float dt,const char *str,size_t len,int line)
{
kate_curve *kc;
kate_float ptr=(kate_float)-0.5;
int ret;
if (dt<0) {
fprintf(stderr, _("WARNING - line %d: lyrics times must not be decreasing\n"), line);
return;
}
/* work out how many glyphs we have */
while (len>0) {
ret=kate_text_get_character(kate_utf8,&str,&len);
if (ret<0) {
fprintf(stderr, _("WARNING - line %d: failed to get UTF-8 glyph from string\n"), line);
return;
}
ptr+=(kate_float)1.0;
}
/* ptr now points to the middle of the glyph we're at */
kc=(kate_curve*)malloc(sizeof(kate_curve));
kate_curve_init(kc);
kc->type=kate_curve_static;
kc->npts=1;
kc->pts=(kate_float*)malloc(2*sizeof(kate_float));
kc->pts[0]=ptr;
kc->pts[1]=(kate_float)0;
km->ncurves++;
km->curves=(kate_curve**)realloc(km->curves,km->ncurves*sizeof(kate_curve*));
km->durations=(kate_float*)realloc(km->durations,km->ncurves*sizeof(kate_float));
km->curves[km->ncurves-1]=kc;
km->durations[km->ncurves-1]=dt;
}
static kate_motion *process_enhanced_lrc_tags(char *str,kate_float start_time,kate_float end_time,int line)
{
char *start,*end;
int ret;
int m,s,cs;
kate_motion *km=NULL;
kate_float current_time = start_time;
if (!str) return NULL;
start=str;
while (1) {
start=strchr(start,'<');
if (!start) break;
end=strchr(start+1,'>');
if (!end) break;
/* we found a <> pair, parse it */
ret=sscanf(start,"<%d:%d.%d>",&m,&s,&cs);
/* remove the <> tag from input to get raw text */
memmove(start,end+1,strlen(end+1)+1);
if (ret!=3 || (m|s|cs)<0) {
fprintf(stderr, _("WARNING - line %d: failed to process enhanced LRC tag (%*.*s) - ignored\n"),line,(int)(end-start+1),(int)(end-start+1),start);
}
else {
kate_float tag_time=hmsms2s(0,m,s,cs*10);
/* if this is the first tag in this line, create a kate motion */
if (!km) {
km=(kate_motion*)malloc(sizeof(kate_motion));
if (!km) {
fprintf(stderr, _("WARNING: failed to allocate memory - enhanced LRC tag will be ignored\n"));
}
else {
kate_motion_init(km);
km->semantics=kate_motion_semantics_glyph_pointer_1;
}
}
/* add to the kate motion */
if (km) {
add_kate_karaoke_tag(km,tag_time-current_time,str,start-str,line);
current_time = tag_time;
}
}
}
/* if we've found karaoke info, extend the motion to the end time */
if (km) {
add_kate_karaoke_tag(km,end_time-current_time,str,strlen(str),line);
}
return km;
}
static oe_lyrics *load_lrc_lyrics(FILE *f)
{
oe_lyrics *lyrics;
static char str[4096];
static char lyrics_line[4096]="";
int m,s,cs;
double t,start_time = -1.0;
int offset;
int ret;
unsigned line=0;
kate_motion *km;
if (!f) return NULL;
/* skip headers */
fgets2(str,sizeof(str),f);
++line;
while (!feof(f)) {
ret = sscanf(str, "[%d:%d.%d]%n\n",&m,&s,&cs,&offset);
if (ret == 3)
break;
fgets2(str,sizeof(str),f);
++line;
}
if (feof(f)) {
fprintf(stderr,_("ERROR - line %u: Syntax error: %s\n"),line,str);
return NULL;
}
lyrics=(oe_lyrics*)malloc(sizeof(oe_lyrics));
if (!lyrics) return NULL;
lyrics->count = 0;
lyrics->lyrics = NULL;
lyrics->karaoke = 0;
while (!feof(f)) {
/* ignore empty lines */
if (!is_line_empty(str)) {
ret=sscanf(str, "[%d:%d.%d]%n\n",&m,&s,&cs,&offset);
if (ret != 3 || (m|s|cs)<0) {
fprintf(stderr,_("ERROR - line %u: Syntax error: %s\n"),line,str);
free_lyrics(lyrics);
return NULL;
}
t=hmsms2s(0,m,s,cs*10);
if (start_time>=0.0 && !is_line_empty(lyrics_line)) {
km=process_enhanced_lrc_tags(lyrics_line,start_time,t,line);
if (km) {
lyrics->karaoke = 1;
}
if (add_lyrics(lyrics,lyrics_line,km,start_time,t) < 0) {
free_lyrics(lyrics);
return NULL;
}
}
strncpy(lyrics_line,str+offset,sizeof(lyrics_line));
lyrics_line[sizeof(lyrics_line)-1]=0;
start_time=t;
}
fgets2(str,sizeof(str),f);
++line;
}
return lyrics;
}
/* very weak checks, but we only support two formats, so it's ok */
lyrics_format probe_lyrics_format(FILE *f)
{
int dummy_int;
static char str[4096];
lyrics_format format=lf_unknown;
long pos;
if (!f) return lf_unknown;
pos=ftell(f);
fgets2(str,sizeof(str),f);
/* srt */
if (sscanf(str, "%d\n", &dummy_int) == 1 && dummy_int>=0)
format=lf_srt;
/* lrc */
if (str[0] == '[')
format=lf_lrc;
fseek(f,pos,SEEK_SET);
return format;
}
#endif
oe_lyrics *load_lyrics(const char *filename)
{
#ifdef HAVE_KATE
static char str[4096];
int ret;
oe_lyrics *lyrics=NULL;
FILE *f;
if (!filename) {
fprintf(stderr,_("ERROR: No lyrics filename to load from\n"));
return NULL;
}
f = fopen(filename, "r");
if (!f) {
fprintf(stderr,_("ERROR: Failed to open lyrics file %s (%s)\n"), filename, strerror(errno));
return NULL;
}
/* first, check for a BOM */
ret=fread(str,1,3,f);
if (ret<3 || memcmp(str,"\xef\xbb\xbf",3)) {
/* No BOM, rewind */
fseek(f,0,SEEK_SET);
}
switch (probe_lyrics_format(f)) {
case lf_srt:
lyrics = load_srt_lyrics(f);
break;
case lf_lrc:
lyrics = load_lrc_lyrics(f);
break;
default:
fprintf(stderr, _("ERROR: Failed to load %s - can't determine format\n"), filename);
break;
}
fclose(f);
return lyrics;
#else
return NULL;
#endif
}
void free_lyrics(oe_lyrics *lyrics)
{
size_t n,c;
if (lyrics) {
for (n=0; n<lyrics->count; ++n) {
oe_lyrics_item *li=&lyrics->lyrics[n];
free(li->text);
if (li->km) {
for (c=0; c<li->km->ncurves; ++c) {
free(li->km->curves[c]->pts);
free(li->km->curves[c]);
}
free(li->km->curves);
free(li->km->durations);
free(li->km);
}
}
free(lyrics->lyrics);
free(lyrics);
}
}
const oe_lyrics_item *get_lyrics(const oe_lyrics *lyrics, double t, size_t *idx)
{
if (!lyrics || *idx>=lyrics->count) return NULL;
if (lyrics->lyrics[*idx].t0 > t) return NULL;
return &lyrics->lyrics[(*idx)++];
}
#ifndef __LYRICS_H
#define __LYRICS_H
#include <stddef.h>
#ifdef HAVE_KATE
#include <kate/kate.h>
#endif
typedef struct oe_lyrics_item {
char *text;
size_t len;
double t0;
double t1;
#ifdef HAVE_KATE
kate_motion *km;
#endif
} oe_lyrics_item;
typedef struct oe_lyrics {
size_t count;
oe_lyrics_item *lyrics;
int karaoke;
} oe_lyrics;
extern oe_lyrics *load_lyrics(const char *filename);
extern void free_lyrics(oe_lyrics *lyrics);
extern const oe_lyrics_item *get_lyrics(const oe_lyrics *lyrics, double t, size_t *idx);
#endif
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment