From e15ae9a60043d8d9c1b110607c5059f13ca86421 Mon Sep 17 00:00:00 2001 From: eugeni Date: Fri, 7 Jul 2006 18:26:51 +0000 Subject: Initial libass release (without mencoder support). git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@18942 b3059339-0415-0410-9bf9-f77b7e298cf2 --- libass/Makefile | 50 ++ libass/ass.c | 857 +++++++++++++++++++++++ libass/ass.h | 156 +++++ libass/ass_cache.c | 209 ++++++ libass/ass_cache.h | 50 ++ libass/ass_fontconfig.c | 192 ++++++ libass/ass_fontconfig.h | 11 + libass/ass_mp.c | 10 + libass/ass_mp.h | 12 + libass/ass_render.c | 1766 +++++++++++++++++++++++++++++++++++++++++++++++ libass/ass_types.h | 83 +++ libass/ass_utils.c | 63 ++ libass/ass_utils.h | 9 + 13 files changed, 3468 insertions(+) create mode 100644 libass/Makefile create mode 100644 libass/ass.c create mode 100644 libass/ass.h create mode 100644 libass/ass_cache.c create mode 100644 libass/ass_cache.h create mode 100644 libass/ass_fontconfig.c create mode 100644 libass/ass_fontconfig.h create mode 100644 libass/ass_mp.c create mode 100644 libass/ass_mp.h create mode 100644 libass/ass_render.c create mode 100644 libass/ass_types.h create mode 100644 libass/ass_utils.c create mode 100644 libass/ass_utils.h (limited to 'libass') diff --git a/libass/Makefile b/libass/Makefile new file mode 100644 index 0000000000..0604dd32e2 --- /dev/null +++ b/libass/Makefile @@ -0,0 +1,50 @@ + +include ../config.mak + +LIBNAME=libass.a + +LIBS=$(LIBNAME) + +SRCS=ass.c ass_cache.c ass_fontconfig.c ass_render.c ass_utils.c ass_mp.c + +OBJS=$(SRCS:.c=.o) + +CFLAGS = $(OPTFLAGS) \ + -I. -I.. \ + -I../libmpcodecs \ + $(EXTRA_INC) \ + -D_GNU_SOURCE \ + $(FREETYPE_INC) \ + $(FONTCONFIG_INC) \ + +.SUFFIXES: .c .o + +# .PHONY: all clean + +.c.o: + $(CC) -c $(CFLAGS) -o $@ $< + +all: $(LIBS) + +$(LIBNAME): $(OBJS) + $(AR) r $(LIBNAME) $(OBJS) + $(RANLIB) $(LIBNAME) + +clean: + rm -f *.o *.a *~ + +distclean: clean + rm -f .depend + +dep: depend + +depend: + $(CC) -MM $(CFLAGS) $(SRCS) 1>.depend + +# +# include dependency files if they exist +# +ifneq ($(wildcard .depend),) +include .depend +endif + diff --git a/libass/ass.c b/libass/ass.c new file mode 100644 index 0000000000..4f9b8908ec --- /dev/null +++ b/libass/ass.c @@ -0,0 +1,857 @@ +#include "config.h" + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#ifdef HAVE_ENCA +#include "subreader.h" // for guess_buffer_cp +#endif + +#ifdef USE_ICONV +#include +extern char *sub_cp; +#endif + +#include "mp_msg.h" +#include "ass.h" +#include "ass_utils.h" +#include "libvo/sub.h" // for utf8_get_char + +char *get_path(char *); + +#define ASS_STYLES_ALLOC 20 +#define ASS_EVENTS_ALLOC 200 + +void ass_free_track(ass_track_t* track) { + int i; + + if (track->style_format) + free(track->style_format); + if (track->event_format) + free(track->event_format); + if (track->styles) { + for (i = 0; i < track->n_styles; ++i) { + ass_style_t* style = track->styles + i; + if (style->Name) + free(style->Name); + if (style->FontName) + free(style->FontName); + } + free(track->styles); + } + if (track->events) { + for (i = 0; i < track->n_events; ++i) { + ass_event_t* event = track->events + i; + if (event->Name) + free(event->Name); + if (event->Effect) + free(event->Effect); + if (event->Text) + free(event->Text); + } + free(track->events); + } +} + +/// \brief Allocate a new style struct +/// \param track track +/// \return style id +int ass_alloc_style(ass_track_t* track) { + int sid; + + assert(track->n_styles <= track->max_styles); + + if (track->n_styles == track->max_styles) { + track->max_styles += ASS_STYLES_ALLOC; + track->styles = (ass_style_t*)realloc(track->styles, sizeof(ass_style_t)*track->max_styles); + } + + sid = track->n_styles++; + memset(track->styles + sid, 0, sizeof(ass_style_t)); + return sid; +} + +/// \brief Allocate a new event struct +/// \param track track +/// \return event id +int ass_alloc_event(ass_track_t* track) { + int eid; + + assert(track->n_events <= track->max_events); + + if (track->n_events == track->max_events) { + track->max_events += ASS_EVENTS_ALLOC; + track->events = (ass_event_t*)realloc(track->events, sizeof(ass_event_t)*track->max_events); + } + + eid = track->n_events++; + memset(track->events + eid, 0, sizeof(ass_event_t)); + return eid; +} + +static void free_event(ass_track_t* track, int eid) { + if (track->n_events > eid + 1) // not last event + memcpy(track->events + eid, track->events + eid + 1, sizeof(ass_event_t) * (track->n_events - eid - 1)); + track->n_events--; +} + +static int events_compare_f(const void* a_, const void* b_) { + ass_event_t* a = (ass_event_t*)a_; + ass_event_t* b = (ass_event_t*)b_; + if (a->Start < b->Start) + return -1; + else if (a->Start > b->Start) + return 1; + else + return 0; +} + +/// \brief Sort events by start time +/// \param tid track id +static void sort_events(ass_track_t* track) { + qsort(track->events, track->n_events, sizeof(ass_event_t), events_compare_f); +} + +// ============================================================================================== + +static void skip_spaces(char** str) { + char* p = *str; + while ((*p==' ') || (*p=='\t')) + ++p; + *str = p; +} + +static void rskip_spaces(char** str, char* limit) { + char* p = *str; + while ((p >= limit) && ((*p==' ') || (*p=='\t'))) + --p; + *str = p; +} + +/** + * \brief find style by name + * \param track track + * \param name style name + * \return index in track->styles + * Returnes 0 if no styles found => expects at least 1 style. + * Parsing code always adds "Default" style in the end. + */ +static int lookup_style(ass_track_t* track, char* name) { + int i; + for (i=0; in_styles; ++i) { + // FIXME: mb strcasecmp ? + if (strcmp(track->styles[i].Name, name) == 0) + return i; + } + i = track->default_style; + mp_msg(MSGT_GLOBAL, MSGL_WARN, "[%p] Warning: no style named '%s' found, using '%s'\n", track, name, track->styles[i].Name); + return i; // use the first style +} + +static uint32_t string2color(char* p) { + uint32_t tmp; + (void)strtocolor(&p, &tmp); + return tmp; +} + +static long long string2timecode(char* p) { + unsigned h, m, s, ms; + long long tm; + int res = sscanf(p, "%1d:%2d:%2d.%2d", &h, &m, &s, &ms); + if (res < 4) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "bad timestamp\n"); + return 0; + } + tm = ((h * 60 + m) * 60 + s) * 1000 + ms * 10; + return tm; +} + +/** + * \brief converts numpad-style align to align. + */ +static int numpad2align(int val) { + int res, v; + v = (val - 1) / 3; // 0, 1 or 2 for vertical alignment + if (v != 0) v = 3 - v; + res = ((val - 1) % 3) + 1; // horizontal alignment + res += v*4; + return res; +} + +#define NEXT(str,token) \ + token = next_token(&str); \ + if (!token) break; + +#define ANYVAL(name,func) \ + } else if (strcasecmp(tname, #name) == 0) { \ + target->name = func(token); \ + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "%s = %s\n", #name, token); +#define STRVAL(name) ANYVAL(name,strdup) +#define COLORVAL(name) ANYVAL(name,string2color) +#define INTVAL(name) ANYVAL(name,atoi) +#define FPVAL(name) ANYVAL(name,atof) +#define TIMEVAL(name) ANYVAL(name,string2timecode) +#define STYLEVAL(name) \ + } else if (strcasecmp(tname, #name) == 0) { \ + target->name = lookup_style(track, token); \ + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "%s = %s\n", #name, token); + +#define ALIAS(alias,name) \ + if (strcasecmp(tname, #alias) == 0) {tname = #name;} + +static char* next_token(char** str) { + char* p = *str; + char* start; + skip_spaces(&p); + if (*p == '\0') { + *str = p; + return 0; + } + start = p; // start of the token + for (; (*p != '\0') && (*p != ','); ++p) {} + if (*p == '\0') { + *str = p; // eos found, str will point to '\0' at exit + } else { + *p = '\0'; + *str = p + 1; // ',' found, str will point to the next char (beginning of the next token) + } + --p; // end of current token + rskip_spaces(&p, start); + if (p < start) + p = start; // empty token + else + ++p; // the first space character, or '\0' + *p = '\0'; + return start; +} +/** + * \brief Parse the tail of Dialogue line + * \param track track + * \param event parsed data goes here + * \param str string to parse, zero-terminated + * \param n_ignored number of format options to skip at the beginning +*/ +static int process_event_tail(ass_track_t* track, ass_event_t* event, char* str, int n_ignored) +{ + char* token; + char* tname; + char* p = str; + int i; + ass_event_t* target = event; + + char* format = strdup(track->event_format); + char* q = format; // format scanning pointer + + for (i = 0; i < n_ignored; ++i) { + NEXT(q, tname); + } + + while (1) { + NEXT(q, tname); + if (strcasecmp(tname, "Text") == 0) { + char* last; + event->Text = strdup(p); + last = event->Text + strlen(event->Text) - 1; + if (*last == '\r') + *last = 0; + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "Text = %s\n", event->Text); + event->Duration -= event->Start; + free(format); + return 0; // "Text" is always the last + } + NEXT(p, token); + + ALIAS(End,Duration) // temporarily store end timecode in event->Duration + if (0) { // cool ;) + STYLEVAL(Style) + STRVAL(Name) + STRVAL(Effect) + INTVAL(MarginL) + INTVAL(MarginR) + INTVAL(MarginV) + TIMEVAL(Start) + TIMEVAL(Duration) + } + } + free(format); + return 1; +} + +/** + * \brief Parse the Style line + * \param track track + * \param str string to parse, zero-terminated + * Allocates a new style struct. +*/ +static int process_style(ass_track_t* track, char *str) +{ + + char* token; + char* tname; + char* p = str; + char* format; + char* q; // format scanning pointer + int sid; + ass_style_t* style; + ass_style_t* target; + + if (!track->style_format) { + // no style format header + // probably an ancient script version + if (track->track_type == TRACK_TYPE_SSA) + track->style_format = strdup("Name, Fontname, Fontsize, PrimaryColour, SecondaryColour," + "TertiaryColour, BackColour, Bold, Italic, BorderStyle, Outline," + "Shadow, Alignment, MarginL, MarginR, MarginV, AlphaLevel, Encoding"); + else + track->style_format = strdup("Name, Fontname, Fontsize, PrimaryColour, SecondaryColour," + "OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut," + "ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow," + "Alignment, MarginL, MarginR, MarginV, Encoding"); + } + + q = format = strdup(track->style_format); + + mp_msg(MSGT_GLOBAL, MSGL_V, "[%p] Style: %s\n", track, str); + + sid = ass_alloc_style(track); + + style = track->styles + sid; + target = style; +// fill style with some default values + style->ScaleX = 100.; + style->ScaleY = 100.; + + while (1) { + NEXT(q, tname); + NEXT(p, token); + +// ALIAS(TertiaryColour,OutlineColour) // ignore TertiaryColour; it appears only in SSA, and is overridden by BackColour + + if (0) { // cool ;) + STRVAL(Name) + if ((strcmp(target->Name, "Default")==0) || (strcmp(target->Name, "*Default")==0)) + track->default_style = sid; + STRVAL(FontName) + COLORVAL(PrimaryColour) + COLORVAL(SecondaryColour) + COLORVAL(OutlineColour) // TertiaryColor + COLORVAL(BackColour) + // SSA uses BackColour for both outline and shadow + // this will destroy SSA's TertiaryColour, but i'm not going to use it anyway + if (track->track_type == TRACK_TYPE_SSA) + target->OutlineColour = target->BackColour; + INTVAL(FontSize) + INTVAL(Bold) + INTVAL(Italic) + INTVAL(Underline) + INTVAL(StrikeOut) + INTVAL(Spacing) + INTVAL(Angle) + INTVAL(BorderStyle) + INTVAL(Alignment) + if (track->track_type == TRACK_TYPE_ASS) + target->Alignment = numpad2align(target->Alignment); + INTVAL(MarginL) + INTVAL(MarginR) + INTVAL(MarginV) + INTVAL(Encoding) + FPVAL(ScaleX) + FPVAL(ScaleY) + FPVAL(Outline) + FPVAL(Shadow) + } + } + style->ScaleX /= 100.; + style->ScaleY /= 100.; + if (!style->Name) + style->Name = strdup("Default"); + if (!style->FontName) + style->FontName = strdup("Arial"); + free(format); + return 0; + +} + +/** + * \brief Parse a header line + * \param track track + * \param str string to parse, zero-terminated +*/ +static int process_header_line(ass_track_t* track, char *str) +{ + static int events_section_started = 0; + + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "=== Header: %s\n", str); + if (strncmp(str, "PlayResX:", 9)==0) { + track->PlayResX = atoi(str + 9); + } else if (strncmp(str,"PlayResY:", 9)==0) { + track->PlayResY = atoi(str + 9); + } else if (strncmp(str,"Timer:", 6)==0) { + track->Timer = atof(str + 6); + } else if (strstr(str,"Styles]")) { + events_section_started = 0; + if (strchr(str, '+')) + track->track_type = TRACK_TYPE_ASS; + else + track->track_type = TRACK_TYPE_SSA; + } else if (strncmp(str,"[Events]", 8)==0) { + events_section_started = 1; + } else if (strncmp(str,"Format:", 7)==0) { + char* p = str + 7; + skip_spaces(&p); + if (events_section_started) { + track->event_format = strdup(p); + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "Event format: %s\n", track->event_format); + } else { + track->style_format = strdup(p); + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "Style format: %s\n", track->style_format); + } + } else if (strncmp(str,"Style:", 6)==0) { + char* p = str + 6; + skip_spaces(&p); + process_style(track, p); + } else if (strncmp(str,"WrapStyle:", 10)==0) { + track->WrapStyle = atoi(str + 10); + } + return 0; +} + +/** + * \brief Process CodecPrivate section of subtitle stream + * \param track track + * \param data string to parse + * \param size length of data + CodecPrivate section contains [Stream Info] and [V4+ Styles] sections +*/ +void ass_process_chunk(ass_track_t* track, char *data, int size) +{ + char* str = malloc(size + 1); + char* p; + int sid; + + memcpy(str, data, size); + str[size] = '\0'; + + p = str; + while(1) { + char* q; + for (;((*p=='\r')||(*p=='\n'));++p) {} + for (q=p; ((*q!='\0')&&(*q!='\r')&&(*q!='\n')); ++q) {}; + if (q==p) + break; + if (*q != '\0') + *(q++) = '\0'; + process_header_line(track, p); + if (*q == '\0') + break; + p = q; + } + free(str); + + // add "Default" style to the end + // will be used if track does not contain a default style (or even does not contain styles at all) + sid = ass_alloc_style(track); + track->styles[sid].Name = strdup("Default"); + track->styles[sid].FontName = strdup("Arial"); + + if (!track->event_format) { + // probably an mkv produced by ancient mkvtoolnix + // such files don't have [Events] and Format: headers + if (track->track_type == TRACK_TYPE_SSA) + track->event_format = strdup("Format: Marked, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text"); + else + track->event_format = strdup("Format: Layer, Start, End, Style, Actor, MarginL, MarginR, MarginV, Effect, Text"); + } +} + +static int check_duplicate_event(ass_track_t* track, int ReadOrder) +{ + int i; + for (i = 0; in_events - 1; ++i) // ignoring last event, it is the one we are comparing with + if (track->events[i].ReadOrder == ReadOrder) + return 1; + return 0; +} + +/** + * \brief Process a chunk of subtitle stream data. In matroska, this containes exactly 1 event (or a commentary) + * \param track track + * \param data string to parse + * \param size length of data + * \param timecode starting time of the event (milliseconds) + * \param duration duration of the event (milliseconds) +*/ +void ass_process_line(ass_track_t* track, char *data, int size, long long timecode, long long duration) +{ + char* str; + int eid; + char* p; + char* token; + ass_event_t* event; + + if (!track->event_format) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "Event format header missing\n"); + return; + } + + str = malloc(size + 1); + memcpy(str, data, size); + str[size] = '\0'; + mp_msg(MSGT_GLOBAL, MSGL_V, "\nline at timecode %lld, duration %lld: \n%s\n", timecode, duration, str); + + eid = ass_alloc_event(track); + event = track->events + eid; + + p = str; + + do { + NEXT(p, token); + event->ReadOrder = atoi(token); + if (check_duplicate_event(track, event->ReadOrder)) + break; + + NEXT(p, token); + event->Layer = atoi(token); + + process_event_tail(track, event, p, 3); + + event->Start = timecode; + event->Duration = duration; + + free(str); + return; +// dump_events(tid); + } while (0); + // some error + free_event(track, eid); + free(str); +} + +/** + * \brief Process a line from external file. + * \param track track + * \param str string to parse + * \param size length of data +*/ +static void ass_process_external_line(ass_track_t* track, char *str, int size) +{ + int eid; + ass_event_t* event; + + eid = ass_alloc_event(track); + event = track->events + eid; + + if (strncmp("Dialogue:", str, 9) != 0) + return; + + str += 9; + while (*str == ' ') {++str;} + + process_event_tail(track, event, str, 0); +} + +#ifdef USE_ICONV +/** \brief recode buffer to utf-8 + * constraint: sub_cp != 0 + * \param data pointer to text buffer + * \param size buffer size + * \return a pointer to recoded buffer, caller is responsible for freeing it +**/ +static char* sub_recode(char* data, size_t size) +{ + static iconv_t icdsc = (iconv_t)(-1); + char* tocp = "UTF-8"; + char* outbuf; + assert(sub_cp); + + { + char* cp_tmp = sub_cp; +#ifdef HAVE_ENCA + char enca_lang[3], enca_fallback[100]; + if (sscanf(sub_cp, "enca:%2s:%99s", enca_lang, enca_fallback) == 2 + || sscanf(sub_cp, "ENCA:%2s:%99s", enca_lang, enca_fallback) == 2) { + cp_tmp = guess_buffer_cp((unsigned char*)data, size, enca_lang, enca_fallback); + } +#endif + if ((icdsc = iconv_open (tocp, cp_tmp)) != (iconv_t)(-1)){ + mp_msg(MSGT_SUBREADER,MSGL_V,"LIBSUB: opened iconv descriptor.\n"); + } else + mp_msg(MSGT_SUBREADER,MSGL_ERR,"LIBSUB: error opening iconv descriptor.\n"); +#ifdef HAVE_ENCA + if (cp_tmp) free(cp_tmp); +#endif + } + + { + size_t osize = size; + size_t ileft = size; + size_t oleft = size - 1; + char* ip; + char* op; + size_t rc; + + outbuf = malloc(size); + ip = data; + op = outbuf; + + while (ileft) { + rc = iconv(icdsc, &ip, &ileft, &op, &oleft); + if (rc == (size_t)(-1)) { + if (errno == E2BIG) { + int offset = op - outbuf; + outbuf = (char*)realloc(outbuf, osize + size); + op = outbuf + offset; + osize += size; + oleft += size; + } else { + mp_msg(MSGT_SUBREADER, MSGL_WARN, "LIBSUB: error recoding file.\n"); + return NULL; + } + } + } + outbuf[osize - oleft - 1] = 0; + } + + if (icdsc != (iconv_t)(-1)) { + (void)iconv_close(icdsc); + icdsc = (iconv_t)(-1); + mp_msg(MSGT_SUBREADER,MSGL_V,"LIBSUB: closed iconv descriptor.\n"); + } + + return outbuf; +} +#endif // ICONV + +/** + * \brief Read subtitles from file. + * \param fname file name + * \return newly allocated track +*/ +ass_track_t* ass_read_file(char* fname) +{ + int res; + long sz; + long bytes_read; + char* buf; + char* p; + int events_reached; + ass_track_t* track; + + FILE* fp = fopen(fname, "rb"); + if (!fp) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "ass_read_file(%s): fopen failed\n", fname); + return 0; + } + res = fseek(fp, 0, SEEK_END); + if (res == -1) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "ass_read_file(%s): fseek failed\n", fname); + fclose(fp); + return 0; + } + + sz = ftell(fp); + rewind(fp); + + if (sz > 10*1024*1024) { + mp_msg(MSGT_GLOBAL, MSGL_INFO, "ass_read_file(%s): Refusing to load subtitles larger than 10M\n", fname); + fclose(fp); + return 0; + } + + mp_msg(MSGT_GLOBAL, MSGL_V, "file size: %ld\n", sz); + + buf = malloc(sz + 1); + assert(buf); + bytes_read = 0; + do { + res = fread(buf + bytes_read, 1, sz - bytes_read, fp); + if (res <= 0) { + mp_msg(MSGT_GLOBAL, MSGL_INFO, "Read failed, %d: %s\n", errno, strerror(errno)); + fclose(fp); + free(buf); + return 0; + } + bytes_read += res; + } while (sz - bytes_read > 0); + buf[sz] = '\0'; + fclose(fp); + +#ifdef USE_ICONV + if (sub_cp) { + char* tmpbuf = sub_recode(buf, sz); + free(buf); + if (!tmpbuf) + return 0; + buf = tmpbuf; + } +#endif + + track = ass_new_track(); + track->name = strdup(fname); + + // process header + events_reached = 0; + p = buf; + while (p && (*p)) { + while (*p == '\n') {++p;} + if (strncmp(p, "[Events]", 8) == 0) { + events_reached = 1; + } else if ((strncmp(p, "Format:", 7) == 0) && (events_reached)) { + p = strchr(p, '\n'); + if (p == 0) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "Incomplete subtitles\n"); + free(buf); + return 0; + } + ass_process_chunk(track, buf, p - buf + 1); + ++p; + break; + } + p = strchr(p, '\n'); + } + // process events + while (p && (*p)) { + char* next; + int len; + while (*p == '\n') {++p;} + next = strchr(p, '\n'); + len = 0; + if (next) { + len = next - p; + *next = 0; + } else { + len = strlen(p); + } + ass_process_external_line(track, p, len); + if (next) { + p = next + 1; + continue; + } else + break; + } + + free(buf); + + if (!events_reached) { + ass_free_track(track); + return 0; + } + + mp_msg(MSGT_GLOBAL, MSGL_INFO, "LIBASS: added subtitle file: %s (%d styles, %d events)\n", fname, track->n_styles, track->n_events); + + sort_events(track); + +// dump_events(forced_tid); + return track; +} + +static char* validate_fname(char* name) +{ + char* fname; + char* p; + char* q; + unsigned code; + int sz = strlen(name); + + q = fname = malloc(sz + 1); + p = name; + while (*p) { + code = utf8_get_char(&p); + if (code == 0) + break; + if ( (code > 0x7F) || + (code == '\\') || + (code == '/') || + (code == ':') || + (code == '*') || + (code == '?') || + (code == '<') || + (code == '>') || + (code == '|') || + (code == 0)) + { + *q++ = '_'; + } else { + *q++ = code; + } + if (p - name > sz) + break; + } + *q = 0; + return fname; +} + +/** + * \brief Process embedded matroska font. Saves it to ~/.mplayer/fonts. + * \param name attachment name + * \param data binary font data + * \param data_size data size +*/ +void ass_process_font(const char* name, char* data, int data_size) +{ + char buf[1000]; + FILE* fp = 0; + int rc; + struct stat st; + char* fname; + + char* fonts_dir = get_path("fonts"); + rc = stat(fonts_dir, &st); + if (rc) { + int res; +#ifndef __MINGW32__ + res = mkdir(fonts_dir, 0700); +#else + res = mkdir(fonts_dir); +#endif + if (res) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "Failed to create: %s\n", fonts_dir); + } + } else if (!S_ISDIR(st.st_mode)) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "Not a directory: %s\n", fonts_dir); + } + + fname = validate_fname((char*)name); + + snprintf(buf, 1000, "%s/%s", fonts_dir, fname); + free(fname); + free(fonts_dir); + + fp = fopen(buf, "wb"); + if (!fp) return; + + fwrite(data, data_size, 1, fp); + fclose(fp); +} + +long long ass_step_sub(ass_track_t* track, long long now, int movement) { + int i; + + if (movement == 0) return 0; + if (track->n_events == 0) return 0; + + if (movement < 0) + for (i = 0; (i < track->n_events) && ((long long)(track->events[i].Start + track->events[i].Duration) <= now); ++i) {} + else + for (i = track->n_events - 1; (i >= 0) && ((long long)(track->events[i].Start) > now); --i) {} + + // -1 and n_events are ok + assert(i >= -1); assert(i <= track->n_events); + i += movement; + if (i < 0) i = 0; + if (i >= track->n_events) i = track->n_events - 1; + return ((long long)track->events[i].Start) - now; +} + +ass_track_t* ass_new_track(void) { + ass_track_t* track = calloc(1, sizeof(ass_track_t)); + return track; +} + diff --git a/libass/ass.h b/libass/ass.h new file mode 100644 index 0000000000..6049fd184b --- /dev/null +++ b/libass/ass.h @@ -0,0 +1,156 @@ +#ifndef __ASS_H__ +#define __ASS_H__ + +#include "ass_types.h" + +/// Libass "library object". Contents are private. +typedef struct ass_instance_s ass_instance_t; + +/// used in ass_configure +typedef struct ass_settings_s { + int frame_width; + int frame_height; + double font_size_coeff; // font size multiplier + double line_spacing; // additional line spacing (in frame pixels) + int top_margin; // height of top margin. Everything except toptitles is shifted down by top_margin. + int bottom_margin; // height of bottom margin. (frame_height - top_margin - bottom_margin) is original video height. + double aspect; // frame aspect ratio, d_width / d_height. +} ass_settings_t; + +/// a linked list of images produced by ass renderer +typedef struct ass_image_s { + int w, h; // bitmap width/height + int stride; // bitmap stride + unsigned char* bitmap; // 1bpp stride*h alpha buffer + uint32_t color; // RGBA + int dst_x, dst_y; // bitmap placement inside the video frame + + struct ass_image_s* next; // linked list +} ass_image_t; + +/** + * \brief initialize the library + * \return library handle or NULL if failed + */ +ass_instance_t* ass_init(void); + +/** + * \brief finalize the library + * \param priv library handle + */ +void ass_done(ass_instance_t* priv); + +/** + * \brief configure the library + * \param priv library handle + * \param config struct with configuration parameters. Caller is free to reuse it after this function returns. + */ +void ass_configure(ass_instance_t* priv, const ass_settings_t* config); + +/** + * \brief start rendering a frame + * \param priv library + * \param track subtitle track + * \param now video timestamp in milliseconds + */ +int ass_start_frame(ass_instance_t *priv, ass_track_t* track, long long now); + +/** + * \brief render a single event + * uses library, track and timestamp from the previous call to ass_start_frame + */ +int ass_render_event(ass_event_t* event); + +/** + * \brief done rendering frame, give out the results + * \return a list of images for blending + */ +ass_image_t* ass_end_frame(void); // returns linked list of images to render + +/** + * \brief render a frame, producing a list of ass_image_t + * \param priv library + * \param track subtitle track + * \param now video timestamp in milliseconds + * This function is equivalent to + * ass_start_frame() + * for events: start <= now < end: + * ass_render_event() + * ass_end_frame() + */ +ass_image_t* ass_render_frame(ass_instance_t *priv, ass_track_t* track, long long now); + + +// The following functions operate on track objects and do not need an ass_instance // + +/** + * \brief allocate a new empty track object + * \return pointer to empty track + */ +ass_track_t* ass_new_track(void); + +/** + * \brief deallocate track and all its child objects (styles and events) + * \param track track to deallocate + */ +void ass_free_track(ass_track_t* track); + +/** + * \brief allocate new style + * \param track track + * \return newly allocated style id + */ +int ass_alloc_style(ass_track_t* track); + +/** + * \brief allocate new event + * \param track track + * \return newly allocated event id + */ +int ass_alloc_event(ass_track_t* track); + +/** + * \brief Process Codec Private section of subtitle stream + * \param track target track + * \param data string to parse + * \param size length of data + */ +void ass_process_chunk(ass_track_t* track, char *data, int size); + +/** + * \brief Process a chunk of subtitle stream data. In matroska, this containes exactly 1 event (or a commentary) + * \param track track + * \param data string to parse + * \param size length of data + * \param timecode starting time of the event (milliseconds) + * \param duration duration of the event (milliseconds) +*/ +void ass_process_line(ass_track_t* track, char *data, int size, long long timecode, long long duration); + +/** + * \brief Read subtitles from file. + * \param fname file name + * \return newly allocated track +*/ +ass_track_t* ass_read_file(char* fname); + +/** + * \brief Process embedded matroska font. Saves it to ~/.mplayer/fonts. + * \param name attachment name + * \param data binary font data + * \param data_size data size +*/ +void ass_process_font(const char* name, char* data, int data_size); + +/** + * \brief Calculates timeshift from now to the start of some other subtitle event, depending on movement parameter + * \param track subtitle track + * \param now current time, ms + * \param movement how many events to skip from the one currently displayed + * +2 means "the one after the next", -1 means "previous" + * \return timeshift, ms + */ +long long ass_step_sub(ass_track_t* track, long long now, int movement); + +#endif + diff --git a/libass/ass_cache.c b/libass/ass_cache.c new file mode 100644 index 0000000000..5d400ed6e7 --- /dev/null +++ b/libass/ass_cache.c @@ -0,0 +1,209 @@ +#include "config.h" + +#include +#include FT_FREETYPE_H + +#include + +#include "mp_msg.h" +#include "ass_fontconfig.h" +#include "ass_cache.h" + + +typedef struct face_cache_item_s { + face_desc_t desc; + char* path; + int index; + FT_Face face; +} face_cache_item_t; + +#define MAX_FACE_CACHE_SIZE 100 + +static face_cache_item_t* face_cache; +static int face_cache_size; + +extern int no_more_font_messages; + +static int font_compare(face_desc_t* a, face_desc_t* b) { + if (strcmp(a->family, b->family) != 0) + return 0; + if (a->bold != b->bold) + return 0; + if (a->italic != b->italic) + return 0; + return 1; +} + +/** + * \brief Get a face object, either from cache or created through FreeType+FontConfig. + * \param library FreeType library object + * \param fontconfig_priv fontconfig private data + * \param desc required face description + * \param face out: the face object +*/ +int ass_new_face(FT_Library library, void* fontconfig_priv, face_desc_t* desc, /*out*/ FT_Face* face) +{ + FT_Error error; + int i; + char* path; + int index; + face_cache_item_t* item; + + for (i=0; ifamily, desc->bold, desc->italic, &index); + + error = FT_New_Face(library, path, index, face); + if (error) { + if (!no_more_font_messages) + mp_msg(MSGT_GLOBAL, MSGL_WARN, "Error opening font: %s, %d\n", path, index); + no_more_font_messages = 1; + return 1; + } + + item = face_cache + face_cache_size; + item->path = strdup(path); + item->index = index; + item->face = *face; + memcpy(&(item->desc), desc, sizeof(face_desc_t)); + face_cache_size++; + return 0; +} + +void ass_face_cache_init(void) +{ + face_cache = calloc(MAX_FACE_CACHE_SIZE, sizeof(face_cache_item_t)); + face_cache_size = 0; +} + +void ass_face_cache_done(void) +{ + int i; + for (i = 0; i < face_cache_size; ++i) { + face_cache_item_t* item = face_cache + i; + if (item->face) FT_Done_Face(item->face); + if (item->path) free(item->path); + // FIXME: free desc ? + } + free(face_cache); + face_cache_size = 0; +} + +//--------------------------------- +// glyph cache + +#define GLYPH_HASH_SIZE (0xFFFF + 13) + +typedef struct glyph_hash_item_s { + glyph_hash_key_t key; + glyph_hash_val_t val; + struct glyph_hash_item_s* next; +} glyph_hash_item_t; + +typedef glyph_hash_item_t* glyph_hash_item_p; + +static glyph_hash_item_p* glyph_hash_root; +static int glyph_hash_size; + +static int glyph_compare(glyph_hash_key_t* a, glyph_hash_key_t* b) { + if (memcmp(a, b, sizeof(glyph_hash_key_t)) == 0) + return 1; + else + return 0; +} + +static unsigned glyph_hash(glyph_hash_key_t* key) { + unsigned val = 0; + unsigned i; + for (i = 0; i < sizeof(key->face); ++i) + val += *(unsigned char *)(&(key->face) + i); + val <<= 21; + + if (key->bitmap) val &= 0x80000000; + val += key->index; + val += key->size << 8; + val += key->outline << 3; + val += key->advance.x << 10; + val += key->advance.y << 16; + val += key->bold << 1; + val += key->italic << 20; + return val; +} + +/** + * \brief Add a glyph to glyph cache. + * \param key hash key + * \param val hash val: 2 bitmap glyphs + some additional info +*/ +void cache_add_glyph(glyph_hash_key_t* key, glyph_hash_val_t* val) +{ + unsigned hash = glyph_hash(key); + glyph_hash_item_t** next = glyph_hash_root + (hash % GLYPH_HASH_SIZE); + while (*next) { + if (glyph_compare(key, &((*next)->key))) + return; + next = &((*next)->next); + assert(next); + } + (*next) = malloc(sizeof(glyph_hash_item_t)); +// (*next)->desc = glyph_key_copy(key, &((*next)->key)); + memcpy(&((*next)->key), key, sizeof(glyph_hash_key_t)); + memcpy(&((*next)->val), val, sizeof(glyph_hash_val_t)); + (*next)->next = 0; + + glyph_hash_size ++; +/* if (glyph_hash_size && (glyph_hash_size % 25 == 0)) { + printf("\nGlyph cache: %d entries, %d bytes\n", glyph_hash_size, glyph_hash_size * sizeof(glyph_hash_item_t)); + } */ +} + +/** + * \brief Get a glyph from glyph cache. + * \param key hash key + * \return requested hash val or 0 if not found +*/ +glyph_hash_val_t* cache_find_glyph(glyph_hash_key_t* key) +{ + unsigned hash = glyph_hash(key); + glyph_hash_item_t* item = glyph_hash_root[hash % GLYPH_HASH_SIZE]; + while (item) { + if (glyph_compare(key, &(item->key))) { + return &(item->val); + } + item = item->next; + } + return 0; +} + +void ass_glyph_cache_init(void) +{ + glyph_hash_root = calloc(GLYPH_HASH_SIZE, sizeof(glyph_hash_item_p)); + glyph_hash_size = 0; +} + +void ass_glyph_cache_done(void) +{ + int i; + for (i = 0; i < GLYPH_HASH_SIZE; ++i) { + glyph_hash_item_t* item = glyph_hash_root[i]; + while (item) { + glyph_hash_item_t* next = item->next; + if (item->val.glyph) FT_Done_Glyph(item->val.glyph); + if (item->val.outline_glyph) FT_Done_Glyph(item->val.outline_glyph); + free(item); + item = next; + } + } + free(glyph_hash_root); + glyph_hash_size = 0; +} + diff --git a/libass/ass_cache.h b/libass/ass_cache.h new file mode 100644 index 0000000000..07038b5e47 --- /dev/null +++ b/libass/ass_cache.h @@ -0,0 +1,50 @@ +#ifndef __ASS_CACHE_H__ +#define __ASS_CACHE_H__ + +#include +#include FT_FREETYPE_H +#include FT_STROKER_H +#include FT_GLYPH_H + +// font cache +typedef struct face_desc_s { + char* family; + unsigned bold; + unsigned italic; +} face_desc_t; + +void ass_face_cache_init(void); +int ass_new_face(FT_Library library, void* fontconfig_priv, face_desc_t* desc, /*out*/ FT_Face* face); +void ass_face_cache_done(void); + + +// describes a glyph; glyphs with equivalents structs are considered identical +typedef struct glyph_hash_key_s { + char bitmap; // bool : true = bitmap, false = outline + FT_Face face; + int size; // font size + int index; // glyph index in the face + unsigned outline; // border width, 16.16 fixed point value + int bold, italic; + + // the following affects bitmap glyphs only + unsigned scale_x, scale_y; // 16.16 + int angle; // signed 16.16 + + FT_Vector advance; // subpixel shift vector +} glyph_hash_key_t; + +typedef struct glyph_hash_val_s { + FT_Glyph glyph; // the actual glyphs + FT_Glyph outline_glyph; + FT_BBox bbox_scaled; // bbox after scaling, but before rotation + FT_Vector advance; // 26.6, advance distance to the next glyph in line +} glyph_hash_val_t; + +void ass_glyph_cache_init(void); +void cache_add_glyph(glyph_hash_key_t* key, glyph_hash_val_t* val); +glyph_hash_val_t* cache_find_glyph(glyph_hash_key_t* key); +void ass_glyph_cache_done(void); + +#endif + diff --git a/libass/ass_fontconfig.c b/libass/ass_fontconfig.c new file mode 100644 index 0000000000..e75bcbc058 --- /dev/null +++ b/libass/ass_fontconfig.c @@ -0,0 +1,192 @@ +#include "config.h" + +#include +#include +#include +#include +#include +#include + +#include "mp_msg.h" +#include "ass_fontconfig.h" + +#ifdef HAVE_FONTCONFIG +#include +#endif + +struct fc_instance_s { +#ifdef HAVE_FONTCONFIG + FcConfig* config; +#endif + char* family_default; + char* path_default; + int index_default; +}; + +extern int no_more_font_messages; + +#ifdef HAVE_FONTCONFIG +/** + * \brief Low-level font selection. + * \param priv private data + * \param family font family + * \param bold font weight value + * \param italic font slant value + * \param index out: font index inside a file + * \return font file path +*/ +static char* _select_font(fc_instance_t* priv, const char* family, unsigned bold, unsigned italic, int* index) +{ + FcBool rc; + FcResult result; + FcPattern *pat, *rpat; + int val_i; + FcChar8* val_s; + char buf[2000]; + + *index = 0; + + snprintf(buf, 2000, "%s:outline=True:slant=%u:weight=%u", family, italic, bold); + + pat = FcNameParse((const FcChar8*)buf); + if (!pat) + return 0; + + FcDefaultSubstitute(pat); + + rc = FcConfigSubstitute(priv->config, pat, FcMatchPattern); + if (!rc) + return 0; + + rpat = FcFontMatch(priv->config, pat, &result); + if (!rpat) + return 0; + + result = FcPatternGetInteger(rpat, FC_INDEX, 0, &val_i); + if (result != FcResultMatch) + return 0; + *index = val_i; + + result = FcPatternGetString(rpat, FC_FAMILY, 0, &val_s); + if (result != FcResultMatch) + return 0; + + if (strcasecmp((const char*)val_s, family) != 0) + mp_msg(MSGT_GLOBAL, MSGL_WARN, "fontconfig: selected font family is not the requested one: '%s' != '%s'\n", + (const char*)val_s, family); + + result = FcPatternGetString(rpat, FC_FILE, 0, &val_s); + if (result != FcResultMatch) + return 0; + + return strdup((const char*)val_s); +} + +/** + * \brief Find a font. Use default family or path if necessary. + * \param priv_ private data + * \param family font family + * \param bold font weight value + * \param italic font slant value + * \param index out: font index inside a file + * \return font file path +*/ +char* fontconfig_select(fc_instance_t* priv, const char* family, unsigned bold, unsigned italic, int* index) +{ + char* res = 0; + if (family && *family) + res = _select_font(priv, family, bold, italic, index); + if (!res && priv->family_default) { + res = _select_font(priv, priv->family_default, bold, italic, index); + if (res && !no_more_font_messages) + mp_msg(MSGT_GLOBAL, MSGL_WARN, "fontconfig_select: using default font family: (%s, %d, %d) -> %s, %d\n", + family, bold, italic, res, *index); + } + if (!res && priv->path_default) { + res = priv->path_default; + *index = priv->index_default; + if (!no_more_font_messages) + mp_msg(MSGT_GLOBAL, MSGL_WARN, "fontconfig_select: using default font: (%s, %d, %d) -> %s, %d\n", + family, bold, italic, res, *index); + } + if (!res) { + res = _select_font(priv, "Arial", bold, italic, index); + if (res && !no_more_font_messages) + mp_msg(MSGT_GLOBAL, MSGL_WARN, "fontconfig_select: using 'Arial' font family: (%s, %d, %d) -> %s, %d\n", + family, bold, italic, res, *index); + } + if (res) + mp_msg(MSGT_GLOBAL, MSGL_V, "fontconfig_select: (%s, %d, %d) -> %s, %d\n", + family, bold, italic, res, *index); + return res; +} + +/** + * \brief Init fontconfig. + * \param dir additional directoryu for fonts + * \param family default font family + * \param path default font path + * \return pointer to fontconfig private data +*/ +fc_instance_t* fontconfig_init(const char* dir, const char* family, const char* path) +{ + int rc; + struct stat st; + fc_instance_t* priv = calloc(1, sizeof(fc_instance_t)); + + rc = FcInit(); + assert(rc); + + priv->config = FcConfigGetCurrent(); + if (!priv->config) { + mp_msg(MSGT_GLOBAL, MSGL_FATAL, "FcInitLoadConfigAndFonts failed\n"); + return 0; + } + + rc = FcConfigAppFontAddDir(priv->config, (const FcChar8*)dir); + if (!rc) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "FcConfigAppFontAddDir failed\n"); + } + + priv->family_default = family ? strdup(family) : 0; + priv->index_default = 0; + + rc = stat(path, &st); + if (!rc && S_ISREG(st.st_mode)) + priv->path_default = path ? strdup(path) : 0; + else + priv->path_default = 0; + + return priv; +} + +#else + +char* fontconfig_select(fc_instance_t* priv, const char* family, unsigned bold, unsigned italic, int* index) +{ + *index = priv->index_default; + return priv->path_default; +} + +fc_instance_t* fontconfig_init(const char* dir, const char* family, const char* path) +{ + mp_msg(MSGT_GLOBAL, MSGL_WARN, "Fontconfig disabled, only default font will be used\n"); + + fc_instance_t* priv = calloc(1, sizeof(fc_instance_t)); + + priv->path_default = strdup(path); + priv->index_default = 0; + return priv; +} + +#endif + +void fontconfig_done(fc_instance_t* priv) +{ + // don't call FcFini() here, library can still be used by some code + if (priv && priv->path_default) free(priv->path_default); + if (priv && priv->family_default) free(priv->family_default); + if (priv) free(priv); +} + + diff --git a/libass/ass_fontconfig.h b/libass/ass_fontconfig.h new file mode 100644 index 0000000000..d8d48c1742 --- /dev/null +++ b/libass/ass_fontconfig.h @@ -0,0 +1,11 @@ +#ifndef __ASS_FONTCONFIG_H__ +#define __ASS_FONTCONFIG_H__ + +typedef struct fc_instance_s fc_instance_t; + +fc_instance_t* fontconfig_init(const char* dir, const char* family, const char* path); +char* fontconfig_select(fc_instance_t* priv, const char* family, unsigned bold, unsigned italic, int* index); +void fontconfig_done(fc_instance_t* priv); + +#endif + diff --git a/libass/ass_mp.c b/libass/ass_mp.c new file mode 100644 index 0000000000..f00ecc10b3 --- /dev/null +++ b/libass/ass_mp.c @@ -0,0 +1,10 @@ +#include "ass_mp.h" + +// libass-related command line options +int ass_enabled = 0; +float ass_font_scale = 1.; +float ass_line_spacing = 0.; +int ass_top_margin = 0; +int ass_bottom_margin = 0; +int extract_embedded_fonts = 0; + diff --git a/libass/ass_mp.h b/libass/ass_mp.h new file mode 100644 index 0000000000..017d78a656 --- /dev/null +++ b/libass/ass_mp.h @@ -0,0 +1,12 @@ +#ifndef __ASS_OPTIONS_H__ +#define __ASS_OPTIONS_H__ + +extern int ass_enabled; +extern float ass_font_scale; +extern float ass_line_spacing; +extern int ass_top_margin; +extern int ass_bottom_margin; +extern int extract_embedded_fonts; + +#endif + diff --git a/libass/ass_render.c b/libass/ass_render.c new file mode 100644 index 0000000000..1723394830 --- /dev/null +++ b/libass/ass_render.c @@ -0,0 +1,1766 @@ +#include "config.h" + +#include +#include +#include +#include +#include FT_FREETYPE_H +#include FT_STROKER_H +#include FT_GLYPH_H +#include FT_SYNTHESIS_H + +#include "mp_msg.h" + +#include "ass.h" +#include "ass_cache.h" +#include "ass_utils.h" +#include "ass_fontconfig.h" + +#include "libvo/sub.h" // for utf8_get_char + +#define MAX_GLYPHS 1000 +#define MAX_LINES 100 + +char *get_path(char *); + +extern char *font_name; +#ifdef HAVE_FONTCONFIG +extern int font_fontconfig; +#else +static int font_fontconfig = 0; +#endif + +struct ass_instance_s { + FT_Library library; + fc_instance_t* fontconfig_priv; + ass_settings_t settings; + + ass_image_t* images_root; // rendering result is stored here + int n_images; + int max_images; +}; + +int no_more_font_messages = 0; // don't print font warnings + +typedef enum {EF_NONE = 0, EF_KARAOKE, EF_KARAOKE_KF, EF_KARAOKE_KO} effect_t; + +// describes a glyph +// glyph_info_t and text_info_t are used for text centering and word-wrapping operations +typedef struct glyph_info_s { + unsigned symbol; + FT_Glyph glyph; + FT_Glyph outline_glyph; + FT_BBox bbox; + FT_Vector pos; + char linebreak; // the first (leading) glyph of some line ? + uint32_t c1, c2, c3, c4; // colors + char bitmap; // bool + FT_Vector advance; // 26.6 + effect_t effect_type; + int effect_timing; + int asc, desc; // font max ascender and descender +// int height; + + glyph_hash_key_t hash_key; +} glyph_info_t; + +typedef struct line_info_s { + int asc, desc; +} line_info_t; + +typedef struct text_info_s { + glyph_info_t* glyphs; + int length; + line_info_t lines[MAX_LINES]; + int n_lines; + int height; +} text_info_t; + + +// Renderer state. +// Values like current font face, color, screen position, clipping and so on are stored here. +typedef struct render_context_s { + ass_event_t* event; + ass_style_t* style; + + FT_Face face; + char* font_path; + int font_size; + + FT_Stroker stroker; + int alignment; // alignment overrides go here; if zero, style value will be used + double rotation; + enum { EVENT_NORMAL, // "normal" top-, sub- or mid- title + EVENT_POSITIONED // happens after pos(,), margins are ignored + } evt_type; + int pos_x, pos_y; // position + int org_x, org_y; // origin + double scale_x, scale_y; + int hspacing; // distance between letters, in pixels + double border; // outline width + uint32_t c1, c2, c3, c4; // colors(Primary, Secondary, so on) in RGBA + int clip_x0, clip_y0, clip_x1, clip_y1; + char detect_collisions; + + effect_t effect_type; + int effect_timing; + + // face properties + char* family; + unsigned bold; + unsigned italic; + +} render_context_t; + +// frame-global data +typedef struct frame_context_s { + ass_instance_t* ass_priv; + int width, height; // screen dimensions + int orig_height; // frame height ( = screen height - margins ) + ass_track_t* track; + int add_bottom_margin; // additional margin, used to shift subtitles in case of collision + int add_top_margin; + long long time; // frame's timestamp, ms + double font_scale_x; // x scale applied to all glyphs to preserve text aspect ratio +} frame_context_t; + +static ass_instance_t* ass_instance; +static ass_settings_t* global_settings; +static text_info_t text_info; +static render_context_t render_context; +static frame_context_t frame_context; + +static void ass_lazy_track_init(void) +{ + ass_track_t* track = frame_context.track; + if (track->PlayResX && track->PlayResY) + return; + if (!track->PlayResX && !track->PlayResY) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "Neither PlayResX nor PlayResY defined. Assuming 384x288. \n"); + track->PlayResX = 384; + track->PlayResY = 288; + } else { + double orig_aspect = (global_settings->aspect * frame_context.height) / frame_context.orig_height; + if (!track->PlayResY) { + track->PlayResY = track->PlayResX / orig_aspect + .5; + mp_msg(MSGT_GLOBAL, MSGL_WARN, "PlayResY undefined, setting %d \n", track->PlayResY); + } else if (!track->PlayResX) { + track->PlayResX = track->PlayResY * orig_aspect + .5; + mp_msg(MSGT_GLOBAL, MSGL_WARN, "PlayResX undefined, setting %d \n", track->PlayResX); + } + } +} + +ass_instance_t* ass_init(void) +{ + char* family = 0; + char* path = 0; + char* fonts_path = 0; + int error; + fc_instance_t* fc_priv; + FT_Library ft; + ass_instance_t* priv; + + if (font_fontconfig && font_name) + family = strdup(font_name); + + if (!font_fontconfig && font_name) + path = strdup(font_name); + else + path = get_path("subfont.ttf"); + + fonts_path = get_path("fonts"); + + fc_priv = fontconfig_init(fonts_path, family, path); + + free(fonts_path); + if (path) free(path); + if (family) free(family); + + if (!fc_priv) + return 0; + + error = FT_Init_FreeType( &ft ); + if ( error ) { + mp_msg(MSGT_GLOBAL, MSGL_FATAL, "FT_Init_FreeType failed\n"); + fontconfig_done(fc_priv); + return 0; + } + + priv = calloc(1, sizeof(ass_instance_t)); + if (!priv) { + FT_Done_FreeType(ft); + fontconfig_done(fc_priv); + return 0; + } + priv->library = ft; + priv->fontconfig_priv = fc_priv; + // images_root and related stuff is zero-filled in calloc + + ass_face_cache_init(); + ass_glyph_cache_init(); + + text_info.glyphs = calloc(MAX_GLYPHS, sizeof(glyph_info_t)); + + return priv; +} + +void ass_done(ass_instance_t* priv) +{ + ass_face_cache_done(); + ass_glyph_cache_done(); + if (priv && priv->library) FT_Done_FreeType(priv->library); + if (priv && priv->fontconfig_priv) fontconfig_done(priv->fontconfig_priv); + if (priv) free(priv); + if (text_info.glyphs) free(text_info.glyphs); +} + +/** + * \brief Create a new ass_image_t and append it to images_root + * Parameters are the same as ass_image_t fields. + */ +static void my_draw_bitmap(unsigned char* bitmap, int bitmap_w, int bitmap_h, int stride, int dst_x, int dst_y, uint32_t color) +{ + ass_instance_t* priv = ass_instance; + ass_image_t* img; + + assert(priv->n_images <= priv->max_images); + if (priv->n_images == priv->max_images) { + if (!priv->max_images) priv->max_images = 100; + else priv->max_images *= 2; + priv->images_root = (ass_image_t*)realloc(priv->images_root, priv->max_images * sizeof(ass_image_t)); + } + assert(priv->images_root); + img = priv->images_root + priv->n_images; + + img->w = bitmap_w; + img->h = bitmap_h; + img->stride = stride; + img->bitmap = bitmap; + img->color = color; + img->dst_x = dst_x; + img->dst_y = dst_y; + + priv->n_images++; +} + +/** + * \brief convert bitmap glyph into ass_image_t struct(s) + * \param bit freetype bitmap glyph, FT_PIXEL_MODE_GRAY + * \param dst_x bitmap x coordinate in video frame + * \param dst_y bitmap y coordinate in video frame + * \param color first color, RGBA + * \param color2 second color, RGBA + * \param brk x coordinate relative to glyph origin, color is used to the left of brk, color2 - to the right + * Performs clipping. Uses my_draw_bitmap for actual bitmap convertion. + */ +static int render_glyph(FT_BitmapGlyph bit, int dst_x, int dst_y, uint32_t color, uint32_t color2, int brk) +{ + // brk is relative to dst_x + // color = color left of brk + // color2 = color right of brk + int b_x0, b_y0, b_x1, b_y1; // visible part of the bitmap + int clip_x0, clip_y0, clip_x1, clip_y1; + int tmp; + FT_Bitmap* bitmap; + + bitmap = &(bit->bitmap); + dst_x += bit->left; + dst_y -= bit->top; + brk -= bit->left; + + if (bitmap->pixel_mode != FT_PIXEL_MODE_GRAY) { + mp_msg(MSGT_GLOBAL, MSGL_WARN, "Unsupported pixel mode: %d\n", (int)(bitmap->pixel_mode)); + return 1; + } + + // clipping + clip_x0 = render_context.clip_x0; + clip_y0 = render_context.clip_y0; + clip_x1 = render_context.clip_x1; + clip_y1 = render_context.clip_y1; + b_x0 = 0; + b_y0 = 0; + b_x1 = bitmap->width; + b_y1 = bitmap->rows; + + tmp = dst_x - clip_x0; + if (tmp < 0) { + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "clip left\n"); + b_x0 = - tmp; + } + tmp = dst_y - clip_y0; + if (tmp < 0) { + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "clip top\n"); + b_y0 = - tmp; + } + tmp = clip_x1 - dst_x - bitmap->width; + if (tmp < 0) { + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "clip right\n"); + b_x1 = bitmap->width + tmp; + } + tmp = clip_y1 - dst_y - bitmap->rows; + if (tmp < 0) { + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "clip bottom\n"); + b_y1 = bitmap->rows + tmp; + } + + if ((b_y0 >= b_y1) || (b_x0 >= b_x1)) + return 0; + + if (brk > b_x0) { // draw left part + if (brk > b_x1) brk = b_x1; + my_draw_bitmap(bitmap->buffer + bitmap->pitch * b_y0 + b_x0, + brk - b_x0, b_y1 - b_y0, bitmap->pitch, + dst_x + b_x0, dst_y + b_y0, color); + } + if (brk < b_x1) { // draw right part + if (brk < b_x0) brk = b_x0; + my_draw_bitmap(bitmap->buffer + bitmap->pitch * b_y0 + brk, + b_x1 - brk, b_y1 - b_y0, bitmap->pitch, + dst_x + brk, dst_y + b_y0, color2); + + } + return 0; +} + +/** + * \brief Render text_info_t struct into ass_images_t list + * Rasterize glyphs and put them in glyph cache. + */ +static int render_text(text_info_t* text_info, int dst_x, int dst_y) +{ + int pen_x, pen_y; + int error, error2; + int i; + FT_Glyph image; + FT_BitmapGlyph bit; + glyph_hash_val_t hash_val; + + for (i = 0; i < text_info->length; ++i) { + if (text_info->glyphs[i].bitmap != 1) { + if ((text_info->glyphs[i].symbol == '\n') || (text_info->glyphs[i].symbol == 0)) + continue; + error = FT_Glyph_To_Bitmap( &(text_info->glyphs[i].outline_glyph), FT_RENDER_MODE_NORMAL, 0, 1); + error2 = FT_Glyph_To_Bitmap( &(text_info->glyphs[i].glyph), FT_RENDER_MODE_NORMAL, 0, 1); + + if (error || error2) { + FT_Done_Glyph(text_info->glyphs[i].outline_glyph); + FT_Done_Glyph(text_info->glyphs[i].glyph); + mp_msg(MSGT_GLOBAL, MSGL_WARN, "FT_Glyph_To_Bitmap error %d %d, symbol %d, index %d\n", + error, error2, text_info->glyphs[i].symbol, text_info->glyphs[i].hash_key.index); + text_info->glyphs[i].symbol = 0; // do not render + continue; + } + // cache + text_info->glyphs[i].hash_key.bitmap = 1; // other hash_key fields were set in get_glyph() + hash_val.bbox_scaled = text_info->glyphs[i].bbox; + hash_val.outline_glyph = text_info->glyphs[i].outline_glyph; + hash_val.glyph = text_info->glyphs[i].glyph; + hash_val.advance.x = text_info->glyphs[i].advance.x; + hash_val.advance.y = text_info->glyphs[i].advance.y; + cache_add_glyph(&(text_info->glyphs[i].hash_key), &hash_val); + } + } + + for (i = 0; i < text_info->length; ++i) { + glyph_info_t* info = text_info->glyphs + i; + if ((info->symbol == 0) || (info->symbol == '\n')) + continue; + + pen_x = dst_x + info->pos.x; + pen_y = dst_y + info->pos.y; + image = info->outline_glyph; + bit = (FT_BitmapGlyph)image; + + if ((info->effect_type == EF_KARAOKE_KO) && (info->effect_timing <= info->bbox.xMax)) { + // do nothing + } else + render_glyph(bit, pen_x, pen_y, info->c3, 0, 1000000); + } + for (i = 0; i < text_info->length; ++i) { + glyph_info_t* info = text_info->glyphs + i; + if ((info->symbol == 0) || (info->symbol == '\n')) + continue; + + pen_x = dst_x + info->pos.x; + pen_y = dst_y + info->pos.y; + image = info->glyph; + bit = (FT_BitmapGlyph)image; + + if ((info->effect_type == EF_KARAOKE) || (info->effect_type == EF_KARAOKE_KO)) { + if (info->effect_timing > info->bbox.xMax) + render_glyph(bit, pen_x, pen_y, info->c1, 0, 1000000); + else + render_glyph(bit, pen_x, pen_y, info->c2, 0, 1000000); + } else if (info->effect_type == EF_KARAOKE_KF) { + render_glyph(bit, pen_x, pen_y, info->c1, info->c2, info->effect_timing); + } else + render_glyph(bit, pen_x, pen_y, info->c1, 0, 1000000); + } + + return 0; +} + +/** + * \brief Mapping between script and screen coordinates + */ +static int x2scr(int x) { + return x*frame_context.width / frame_context.track->PlayResX; +} +/** + * \brief Mapping between script and screen coordinates + */ +static int y2scr(int y) { + return y * frame_context.orig_height / frame_context.track->PlayResY + global_settings->top_margin; +} +// the same for toptitles +static int y2scr_top(int y) { + return y * frame_context.orig_height / frame_context.track->PlayResY; +} +// the same for subtitles +static int y2scr_sub(int y) { + return y * frame_context.orig_height / frame_context.track->PlayResY + global_settings->top_margin + global_settings->bottom_margin; +} + +static void vmirror_bbox(FT_BBox* orig, FT_BBox* pbbox) { + pbbox->xMin = orig->xMin; + pbbox->xMax = orig->xMax; + pbbox->yMin = - orig->yMax; + pbbox->yMax = - orig->yMin; +} + +static void compute_string_bbox( text_info_t* info, FT_BBox *abbox ) { + FT_BBox bbox; + int n; + + /* initialize string bbox to "empty" values */ + bbox.xMin = bbox.yMin = 32000; + bbox.xMax = bbox.yMax = -32000; + + /* for each glyph image, compute its bounding box, */ + /* translate it, and grow the string bbox */ + for ( n = 0; n < info->length; n++ ) { + FT_BBox glyph_bbox; + vmirror_bbox( &(info->glyphs[n].bbox), &glyph_bbox ); + glyph_bbox.xMin += info->glyphs[n].pos.x; + glyph_bbox.xMax += info->glyphs[n].pos.x; + glyph_bbox.yMin += info->glyphs[n].pos.y; + glyph_bbox.yMax += info->glyphs[n].pos.y; + if ( glyph_bbox.xMin < bbox.xMin ) bbox.xMin = glyph_bbox.xMin; + if ( glyph_bbox.yMin < bbox.yMin ) bbox.yMin = glyph_bbox.yMin; + if ( glyph_bbox.xMax > bbox.xMax ) bbox.xMax = glyph_bbox.xMax; + if ( glyph_bbox.yMax > bbox.yMax ) bbox.yMax = glyph_bbox.yMax; + } + + /* check that we really grew the string bbox */ + if ( bbox.xMin > bbox.xMax ) { + bbox.xMin = 0; + bbox.yMin = 0; + bbox.xMax = 0; + bbox.yMax = 0; + } + + /* return string bbox */ + *abbox = bbox; +} + + +/** + * \brief Check if starting part of (*p) matches sample. If true, shift p to the first symbol after the matching part. + */ +static inline int mystrcmp(char** p, char* sample) { + int len = strlen(sample); + if (strncmp(*p, sample, len) == 0) { + (*p) += len; + return 1; + } else + return 0; +} + +static void change_font_size(int sz) +{ + double size = (double)sz * global_settings->font_size_coeff * 0.8; + size *= frame_context.height; + size /= frame_context.track->PlayResY; + + if (size < 1) + size = 1; + else if (size > frame_context.height * 2) + size = frame_context.height * 2; + + FT_Set_Pixel_Sizes(render_context.face, 0, size); + + render_context.font_size = sz; +} + +/** + * \brief Change current font, using setting from render_context. + */ +static void update_font(void) +{ + int error; + unsigned val; + ass_instance_t* priv = frame_context.ass_priv; + face_desc_t desc; + desc.family = strdup(render_context.family); + + val = render_context.bold; + // 0 = normal, 1 = bold, >1 = exact weight + if (val == 0) val = 80; // normal + else if (val == 1) val = 200; // bold + desc.bold = val; + + val = render_context.italic; + if (val == 0) val = 0; // normal + else if (val == 1) val = 110; //italic + desc.italic = val; + + error = ass_new_face(priv->library, priv->fontconfig_priv, &desc, &(render_context.face)); + if (error) { + render_context.face = 0; + } + + if (render_context.face) + { + change_font_size(render_context.font_size); + } +} + +/** + * \brief Change border width + */ +static void change_border(double border) +{ + if (!render_context.stroker) { + if (!no_more_font_messages) + mp_msg(MSGT_GLOBAL, MSGL_WARN, "No stroker!\n"); + } else { + render_context.border = border; + FT_Stroker_Set( render_context.stroker, + (int)(64 * border), + FT_STROKER_LINECAP_ROUND, + FT_STROKER_LINEJOIN_ROUND, + 0 ); + } +} + +#define _r(c) ((c)>>24) +#define _g(c) (((c)>>16)&0xFF) +#define _b(c) (((c)>>8)&0xFF) +#define _a(c) ((c)&0xFF) + +/** + * \brief Calculate a weighted average of two colors + * calculates c1*(1-a) + c2*a, but separately for each component except alpha + */ +static void change_color(uint32_t* var, uint32_t new, double pwr) +{ + (*var)= ((uint32_t)(_r(*var) * (1 - pwr) + _r(new) * pwr) << 24) + + ((uint32_t)(_g(*var) * (1 - pwr) + _g(new) * pwr) << 16) + + ((uint32_t)(_b(*var) * (1 - pwr) + _b(new) * pwr) << 8) + + _a(*var); +} + +// like change_color, but for alpha component only +static void change_alpha(uint32_t* var, uint32_t new, double pwr) +{ + *var = (_r(*var) << 24) + (_g(*var) << 16) + (_b(*var) << 8) + (_a(*var) * (1 - pwr) + _a(new) * pwr); +} + + +/** + * \brief Calculate alpha value by piecewise linear function + * Used for \fad, \fade implementation. + */ +static void interpolate_alpha(long long now, + long long t1, long long t2, long long t3, long long t4, + unsigned a1, unsigned a2, unsigned a3) +{ + unsigned a; + double cf; + if (now <= t1) { + a = a1; + } else if (now >= t4) { + a = a3; + } else if (now < t2) { // and > t1 + cf = ((double)(now - t1)) / (t2 - t1); + a = a1 * (1 - cf) + a2 * cf; + } else if (now > t3) { + cf = ((double)(now - t3)) / (t4 - t3); + a = a2 * (1 - cf) + a3 * cf; + } else { // t2 <= now <= t3 + a = a2; + } + + + change_alpha(&render_context.c1, a, 1.); + change_alpha(&render_context.c2, a, 1.); + change_alpha(&render_context.c3, a, 1.); + change_alpha(&render_context.c4, a, 1.); +} + +/** + * \brief Parse style override tag. + * \param p string to parse + * \param pwr multiplier for some tag effects (comes from \t tags) + */ +static char* parse_tag(char* p, double pwr) { +#define skip_all(x) if (*p == (x)) ++p; else { \ + while ((*p != (x)) && (*p != '}') && (*p != 0)) {++p;} } +#define skip(x) if (*p == (x)) ++p; else { return p; } + + skip_all('\\'); + if ((*p == '}') || (*p == 0)) + return p; + + if (mystrcmp(&p, "fsc")) { + char tp = *p++; + double val; + if (tp == 'x') { + if (mystrtod(&p, &val)) { + val /= 100; + render_context.scale_x = (val - 1.) * pwr + 1.; + } else + render_context.scale_x = render_context.style->ScaleX; + } else if (tp == 'y') { + if (mystrtod(&p, &val)) { + val /= 100; + render_context.scale_y = (val - 1.) * pwr + 1.; + } else + render_context.scale_y = render_context.style->ScaleY; + } + } else if (mystrcmp(&p, "fsp")) { + int val; + if (mystrtoi(&p, 10, &val)) + render_context.hspacing = val * pwr; + else + render_context.hspacing = 0; + } else if (mystrcmp(&p, "fs")) { + int val; + if (mystrtoi(&p, 10, &val)) + val = render_context.font_size * ( 1 - pwr ) + val * pwr; + else + val = render_context.style->FontSize; + if (render_context.face) + change_font_size(val); + } else if (mystrcmp(&p, "bord")) { + double val; + if (mystrtod(&p, &val)) + val = render_context.border * ( 1 - pwr ) + val * pwr; + else + val = (render_context.style->BorderStyle == 1) ? render_context.style->Outline : 1.; + change_border(val); + } else if (mystrcmp(&p, "move")) { + int x1, x2, y1, y2; + long long t1, delta_t, t; + int x, y; + double k; + skip('('); + x1 = strtol(p, &p, 10); + skip(','); + y1 = strtol(p, &p, 10); + skip(','); + x2 = strtol(p, &p, 10); + skip(','); + y2 = strtol(p, &p, 10); + skip(')'); // FIXME: 2 more optional args + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "movement: (%d, %d) -> (%d, %d)\n", x1, y1, x2, y2); + t1 = render_context.event->Start; + delta_t = render_context.event->Duration; + t = frame_context.time; // FIXME: move to render_context + k = ((double)(t - t1)) / delta_t; + x = k * (x2 - x1) + x1; + y = k * (y2 - y1) + y1; + render_context.pos_x = x; + render_context.pos_y = y; + render_context.detect_collisions = 0; + render_context.evt_type = EVENT_POSITIONED; + } else if (mystrcmp(&p, "frx") || mystrcmp(&p, "fry")) { + int val; + mystrtoi(&p, 10, &val); + mp_msg(MSGT_GLOBAL, MSGL_V, "frx/fry unimplemented \n"); + } else if (mystrcmp(&p, "frz") || mystrcmp(&p, "fr")) { + double angle; + int val; + val = strtol(p, &p, 10); + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "setting rotation to %.2f\n", val * pwr); + angle = M_PI * val / 180; + render_context.rotation = angle * pwr; + } else if (mystrcmp(&p, "fn")) { + char* start = p; + char* family; + skip_all('\\'); + family = malloc(p - start + 1); + strncpy(family, start, p - start); + family[p - start] = '\0'; + if (render_context.family) + free(render_context.family); + render_context.family = family; + update_font(); + } else if (mystrcmp(&p, "alpha")) { + uint32_t val; + if (strtocolor(&p, &val)) { + unsigned char a = val >> 24; + change_alpha(&render_context.c1, a, pwr); + change_alpha(&render_context.c2, a, pwr); + change_alpha(&render_context.c3, a, pwr); + change_alpha(&render_context.c4, a, pwr); + } else { + change_alpha(&render_context.c1, render_context.style->PrimaryColour, pwr); + change_alpha(&render_context.c2, render_context.style->SecondaryColour, pwr); + change_alpha(&render_context.c3, render_context.style->OutlineColour, pwr); + change_alpha(&render_context.c4, render_context.style->BackColour, pwr); + } + // FIXME: simplify + } else if (mystrcmp(&p, "an")) { + int val = strtol(p, &p, 10); + int v = (val - 1) / 3; // 0, 1 or 2 for vertical alignment + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "an %d\n", val); + if (v != 0) v = 3 - v; + val = ((val - 1) % 3) + 1; // horizontal alignment + val += v*4; + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "align %d\n", val); + render_context.alignment = val; + } else if (mystrcmp(&p, "a")) { + int val = strtol(p, &p, 10); + render_context.alignment = val; + } else if (mystrcmp(&p, "pos")) { + int v1, v2; + skip('('); + v1 = strtol(p, &p, 10); + skip(','); + v2 = strtol(p, &p, 10); + skip(')'); + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "pos(%d, %d)\n", v1, v2); + render_context.evt_type = EVENT_POSITIONED; + render_context.detect_collisions = 0; + render_context.pos_x = v1; + render_context.pos_y = v2; + } else if (mystrcmp(&p, "fade")) { + int a1, a2, a3, v1, v2, v3, v4; + skip('('); + a1 = strtol(p, &p, 10); + skip(','); + a2 = strtol(p, &p, 10); + skip(','); + a3 = strtol(p, &p, 10); + skip(','); + v1 = strtol(p, &p, 10); + skip(','); + v2 = strtol(p, &p, 10); + skip(','); + v3 = strtol(p, &p, 10); + skip(','); + v4 = strtol(p, &p, 10); + skip(')'); + interpolate_alpha(frame_context.time - render_context.event->Start, v1, v2, v3, v4, a1, a2, a3); + } else if (mystrcmp(&p, "fad")) { + int v1, v2; + long long now, t1, t2, t3, t4; + skip('('); + v1 = strtol(p, &p, 10); + skip(','); + v2 = strtol(p, &p, 10); + skip(')'); + now = frame_context.time; + t1 = render_context.event->Start; + t2 = t1 + v1; + t4 = render_context.event->Start + render_context.event->Duration; + t3 = t4 - v2; + interpolate_alpha(now, t1, t2, t3, t4, 0xFF, 0, 0xFF); + } else if (mystrcmp(&p, "org")) { + int v1, v2; + skip('('); + v1 = strtol(p, &p, 10); + skip(','); + v2 = strtol(p, &p, 10); + skip(')'); + mp_msg(MSGT_GLOBAL, MSGL_DBG2, "org(%d, %d)\n", v1, v2); + // render_context.evt_type = EVENT_POSITIONED; + render_context.org_x = v1; + render_c