1 /*****************************************************************************
2 * dvbsub.c : DVB subtitles decoder
3 * DVB subtitles encoder (developed for Anevia, www.anevia.com)
4 *****************************************************************************
5 * Copyright (C) 2003 ANEVIA
6 * Copyright (C) 2003-2005 the VideoLAN team
9 * Authors: Gildas Bazin <gbazin@videolan.org>
10 * Damien LUCAS <damien.lucas@anevia.com>
11 * Laurent Aimar <fenrir@via.ecp.fr>
12 * Jean-Paul Saman <jpsaman #_at_# m2x dot nl>
13 * Derk-Jan Hartman <hartman #at# videolan dot org>
15 * This program is free software; you can redistribute it and/or modify
16 * it under the terms of the GNU General Public License as published by
17 * the Free Software Foundation; either version 2 of the License, or
18 * (at your option) any later version.
20 * This program is distributed in the hope that it will be useful,
21 * but WITHOUT ANY WARRANTY; without even the implied warranty of
22 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
23 * GNU General Public License for more details.
25 * You should have received a copy of the GNU General Public License
26 * along with this program; if not, write to the Free Software
27 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
28 *****************************************************************************/
29 /*****************************************************************************
33 * DVB subtitles coded as strings of characters are not handled correctly.
34 * The character codes in the string should actually be indexes referring to a
35 * character table identified in the subtitle descriptor.
37 * The spec is quite vague in this area, but what is meant is perhaps that it
38 * refers to the character index in the codepage belonging to the language specified
39 * in the subtitle descriptor. Potentially it's designed for widechar
40 * (but not for UTF-*) codepages.
41 *****************************************************************************/
44 #include <vlc_codec.h>
49 /* #define DEBUG_DVBSUB 1 */
51 #define POSX_TEXT N_("Decoding X coordinate")
52 #define POSX_LONGTEXT N_("X coordinate of the rendered subtitle")
54 #define POSY_TEXT N_("Decoding Y coordinate")
55 #define POSY_LONGTEXT N_("Y coordinate of the rendered subtitle")
57 #define POS_TEXT N_("Subpicture position")
58 #define POS_LONGTEXT N_( \
59 "You can enforce the subpicture position on the video " \
60 "(0=center, 1=left, 2=right, 4=top, 8=bottom, you can " \
61 "also use combinations of these values, e.g. 6=top-right).")
63 #define ENC_POSX_TEXT N_("Encoding X coordinate")
64 #define ENC_POSX_LONGTEXT N_("X coordinate of the encoded subtitle" )
65 #define ENC_POSY_TEXT N_("Encoding Y coordinate")
66 #define ENC_POSY_LONGTEXT N_("Y coordinate of the encoded subtitle" )
68 static int pi_pos_values[] = { 0, 1, 2, 4, 8, 5, 6, 9, 10 };
69 static const char *ppsz_pos_descriptions[] =
70 { N_("Center"), N_("Left"), N_("Right"), N_("Top"), N_("Bottom"),
71 N_("Top-Left"), N_("Top-Right"), N_("Bottom-Left"), N_("Bottom-Right") };
73 /*****************************************************************************
75 *****************************************************************************/
76 static int Open ( vlc_object_t * );
77 static void Close( vlc_object_t * );
78 static subpicture_t *Decode( decoder_t *, block_t ** );
80 static int OpenEncoder ( vlc_object_t * );
81 static void CloseEncoder( vlc_object_t * );
82 static block_t *Encode ( encoder_t *, subpicture_t * );
85 # define DVBSUB_CFG_PREFIX "dvbsub-"
86 set_description( _("DVB subtitles decoder") );
87 set_capability( "decoder", 50 );
88 set_category( CAT_INPUT );
89 set_subcategory( SUBCAT_INPUT_SCODEC );
90 set_callbacks( Open, Close );
92 add_integer( DVBSUB_CFG_PREFIX "position", 8, NULL, POS_TEXT, POS_LONGTEXT, VLC_TRUE );
93 change_integer_list( pi_pos_values, ppsz_pos_descriptions, 0 );
94 add_integer( DVBSUB_CFG_PREFIX "x", -1, NULL, POSX_TEXT, POSX_LONGTEXT, VLC_FALSE );
95 add_integer( DVBSUB_CFG_PREFIX "y", -1, NULL, POSY_TEXT, POSY_LONGTEXT, VLC_FALSE );
97 # define ENC_CFG_PREFIX "sout-dvbsub-"
99 set_description( _("DVB subtitles encoder") );
100 set_capability( "encoder", 100 );
101 set_callbacks( OpenEncoder, CloseEncoder );
103 add_integer( ENC_CFG_PREFIX "x", -1, NULL, ENC_POSX_TEXT, ENC_POSX_LONGTEXT, VLC_FALSE );
104 add_integer( ENC_CFG_PREFIX "y", -1, NULL, ENC_POSY_TEXT, ENC_POSY_LONGTEXT, VLC_FALSE );
105 add_obsolete_integer( ENC_CFG_PREFIX "timeout" ); /* Suppressed since 0.8.5 */
108 static const char *ppsz_enc_options[] = { "x", "y", NULL };
110 /****************************************************************************
112 ****************************************************************************
113 * Those structures refer closely to the ETSI 300 743 Object model
114 ****************************************************************************/
116 /* The object definition gives the position of the object in a region [7.2.5] */
117 typedef struct dvbsub_objectdef_s
125 char *psz_text; /* for string of characters objects */
127 } dvbsub_objectdef_t;
129 /* The entry in the palette CLUT */
139 /* The displays dimensions [7.2.1] */
140 typedef struct dvbsub_display_s
148 vlc_bool_t b_windowed;
157 typedef struct dvbsub_clut_s
161 dvbsub_color_t c_2b[4];
162 dvbsub_color_t c_4b[16];
163 dvbsub_color_t c_8b[256];
165 struct dvbsub_clut_s *p_next;
169 /* The Region is an aera on the image [7.2.3]
170 * with a list of the object definitions associated and a CLUT */
171 typedef struct dvbsub_region_s
186 dvbsub_objectdef_t *p_object_defs;
188 struct dvbsub_region_s *p_next;
192 /* The object definition gives the position of the object in a region */
193 typedef struct dvbsub_regiondef_s
199 } dvbsub_regiondef_t;
201 /* The page defines the list of regions [7.2.2] */
205 int i_timeout; /* in seconds */
210 dvbsub_regiondef_t *p_region_defs;
218 /* Decoder internal data */
223 vlc_bool_t b_absolute;
229 dvbsub_page_t *p_page;
230 dvbsub_region_t *p_regions;
231 dvbsub_clut_t *p_cluts;
232 dvbsub_display_t *p_display;
233 dvbsub_clut_t default_clut;
237 /* List of different SEGMENT TYPES */
238 /* According to EN 300-743, table 2 */
239 #define DVBSUB_ST_PAGE_COMPOSITION 0x10
240 #define DVBSUB_ST_REGION_COMPOSITION 0x11
241 #define DVBSUB_ST_CLUT_DEFINITION 0x12
242 #define DVBSUB_ST_OBJECT_DATA 0x13
243 #define DVBSUB_ST_DISPLAY_DEFINITION 0x14
244 #define DVBSUB_ST_ENDOFDISPLAY 0x80
245 #define DVBSUB_ST_STUFFING 0xff
246 /* List of different OBJECT TYPES */
247 /* According to EN 300-743, table 6 */
248 #define DVBSUB_OT_BASIC_BITMAP 0x00
249 #define DVBSUB_OT_BASIC_CHAR 0x01
250 #define DVBSUB_OT_COMPOSITE_STRING 0x02
251 /* Pixel DATA TYPES */
252 /* According to EN 300-743, table 9 */
253 #define DVBSUB_DT_2BP_CODE_STRING 0x10
254 #define DVBSUB_DT_4BP_CODE_STRING 0x11
255 #define DVBSUB_DT_8BP_CODE_STRING 0x12
256 #define DVBSUB_DT_24_TABLE_DATA 0x20
257 #define DVBSUB_DT_28_TABLE_DATA 0x21
258 #define DVBSUB_DT_48_TABLE_DATA 0x22
259 #define DVBSUB_DT_END_LINE 0xf0
260 /* List of different Page Composition Segment state */
261 /* According to EN 300-743, 7.2.1 table 3 */
262 #define DVBSUB_PCS_STATE_ACQUISITION 0x01
263 #define DVBSUB_PCS_STATE_CHANGE 0x02
265 /*****************************************************************************
267 *****************************************************************************/
268 static void decode_segment( decoder_t *, bs_t * );
269 static void decode_page_composition( decoder_t *, bs_t * );
270 static void decode_region_composition( decoder_t *, bs_t * );
271 static void decode_object( decoder_t *, bs_t * );
272 static void decode_display_definition( decoder_t *, bs_t * );
273 static void decode_clut( decoder_t *, bs_t * );
274 static void free_all( decoder_t * );
276 static void default_clut_init( decoder_t * );
278 static subpicture_t *render( decoder_t * );
280 /*****************************************************************************
281 * Open: probe the decoder and return score
282 *****************************************************************************
283 * Tries to launch a decoder and return score so that the interface is able
285 *****************************************************************************/
286 static int Open( vlc_object_t *p_this )
288 decoder_t *p_dec = (decoder_t *) p_this;
289 decoder_sys_t *p_sys;
293 if( p_dec->fmt_in.i_codec != VLC_FOURCC('d','v','b','s') )
298 p_dec->pf_decode_sub = Decode;
299 p_sys = p_dec->p_sys = malloc( sizeof(decoder_sys_t) );
302 msg_Err( p_dec, "out of memory" );
305 memset( p_sys, 0, sizeof(decoder_sys_t) );
307 p_sys->i_pts = (mtime_t) 0;
308 p_sys->i_id = p_dec->fmt_in.subs.dvb.i_id & 0xFFFF;
309 p_sys->i_ancillary_id = p_dec->fmt_in.subs.dvb.i_id >> 16;
311 p_sys->p_regions = NULL;
312 p_sys->p_cluts = NULL;
313 p_sys->p_page = NULL;
314 p_sys->p_display = NULL;
316 var_Create( p_this, DVBSUB_CFG_PREFIX "position",
317 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
318 var_Get( p_this, DVBSUB_CFG_PREFIX "position", &val );
319 p_sys->i_spu_position = val.i_int;
320 var_Create( p_this, DVBSUB_CFG_PREFIX "x",
321 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
322 var_Get( p_this, DVBSUB_CFG_PREFIX "x", &val );
324 var_Create( p_this, DVBSUB_CFG_PREFIX "y",
325 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
326 var_Get( p_this, DVBSUB_CFG_PREFIX "y", &val );
329 /* Check if subpicture position was overridden */
330 p_sys->b_absolute = VLC_TRUE;
331 p_sys->i_spu_x = p_sys->i_spu_y = 0;
333 if( i_posx >= 0 && i_posy >= 0 )
335 p_sys->b_absolute = VLC_FALSE;
336 p_sys->i_spu_x = i_posx;
337 p_sys->i_spu_y = i_posy;
340 es_format_Init( &p_dec->fmt_out, SPU_ES, VLC_FOURCC( 'd','v','b','s' ) );
342 default_clut_init( p_dec );
347 /*****************************************************************************
349 *****************************************************************************/
350 static void Close( vlc_object_t *p_this )
352 decoder_t *p_dec = (decoder_t*) p_this;
353 decoder_sys_t *p_sys = p_dec->p_sys;
355 var_Destroy( p_this, DVBSUB_CFG_PREFIX "x" );
356 var_Destroy( p_this, DVBSUB_CFG_PREFIX "y" );
357 var_Destroy( p_this, DVBSUB_CFG_PREFIX "position" );
363 /*****************************************************************************
365 *****************************************************************************/
366 static subpicture_t *Decode( decoder_t *p_dec, block_t **pp_block )
368 decoder_sys_t *p_sys = p_dec->p_sys;
370 subpicture_t *p_spu = NULL;
372 if( pp_block == NULL || *pp_block == NULL ) return NULL;
376 p_sys->i_pts = p_block->i_pts;
377 if( p_sys->i_pts <= 0 )
380 /* Some DVB channels send stuffing segments in non-dated packets so
381 * don't complain too loudly. */
382 msg_Warn( p_dec, "non dated subtitle" );
384 block_Release( p_block );
388 bs_init( &p_sys->bs, p_block->p_buffer, p_block->i_buffer );
390 if( bs_read( &p_sys->bs, 8 ) != 0x20 ) /* Data identifier */
392 msg_Dbg( p_dec, "invalid data identifier" );
393 block_Release( p_block );
397 if( bs_read( &p_sys->bs, 8 ) ) /* Subtitle stream id */
399 msg_Dbg( p_dec, "invalid subtitle stream id" );
400 block_Release( p_block );
405 msg_Dbg( p_dec, "subtitle packet received: "I64Fd, p_sys->i_pts );
408 p_sys->b_page = VLC_FALSE;
409 while( bs_show( &p_sys->bs, 8 ) == 0x0f ) /* Sync byte */
411 decode_segment( p_dec, &p_sys->bs );
414 if( bs_read( &p_sys->bs, 8 ) != 0xff ) /* End marker */
416 msg_Warn( p_dec, "end marker not found (corrupted subtitle ?)" );
417 block_Release( p_block );
421 /* Check if the page is to be displayed */
422 if( p_sys->p_page && p_sys->b_page ) p_spu = render( p_dec );
424 block_Release( p_block );
429 /* following functions are local */
431 /*****************************************************************************
432 * default_clut_init: default clut as defined in EN 300-743 section 10
433 *****************************************************************************/
434 static void default_clut_init( decoder_t *p_dec )
436 decoder_sys_t *p_sys = p_dec->p_sys;
439 #define RGB_TO_Y(r, g, b) ((int16_t) 77 * r + 150 * g + 29 * b) / 256;
440 #define RGB_TO_U(r, g, b) ((int16_t) -44 * r - 87 * g + 131 * b) / 256;
441 #define RGB_TO_V(r, g, b) ((int16_t) 131 * r - 110 * g - 21 * b) / 256;
444 for( i = 0; i < 4; i++ )
446 uint8_t R = 0, G = 0, B = 0, T = 0;
448 if( !(i & 0x2) && !(i & 0x1) ) T = 0xFF;
449 else if( !(i & 0x2) && (i & 0x1) ) R = G = B = 0xFF;
450 else if( (i & 0x2) && !(i & 0x1) ) R = G = B = 0;
451 else R = G = B = 0x7F;
453 p_sys->default_clut.c_2b[i].Y = RGB_TO_Y(R,G,B);
454 p_sys->default_clut.c_2b[i].Cb = RGB_TO_V(R,G,B);
455 p_sys->default_clut.c_2b[i].Cr = RGB_TO_U(R,G,B);
456 p_sys->default_clut.c_2b[i].T = T;
459 /* 16 entries CLUT */
460 for( i = 0; i < 16; i++ )
462 uint8_t R = 0, G = 0, B = 0, T = 0;
466 if( !(i & 0x4) && !(i & 0x2) && !(i & 0x1) )
472 R = (i & 0x1) ? 0xFF : 0;
473 G = (i & 0x2) ? 0xFF : 0;
474 B = (i & 0x4) ? 0xFF : 0;
479 R = (i & 0x1) ? 0x7F : 0;
480 G = (i & 0x2) ? 0x7F : 0;
481 B = (i & 0x4) ? 0x7F : 0;
484 p_sys->default_clut.c_4b[i].Y = RGB_TO_Y(R,G,B);
485 p_sys->default_clut.c_4b[i].Cr = RGB_TO_V(R,G,B);
486 p_sys->default_clut.c_4b[i].Cb = RGB_TO_U(R,G,B);
487 p_sys->default_clut.c_4b[i].T = T;
490 /* 256 entries CLUT */
491 memset( p_sys->default_clut.c_8b, 0xFF, 256 * sizeof(dvbsub_color_t) );
494 static void decode_segment( decoder_t *p_dec, bs_t *s )
496 decoder_sys_t *p_sys = p_dec->p_sys;
501 /* sync_byte (already checked) */
505 i_type = bs_read( s, 8 );
508 i_page_id = bs_read( s, 16 );
511 i_size = bs_show( s, 16 );
513 if( i_page_id != p_sys->i_id && i_page_id != p_sys->i_ancillary_id )
516 msg_Dbg( p_dec, "subtitle skipped (page id: %i, %i)",
517 i_page_id, p_sys->i_id );
519 bs_skip( s, 8 * ( 2 + i_size ) );
523 if( p_sys->i_ancillary_id != p_sys->i_id &&
524 i_type == DVBSUB_ST_PAGE_COMPOSITION &&
525 i_page_id == p_sys->i_ancillary_id )
528 msg_Dbg( p_dec, "skipped invalid ancillary subtitle packet" );
530 bs_skip( s, 8 * ( 2 + i_size ) );
535 if( i_page_id == p_sys->i_id )
536 msg_Dbg( p_dec, "segment (id: %i)", i_page_id );
538 msg_Dbg( p_dec, "ancillary segment (id: %i)", i_page_id );
543 case DVBSUB_ST_PAGE_COMPOSITION:
545 msg_Dbg( p_dec, "decode_page_composition" );
547 decode_page_composition( p_dec, s );
550 case DVBSUB_ST_REGION_COMPOSITION:
552 msg_Dbg( p_dec, "decode_region_composition" );
554 decode_region_composition( p_dec, s );
557 case DVBSUB_ST_CLUT_DEFINITION:
559 msg_Dbg( p_dec, "decode_clut" );
561 decode_clut( p_dec, s );
564 case DVBSUB_ST_OBJECT_DATA:
566 msg_Dbg( p_dec, "decode_object" );
568 decode_object( p_dec, s );
571 case DVBSUB_ST_DISPLAY_DEFINITION:
573 msg_Dbg( p_dec, "decode_display_definition" );
575 decode_display_definition( p_dec, s );
578 case DVBSUB_ST_ENDOFDISPLAY:
580 msg_Dbg( p_dec, "end of display" );
582 bs_skip( s, 8 * ( 2 + i_size ) );
585 case DVBSUB_ST_STUFFING:
587 msg_Dbg( p_dec, "skip stuffing" );
589 bs_skip( s, 8 * ( 2 + i_size ) );
593 msg_Warn( p_dec, "unsupported segment type: (%04x)", i_type );
594 bs_skip( s, 8 * ( 2 + i_size ) );
599 static void decode_clut( decoder_t *p_dec, bs_t *s )
601 decoder_sys_t *p_sys = p_dec->p_sys;
602 uint16_t i_segment_length;
603 uint16_t i_processed_length;
604 dvbsub_clut_t *p_clut, *p_next;
607 i_segment_length = bs_read( s, 16 );
608 i_id = bs_read( s, 8 );
609 i_version = bs_read( s, 4 );
611 /* Check if we already have this clut */
612 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut->p_next )
614 if( p_clut->i_id == i_id ) break;
617 /* Check version number */
618 if( p_clut && p_clut->i_version == i_version )
621 bs_skip( s, 8 * i_segment_length - 12 );
628 msg_Dbg( p_dec, "new clut: %i", i_id );
630 p_clut = malloc( sizeof(dvbsub_clut_t) );
631 p_clut->p_next = p_sys->p_cluts;
632 p_sys->p_cluts = p_clut;
635 /* Initialize to default clut */
636 p_next = p_clut->p_next;
637 *p_clut = p_sys->default_clut;
638 p_clut->p_next = p_next;
640 /* We don't have this version of the CLUT: Parse it */
641 p_clut->i_version = i_version;
643 bs_skip( s, 4 ); /* Reserved bits */
644 i_processed_length = 2;
645 while( i_processed_length < i_segment_length )
647 uint8_t y, cb, cr, t;
651 i_id = bs_read( s, 8 );
652 i_type = bs_read( s, 3 );
656 if( bs_read( s, 1 ) )
659 cr = bs_read( s, 8 );
660 cb = bs_read( s, 8 );
662 i_processed_length += 6;
666 y = bs_read( s, 6 ) << 2;
667 cr = bs_read( s, 4 ) << 4;
668 cb = bs_read( s, 4 ) << 4;
669 t = bs_read( s, 2 ) << 6;
670 i_processed_length += 4;
673 /* We are not entirely compliant here as full transparency is indicated
674 * with a luma value of zero, not a transparency value of 0xff
675 * (full transparency would actually be 0xff + 1). */
682 /* According to EN 300-743 section 7.2.3 note 1, type should
683 * not have more than 1 bit set to one, but some streams don't
684 * respect this note. */
685 if( i_type & 0x04 && i_id < 4 )
687 p_clut->c_2b[i_id].Y = y;
688 p_clut->c_2b[i_id].Cr = cr;
689 p_clut->c_2b[i_id].Cb = cb;
690 p_clut->c_2b[i_id].T = t;
692 if( i_type & 0x02 && i_id < 16 )
694 p_clut->c_4b[i_id].Y = y;
695 p_clut->c_4b[i_id].Cr = cr;
696 p_clut->c_4b[i_id].Cb = cb;
697 p_clut->c_4b[i_id].T = t;
701 p_clut->c_8b[i_id].Y = y;
702 p_clut->c_8b[i_id].Cr = cr;
703 p_clut->c_8b[i_id].Cb = cb;
704 p_clut->c_8b[i_id].T = t;
709 static void decode_page_composition( decoder_t *p_dec, bs_t *s )
711 decoder_sys_t *p_sys = p_dec->p_sys;
712 int i_version, i_state, i_segment_length, i_timeout, i;
714 /* A page is composed by 0 or more region */
715 i_segment_length = bs_read( s, 16 );
716 i_timeout = bs_read( s, 8 );
717 i_version = bs_read( s, 4 );
718 i_state = bs_read( s, 2 );
719 bs_skip( s, 2 ); /* Reserved */
721 if( i_state == DVBSUB_PCS_STATE_CHANGE )
723 /* End of an epoch, reset decoder buffer */
725 msg_Dbg( p_dec, "page composition mode change" );
729 else if( !p_sys->p_page && i_state != DVBSUB_PCS_STATE_ACQUISITION &&
730 i_state != DVBSUB_PCS_STATE_CHANGE )
732 /* Not a full PCS, we need to wait for one */
733 msg_Dbg( p_dec, "didn't receive an acquisition page yet" );
736 /* Try to start decoding even without an acquisition page */
737 bs_skip( s, 8 * (i_segment_length - 2) );
743 if( i_state == DVBSUB_PCS_STATE_ACQUISITION )
744 msg_Dbg( p_dec, "acquisition page composition" );
747 /* Check version number */
748 if( p_sys->p_page && p_sys->p_page->i_version == i_version )
750 bs_skip( s, 8 * (i_segment_length - 2) );
753 else if( p_sys->p_page )
755 if( p_sys->p_page->i_region_defs )
756 free( p_sys->p_page->p_region_defs );
757 p_sys->p_page->i_region_defs = 0;
763 msg_Dbg( p_dec, "new page" );
765 /* Allocate a new page */
766 p_sys->p_page = malloc( sizeof(dvbsub_page_t) );
769 p_sys->p_page->i_version = i_version;
770 p_sys->p_page->i_timeout = i_timeout;
771 p_sys->b_page = VLC_TRUE;
773 /* Number of regions */
774 p_sys->p_page->i_region_defs = (i_segment_length - 2) / 6;
776 if( p_sys->p_page->i_region_defs == 0 ) return;
778 p_sys->p_page->p_region_defs =
779 malloc( p_sys->p_page->i_region_defs * sizeof(dvbsub_region_t) );
780 for( i = 0; i < p_sys->p_page->i_region_defs; i++ )
782 p_sys->p_page->p_region_defs[i].i_id = bs_read( s, 8 );
783 bs_skip( s, 8 ); /* Reserved */
784 p_sys->p_page->p_region_defs[i].i_x = bs_read( s, 16 );
785 p_sys->p_page->p_region_defs[i].i_y = bs_read( s, 16 );
788 msg_Dbg( p_dec, "page_composition, region %i (%i,%i)",
789 i, p_sys->p_page->p_region_defs[i].i_x,
790 p_sys->p_page->p_region_defs[i].i_y );
795 static void decode_region_composition( decoder_t *p_dec, bs_t *s )
797 decoder_sys_t *p_sys = p_dec->p_sys;
798 dvbsub_region_t *p_region, **pp_region = &p_sys->p_regions;
799 int i_segment_length, i_processed_length, i_id, i_version;
800 int i_width, i_height, i_level_comp, i_depth, i_clut;
801 int i_8_bg, i_4_bg, i_2_bg;
804 i_segment_length = bs_read( s, 16 );
805 i_id = bs_read( s, 8 );
806 i_version = bs_read( s, 4 );
808 /* Check if we already have this region */
809 for( p_region = p_sys->p_regions; p_region != NULL;
810 p_region = p_region->p_next )
812 pp_region = &p_region->p_next;
813 if( p_region->i_id == i_id ) break;
816 /* Check version number */
817 if( p_region && p_region->i_version == i_version )
819 bs_skip( s, 8 * (i_segment_length - 1) - 4 );
826 msg_Dbg( p_dec, "new region: %i", i_id );
828 p_region = *pp_region = malloc( sizeof(dvbsub_region_t) );
829 memset( p_region, 0, sizeof(dvbsub_region_t) );
830 p_region->p_object_defs = NULL;
831 p_region->p_pixbuf = NULL;
832 p_region->p_next = NULL;
835 /* Region attributes */
836 p_region->i_id = i_id;
837 p_region->i_version = i_version;
838 b_fill = bs_read( s, 1 );
839 bs_skip( s, 3 ); /* Reserved */
841 i_width = bs_read( s, 16 );
842 i_height = bs_read( s, 16 );
844 msg_Dbg( p_dec, " width=%d height=%d", i_width, i_height );
846 i_level_comp = bs_read( s, 3 );
847 i_depth = bs_read( s, 3 );
848 bs_skip( s, 2 ); /* Reserved */
849 i_clut = bs_read( s, 8 );
851 i_8_bg = bs_read( s, 8 );
852 i_4_bg = bs_read( s, 4 );
853 i_2_bg = bs_read( s, 2 );
854 bs_skip( s, 2 ); /* Reserved */
856 /* Free old object defs */
857 while( p_region->i_object_defs )
859 int i = p_region->i_object_defs - 1;
860 if( p_region->p_object_defs[i].psz_text )
861 free( p_region->p_object_defs[i].psz_text );
862 if( !i ) free( p_region->p_object_defs );
864 p_region->i_object_defs--;
866 p_region->p_object_defs = NULL;
868 /* Extra sanity checks */
869 if( p_region->i_width != i_width || p_region->i_height != i_height )
871 if( p_region->p_pixbuf )
873 msg_Dbg( p_dec, "region size changed (%dx%d->%dx%d)",
874 p_region->i_width, p_region->i_height, i_width, i_height );
875 free( p_region->p_pixbuf );
878 p_region->p_pixbuf = malloc( i_height * i_width );
879 p_region->i_depth = 0;
882 if( p_region->i_depth && (p_region->i_depth != i_depth ||
883 p_region->i_level_comp != i_level_comp || p_region->i_clut != i_clut) )
885 msg_Dbg( p_dec, "region parameters changed (not allowed)" );
888 /* Erase background of region */
891 int i_background = (p_region->i_depth == 1) ? i_2_bg :
892 (p_region->i_depth == 2) ? i_4_bg : i_8_bg;
893 memset( p_region->p_pixbuf, i_background, i_width * i_height );
896 p_region->i_width = i_width;
897 p_region->i_height = i_height;
898 p_region->i_level_comp = i_level_comp;
899 p_region->i_depth = i_depth;
900 p_region->i_clut = i_clut;
902 /* List of objects in the region */
903 i_processed_length = 10;
904 while( i_processed_length < i_segment_length )
906 dvbsub_objectdef_t *p_obj;
908 /* We create a new object */
909 p_region->i_object_defs++;
910 p_region->p_object_defs =
911 realloc( p_region->p_object_defs,
912 sizeof(dvbsub_objectdef_t) * p_region->i_object_defs );
914 /* We parse object properties */
915 p_obj = &p_region->p_object_defs[p_region->i_object_defs - 1];
916 p_obj->i_id = bs_read( s, 16 );
917 p_obj->i_type = bs_read( s, 2 );
918 bs_skip( s, 2 ); /* Provider */
919 p_obj->i_x = bs_read( s, 12 );
920 bs_skip( s, 4 ); /* Reserved */
921 p_obj->i_y = bs_read( s, 12 );
924 i_processed_length += 6;
926 if( p_obj->i_type == DVBSUB_OT_BASIC_CHAR ||
927 p_obj->i_type == DVBSUB_OT_COMPOSITE_STRING )
929 p_obj->i_fg_pc = bs_read( s, 8 );
930 p_obj->i_bg_pc = bs_read( s, 8 );
931 i_processed_length += 2;
936 /* ETSI 300 743 [7.2.1] */
937 static void decode_display_definition( decoder_t *p_dec, bs_t *s )
939 decoder_sys_t *p_sys = p_dec->p_sys;
940 uint16_t i_segment_length;
941 uint16_t i_processed_length = 40;
942 dvbsub_display_t *p_display;
943 dvbsub_display_t *p_old = p_sys->p_display;
946 i_segment_length = bs_read( s, 16 );
947 i_version = bs_read( s, 4 );
949 /* Check version number */
950 if( p_old && p_old->i_version == i_version )
952 /* The definition did not change */
953 bs_skip( s, 8*i_segment_length - 4 );
958 msg_Dbg( p_dec, "new display definition: %i", i_version );
960 p_display = malloc( sizeof(dvbsub_display_t) );
962 /* We don't have this version of the display definition: Parse it */
963 p_display->i_version = i_version;
964 p_display->b_windowed = bs_read( s, 1 );
965 bs_skip( s, 3 ); /* Reserved bits */
966 p_display->i_width = bs_read( s, 16 )+1;
967 p_display->i_height = bs_read( s, 16 )+1;
969 if( p_display->b_windowed )
972 msg_Dbg( p_dec, "display definition with offsets (windowed)" );
974 /* Coordinates are measured from the top left corner */
975 p_display->i_x = bs_read( s, 16 );
976 p_display->i_max_x = bs_read( s, 16 );
977 p_display->i_y = bs_read( s, 16 );
978 p_display->i_max_y = bs_read( s, 16 );
979 i_processed_length += 64;
982 p_sys->p_display = p_display;
983 if( p_old ) free( p_old );
985 if( i_processed_length != i_segment_length*8 )
987 msg_Err( p_dec, "processed length %d != segment length %d", i_processed_length, i_segment_length );
991 msg_Dbg( p_dec, "version: %d, width: %d, height: %d", p_display->i_version, p_display->i_width, p_display->i_height );
992 if( p_display->b_windowed )
993 msg_Dbg( p_dec, "xmin: %d, xmax: %d, ymin: %d, ymax: %d", p_display->i_x, p_display->i_max_x, p_display->i_y, p_display->i_max_y );
997 static void dvbsub_render_pdata( decoder_t *, dvbsub_region_t *, int, int,
999 static void dvbsub_pdata2bpp( bs_t *, uint8_t *, int, int * );
1000 static void dvbsub_pdata4bpp( bs_t *, uint8_t *, int, int * );
1001 static void dvbsub_pdata8bpp( bs_t *, uint8_t *, int, int * );
1003 static void decode_object( decoder_t *p_dec, bs_t *s )
1005 decoder_sys_t *p_sys = p_dec->p_sys;
1006 dvbsub_region_t *p_region;
1007 int i_segment_length, i_coding_method, i_version, i_id, i;
1008 vlc_bool_t b_non_modify_color;
1010 /* ETSI 300-743 paragraph 7.2.4
1011 * sync_byte, segment_type and page_id have already been processed.
1013 i_segment_length = bs_read( s, 16 );
1014 i_id = bs_read( s, 16 );
1015 i_version = bs_read( s, 4 );
1016 i_coding_method = bs_read( s, 2 );
1018 if( i_coding_method > 1 )
1020 msg_Dbg( p_dec, "unknown DVB subtitling coding %d is not handled!", i_coding_method );
1021 bs_skip( s, 8 * (i_segment_length - 2) - 6 );
1025 /* Check if the object needs to be rendered in at least one
1027 for( p_region = p_sys->p_regions; p_region != NULL;
1028 p_region = p_region->p_next )
1030 for( i = 0; i < p_region->i_object_defs; i++ )
1031 if( p_region->p_object_defs[i].i_id == i_id ) break;
1033 if( i != p_region->i_object_defs ) break;
1037 bs_skip( s, 8 * (i_segment_length - 2) - 6 );
1042 msg_Dbg( p_dec, "new object: %i", i_id );
1045 b_non_modify_color = bs_read( s, 1 );
1046 bs_skip( s, 1 ); /* Reserved */
1048 if( i_coding_method == 0x00 )
1050 int i_topfield, i_bottomfield;
1051 uint8_t *p_topfield, *p_bottomfield;
1053 i_topfield = bs_read( s, 16 );
1054 i_bottomfield = bs_read( s, 16 );
1055 p_topfield = s->p_start + bs_pos( s ) / 8;
1056 p_bottomfield = p_topfield + i_topfield;
1058 bs_skip( s, 8 * (i_segment_length - 7) );
1061 if( i_segment_length < i_topfield + i_bottomfield + 7 ||
1062 p_topfield + i_topfield + i_bottomfield > s->p_end )
1064 msg_Dbg( p_dec, "corrupted object data" );
1068 for( p_region = p_sys->p_regions; p_region != NULL;
1069 p_region = p_region->p_next )
1071 for( i = 0; i < p_region->i_object_defs; i++ )
1073 if( p_region->p_object_defs[i].i_id != i_id ) continue;
1075 dvbsub_render_pdata( p_dec, p_region,
1076 p_region->p_object_defs[i].i_x,
1077 p_region->p_object_defs[i].i_y,
1078 p_topfield, i_topfield );
1082 dvbsub_render_pdata( p_dec, p_region,
1083 p_region->p_object_defs[i].i_x,
1084 p_region->p_object_defs[i].i_y + 1,
1085 p_bottomfield, i_bottomfield );
1089 /* Duplicate the top field */
1090 dvbsub_render_pdata( p_dec, p_region,
1091 p_region->p_object_defs[i].i_x,
1092 p_region->p_object_defs[i].i_y + 1,
1093 p_topfield, i_topfield );
1100 /* DVB subtitling as characters */
1101 int i_number_of_codes = bs_read( s, 8 );
1102 uint8_t* p_start = s->p_start + bs_pos( s ) / 8;
1105 if( i_segment_length < i_number_of_codes*2 + 4 ||
1106 p_start + i_number_of_codes*2 > s->p_end )
1108 msg_Dbg( p_dec, "corrupted object data" );
1112 for( p_region = p_sys->p_regions; p_region != NULL;
1113 p_region = p_region->p_next )
1115 for( i = 0; i < p_region->i_object_defs; i++ )
1119 if( p_region->p_object_defs[i].i_id != i_id ) continue;
1121 p_region->p_object_defs[i].psz_text =
1122 realloc( p_region->p_object_defs[i].psz_text,
1123 i_number_of_codes + 1 );
1125 /* FIXME 16bits -> char ??? See Preamble */
1126 for( j = 0; j < i_number_of_codes; j++ )
1128 p_region->p_object_defs[i].psz_text[j] = (char)(bs_read( s, 16 ) & 0xFF);
1130 p_region->p_object_defs[i].psz_text[j] = 0;
1136 msg_Dbg( p_dec, "end object: %i", i_id );
1140 static void dvbsub_render_pdata( decoder_t *p_dec, dvbsub_region_t *p_region,
1142 uint8_t *p_field, int i_field )
1149 if( !p_region->p_pixbuf )
1151 msg_Err( p_dec, "region %i has no pixel buffer!", p_region->i_id );
1154 if( i_y < 0 || i_x < 0 || i_y >= p_region->i_height ||
1155 i_x >= p_region->i_width )
1157 msg_Dbg( p_dec, "invalid offset (%i,%i)", i_x, i_y );
1161 p_pixbuf = p_region->p_pixbuf + i_y * p_region->i_width;
1162 bs_init( &bs, p_field, i_field );
1164 while( !bs_eof( &bs ) )
1167 if( i_y >= p_region->i_height ) return;
1169 switch( bs_read( &bs, 8 ) )
1172 dvbsub_pdata2bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1177 dvbsub_pdata4bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1182 dvbsub_pdata8bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1189 /* We don't use map tables */
1192 case 0xf0: /* End of line code */
1193 p_pixbuf += 2*p_region->i_width;
1194 i_offset = 0; i_y += 2;
1200 static void dvbsub_pdata2bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1202 vlc_bool_t b_stop = 0;
1204 while( !b_stop && !bs_eof( s ) )
1206 int i_count = 0, i_color = 0;
1208 if( (i_color = bs_read( s, 2 )) != 0x00 )
1214 if( bs_read( s, 1 ) == 0x01 ) // Switch1
1216 i_count = 3 + bs_read( s, 3 );
1217 i_color = bs_read( s, 2 );
1221 if( bs_read( s, 1 ) == 0x00 ) //Switch2
1223 switch( bs_read( s, 2 ) ) //Switch3
1232 i_count = 12 + bs_read( s, 4 );
1233 i_color = bs_read( s, 2 );
1236 i_count = 29 + bs_read( s, 8 );
1237 i_color = bs_read( s, 2 );
1245 /* 1 pixel color 0 */
1251 if( !i_count ) continue;
1254 if( i_count + *pi_off > i_width ) break;
1256 if( i_count == 1 ) p[*pi_off] = i_color;
1257 else memset( p + *pi_off, i_color, i_count );
1259 (*pi_off) += i_count;
1265 static void dvbsub_pdata4bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1267 vlc_bool_t b_stop = 0;
1269 while( !b_stop && !bs_eof( s ) )
1271 int i_count = 0, i_color = 0;
1273 if( (i_color = bs_read( s, 4 )) != 0x00 )
1280 if( bs_read( s, 1 ) == 0x00 ) // Switch1
1282 if( bs_show( s, 3 ) != 0x00 )
1284 i_count = 2 + bs_read( s, 3 );
1294 if( bs_read( s, 1 ) == 0x00) //Switch2
1296 i_count = 4 + bs_read( s, 2 );
1297 i_color = bs_read( s, 4 );
1301 switch ( bs_read( s, 2 ) ) //Switch3
1310 i_count = 9 + bs_read( s, 4 );
1311 i_color = bs_read( s, 4 );
1314 i_count= 25 + bs_read( s, 8 );
1315 i_color = bs_read( s, 4 );
1322 if( !i_count ) continue;
1325 if( i_count + *pi_off > i_width ) break;
1327 if( i_count == 1 ) p[*pi_off] = i_color;
1328 else memset( p + *pi_off, i_color, i_count );
1330 (*pi_off) += i_count;
1336 static void dvbsub_pdata8bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1338 vlc_bool_t b_stop = 0;
1340 while( !b_stop && !bs_eof( s ) )
1342 int i_count = 0, i_color = 0;
1344 if( (i_color = bs_read( s, 8 )) != 0x00 )
1351 if( bs_read( s, 1 ) == 0x00 ) // Switch1
1353 if( bs_show( s, 7 ) != 0x00 )
1355 i_count = bs_read( s, 7 );
1365 i_count = bs_read( s, 7 );
1366 i_color = bs_read( s, 8 );
1370 if( !i_count ) continue;
1373 if( i_count + *pi_off > i_width ) break;
1375 if( i_count == 1 ) p[*pi_off] = i_color;
1376 else memset( p + *pi_off, i_color, i_count );
1378 (*pi_off) += i_count;
1384 static void free_all( decoder_t *p_dec )
1386 decoder_sys_t *p_sys = p_dec->p_sys;
1387 dvbsub_region_t *p_reg, *p_reg_next;
1388 dvbsub_clut_t *p_clut, *p_clut_next;
1390 if( p_sys->p_display ) free( p_sys->p_display );
1392 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut_next )
1394 p_clut_next = p_clut->p_next;
1397 p_sys->p_cluts = NULL;
1399 for( p_reg = p_sys->p_regions; p_reg != NULL; p_reg = p_reg_next )
1403 p_reg_next = p_reg->p_next;
1404 for( i = 0; i < p_reg->i_object_defs; i++ )
1405 if( p_reg->p_object_defs[i].psz_text )
1406 free( p_reg->p_object_defs[i].psz_text );
1407 if( p_reg->i_object_defs ) free( p_reg->p_object_defs );
1408 if( p_reg->p_pixbuf ) free( p_reg->p_pixbuf );
1411 p_sys->p_regions = NULL;
1415 if( p_sys->p_page->i_region_defs )
1416 free( p_sys->p_page->p_region_defs );
1417 free( p_sys->p_page );
1419 p_sys->p_page = NULL;
1422 static subpicture_t *render( decoder_t *p_dec )
1424 decoder_sys_t *p_sys = p_dec->p_sys;
1425 subpicture_t *p_spu;
1426 subpicture_region_t **pp_spu_region;
1427 int i, j, i_timeout = 0;
1429 /* Allocate the subpicture internal data. */
1430 p_spu = p_dec->pf_spu_buffer_new( p_dec );
1431 if( !p_spu ) return NULL;
1433 pp_spu_region = &p_spu->p_region;
1435 /* Loop on region definitions */
1438 msg_Dbg( p_dec, "rendering %i regions", p_sys->p_page->i_region_defs );
1441 for( i = 0; p_sys->p_page && i < p_sys->p_page->i_region_defs; i++ )
1443 dvbsub_region_t *p_region;
1444 dvbsub_regiondef_t *p_regiondef;
1445 dvbsub_clut_t *p_clut;
1446 dvbsub_color_t *p_color;
1447 subpicture_region_t *p_spu_region;
1448 uint8_t *p_src, *p_dst;
1452 i_timeout = p_sys->p_page->i_timeout;
1454 p_regiondef = &p_sys->p_page->p_region_defs[i];
1457 msg_Dbg( p_dec, "rendering region %i (%i,%i)", i,
1458 p_regiondef->i_x, p_regiondef->i_y );
1461 /* Find associated region */
1462 for( p_region = p_sys->p_regions; p_region != NULL;
1463 p_region = p_region->p_next )
1465 if( p_regiondef->i_id == p_region->i_id ) break;
1470 msg_Dbg( p_dec, "region %i not found", p_regiondef->i_id );
1474 /* Find associated CLUT */
1475 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut->p_next )
1477 if( p_region->i_clut == p_clut->i_id ) break;
1481 msg_Dbg( p_dec, "clut %i not found", p_region->i_clut );
1485 /* Create new SPU region */
1486 memset( &fmt, 0, sizeof(video_format_t) );
1487 fmt.i_chroma = VLC_FOURCC('Y','U','V','P');
1488 fmt.i_aspect = 0; /* 0 means use aspect ratio of background video */
1489 fmt.i_width = fmt.i_visible_width = p_region->i_width;
1490 fmt.i_height = fmt.i_visible_height = p_region->i_height;
1491 fmt.i_x_offset = fmt.i_y_offset = 0;
1492 p_spu_region = p_spu->pf_create_region( VLC_OBJECT(p_dec), &fmt );
1495 msg_Err( p_dec, "cannot allocate SPU region" );
1498 p_spu_region->i_x = p_regiondef->i_x;
1499 p_spu_region->i_y = p_regiondef->i_y;
1500 p_spu_region->i_align = p_sys->i_spu_position;
1501 *pp_spu_region = p_spu_region;
1502 pp_spu_region = &p_spu_region->p_next;
1505 fmt.p_palette->i_entries = p_region->i_depth == 1 ? 4 :
1506 p_region->i_depth == 2 ? 16 : 256;
1507 p_color = (p_region->i_depth == 1) ? p_clut->c_2b :
1508 (p_region->i_depth == 2) ? p_clut->c_4b : p_clut->c_8b;
1509 for( j = 0; j < fmt.p_palette->i_entries; j++ )
1511 fmt.p_palette->palette[j][0] = p_color[j].Y;
1512 fmt.p_palette->palette[j][1] = p_color[j].Cb; /* U == Cb */
1513 fmt.p_palette->palette[j][2] = p_color[j].Cr; /* V == Cr */
1514 fmt.p_palette->palette[j][3] = 0xff - p_color[j].T;
1517 p_src = p_region->p_pixbuf;
1518 p_dst = p_spu_region->picture.Y_PIXELS;
1519 i_pitch = p_spu_region->picture.Y_PITCH;
1521 /* Copy pixel buffer */
1522 for( j = 0; j < p_region->i_height; j++ )
1524 memcpy( p_dst, p_src, p_region->i_width );
1525 p_src += p_region->i_width;
1529 /* Check subtitles encoded as strings of characters
1530 * (since there are not rendered in the pixbuffer) */
1531 for( j = 0; j < p_region->i_object_defs; j++ )
1533 dvbsub_objectdef_t *p_object_def = &p_region->p_object_defs[j];
1535 if( p_object_def->i_type != 1 || !p_object_def->psz_text )
1538 /* Create new SPU region */
1539 memset( &fmt, 0, sizeof(video_format_t) );
1540 fmt.i_chroma = VLC_FOURCC('T','E','X','T');
1541 fmt.i_aspect = VOUT_ASPECT_FACTOR;
1542 fmt.i_width = fmt.i_visible_width = p_region->i_width;
1543 fmt.i_height = fmt.i_visible_height = p_region->i_height;
1544 fmt.i_x_offset = fmt.i_y_offset = 0;
1545 p_spu_region = p_spu->pf_create_region( VLC_OBJECT(p_dec), &fmt );
1548 msg_Err( p_dec, "cannot allocate SPU region" );
1552 p_spu_region->psz_text = strdup( p_object_def->psz_text );
1553 p_spu_region->i_x = p_regiondef->i_x + p_object_def->i_x;
1554 p_spu_region->i_y = p_regiondef->i_y + p_object_def->i_y;
1555 p_spu_region->i_align = p_sys->i_spu_position;
1556 *pp_spu_region = p_spu_region;
1557 pp_spu_region = &p_spu_region->p_next;
1561 /* Set the pf_render callback */
1562 p_spu->i_start = p_sys->i_pts;
1563 p_spu->b_ephemer = VLC_TRUE;
1564 p_spu->b_pausable = VLC_TRUE;
1565 //p_spu->b_fade = VLC_TRUE;
1566 //p_spu->i_stop = p_spu->i_start + (mtime_t) (i_timeout * 1000000);
1568 /* Correct positioning of SPU */
1569 p_spu->b_absolute = p_sys->b_absolute;
1570 p_spu->i_flags = p_sys->i_spu_position;
1571 p_spu->i_x = p_sys->i_spu_x;
1572 p_spu->i_y = p_sys->i_spu_y;
1573 p_spu->i_original_picture_width = 720;
1574 p_spu->i_original_picture_height = 576;
1576 if( p_sys->p_display )
1578 p_spu->i_original_picture_width = p_sys->p_display->i_width;
1579 p_spu->i_original_picture_height = p_sys->p_display->i_height;
1581 if( p_sys->p_display->b_windowed )
1583 /* TODO: check that this actually works */
1584 p_spu->i_original_picture_width = p_sys->p_display->i_max_x - p_sys->p_display->i_x;
1585 p_spu->i_original_picture_height = p_sys->p_display->i_max_y - p_sys->p_display->i_y;
1586 p_spu->i_x += p_sys->p_display->i_x;
1587 p_spu->i_y += p_sys->p_display->i_y;
1594 /*****************************************************************************
1595 * encoder_sys_t : encoder descriptor
1596 *****************************************************************************/
1597 typedef struct encoder_region_t
1604 struct encoder_sys_t
1606 unsigned int i_page_ver;
1607 unsigned int i_region_ver;
1608 unsigned int i_clut_ver;
1611 encoder_region_t *p_regions;
1615 /* subpicture positioning */
1620 static void encode_page_composition( encoder_t *, bs_t *, subpicture_t * );
1621 static void encode_clut( encoder_t *, bs_t *, subpicture_t * );
1622 static void encode_region_composition( encoder_t *, bs_t *, subpicture_t * );
1623 static void encode_object( encoder_t *, bs_t *, subpicture_t * );
1625 /*****************************************************************************
1626 * OpenEncoder: probe the encoder and return score
1627 *****************************************************************************/
1628 static int OpenEncoder( vlc_object_t *p_this )
1630 encoder_t *p_enc = (encoder_t *)p_this;
1631 encoder_sys_t *p_sys;
1634 if( p_enc->fmt_out.i_codec != VLC_FOURCC('d','v','b','s') &&
1637 return VLC_EGENERIC;
1640 /* Allocate the memory needed to store the decoder's structure */
1641 if( ( p_sys = (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
1643 msg_Err( p_enc, "out of memory" );
1646 p_enc->p_sys = p_sys;
1648 p_enc->pf_encode_sub = Encode;
1649 p_enc->fmt_out.i_codec = VLC_FOURCC('d','v','b','s');
1650 p_enc->fmt_out.subs.dvb.i_id = 1 << 16 | 1;
1652 config_ChainParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
1654 p_sys->i_page_ver = 0;
1655 p_sys->i_region_ver = 0;
1656 p_sys->i_clut_ver = 0;
1657 p_sys->i_regions = 0;
1658 p_sys->p_regions = 0;
1660 var_Create( p_this, ENC_CFG_PREFIX "x", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
1661 var_Get( p_this, ENC_CFG_PREFIX "x", &val );
1662 p_sys->i_offset_x = val.i_int;
1663 var_Create( p_this, ENC_CFG_PREFIX "y", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
1664 var_Get( p_this, ENC_CFG_PREFIX "y", &val );
1665 p_sys->i_offset_y = val.i_int;
1670 /* FIXME: this routine is a hack to convert VLC_FOURCC('Y','U','V','A')
1671 * into VLC_FOURCC('Y','U','V','P')
1673 static subpicture_t *YuvaYuvp( encoder_t *p_enc, subpicture_t *p_subpic )
1675 subpicture_region_t *p_region = NULL;
1677 if( !p_subpic ) return NULL;
1679 for( p_region = p_subpic->p_region; p_region; p_region = p_region->p_next )
1681 video_format_t *p_fmt = &p_region->fmt;
1682 int i = 0, j = 0, n = 0, p = 0;
1683 int i_max_entries = 256;
1685 #ifdef RANDOM_DITHERING
1686 int i_seed = 0xdeadbeef; /* random seed */
1690 int i_pixels = p_region->picture.p[0].i_visible_lines
1691 * p_region->picture.p[0].i_pitch;
1692 int i_iterator = p_region->picture.p[0].i_visible_lines * 3 / 4
1693 * p_region->picture.p[0].i_pitch
1694 + p_region->picture.p[0].i_pitch * 1 / 3;
1695 int i_tolerance = 0;
1698 msg_Dbg( p_enc, "YuvaYuvp: i_pixels=%d, i_iterator=%d", i_pixels, i_iterator );
1700 p_fmt->i_chroma = VLC_FOURCC('Y','U','V','P');
1701 p_fmt->p_palette = (video_palette_t *) malloc( sizeof( video_palette_t ) );
1702 if( !p_fmt->p_palette ) break;
1703 p_fmt->p_palette->i_entries = 0;
1705 /* Find best iterator using Euclide’s algorithm */
1706 for( ; i_iterator > 1 ; i_iterator-- )
1725 /* Count colors, build best palette */
1726 for( i_tolerance = 0; i_tolerance < 128; i_tolerance++ )
1728 vlc_bool_t b_success = VLC_TRUE;
1729 p_fmt->p_palette->i_entries = 0;
1731 for( i = 0; i < i_pixels ; )
1734 y = p_region->picture.p[0].p_pixels[i];
1735 u = p_region->picture.p[1].p_pixels[i];
1736 v = p_region->picture.p[2].p_pixels[i];
1737 a = p_region->picture.p[3].p_pixels[i];
1738 for( j = 0; j < p_fmt->p_palette->i_entries; j++ )
1740 if( abs((int)p_fmt->p_palette->palette[j][0] - (int)y) <= i_tolerance &&
1741 abs((int)p_fmt->p_palette->palette[j][1] - (int)u) <= i_tolerance &&
1742 abs((int)p_fmt->p_palette->palette[j][2] - (int)v) <= i_tolerance &&
1743 abs((int)p_fmt->p_palette->palette[j][3] - (int)a) <= i_tolerance / 2 )
1748 if( j == p_fmt->p_palette->i_entries )
1750 p_fmt->p_palette->palette[j][0] = y;
1751 p_fmt->p_palette->palette[j][1] = u;
1752 p_fmt->p_palette->palette[j][2] = v;
1753 p_fmt->p_palette->palette[j][3] = a;
1754 p_fmt->p_palette->i_entries++;
1756 if( p_fmt->p_palette->i_entries >= i_max_entries )
1758 b_success = VLC_FALSE;
1775 msg_Dbg( p_enc, "best palette has %d colors", p_fmt->p_palette->i_entries );
1778 #ifndef RANDOM_DITHERING
1779 pi_delta = malloc( ( p_region->picture.p[0].i_pitch + 1 )
1780 * sizeof(int) * 4 );
1781 for( i = 0; i < (p_region->picture.p[0].i_pitch + 1) * 4 ; i++ )
1787 /* Fill image with our new colours */
1788 for( p = 0; p < p_region->picture.p[0].i_visible_lines ; p++ )
1790 int i_ydelta = 0, i_udelta = 0, i_vdelta = 0, i_adelta = 0;
1792 for( n = 0; n < p_region->picture.p[0].i_pitch ; n++ )
1794 int i_offset = p * p_region->picture.p[0].i_pitch + n;
1796 int i_mindist, i_best;
1798 y = (int)p_region->picture.p[0].p_pixels[i_offset];
1799 u = (int)p_region->picture.p[1].p_pixels[i_offset];
1800 v = (int)p_region->picture.p[2].p_pixels[i_offset];
1801 a = (int)p_region->picture.p[3].p_pixels[i_offset];
1803 /* Add dithering compensation */
1804 #ifdef RANDOM_DITHERING
1805 y += ((i_seed & 0xff) - 0x80) * i_tolerance / 0x80;
1806 u += (((i_seed >> 8) & 0xff) - 0x80) * i_tolerance / 0x80;
1807 v += (((i_seed >> 16) & 0xff) - 0x80) * i_tolerance / 0x80;
1808 a += (((i_seed >> 24) & 0xff) - 0x80) * i_tolerance / 0x80;
1810 y += i_ydelta + pi_delta[ n * 4 ];
1811 u += i_udelta + pi_delta[ n * 4 + 1 ];
1812 v += i_vdelta + pi_delta[ n * 4 + 2 ];
1813 a += i_adelta + pi_delta[ n * 4 + 3 ];
1816 /* Find best colour in palette */
1817 for( i_mindist = 99999999, i_best = 0, j = 0; j < p_fmt->p_palette->i_entries; j++ )
1821 i_dist += abs((int)p_fmt->p_palette->palette[j][0] - y);
1822 i_dist += abs((int)p_fmt->p_palette->palette[j][1] - u);
1823 i_dist += abs((int)p_fmt->p_palette->palette[j][2] - v);
1824 i_dist += 2 * abs((int)p_fmt->p_palette->palette[j][3] - a);
1826 if( i_dist < i_mindist )
1833 /* Set pixel to best color */
1834 p_region->picture.p[0].p_pixels[i_offset] = i_best;
1836 /* Update dithering state */
1837 #ifdef RANDOM_DITHERING
1838 i_seed = (i_seed * 0x1283837) ^ 0x789479 ^ (i_seed >> 13);
1840 i_ydelta = y - (int)p_fmt->p_palette->palette[i_best][0];
1841 i_udelta = u - (int)p_fmt->p_palette->palette[i_best][1];
1842 i_vdelta = v - (int)p_fmt->p_palette->palette[i_best][2];
1843 i_adelta = a - (int)p_fmt->p_palette->palette[i_best][3];
1844 pi_delta[ n * 4 ] = i_ydelta * 3 / 8;
1845 pi_delta[ n * 4 + 1 ] = i_udelta * 3 / 8;
1846 pi_delta[ n * 4 + 2 ] = i_vdelta * 3 / 8;
1847 pi_delta[ n * 4 + 3 ] = i_adelta * 3 / 8;
1848 i_ydelta = i_ydelta * 5 / 8;
1849 i_udelta = i_udelta * 5 / 8;
1850 i_vdelta = i_vdelta * 5 / 8;
1851 i_adelta = i_adelta * 5 / 8;
1855 #ifndef RANDOM_DITHERING
1860 for( i = p_fmt->p_palette->i_entries; i < i_max_entries; i++ )
1862 p_fmt->p_palette->palette[i][0] = 0;
1863 p_fmt->p_palette->palette[i][1] = 0;
1864 p_fmt->p_palette->palette[i][2] = 0;
1865 p_fmt->p_palette->palette[i][3] = 0;
1867 p_fmt->p_palette->i_entries = i_max_entries;
1869 msg_Dbg( p_enc, "best palette has %d colors", p_fmt->p_palette->i_entries );
1875 /****************************************************************************
1876 * Encode: the whole thing
1877 ****************************************************************************/
1878 static block_t *Encode( encoder_t *p_enc, subpicture_t *p_subpic )
1880 subpicture_t *p_temp = NULL;
1881 subpicture_region_t *p_region = NULL;
1882 bs_t bits, *s = &bits;
1885 if( !p_subpic || !p_subpic->p_region ) return NULL;
1887 /* FIXME: this is a hack to convert VLC_FOURCC('Y','U','V','A') into
1888 * VLC_FOURCC('Y','U','V','P')
1890 p_region = p_subpic->p_region;
1891 if( p_region->fmt.i_chroma == VLC_FOURCC('Y','U','V','A') )
1893 p_temp = YuvaYuvp( p_enc, p_subpic );
1896 msg_Err( p_enc, "no picture in subpicture" );
1899 p_region = p_subpic->p_region;
1903 if( !p_region ) return NULL;
1905 if( p_region->fmt.i_chroma != VLC_FOURCC('T','E','X','T') &&
1906 p_region->fmt.i_chroma != VLC_FOURCC('Y','U','V','P') ) return NULL;
1908 if( p_region->fmt.p_palette )
1910 switch( p_region->fmt.p_palette->i_entries )
1918 msg_Err( p_enc, "subpicture palette (%d) not handled",
1919 p_region->fmt.p_palette->i_entries );
1926 msg_Dbg( p_enc, "encoding subpicture" );
1928 p_block = block_New( p_enc, 64000 );
1929 bs_init( s, p_block->p_buffer, p_block->i_buffer );
1931 bs_write( s, 8, 0x20 ); /* Data identifier */
1932 bs_write( s, 8, 0x0 ); /* Subtitle stream id */
1934 encode_page_composition( p_enc, s, p_subpic );
1935 encode_region_composition( p_enc, s, p_subpic );
1936 encode_clut( p_enc, s, p_subpic );
1937 encode_object( p_enc, s, p_subpic );
1939 /* End of display */
1940 bs_write( s, 8, 0x0f ); /* Sync byte */
1941 bs_write( s, 8, DVBSUB_ST_ENDOFDISPLAY ); /* Segment type */
1942 bs_write( s, 16, 1 ); /* Page id */
1943 bs_write( s, 16, 0 ); /* Segment length */
1945 bs_write( s, 8, 0xff );/* End marker */
1946 p_block->i_buffer = bs_pos( s ) / 8;
1947 p_block->i_pts = p_block->i_dts = p_subpic->i_start;
1948 if( !p_subpic->b_ephemer && p_subpic->i_stop > p_subpic->i_start )
1950 block_t *p_block_stop;
1952 p_block->i_length = p_subpic->i_stop - p_subpic->i_start;
1954 /* Send another (empty) subtitle to signal the end of display */
1955 p_block_stop = block_New( p_enc, 64000 );
1956 bs_init( s, p_block_stop->p_buffer, p_block_stop->i_buffer );
1957 bs_write( s, 8, 0x20 ); /* Data identifier */
1958 bs_write( s, 8, 0x0 ); /* Subtitle stream id */
1959 encode_page_composition( p_enc, s, 0 );
1960 bs_write( s, 8, 0x0f ); /* Sync byte */
1961 bs_write( s, 8, DVBSUB_ST_ENDOFDISPLAY ); /* Segment type */
1962 bs_write( s, 16, 1 ); /* Page id */
1963 bs_write( s, 16, 0 ); /* Segment length */
1964 bs_write( s, 8, 0xff );/* End marker */
1965 p_block_stop->i_buffer = bs_pos( s ) / 8;
1966 p_block_stop->i_pts = p_block_stop->i_dts = p_subpic->i_stop;
1967 block_ChainAppend( &p_block, p_block_stop );
1968 p_block_stop->i_length = 100000; /* p_subpic->i_stop - p_subpic->i_start; */
1971 msg_Dbg( p_enc, "subpicture encoded properly" );
1976 /*****************************************************************************
1977 * CloseEncoder: encoder destruction
1978 *****************************************************************************/
1979 static void CloseEncoder( vlc_object_t *p_this )
1981 encoder_t *p_enc = (encoder_t *)p_this;
1982 encoder_sys_t *p_sys = p_enc->p_sys;
1984 var_Destroy( p_this , ENC_CFG_PREFIX "x" );
1985 var_Destroy( p_this , ENC_CFG_PREFIX "y" );
1986 var_Destroy( p_this , ENC_CFG_PREFIX "timeout" );
1988 if( p_sys->i_regions ) free( p_sys->p_regions );
1992 static void encode_page_composition( encoder_t *p_enc, bs_t *s,
1993 subpicture_t *p_subpic )
1995 encoder_sys_t *p_sys = p_enc->p_sys;
1996 subpicture_region_t *p_region;
1997 vlc_bool_t b_mode_change = VLC_FALSE;
1998 int i_regions, i_timeout;
2000 bs_write( s, 8, 0x0f ); /* Sync byte */
2001 bs_write( s, 8, DVBSUB_ST_PAGE_COMPOSITION ); /* Segment type */
2002 bs_write( s, 16, 1 ); /* Page id */
2004 for( i_regions = 0, p_region = p_subpic ? p_subpic->p_region : 0;
2005 p_region; p_region = p_region->p_next, i_regions++ )
2007 if( i_regions >= p_sys->i_regions )
2009 encoder_region_t region;
2010 region.i_width = region.i_height = 0;
2012 realloc( p_sys->p_regions, sizeof(encoder_region_t) *
2013 (p_sys->i_regions + 1) );
2014 p_sys->p_regions[p_sys->i_regions++] = region;
2017 if( ( p_sys->p_regions[i_regions].i_width <
2018 (int)p_region->fmt.i_visible_width ) ||
2019 ( p_sys->p_regions[i_regions].i_width >
2020 (int)p_region->fmt.i_visible_width ) )
2022 b_mode_change = VLC_TRUE;
2023 msg_Dbg( p_enc, "region %i width change: %i -> %i",
2024 i_regions, p_sys->p_regions[i_regions].i_width,
2025 p_region->fmt.i_visible_width );
2026 p_sys->p_regions[i_regions].i_width =
2027 p_region->fmt.i_visible_width;
2029 if( p_sys->p_regions[i_regions].i_height <
2030 (int)p_region->fmt.i_visible_height )
2032 b_mode_change = VLC_TRUE;
2033 msg_Dbg( p_enc, "region %i height change: %i -> %i",
2034 i_regions, p_sys->p_regions[i_regions].i_height,
2035 p_region->fmt.i_visible_height );
2036 p_sys->p_regions[i_regions].i_height =
2037 p_region->fmt.i_visible_height;
2041 bs_write( s, 16, i_regions * 6 + 2 ); /* Segment length */
2044 if( p_subpic && !p_subpic->b_ephemer &&
2045 p_subpic->i_stop > p_subpic->i_start )
2047 i_timeout = (p_subpic->i_stop - p_subpic->i_start) / 1000000;
2050 bs_write( s, 8, i_timeout ); /* Timeout */
2051 bs_write( s, 4, p_sys->i_page_ver++ );
2052 bs_write( s, 2, b_mode_change ?
2053 DVBSUB_PCS_STATE_CHANGE : DVBSUB_PCS_STATE_ACQUISITION );
2054 bs_write( s, 2, 0 ); /* Reserved */
2056 for( i_regions = 0, p_region = p_subpic ? p_subpic->p_region : 0;
2057 p_region; p_region = p_region->p_next, i_regions++ )
2059 bs_write( s, 8, i_regions );
2060 bs_write( s, 8, 0 ); /* Reserved */
2061 if( (p_sys->i_offset_x > 0) && (p_sys->i_offset_y > 0) )
2063 bs_write( s, 16, p_sys->i_offset_x ); /* override x position */
2064 bs_write( s, 16, p_sys->i_offset_y ); /* override y position */
2068 bs_write( s, 16, p_subpic->i_x + p_region->i_x );
2069 bs_write( s, 16, p_subpic->i_y + p_region->i_y );
2074 static void encode_clut( encoder_t *p_enc, bs_t *s, subpicture_t *p_subpic )
2076 encoder_sys_t *p_sys = p_enc->p_sys;
2077 subpicture_region_t *p_region = p_subpic->p_region;
2078 video_palette_t *p_pal, pal;
2082 if( !p_region ) return;
2084 if( p_region->fmt.i_chroma == VLC_FOURCC('Y','U','V','P') )
2086 p_pal = p_region->fmt.p_palette;
2091 for( i = 0; i < 4; i++ )
2093 pal.palette[i][0] = 0;
2094 pal.palette[i][1] = 0;
2095 pal.palette[i][2] = 0;
2096 pal.palette[i][3] = 0;
2101 bs_write( s, 8, 0x0f ); /* Sync byte */
2102 bs_write( s, 8, DVBSUB_ST_CLUT_DEFINITION ); /* Segment type */
2103 bs_write( s, 16, 1 ); /* Page id */
2105 bs_write( s, 16, p_pal->i_entries * 6 + 2 ); /* Segment length */
2106 bs_write( s, 8, 1 ); /* Clut id */
2107 bs_write( s, 4, p_sys->i_clut_ver++ );
2108 bs_write( s, 4, 0 ); /* Reserved */
2110 for( i = 0; i < p_pal->i_entries; i++ )
2112 bs_write( s, 8, i ); /* Clut entry id */
2113 bs_write( s, 1, p_pal->i_entries == 4 ); /* 2bit/entry flag */
2114 bs_write( s, 1, p_pal->i_entries == 16 ); /* 4bit/entry flag */
2115 bs_write( s, 1, p_pal->i_entries == 256 ); /* 8bit/entry flag */
2116 bs_write( s, 4, 0 ); /* Reserved */
2117 bs_write( s, 1, 1 ); /* Full range flag */
2118 bs_write( s, 8, p_pal->palette[i][3] ? /* Y value */
2119 (p_pal->palette[i][0] ? p_pal->palette[i][0] : 16) : 0 );
2120 bs_write( s, 8, p_pal->palette[i][1] ); /* Cr value */
2121 bs_write( s, 8, p_pal->palette[i][2] ); /* Cb value */
2122 bs_write( s, 8, 0xff - p_pal->palette[i][3] ); /* T value */
2126 static void encode_region_composition( encoder_t *p_enc, bs_t *s,
2127 subpicture_t *p_subpic )
2129 encoder_sys_t *p_sys = p_enc->p_sys;
2130 subpicture_region_t *p_region;
2133 for( i_region = 0, p_region = p_subpic->p_region; p_region;
2134 p_region = p_region->p_next, i_region++ )
2136 int i_entries = 4, i_depth = 0x1, i_bg = 0;
2138 p_region->fmt.i_chroma == VLC_FOURCC('T','E','X','T');
2142 video_palette_t *p_pal = p_region->fmt.p_palette;
2146 msg_Err( p_enc, "subpicture has no palette - ignoring it" );
2150 i_entries = p_pal->i_entries;
2151 i_depth = i_entries == 4 ? 0x1 : i_entries == 16 ? 0x2 : 0x3;
2153 for( i_bg = 0; i_bg < p_pal->i_entries; i_bg++ )
2155 if( !p_pal->palette[i_bg][3] ) break;
2159 bs_write( s, 8, 0x0f ); /* Sync byte */
2160 bs_write( s, 8, DVBSUB_ST_REGION_COMPOSITION ); /* Segment type */
2161 bs_write( s, 16, 1 ); /* Page id */
2163 bs_write( s, 16, 10 + 6 + (b_text ? 2 : 0) ); /* Segment length */
2164 bs_write( s, 8, i_region );
2165 bs_write( s, 4, p_sys->i_region_ver++ );
2167 /* Region attributes */
2168 bs_write( s, 1, i_bg < i_entries ); /* Fill */
2169 bs_write( s, 3, 0 ); /* Reserved */
2170 bs_write( s, 16, p_sys->p_regions[i_region].i_width );
2171 bs_write( s, 16, p_sys->p_regions[i_region].i_height );
2172 bs_write( s, 3, i_depth ); /* Region level of compatibility */
2173 bs_write( s, 3, i_depth ); /* Region depth */
2174 bs_write( s, 2, 0 ); /* Reserved */
2175 bs_write( s, 8, 1 ); /* Clut id */
2176 bs_write( s, 8, i_bg ); /* region 8bit pixel code */
2177 bs_write( s, 4, i_bg ); /* region 4bit pixel code */
2178 bs_write( s, 2, i_bg ); /* region 2bit pixel code */
2179 bs_write( s, 2, 0 ); /* Reserved */
2181 /* In our implementation we only have 1 object per region */
2182 bs_write( s, 16, i_region );
2183 bs_write( s, 2, b_text ? DVBSUB_OT_BASIC_CHAR:DVBSUB_OT_BASIC_BITMAP );
2184 bs_write( s, 2, 0 ); /* object provider flag */
2185 bs_write( s, 12, 0 );/* object horizontal position */
2186 bs_write( s, 4, 0 ); /* Reserved */
2187 bs_write( s, 12, 0 );/* object vertical position */
2191 bs_write( s, 8, 1 ); /* foreground pixel code */
2192 bs_write( s, 8, 0 ); /* background pixel code */
2197 static void encode_pixel_data( encoder_t *p_enc, bs_t *s,
2198 subpicture_region_t *p_region,
2201 static void encode_object( encoder_t *p_enc, bs_t *s, subpicture_t *p_subpic )
2203 encoder_sys_t *p_sys = p_enc->p_sys;
2204 subpicture_region_t *p_region;
2207 int i_length_pos, i_update_pos, i_pixel_data_pos;
2209 for( i_region = 0, p_region = p_subpic->p_region; p_region;
2210 p_region = p_region->p_next, i_region++ )
2212 bs_write( s, 8, 0x0f ); /* Sync byte */
2213 bs_write( s, 8, DVBSUB_ST_OBJECT_DATA ); /* Segment type */
2214 bs_write( s, 16, 1 ); /* Page id */
2216 i_length_pos = bs_pos( s );
2217 bs_write( s, 16, 0 ); /* Segment length */
2218 bs_write( s, 16, i_region ); /* Object id */
2219 bs_write( s, 4, p_sys->i_region_ver++ );
2221 /* object coding method */
2222 switch( p_region->fmt.i_chroma )
2224 case VLC_FOURCC( 'Y','U','V','P' ):
2225 bs_write( s, 2, 0 );
2227 case VLC_FOURCC( 'T','E','X','T' ):
2228 bs_write( s, 2, 1 );
2231 msg_Err( p_enc, "FOURCC %d not supported by encoder.", p_region->fmt.i_chroma );
2235 bs_write( s, 1, 0 ); /* non modifying color flag */
2236 bs_write( s, 1, 0 ); /* Reserved */
2238 if( p_region->fmt.i_chroma == VLC_FOURCC( 'T','E','X','T' ) )
2242 if( !p_region->psz_text ) continue;
2244 i_size = __MIN( strlen( p_region->psz_text ), 256 );
2246 bs_write( s, 8, i_size ); /* number of characters in string */
2247 for( i = 0; i < i_size; i++ )
2249 bs_write( s, 16, p_region->psz_text[i] );
2252 /* Update segment length */
2253 SetWBE( &s->p_start[i_length_pos/8],
2254 (bs_pos(s) - i_length_pos)/8 -2 );
2258 /* Coding of a bitmap object */
2259 i_update_pos = bs_pos( s );
2260 bs_write( s, 16, 0 ); /* topfield data block length */
2261 bs_write( s, 16, 0 ); /* bottomfield data block length */
2264 i_pixel_data_pos = bs_pos( s );
2265 encode_pixel_data( p_enc, s, p_region, VLC_TRUE );
2266 i_pixel_data_pos = ( bs_pos( s ) - i_pixel_data_pos ) / 8;
2267 SetWBE( &s->p_start[i_update_pos/8], i_pixel_data_pos );
2270 i_pixel_data_pos = bs_pos( s );
2271 encode_pixel_data( p_enc, s, p_region, VLC_FALSE );
2272 i_pixel_data_pos = ( bs_pos( s ) - i_pixel_data_pos ) / 8;
2273 SetWBE( &s->p_start[i_update_pos/8+2], i_pixel_data_pos );
2275 /* Stuffing for word alignment */
2277 if( bs_pos( s ) % 16 ) bs_write( s, 8, 0 );
2279 /* Update segment length */
2280 SetWBE( &s->p_start[i_length_pos/8], (bs_pos(s) - i_length_pos)/8 -2 );
2284 static void encode_pixel_line_2bp( encoder_t *p_enc, bs_t *s,
2285 subpicture_region_t *p_region,
2287 static void encode_pixel_line_4bp( encoder_t *p_enc, bs_t *s,
2288 subpicture_region_t *p_region,
2290 static void encode_pixel_line_8bp( encoder_t *p_enc, bs_t *s,
2291 subpicture_region_t *p_region,
2293 static void encode_pixel_data( encoder_t *p_enc, bs_t *s,
2294 subpicture_region_t *p_region,
2297 unsigned int i_line;
2300 if( p_region->fmt.i_chroma != VLC_FOURCC('Y','U','V','P') ) return;
2302 /* Encode line by line */
2303 for( i_line = !b_top; i_line < p_region->fmt.i_visible_height;
2306 switch( p_region->fmt.p_palette->i_entries )
2312 bs_write( s, 8, 0x10 ); /* 2 bit/pixel code string */
2313 encode_pixel_line_2bp( p_enc, s, p_region, i_line );
2317 bs_write( s, 8, 0x11 ); /* 4 bit/pixel code string */
2318 encode_pixel_line_4bp( p_enc, s, p_region, i_line );
2322 bs_write( s, 8, 0x12 ); /* 8 bit/pixel code string */
2323 encode_pixel_line_8bp( p_enc, s, p_region, i_line );
2327 msg_Err( p_enc, "subpicture palette (%i) not handled",
2328 p_region->fmt.p_palette->i_entries );
2332 bs_write( s, 8, 0xf0 ); /* End of object line code */
2336 static void encode_pixel_line_2bp( encoder_t *p_enc, bs_t *s,
2337 subpicture_region_t *p_region,
2340 unsigned int i, i_length = 0;
2341 int i_pitch = p_region->picture.p->i_pitch;
2342 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2343 int i_last_pixel = p_data[0];
2345 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2347 if( i != p_region->fmt.i_visible_width &&
2348 p_data[i] == i_last_pixel && i_length != 284 )
2354 if( i_length == 1 || i_length == 11 || i_length == 28 )
2356 /* 2bit/pixel code */
2357 if( i_last_pixel ) bs_write( s, 2, i_last_pixel );
2360 bs_write( s, 2, 0 );
2361 bs_write( s, 1, 0 );
2362 bs_write( s, 1, 1 ); /* pseudo color 0 */
2371 bs_write( s, 2, i_last_pixel );
2372 bs_write( s, 2, i_last_pixel );
2376 bs_write( s, 2, 0 );
2377 bs_write( s, 1, 0 );
2378 bs_write( s, 1, 0 );
2379 bs_write( s, 2, 1 ); /* 2 * pseudo color 0 */
2382 else if( i_length > 2 )
2384 bs_write( s, 2, 0 );
2385 if( i_length <= 10 )
2387 bs_write( s, 1, 1 );
2388 bs_write( s, 3, i_length - 3 );
2389 bs_write( s, 2, i_last_pixel );
2393 bs_write( s, 1, 0 );
2394 bs_write( s, 1, 0 );
2396 if( i_length <= 27 )
2398 bs_write( s, 2, 2 );
2399 bs_write( s, 4, i_length - 12 );
2400 bs_write( s, 2, i_last_pixel );
2404 bs_write( s, 2, 3 );
2405 bs_write( s, 8, i_length - 29 );
2406 bs_write( s, 2, i_last_pixel );
2411 if( i == p_region->fmt.i_visible_width ) break;
2413 i_last_pixel = p_data[i];
2418 bs_write( s, 2, 0 );
2419 bs_write( s, 1, 0 );
2420 bs_write( s, 1, 0 );
2421 bs_write( s, 2, 0 );
2427 static void encode_pixel_line_4bp( encoder_t *p_enc, bs_t *s,
2428 subpicture_region_t *p_region,
2431 unsigned int i, i_length = 0;
2432 int i_pitch = p_region->picture.p->i_pitch;
2433 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2434 int i_last_pixel = p_data[0];
2436 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2438 if( i != p_region->fmt.i_visible_width &&
2439 p_data[i] == i_last_pixel && i_length != 280 )
2445 if( i_length == 1 || (i_length == 3 && i_last_pixel) || i_length == 8 )
2447 /* 4bit/pixel code */
2448 if( i_last_pixel ) bs_write( s, 4, i_last_pixel );
2451 bs_write( s, 4, 0 );
2452 bs_write( s, 1, 1 );
2453 bs_write( s, 1, 1 );
2454 bs_write( s, 2, 0 ); /* pseudo color 0 */
2463 bs_write( s, 4, i_last_pixel );
2464 bs_write( s, 4, i_last_pixel );
2468 bs_write( s, 4, 0 );
2469 bs_write( s, 1, 1 );
2470 bs_write( s, 1, 1 );
2471 bs_write( s, 2, 1 ); /* 2 * pseudo color 0 */
2474 else if( !i_last_pixel && i_length >= 3 && i_length <= 9 )
2476 bs_write( s, 4, 0 );
2477 bs_write( s, 1, 0 );
2478 bs_write( s, 3, i_length - 2 ); /* (i_length - 2) * color 0 */
2480 else if( i_length > 2 )
2482 bs_write( s, 4, 0 );
2483 bs_write( s, 1, 1 );
2487 bs_write( s, 1, 0 );
2488 bs_write( s, 2, i_length - 4 );
2489 bs_write( s, 4, i_last_pixel );
2493 bs_write( s, 1, 1 );
2495 if( i_length <= 24 )
2497 bs_write( s, 2, 2 );
2498 bs_write( s, 4, i_length - 9 );
2499 bs_write( s, 4, i_last_pixel );
2503 bs_write( s, 2, 3 );
2504 bs_write( s, 8, i_length - 25 );
2505 bs_write( s, 4, i_last_pixel );
2510 if( i == p_region->fmt.i_visible_width ) break;
2512 i_last_pixel = p_data[i];
2517 bs_write( s, 8, 0 );
2523 static void encode_pixel_line_8bp( encoder_t *p_enc, bs_t *s,
2524 subpicture_region_t *p_region,
2527 unsigned int i, i_length = 0;
2528 int i_pitch = p_region->picture.p->i_pitch;
2529 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2530 int i_last_pixel = p_data[0];
2532 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2534 if( i != p_region->fmt.i_visible_width &&
2535 p_data[i] == i_last_pixel && i_length != 127 )
2541 if( i_length == 1 && i_last_pixel )
2543 /* 8bit/pixel code */
2544 bs_write( s, 8, i_last_pixel );
2546 else if( i_length == 2 && i_last_pixel )
2548 /* 8bit/pixel code */
2549 bs_write( s, 8, i_last_pixel );
2550 bs_write( s, 8, i_last_pixel );
2552 else if( i_length <= 127 )
2554 bs_write( s, 8, 0 );
2558 bs_write( s, 1, 0 );
2559 bs_write( s, 7, i_length ); /* pseudo color 0 */
2563 bs_write( s, 1, 1 );
2564 bs_write( s, 7, i_length );
2565 bs_write( s, 8, i_last_pixel );
2569 if( i == p_region->fmt.i_visible_width ) break;
2571 i_last_pixel = p_data[i];
2576 bs_write( s, 8, 0 );
2577 bs_write( s, 8, 0 );