git.sesse.net Git - vlc/blob - modules/codec/libass.c

   1 /*****************************************************************************
   2  * SSA/ASS subtitle decoder using libass.
   3  *****************************************************************************
   4  * Copyright (C) 2008 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@videolan.org>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27
  28 #ifdef HAVE_CONFIG_H
  29 #   include "config.h"
  30 #endif
  31
  32 #include <string.h>
  33 #include <limits.h>
  34 #include <assert.h>
  35 #include <math.h>
  36
  37 #include <vlc_common.h>
  38 #include <vlc_plugin.h>
  39 #include <vlc_vout.h>
  40 #include <vlc_codec.h>
  41 #include <vlc_osd.h>
  42 #include <vlc_input.h>
  43
  44 #include <ass/ass.h>
  45
  46 /*****************************************************************************
  47  * Module descriptor
  48  *****************************************************************************/
  49 static int  Create ( vlc_object_t * );
  50 static void Destroy( vlc_object_t * );
  51
  52 vlc_module_begin();
  53     set_shortname( N_("Subtitles (advanced)"));
  54     set_description( N_("Subtitle renderers using libass") );
  55     set_capability( "decoder", 100 );
  56     set_category( CAT_INPUT );
  57     set_subcategory( SUBCAT_INPUT_SCODEC );
  58     set_callbacks( Create, Destroy );
  59 vlc_module_end();
  60
  61 /*****************************************************************************
  62  * Local prototypes
  63  *****************************************************************************/
  64 static subpicture_t *DecodeBlock( decoder_t *, block_t ** );
  65 static void DestroySubpicture( subpicture_t * );
  66 static void PreRender( video_format_t *, spu_t *, subpicture_t * );
  67 static void UpdateRegions( video_format_t *, spu_t *,
  68                            subpicture_t *, mtime_t );
  69
  70 /* Yes libass sux with threads */
  71 typedef struct
  72 {
  73     vlc_object_t   *p_libvlc;
  74
  75     vlc_mutex_t     *p_lock;
  76     int             i_refcount;
  77     ass_library_t   *p_library;
  78     ass_renderer_t  *p_renderer;
  79     video_format_t  fmt;
  80 } ass_handle_t;
  81 static ass_handle_t *AssHandleYield( decoder_t *p_dec );
  82 static void AssHandleRelease( ass_handle_t * );
  83
  84 /* */
  85 struct decoder_sys_t
  86 {
  87     /* decoder_sys_t is shared between decoder and spu units */
  88     vlc_mutex_t  lock;
  89     int          i_refcount;
  90
  91     /* */
  92     ass_handle_t *p_ass;
  93
  94     /* */
  95     ass_track_t  *p_track;
  96
  97     /* */
  98     subpicture_t *p_spu_final;
  99 };
 100 static void DecSysRelease( decoder_sys_t *p_sys );
 101 static void DecSysYield( decoder_sys_t *p_sys );
 102
 103 struct subpicture_sys_t
 104 {
 105     decoder_sys_t *p_dec_sys;
 106     void *p_subs_data;
 107     int i_subs_len;
 108 };
 109
 110 typedef struct
 111 {
 112     int x0;
 113     int y0;
 114     int x1;
 115     int y1;
 116 } rectangle_t;
 117
 118 static int BuildRegions( spu_t *p_spu, rectangle_t *p_region, int i_max_region, ass_image_t *p_img_list, int i_width, int i_height );
 119 static void SubpictureReleaseRegions( spu_t *p_spu, subpicture_t *p_subpic );
 120 static void RegionDraw( subpicture_region_t *p_region, ass_image_t *p_img );
 121
 122 //#define DEBUG_REGION
 123
 124 /*****************************************************************************
 125  * Create: Open libass decoder.
 126  *****************************************************************************/
 127 static int Create( vlc_object_t *p_this )
 128 {
 129     decoder_t *p_dec = (decoder_t *)p_this;
 130     decoder_sys_t *p_sys;
 131     ass_track_t *p_track;
 132
 133     if( p_dec->fmt_in.i_codec != VLC_FOURCC('s','s','a',' ') )
 134         return VLC_EGENERIC;
 135
 136     p_dec->pf_decode_sub = DecodeBlock;
 137
 138     p_dec->p_sys = p_sys = malloc( sizeof( decoder_sys_t ) );
 139     if( !p_sys )
 140         return VLC_ENOMEM;
 141
 142     /* */
 143     p_sys->p_ass = AssHandleYield( p_dec );
 144     if( !p_sys->p_ass )
 145     {
 146         free( p_sys );
 147         return VLC_EGENERIC;
 148     }
 149     vlc_mutex_init( &p_sys->lock );
 150     p_sys->i_refcount = 1;
 151
 152     /* Add a track */
 153     vlc_mutex_lock( p_sys->p_ass->p_lock );
 154     p_sys->p_track = p_track = ass_new_track( p_sys->p_ass->p_library );
 155     if( !p_track )
 156     {
 157         vlc_mutex_unlock( p_sys->p_ass->p_lock );
 158         DecSysRelease( p_sys );
 159         return VLC_EGENERIC;
 160     }
 161     ass_process_codec_private( p_track, p_dec->fmt_in.p_extra, p_dec->fmt_in.i_extra );
 162     vlc_mutex_unlock( p_sys->p_ass->p_lock );
 163
 164     return VLC_SUCCESS;
 165 }
 166
 167 /*****************************************************************************
 168  * Destroy: finish
 169  *****************************************************************************/
 170 static void Destroy( vlc_object_t *p_this )
 171 {
 172     decoder_t *p_dec = (decoder_t *)p_this;
 173
 174     DecSysRelease( p_dec->p_sys );
 175 }
 176
 177 static void DecSysYield( decoder_sys_t *p_sys )
 178 {
 179     vlc_mutex_lock( &p_sys->lock );
 180     p_sys->i_refcount++;
 181     vlc_mutex_unlock( &p_sys->lock );
 182 }
 183 static void DecSysRelease( decoder_sys_t *p_sys )
 184 {
 185     /* */
 186     vlc_mutex_lock( &p_sys->lock );
 187     p_sys->i_refcount--;
 188     if( p_sys->i_refcount > 0 )
 189     {
 190         vlc_mutex_unlock( &p_sys->lock );
 191         return;
 192     }
 193     vlc_mutex_unlock( &p_sys->lock );
 194     vlc_mutex_destroy( &p_sys->lock );
 195
 196     vlc_mutex_lock( p_sys->p_ass->p_lock );
 197     if( p_sys->p_track )
 198         ass_free_track( p_sys->p_track );
 199     vlc_mutex_unlock( p_sys->p_ass->p_lock );
 200
 201     AssHandleRelease( p_sys->p_ass );
 202     free( p_sys );
 203 }
 204
 205 /****************************************************************************
 206  * DecodeBlock:
 207  ****************************************************************************/
 208 static subpicture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
 209 {
 210     decoder_sys_t *p_sys = p_dec->p_sys;
 211
 212     subpicture_t *p_spu = NULL;
 213     block_t *p_block;
 214
 215     if( !pp_block || *pp_block == NULL )
 216         return NULL;
 217
 218     p_block = *pp_block;
 219     if( p_block->i_flags & (BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) )
 220     {
 221         msg_Dbg( p_dec, "Resetting libass track after time discontinuity" );
 222         /* We need to reset our tracks for the time discontinuity to be
 223          * handled */
 224         vlc_mutex_lock( p_sys->p_ass->p_lock );
 225         if( p_sys->p_track )
 226             ass_free_track( p_sys->p_track );
 227
 228         p_sys->p_track = ass_new_track( p_sys->p_ass->p_library );
 229         if( p_sys->p_track  )
 230             ass_process_codec_private( p_sys->p_track,
 231                                        p_dec->fmt_in.p_extra, p_dec->fmt_in.i_extra );
 232         vlc_mutex_unlock( p_sys->p_ass->p_lock );
 233
 234         block_Release( p_block );
 235         return NULL;
 236     }
 237     if( p_block->i_rate != 0 )
 238         p_block->i_length = p_block->i_length * p_block->i_rate / INPUT_RATE_DEFAULT;
 239     *pp_block = NULL;
 240
 241     if( p_block->i_buffer == 0 || p_block->p_buffer[0] == '\0' )
 242     {
 243         block_Release( p_block );
 244         return NULL;
 245     }
 246
 247     p_spu = p_dec->pf_spu_buffer_new( p_dec );
 248     if( !p_spu )
 249     {
 250         msg_Warn( p_dec, "can't get spu buffer" );
 251         block_Release( p_block );
 252         return NULL;
 253     }
 254
 255     p_spu->p_sys = malloc( sizeof( subpicture_sys_t ));
 256     if( !p_spu->p_sys )
 257     {
 258         p_dec->pf_spu_buffer_del( p_dec, p_spu );
 259         block_Release( p_block );
 260         return NULL;
 261     }
 262
 263     p_spu->p_sys->i_subs_len = p_block->i_buffer;
 264     p_spu->p_sys->p_subs_data = malloc( p_block->i_buffer );
 265     if( !p_spu->p_sys->p_subs_data )
 266     {
 267         free( p_spu->p_sys );
 268         p_dec->pf_spu_buffer_del( p_dec, p_spu );
 269         block_Release( p_block );
 270         return NULL;
 271     }
 272     memcpy( p_spu->p_sys->p_subs_data, p_block->p_buffer,
 273             p_block->i_buffer );
 274
 275     p_spu->i_x = 0;
 276     p_spu->i_y = 0;
 277     p_spu->i_start = p_block->i_pts;
 278     p_spu->i_stop = p_block->i_pts + p_block->i_length;
 279     p_spu->b_ephemer = true;
 280     p_spu->b_absolute = true;
 281
 282     vlc_mutex_lock( p_sys->p_ass->p_lock );
 283     if( p_sys->p_track )
 284     {
 285         ass_process_chunk( p_sys->p_track, p_spu->p_sys->p_subs_data, p_spu->p_sys->i_subs_len,
 286                            p_spu->i_start / 1000, (p_spu->i_stop-p_spu->i_start) / 1000 );
 287     }
 288     vlc_mutex_unlock( p_sys->p_ass->p_lock );
 289
 290     p_spu->pf_pre_render = PreRender;
 291     p_spu->pf_update_regions = UpdateRegions;
 292     p_spu->pf_destroy = DestroySubpicture;
 293     p_spu->p_sys->p_dec_sys = p_sys;
 294
 295     DecSysYield( p_sys );
 296
 297     block_Release( p_block );
 298
 299     return p_spu;
 300 }
 301
 302 /****************************************************************************
 303  *
 304  ****************************************************************************/
 305 static void DestroySubpicture( subpicture_t *p_subpic )
 306 {
 307     DecSysRelease( p_subpic->p_sys->p_dec_sys );
 308
 309     free( p_subpic->p_sys->p_subs_data );
 310     free( p_subpic->p_sys );
 311 }
 312
 313 static void PreRender( video_format_t *p_fmt, spu_t *p_spu,
 314                        subpicture_t *p_subpic )
 315 {
 316     decoder_sys_t *p_dec_sys = p_subpic->p_sys->p_dec_sys;
 317
 318     p_dec_sys->p_spu_final = p_subpic;
 319     VLC_UNUSED(p_fmt);
 320     VLC_UNUSED(p_spu);
 321 }
 322
 323 static void UpdateRegions( video_format_t *p_fmt, spu_t *p_spu,
 324                            subpicture_t *p_subpic, mtime_t i_ts )
 325 {
 326     decoder_sys_t *p_sys = p_subpic->p_sys->p_dec_sys;
 327     ass_handle_t *p_ass = p_sys->p_ass;
 328
 329     video_format_t fmt;
 330     bool b_fmt_changed;
 331
 332     if( p_subpic != p_sys->p_spu_final )
 333     {
 334         SubpictureReleaseRegions( p_spu, p_subpic );
 335         return;
 336     }
 337
 338     vlc_mutex_lock( p_ass->p_lock );
 339
 340     /* */
 341     fmt = *p_fmt;
 342     fmt.i_chroma = VLC_FOURCC('R','G','B','A');
 343     fmt.i_width = fmt.i_visible_width;
 344     fmt.i_height = fmt.i_visible_height;
 345     fmt.i_bits_per_pixel = 0;
 346     fmt.i_x_offset = fmt.i_y_offset = 0;
 347     fmt.i_sar_num = 1;
 348     fmt.i_sar_den = 1;
 349
 350     b_fmt_changed = memcmp( &fmt, &p_ass->fmt, sizeof(fmt) ) != 0;
 351     if( b_fmt_changed )
 352     {
 353         ass_set_frame_size( p_ass->p_renderer, fmt.i_width, fmt.i_height );
 354         ass_set_aspect_ratio( p_ass->p_renderer, 1.0 ); // TODO ?
 355
 356         p_ass->fmt = fmt;
 357     }
 358
 359     /* */
 360     int i_changed;
 361     ass_image_t *p_img = ass_render_frame( p_ass->p_renderer, p_sys->p_track, i_ts/1000, &i_changed );
 362
 363     if( !i_changed && !b_fmt_changed )
 364     {
 365         vlc_mutex_unlock( p_ass->p_lock );
 366         return;
 367     }
 368
 369     /* */
 370     p_subpic->i_original_picture_height = fmt.i_height;
 371     p_subpic->i_original_picture_width = fmt.i_width;
 372     SubpictureReleaseRegions( p_spu, p_subpic );
 373
 374     /* XXX to improve efficiency we merge regions that are close minimizing
 375      * the lost surface.
 376      * libass tends to create a lot of small regions and thus spu engine
 377      * reinstanciate a lot the scaler, and as we do not support subpel blending
 378      * it looks ugly (text unaligned).
 379      */
 380     const int i_max_region = 4;
 381     rectangle_t region[i_max_region];
 382     const int i_region = BuildRegions( p_spu, region, i_max_region, p_img, fmt.i_width, fmt.i_height );
 383
 384     if( i_region <= 0 )
 385     {
 386         vlc_mutex_unlock( p_ass->p_lock );
 387         return;
 388     }
 389
 390     /* Allocate the regions and draw them */
 391     subpicture_region_t *pp_region[i_max_region];
 392     subpicture_region_t **pp_region_last = &p_subpic->p_region;
 393
 394     for( int i = 0; i < i_region; i++ )
 395     {
 396         subpicture_region_t *r;
 397         video_format_t fmt_region;
 398
 399         /* */
 400         fmt_region = fmt;
 401         fmt_region.i_width =
 402         fmt_region.i_visible_width  = region[i].x1 - region[i].x0;
 403         fmt_region.i_height =
 404         fmt_region.i_visible_height = region[i].y1 - region[i].y0;
 405
 406         pp_region[i] = r = p_subpic->pf_create_region( VLC_OBJECT(p_spu), &fmt_region );
 407         if( !r )
 408             break;
 409         r->i_x = region[i].x0;
 410         r->i_y = region[i].y0;
 411         r->i_align = SUBPICTURE_ALIGN_TOP | SUBPICTURE_ALIGN_LEFT;
 412
 413         /* */
 414         RegionDraw( r, p_img );
 415
 416         /* */
 417         *pp_region_last = r;
 418         pp_region_last = &r->p_next;
 419     }
 420     vlc_mutex_unlock( p_ass->p_lock );
 421 }
 422
 423 static rectangle_t r_create( int x0, int y0, int x1, int y1 )
 424 {
 425     rectangle_t r = { x0, y0, x1, y1 };
 426     return r;
 427 }
 428 static rectangle_t r_img( const ass_image_t *p_img )
 429 {
 430     return r_create( p_img->dst_x, p_img->dst_y, p_img->dst_x+p_img->w, p_img->dst_y+p_img->h );
 431 }
 432 static void r_add( rectangle_t *r, const rectangle_t *n )
 433 {
 434     r->x0 = __MIN( r->x0, n->x0 );
 435     r->y0 = __MIN( r->y0, n->y0 );
 436     r->x1 = __MAX( r->x1, n->x1 );
 437     r->y1 = __MAX( r->y1, n->y1 );
 438 }
 439 static int r_surface( const rectangle_t *r )
 440 {
 441     return (r->x1-r->x0) * (r->y1-r->y0);
 442 }
 443 static bool r_overlap( const rectangle_t *a, const rectangle_t *b, int i_dx, int i_dy )
 444 {
 445     return  __MAX(a->x0-i_dx, b->x0) < __MIN( a->x1+i_dx, b->x1 ) &&
 446             __MAX(a->y0-i_dy, b->y0) < __MIN( a->y1+i_dy, b->y1 );
 447 }
 448
 449 static int BuildRegions( spu_t *p_spu, rectangle_t *p_region, int i_max_region, ass_image_t *p_img_list, int i_width, int i_height )
 450 {
 451     ass_image_t *p_tmp;
 452     int i_count;
 453
 454     VLC_UNUSED(p_spu);
 455
 456 #ifdef DEBUG_REGION
 457     int64_t i_ck_start = mdate();
 458 #endif
 459
 460     for( p_tmp = p_img_list, i_count = 0; p_tmp != NULL; p_tmp = p_tmp->next )
 461         i_count++;
 462     if( i_count <= 0 )
 463         return 0;
 464
 465     ass_image_t **pp_img = calloc( i_count, sizeof(*pp_img) );
 466     if( !pp_img )
 467         return 0;
 468
 469     for( p_tmp = p_img_list, i_count = 0; p_tmp != NULL; p_tmp = p_tmp->next, i_count++ )
 470         pp_img[i_count] = p_tmp;
 471
 472     /* */
 473     const int i_w_inc = __MAX( ( i_width + 49 ) / 50, 32 );
 474     const int i_h_inc = __MAX( ( i_height + 99 ) / 100, 32 );
 475     int i_maxh = i_w_inc;
 476     int i_maxw = i_h_inc;
 477     int i_region;
 478     rectangle_t region[i_max_region+1];
 479
 480     i_region = 0;
 481     for( int i_used = 0; i_used < i_count; )
 482     {
 483         int n;
 484         for( n = 0; n < i_count; n++ )
 485         {
 486             if( pp_img[n] )
 487                 break;
 488         }
 489         assert( i_region < i_max_region + 1 );
 490         region[i_region++] = r_img( pp_img[n] );
 491         pp_img[n] = NULL; i_used++;
 492
 493         bool b_ok;
 494         do {
 495             b_ok = false;
 496             for( n = 0; n < i_count; n++ )
 497             {
 498                 ass_image_t *p_img = pp_img[n];
 499                 if( !p_img )
 500                     continue;
 501                 rectangle_t r = r_img( p_img );
 502
 503                 int k;
 504                 int i_best = -1;
 505                 int i_best_s = INT_MAX;
 506                 for( k = 0; k < i_region; k++ )
 507                 {
 508                     if( !r_overlap( &region[k], &r, i_maxw, i_maxh ) )
 509                         continue;
 510                     int s = r_surface( &r );
 511                     if( s < i_best_s )
 512                     {
 513                         i_best_s = s;
 514                         i_best = k;
 515                     }
 516                 }
 517                 if( i_best >= 0 )
 518                 {
 519                     r_add( &region[i_best], &r );
 520                     pp_img[n] = NULL; i_used++;
 521                     b_ok = true;
 522                 }
 523             }
 524         } while( b_ok );
 525
 526         if( i_region > i_max_region )
 527         {
 528             int i_best_i = -1;
 529             int i_best_j = -1;
 530             int i_best_ds = INT_MAX;
 531
 532             /* merge best */
 533             for( int i = 0; i < i_region; i++ )
 534             {
 535                 for( int j = i+1; j < i_region; j++ )
 536                 {
 537                     rectangle_t n = region[i];
 538                     r_add( &n, &region[j] );
 539                     int ds = r_surface( &n ) - r_surface( &region[i] ) - r_surface( &region[j] );
 540
 541                     if( ds < i_best_ds )
 542                     {
 543                         i_best_i = i;
 544                         i_best_j = j;
 545                         i_best_ds = ds;
 546                     }
 547                 }
 548             }
 549 #ifdef DEBUG_REGION
 550             msg_Err( p_spu, "Merging %d and %d", i_best_i, i_best_j );
 551 #endif
 552             r_add( &region[i_best_i], &region[i_best_j] );
 553
 554             if( i_best_j+1 < i_region )
 555                 memmove( &region[i_best_j], &region[i_best_j+1], sizeof(*region) * ( i_region - (i_best_j+1)  ) );
 556             i_region--;
 557         }
 558     }
 559
 560     /* */
 561     for( int n = 0; n < i_region; n++ )
 562         p_region[n] = region[n];
 563
 564 #ifdef DEBUG_REGION
 565     int64_t i_ck_time = mdate() - i_ck_start;
 566     msg_Err( p_spu, "ASS: %d objects merged into %d region in %d micros", i_count, i_region, (int)(i_ck_time) );
 567 #endif
 568
 569     free( pp_img );
 570
 571     return i_region;
 572 }
 573
 574 static void RegionDraw( subpicture_region_t *p_region, ass_image_t *p_img )
 575 {
 576     const plane_t *p = &p_region->picture.p[0];
 577     const int i_x = p_region->i_x;
 578     const int i_y = p_region->i_y;
 579     const int i_width  = p_region->fmt.i_width;
 580     const int i_height = p_region->fmt.i_height;
 581
 582     memset( p->p_pixels, 0x00, p->i_pitch * p->i_lines );
 583     for( ; p_img != NULL; p_img = p_img->next )
 584     {
 585         if( p_img->dst_x < i_x || p_img->dst_x + p_img->w > i_x + i_width ||
 586             p_img->dst_y < i_y || p_img->dst_y + p_img->h > i_y + i_height )
 587             continue;
 588
 589         const int r = (p_img->color >> 24)&0xff;
 590         const int g = (p_img->color >> 16)&0xff;
 591         const int b = (p_img->color >>  8)&0xff;
 592         const int a = (p_img->color      )&0xff;
 593         int x, y;
 594
 595         for( y = 0; y < p_img->h; y++ )
 596         {
 597             for( x = 0; x < p_img->w; x++ )
 598             {
 599                 const int alpha = p_img->bitmap[y*p_img->stride+x];
 600                 const int an = (255 - a) * alpha / 255;
 601
 602                 uint8_t *p_rgba = &p->p_pixels[(y+p_img->dst_y-i_y) * p->i_pitch + 4 * (x+p_img->dst_x-i_x)];
 603
 604                 /* Native endianness, but RGBA ordering */
 605                 p_rgba[0] = ( p_rgba[0] * (255-an) + r * an ) / 255;
 606                 p_rgba[1] = ( p_rgba[1] * (255-an) + g * an ) / 255;
 607                 p_rgba[2] = ( p_rgba[2] * (255-an) + b * an ) / 255;
 608                 p_rgba[3] = 255 - ( 255 - p_rgba[3] ) * ( 255 - an ) / 255;
 609             }
 610         }
 611     }
 612
 613 #ifdef DEBUG_REGION
 614     /* XXX Draw a box for debug */
 615 #define P(x,y) ((uint32_t*)&p->p_pixels[(y)*p->i_pitch + 4*(x)])
 616     for( int y = 0; y < p->i_lines; y++ )
 617         *P(0,y) = *P(p->i_visible_pitch/4-1,y) = 0xff000000;
 618     for( int x = 0; x < p->i_visible_pitch; x++ )
 619         *P(x/4,0) = *P(x/4,p->i_visible_lines-1) = 0xff000000;
 620 #undef P
 621 #endif
 622 }
 623
 624
 625 static void SubpictureReleaseRegions( spu_t *p_spu, subpicture_t *p_subpic )
 626 {
 627     while( p_subpic->p_region )
 628     {
 629         subpicture_region_t *p_region = p_subpic->p_region;
 630         p_subpic->p_region = p_region->p_next;
 631         spu_DestroyRegion( p_spu, p_region );
 632     }
 633     p_subpic->p_region = NULL;
 634 }
 635
 636 /* */
 637 static ass_handle_t *AssHandleYield( decoder_t *p_dec )
 638 {
 639     vlc_mutex_t *p_lock = var_AcquireMutex( "libass" );
 640     if( !p_lock )
 641         return NULL;
 642
 643     ass_handle_t *p_ass = NULL;
 644     ass_library_t *p_library = NULL;
 645     ass_renderer_t *p_renderer = NULL;
 646     vlc_value_t val;
 647
 648     var_Create( p_dec->p_libvlc, "libass-handle", VLC_VAR_ADDRESS );
 649     if( var_Get( p_dec->p_libvlc, "libass-handle", &val ) )
 650         val.p_address = NULL;
 651
 652     if( val.p_address )
 653     {
 654         p_ass = val.p_address;
 655
 656         p_ass->i_refcount++;
 657
 658         vlc_mutex_unlock( p_lock );
 659         return p_ass;
 660     }
 661
 662     /* */
 663     p_ass = malloc( sizeof(*p_ass) );
 664     if( !p_ass )
 665         goto error;
 666
 667     /* */
 668     p_ass->p_libvlc = VLC_OBJECT(p_dec->p_libvlc);
 669     p_ass->p_lock = p_lock;
 670     p_ass->i_refcount = 1;
 671
 672     /* Create libass library */
 673     p_ass->p_library = p_library = ass_library_init();
 674     if( !p_library )
 675         goto error;
 676
 677     /* load attachments */
 678     input_attachment_t  **pp_attachments;
 679     int                   i_attachments;
 680
 681     if( decoder_GetInputAttachments( p_dec, &pp_attachments, &i_attachments ))
 682     {
 683         i_attachments = 0;
 684         pp_attachments = NULL;
 685     }
 686     for( int k = 0; k < i_attachments; k++ )
 687     {
 688         input_attachment_t *p_attach = pp_attachments[k];
 689
 690         if( !strcasecmp( p_attach->psz_mime, "application/x-truetype-font" ) )
 691         {
 692             msg_Dbg( p_dec, "adding embedded font %s", p_attach->psz_name );
 693
 694             ass_add_font( p_ass->p_library, p_attach->psz_name, p_attach->p_data, p_attach->i_data );
 695         }
 696         vlc_input_attachment_Delete( p_attach );
 697     }
 698     free( pp_attachments );
 699
 700     char *psz_font_dir = config_GetCacheDir();
 701     if( !psz_font_dir )
 702         goto error;
 703     ass_set_fonts_dir( p_library, psz_font_dir );
 704     free( psz_font_dir );
 705
 706     ass_set_extract_fonts( p_library, true );
 707     ass_set_style_overrides( p_library, NULL );
 708
 709     /* Create the renderer */
 710     p_ass->p_renderer = p_renderer = ass_renderer_init( p_library );
 711     if( !p_renderer )
 712         goto error;
 713
 714     ass_set_use_margins( p_renderer, false);
 715     //if( false )
 716     //    ass_set_margins( p_renderer, int t, int b, int l, int r);
 717     ass_set_hinting( p_renderer, ASS_HINTING_NATIVE ); // No idea
 718     ass_set_font_scale( p_renderer, 1.0 );
 719     ass_set_line_spacing( p_renderer, 0.0 );
 720
 721     const char *psz_font = NULL; /* We don't ship a default font with VLC */
 722     const char *psz_family = "Arial"; /* Use Arial if we can't find anything more suitable */
 723 #ifdef HAVE_FONTCONFIG
 724     ass_set_fonts( p_renderer, psz_font, psz_family );  // setup default font/family
 725 #else
 726     /* FIXME you HAVE to give him a font if no fontconfig */
 727     ass_set_fonts_nofc( p_renderer, psz_font, psz_family );
 728 #endif
 729     memset( &p_ass->fmt, 0, sizeof(p_ass->fmt) );
 730
 731     /* */
 732     val.p_address = p_ass;
 733     var_Set( p_dec->p_libvlc, "libass-handle", val );
 734
 735     /* */
 736     vlc_mutex_unlock( p_ass->p_lock );
 737     return p_ass;
 738
 739 error:
 740     if( p_renderer )
 741         ass_renderer_done( p_renderer );
 742     if( p_library )
 743         ass_library_done( p_library );
 744
 745     free( p_ass );
 746     vlc_mutex_unlock( p_lock );
 747     return NULL;
 748 }
 749 static void AssHandleRelease( ass_handle_t *p_ass )
 750 {
 751     vlc_mutex_lock( p_ass->p_lock );
 752     p_ass->i_refcount--;
 753     if( p_ass->i_refcount > 0 )
 754     {
 755         vlc_mutex_unlock( p_ass->p_lock );
 756         return;
 757     }
 758
 759     ass_renderer_done( p_ass->p_renderer );
 760     ass_library_done( p_ass->p_library );
 761
 762     vlc_value_t val;
 763     val.p_address = NULL;
 764     var_Set( p_ass->p_libvlc, "libass-handle", val );
 765
 766     vlc_mutex_unlock( p_ass->p_lock );
 767     free( p_ass );
 768 }
 769