2011-02-21 02:02:29 +02:00
/*
* Copyright ( c ) 2011 Stefano Sabatini
* Copyright ( c ) 2010 S . N . Hemanth Meenakshisundaram
* Copyright ( c ) 2003 Gustavo Sverzut Barbieri < gsbarbieri @ yahoo . com . br >
*
* 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
*/
/**
* @ file
2011-10-30 19:56:57 +03:00
* drawtext filter , based on the original vhook / drawtext . c
2011-02-21 02:02:29 +02:00
* filter by Gustavo Sverzut Barbieri
*/
2014-02-03 09:46:11 +03:00
# include "config.h"
2014-04-12 17:02:50 +03:00
2014-02-03 09:46:11 +03:00
# if HAVE_SYS_TIME_H
2011-02-21 02:02:29 +02:00
# include <sys/time.h>
2014-02-03 09:46:11 +03:00
# endif
2014-04-17 04:42:17 +03:00
# include <sys/types.h>
2014-04-12 17:02:50 +03:00
# include <sys/stat.h>
2011-02-21 02:02:29 +02:00
# include <time.h>
2014-07-08 03:48:51 +03:00
# if HAVE_UNISTD_H
2014-04-12 17:02:50 +03:00
# include <unistd.h>
2014-07-08 03:48:51 +03:00
# endif
2014-07-04 12:16:16 +03:00
# include <fenv.h>
2014-04-12 17:02:50 +03:00
# if CONFIG_LIBFONTCONFIG
# include <fontconfig/fontconfig.h>
# endif
2011-02-21 02:02:29 +02:00
2011-12-06 13:19:11 +03:00
# include "libavutil/avstring.h"
2012-11-10 18:06:32 +03:00
# include "libavutil/bprint.h"
2012-08-16 10:22:31 +03:00
# include "libavutil/common.h"
2011-02-21 02:02:29 +02:00
# include "libavutil/file.h"
2011-12-04 04:27:13 +03:00
# include "libavutil/eval.h"
2011-02-21 02:02:29 +02:00
# include "libavutil/opt.h"
2011-12-05 02:56:21 +03:00
# include "libavutil/random_seed.h"
2011-02-21 02:02:29 +02:00
# include "libavutil/parseutils.h"
2012-01-16 13:41:39 +03:00
# include "libavutil/timecode.h"
2014-10-24 10:59:34 +03:00
# include "libavutil/time_internal.h"
2011-02-21 02:02:29 +02:00
# include "libavutil/tree.h"
2011-12-05 02:56:21 +03:00
# include "libavutil/lfg.h"
2011-02-21 02:02:29 +02:00
# include "avfilter.h"
# include "drawutils.h"
2012-05-30 11:12:55 +03:00
# include "formats.h"
2012-06-12 21:12:42 +03:00
# include "internal.h"
2012-05-19 11:37:56 +03:00
# include "video.h"
2011-02-21 02:02:29 +02:00
2014-06-21 07:41:45 +03:00
# if CONFIG_LIBFRIBIDI
# include <fribidi.h>
# endif
2011-02-21 02:02:29 +02:00
# include <ft2build.h>
# include FT_FREETYPE_H
# include FT_GLYPH_H
2014-01-18 08:58:34 +03:00
# include FT_STROKER_H
2011-02-21 02:02:29 +02:00
2012-02-20 11:42:33 +03:00
static const char * const var_names [ ] = {
2012-05-01 12:37:15 +03:00
" dar " ,
" hsub " , " vsub " ,
" line_h " , " lh " , ///< line height, same as max_glyph_h
2011-12-04 17:29:26 +03:00
" main_h " , " h " , " H " , ///< height of the input video
2012-05-01 12:37:15 +03:00
" main_w " , " w " , " W " , ///< width of the input video
2011-09-18 03:41:56 +03:00
" max_glyph_a " , " ascent " , ///< max glyph ascent
" max_glyph_d " , " descent " , ///< min glyph descent
2012-05-01 12:37:15 +03:00
" max_glyph_h " , ///< max glyph height
" max_glyph_w " , ///< max glyph width
" n " , ///< number of frame
2011-09-18 03:41:56 +03:00
" sar " ,
2012-05-01 12:37:15 +03:00
" t " , ///< timestamp expressed in seconds
" text_h " , " th " , ///< height of the rendered text
" text_w " , " tw " , ///< width of the rendered text
2011-09-18 03:41:56 +03:00
" x " ,
" y " ,
2013-04-23 04:27:42 +03:00
" pict_type " ,
2011-09-18 03:41:56 +03:00
NULL
} ;
2012-02-20 11:42:33 +03:00
static const char * const fun2_names [ ] = {
" rand "
2011-12-05 02:56:21 +03:00
} ;
static double drand ( void * opaque , double min , double max )
{
2011-12-07 22:30:55 +03:00
return min + ( max - min ) / UINT_MAX * av_lfg_get ( opaque ) ;
2011-12-05 02:56:21 +03:00
}
typedef double ( * eval_func2 ) ( void * , double a , double b ) ;
static const eval_func2 fun2 [ ] = {
drand ,
NULL
} ;
2011-09-18 03:41:56 +03:00
enum var_name {
2012-05-01 12:37:15 +03:00
VAR_DAR ,
VAR_HSUB , VAR_VSUB ,
VAR_LINE_H , VAR_LH ,
2011-12-04 17:29:26 +03:00
VAR_MAIN_H , VAR_h , VAR_H ,
2012-05-01 12:37:15 +03:00
VAR_MAIN_W , VAR_w , VAR_W ,
2011-09-18 03:41:56 +03:00
VAR_MAX_GLYPH_A , VAR_ASCENT ,
VAR_MAX_GLYPH_D , VAR_DESCENT ,
2012-05-01 12:37:15 +03:00
VAR_MAX_GLYPH_H ,
VAR_MAX_GLYPH_W ,
VAR_N ,
2011-09-18 03:41:56 +03:00
VAR_SAR ,
2012-05-01 12:37:15 +03:00
VAR_T ,
VAR_TEXT_H , VAR_TH ,
VAR_TEXT_W , VAR_TW ,
2011-09-18 03:41:56 +03:00
VAR_X ,
VAR_Y ,
2013-04-23 04:27:42 +03:00
VAR_PICT_TYPE ,
2011-09-18 03:41:56 +03:00
VAR_VARS_NB
} ;
2012-11-10 21:46:37 +03:00
enum expansion_mode {
EXP_NONE ,
EXP_NORMAL ,
EXP_STRFTIME ,
} ;
2014-04-11 12:54:15 +03:00
typedef struct DrawTextContext {
2011-02-21 02:02:29 +02:00
const AVClass * class ;
2015-03-08 21:59:51 +02:00
int exp_mode ; ///< expansion mode to use for the text
2011-09-23 16:37:47 +03:00
int reinit ; ///< tells if the filter is being reinited
2014-04-12 17:02:50 +03:00
# if CONFIG_LIBFONTCONFIG
uint8_t * font ; ///< font to be used
# endif
2011-02-21 02:02:29 +02:00
uint8_t * fontfile ; ///< font to be used
uint8_t * text ; ///< text to be drawn
2012-11-10 18:06:32 +03:00
AVBPrint expanded_text ; ///< used to contain the expanded text
2014-07-04 21:54:53 +03:00
uint8_t * fontcolor_expr ; ///< fontcolor expression to evaluate
AVBPrint expanded_fontcolor ; ///< used to contain the expanded fontcolor spec
2011-02-21 02:02:29 +02:00
int ft_load_flags ; ///< flags used for loading fonts, see FT_LOAD_*
FT_Vector * positions ; ///< positions for each element in the text
2011-05-16 21:11:50 +03:00
size_t nb_positions ; ///< number of elements of positions array
2011-02-21 02:02:29 +02:00
char * textfile ; ///< file with text to be drawn
2011-09-20 03:17:08 +03:00
int x ; ///< x position to start drawing text
int y ; ///< y position to start drawing text
2011-09-18 03:41:56 +03:00
int max_glyph_w ; ///< max glyph width
2012-03-13 01:30:13 +03:00
int max_glyph_h ; ///< max glyph height
2011-02-22 02:41:52 +02:00
int shadowx , shadowy ;
2014-01-18 08:58:34 +03:00
int borderw ; ///< border width
2011-02-21 02:02:29 +02:00
unsigned int fontsize ; ///< font size to use
2017-01-19 01:09:22 +02:00
int line_spacing ; ///< lines spacing in pixels
2011-02-21 02:02:29 +02:00
short int draw_box ; ///< draw box around text - true or false
2015-03-31 00:52:47 +02:00
int boxborderw ; ///< box border width
2011-02-21 02:02:29 +02:00
int use_kerning ; ///< font kerning is used - true/false
int tabsize ; ///< tab size
2012-02-05 15:41:01 +03:00
int fix_bounds ; ///< do we let it go out of frame bounds - t/f
2011-02-21 02:02:29 +02:00
2012-03-28 16:27:07 +03:00
FFDrawContext dc ;
FFDrawColor fontcolor ; ///< foreground color
FFDrawColor shadowcolor ; ///< shadow color
2014-01-18 08:58:34 +03:00
FFDrawColor bordercolor ; ///< border color
2012-03-28 16:27:07 +03:00
FFDrawColor boxcolor ; ///< background color
2011-02-21 02:02:29 +02:00
FT_Library library ; ///< freetype font library handle
FT_Face face ; ///< freetype font face handle
2014-01-18 08:58:34 +03:00
FT_Stroker stroker ; ///< freetype stroker handle
2011-02-21 02:02:29 +02:00
struct AVTreeNode * glyphs ; ///< rendered glyphs, stored using the UTF-32 char code
2011-12-04 04:27:13 +03:00
char * x_expr ; ///< expression for x position
char * y_expr ; ///< expression for y position
AVExpr * x_pexpr , * y_pexpr ; ///< parsed expressions for x and y
2011-08-31 00:37:49 +03:00
int64_t basetime ; ///< base pts time in the real world for display
2011-09-18 03:41:56 +03:00
double var_values [ VAR_VARS_NB ] ;
2015-04-11 19:04:25 +02:00
char * a_expr ;
AVExpr * a_pexpr ;
int alpha ;
2011-12-05 02:56:21 +03:00
AVLFG prng ; ///< random
2012-01-16 13:41:39 +03:00
char * tc_opt_string ; ///< specified timecode option string
AVRational tc_rate ; ///< frame rate for timecode
AVTimecode tc ; ///< timecode context
2012-03-12 11:16:04 +03:00
int tc24hmax ; ///< 1 if timecode is wrapped to 24 hours, 0 otherwise
2012-12-02 21:58:39 +03:00
int reload ; ///< reload text file for each frame
2013-05-21 23:14:46 +03:00
int start_number ; ///< starting frame number for n/frame_num var
2014-06-21 07:41:45 +03:00
# if CONFIG_LIBFRIBIDI
int text_shaping ; ///< 1 to shape the text before drawing it
# endif
2013-07-08 21:55:08 +03:00
AVDictionary * metadata ;
2011-02-21 02:02:29 +02:00
} DrawTextContext ;
# define OFFSET(x) offsetof(DrawTextContext, x)
2012-08-13 14:40:01 +03:00
# define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM
2011-02-21 02:02:29 +02:00
static const AVOption drawtext_options [ ] = {
2013-04-15 22:49:08 +03:00
{ " fontfile " , " set font file " , OFFSET ( fontfile ) , AV_OPT_TYPE_STRING , { . str = NULL } , CHAR_MIN , CHAR_MAX , FLAGS } ,
{ " text " , " set text " , OFFSET ( text ) , AV_OPT_TYPE_STRING , { . str = NULL } , CHAR_MIN , CHAR_MAX , FLAGS } ,
{ " textfile " , " set text file " , OFFSET ( textfile ) , AV_OPT_TYPE_STRING , { . str = NULL } , CHAR_MIN , CHAR_MAX , FLAGS } ,
2013-05-13 18:02:59 +03:00
{ " fontcolor " , " set foreground color " , OFFSET ( fontcolor . rgba ) , AV_OPT_TYPE_COLOR , { . str = " black " } , CHAR_MIN , CHAR_MAX , FLAGS } ,
2014-07-04 21:54:53 +03:00
{ " fontcolor_expr " , " set foreground color expression " , OFFSET ( fontcolor_expr ) , AV_OPT_TYPE_STRING , { . str = " " } , CHAR_MIN , CHAR_MAX , FLAGS } ,
2013-05-13 18:02:59 +03:00
{ " boxcolor " , " set box color " , OFFSET ( boxcolor . rgba ) , AV_OPT_TYPE_COLOR , { . str = " white " } , CHAR_MIN , CHAR_MAX , FLAGS } ,
2014-01-18 08:58:34 +03:00
{ " bordercolor " , " set border color " , OFFSET ( bordercolor . rgba ) , AV_OPT_TYPE_COLOR , { . str = " black " } , CHAR_MIN , CHAR_MAX , FLAGS } ,
2013-05-13 18:02:59 +03:00
{ " shadowcolor " , " set shadow color " , OFFSET ( shadowcolor . rgba ) , AV_OPT_TYPE_COLOR , { . str = " black " } , CHAR_MIN , CHAR_MAX , FLAGS } ,
2015-09-08 23:55:48 +02:00
{ " box " , " set box " , OFFSET ( draw_box ) , AV_OPT_TYPE_BOOL , { . i64 = 0 } , 0 , 1 , FLAGS } ,
2015-03-31 00:52:47 +02:00
{ " boxborderw " , " set box border width " , OFFSET ( boxborderw ) , AV_OPT_TYPE_INT , { . i64 = 0 } , INT_MIN , INT_MAX , FLAGS } ,
2017-01-19 01:09:22 +02:00
{ " line_spacing " , " set line spacing in pixels " , OFFSET ( line_spacing ) , AV_OPT_TYPE_INT , { . i64 = 0 } , INT_MIN , INT_MAX , FLAGS } ,
2013-04-15 22:49:08 +03:00
{ " fontsize " , " set font size " , OFFSET ( fontsize ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , INT_MAX , FLAGS } ,
{ " x " , " set x expression " , OFFSET ( x_expr ) , AV_OPT_TYPE_STRING , { . str = " 0 " } , CHAR_MIN , CHAR_MAX , FLAGS } ,
{ " y " , " set y expression " , OFFSET ( y_expr ) , AV_OPT_TYPE_STRING , { . str = " 0 " } , CHAR_MIN , CHAR_MAX , FLAGS } ,
2015-12-21 17:42:14 +02:00
{ " shadowx " , " set shadow x offset " , OFFSET ( shadowx ) , AV_OPT_TYPE_INT , { . i64 = 0 } , INT_MIN , INT_MAX , FLAGS } ,
{ " shadowy " , " set shadow y offset " , OFFSET ( shadowy ) , AV_OPT_TYPE_INT , { . i64 = 0 } , INT_MIN , INT_MAX , FLAGS } ,
2014-01-18 08:58:34 +03:00
{ " borderw " , " set border width " , OFFSET ( borderw ) , AV_OPT_TYPE_INT , { . i64 = 0 } , INT_MIN , INT_MAX , FLAGS } ,
2013-04-15 22:49:08 +03:00
{ " tabsize " , " set tab size " , OFFSET ( tabsize ) , AV_OPT_TYPE_INT , { . i64 = 4 } , 0 , INT_MAX , FLAGS } ,
{ " basetime " , " set base time " , OFFSET ( basetime ) , AV_OPT_TYPE_INT64 , { . i64 = AV_NOPTS_VALUE } , INT64_MIN , INT64_MAX , FLAGS } ,
2014-04-12 17:02:50 +03:00
# if CONFIG_LIBFONTCONFIG
{ " font " , " Font name " , OFFSET ( font ) , AV_OPT_TYPE_STRING , { . str = " Sans " } , . flags = FLAGS } ,
# endif
2013-04-15 22:49:08 +03:00
{ " expansion " , " set the expansion mode " , OFFSET ( exp_mode ) , AV_OPT_TYPE_INT , { . i64 = EXP_NORMAL } , 0 , 2 , FLAGS , " expansion " } ,
{ " none " , " set no expansion " , OFFSET ( exp_mode ) , AV_OPT_TYPE_CONST , { . i64 = EXP_NONE } , 0 , 0 , FLAGS , " expansion " } ,
{ " normal " , " set normal expansion " , OFFSET ( exp_mode ) , AV_OPT_TYPE_CONST , { . i64 = EXP_NORMAL } , 0 , 0 , FLAGS , " expansion " } ,
{ " strftime " , " set strftime expansion (deprecated) " , OFFSET ( exp_mode ) , AV_OPT_TYPE_CONST , { . i64 = EXP_STRFTIME } , 0 , 0 , FLAGS , " expansion " } ,
{ " timecode " , " set initial timecode " , OFFSET ( tc_opt_string ) , AV_OPT_TYPE_STRING , { . str = NULL } , CHAR_MIN , CHAR_MAX , FLAGS } ,
2015-09-08 23:55:48 +02:00
{ " tc24hmax " , " set 24 hours max (timecode only) " , OFFSET ( tc24hmax ) , AV_OPT_TYPE_BOOL , { . i64 = 0 } , 0 , 1 , FLAGS } ,
2013-04-15 22:49:08 +03:00
{ " timecode_rate " , " set rate (timecode only) " , OFFSET ( tc_rate ) , AV_OPT_TYPE_RATIONAL , { . dbl = 0 } , 0 , INT_MAX , FLAGS } ,
{ " r " , " set rate (timecode only) " , OFFSET ( tc_rate ) , AV_OPT_TYPE_RATIONAL , { . dbl = 0 } , 0 , INT_MAX , FLAGS } ,
{ " rate " , " set rate (timecode only) " , OFFSET ( tc_rate ) , AV_OPT_TYPE_RATIONAL , { . dbl = 0 } , 0 , INT_MAX , FLAGS } ,
2015-09-08 23:55:48 +02:00
{ " reload " , " reload text file for each frame " , OFFSET ( reload ) , AV_OPT_TYPE_BOOL , { . i64 = 0 } , 0 , 1 , FLAGS } ,
2015-04-11 19:04:25 +02:00
{ " alpha " , " apply alpha while rendering " , OFFSET ( a_expr ) , AV_OPT_TYPE_STRING , { . str = " 1 " } , . flags = FLAGS } ,
2015-09-08 23:55:48 +02:00
{ " fix_bounds " , " check and fix text coords to avoid clipping " , OFFSET ( fix_bounds ) , AV_OPT_TYPE_BOOL , { . i64 = 1 } , 0 , 1 , FLAGS } ,
2013-05-21 23:14:46 +03:00
{ " start_number " , " start frame number for n/frame_num variable " , OFFSET ( start_number ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , INT_MAX , FLAGS } ,
2011-02-21 02:02:29 +02:00
2014-06-21 07:41:45 +03:00
# if CONFIG_LIBFRIBIDI
2015-09-08 23:55:48 +02:00
{ " text_shaping " , " attempt to shape text before drawing " , OFFSET ( text_shaping ) , AV_OPT_TYPE_BOOL , { . i64 = 1 } , 0 , 1 , FLAGS } ,
2014-06-21 07:41:45 +03:00
# endif
2013-02-25 23:21:29 +03:00
/* FT_LOAD_* flags */
2014-01-10 16:03:12 +03:00
{ " ft_load_flags " , " set font loading flags for libfreetype " , OFFSET ( ft_load_flags ) , AV_OPT_TYPE_FLAGS , { . i64 = FT_LOAD_DEFAULT } , 0 , INT_MAX , FLAGS , " ft_load_flags " } ,
2013-02-25 23:21:29 +03:00
{ " default " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_DEFAULT } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " no_scale " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_NO_SCALE } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " no_hinting " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_NO_HINTING } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " render " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_RENDER } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " no_bitmap " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_NO_BITMAP } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " vertical_layout " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_VERTICAL_LAYOUT } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " force_autohint " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_FORCE_AUTOHINT } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " crop_bitmap " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_CROP_BITMAP } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " pedantic " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_PEDANTIC } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " ignore_global_advance_width " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_IGNORE_GLOBAL_ADVANCE_WIDTH } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " no_recurse " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_NO_RECURSE } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " ignore_transform " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_IGNORE_TRANSFORM } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " monochrome " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_MONOCHROME } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " linear_design " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_LINEAR_DESIGN } , . flags = FLAGS , . unit = " ft_load_flags " } ,
{ " no_autohint " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = FT_LOAD_NO_AUTOHINT } , . flags = FLAGS , . unit = " ft_load_flags " } ,
2013-09-07 15:13:50 +03:00
{ NULL }
2011-02-21 02:02:29 +02:00
} ;
2012-06-22 15:33:09 +03:00
AVFILTER_DEFINE_CLASS ( drawtext ) ;
2011-02-21 02:02:29 +02:00
# undef __FTERRORS_H__
# define FT_ERROR_START_LIST {
# define FT_ERRORDEF(e, v, s) { (e), (s) },
# define FT_ERROR_END_LIST { 0, NULL } };
2014-08-31 11:40:39 +03:00
static const struct ft_error
2011-02-21 02:02:29 +02:00
{
int err ;
const char * err_msg ;
2014-08-31 11:40:39 +03:00
} ft_errors [ ] =
2011-02-21 02:02:29 +02:00
# include FT_ERRORS_H
# define FT_ERRMSG(e) ft_errors[e].err_msg
2014-04-11 12:54:15 +03:00
typedef struct Glyph {
2014-07-03 18:58:38 +03:00
FT_Glyph glyph ;
2014-07-02 15:39:50 +03:00
FT_Glyph border_glyph ;
2011-02-21 02:02:29 +02:00
uint32_t code ;
FT_Bitmap bitmap ; ///< array holding bitmaps of font
2014-01-18 08:58:34 +03:00
FT_Bitmap border_bitmap ; ///< array holding bitmaps of font border
2011-02-21 02:02:29 +02:00
FT_BBox bbox ;
int advance ;
int bitmap_left ;
int bitmap_top ;
} Glyph ;
2015-10-24 23:53:21 +02:00
static int glyph_cmp ( const void * key , const void * b )
2011-02-21 02:02:29 +02:00
{
const Glyph * a = key , * bb = b ;
int64_t diff = ( int64_t ) a - > code - ( int64_t ) bb - > code ;
return diff > 0 ? 1 : diff < 0 ? - 1 : 0 ;
}
/**
* Load glyphs corresponding to the UTF - 32 codepoint code .
*/
static int load_glyph ( AVFilterContext * ctx , Glyph * * glyph_ptr , uint32_t code )
{
2013-03-18 22:44:36 +03:00
DrawTextContext * s = ctx - > priv ;
2014-01-10 16:03:12 +03:00
FT_BitmapGlyph bitmapglyph ;
2011-02-21 02:02:29 +02:00
Glyph * glyph ;
struct AVTreeNode * node = NULL ;
int ret ;
2013-03-18 22:44:36 +03:00
/* load glyph into s->face->glyph */
if ( FT_Load_Char ( s - > face , code , s - > ft_load_flags ) )
2011-02-21 02:02:29 +02:00
return AVERROR ( EINVAL ) ;
2014-07-03 18:58:38 +03:00
glyph = av_mallocz ( sizeof ( * glyph ) ) ;
if ( ! glyph ) {
2011-02-21 02:02:29 +02:00
ret = AVERROR ( ENOMEM ) ;
goto error ;
}
glyph - > code = code ;
2014-07-03 18:58:38 +03:00
if ( FT_Get_Glyph ( s - > face - > glyph , & glyph - > glyph ) ) {
2011-02-21 02:02:29 +02:00
ret = AVERROR ( EINVAL ) ;
goto error ;
}
2014-01-18 08:58:34 +03:00
if ( s - > borderw ) {
2014-07-03 18:58:38 +03:00
glyph - > border_glyph = glyph - > glyph ;
2014-07-02 15:39:50 +03:00
if ( FT_Glyph_StrokeBorder ( & glyph - > border_glyph , s - > stroker , 0 , 0 ) | |
FT_Glyph_To_Bitmap ( & glyph - > border_glyph , FT_RENDER_MODE_NORMAL , 0 , 1 ) ) {
2014-01-18 08:58:34 +03:00
ret = AVERROR_EXTERNAL ;
goto error ;
}
2014-07-02 15:39:50 +03:00
bitmapglyph = ( FT_BitmapGlyph ) glyph - > border_glyph ;
2014-01-18 08:58:34 +03:00
glyph - > border_bitmap = bitmapglyph - > bitmap ;
}
2014-07-03 18:58:38 +03:00
if ( FT_Glyph_To_Bitmap ( & glyph - > glyph , FT_RENDER_MODE_NORMAL , 0 , 1 ) ) {
2014-01-10 16:03:12 +03:00
ret = AVERROR_EXTERNAL ;
goto error ;
}
2014-07-03 18:58:38 +03:00
bitmapglyph = ( FT_BitmapGlyph ) glyph - > glyph ;
2011-02-21 02:02:29 +02:00
2014-01-10 16:03:12 +03:00
glyph - > bitmap = bitmapglyph - > bitmap ;
glyph - > bitmap_left = bitmapglyph - > left ;
glyph - > bitmap_top = bitmapglyph - > top ;
2013-03-18 22:44:36 +03:00
glyph - > advance = s - > face - > glyph - > advance . x > > 6 ;
2011-02-21 02:02:29 +02:00
/* measure text height to calculate text_height (or the maximum text height) */
2014-07-03 18:58:38 +03:00
FT_Glyph_Get_CBox ( glyph - > glyph , ft_glyph_bbox_pixels , & glyph - > bbox ) ;
2011-02-21 02:02:29 +02:00
/* cache the newly created glyph */
2012-11-02 12:56:30 +03:00
if ( ! ( node = av_tree_node_alloc ( ) ) ) {
2011-02-21 02:02:29 +02:00
ret = AVERROR ( ENOMEM ) ;
goto error ;
}
2013-03-18 22:44:36 +03:00
av_tree_insert ( & s - > glyphs , glyph , glyph_cmp , & node ) ;
2011-02-21 02:02:29 +02:00
if ( glyph_ptr )
* glyph_ptr = glyph ;
return 0 ;
error :
if ( glyph )
av_freep ( & glyph - > glyph ) ;
2014-07-02 15:39:50 +03:00
2011-02-21 02:02:29 +02:00
av_freep ( & glyph ) ;
av_freep ( & node ) ;
return ret ;
}
2014-04-17 04:42:17 +03:00
static int load_font_file ( AVFilterContext * ctx , const char * path , int index )
2012-04-07 16:09:16 +03:00
{
2013-05-16 11:33:31 +03:00
DrawTextContext * s = ctx - > priv ;
2012-04-07 16:09:16 +03:00
int err ;
2013-05-16 11:33:31 +03:00
err = FT_New_Face ( s - > library , path , index , & s - > face ) ;
2012-04-07 16:09:16 +03:00
if ( err ) {
2015-12-21 17:07:22 +02:00
# if !CONFIG_LIBFONTCONFIG
2014-04-17 04:42:17 +03:00
av_log ( ctx , AV_LOG_ERROR , " Could not load font \" %s \" : %s \n " ,
s - > fontfile , FT_ERRMSG ( err ) ) ;
2015-12-21 17:07:22 +02:00
# endif
2012-04-07 16:09:16 +03:00
return AVERROR ( EINVAL ) ;
}
return 0 ;
}
2014-04-17 04:42:17 +03:00
# if CONFIG_LIBFONTCONFIG
static int load_font_fontconfig ( AVFilterContext * ctx )
2012-04-07 16:09:16 +03:00
{
2013-05-16 11:33:31 +03:00
DrawTextContext * s = ctx - > priv ;
2012-04-07 16:09:16 +03:00
FcConfig * fontconfig ;
2014-04-12 17:02:50 +03:00
FcPattern * pat , * best ;
2012-04-07 16:09:16 +03:00
FcResult result = FcResultMatch ;
FcChar8 * filename ;
2014-04-17 04:42:17 +03:00
int index ;
2012-04-07 16:09:16 +03:00
double size ;
2014-04-12 17:02:50 +03:00
int err = AVERROR ( ENOENT ) ;
2012-04-07 16:09:16 +03:00
fontconfig = FcInitLoadConfigAndFonts ( ) ;
if ( ! fontconfig ) {
2014-04-17 04:42:17 +03:00
av_log ( ctx , AV_LOG_ERROR , " impossible to init fontconfig \n " ) ;
2014-04-12 17:02:50 +03:00
return AVERROR_UNKNOWN ;
2012-04-07 16:09:16 +03:00
}
2014-04-17 04:42:17 +03:00
pat = FcNameParse ( s - > fontfile ? s - > fontfile :
2012-04-07 16:09:16 +03:00
( uint8_t * ) ( intptr_t ) " default " ) ;
2014-04-17 04:42:17 +03:00
if ( ! pat ) {
av_log ( ctx , AV_LOG_ERROR , " could not parse fontconfig pat " ) ;
2012-04-07 16:09:16 +03:00
return AVERROR ( EINVAL ) ;
}
2014-04-12 17:02:50 +03:00
FcPatternAddString ( pat , FC_FAMILY , s - > font ) ;
2014-04-17 04:42:17 +03:00
if ( s - > fontsize )
FcPatternAddDouble ( pat , FC_SIZE , ( double ) s - > fontsize ) ;
2014-04-12 17:02:50 +03:00
FcDefaultSubstitute ( pat ) ;
2014-04-17 04:42:17 +03:00
if ( ! FcConfigSubstitute ( fontconfig , pat , FcMatchPattern ) ) {
av_log ( ctx , AV_LOG_ERROR , " could not substitue fontconfig options " ) ; /* very unlikely */
2014-04-12 17:02:50 +03:00
FcPatternDestroy ( pat ) ;
return AVERROR ( ENOMEM ) ;
2012-04-07 16:09:16 +03:00
}
2014-04-12 17:02:50 +03:00
2014-04-17 04:42:17 +03:00
best = FcFontMatch ( fontconfig , pat , & result ) ;
2014-04-12 17:02:50 +03:00
FcPatternDestroy ( pat ) ;
2014-04-17 04:42:17 +03:00
if ( ! best | | result ! = FcResultMatch ) {
av_log ( ctx , AV_LOG_ERROR ,
" Cannot find a valid font for the family %s \n " ,
2014-04-12 17:02:50 +03:00
s - > font ) ;
goto fail ;
2012-04-07 16:09:16 +03:00
}
2014-04-12 17:02:50 +03:00
2014-04-17 04:42:17 +03:00
if (
FcPatternGetInteger ( best , FC_INDEX , 0 , & index ) ! = FcResultMatch | |
FcPatternGetDouble ( best , FC_SIZE , 0 , & size ) ! = FcResultMatch ) {
av_log ( ctx , AV_LOG_ERROR , " impossible to find font information " ) ;
2012-04-07 16:09:16 +03:00
return AVERROR ( EINVAL ) ;
}
2014-04-12 17:02:50 +03:00
2014-04-17 04:42:17 +03:00
if ( FcPatternGetString ( best , FC_FILE , 0 , & filename ) ! = FcResultMatch ) {
2014-04-12 17:02:50 +03:00
av_log ( ctx , AV_LOG_ERROR , " No file path for %s \n " ,
s - > font ) ;
goto fail ;
}
2012-04-07 16:09:16 +03:00
av_log ( ctx , AV_LOG_INFO , " Using \" %s \" \n " , filename ) ;
2013-05-16 11:33:31 +03:00
if ( ! s - > fontsize )
s - > fontsize = size + 0.5 ;
2014-04-12 17:02:50 +03:00
2014-04-17 04:42:17 +03:00
err = load_font_file ( ctx , filename , index ) ;
2012-04-07 16:09:16 +03:00
if ( err )
return err ;
FcConfigDestroy ( fontconfig ) ;
2014-04-12 17:02:50 +03:00
fail :
FcPatternDestroy ( best ) ;
return err ;
2012-04-07 16:09:16 +03:00
}
# endif
static int load_font ( AVFilterContext * ctx )
{
2013-05-16 11:33:31 +03:00
DrawTextContext * s = ctx - > priv ;
2012-04-07 16:09:16 +03:00
int err ;
/* load the face, and set up the encoding, which is by default UTF-8 */
2014-04-17 04:42:17 +03:00
err = load_font_file ( ctx , s - > fontfile , 0 ) ;
2012-04-07 16:09:16 +03:00
if ( ! err )
return 0 ;
2014-04-17 04:42:17 +03:00
# if CONFIG_LIBFONTCONFIG
err = load_font_fontconfig ( ctx ) ;
2012-04-07 16:09:16 +03:00
if ( ! err )
return 0 ;
# endif
return err ;
}
2012-12-02 21:58:39 +03:00
static int load_textfile ( AVFilterContext * ctx )
{
2013-05-16 11:33:31 +03:00
DrawTextContext * s = ctx - > priv ;
2012-12-02 21:58:39 +03:00
int err ;
uint8_t * textbuf ;
2014-04-27 02:54:32 +03:00
uint8_t * tmp ;
2012-12-02 21:58:39 +03:00
size_t textbuf_size ;
2013-05-16 11:33:31 +03:00
if ( ( err = av_file_map ( s - > textfile , & textbuf , & textbuf_size , 0 , ctx ) ) < 0 ) {
2012-12-02 21:58:39 +03:00
av_log ( ctx , AV_LOG_ERROR ,
" The text file '%s' could not be read or is empty \n " ,
2013-05-16 11:33:31 +03:00
s - > textfile ) ;
2012-12-02 21:58:39 +03:00
return err ;
}
2014-10-18 22:31:08 +03:00
if ( textbuf_size > SIZE_MAX - 1 | | ! ( tmp = av_realloc ( s - > text , textbuf_size + 1 ) ) ) {
2014-04-27 02:54:32 +03:00
av_file_unmap ( textbuf , textbuf_size ) ;
2012-12-02 21:58:39 +03:00
return AVERROR ( ENOMEM ) ;
2014-04-27 02:54:32 +03:00
}
s - > text = tmp ;
2013-05-16 11:33:31 +03:00
memcpy ( s - > text , textbuf , textbuf_size ) ;
s - > text [ textbuf_size ] = 0 ;
2012-12-02 21:58:39 +03:00
av_file_unmap ( textbuf , textbuf_size ) ;
return 0 ;
}
2014-06-21 07:41:45 +03:00
static inline int is_newline ( uint32_t c )
{
return c = = ' \n ' | | c = = ' \r ' | | c = = ' \f ' | | c = = ' \v ' ;
}
# if CONFIG_LIBFRIBIDI
static int shape_text ( AVFilterContext * ctx )
{
DrawTextContext * s = ctx - > priv ;
uint8_t * tmp ;
int ret = AVERROR ( ENOMEM ) ;
static const FriBidiFlags flags = FRIBIDI_FLAGS_DEFAULT |
FRIBIDI_FLAGS_ARABIC ;
FriBidiChar * unicodestr = NULL ;
FriBidiStrIndex len ;
FriBidiParType direction = FRIBIDI_PAR_LTR ;
FriBidiStrIndex line_start = 0 ;
FriBidiStrIndex line_end = 0 ;
FriBidiLevel * embedding_levels = NULL ;
FriBidiArabicProp * ar_props = NULL ;
FriBidiCharType * bidi_types = NULL ;
FriBidiStrIndex i , j ;
len = strlen ( s - > text ) ;
if ( ! ( unicodestr = av_malloc_array ( len , sizeof ( * unicodestr ) ) ) ) {
goto out ;
}
len = fribidi_charset_to_unicode ( FRIBIDI_CHAR_SET_UTF8 ,
s - > text , len , unicodestr ) ;
bidi_types = av_malloc_array ( len , sizeof ( * bidi_types ) ) ;
if ( ! bidi_types ) {
goto out ;
}
fribidi_get_bidi_types ( unicodestr , len , bidi_types ) ;
embedding_levels = av_malloc_array ( len , sizeof ( * embedding_levels ) ) ;
if ( ! embedding_levels ) {
goto out ;
}
if ( ! fribidi_get_par_embedding_levels ( bidi_types , len , & direction ,
embedding_levels ) ) {
goto out ;
}
ar_props = av_malloc_array ( len , sizeof ( * ar_props ) ) ;
if ( ! ar_props ) {
goto out ;
}
fribidi_get_joining_types ( unicodestr , len , ar_props ) ;
fribidi_join_arabic ( bidi_types , len , embedding_levels , ar_props ) ;
fribidi_shape ( flags , embedding_levels , len , ar_props , unicodestr ) ;
for ( line_end = 0 , line_start = 0 ; line_end < len ; line_end + + ) {
if ( is_newline ( unicodestr [ line_end ] ) | | line_end = = len - 1 ) {
if ( ! fribidi_reorder_line ( flags , bidi_types ,
line_end - line_start + 1 , line_start ,
direction , embedding_levels , unicodestr ,
NULL ) ) {
goto out ;
}
line_start = line_end + 1 ;
}
}
/* Remove zero-width fill chars put in by libfribidi */
for ( i = 0 , j = 0 ; i < len ; i + + )
if ( unicodestr [ i ] ! = FRIBIDI_CHAR_FILL )
unicodestr [ j + + ] = unicodestr [ i ] ;
len = j ;
if ( ! ( tmp = av_realloc ( s - > text , ( len * 4 + 1 ) * sizeof ( * s - > text ) ) ) ) {
/* Use len * 4, as a unicode character can be up to 4 bytes in UTF-8 */
goto out ;
}
s - > text = tmp ;
len = fribidi_unicode_to_charset ( FRIBIDI_CHAR_SET_UTF8 ,
unicodestr , len , s - > text ) ;
ret = 0 ;
out :
av_free ( unicodestr ) ;
av_free ( embedding_levels ) ;
av_free ( ar_props ) ;
av_free ( bidi_types ) ;
return ret ;
}
# endif
2013-03-13 10:26:39 +03:00
static av_cold int init ( AVFilterContext * ctx )
2011-02-21 02:02:29 +02:00
{
int err ;
2013-03-18 22:44:36 +03:00
DrawTextContext * s = ctx - > priv ;
2011-02-21 02:02:29 +02:00
Glyph * glyph ;
2014-04-17 04:42:17 +03:00
if ( ! s - > fontfile & & ! CONFIG_LIBFONTCONFIG ) {
2011-02-21 02:02:29 +02:00
av_log ( ctx , AV_LOG_ERROR , " No font filename provided \n " ) ;
return AVERROR ( EINVAL ) ;
}
2013-03-18 22:44:36 +03:00
if ( s - > textfile ) {
if ( s - > text ) {
2011-02-21 02:02:29 +02:00
av_log ( ctx , AV_LOG_ERROR ,
" Both text and text file provided. Please provide only one \n " ) ;
return AVERROR ( EINVAL ) ;
}
2012-12-02 21:58:39 +03:00
if ( ( err = load_textfile ( ctx ) ) < 0 )
2011-02-21 02:02:29 +02:00
return err ;
}
2013-05-16 11:33:31 +03:00
if ( s - > reload & & ! s - > textfile )
2012-12-02 21:58:39 +03:00
av_log ( ctx , AV_LOG_WARNING , " No file to reload \n " ) ;
2013-05-16 11:33:31 +03:00
if ( s - > tc_opt_string ) {
int ret = av_timecode_init_from_string ( & s - > tc , s - > tc_rate ,
s - > tc_opt_string , ctx ) ;
2012-01-16 13:41:39 +03:00
if ( ret < 0 )
return ret ;
2013-05-16 11:33:31 +03:00
if ( s - > tc24hmax )
s - > tc . flags | = AV_TIMECODE_FLAG_24HOURSMAX ;
if ( ! s - > text )
s - > text = av_strdup ( " " ) ;
2011-12-06 13:19:11 +03:00
}
2013-03-18 22:44:36 +03:00
if ( ! s - > text ) {
2011-02-21 02:02:29 +02:00
av_log ( ctx , AV_LOG_ERROR ,
2011-12-06 13:19:11 +03:00
" Either text, a valid file or a timecode must be provided \n " ) ;
2011-02-21 02:02:29 +02:00
return AVERROR ( EINVAL ) ;
}
2015-12-21 16:54:20 +02:00
# if CONFIG_LIBFRIBIDI
if ( s - > text_shaping )
if ( ( err = shape_text ( ctx ) ) < 0 )
return err ;
# endif
2013-03-18 22:44:36 +03:00
if ( ( err = FT_Init_FreeType ( & ( s - > library ) ) ) ) {
2011-02-21 02:02:29 +02:00
av_log ( ctx , AV_LOG_ERROR ,
" Could not load FreeType: %s \n " , FT_ERRMSG ( err ) ) ;
return AVERROR ( EINVAL ) ;
}
2012-04-07 16:09:16 +03:00
err = load_font ( ctx ) ;
if ( err )
return err ;
2013-05-16 11:33:31 +03:00
if ( ! s - > fontsize )
s - > fontsize = 16 ;
2013-03-18 22:44:36 +03:00
if ( ( err = FT_Set_Pixel_Sizes ( s - > face , 0 , s - > fontsize ) ) ) {
2011-02-21 02:02:29 +02:00
av_log ( ctx , AV_LOG_ERROR , " Could not set font size to %d pixels: %s \n " ,
2013-03-18 22:44:36 +03:00
s - > fontsize , FT_ERRMSG ( err ) ) ;
2011-02-21 02:02:29 +02:00
return AVERROR ( EINVAL ) ;
}
2014-01-18 08:58:34 +03:00
if ( s - > borderw ) {
if ( FT_Stroker_New ( s - > library , & s - > stroker ) ) {
av_log ( ctx , AV_LOG_ERROR , " Coult not init FT stroker \n " ) ;
return AVERROR_EXTERNAL ;
}
FT_Stroker_Set ( s - > stroker , s - > borderw < < 6 , FT_STROKER_LINECAP_ROUND ,
FT_STROKER_LINEJOIN_ROUND , 0 ) ;
}
2013-03-18 22:44:36 +03:00
s - > use_kerning = FT_HAS_KERNING ( s - > face ) ;
2011-02-21 02:02:29 +02:00
/* load the fallback glyph with code 0 */
load_glyph ( ctx , NULL , 0 ) ;
/* set the tabsize in pixels */
2011-11-21 09:50:03 +03:00
if ( ( err = load_glyph ( ctx , & glyph , ' ' ) ) < 0 ) {
2011-02-21 02:02:29 +02:00
av_log ( ctx , AV_LOG_ERROR , " Could not set tabsize. \n " ) ;
return err ;
}
2013-03-18 22:44:36 +03:00
s - > tabsize * = glyph - > advance ;
2011-02-21 02:02:29 +02:00
2013-05-16 11:33:31 +03:00
if ( s - > exp_mode = = EXP_STRFTIME & &
( strchr ( s - > text , ' % ' ) | | strchr ( s - > text , ' \\ ' ) ) )
2012-11-10 21:46:37 +03:00
av_log ( ctx , AV_LOG_WARNING , " expansion=strftime is deprecated. \n " ) ;
2013-05-16 11:33:31 +03:00
av_bprint_init ( & s - > expanded_text , 0 , AV_BPRINT_SIZE_UNLIMITED ) ;
2014-07-04 21:54:53 +03:00
av_bprint_init ( & s - > expanded_fontcolor , 0 , AV_BPRINT_SIZE_UNLIMITED ) ;
2012-11-10 18:06:32 +03:00
2011-02-21 02:02:29 +02:00
return 0 ;
}
static int query_formats ( AVFilterContext * ctx )
{
2015-04-03 19:55:18 +02:00
return ff_set_common_formats ( ctx , ff_draw_supported_pixel_formats ( 0 ) ) ;
2011-02-21 02:02:29 +02:00
}
static int glyph_enu_free ( void * opaque , void * elem )
{
2012-04-10 10:55:11 +03:00
Glyph * glyph = elem ;
2014-07-03 18:58:38 +03:00
FT_Done_Glyph ( glyph - > glyph ) ;
2014-07-02 15:39:50 +03:00
FT_Done_Glyph ( glyph - > border_glyph ) ;
2011-02-21 02:02:29 +02:00
av_free ( elem ) ;
return 0 ;
}
static av_cold void uninit ( AVFilterContext * ctx )
{
2013-03-18 22:44:36 +03:00
DrawTextContext * s = ctx - > priv ;
2011-02-21 02:02:29 +02:00
2013-03-19 00:13:35 +03:00
av_expr_free ( s - > x_pexpr ) ;
av_expr_free ( s - > y_pexpr ) ;
2017-03-20 21:40:29 +02:00
av_expr_free ( s - > a_pexpr ) ;
s - > x_pexpr = s - > y_pexpr = s - > a_pexpr = NULL ;
2013-03-18 22:44:36 +03:00
av_freep ( & s - > positions ) ;
2013-05-16 11:33:31 +03:00
s - > nb_positions = 0 ;
2011-09-23 15:13:47 +03:00
2013-04-10 19:27:40 +03:00
2013-03-18 22:44:36 +03:00
av_tree_enumerate ( s - > glyphs , NULL , NULL , glyph_enu_free ) ;
av_tree_destroy ( s - > glyphs ) ;
2013-05-16 11:33:31 +03:00
s - > glyphs = NULL ;
2011-09-23 15:13:47 +03:00
2013-03-18 22:44:36 +03:00
FT_Done_Face ( s - > face ) ;
2014-01-18 08:58:34 +03:00
FT_Stroker_Done ( s - > stroker ) ;
2013-03-18 22:44:36 +03:00
FT_Done_FreeType ( s - > library ) ;
2012-11-10 18:06:32 +03:00
2013-05-16 11:33:31 +03:00
av_bprint_finalize ( & s - > expanded_text , NULL ) ;
2014-07-04 21:54:53 +03:00
av_bprint_finalize ( & s - > expanded_fontcolor , NULL ) ;
2011-02-21 02:02:29 +02:00
}
static int config_input ( AVFilterLink * inlink )
{
2011-09-18 03:41:56 +03:00
AVFilterContext * ctx = inlink - > dst ;
2013-03-18 22:44:36 +03:00
DrawTextContext * s = ctx - > priv ;
2011-02-21 02:02:29 +02:00
int ret ;
2016-09-14 01:37:13 +02:00
ff_draw_init ( & s - > dc , inlink - > format , FF_DRAW_PROCESS_ALPHA ) ;
2013-05-16 11:33:31 +03:00
ff_draw_color ( & s - > dc , & s - > fontcolor , s - > fontcolor . rgba ) ;
ff_draw_color ( & s - > dc , & s - > shadowcolor , s - > shadowcolor . rgba ) ;
2014-01-18 08:58:34 +03:00
ff_draw_color ( & s - > dc , & s - > bordercolor , s - > bordercolor . rgba ) ;
2013-05-16 11:33:31 +03:00
ff_draw_color ( & s - > dc , & s - > boxcolor , s - > boxcolor . rgba ) ;
s - > var_values [ VAR_w ] = s - > var_values [ VAR_W ] = s - > var_values [ VAR_MAIN_W ] = inlink - > w ;
s - > var_values [ VAR_h ] = s - > var_values [ VAR_H ] = s - > var_values [ VAR_MAIN_H ] = inlink - > h ;
s - > var_values [ VAR_SAR ] = inlink - > sample_aspect_ratio . num ? av_q2d ( inlink - > sample_aspect_ratio ) : 1 ;
s - > var_values [ VAR_DAR ] = ( double ) inlink - > w / inlink - > h * s - > var_values [ VAR_SAR ] ;
s - > var_values [ VAR_HSUB ] = 1 < < s - > dc . hsub_max ;
s - > var_values [ VAR_VSUB ] = 1 < < s - > dc . vsub_max ;
s - > var_values [ VAR_X ] = NAN ;
s - > var_values [ VAR_Y ] = NAN ;
s - > var_values [ VAR_T ] = NAN ;
2011-12-01 13:27:19 +03:00
2013-05-16 11:33:31 +03:00
av_lfg_init ( & s - > prng , av_get_random_seed ( ) ) ;
2011-12-01 13:27:19 +03:00
2013-05-17 12:28:15 +03:00
av_expr_free ( s - > x_pexpr ) ;
av_expr_free ( s - > y_pexpr ) ;
2017-03-20 21:40:29 +02:00
av_expr_free ( s - > a_pexpr ) ;
s - > x_pexpr = s - > y_pexpr = s - > a_pexpr = NULL ;
2013-08-20 21:56:39 +03:00
2013-05-16 11:33:31 +03:00
if ( ( ret = av_expr_parse ( & s - > x_pexpr , s - > x_expr , var_names ,
2011-12-08 02:23:37 +03:00
NULL , NULL , fun2_names , fun2 , 0 , ctx ) ) < 0 | |
2013-05-16 11:33:31 +03:00
( ret = av_expr_parse ( & s - > y_pexpr , s - > y_expr , var_names ,
2015-04-21 20:20:58 +02:00
NULL , NULL , fun2_names , fun2 , 0 , ctx ) ) < 0 | |
( ret = av_expr_parse ( & s - > a_pexpr , s - > a_expr , var_names ,
2011-12-08 02:23:37 +03:00
NULL , NULL , fun2_names , fun2 , 0 , ctx ) ) < 0 )
2011-09-18 03:41:56 +03:00
return AVERROR ( EINVAL ) ;
2011-02-21 02:02:29 +02:00
return 0 ;
}
2011-08-28 21:47:33 +03:00
static int command ( AVFilterContext * ctx , const char * cmd , const char * arg , char * res , int res_len , int flags )
{
2013-05-16 11:33:31 +03:00
DrawTextContext * s = ctx - > priv ;
2011-09-23 16:37:47 +03:00
2011-09-25 23:38:32 +03:00
if ( ! strcmp ( cmd , " reinit " ) ) {
2011-09-04 21:47:17 +03:00
int ret ;
2011-08-28 21:47:33 +03:00
uninit ( ctx ) ;
2013-05-16 11:33:31 +03:00
s - > reinit = 1 ;
2014-02-01 16:47:25 +03:00
if ( ( ret = av_set_options_string ( ctx , arg , " = " , " : " ) ) < 0 )
return ret ;
2013-04-12 12:13:33 +03:00
if ( ( ret = init ( ctx ) ) < 0 )
2011-09-04 21:47:17 +03:00
return ret ;
return config_input ( ctx - > inputs [ 0 ] ) ;
2011-08-28 21:47:33 +03:00
}
return AVERROR ( ENOSYS ) ;
}
2013-04-23 04:27:42 +03:00
static int func_pict_type ( AVFilterContext * ctx , AVBPrint * bp ,
char * fct , unsigned argc , char * * argv , int tag )
{
2013-05-16 11:33:31 +03:00
DrawTextContext * s = ctx - > priv ;
2013-04-23 04:27:42 +03:00
2013-05-16 11:33:31 +03:00
av_bprintf ( bp , " %c " , av_get_picture_type_char ( s - > var_values [ VAR_PICT_TYPE ] ) ) ;
2013-04-23 04:27:42 +03:00
return 0 ;
}
2012-11-10 21:46:37 +03:00
static int func_pts ( AVFilterContext * ctx , AVBPrint * bp ,
char * fct , unsigned argc , char * * argv , int tag )
{
2013-05-16 11:33:31 +03:00
DrawTextContext * s = ctx - > priv ;
2014-04-25 15:01:43 +03:00
const char * fmt ;
double pts = s - > var_values [ VAR_T ] ;
int ret ;
2012-11-10 21:46:37 +03:00
2014-04-25 15:01:43 +03:00
fmt = argc > = 1 ? argv [ 0 ] : " flt " ;
if ( argc > = 2 ) {
int64_t delta ;
if ( ( ret = av_parse_time ( & delta , argv [ 1 ] , 1 ) ) < 0 ) {
av_log ( ctx , AV_LOG_ERROR , " Invalid delta '%s' \n " , argv [ 1 ] ) ;
return ret ;
}
pts + = ( double ) delta / AV_TIME_BASE ;
}
if ( ! strcmp ( fmt , " flt " ) ) {
2016-09-26 22:23:45 +02:00
av_bprintf ( bp , " %.6f " , pts ) ;
2014-04-25 15:01:43 +03:00
} else if ( ! strcmp ( fmt , " hms " ) ) {
if ( isnan ( pts ) ) {
av_bprintf ( bp , " ??:??:??.??? " ) ;
} else {
2015-12-16 21:39:37 +02:00
int64_t ms = llrint ( pts * 1000 ) ;
2014-04-25 15:01:43 +03:00
char sign = ' ' ;
if ( ms < 0 ) {
sign = ' - ' ;
ms = - ms ;
}
av_bprintf ( bp , " %c%02d:%02d:%02d.%03d " , sign ,
( int ) ( ms / ( 60 * 60 * 1000 ) ) ,
( int ) ( ms / ( 60 * 1000 ) ) % 60 ,
( int ) ( ms / 1000 ) % 60 ,
2016-01-03 23:55:31 +02:00
( int ) ( ms % 1000 ) ) ;
2014-04-25 15:01:43 +03:00
}
2015-10-10 20:52:34 +02:00
} else if ( ! strcmp ( fmt , " localtime " ) | |
! strcmp ( fmt , " gmtime " ) ) {
struct tm tm ;
time_t ms = ( time_t ) pts ;
const char * timefmt = argc > = 3 ? argv [ 2 ] : " %Y-%m-%d %H:%M:%S " ;
if ( ! strcmp ( fmt , " localtime " ) )
localtime_r ( & ms , & tm ) ;
else
gmtime_r ( & ms , & tm ) ;
av_bprint_strftime ( bp , timefmt , & tm ) ;
2014-04-25 15:01:43 +03:00
} else {
av_log ( ctx , AV_LOG_ERROR , " Invalid format '%s' \n " , fmt ) ;
return AVERROR ( EINVAL ) ;
}
2012-11-10 21:46:37 +03:00
return 0 ;
}
2012-11-22 14:08:45 +03:00
static int func_frame_num ( AVFilterContext * ctx , AVBPrint * bp ,
char * fct , unsigned argc , char * * argv , int tag )
{
2013-05-16 11:33:31 +03:00
DrawTextContext * s = ctx - > priv ;
2012-11-22 14:08:45 +03:00
2013-06-19 00:38:17 +03:00
av_bprintf ( bp , " %d " , ( int ) s - > var_values [ VAR_N ] ) ;
2012-11-22 14:08:45 +03:00
return 0 ;
}
2013-07-08 21:55:08 +03:00
static int func_metadata ( AVFilterContext * ctx , AVBPrint * bp ,
char * fct , unsigned argc , char * * argv , int tag )
{
DrawTextContext * s = ctx - > priv ;
AVDictionaryEntry * e = av_dict_get ( s - > metadata , argv [ 0 ] , NULL , 0 ) ;
if ( e & & e - > value )
av_bprintf ( bp , " %s " , e - > value ) ;
2016-04-07 10:20:53 +02:00
else if ( argc > = 2 )
av_bprintf ( bp , " %s " , argv [ 1 ] ) ;
2013-07-08 21:55:08 +03:00
return 0 ;
}
2012-11-10 21:46:37 +03:00
static int func_strftime ( AVFilterContext * ctx , AVBPrint * bp ,
char * fct , unsigned argc , char * * argv , int tag )
{
const char * fmt = argc ? argv [ 0 ] : " %Y-%m-%d %H:%M:%S " ;
time_t now ;
struct tm tm ;
time ( & now ) ;
if ( tag = = ' L ' )
localtime_r ( & now , & tm ) ;
else
2014-11-02 20:20:26 +02:00
tm = * gmtime_r ( & now , & tm ) ;
2012-11-10 21:46:37 +03:00
av_bprint_strftime ( bp , fmt , & tm ) ;
return 0 ;
}
2012-11-27 13:27:12 +03:00
static int func_eval_expr ( AVFilterContext * ctx , AVBPrint * bp ,
char * fct , unsigned argc , char * * argv , int tag )
{
2013-03-18 22:44:36 +03:00
DrawTextContext * s = ctx - > priv ;
2012-11-27 13:27:12 +03:00
double res ;
int ret ;
2013-05-16 11:33:31 +03:00
ret = av_expr_parse_and_eval ( & res , argv [ 0 ] , var_names , s - > var_values ,
2012-11-27 13:27:12 +03:00
NULL , NULL , fun2_names , fun2 ,
2013-05-16 11:33:31 +03:00
& s - > prng , 0 , ctx ) ;
2012-11-27 13:27:12 +03:00
if ( ret < 0 )
av_log ( ctx , AV_LOG_ERROR ,
" Expression '%s' for the expr text expansion function is not valid \n " ,
argv [ 0 ] ) ;
else
av_bprintf ( bp , " %f " , res ) ;
return ret ;
}
2014-07-04 12:16:16 +03:00
static int func_eval_expr_int_format ( AVFilterContext * ctx , AVBPrint * bp ,
char * fct , unsigned argc , char * * argv , int tag )
{
DrawTextContext * s = ctx - > priv ;
double res ;
int intval ;
int ret ;
unsigned int positions = 0 ;
char fmt_str [ 30 ] = " % " ;
/*
* argv [ 0 ] expression to be converted to ` int `
* argv [ 1 ] format : ' x ' , ' X ' , ' d ' or ' u '
* argv [ 2 ] positions printed ( optional )
*/
ret = av_expr_parse_and_eval ( & res , argv [ 0 ] , var_names , s - > var_values ,
NULL , NULL , fun2_names , fun2 ,
& s - > prng , 0 , ctx ) ;
if ( ret < 0 ) {
av_log ( ctx , AV_LOG_ERROR ,
" Expression '%s' for the expr text expansion function is not valid \n " ,
argv [ 0 ] ) ;
return ret ;
}
if ( ! strchr ( " xXdu " , argv [ 1 ] [ 0 ] ) ) {
av_log ( ctx , AV_LOG_ERROR , " Invalid format '%c' specified, "
" allowed values: 'x', 'X', 'd', 'u' \n " , argv [ 1 ] [ 0 ] ) ;
return AVERROR ( EINVAL ) ;
}
if ( argc = = 3 ) {
ret = sscanf ( argv [ 2 ] , " %u " , & positions ) ;
if ( ret ! = 1 ) {
2014-07-18 01:00:40 +03:00
av_log ( ctx , AV_LOG_ERROR , " expr_int_format(): Invalid number of positions "
2014-07-04 12:16:16 +03:00
" to print: '%s' \n " , argv [ 2 ] ) ;
return AVERROR ( EINVAL ) ;
}
}
feclearexcept ( FE_ALL_EXCEPT ) ;
intval = res ;
if ( ( ret = fetestexcept ( FE_INVALID | FE_OVERFLOW | FE_UNDERFLOW ) ) ) {
av_log ( ctx , AV_LOG_ERROR , " Conversion of floating-point result to int failed. Control register: 0x%08x. Conversion result: %d \n " , ret , intval ) ;
return AVERROR ( EINVAL ) ;
}
if ( argc = = 3 )
av_strlcatf ( fmt_str , sizeof ( fmt_str ) , " 0%u " , positions ) ;
av_strlcatf ( fmt_str , sizeof ( fmt_str ) , " %c " , argv [ 1 ] [ 0 ] ) ;
av_log ( ctx , AV_LOG_DEBUG , " Formatting value %f (expr '%s') with spec '%s' \n " ,
res , argv [ 0 ] , fmt_str ) ;
av_bprintf ( bp , fmt_str , intval ) ;
return 0 ;
}
2012-11-10 21:46:37 +03:00
static const struct drawtext_function {
const char * name ;
unsigned argc_min , argc_max ;
2013-05-23 16:50:39 +03:00
int tag ; /**< opaque argument to func */
2012-11-10 21:46:37 +03:00
int ( * func ) ( AVFilterContext * , AVBPrint * , char * , unsigned , char * * , int ) ;
} functions [ ] = {
2012-11-27 13:27:12 +03:00
{ " expr " , 1 , 1 , 0 , func_eval_expr } ,
{ " e " , 1 , 1 , 0 , func_eval_expr } ,
2014-07-18 01:00:40 +03:00
{ " expr_int_format " , 2 , 3 , 0 , func_eval_expr_int_format } ,
2014-07-04 12:16:16 +03:00
{ " eif " , 2 , 3 , 0 , func_eval_expr_int_format } ,
2013-04-23 04:27:42 +03:00
{ " pict_type " , 0 , 0 , 0 , func_pict_type } ,
2015-10-10 20:52:34 +02:00
{ " pts " , 0 , 3 , 0 , func_pts } ,
2012-11-10 21:46:37 +03:00
{ " gmtime " , 0 , 1 , ' G ' , func_strftime } ,
{ " localtime " , 0 , 1 , ' L ' , func_strftime } ,
2012-11-22 14:08:45 +03:00
{ " frame_num " , 0 , 0 , 0 , func_frame_num } ,
{ " n " , 0 , 0 , 0 , func_frame_num } ,
2016-04-07 10:20:53 +02:00
{ " metadata " , 1 , 2 , 0 , func_metadata } ,
2012-11-10 21:46:37 +03:00
} ;
static int eval_function ( AVFilterContext * ctx , AVBPrint * bp , char * fct ,
unsigned argc , char * * argv )
{
unsigned i ;
for ( i = 0 ; i < FF_ARRAY_ELEMS ( functions ) ; i + + ) {
if ( strcmp ( fct , functions [ i ] . name ) )
continue ;
if ( argc < functions [ i ] . argc_min ) {
av_log ( ctx , AV_LOG_ERROR , " %%{%s} requires at least %d arguments \n " ,
fct , functions [ i ] . argc_min ) ;
return AVERROR ( EINVAL ) ;
}
if ( argc > functions [ i ] . argc_max ) {
av_log ( ctx , AV_LOG_ERROR , " %%{%s} requires at most %d arguments \n " ,
fct , functions [ i ] . argc_max ) ;
return AVERROR ( EINVAL ) ;
}
break ;
}
if ( i > = FF_ARRAY_ELEMS ( functions ) ) {
av_log ( ctx , AV_LOG_ERROR , " %%{%s} is not known \n " , fct ) ;
return AVERROR ( EINVAL ) ;
}
return functions [ i ] . func ( ctx , bp , fct , argc , argv , functions [ i ] . tag ) ;
}
static int expand_function ( AVFilterContext * ctx , AVBPrint * bp , char * * rtext )
{
const char * text = * rtext ;
char * argv [ 16 ] = { NULL } ;
unsigned argc = 0 , i ;
int ret ;
if ( * text ! = ' { ' ) {
av_log ( ctx , AV_LOG_ERROR , " Stray %% near '%s' \n " , text ) ;
return AVERROR ( EINVAL ) ;
}
text + + ;
while ( 1 ) {
if ( ! ( argv [ argc + + ] = av_get_token ( & text , " :} " ) ) ) {
ret = AVERROR ( ENOMEM ) ;
goto end ;
}
if ( ! * text ) {
av_log ( ctx , AV_LOG_ERROR , " Unterminated %%{} near '%s' \n " , * rtext ) ;
ret = AVERROR ( EINVAL ) ;
goto end ;
}
if ( argc = = FF_ARRAY_ELEMS ( argv ) )
av_freep ( & argv [ - - argc ] ) ; /* error will be caught later */
if ( * text = = ' } ' )
break ;
text + + ;
}
if ( ( ret = eval_function ( ctx , bp , argv [ 0 ] , argc - 1 , argv + 1 ) ) < 0 )
goto end ;
ret = 0 ;
* rtext = ( char * ) text + 1 ;
end :
for ( i = 0 ; i < argc ; i + + )
av_freep ( & argv [ i ] ) ;
return ret ;
}
2014-07-04 21:54:53 +03:00
static int expand_text ( AVFilterContext * ctx , char * text , AVBPrint * bp )
2012-11-10 21:46:37 +03:00
{
int ret ;
av_bprint_clear ( bp ) ;
while ( * text ) {
if ( * text = = ' \\ ' & & text [ 1 ] ) {
av_bprint_chars ( bp , text [ 1 ] , 1 ) ;
text + = 2 ;
} else if ( * text = = ' % ' ) {
text + + ;
if ( ( ret = expand_function ( ctx , bp , & text ) ) < 0 )
return ret ;
} else {
av_bprint_chars ( bp , * text , 1 ) ;
text + + ;
}
}
if ( ! av_bprint_is_complete ( bp ) )
return AVERROR ( ENOMEM ) ;
return 0 ;
}
2013-03-18 22:44:36 +03:00
static int draw_glyphs ( DrawTextContext * s , AVFrame * frame ,
2014-06-22 17:31:09 +03:00
int width , int height ,
2015-04-21 20:20:58 +02:00
FFDrawColor * color ,
int x , int y , int borderw )
2011-02-22 02:11:35 +02:00
{
2013-05-16 11:33:31 +03:00
char * text = s - > expanded_text . str ;
2011-02-22 02:11:35 +02:00
uint32_t code = 0 ;
2011-09-18 03:41:56 +03:00
int i , x1 , y1 ;
2011-02-22 02:11:35 +02:00
uint8_t * p ;
Glyph * glyph = NULL ;
for ( i = 0 , p = text ; * p ; i + + ) {
2014-01-18 08:58:34 +03:00
FT_Bitmap bitmap ;
2011-02-22 02:11:35 +02:00
Glyph dummy = { 0 } ;
GET_UTF8 ( code , * p + + , continue ; ) ;
/* skip new line chars, just go to new line */
if ( code = = ' \n ' | | code = = ' \r ' | | code = = ' \t ' )
continue ;
dummy . code = code ;
2015-10-23 17:46:41 +02:00
glyph = av_tree_find ( s - > glyphs , & dummy , glyph_cmp , NULL ) ;
2011-02-22 02:11:35 +02:00
2014-01-18 08:58:34 +03:00
bitmap = borderw ? glyph - > border_bitmap : glyph - > bitmap ;
2011-02-22 02:11:35 +02:00
if ( glyph - > bitmap . pixel_mode ! = FT_PIXEL_MODE_MONO & &
glyph - > bitmap . pixel_mode ! = FT_PIXEL_MODE_GRAY )
return AVERROR ( EINVAL ) ;
2014-01-18 08:58:34 +03:00
x1 = s - > positions [ i ] . x + s - > x + x - borderw ;
y1 = s - > positions [ i ] . y + s - > y + y - borderw ;
2011-09-18 03:41:56 +03:00
2013-05-16 11:33:31 +03:00
ff_blend_mask ( & s - > dc , color ,
2013-03-10 03:30:30 +03:00
frame - > data , frame - > linesize , width , height ,
2014-01-18 08:58:34 +03:00
bitmap . buffer , bitmap . pitch ,
bitmap . width , bitmap . rows ,
bitmap . pixel_mode = = FT_PIXEL_MODE_MONO ? 0 : 3 ,
2012-03-28 16:27:07 +03:00
0 , x1 , y1 ) ;
2011-02-22 02:11:35 +02:00
}
return 0 ;
}
2011-12-01 13:43:11 +03:00
2015-04-21 20:20:58 +02:00
static void update_color_with_alpha ( DrawTextContext * s , FFDrawColor * color , const FFDrawColor incolor )
{
* color = incolor ;
color - > rgba [ 3 ] = ( color - > rgba [ 3 ] * s - > alpha ) / 255 ;
ff_draw_color ( & s - > dc , color , color - > rgba ) ;
2011-12-01 13:43:11 +03:00
}
2015-04-11 19:04:25 +02:00
static void update_alpha ( DrawTextContext * s )
{
double alpha = av_expr_eval ( s - > a_pexpr , s - > var_values , & s - > prng ) ;
if ( isnan ( alpha ) )
return ;
if ( alpha > = 1.0 )
s - > alpha = 255 ;
else if ( alpha < = 0 )
s - > alpha = 0 ;
else
s - > alpha = 256 * alpha ;
}
2012-11-28 10:41:07 +03:00
static int draw_text ( AVFilterContext * ctx , AVFrame * frame ,
2011-02-21 02:02:29 +02:00
int width , int height )
{
2013-03-18 22:44:36 +03:00
DrawTextContext * s = ctx - > priv ;
2013-04-15 18:42:04 +03:00
AVFilterLink * inlink = ctx - > inputs [ 0 ] ;
2013-05-16 11:33:31 +03:00
2011-02-21 02:02:29 +02:00
uint32_t code = 0 , prev_code = 0 ;
2011-02-22 02:11:35 +02:00
int x = 0 , y = 0 , i = 0 , ret ;
2011-09-18 03:41:56 +03:00
int max_text_line_w = 0 , len ;
int box_w , box_h ;
2013-10-07 11:40:17 +03:00
char * text ;
2011-02-21 02:02:29 +02:00
uint8_t * p ;
int y_min = 32000 , y_max = - 32000 ;
2011-09-18 03:41:56 +03:00
int x_min = 32000 , x_max = - 32000 ;
2011-02-21 02:02:29 +02:00
FT_Vector delta ;
Glyph * glyph = NULL , * prev_glyph = NULL ;
Glyph dummy = { 0 } ;
2011-05-17 00:48:00 +03:00
time_t now = time ( 0 ) ;
struct tm ltime ;
2013-05-16 11:33:31 +03:00
AVBPrint * bp = & s - > expanded_text ;
2012-11-10 18:06:32 +03:00
2015-04-21 20:20:58 +02:00
FFDrawColor fontcolor ;
FFDrawColor shadowcolor ;
FFDrawColor bordercolor ;
FFDrawColor boxcolor ;
2012-11-10 18:06:32 +03:00
av_bprint_clear ( bp ) ;
2011-05-17 00:48:00 +03:00
2013-05-16 11:33:31 +03:00
if ( s - > basetime ! = AV_NOPTS_VALUE )
now = frame - > pts * av_q2d ( ctx - > inputs [ 0 ] - > time_base ) + s - > basetime / 1000000 ;
2011-08-31 00:37:49 +03:00
2013-05-16 11:33:31 +03:00
switch ( s - > exp_mode ) {
2012-11-10 21:46:37 +03:00
case EXP_NONE :
2013-05-16 11:33:31 +03:00
av_bprintf ( bp , " %s " , s - > text ) ;
2012-11-10 21:46:37 +03:00
break ;
case EXP_NORMAL :
2014-07-04 21:54:53 +03:00
if ( ( ret = expand_text ( ctx , s - > text , & s - > expanded_text ) ) < 0 )
2012-11-10 21:46:37 +03:00
return ret ;
break ;
case EXP_STRFTIME :
localtime_r ( & now , & ltime ) ;
2013-05-16 11:33:31 +03:00
av_bprint_strftime ( bp , s - > text , & ltime ) ;
2012-11-10 21:46:37 +03:00
break ;
}
2011-05-16 21:11:50 +03:00
2013-05-16 11:33:31 +03:00
if ( s - > tc_opt_string ) {
2012-01-16 13:41:39 +03:00
char tcbuf [ AV_TIMECODE_STR_SIZE ] ;
2016-08-30 15:28:41 +02:00
av_timecode_make_string ( & s - > tc , tcbuf , inlink - > frame_count_out ) ;
2012-11-10 18:06:32 +03:00
av_bprint_clear ( bp ) ;
2013-05-16 11:33:31 +03:00
av_bprintf ( bp , " %s%s " , s - > text , tcbuf ) ;
2011-12-06 13:19:11 +03:00
}
2012-11-10 18:06:32 +03:00
if ( ! av_bprint_is_complete ( bp ) )
2011-05-17 00:48:00 +03:00
return AVERROR ( ENOMEM ) ;
2013-05-16 11:33:31 +03:00
text = s - > expanded_text . str ;
if ( ( len = s - > expanded_text . len ) > s - > nb_positions ) {
if ( ! ( s - > positions =
av_realloc ( s - > positions , len * sizeof ( * s - > positions ) ) ) )
2011-02-21 02:02:29 +02:00
return AVERROR ( ENOMEM ) ;
2013-05-16 11:33:31 +03:00
s - > nb_positions = len ;
2011-02-21 02:02:29 +02:00
}
2014-07-04 21:54:53 +03:00
if ( s - > fontcolor_expr [ 0 ] ) {
/* If expression is set, evaluate and replace the static value */
av_bprint_clear ( & s - > expanded_fontcolor ) ;
if ( ( ret = expand_text ( ctx , s - > fontcolor_expr , & s - > expanded_fontcolor ) ) < 0 )
return ret ;
if ( ! av_bprint_is_complete ( & s - > expanded_fontcolor ) )
return AVERROR ( ENOMEM ) ;
av_log ( s , AV_LOG_DEBUG , " Evaluated fontcolor is '%s' \n " , s - > expanded_fontcolor . str ) ;
ret = av_parse_color ( s - > fontcolor . rgba , s - > expanded_fontcolor . str , - 1 , s ) ;
if ( ret )
return ret ;
ff_draw_color ( & s - > dc , & s - > fontcolor , s - > fontcolor . rgba ) ;
}
2011-09-18 03:41:56 +03:00
x = 0 ;
y = 0 ;
2011-02-21 02:02:29 +02:00
/* load and cache glyphs */
2011-05-16 21:11:50 +03:00
for ( i = 0 , p = text ; * p ; i + + ) {
2011-02-21 02:02:29 +02:00
GET_UTF8 ( code , * p + + , continue ; ) ;
/* get glyph */
dummy . code = code ;
2013-05-16 11:33:31 +03:00
glyph = av_tree_find ( s - > glyphs , & dummy , glyph_cmp , NULL ) ;
2012-02-07 03:40:29 +03:00
if ( ! glyph ) {
2016-04-24 12:30:20 +02:00
ret = load_glyph ( ctx , & glyph , code ) ;
if ( ret < 0 )
return ret ;
2012-02-07 03:40:29 +03:00
}
2011-02-21 02:02:29 +02:00
y_min = FFMIN ( glyph - > bbox . yMin , y_min ) ;
y_max = FFMAX ( glyph - > bbox . yMax , y_max ) ;
2011-09-18 03:41:56 +03:00
x_min = FFMIN ( glyph - > bbox . xMin , x_min ) ;
x_max = FFMAX ( glyph - > bbox . xMax , x_max ) ;
2011-02-21 02:02:29 +02:00
}
2013-05-16 11:33:31 +03:00
s - > max_glyph_h = y_max - y_min ;
s - > max_glyph_w = x_max - x_min ;
2011-02-21 02:02:29 +02:00
/* compute and save position for each glyph */
glyph = NULL ;
2011-05-16 21:11:50 +03:00
for ( i = 0 , p = text ; * p ; i + + ) {
2011-02-21 02:02:29 +02:00
GET_UTF8 ( code , * p + + , continue ; ) ;
/* skip the \n in the sequence \r\n */
if ( prev_code = = ' \r ' & & code = = ' \n ' )
continue ;
prev_code = code ;
if ( is_newline ( code ) ) {
2013-05-16 11:33:31 +03:00
2011-09-18 03:41:56 +03:00
max_text_line_w = FFMAX ( max_text_line_w , x ) ;
2017-01-19 01:09:22 +02:00
y + = s - > max_glyph_h + s - > line_spacing ;
2011-09-18 03:41:56 +03:00
x = 0 ;
2011-02-21 02:02:29 +02:00
continue ;
}
/* get glyph */
prev_glyph = glyph ;
dummy . code = code ;
2013-05-16 11:33:31 +03:00
glyph = av_tree_find ( s - > glyphs , & dummy , glyph_cmp , NULL ) ;
2011-02-21 02:02:29 +02:00
/* kerning */
2013-05-16 11:33:31 +03:00
if ( s - > use_kerning & & prev_glyph & & glyph - > code ) {
FT_Get_Kerning ( s - > face , prev_glyph - > code , glyph - > code ,
2011-02-21 02:02:29 +02:00
ft_kerning_default , & delta ) ;
x + = delta . x > > 6 ;
}
/* save position */
2013-05-16 11:33:31 +03:00
s - > positions [ i ] . x = x + glyph - > bitmap_left ;
s - > positions [ i ] . y = y - glyph - > bitmap_top + y_max ;
if ( code = = ' \t ' ) x = ( x / s - > tabsize + 1 ) * s - > tabsize ;
2011-02-21 02:02:29 +02:00
else x + = glyph - > advance ;
}
2011-09-18 03:41:56 +03:00
max_text_line_w = FFMAX ( x , max_text_line_w ) ;
2013-05-16 11:33:31 +03:00
s - > var_values [ VAR_TW ] = s - > var_values [ VAR_TEXT_W ] = max_text_line_w ;
s - > var_values [ VAR_TH ] = s - > var_values [ VAR_TEXT_H ] = y + s - > max_glyph_h ;
2011-09-18 03:41:56 +03:00
2013-05-16 11:33:31 +03:00
s - > var_values [ VAR_MAX_GLYPH_W ] = s - > max_glyph_w ;
s - > var_values [ VAR_MAX_GLYPH_H ] = s - > max_glyph_h ;
s - > var_values [ VAR_MAX_GLYPH_A ] = s - > var_values [ VAR_ASCENT ] = y_max ;
s - > var_values [ VAR_MAX_GLYPH_D ] = s - > var_values [ VAR_DESCENT ] = y_min ;
2011-09-18 03:41:56 +03:00
2013-05-16 11:33:31 +03:00
s - > var_values [ VAR_LINE_H ] = s - > var_values [ VAR_LH ] = s - > max_glyph_h ;
2011-09-18 03:41:56 +03:00
2013-05-16 11:33:31 +03:00
s - > x = s - > var_values [ VAR_X ] = av_expr_eval ( s - > x_pexpr , s - > var_values , & s - > prng ) ;
s - > y = s - > var_values [ VAR_Y ] = av_expr_eval ( s - > y_pexpr , s - > var_values , & s - > prng ) ;
s - > x = s - > var_values [ VAR_X ] = av_expr_eval ( s - > x_pexpr , s - > var_values , & s - > prng ) ;
2011-09-18 03:41:56 +03:00
2015-04-11 19:04:25 +02:00
update_alpha ( s ) ;
2015-04-21 20:20:58 +02:00
update_color_with_alpha ( s , & fontcolor , s - > fontcolor ) ;
update_color_with_alpha ( s , & shadowcolor , s - > shadowcolor ) ;
update_color_with_alpha ( s , & bordercolor , s - > bordercolor ) ;
update_color_with_alpha ( s , & boxcolor , s - > boxcolor ) ;
2011-09-18 03:41:56 +03:00
box_w = FFMIN ( width - 1 , max_text_line_w ) ;
2013-05-16 11:33:31 +03:00
box_h = FFMIN ( height - 1 , y + s - > max_glyph_h ) ;
2011-02-21 02:02:29 +02:00
/* draw box */
2013-03-18 22:44:36 +03:00
if ( s - > draw_box )
2015-04-21 20:20:58 +02:00
ff_blend_rectangle ( & s - > dc , & boxcolor ,
2013-03-10 03:30:30 +03:00
frame - > data , frame - > linesize , width , height ,
2015-03-31 00:52:47 +02:00
s - > x - s - > boxborderw , s - > y - s - > boxborderw ,
box_w + s - > boxborderw * 2 , box_h + s - > boxborderw * 2 ) ;
2011-02-21 02:02:29 +02:00
2013-03-18 22:44:36 +03:00
if ( s - > shadowx | | s - > shadowy ) {
2014-06-22 17:31:09 +03:00
if ( ( ret = draw_glyphs ( s , frame , width , height ,
2015-04-21 20:20:58 +02:00
& shadowcolor , s - > shadowx , s - > shadowy , 0 ) ) < 0 )
2011-02-22 02:41:52 +02:00
return ret ;
}
2014-01-18 08:58:34 +03:00
if ( s - > borderw ) {
2014-06-22 17:31:09 +03:00
if ( ( ret = draw_glyphs ( s , frame , width , height ,
2015-04-21 20:20:58 +02:00
& bordercolor , 0 , 0 , s - > borderw ) ) < 0 )
2014-01-18 08:58:34 +03:00
return ret ;
}
2014-06-22 17:31:09 +03:00
if ( ( ret = draw_glyphs ( s , frame , width , height ,
2015-04-21 20:20:58 +02:00
& fontcolor , 0 , 0 , 0 ) ) < 0 )
2011-02-22 02:11:35 +02:00
return ret ;
2011-02-21 02:02:29 +02:00
return 0 ;
}
2012-11-28 10:41:07 +03:00
static int filter_frame ( AVFilterLink * inlink , AVFrame * frame )
2011-02-21 02:02:29 +02:00
{
2011-09-18 03:41:56 +03:00
AVFilterContext * ctx = inlink - > dst ;
2011-09-23 15:28:26 +03:00
AVFilterLink * outlink = ctx - > outputs [ 0 ] ;
2013-03-18 22:44:36 +03:00
DrawTextContext * s = ctx - > priv ;
2012-07-14 10:25:33 +03:00
int ret ;
2011-02-21 02:02:29 +02:00
2014-06-21 07:41:45 +03:00
if ( s - > reload ) {
2015-04-10 16:28:45 +02:00
if ( ( ret = load_textfile ( ctx ) ) < 0 ) {
av_frame_free ( & frame ) ;
2012-12-02 21:58:39 +03:00
return ret ;
2015-04-10 16:28:45 +02:00
}
2014-06-21 07:41:45 +03:00
# if CONFIG_LIBFRIBIDI
if ( s - > text_shaping )
2015-04-10 16:28:45 +02:00
if ( ( ret = shape_text ( ctx ) ) < 0 ) {
av_frame_free ( & frame ) ;
2014-06-21 07:41:45 +03:00
return ret ;
2015-04-10 16:28:45 +02:00
}
2014-06-21 07:41:45 +03:00
# endif
}
2012-12-02 21:58:39 +03:00
2016-08-30 15:28:41 +02:00
s - > var_values [ VAR_N ] = inlink - > frame_count_out + s - > start_number ;
2013-03-18 22:44:36 +03:00
s - > var_values [ VAR_T ] = frame - > pts = = AV_NOPTS_VALUE ?
2012-11-28 20:07:51 +03:00
NAN : frame - > pts * av_q2d ( inlink - > time_base ) ;
2011-09-18 03:41:56 +03:00
2013-05-16 11:33:31 +03:00
s - > var_values [ VAR_PICT_TYPE ] = frame - > pict_type ;
2013-07-08 21:55:08 +03:00
s - > metadata = av_frame_get_metadata ( frame ) ;
2013-04-23 04:27:42 +03:00
2013-03-10 03:30:30 +03:00
draw_text ( ctx , frame , frame - > width , frame - > height ) ;
2011-09-18 03:41:56 +03:00
av_log ( ctx , AV_LOG_DEBUG , " n:%d t:%f text_w:%d text_h:%d x:%d y:%d \n " ,
2013-05-16 11:33:31 +03:00
( int ) s - > var_values [ VAR_N ] , s - > var_values [ VAR_T ] ,
( int ) s - > var_values [ VAR_TEXT_W ] , ( int ) s - > var_values [ VAR_TEXT_H ] ,
s - > x , s - > y ) ;
2011-09-18 03:41:56 +03:00
2012-12-06 15:48:00 +03:00
return ff_filter_frame ( outlink , frame ) ;
2011-02-21 02:02:29 +02:00
}
2012-07-24 16:14:01 +03:00
static const AVFilterPad avfilter_vf_drawtext_inputs [ ] = {
{
2013-09-07 15:13:50 +03:00
. name = " default " ,
. type = AVMEDIA_TYPE_VIDEO ,
. filter_frame = filter_frame ,
. config_props = config_input ,
. needs_writable = 1 ,
2012-07-24 16:14:01 +03:00
} ,
{ NULL }
} ;
static const AVFilterPad avfilter_vf_drawtext_outputs [ ] = {
{
. name = " default " ,
. type = AVMEDIA_TYPE_VIDEO ,
} ,
{ NULL }
} ;
2013-10-28 09:44:24 +03:00
AVFilter ff_vf_drawtext = {
2011-02-21 02:02:29 +02:00
. name = " drawtext " ,
. description = NULL_IF_CONFIG_SMALL ( " Draw text on top of video frames using libfreetype library. " ) ,
. priv_size = sizeof ( DrawTextContext ) ,
2013-02-25 23:21:29 +03:00
. priv_class = & drawtext_class ,
2011-02-21 02:02:29 +02:00
. init = init ,
. uninit = uninit ,
. query_formats = query_formats ,
2013-09-07 15:13:50 +03:00
. inputs = avfilter_vf_drawtext_inputs ,
. outputs = avfilter_vf_drawtext_outputs ,
2011-08-28 21:47:33 +03:00
. process_command = command ,
2013-09-07 15:13:50 +03:00
. flags = AVFILTER_FLAG_SUPPORT_TIMELINE_GENERIC ,
2011-02-21 02:02:29 +02:00
} ;