From 0ea721f7ce81357bc4ec6eea609cd50482c3d15b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Diego=20=27Flameeyes=27=20Petten=C3=B2?= Date: Sat, 2 Dec 2006 01:19:48 +0000 Subject: Start working on a branch where FFmpeg is not copied, patched and carved to be built with automake but instead imported inline and built using its own build system. This is an import of a slightly modified FFmpeg current tree. xine-lib builds, install and run fine with it, but there are of course plenty of things that needs to be fixed before it can even be considered for a 1.2.x series. Work will continue in the next days of course. CVS patchset: 8397 CVS date: 2006/12/02 01:19:48 --- contrib/ffmpeg/vhook/drawtext.c | 531 ++++++++++++++++++++++++++++++++++++++++ 1 file changed, 531 insertions(+) create mode 100644 contrib/ffmpeg/vhook/drawtext.c (limited to 'contrib/ffmpeg/vhook/drawtext.c') diff --git a/contrib/ffmpeg/vhook/drawtext.c b/contrib/ffmpeg/vhook/drawtext.c new file mode 100644 index 000000000..081847620 --- /dev/null +++ b/contrib/ffmpeg/vhook/drawtext.c @@ -0,0 +1,531 @@ +/* + * drawtext.c: print text over the screen + ****************************************************************************** + * Options: + * -f font filename (MANDATORY!!!) + * -s font size in pixels [default 16] + * -b print background + * -o outline glyphs (use the bg color) + * -x x position ( >= 0) [default 0] + * -y y position ( >= 0) [default 0] + * -t text to print (will be passed to strftime()) + * MANDATORY: will be used even when -T is used. + * in this case, -t will be used if some error + * occurs + * -T file with the text (re-read every frame) + * -c <#RRGGBB> foreground color ('internet' way) [default #ffffff] + * -C <#RRGGBB> background color ('internet' way) [default #000000] + * + ****************************************************************************** + * Features: + * - True Type, Type1 and others via FreeType2 library + * - Font kerning (better output) + * - Line Wrap (if the text doesn't fit, the next char go to the next line) + * - Background box + * - Outline + ****************************************************************************** + * Author: Gustavo Sverzut Barbieri + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#define MAXSIZE_TEXT 1024 + +#include "framehook.h" + +#include +#include +#include +#include +#include +#include +#undef time +#include +#include + +#include +#include FT_FREETYPE_H +#include FT_GLYPH_H + +#define SCALEBITS 10 +#define ONE_HALF (1 << (SCALEBITS - 1)) +#define FIX(x) ((int) ((x) * (1<> SCALEBITS; \ + yuv_color[2] = ((FIX(0.50000) * rgb_color[0] - FIX(0.41869) * rgb_color[1] - FIX(0.08131) * rgb_color[2] + ONE_HALF - 1) >> SCALEBITS) + 128; \ + yuv_color[1] = ((- FIX(0.16874) * rgb_color[0] - FIX(0.33126) * rgb_color[1] + FIX(0.50000) * rgb_color[2] + ONE_HALF - 1) >> SCALEBITS) + 128; \ +} while (0) + +#define COPY_3(dst,src) { \ + dst[0]=src[0]; \ + dst[1]=src[1]; \ + dst[2]=src[2]; \ +} + + + +#define SET_PIXEL(picture, yuv_color, x, y) { \ + picture->data[0][ (x) + (y)*picture->linesize[0] ] = yuv_color[0]; \ + picture->data[1][ ((x/2) + (y/2)*picture->linesize[1]) ] = yuv_color[1]; \ + picture->data[2][ ((x/2) + (y/2)*picture->linesize[2]) ] = yuv_color[2]; \ +} + +#define GET_PIXEL(picture, yuv_color, x, y) { \ + yuv_color[0] = picture->data[0][ (x) + (y)*picture->linesize[0] ]; \ + yuv_color[1] = picture->data[1][ (x/2) + (y/2)*picture->linesize[1] ]; \ + yuv_color[2] = picture->data[2][ (x/2) + (y/2)*picture->linesize[2] ]; \ +} + + +typedef struct { + unsigned char *text; + char *file; + unsigned int x; + unsigned int y; + int bg; + int outline; + unsigned char bgcolor[3]; /* YUV */ + unsigned char fgcolor[3]; /* YUV */ + FT_Library library; + FT_Face face; + FT_Glyph glyphs[ 255 ]; + FT_Bitmap bitmaps[ 255 ]; + int advance[ 255 ]; + int bitmap_left[ 255 ]; + int bitmap_top[ 255 ]; + unsigned int glyphs_index[ 255 ]; + int text_height; + int baseline; + int use_kerning; +} ContextInfo; + + +void Release(void *ctx) +{ + if (ctx) + av_free(ctx); +} + + +static int ParseColor(char *text, unsigned char yuv_color[3]) +{ + char tmp[3]; + unsigned char rgb_color[3]; + int i; + + tmp[2] = '\0'; + + if ((!text) || (strlen(text) != 7) || (text[0] != '#') ) + return -1; + + for (i=0; i < 3; i++) + { + tmp[0] = text[i*2+1]; + tmp[1] = text[i*2+2]; + + rgb_color[i] = strtol(tmp, NULL, 16); + } + + RGB_TO_YUV(rgb_color, yuv_color); + + return 0; +} + +int Configure(void **ctxp, int argc, char *argv[]) +{ + int c; + int error; + ContextInfo *ci=NULL; + char *font=NULL; + unsigned int size=16; + FT_BBox bbox; + int yMax, yMin; + *ctxp = av_mallocz(sizeof(ContextInfo)); + ci = (ContextInfo *) *ctxp; + + /* configure Context Info */ + ci->text = NULL; + ci->file = NULL; + ci->x = ci->y = 0; + ci->fgcolor[0]=255; + ci->fgcolor[1]=128; + ci->fgcolor[2]=128; + ci->bgcolor[0]=0; + ci->fgcolor[1]=128; + ci->fgcolor[2]=128; + ci->bg = 0; + ci->outline = 0; + ci->text_height = 0; + + optind = 0; + while ((c = getopt(argc, argv, "f:t:T:x:y:s:c:C:bo")) > 0) { + switch (c) { + case 'f': + font = optarg; + break; + case 't': + ci->text = av_strdup(optarg); + break; + case 'T': + ci->file = av_strdup(optarg); + break; + case 'x': + ci->x = (unsigned int) atoi(optarg); + break; + case 'y': + ci->y = (unsigned int) atoi(optarg); + break; + case 's': + size = (unsigned int) atoi(optarg); + break; + case 'c': + if (ParseColor(optarg, ci->fgcolor) == -1) + { + av_log(NULL, AV_LOG_ERROR, "Invalid foreground color: '%s'. You must specify the color in the internet way(packaged hex): #RRGGBB, ie: -c #ffffff (for white foreground)\n", optarg); + return -1; + } + break; + case 'C': + if (ParseColor(optarg, ci->bgcolor) == -1) + { + av_log(NULL, AV_LOG_ERROR, "Invalid foreground color: '%s'. You must specify the color in the internet way(packaged hex): #RRGGBB, ie: -c #ffffff (for white foreground)\n", optarg); + return -1; + } + break; + case 'b': + ci->bg=1; + break; + case 'o': + ci->outline=1; + break; + case '?': + av_log(NULL, AV_LOG_ERROR, "Unrecognized argument '%s'\n", argv[optind]); + return -1; + } + } + + if (!ci->text) + { + av_log(NULL, AV_LOG_ERROR, "No text provided (-t text)\n"); + return -1; + } + + if (ci->file) + { + FILE *fp; + if ((fp=fopen(ci->file, "r")) == NULL) + { + av_log(NULL, AV_LOG_INFO, "WARNING: The file could not be opened. Using text provided with -t switch: %s", strerror(errno)); + } + else + { + fclose(fp); + } + } + + if (!font) + { + av_log(NULL, AV_LOG_ERROR, "No font file provided! (-f filename)\n"); + return -1; + } + + if ((error = FT_Init_FreeType(&(ci->library))) != 0) + { + av_log(NULL, AV_LOG_ERROR, "Could not load FreeType (error# %d).\n", error); + return -1; + } + + if ((error = FT_New_Face( ci->library, font, 0, &(ci->face) )) != 0) + { + av_log(NULL, AV_LOG_ERROR, "Could not load face: %s (error# %d).\n", font, error); + return -1; + } + + if ((error = FT_Set_Pixel_Sizes( ci->face, 0, size)) != 0) + { + av_log(NULL, AV_LOG_ERROR, "Could not set font size to %d pixels (error# %d).\n", size, error); + return -1; + } + + ci->use_kerning = FT_HAS_KERNING(ci->face); + + /* load and cache glyphs */ + yMax = -32000; + yMin = 32000; + for (c=0; c < 256; c++) + { + /* Load char */ + error = FT_Load_Char( ci->face, (unsigned char) c, FT_LOAD_RENDER | FT_LOAD_MONOCHROME ); + if (error) continue; /* ignore errors */ + + /* Save bitmap */ + ci->bitmaps[c] = ci->face->glyph->bitmap; + /* Save bitmap left */ + ci->bitmap_left[c] = ci->face->glyph->bitmap_left; + /* Save bitmap top */ + ci->bitmap_top[c] = ci->face->glyph->bitmap_top; + + /* Save advance */ + ci->advance[c] = ci->face->glyph->advance.x >> 6; + + /* Save glyph */ + error = FT_Get_Glyph( ci->face->glyph, &(ci->glyphs[c]) ); + /* Save glyph index */ + ci->glyphs_index[c] = FT_Get_Char_Index( ci->face, (unsigned char) c ); + + /* Measure text height to calculate text_height (or the maximum text height) */ + FT_Glyph_Get_CBox( ci->glyphs[ c ], ft_glyph_bbox_pixels, &bbox ); + if (bbox.yMax > yMax) + yMax = bbox.yMax; + if (bbox.yMin < yMin) + yMin = bbox.yMin; + + } + + ci->text_height = yMax - yMin; + ci->baseline = yMax; + + return 0; +} + + + + +static inline void draw_glyph(AVPicture *picture, FT_Bitmap *bitmap, unsigned int x, unsigned int y, unsigned int width, unsigned int height, unsigned char yuv_fgcolor[3], unsigned char yuv_bgcolor[3], int outline) +{ + int r, c; + int spixel, dpixel[3], in_glyph=0; + + if (bitmap->pixel_mode == ft_pixel_mode_mono) + { + in_glyph = 0; + for (r=0; (r < bitmap->rows) && (r+y < height); r++) + { + for (c=0; (c < bitmap->width) && (c+x < width); c++) + { + /* pixel in the picture (destination) */ + GET_PIXEL(picture, dpixel, (c+x), (y+r)); + + /* pixel in the glyph bitmap (source) */ + spixel = bitmap->buffer[r*bitmap->pitch +c/8] & (0x80>>(c%8)); + + if (spixel) + COPY_3(dpixel, yuv_fgcolor); + + if (outline) + { + /* border detection: */ + if ( (!in_glyph) && (spixel) ) + /* left border detected */ + { + in_glyph = 1; + /* draw left pixel border */ + if (c-1 >= 0) + SET_PIXEL(picture, yuv_bgcolor, (c+x-1), (y+r)); + } + else if ( (in_glyph) && (!spixel) ) + /* right border detected */ + { + in_glyph = 0; + /* 'draw' right pixel border */ + COPY_3(dpixel, yuv_bgcolor); + } + + if (in_glyph) + /* see if we have a top/bottom border */ + { + /* top */ + if ( (r-1 >= 0) && (! bitmap->buffer[(r-1)*bitmap->pitch +c/8] & (0x80>>(c%8))) ) + /* we have a top border */ + SET_PIXEL(picture, yuv_bgcolor, (c+x), (y+r-1)); + + /* bottom */ + if ( (r+1 < height) && (! bitmap->buffer[(r+1)*bitmap->pitch +c/8] & (0x80>>(c%8))) ) + /* we have a bottom border */ + SET_PIXEL(picture, yuv_bgcolor, (c+x), (y+r+1)); + + } + } + + SET_PIXEL(picture, dpixel, (c+x), (y+r)); + } + } + } +} + + +static inline void draw_box(AVPicture *picture, unsigned int x, unsigned int y, unsigned int width, unsigned int height, unsigned char yuv_color[3]) +{ + int i, j; + + for (j = 0; (j < height); j++) + for (i = 0; (i < width); i++) + { + SET_PIXEL(picture, yuv_color, (i+x), (y+j)); + } + +} + + + + +void Process(void *ctx, AVPicture *picture, enum PixelFormat pix_fmt, int width, int height, int64_t pts) +{ + ContextInfo *ci = (ContextInfo *) ctx; + FT_Face face = ci->face; + FT_GlyphSlot slot = face->glyph; + unsigned char *text = ci->text; + unsigned char c; + int x = 0, y = 0, i=0, size=0; + unsigned char buff[MAXSIZE_TEXT]; + unsigned char tbuff[MAXSIZE_TEXT]; + time_t now = time(0); + int str_w, str_w_max; + FT_Vector pos[MAXSIZE_TEXT]; + FT_Vector delta; + + if (ci->file) + { + int fd = open(ci->file, O_RDONLY); + + if (fd < 0) + { + text = ci->text; + av_log(NULL, AV_LOG_INFO, "WARNING: The file could not be opened. Using text provided with -t switch: %s", strerror(errno)); + } + else + { + int l = read(fd, tbuff, sizeof(tbuff) - 1); + + if (l >= 0) + { + tbuff[l] = 0; + text = tbuff; + } + else + { + text = ci->text; + av_log(NULL, AV_LOG_INFO, "WARNING: The file could not be read. Using text provided with -t switch: %s", strerror(errno)); + } + close(fd); + } + } + else + { + text = ci->text; + } + + strftime(buff, sizeof(buff), text, localtime(&now)); + + text = buff; + + size = strlen(text); + + + + + /* measure string size and save glyphs position*/ + str_w = str_w_max = 0; + x = ci->x; + y = ci->y; + for (i=0; i < size; i++) + { + c = text[i]; + + /* kerning */ + if ( (ci->use_kerning) && (i > 0) && (ci->glyphs_index[c]) ) + { + FT_Get_Kerning( ci->face, + ci->glyphs_index[ text[i-1] ], + ci->glyphs_index[c], + ft_kerning_default, + &delta ); + + x += delta.x >> 6; + } + + if (( (x + ci->advance[ c ]) >= width ) || ( c == '\n' )) + { + str_w = width - ci->x - 1; + + y += ci->text_height; + x = ci->x; + } + + + /* save position */ + pos[i].x = x + ci->bitmap_left[c]; + pos[i].y = y - ci->bitmap_top[c] + ci->baseline; + + + x += ci->advance[c]; + + + if (str_w > str_w_max) + str_w_max = str_w; + + } + + + + + if (ci->bg) + { + /* Check if it doesn't pass the limits */ + if ( str_w_max + ci->x >= width ) + str_w_max = width - ci->x - 1; + if ( y >= height ) + y = height - 1 - 2*ci->y; + + /* Draw Background */ + draw_box( picture, ci->x, ci->y, str_w_max, y - ci->y, ci->bgcolor ); + } + + + + /* Draw Glyphs */ + for (i=0; i < size; i++) + { + c = text[i]; + + if ( + ( (c == '_') && (text == ci->text) ) || /* skip '_' (consider as space) + IF text was specified in cmd line + (which doesn't like neasted quotes) */ + ( c == '\n' ) /* Skip new line char, just go to new line */ + ) + continue; + + /* now, draw to our target surface */ + draw_glyph( picture, + &(ci->bitmaps[ c ]), + pos[i].x, + pos[i].y, + width, + height, + ci->fgcolor, + ci->bgcolor, + ci->outline ); + + /* increment pen position */ + x += slot->advance.x >> 6; + } + + +} + -- cgit v1.2.3