]> git.sesse.net Git - vlc/blob - modules/codec/subtitles/subsusf.c
54b303194daa182f53ea3248fd6b6715fcc2d8ba
[vlc] / modules / codec / subtitles / subsusf.c
1 /*****************************************************************************
2  * subsusf.c : USF subtitles decoder
3  *****************************************************************************
4  * Copyright (C) 2000-2006 the VideoLAN team
5  * $Id$
6  *
7  * Authors: Bernie Purcell <bitmap@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 #ifdef HAVE_CONFIG_H
24 # include "config.h"
25 #endif
26
27 #include "subsdec.h"
28 #include <vlc_plugin.h>
29 #include <assert.h>
30
31 /*****************************************************************************
32  * Local prototypes
33  *****************************************************************************/
34 static int  OpenDecoder   ( vlc_object_t * );
35 static void CloseDecoder  ( vlc_object_t * );
36
37 static subpicture_t *DecodeBlock   ( decoder_t *, block_t ** );
38 static char         *CreatePlainText( char * );
39 static int           ParseImageAttachments( decoder_t *p_dec );
40
41 static subpicture_t        *ParseText     ( decoder_t *, block_t * );
42 static void                 ParseUSFHeader( decoder_t * );
43 static subpicture_region_t *ParseUSFString( decoder_t *, char * );
44 static subpicture_region_t *LoadEmbeddedImage( decoder_t *p_dec, const char *psz_filename, int i_transparent_color );
45
46 /*****************************************************************************
47  * Module descriptor.
48  *****************************************************************************/
49
50 vlc_module_begin ()
51     set_capability( "decoder", 40 )
52     set_shortname( N_("USFSubs"))
53     set_description( N_("USF subtitles decoder") )
54     set_callbacks( OpenDecoder, CloseDecoder )
55     set_category( CAT_INPUT )
56     set_subcategory( SUBCAT_INPUT_SCODEC )
57     /* We inherit subsdec-align and subsdec-formatted from subsdec.c */
58 vlc_module_end ()
59
60 /*****************************************************************************
61  * OpenDecoder: probe the decoder and return score
62  *****************************************************************************
63  * Tries to launch a decoder and return score so that the interface is able
64  * to chose.
65  *****************************************************************************/
66 static int OpenDecoder( vlc_object_t *p_this )
67 {
68     decoder_t     *p_dec = (decoder_t*)p_this;
69     decoder_sys_t *p_sys;
70
71     if( p_dec->fmt_in.i_codec != VLC_CODEC_USF )
72         return VLC_EGENERIC;
73
74     /* Allocate the memory needed to store the decoder's structure */
75     if( ( p_dec->p_sys = p_sys = calloc(1, sizeof(decoder_sys_t)) ) == NULL )
76         return VLC_ENOMEM;
77
78     p_dec->pf_decode_sub = DecodeBlock;
79     p_dec->fmt_out.i_cat = SPU_ES;
80     p_dec->fmt_out.i_codec = 0;
81
82     /* Unused fields of p_sys - not needed for USF decoding */
83     p_sys->b_ass = false;
84     p_sys->iconv_handle = (vlc_iconv_t)-1;
85     p_sys->b_autodetect_utf8 = false;
86
87     /* init of p_sys */
88     p_sys->i_align = 0;
89     p_sys->i_original_height = 0;
90     p_sys->i_original_width = 0;
91     TAB_INIT( p_sys->i_ssa_styles, p_sys->pp_ssa_styles );
92     TAB_INIT( p_sys->i_images, p_sys->pp_images );
93
94     /* USF subtitles are mandated to be UTF-8, so don't need vlc_iconv */
95
96     p_sys->i_align = var_CreateGetInteger( p_dec, "subsdec-align" );
97
98     ParseImageAttachments( p_dec );
99
100     if( var_CreateGetBool( p_dec, "subsdec-formatted" ) )
101     {
102         if( p_dec->fmt_in.i_extra > 0 )
103             ParseUSFHeader( p_dec );
104     }
105
106     return VLC_SUCCESS;
107 }
108
109 /****************************************************************************
110  * DecodeBlock: the whole thing
111  ****************************************************************************
112  * This function must be fed with complete subtitles units.
113  ****************************************************************************/
114 static subpicture_t *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
115 {
116     subpicture_t *p_spu;
117     block_t *p_block;
118
119     if( !pp_block || *pp_block == NULL )
120         return NULL;
121
122     p_block = *pp_block;
123
124     p_spu = ParseText( p_dec, p_block );
125
126     block_Release( p_block );
127     *pp_block = NULL;
128
129     return p_spu;
130 }
131
132 /*****************************************************************************
133  * CloseDecoder: clean up the decoder
134  *****************************************************************************/
135 static void CloseDecoder( vlc_object_t *p_this )
136 {
137     decoder_t *p_dec = (decoder_t *)p_this;
138     decoder_sys_t *p_sys = p_dec->p_sys;
139
140     if( p_sys->pp_ssa_styles )
141     {
142         int i;
143         for( i = 0; i < p_sys->i_ssa_styles; i++ )
144         {
145             if( !p_sys->pp_ssa_styles[i] )
146                 continue;
147
148             free( p_sys->pp_ssa_styles[i]->psz_stylename );
149             //FIXME: Make font_style a pointer and use text_style_* functions
150             free( p_sys->pp_ssa_styles[i]->font_style.psz_fontname );
151             free( p_sys->pp_ssa_styles[i] );
152         }
153         TAB_CLEAN( p_sys->i_ssa_styles, p_sys->pp_ssa_styles );
154     }
155     if( p_sys->pp_images )
156     {
157         int i;
158         for( i = 0; i < p_sys->i_images; i++ )
159         {
160             if( !p_sys->pp_images[i] )
161                 continue;
162
163             if( p_sys->pp_images[i]->p_pic )
164                 picture_Release( p_sys->pp_images[i]->p_pic );
165             free( p_sys->pp_images[i]->psz_filename );
166
167             free( p_sys->pp_images[i] );
168         }
169         TAB_CLEAN( p_sys->i_images, p_sys->pp_images );
170     }
171
172     free( p_sys );
173 }
174
175 /*****************************************************************************
176  * ParseText: parse an text subtitle packet and send it to the video output
177  *****************************************************************************/
178 static subpicture_t *ParseText( decoder_t *p_dec, block_t *p_block )
179 {
180     decoder_sys_t *p_sys = p_dec->p_sys;
181     subpicture_t *p_spu = NULL;
182     char *psz_subtitle = NULL;
183
184     /* We cannot display a subpicture with no date */
185     if( p_block->i_pts == 0 )
186     {
187         msg_Warn( p_dec, "subtitle without a date" );
188         return NULL;
189     }
190
191     /* Check validity of packet data */
192     /* An "empty" line containing only \0 can be used to force
193        and ephemer picture from the screen */
194     if( p_block->i_buffer < 1 )
195     {
196         msg_Warn( p_dec, "no subtitle data" );
197         return NULL;
198     }
199
200     /* Should be resiliant against bad subtitles */
201     psz_subtitle = strndup( (const char *)p_block->p_buffer,
202                             p_block->i_buffer );
203     if( psz_subtitle == NULL )
204         return NULL;
205
206     /* USF Subtitles are mandated to be UTF-8 -- make sure it is */
207     if (EnsureUTF8( psz_subtitle ) == NULL)
208     {
209         msg_Err( p_dec, "USF subtitles must be in UTF-8 format.\n"
210                  "This stream contains USF subtitles which aren't." );
211     }
212
213     /* Create the subpicture unit */
214     p_spu = decoder_NewSubpicture( p_dec );
215     if( !p_spu )
216     {
217         msg_Warn( p_dec, "can't get spu buffer" );
218         free( psz_subtitle );
219         return NULL;
220     }
221
222     /* Decode USF strings */
223     p_spu->p_region = ParseUSFString( p_dec, psz_subtitle );
224
225     p_spu->i_start = p_block->i_pts;
226     p_spu->i_stop = p_block->i_pts + p_block->i_length;
227     p_spu->b_ephemer = (p_block->i_length == 0);
228     p_spu->b_absolute = false;
229     p_spu->i_original_picture_width = p_sys->i_original_width;
230     p_spu->i_original_picture_height = p_sys->i_original_height;
231
232     free( psz_subtitle );
233
234     return p_spu;
235 }
236
237 static char *GrabAttributeValue( const char *psz_attribute,
238                                  const char *psz_tag_start )
239 {
240     if( psz_attribute && psz_tag_start )
241     {
242         char *psz_tag_end = strchr( psz_tag_start, '>' );
243         char *psz_found   = strcasestr( psz_tag_start, psz_attribute );
244
245         if( psz_found )
246         {
247             psz_found += strlen( psz_attribute );
248
249             if(( *(psz_found++) == '=' ) &&
250                ( *(psz_found++) == '\"' ))
251             {
252                 if( psz_found < psz_tag_end )
253                 {
254                     int   i_len = strcspn( psz_found, "\"" );
255                     return strndup( psz_found, i_len );
256                 }
257             }
258         }
259     }
260     return NULL;
261 }
262
263 static ssa_style_t *ParseStyle( decoder_sys_t *p_sys, char *psz_subtitle )
264 {
265     ssa_style_t *p_ssa_style = NULL;
266     char        *psz_style = GrabAttributeValue( "style", psz_subtitle );
267
268     if( psz_style )
269     {
270         int i;
271
272         for( i = 0; i < p_sys->i_ssa_styles; i++ )
273         {
274             if( !strcmp( p_sys->pp_ssa_styles[i]->psz_stylename, psz_style ) )
275                 p_ssa_style = p_sys->pp_ssa_styles[i];
276         }
277         free( psz_style );
278     }
279     return p_ssa_style;
280 }
281
282 static int ParsePositionAttributeList( char *psz_subtitle, int *i_align,
283                                        int *i_x, int *i_y )
284 {
285     int   i_mask = 0;
286
287     char *psz_align    = GrabAttributeValue( "alignment", psz_subtitle );
288     char *psz_margin_x = GrabAttributeValue( "horizontal-margin", psz_subtitle );
289     char *psz_margin_y = GrabAttributeValue( "vertical-margin", psz_subtitle );
290     /* -- UNSUPPORTED
291     char *psz_relative = GrabAttributeValue( "relative-to", psz_subtitle );
292     char *psz_rotate_x = GrabAttributeValue( "rotate-x", psz_subtitle );
293     char *psz_rotate_y = GrabAttributeValue( "rotate-y", psz_subtitle );
294     char *psz_rotate_z = GrabAttributeValue( "rotate-z", psz_subtitle );
295     */
296
297     *i_align = SUBPICTURE_ALIGN_BOTTOM;
298     *i_x = 0;
299     *i_y = 0;
300
301     if( psz_align )
302     {
303         if( !strcasecmp( "TopLeft", psz_align ) )
304             *i_align = SUBPICTURE_ALIGN_TOP | SUBPICTURE_ALIGN_LEFT;
305         else if( !strcasecmp( "TopCenter", psz_align ) )
306             *i_align = SUBPICTURE_ALIGN_TOP;
307         else if( !strcasecmp( "TopRight", psz_align ) )
308             *i_align = SUBPICTURE_ALIGN_TOP | SUBPICTURE_ALIGN_RIGHT;
309         else if( !strcasecmp( "MiddleLeft", psz_align ) )
310             *i_align = SUBPICTURE_ALIGN_LEFT;
311         else if( !strcasecmp( "MiddleCenter", psz_align ) )
312             *i_align = 0;
313         else if( !strcasecmp( "MiddleRight", psz_align ) )
314             *i_align = SUBPICTURE_ALIGN_RIGHT;
315         else if( !strcasecmp( "BottomLeft", psz_align ) )
316             *i_align = SUBPICTURE_ALIGN_BOTTOM | SUBPICTURE_ALIGN_LEFT;
317         else if( !strcasecmp( "BottomCenter", psz_align ) )
318             *i_align = SUBPICTURE_ALIGN_BOTTOM;
319         else if( !strcasecmp( "BottomRight", psz_align ) )
320             *i_align = SUBPICTURE_ALIGN_BOTTOM | SUBPICTURE_ALIGN_RIGHT;
321
322         i_mask |= ATTRIBUTE_ALIGNMENT;
323         free( psz_align );
324     }
325     if( psz_margin_x )
326     {
327         *i_x = atoi( psz_margin_x );
328         if( strchr( psz_margin_x, '%' ) )
329             i_mask |= ATTRIBUTE_X_PERCENT;
330         else
331             i_mask |= ATTRIBUTE_X;
332
333         free( psz_margin_x );
334     }
335     if( psz_margin_y )
336     {
337         *i_y = atoi( psz_margin_y );
338         if( strchr( psz_margin_y, '%' ) )
339             i_mask |= ATTRIBUTE_Y_PERCENT;
340         else
341             i_mask |= ATTRIBUTE_Y;
342
343         free( psz_margin_y );
344     }
345     return i_mask;
346 }
347
348 static void SetupPositions( subpicture_region_t *p_region, char *psz_subtitle )
349 {
350     int           i_mask = 0;
351     int           i_align;
352     int           i_x, i_y;
353
354     i_mask = ParsePositionAttributeList( psz_subtitle, &i_align, &i_x, &i_y );
355
356     if( i_mask & ATTRIBUTE_ALIGNMENT )
357         p_region->i_align = i_align;
358
359     /* TODO: Setup % based offsets properly, without adversely affecting
360      *       everything else in vlc. Will address with separate patch, to
361      *       prevent this one being any more complicated.
362      */
363     if( i_mask & ATTRIBUTE_X )
364         p_region->i_x = i_x;
365     else if( i_mask & ATTRIBUTE_X_PERCENT )
366         p_region->i_x = 0;
367
368     if( i_mask & ATTRIBUTE_Y )
369         p_region->i_y = i_y;
370     else if( i_mask & ATTRIBUTE_Y_PERCENT )
371         p_region->i_y = 0;
372 }
373
374 static subpicture_region_t *CreateTextRegion( decoder_t *p_dec,
375                                               char *psz_subtitle,
376                                               int i_len,
377                                               int i_sys_align )
378 {
379     decoder_sys_t        *p_sys = p_dec->p_sys;
380     subpicture_region_t  *p_text_region;
381     video_format_t        fmt;
382
383     /* Create a new subpicture region */
384     memset( &fmt, 0, sizeof(video_format_t) );
385     fmt.i_chroma = VLC_CODEC_TEXT;
386     fmt.i_aspect = 0;
387     fmt.i_width = fmt.i_height = 0;
388     fmt.i_x_offset = fmt.i_y_offset = 0;
389     p_text_region = subpicture_region_New( &fmt );
390
391     if( p_text_region != NULL )
392     {
393         ssa_style_t  *p_ssa_style = NULL;
394
395         p_text_region->psz_text = NULL;
396         p_text_region->psz_html = strndup( psz_subtitle, i_len );
397         if( ! p_text_region->psz_html )
398         {
399             subpicture_region_Delete( p_text_region );
400             return NULL;
401         }
402
403         p_ssa_style = ParseStyle( p_sys, p_text_region->psz_html );
404         if( !p_ssa_style )
405         {
406             int i;
407
408             for( i = 0; i < p_sys->i_ssa_styles; i++ )
409             {
410                 if( !strcasecmp( p_sys->pp_ssa_styles[i]->psz_stylename, "Default" ) )
411                     p_ssa_style = p_sys->pp_ssa_styles[i];
412             }
413         }
414
415         if( p_ssa_style )
416         {
417             msg_Dbg( p_dec, "style is: %s", p_ssa_style->psz_stylename );
418
419             p_text_region->p_style = text_style_Duplicate( &p_ssa_style->font_style );
420             p_text_region->i_align = p_ssa_style->i_align;
421
422             /* TODO: Setup % based offsets properly, without adversely affecting
423              *       everything else in vlc. Will address with separate patch,
424              *       to prevent this one being any more complicated.
425
426                      * p_ssa_style->i_margin_percent_h;
427                      * p_ssa_style->i_margin_percent_v;
428              */
429             p_text_region->i_x         = p_ssa_style->i_margin_h;
430             p_text_region->i_y         = p_ssa_style->i_margin_v;
431
432         }
433         else
434         {
435             p_text_region->i_align = SUBPICTURE_ALIGN_BOTTOM | i_sys_align;
436             p_text_region->i_x = i_sys_align ? 20 : 0;
437             p_text_region->i_y = 10;
438         }
439         /* Look for position arguments which may override the style-based
440          * defaults.
441          */
442         SetupPositions( p_text_region, psz_subtitle );
443
444         p_text_region->p_next = NULL;
445     }
446     return p_text_region;
447 }
448
449 static int ParseImageAttachments( decoder_t *p_dec )
450 {
451     decoder_sys_t        *p_sys = p_dec->p_sys;
452     input_attachment_t  **pp_attachments;
453     int                   i_attachments_cnt;
454     int                   k = 0;
455
456     if( VLC_SUCCESS != decoder_GetInputAttachments( p_dec, &pp_attachments, &i_attachments_cnt ))
457         return VLC_EGENERIC;
458
459     for( k = 0; k < i_attachments_cnt; k++ )
460     {
461         input_attachment_t *p_attach = pp_attachments[k];
462
463         vlc_fourcc_t type = image_Mime2Fourcc( p_attach->psz_mime );
464
465         if( ( type != 0 ) &&
466             ( p_attach->i_data > 0 ) &&
467             ( p_attach->p_data != NULL ) )
468         {
469             picture_t         *p_pic = NULL;
470             image_handler_t   *p_image;
471
472             p_image = image_HandlerCreate( p_dec );
473             if( p_image != NULL )
474             {
475                 block_t   *p_block;
476
477                 p_block = block_New( p_image->p_parent, p_attach->i_data );
478
479                 if( p_block != NULL )
480                 {
481                     video_format_t     fmt_in;
482                     video_format_t     fmt_out;
483
484                     memcpy( p_block->p_buffer, p_attach->p_data, p_attach->i_data );
485
486                     memset( &fmt_in,  0, sizeof( video_format_t));
487                     memset( &fmt_out, 0, sizeof( video_format_t));
488
489                     fmt_in.i_chroma  = type;
490                     fmt_out.i_chroma = VLC_CODEC_YUVA;
491
492                     /* Find a suitable decoder module */
493                     if( module_exists( "sdl_image" ) )
494                     {
495                         /* ffmpeg thinks it can handle bmp properly but it can't (at least
496                          * not all of them), so use sdl_image if it is available */
497
498                         vlc_value_t val;
499
500                         var_Create( p_dec, "codec", VLC_VAR_MODULE | VLC_VAR_DOINHERIT );
501                         val.psz_string = (char*) "sdl_image";
502                         var_Set( p_dec, "codec", val );
503                     }
504
505                     p_pic = image_Read( p_image, p_block, &fmt_in, &fmt_out );
506                     var_Destroy( p_dec, "codec" );
507                 }
508
509                 image_HandlerDelete( p_image );
510             }
511             if( p_pic )
512             {
513                 image_attach_t *p_picture = malloc( sizeof(image_attach_t) );
514
515                 if( p_picture )
516                 {
517                     p_picture->psz_filename = strdup( p_attach->psz_name );
518                     p_picture->p_pic = p_pic;
519
520                     TAB_APPEND( p_sys->i_images, p_sys->pp_images, p_picture );
521                 }
522             }
523         }
524         vlc_input_attachment_Delete( pp_attachments[ k ] );
525     }
526     free( pp_attachments );
527
528     return VLC_SUCCESS;
529 }
530
531 static void ParseUSFHeaderTags( decoder_t *p_dec, xml_reader_t *p_xml_reader )
532 {
533     decoder_sys_t *p_sys = p_dec->p_sys;
534     char *psz_node;
535     ssa_style_t *p_ssa_style = NULL;
536     int i_style_level = 0;
537     int i_metadata_level = 0;
538
539     while ( xml_ReaderRead( p_xml_reader ) == 1 )
540     {
541         switch ( xml_ReaderNodeType( p_xml_reader ) )
542         {
543             case XML_READER_TEXT:
544             case XML_READER_NONE:
545                 break;
546             case XML_READER_ENDELEM:
547                 psz_node = xml_ReaderName( p_xml_reader );
548
549                 if( !psz_node )
550                     break;
551                 switch (i_style_level)
552                 {
553                     case 0:
554                         if( !strcasecmp( "metadata", psz_node ) && (i_metadata_level == 1) )
555                         {
556                             i_metadata_level--;
557                         }
558                         break;
559                     case 1:
560                         if( !strcasecmp( "styles", psz_node ) )
561                         {
562                             i_style_level--;
563                         }
564                         break;
565                     case 2:
566                         if( !strcasecmp( "style", psz_node ) )
567                         {
568                             TAB_APPEND( p_sys->i_ssa_styles, p_sys->pp_ssa_styles, p_ssa_style );
569
570                             p_ssa_style = NULL;
571                             i_style_level--;
572                         }
573                         break;
574                 }
575
576                 free( psz_node );
577                 break;
578             case XML_READER_STARTELEM:
579                 psz_node = xml_ReaderName( p_xml_reader );
580
581                 if( !psz_node )
582                     break;
583
584                 if( !strcasecmp( "metadata", psz_node ) && (i_style_level == 0) )
585                 {
586                     i_metadata_level++;
587                 }
588                 else if( !strcasecmp( "resolution", psz_node ) &&
589                          ( i_metadata_level == 1) )
590                 {
591                     while ( xml_ReaderNextAttr( p_xml_reader ) == VLC_SUCCESS )
592                     {
593                         char *psz_name = xml_ReaderName ( p_xml_reader );
594                         char *psz_value = xml_ReaderValue ( p_xml_reader );
595
596                         if( psz_name && psz_value )
597                         {
598                             if( !strcasecmp( "x", psz_name ) )
599                                 p_sys->i_original_width = atoi( psz_value );
600                             else if( !strcasecmp( "y", psz_name ) )
601                                 p_sys->i_original_height = atoi( psz_value );
602                         }
603                         free( psz_name );
604                         free( psz_value );
605                     }
606                 }
607                 else if( !strcasecmp( "styles", psz_node ) && (i_style_level == 0) )
608                 {
609                     i_style_level++;
610                 }
611                 else if( !strcasecmp( "style", psz_node ) && (i_style_level == 1) )
612                 {
613                     i_style_level++;
614
615                     p_ssa_style = calloc( 1, sizeof(ssa_style_t) );
616                     if( !p_ssa_style )
617                     {
618                         free( psz_node );
619                         return;
620                     }
621                     /* All styles are supposed to default to Default, and then
622                      * one or more settings are over-ridden.
623                      * At the moment this only effects styles defined AFTER
624                      * Default in the XML
625                      */
626                     int i;
627                     for( i = 0; i < p_sys->i_ssa_styles; i++ )
628                     {
629                         if( !strcasecmp( p_sys->pp_ssa_styles[i]->psz_stylename, "Default" ) )
630                         {
631                             ssa_style_t *p_default_style = p_sys->pp_ssa_styles[i];
632
633                             memcpy( p_ssa_style, p_default_style, sizeof( ssa_style_t ) );
634                             //FIXME: Make font_style a pointer. Actually we double copy some data here,
635                             //   we use text_style_Copy to avoid copying psz_fontname, though .
636                             text_style_Copy( &p_ssa_style->font_style, &p_default_style->font_style );
637                             p_ssa_style->psz_stylename = NULL;
638                         }
639                     }
640
641                     while ( xml_ReaderNextAttr( p_xml_reader ) == VLC_SUCCESS )
642                     {
643                         char *psz_name = xml_ReaderName ( p_xml_reader );
644                         char *psz_value = xml_ReaderValue ( p_xml_reader );
645
646                         if( psz_name && psz_value )
647                         {
648                             if( !strcasecmp( "name", psz_name ) )
649                                 p_ssa_style->psz_stylename = strdup( psz_value );
650                         }
651                         free( psz_name );
652                         free( psz_value );
653                     }
654                 }
655                 else if( !strcasecmp( "fontstyle", psz_node ) && (i_style_level == 2) )
656                 {
657                     while ( xml_ReaderNextAttr( p_xml_reader ) == VLC_SUCCESS )
658                     {
659                         char *psz_name = xml_ReaderName ( p_xml_reader );
660                         char *psz_value = xml_ReaderValue ( p_xml_reader );
661
662                         if( psz_name && psz_value )
663                         {
664                             if( !strcasecmp( "face", psz_name ) )
665                             {
666                                 free( p_ssa_style->font_style.psz_fontname );
667                                 p_ssa_style->font_style.psz_fontname = strdup( psz_value );
668                             }
669                             else if( !strcasecmp( "size", psz_name ) )
670                             {
671                                 if( ( *psz_value == '+' ) || ( *psz_value == '-' ) )
672                                 {
673                                     int i_value = atoi( psz_value );
674
675                                     if( ( i_value >= -5 ) && ( i_value <= 5 ) )
676                                         p_ssa_style->font_style.i_font_size  +=
677                                             ( i_value * p_ssa_style->font_style.i_font_size ) / 10;
678                                     else if( i_value < -5 )
679                                         p_ssa_style->font_style.i_font_size  = - i_value;
680                                     else if( i_value > 5 )
681                                         p_ssa_style->font_style.i_font_size  = i_value;
682                                 }
683                                 else
684                                     p_ssa_style->font_style.i_font_size  = atoi( psz_value );
685                             }
686                             else if( !strcasecmp( "italic", psz_name ) )
687                             {
688                                 if( !strcasecmp( "yes", psz_value ))
689                                     p_ssa_style->font_style.i_style_flags |= STYLE_ITALIC;
690                                 else
691                                     p_ssa_style->font_style.i_style_flags &= ~STYLE_ITALIC;
692                             }
693                             else if( !strcasecmp( "weight", psz_name ) )
694                             {
695                                 if( !strcasecmp( "bold", psz_value ))
696                                     p_ssa_style->font_style.i_style_flags |= STYLE_BOLD;
697                                 else
698                                     p_ssa_style->font_style.i_style_flags &= ~STYLE_BOLD;
699                             }
700                             else if( !strcasecmp( "underline", psz_name ) )
701                             {
702                                 if( !strcasecmp( "yes", psz_value ))
703                                     p_ssa_style->font_style.i_style_flags |= STYLE_UNDERLINE;
704                                 else
705                                     p_ssa_style->font_style.i_style_flags &= ~STYLE_UNDERLINE;
706                             }
707                             else if( !strcasecmp( "color", psz_name ) )
708                             {
709                                 if( *psz_value == '#' )
710                                 {
711                                     unsigned long col = strtol(psz_value+1, NULL, 16);
712                                     p_ssa_style->font_style.i_font_color = (col & 0x00ffffff);
713                                     p_ssa_style->font_style.i_font_alpha = (col >> 24) & 0xff;
714                                 }
715                             }
716                             else if( !strcasecmp( "outline-color", psz_name ) )
717                             {
718                                 if( *psz_value == '#' )
719                                 {
720                                     unsigned long col = strtol(psz_value+1, NULL, 16);
721                                     p_ssa_style->font_style.i_outline_color = (col & 0x00ffffff);
722                                     p_ssa_style->font_style.i_outline_alpha = (col >> 24) & 0xff;
723                                 }
724                             }
725                             else if( !strcasecmp( "outline-level", psz_name ) )
726                             {
727                                 p_ssa_style->font_style.i_outline_width = atoi( psz_value );
728                             }
729                             else if( !strcasecmp( "shadow-color", psz_name ) )
730                             {
731                                 if( *psz_value == '#' )
732                                 {
733                                     unsigned long col = strtol(psz_value+1, NULL, 16);
734                                     p_ssa_style->font_style.i_shadow_color = (col & 0x00ffffff);
735                                     p_ssa_style->font_style.i_shadow_alpha = (col >> 24) & 0xff;
736                                 }
737                             }
738                             else if( !strcasecmp( "shadow-level", psz_name ) )
739                             {
740                                 p_ssa_style->font_style.i_shadow_width = atoi( psz_value );
741                             }
742                             else if( !strcasecmp( "back-color", psz_name ) )
743                             {
744                                 if( *psz_value == '#' )
745                                 {
746                                     unsigned long col = strtol(psz_value+1, NULL, 16);
747                                     p_ssa_style->font_style.i_karaoke_background_color = (col & 0x00ffffff);
748                                     p_ssa_style->font_style.i_karaoke_background_alpha = (col >> 24) & 0xff;
749                                 }
750                             }
751                             else if( !strcasecmp( "spacing", psz_name ) )
752                             {
753                                 p_ssa_style->font_style.i_spacing = atoi( psz_value );
754                             }
755                         }
756                         free( psz_name );
757                         free( psz_value );
758                     }
759                 }
760                 else if( !strcasecmp( "position", psz_node ) && (i_style_level == 2) )
761                 {
762                     while ( xml_ReaderNextAttr( p_xml_reader ) == VLC_SUCCESS )
763                     {
764                         char *psz_name = xml_ReaderName ( p_xml_reader );
765                         char *psz_value = xml_ReaderValue ( p_xml_reader );
766
767                         if( psz_name && psz_value )
768                         {
769                             if( !strcasecmp( "alignment", psz_name ) )
770                             {
771                                 if( !strcasecmp( "TopLeft", psz_value ) )
772                                     p_ssa_style->i_align = SUBPICTURE_ALIGN_TOP | SUBPICTURE_ALIGN_LEFT;
773                                 else if( !strcasecmp( "TopCenter", psz_value ) )
774                                     p_ssa_style->i_align = SUBPICTURE_ALIGN_TOP;
775                                 else if( !strcasecmp( "TopRight", psz_value ) )
776                                     p_ssa_style->i_align = SUBPICTURE_ALIGN_TOP | SUBPICTURE_ALIGN_RIGHT;
777                                 else if( !strcasecmp( "MiddleLeft", psz_value ) )
778                                     p_ssa_style->i_align = SUBPICTURE_ALIGN_LEFT;
779                                 else if( !strcasecmp( "MiddleCenter", psz_value ) )
780                                     p_ssa_style->i_align = 0;
781                                 else if( !strcasecmp( "MiddleRight", psz_value ) )
782                                     p_ssa_style->i_align = SUBPICTURE_ALIGN_RIGHT;
783                                 else if( !strcasecmp( "BottomLeft", psz_value ) )
784                                     p_ssa_style->i_align = SUBPICTURE_ALIGN_BOTTOM | SUBPICTURE_ALIGN_LEFT;
785                                 else if( !strcasecmp( "BottomCenter", psz_value ) )
786                                     p_ssa_style->i_align = SUBPICTURE_ALIGN_BOTTOM;
787                                 else if( !strcasecmp( "BottomRight", psz_value ) )
788                                     p_ssa_style->i_align = SUBPICTURE_ALIGN_BOTTOM | SUBPICTURE_ALIGN_RIGHT;
789                             }
790                             else if( !strcasecmp( "horizontal-margin", psz_name ) )
791                             {
792                                 if( strchr( psz_value, '%' ) )
793                                 {
794                                     p_ssa_style->i_margin_h = 0;
795                                     p_ssa_style->i_margin_percent_h = atoi( psz_value );
796                                 }
797                                 else
798                                 {
799                                     p_ssa_style->i_margin_h = atoi( psz_value );
800                                     p_ssa_style->i_margin_percent_h = 0;
801                                 }
802                             }
803                             else if( !strcasecmp( "vertical-margin", psz_name ) )
804                             {
805                                 if( strchr( psz_value, '%' ) )
806                                 {
807                                     p_ssa_style->i_margin_v = 0;
808                                     p_ssa_style->i_margin_percent_v = atoi( psz_value );
809                                 }
810                                 else
811                                 {
812                                     p_ssa_style->i_margin_v = atoi( psz_value );
813                                     p_ssa_style->i_margin_percent_v = 0;
814                                 }
815                             }
816                         }
817                         free( psz_name );
818                         free( psz_value );
819                     }
820                 }
821
822                 free( psz_node );
823                 break;
824         }
825     }
826     free( p_ssa_style );
827 }
828
829
830
831 static subpicture_region_t *ParseUSFString( decoder_t *p_dec,
832                                             char *psz_subtitle )
833 {
834     decoder_sys_t        *p_sys = p_dec->p_sys;
835     subpicture_region_t  *p_region_first = NULL;
836     subpicture_region_t  *p_region_upto  = p_region_first;
837
838     while( *psz_subtitle )
839     {
840         if( *psz_subtitle == '<' )
841         {
842             char *psz_end = NULL;
843
844             if(( !strncasecmp( psz_subtitle, "<text ", 6 )) ||
845                ( !strncasecmp( psz_subtitle, "<text>", 6 )))
846             {
847                 psz_end = strcasestr( psz_subtitle, "</text>" );
848
849                 if( psz_end )
850                 {
851                     subpicture_region_t  *p_text_region;
852
853                     psz_end += strcspn( psz_end, ">" ) + 1;
854
855                     p_text_region = CreateTextRegion( p_dec,
856                                                       psz_subtitle,
857                                                       psz_end - psz_subtitle,
858                                                       p_sys->i_align );
859
860                     if( p_text_region )
861                     {
862                         p_text_region->psz_text = CreatePlainText( p_text_region->psz_html );
863
864                         if( ! var_CreateGetBool( p_dec, "subsdec-formatted" ) )
865                         {
866                             free( p_text_region->psz_html );
867                             p_text_region->psz_html = NULL;
868                         }
869                     }
870
871                     if( !p_region_first )
872                     {
873                         p_region_first = p_region_upto = p_text_region;
874                     }
875                     else if( p_text_region )
876                     {
877                         p_region_upto->p_next = p_text_region;
878                         p_region_upto = p_region_upto->p_next;
879                     }
880                 }
881             }
882             else if(( !strncasecmp( psz_subtitle, "<karaoke ", 9 )) ||
883                     ( !strncasecmp( psz_subtitle, "<karaoke>", 9 )))
884             {
885                 psz_end = strcasestr( psz_subtitle, "</karaoke>" );
886
887                 if( psz_end )
888                 {
889                     subpicture_region_t  *p_text_region;
890
891                     psz_end += strcspn( psz_end, ">" ) + 1;
892
893                     p_text_region = CreateTextRegion( p_dec,
894                                                       psz_subtitle,
895                                                       psz_end - psz_subtitle,
896                                                       p_sys->i_align );
897
898                     if( p_text_region )
899                     {
900                         if( ! var_CreateGetBool( p_dec, "subsdec-formatted" ) )
901                         {
902                             free( p_text_region->psz_html );
903                             p_text_region->psz_html = NULL;
904                         }
905                     }
906                     if( !p_region_first )
907                     {
908                         p_region_first = p_region_upto = p_text_region;
909                     }
910                     else if( p_text_region )
911                     {
912                         p_region_upto->p_next = p_text_region;
913                         p_region_upto = p_region_upto->p_next;
914                     }
915                 }
916             }
917             else if(( !strncasecmp( psz_subtitle, "<image ", 7 )) ||
918                     ( !strncasecmp( psz_subtitle, "<image>", 7 )))
919             {
920                 subpicture_region_t *p_image_region = NULL;
921
922                 char *psz_end = strcasestr( psz_subtitle, "</image>" );
923                 char *psz_content = strchr( psz_subtitle, '>' );
924                 int   i_transparent = -1;
925
926                 /* If a colorkey parameter is specified, then we have to map
927                  * that index in the picture through as transparent (it is
928                  * required by the USF spec but is also recommended that if the
929                  * creator really wants a transparent colour that they use a
930                  * type like PNG that properly supports it; this goes doubly
931                  * for VLC because the pictures are stored internally in YUV
932                  * and the resulting colour-matching may not produce the
933                  * desired results.)
934                  */
935                 char *psz_tmp = GrabAttributeValue( "colorkey", psz_subtitle );
936                 if( psz_tmp )
937                 {
938                     if( *psz_tmp == '#' )
939                         i_transparent = strtol( psz_tmp + 1, NULL, 16 ) & 0x00ffffff;
940                     free( psz_tmp );
941                 }
942                 if( psz_content && ( psz_content < psz_end ) )
943                 {
944                     char *psz_filename = strndup( &psz_content[1], psz_end - &psz_content[1] );
945                     if( psz_filename )
946                     {
947                         p_image_region = LoadEmbeddedImage( p_dec,
948                                             psz_filename, i_transparent );
949                         free( psz_filename );
950                     }
951                 }
952
953                 if( psz_end ) psz_end += strcspn( psz_end, ">" ) + 1;
954
955                 if( p_image_region )
956                 {
957                     SetupPositions( p_image_region, psz_subtitle );
958
959                     p_image_region->p_next   = NULL;
960                     p_image_region->psz_text = NULL;
961                     p_image_region->psz_html = NULL;
962
963                 }
964                 if( !p_region_first )
965                 {
966                     p_region_first = p_region_upto = p_image_region;
967                 }
968                 else if( p_image_region )
969                 {
970                     p_region_upto->p_next = p_image_region;
971                     p_region_upto = p_region_upto->p_next;
972                 }
973             }
974             if( psz_end )
975                 psz_subtitle = psz_end - 1;
976
977             psz_subtitle += strcspn( psz_subtitle, ">" );
978         }
979
980         psz_subtitle++;
981     }
982
983     return p_region_first;
984 }
985
986 /*****************************************************************************
987  * ParseUSFHeader: Retrieve global formatting information etc
988  *****************************************************************************/
989 static void ParseUSFHeader( decoder_t *p_dec )
990 {
991     stream_t      *p_sub = NULL;
992     xml_t         *p_xml = NULL;
993     xml_reader_t  *p_xml_reader = NULL;
994
995     p_sub = stream_MemoryNew( VLC_OBJECT(p_dec),
996                               p_dec->fmt_in.p_extra,
997                               p_dec->fmt_in.i_extra,
998                               true );
999     if( !p_sub )
1000         return;
1001
1002     p_xml = xml_Create( p_dec );
1003     if( p_xml )
1004     {
1005         p_xml_reader = xml_ReaderCreate( p_xml, p_sub );
1006         if( p_xml_reader )
1007         {
1008             /* Look for Root Node */
1009             if( xml_ReaderRead( p_xml_reader ) == 1 )
1010             {
1011                 char *psz_node = xml_ReaderName( p_xml_reader );
1012
1013                 if( !strcasecmp( "usfsubtitles", psz_node ) )
1014                     ParseUSFHeaderTags( p_dec, p_xml_reader );
1015
1016                 free( psz_node );
1017             }
1018
1019             xml_ReaderDelete( p_xml, p_xml_reader );
1020         }
1021         xml_Delete( p_xml );
1022     }
1023     stream_Delete( p_sub );
1024 }
1025
1026 /* Function now handles tags which has attribute values, and tries
1027  * to deal with &' commands too. It no longer modifies the string
1028  * in place, so that the original text can be reused
1029  */
1030 static char *StripTags( char *psz_subtitle )
1031 {
1032     char *psz_text_start;
1033     char *psz_text;
1034
1035     psz_text = psz_text_start = malloc( strlen( psz_subtitle ) + 1 );
1036     if( !psz_text_start )
1037         return NULL;
1038
1039     while( *psz_subtitle )
1040     {
1041         /* Mask out any pre-existing LFs in the subtitle */
1042         if( *psz_subtitle == '\n' )
1043             *psz_subtitle = ' ';
1044
1045         if( *psz_subtitle == '<' )
1046         {
1047             if( strncasecmp( psz_subtitle, "<br/>", 5 ) == 0 )
1048                 *psz_text++ = '\n';
1049
1050             psz_subtitle += strcspn( psz_subtitle, ">" );
1051         }
1052         else if( *psz_subtitle == '&' )
1053         {
1054             if( !strncasecmp( psz_subtitle, "&lt;", 4 ))
1055             {
1056                 *psz_text++ = '<';
1057                 psz_subtitle += strcspn( psz_subtitle, ";" );
1058             }
1059             else if( !strncasecmp( psz_subtitle, "&gt;", 4 ))
1060             {
1061                 *psz_text++ = '>';
1062                 psz_subtitle += strcspn( psz_subtitle, ";" );
1063             }
1064             else if( !strncasecmp( psz_subtitle, "&amp;", 5 ))
1065             {
1066                 *psz_text++ = '&';
1067                 psz_subtitle += strcspn( psz_subtitle, ";" );
1068             }
1069             else if( !strncasecmp( psz_subtitle, "&quot;", 6 ))
1070             {
1071                 *psz_text++ = '\"';
1072                 psz_subtitle += strcspn( psz_subtitle, ";" );
1073             }
1074             else
1075             {
1076                 /* Assume it is just a normal ampersand */
1077                 *psz_text++ = '&';
1078             }
1079         }
1080         else
1081         {
1082             *psz_text++ = *psz_subtitle;
1083         }
1084
1085         psz_subtitle++;
1086     }
1087     *psz_text = '\0';
1088     psz_text_start = realloc( psz_text_start, strlen( psz_text_start ) + 1 );
1089
1090     return psz_text_start;
1091 }
1092
1093 /* Turn a HTML subtitle, turn into a plain-text version,
1094  *  complete with sensible whitespace compaction
1095  */
1096
1097 static char *CreatePlainText( char *psz_subtitle )
1098 {
1099     char *psz_text = StripTags( psz_subtitle );
1100     char *s;
1101
1102     if( !psz_text )
1103         return NULL;
1104
1105     s = strpbrk( psz_text, "\t\r\n " );
1106     while( s )
1107     {
1108         int   k;
1109         char  spc = ' ';
1110         int   i_whitespace = strspn( s, "\t\r\n " );
1111
1112         /* Favour '\n' over other whitespaces - if one of these
1113          * occurs in the whitespace use a '\n' as our value,
1114          * otherwise just use a ' '
1115          */
1116         for( k = 0; k < i_whitespace; k++ )
1117             if( s[k] == '\n' ) spc = '\n';
1118
1119         if( i_whitespace > 1 )
1120         {
1121             memmove( &s[1],
1122                      &s[i_whitespace],
1123                      strlen( s ) - i_whitespace + 1 );
1124         }
1125         *s++ = spc;
1126
1127         s = strpbrk( s, "\t\r\n " );
1128     }
1129     return psz_text;
1130 }
1131
1132 /****************************************************************************
1133  * download and resize image located at psz_url
1134  ***************************************************************************/
1135 static subpicture_region_t *LoadEmbeddedImage( decoder_t *p_dec,
1136                                                const char *psz_filename,
1137                                                int i_transparent_color )
1138 {
1139     decoder_sys_t         *p_sys = p_dec->p_sys;
1140     subpicture_region_t   *p_region;
1141     video_format_t         fmt_out;
1142     picture_t             *p_pic = NULL;
1143
1144     for( int k = 0; k < p_sys->i_images; k++ )
1145     {
1146         if( p_sys->pp_images &&
1147             !strcmp( p_sys->pp_images[k]->psz_filename, psz_filename ) )
1148         {
1149             p_pic = p_sys->pp_images[k]->p_pic;
1150             break;
1151         }
1152     }
1153
1154     if( !p_pic )
1155     {
1156         msg_Err( p_dec, "Unable to read image %s", psz_filename );
1157         return NULL;
1158     }
1159
1160     /* Display the feed's image */
1161     memset( &fmt_out, 0, sizeof( video_format_t));
1162
1163     fmt_out.i_chroma = VLC_CODEC_YUVA;
1164     fmt_out.i_aspect = VOUT_ASPECT_FACTOR;
1165     fmt_out.i_sar_num = fmt_out.i_sar_den = 1;
1166     fmt_out.i_width =
1167         fmt_out.i_visible_width = p_pic->format.i_visible_width;
1168     fmt_out.i_height =
1169         fmt_out.i_visible_height = p_pic->format.i_visible_height;
1170
1171     p_region = subpicture_region_New( &fmt_out );
1172     if( !p_region )
1173     {
1174         msg_Err( p_dec, "cannot allocate SPU region" );
1175         return NULL;
1176     }
1177     assert( p_pic->format.i_chroma == VLC_CODEC_YUVA );
1178     /* FIXME the copy is probably not needed anymore */
1179     picture_CopyPixels( p_region->p_picture, p_pic );
1180
1181     /* This isn't the best way to do this - if you really want transparency, then
1182      * you're much better off using an image type that supports it like PNG. The
1183      * spec requires this support though.
1184      */
1185     if( i_transparent_color > 0 )
1186     {
1187         int i_r = ( i_transparent_color >> 16 ) & 0xff;
1188         int i_g = ( i_transparent_color >>  8 ) & 0xff;
1189         int i_b = ( i_transparent_color       ) & 0xff;
1190
1191         /* FIXME it cannot work as the yuv conversion code will probably NOT match
1192          * this one  */
1193         int i_y = ( ( (  66 * i_r + 129 * i_g +  25 * i_b + 128 ) >> 8 ) + 16 );
1194         int i_u =   ( ( -38 * i_r -  74 * i_g + 112 * i_b + 128 ) >> 8 ) + 128 ;
1195         int i_v =   ( ( 112 * i_r -  94 * i_g -  18 * i_b + 128 ) >> 8 ) + 128 ;
1196
1197         assert( p_region->fmt.i_chroma == VLC_CODEC_YUVA );
1198         for( unsigned int y = 0; y < p_region->fmt.i_height; y++ )
1199         {
1200             for( unsigned int x = 0; x < p_region->fmt.i_width; x++ )
1201             {
1202                 if( p_region->p_picture->Y_PIXELS[y*p_region->p_picture->Y_PITCH + x] != i_y ||
1203                     p_region->p_picture->U_PIXELS[y*p_region->p_picture->U_PITCH + x] != i_u ||
1204                     p_region->p_picture->V_PIXELS[y*p_region->p_picture->V_PITCH + x] != i_v )
1205                     continue;
1206                 p_region->p_picture->A_PIXELS[y*p_region->p_picture->A_PITCH + x] = 0;
1207
1208             }
1209         }
1210     }
1211     return p_region;
1212 }