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 *****************************************************************************/
47 #include <vlc_plugin.h>
49 #include <vlc_codec.h>
54 /* #define DEBUG_DVBSUB 1 */
56 #define POSX_TEXT N_("Decoding X coordinate")
57 #define POSX_LONGTEXT N_("X coordinate of the rendered subtitle")
59 #define POSY_TEXT N_("Decoding Y coordinate")
60 #define POSY_LONGTEXT N_("Y coordinate of the rendered subtitle")
62 #define POS_TEXT N_("Subpicture position")
63 #define POS_LONGTEXT N_( \
64 "You can enforce the subpicture position on the video " \
65 "(0=center, 1=left, 2=right, 4=top, 8=bottom, you can " \
66 "also use combinations of these values, e.g. 6=top-right).")
68 #define ENC_POSX_TEXT N_("Encoding X coordinate")
69 #define ENC_POSX_LONGTEXT N_("X coordinate of the encoded subtitle" )
70 #define ENC_POSY_TEXT N_("Encoding Y coordinate")
71 #define ENC_POSY_LONGTEXT N_("Y coordinate of the encoded subtitle" )
73 static const int pi_pos_values[] = { 0, 1, 2, 4, 8, 5, 6, 9, 10 };
74 static const char *const ppsz_pos_descriptions[] =
75 { N_("Center"), N_("Left"), N_("Right"), N_("Top"), N_("Bottom"),
76 N_("Top-Left"), N_("Top-Right"), N_("Bottom-Left"), N_("Bottom-Right") };
78 /*****************************************************************************
80 *****************************************************************************/
81 static int Open ( vlc_object_t * );
82 static void Close( vlc_object_t * );
83 static subpicture_t *Decode( decoder_t *, block_t ** );
85 static int OpenEncoder ( vlc_object_t * );
86 static void CloseEncoder( vlc_object_t * );
87 static block_t *Encode ( encoder_t *, subpicture_t * );
90 # define DVBSUB_CFG_PREFIX "dvbsub-"
91 set_description( N_("DVB subtitles decoder") );
92 set_capability( "decoder", 50 );
93 set_category( CAT_INPUT );
94 set_subcategory( SUBCAT_INPUT_SCODEC );
95 set_callbacks( Open, Close );
97 add_integer( DVBSUB_CFG_PREFIX "position", 8, NULL, POS_TEXT, POS_LONGTEXT, true );
98 change_integer_list( pi_pos_values, ppsz_pos_descriptions, 0 );
99 add_integer( DVBSUB_CFG_PREFIX "x", -1, NULL, POSX_TEXT, POSX_LONGTEXT, false );
100 add_integer( DVBSUB_CFG_PREFIX "y", -1, NULL, POSY_TEXT, POSY_LONGTEXT, false );
102 # define ENC_CFG_PREFIX "sout-dvbsub-"
104 set_description( N_("DVB subtitles encoder") );
105 set_capability( "encoder", 100 );
106 set_callbacks( OpenEncoder, CloseEncoder );
108 add_integer( ENC_CFG_PREFIX "x", -1, NULL, ENC_POSX_TEXT, ENC_POSX_LONGTEXT, false );
109 add_integer( ENC_CFG_PREFIX "y", -1, NULL, ENC_POSY_TEXT, ENC_POSY_LONGTEXT, false );
110 add_obsolete_integer( ENC_CFG_PREFIX "timeout" ); /* Suppressed since 0.8.5 */
113 static const char *const ppsz_enc_options[] = { "x", "y", NULL };
115 /****************************************************************************
117 ****************************************************************************
118 * Those structures refer closely to the ETSI 300 743 Object model
119 ****************************************************************************/
121 /* The object definition gives the position of the object in a region [7.2.5] */
122 typedef struct dvbsub_objectdef_s
130 char *psz_text; /* for string of characters objects */
132 } dvbsub_objectdef_t;
134 /* The entry in the palette CLUT */
144 /* The displays dimensions [7.2.1] */
145 typedef struct dvbsub_display_s
162 typedef struct dvbsub_clut_s
166 dvbsub_color_t c_2b[4];
167 dvbsub_color_t c_4b[16];
168 dvbsub_color_t c_8b[256];
170 struct dvbsub_clut_s *p_next;
174 /* The Region is an aera on the image [7.2.3]
175 * with a list of the object definitions associated and a CLUT */
176 typedef struct dvbsub_region_s
191 dvbsub_objectdef_t *p_object_defs;
193 struct dvbsub_region_s *p_next;
197 /* The object definition gives the position of the object in a region */
198 typedef struct dvbsub_regiondef_s
204 } dvbsub_regiondef_t;
206 /* The page defines the list of regions [7.2.2] */
210 int i_timeout; /* in seconds */
215 dvbsub_regiondef_t *p_region_defs;
223 /* Decoder internal data */
234 dvbsub_page_t *p_page;
235 dvbsub_region_t *p_regions;
236 dvbsub_clut_t *p_cluts;
237 dvbsub_display_t *p_display;
238 dvbsub_clut_t default_clut;
242 /* List of different SEGMENT TYPES */
243 /* According to EN 300-743, table 2 */
244 #define DVBSUB_ST_PAGE_COMPOSITION 0x10
245 #define DVBSUB_ST_REGION_COMPOSITION 0x11
246 #define DVBSUB_ST_CLUT_DEFINITION 0x12
247 #define DVBSUB_ST_OBJECT_DATA 0x13
248 #define DVBSUB_ST_DISPLAY_DEFINITION 0x14
249 #define DVBSUB_ST_ENDOFDISPLAY 0x80
250 #define DVBSUB_ST_STUFFING 0xff
251 /* List of different OBJECT TYPES */
252 /* According to EN 300-743, table 6 */
253 #define DVBSUB_OT_BASIC_BITMAP 0x00
254 #define DVBSUB_OT_BASIC_CHAR 0x01
255 #define DVBSUB_OT_COMPOSITE_STRING 0x02
256 /* Pixel DATA TYPES */
257 /* According to EN 300-743, table 9 */
258 #define DVBSUB_DT_2BP_CODE_STRING 0x10
259 #define DVBSUB_DT_4BP_CODE_STRING 0x11
260 #define DVBSUB_DT_8BP_CODE_STRING 0x12
261 #define DVBSUB_DT_24_TABLE_DATA 0x20
262 #define DVBSUB_DT_28_TABLE_DATA 0x21
263 #define DVBSUB_DT_48_TABLE_DATA 0x22
264 #define DVBSUB_DT_END_LINE 0xf0
265 /* List of different Page Composition Segment state */
266 /* According to EN 300-743, 7.2.1 table 3 */
267 #define DVBSUB_PCS_STATE_ACQUISITION 0x01
268 #define DVBSUB_PCS_STATE_CHANGE 0x02
270 /*****************************************************************************
272 *****************************************************************************/
273 static void decode_segment( decoder_t *, bs_t * );
274 static void decode_page_composition( decoder_t *, bs_t * );
275 static void decode_region_composition( decoder_t *, bs_t * );
276 static void decode_object( decoder_t *, bs_t * );
277 static void decode_display_definition( decoder_t *, bs_t * );
278 static void decode_clut( decoder_t *, bs_t * );
279 static void free_all( decoder_t * );
281 static void default_clut_init( decoder_t * );
283 static subpicture_t *render( decoder_t * );
285 /*****************************************************************************
286 * Open: probe the decoder and return score
287 *****************************************************************************
288 * Tries to launch a decoder and return score so that the interface is able
290 *****************************************************************************/
291 static int Open( vlc_object_t *p_this )
293 decoder_t *p_dec = (decoder_t *) p_this;
294 decoder_sys_t *p_sys;
298 if( p_dec->fmt_in.i_codec != VLC_FOURCC('d','v','b','s') )
303 p_dec->pf_decode_sub = Decode;
304 p_sys = p_dec->p_sys = malloc( sizeof(decoder_sys_t) );
307 memset( p_sys, 0, sizeof(decoder_sys_t) );
309 p_sys->i_pts = (mtime_t) 0;
310 p_sys->i_id = p_dec->fmt_in.subs.dvb.i_id & 0xFFFF;
311 p_sys->i_ancillary_id = p_dec->fmt_in.subs.dvb.i_id >> 16;
313 p_sys->p_regions = NULL;
314 p_sys->p_cluts = NULL;
315 p_sys->p_page = NULL;
316 p_sys->p_display = NULL;
318 var_Create( p_this, DVBSUB_CFG_PREFIX "position",
319 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
320 var_Get( p_this, DVBSUB_CFG_PREFIX "position", &val );
321 p_sys->i_spu_position = val.i_int;
322 var_Create( p_this, DVBSUB_CFG_PREFIX "x",
323 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
324 var_Get( p_this, DVBSUB_CFG_PREFIX "x", &val );
326 var_Create( p_this, DVBSUB_CFG_PREFIX "y",
327 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
328 var_Get( p_this, DVBSUB_CFG_PREFIX "y", &val );
331 /* Check if subpicture position was overridden */
332 p_sys->b_absolute = false;
333 p_sys->i_spu_x = p_sys->i_spu_y = 0;
335 if( ( i_posx >= 0 ) && ( i_posy >= 0 ) )
337 p_sys->b_absolute = true;
338 p_sys->i_spu_x = i_posx;
339 p_sys->i_spu_y = i_posy;
342 es_format_Init( &p_dec->fmt_out, SPU_ES, VLC_FOURCC( 'd','v','b','s' ) );
344 default_clut_init( p_dec );
349 /*****************************************************************************
351 *****************************************************************************/
352 static void Close( vlc_object_t *p_this )
354 decoder_t *p_dec = (decoder_t*) p_this;
355 decoder_sys_t *p_sys = p_dec->p_sys;
357 var_Destroy( p_this, DVBSUB_CFG_PREFIX "x" );
358 var_Destroy( p_this, DVBSUB_CFG_PREFIX "y" );
359 var_Destroy( p_this, DVBSUB_CFG_PREFIX "position" );
365 /*****************************************************************************
367 *****************************************************************************/
368 static subpicture_t *Decode( decoder_t *p_dec, block_t **pp_block )
370 decoder_sys_t *p_sys = p_dec->p_sys;
372 subpicture_t *p_spu = NULL;
374 if( ( pp_block == NULL ) || ( *pp_block == NULL ) ) return NULL;
378 p_sys->i_pts = p_block->i_pts;
379 if( p_sys->i_pts <= 0 )
382 /* Some DVB channels send stuffing segments in non-dated packets so
383 * don't complain too loudly. */
384 msg_Warn( p_dec, "non dated subtitle" );
386 block_Release( p_block );
390 bs_init( &p_sys->bs, p_block->p_buffer, p_block->i_buffer );
392 if( bs_read( &p_sys->bs, 8 ) != 0x20 ) /* Data identifier */
394 msg_Dbg( p_dec, "invalid data identifier" );
395 block_Release( p_block );
399 if( bs_read( &p_sys->bs, 8 ) ) /* Subtitle stream id */
401 msg_Dbg( p_dec, "invalid subtitle stream id" );
402 block_Release( p_block );
407 msg_Dbg( p_dec, "subtitle packet received: %"PRId64, p_sys->i_pts );
410 p_sys->b_page = false;
411 while( bs_show( &p_sys->bs, 8 ) == 0x0f ) /* Sync byte */
413 decode_segment( p_dec, &p_sys->bs );
416 if( bs_read( &p_sys->bs, 8 ) != 0xff ) /* End marker */
418 msg_Warn( p_dec, "end marker not found (corrupted subtitle ?)" );
419 block_Release( p_block );
423 /* Check if the page is to be displayed */
424 if( p_sys->p_page && p_sys->b_page )
425 p_spu = render( p_dec );
427 block_Release( p_block );
432 /* following functions are local */
434 /*****************************************************************************
435 * default_clut_init: default clut as defined in EN 300-743 section 10
436 *****************************************************************************/
437 static void default_clut_init( decoder_t *p_dec )
439 decoder_sys_t *p_sys = p_dec->p_sys;
442 #define RGB_TO_Y(r, g, b) ((int16_t) 77 * r + 150 * g + 29 * b) / 256;
443 #define RGB_TO_U(r, g, b) ((int16_t) -44 * r - 87 * g + 131 * b) / 256;
444 #define RGB_TO_V(r, g, b) ((int16_t) 131 * r - 110 * g - 21 * b) / 256;
447 for( i = 0; i < 4; i++ )
449 uint8_t R = 0, G = 0, B = 0, T = 0;
451 if( !(i & 0x2) && !(i & 0x1) ) T = 0xFF;
452 else if( !(i & 0x2) && (i & 0x1) ) R = G = B = 0xFF;
453 else if( (i & 0x2) && !(i & 0x1) ) R = G = B = 0;
454 else R = G = B = 0x7F;
456 p_sys->default_clut.c_2b[i].Y = RGB_TO_Y(R,G,B);
457 p_sys->default_clut.c_2b[i].Cb = RGB_TO_V(R,G,B);
458 p_sys->default_clut.c_2b[i].Cr = RGB_TO_U(R,G,B);
459 p_sys->default_clut.c_2b[i].T = T;
462 /* 16 entries CLUT */
463 for( i = 0; i < 16; i++ )
465 uint8_t R = 0, G = 0, B = 0, T = 0;
469 if( !(i & 0x4) && !(i & 0x2) && !(i & 0x1) )
475 R = (i & 0x1) ? 0xFF : 0;
476 G = (i & 0x2) ? 0xFF : 0;
477 B = (i & 0x4) ? 0xFF : 0;
482 R = (i & 0x1) ? 0x7F : 0;
483 G = (i & 0x2) ? 0x7F : 0;
484 B = (i & 0x4) ? 0x7F : 0;
487 p_sys->default_clut.c_4b[i].Y = RGB_TO_Y(R,G,B);
488 p_sys->default_clut.c_4b[i].Cr = RGB_TO_V(R,G,B);
489 p_sys->default_clut.c_4b[i].Cb = RGB_TO_U(R,G,B);
490 p_sys->default_clut.c_4b[i].T = T;
493 /* 256 entries CLUT */
494 memset( p_sys->default_clut.c_8b, 0xFF, 256 * sizeof(dvbsub_color_t) );
497 static void decode_segment( decoder_t *p_dec, bs_t *s )
499 decoder_sys_t *p_sys = p_dec->p_sys;
504 /* sync_byte (already checked) */
508 i_type = bs_read( s, 8 );
511 i_page_id = bs_read( s, 16 );
514 i_size = bs_show( s, 16 );
516 if( ( i_page_id != p_sys->i_id ) &&
517 ( i_page_id != p_sys->i_ancillary_id ) )
520 msg_Dbg( p_dec, "subtitle skipped (page id: %i, %i)",
521 i_page_id, p_sys->i_id );
523 bs_skip( s, 8 * ( 2 + i_size ) );
527 if( ( p_sys->i_ancillary_id != p_sys->i_id ) &&
528 ( i_type == DVBSUB_ST_PAGE_COMPOSITION ) &&
529 ( i_page_id == p_sys->i_ancillary_id ) )
532 msg_Dbg( p_dec, "skipped invalid ancillary subtitle packet" );
534 bs_skip( s, 8 * ( 2 + i_size ) );
539 if( i_page_id == p_sys->i_id )
540 msg_Dbg( p_dec, "segment (id: %i)", i_page_id );
542 msg_Dbg( p_dec, "ancillary segment (id: %i)", i_page_id );
547 case DVBSUB_ST_PAGE_COMPOSITION:
549 msg_Dbg( p_dec, "decode_page_composition" );
551 decode_page_composition( p_dec, s );
554 case DVBSUB_ST_REGION_COMPOSITION:
556 msg_Dbg( p_dec, "decode_region_composition" );
558 decode_region_composition( p_dec, s );
561 case DVBSUB_ST_CLUT_DEFINITION:
563 msg_Dbg( p_dec, "decode_clut" );
565 decode_clut( p_dec, s );
568 case DVBSUB_ST_OBJECT_DATA:
570 msg_Dbg( p_dec, "decode_object" );
572 decode_object( p_dec, s );
575 case DVBSUB_ST_DISPLAY_DEFINITION:
577 msg_Dbg( p_dec, "decode_display_definition" );
579 decode_display_definition( p_dec, s );
582 case DVBSUB_ST_ENDOFDISPLAY:
584 msg_Dbg( p_dec, "end of display" );
586 bs_skip( s, 8 * ( 2 + i_size ) );
589 case DVBSUB_ST_STUFFING:
591 msg_Dbg( p_dec, "skip stuffing" );
593 bs_skip( s, 8 * ( 2 + i_size ) );
597 msg_Warn( p_dec, "unsupported segment type: (%04x)", i_type );
598 bs_skip( s, 8 * ( 2 + i_size ) );
603 static void decode_clut( decoder_t *p_dec, bs_t *s )
605 decoder_sys_t *p_sys = p_dec->p_sys;
606 uint16_t i_segment_length;
607 uint16_t i_processed_length;
608 dvbsub_clut_t *p_clut, *p_next;
611 i_segment_length = bs_read( s, 16 );
612 i_id = bs_read( s, 8 );
613 i_version = bs_read( s, 4 );
615 /* Check if we already have this clut */
616 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut->p_next )
618 if( p_clut->i_id == i_id ) break;
621 /* Check version number */
622 if( p_clut && ( p_clut->i_version == i_version ) )
625 bs_skip( s, 8 * i_segment_length - 12 );
632 msg_Dbg( p_dec, "new clut: %i", i_id );
634 p_clut = malloc( sizeof( dvbsub_clut_t ) );
637 p_clut->p_next = p_sys->p_cluts;
638 p_sys->p_cluts = p_clut;
641 /* Initialize to default clut */
642 p_next = p_clut->p_next;
643 *p_clut = p_sys->default_clut;
644 p_clut->p_next = p_next;
646 /* We don't have this version of the CLUT: Parse it */
647 p_clut->i_version = i_version;
649 bs_skip( s, 4 ); /* Reserved bits */
650 i_processed_length = 2;
651 while( i_processed_length < i_segment_length )
653 uint8_t y, cb, cr, t;
657 i_id = bs_read( s, 8 );
658 i_type = bs_read( s, 3 );
662 if( bs_read( s, 1 ) )
665 cr = bs_read( s, 8 );
666 cb = bs_read( s, 8 );
668 i_processed_length += 6;
672 y = bs_read( s, 6 ) << 2;
673 cr = bs_read( s, 4 ) << 4;
674 cb = bs_read( s, 4 ) << 4;
675 t = bs_read( s, 2 ) << 6;
676 i_processed_length += 4;
679 /* We are not entirely compliant here as full transparency is indicated
680 * with a luma value of zero, not a transparency value of 0xff
681 * (full transparency would actually be 0xff + 1). */
688 /* According to EN 300-743 section 7.2.3 note 1, type should
689 * not have more than 1 bit set to one, but some streams don't
690 * respect this note. */
691 if( ( i_type & 0x04 ) && ( i_id < 4 ) )
693 p_clut->c_2b[i_id].Y = y;
694 p_clut->c_2b[i_id].Cr = cr;
695 p_clut->c_2b[i_id].Cb = cb;
696 p_clut->c_2b[i_id].T = t;
698 if( ( i_type & 0x02 ) && ( i_id < 16 ) )
700 p_clut->c_4b[i_id].Y = y;
701 p_clut->c_4b[i_id].Cr = cr;
702 p_clut->c_4b[i_id].Cb = cb;
703 p_clut->c_4b[i_id].T = t;
707 p_clut->c_8b[i_id].Y = y;
708 p_clut->c_8b[i_id].Cr = cr;
709 p_clut->c_8b[i_id].Cb = cb;
710 p_clut->c_8b[i_id].T = t;
715 static void decode_page_composition( decoder_t *p_dec, bs_t *s )
717 decoder_sys_t *p_sys = p_dec->p_sys;
718 int i_version, i_state, i_segment_length, i_timeout, i;
720 /* A page is composed by 0 or more region */
721 i_segment_length = bs_read( s, 16 );
722 i_timeout = bs_read( s, 8 );
723 i_version = bs_read( s, 4 );
724 i_state = bs_read( s, 2 );
725 bs_skip( s, 2 ); /* Reserved */
727 if( i_state == DVBSUB_PCS_STATE_CHANGE )
729 /* End of an epoch, reset decoder buffer */
731 msg_Dbg( p_dec, "page composition mode change" );
735 else if( !p_sys->p_page && ( i_state != DVBSUB_PCS_STATE_ACQUISITION ) &&
736 ( i_state != DVBSUB_PCS_STATE_CHANGE ) )
738 /* Not a full PCS, we need to wait for one */
739 msg_Dbg( p_dec, "didn't receive an acquisition page yet" );
742 /* Try to start decoding even without an acquisition page */
743 bs_skip( s, 8 * (i_segment_length - 2) );
749 if( i_state == DVBSUB_PCS_STATE_ACQUISITION )
750 msg_Dbg( p_dec, "acquisition page composition" );
753 /* Check version number */
754 if( p_sys->p_page && ( p_sys->p_page->i_version == i_version ) )
756 bs_skip( s, 8 * (i_segment_length - 2) );
759 else if( p_sys->p_page )
761 if( p_sys->p_page->i_region_defs )
762 free( p_sys->p_page->p_region_defs );
763 p_sys->p_page->p_region_defs = NULL;
764 p_sys->p_page->i_region_defs = 0;
770 msg_Dbg( p_dec, "new page" );
772 /* Allocate a new page */
773 p_sys->p_page = malloc( sizeof(dvbsub_page_t) );
778 p_sys->p_page->i_version = i_version;
779 p_sys->p_page->i_timeout = i_timeout;
780 p_sys->b_page = true;
782 /* Number of regions */
783 p_sys->p_page->i_region_defs = (i_segment_length - 2) / 6;
785 if( p_sys->p_page->i_region_defs == 0 ) return;
787 p_sys->p_page->p_region_defs =
788 malloc( p_sys->p_page->i_region_defs * sizeof(dvbsub_region_t) );
789 if( p_sys->p_page->p_region_defs )
791 for( i = 0; i < p_sys->p_page->i_region_defs; i++ )
793 p_sys->p_page->p_region_defs[i].i_id = bs_read( s, 8 );
794 bs_skip( s, 8 ); /* Reserved */
795 p_sys->p_page->p_region_defs[i].i_x = bs_read( s, 16 );
796 p_sys->p_page->p_region_defs[i].i_y = bs_read( s, 16 );
799 msg_Dbg( p_dec, "page_composition, region %i (%i,%i)",
800 i, p_sys->p_page->p_region_defs[i].i_x,
801 p_sys->p_page->p_region_defs[i].i_y );
807 static void decode_region_composition( decoder_t *p_dec, bs_t *s )
809 decoder_sys_t *p_sys = p_dec->p_sys;
810 dvbsub_region_t *p_region, **pp_region = &p_sys->p_regions;
811 int i_segment_length, i_processed_length, i_id, i_version;
812 int i_width, i_height, i_level_comp, i_depth, i_clut;
813 int i_8_bg, i_4_bg, i_2_bg;
816 i_segment_length = bs_read( s, 16 );
817 i_id = bs_read( s, 8 );
818 i_version = bs_read( s, 4 );
820 /* Check if we already have this region */
821 for( p_region = p_sys->p_regions; p_region != NULL;
822 p_region = p_region->p_next )
824 pp_region = &p_region->p_next;
825 if( p_region->i_id == i_id ) break;
828 /* Check version number */
829 if( p_region && ( p_region->i_version == i_version ) )
831 bs_skip( s, 8 * (i_segment_length - 1) - 4 );
838 msg_Dbg( p_dec, "new region: %i", i_id );
840 p_region = *pp_region = malloc( sizeof(dvbsub_region_t) );
843 memset( p_region, 0, sizeof(dvbsub_region_t) );
844 p_region->p_object_defs = NULL;
845 p_region->p_pixbuf = NULL;
846 p_region->p_next = NULL;
850 /* Region attributes */
851 p_region->i_id = i_id;
852 p_region->i_version = i_version;
853 b_fill = bs_read( s, 1 );
854 bs_skip( s, 3 ); /* Reserved */
856 i_width = bs_read( s, 16 );
857 i_height = bs_read( s, 16 );
859 msg_Dbg( p_dec, " width=%d height=%d", i_width, i_height );
861 i_level_comp = bs_read( s, 3 );
862 i_depth = bs_read( s, 3 );
863 bs_skip( s, 2 ); /* Reserved */
864 i_clut = bs_read( s, 8 );
866 i_8_bg = bs_read( s, 8 );
867 i_4_bg = bs_read( s, 4 );
868 i_2_bg = bs_read( s, 2 );
869 bs_skip( s, 2 ); /* Reserved */
871 /* Free old object defs */
872 while( p_region->i_object_defs )
874 int i = p_region->i_object_defs - 1;
875 free( p_region->p_object_defs[i].psz_text );
877 free( p_region->p_object_defs );
879 p_region->i_object_defs--;
881 p_region->p_object_defs = NULL;
883 /* Extra sanity checks */
884 if( ( p_region->i_width != i_width ) ||
885 ( p_region->i_height != i_height ) )
887 if( p_region->p_pixbuf )
889 msg_Dbg( p_dec, "region size changed (%dx%d->%dx%d)",
890 p_region->i_width, p_region->i_height, i_width, i_height );
891 free( p_region->p_pixbuf );
894 p_region->p_pixbuf = malloc( i_height * i_width );
895 p_region->i_depth = 0;
898 if( p_region->i_depth &&
899 ( ( p_region->i_depth != i_depth ) ||
900 ( p_region->i_level_comp != i_level_comp ) ||
901 ( p_region->i_clut != i_clut) ) )
903 msg_Dbg( p_dec, "region parameters changed (not allowed)" );
906 /* Erase background of region */
909 int i_background = ( p_region->i_depth == 1 ) ? i_2_bg :
910 ( ( p_region->i_depth == 2 ) ? i_4_bg : i_8_bg );
911 memset( p_region->p_pixbuf, i_background, i_width * i_height );
914 p_region->i_width = i_width;
915 p_region->i_height = i_height;
916 p_region->i_level_comp = i_level_comp;
917 p_region->i_depth = i_depth;
918 p_region->i_clut = i_clut;
920 /* List of objects in the region */
921 i_processed_length = 10;
922 while( i_processed_length < i_segment_length )
924 dvbsub_objectdef_t *p_obj;
926 /* We create a new object */
927 p_region->i_object_defs++;
928 p_region->p_object_defs =
929 realloc( p_region->p_object_defs,
930 sizeof(dvbsub_objectdef_t) * p_region->i_object_defs );
932 /* We parse object properties */
933 p_obj = &p_region->p_object_defs[p_region->i_object_defs - 1];
934 p_obj->i_id = bs_read( s, 16 );
935 p_obj->i_type = bs_read( s, 2 );
936 bs_skip( s, 2 ); /* Provider */
937 p_obj->i_x = bs_read( s, 12 );
938 bs_skip( s, 4 ); /* Reserved */
939 p_obj->i_y = bs_read( s, 12 );
942 i_processed_length += 6;
944 if( ( p_obj->i_type == DVBSUB_OT_BASIC_CHAR ) ||
945 ( p_obj->i_type == DVBSUB_OT_COMPOSITE_STRING ) )
947 p_obj->i_fg_pc = bs_read( s, 8 );
948 p_obj->i_bg_pc = bs_read( s, 8 );
949 i_processed_length += 2;
954 /* ETSI 300 743 [7.2.1] */
955 static void decode_display_definition( decoder_t *p_dec, bs_t *s )
957 decoder_sys_t *p_sys = p_dec->p_sys;
958 uint16_t i_segment_length;
959 uint16_t i_processed_length = 40;
960 dvbsub_display_t *p_display;
961 dvbsub_display_t *p_old = p_sys->p_display;
964 i_segment_length = bs_read( s, 16 );
965 i_version = bs_read( s, 4 );
967 /* Check version number */
968 if( p_old && ( p_old->i_version == i_version ) )
970 /* The definition did not change */
971 bs_skip( s, 8*i_segment_length - 4 );
976 msg_Dbg( p_dec, "new display definition: %i", i_version );
978 p_display = malloc( sizeof(dvbsub_display_t) );
981 /* We don't have this version of the display definition: Parse it */
982 p_display->i_version = i_version;
983 p_display->b_windowed = bs_read( s, 1 );
984 bs_skip( s, 3 ); /* Reserved bits */
985 p_display->i_width = bs_read( s, 16 )+1;
986 p_display->i_height = bs_read( s, 16 )+1;
988 if( p_display->b_windowed )
991 msg_Dbg( p_dec, "display definition with offsets (windowed)" );
993 /* Coordinates are measured from the top left corner */
994 p_display->i_x = bs_read( s, 16 );
995 p_display->i_max_x = bs_read( s, 16 );
996 p_display->i_y = bs_read( s, 16 );
997 p_display->i_max_y = bs_read( s, 16 );
998 i_processed_length += 64;
1002 p_sys->p_display = p_display;
1005 if( i_processed_length != i_segment_length*8 )
1007 msg_Err( p_dec, "processed length %d != segment length %d",
1008 i_processed_length, i_segment_length );
1012 msg_Dbg( p_dec, "version: %d, width: %d, height: %d",
1013 p_display->i_version, p_display->i_width, p_display->i_height );
1014 if( p_display->b_windowed )
1015 msg_Dbg( p_dec, "xmin: %d, xmax: %d, ymin: %d, ymax: %d",
1016 p_display->i_x, p_display->i_max_x, p_display->i_y, p_display->i_max_y );
1020 static void dvbsub_render_pdata( decoder_t *, dvbsub_region_t *, int, int,
1022 static void dvbsub_pdata2bpp( bs_t *, uint8_t *, int, int * );
1023 static void dvbsub_pdata4bpp( bs_t *, uint8_t *, int, int * );
1024 static void dvbsub_pdata8bpp( bs_t *, uint8_t *, int, int * );
1026 static void decode_object( decoder_t *p_dec, bs_t *s )
1028 decoder_sys_t *p_sys = p_dec->p_sys;
1029 dvbsub_region_t *p_region;
1030 int i_segment_length, i_coding_method, i_version, i_id, i;
1031 bool b_non_modify_color;
1033 /* ETSI 300-743 paragraph 7.2.4
1034 * sync_byte, segment_type and page_id have already been processed.
1036 i_segment_length = bs_read( s, 16 );
1037 i_id = bs_read( s, 16 );
1038 i_version = bs_read( s, 4 );
1039 i_coding_method = bs_read( s, 2 );
1041 if( i_coding_method > 1 )
1043 msg_Dbg( p_dec, "unknown DVB subtitling coding %d is not handled!", i_coding_method );
1044 bs_skip( s, 8 * (i_segment_length - 2) - 6 );
1048 /* Check if the object needs to be rendered in at least one
1050 for( p_region = p_sys->p_regions; p_region != NULL;
1051 p_region = p_region->p_next )
1053 for( i = 0; i < p_region->i_object_defs; i++ )
1054 if( p_region->p_object_defs[i].i_id == i_id ) break;
1056 if( i != p_region->i_object_defs ) break;
1060 bs_skip( s, 8 * (i_segment_length - 2) - 6 );
1065 msg_Dbg( p_dec, "new object: %i", i_id );
1068 b_non_modify_color = bs_read( s, 1 );
1069 bs_skip( s, 1 ); /* Reserved */
1071 if( i_coding_method == 0x00 )
1073 int i_topfield, i_bottomfield;
1074 uint8_t *p_topfield, *p_bottomfield;
1076 i_topfield = bs_read( s, 16 );
1077 i_bottomfield = bs_read( s, 16 );
1078 p_topfield = s->p_start + bs_pos( s ) / 8;
1079 p_bottomfield = p_topfield + i_topfield;
1081 bs_skip( s, 8 * (i_segment_length - 7) );
1084 if( ( i_segment_length < ( i_topfield + i_bottomfield + 7 ) ) ||
1085 ( ( p_topfield + i_topfield + i_bottomfield ) > s->p_end ) )
1087 msg_Dbg( p_dec, "corrupted object data" );
1091 for( p_region = p_sys->p_regions; p_region != NULL;
1092 p_region = p_region->p_next )
1094 for( i = 0; i < p_region->i_object_defs; i++ )
1096 if( p_region->p_object_defs[i].i_id != i_id ) continue;
1098 dvbsub_render_pdata( p_dec, p_region,
1099 p_region->p_object_defs[i].i_x,
1100 p_region->p_object_defs[i].i_y,
1101 p_topfield, i_topfield );
1105 dvbsub_render_pdata( p_dec, p_region,
1106 p_region->p_object_defs[i].i_x,
1107 p_region->p_object_defs[i].i_y + 1,
1108 p_bottomfield, i_bottomfield );
1112 /* Duplicate the top field */
1113 dvbsub_render_pdata( p_dec, p_region,
1114 p_region->p_object_defs[i].i_x,
1115 p_region->p_object_defs[i].i_y + 1,
1116 p_topfield, i_topfield );
1123 /* DVB subtitling as characters */
1124 int i_number_of_codes = bs_read( s, 8 );
1125 uint8_t* p_start = s->p_start + bs_pos( s ) / 8;
1128 if( ( i_segment_length < ( i_number_of_codes*2 + 4 ) ) ||
1129 ( ( p_start + i_number_of_codes*2 ) > s->p_end ) )
1131 msg_Dbg( p_dec, "corrupted object data" );
1135 for( p_region = p_sys->p_regions; p_region != NULL;
1136 p_region = p_region->p_next )
1138 for( i = 0; i < p_region->i_object_defs; i++ )
1142 if( p_region->p_object_defs[i].i_id != i_id ) continue;
1144 p_region->p_object_defs[i].psz_text =
1145 realloc( p_region->p_object_defs[i].psz_text,
1146 i_number_of_codes + 1 );
1148 /* FIXME 16bits -> char ??? See Preamble */
1149 for( j = 0; j < i_number_of_codes; j++ )
1151 p_region->p_object_defs[i].psz_text[j] = (char)(bs_read( s, 16 ) & 0xFF);
1153 /* Null terminate the string */
1154 p_region->p_object_defs[i].psz_text[j] = 0;
1160 msg_Dbg( p_dec, "end object: %i", i_id );
1164 static void dvbsub_render_pdata( decoder_t *p_dec, dvbsub_region_t *p_region,
1166 uint8_t *p_field, int i_field )
1173 if( !p_region->p_pixbuf )
1175 msg_Err( p_dec, "region %i has no pixel buffer!", p_region->i_id );
1178 if( i_y < 0 || i_x < 0 || i_y >= p_region->i_height ||
1179 i_x >= p_region->i_width )
1181 msg_Dbg( p_dec, "invalid offset (%i,%i)", i_x, i_y );
1185 p_pixbuf = p_region->p_pixbuf + i_y * p_region->i_width;
1186 bs_init( &bs, p_field, i_field );
1188 while( !bs_eof( &bs ) )
1191 if( i_y >= p_region->i_height ) return;
1193 switch( bs_read( &bs, 8 ) )
1196 dvbsub_pdata2bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1201 dvbsub_pdata4bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1206 dvbsub_pdata8bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1213 /* We don't use map tables */
1216 case 0xf0: /* End of line code */
1217 p_pixbuf += 2*p_region->i_width;
1218 i_offset = 0; i_y += 2;
1224 static void dvbsub_pdata2bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1226 bool b_stop = false;
1228 while( !b_stop && !bs_eof( s ) )
1230 int i_count = 0, i_color = 0;
1232 i_color = bs_read( s, 2 );
1233 if( i_color != 0x00 )
1239 if( bs_read( s, 1 ) == 0x01 ) // Switch1
1241 i_count = 3 + bs_read( s, 3 );
1242 i_color = bs_read( s, 2 );
1246 if( bs_read( s, 1 ) == 0x00 ) //Switch2
1248 switch( bs_read( s, 2 ) ) //Switch3
1257 i_count = 12 + bs_read( s, 4 );
1258 i_color = bs_read( s, 2 );
1261 i_count = 29 + bs_read( s, 8 );
1262 i_color = bs_read( s, 2 );
1270 /* 1 pixel color 0 */
1276 if( !i_count ) continue;
1279 if( ( i_count + *pi_off ) > i_width ) break;
1281 if( i_count == 1 ) p[*pi_off] = i_color;
1282 else memset( ( p + *pi_off ), i_color, i_count );
1284 (*pi_off) += i_count;
1290 static void dvbsub_pdata4bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1292 bool b_stop = false;
1294 while( !b_stop && !bs_eof( s ) )
1296 int i_count = 0, i_color = 0;
1298 i_color = bs_read( s, 4 );
1299 if( i_color != 0x00 )
1306 if( bs_read( s, 1 ) == 0x00 ) // Switch1
1308 if( bs_show( s, 3 ) != 0x00 )
1310 i_count = 2 + bs_read( s, 3 );
1320 if( bs_read( s, 1 ) == 0x00) //Switch2
1322 i_count = 4 + bs_read( s, 2 );
1323 i_color = bs_read( s, 4 );
1327 switch ( bs_read( s, 2 ) ) //Switch3
1336 i_count = 9 + bs_read( s, 4 );
1337 i_color = bs_read( s, 4 );
1340 i_count= 25 + bs_read( s, 8 );
1341 i_color = bs_read( s, 4 );
1348 if( !i_count ) continue;
1351 if( ( i_count + *pi_off ) > i_width ) break;
1353 if( i_count == 1 ) p[*pi_off] = i_color;
1354 else memset( ( p + *pi_off ), i_color, i_count );
1356 (*pi_off) += i_count;
1362 static void dvbsub_pdata8bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1364 bool b_stop = false;
1366 while( !b_stop && !bs_eof( s ) )
1368 int i_count = 0, i_color = 0;
1370 i_color = bs_read( s, 8 );
1371 if( i_color != 0x00 )
1378 if( bs_read( s, 1 ) == 0x00 ) // Switch1
1380 if( bs_show( s, 7 ) != 0x00 )
1382 i_count = bs_read( s, 7 );
1392 i_count = bs_read( s, 7 );
1393 i_color = bs_read( s, 8 );
1397 if( !i_count ) continue;
1400 if( ( i_count + *pi_off ) > i_width ) break;
1402 if( i_count == 1 ) p[*pi_off] = i_color;
1403 else memset( ( p + *pi_off ), i_color, i_count );
1405 (*pi_off) += i_count;
1411 static void free_all( decoder_t *p_dec )
1413 decoder_sys_t *p_sys = p_dec->p_sys;
1414 dvbsub_region_t *p_reg, *p_reg_next;
1415 dvbsub_clut_t *p_clut, *p_clut_next;
1417 free( p_sys->p_display );
1419 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut_next )
1421 p_clut_next = p_clut->p_next;
1424 p_sys->p_cluts = NULL;
1426 for( p_reg = p_sys->p_regions; p_reg != NULL; p_reg = p_reg_next )
1430 p_reg_next = p_reg->p_next;
1431 for( i = 0; i < p_reg->i_object_defs; i++ )
1432 free( p_reg->p_object_defs[i].psz_text );
1433 if( p_reg->i_object_defs ) free( p_reg->p_object_defs );
1434 free( p_reg->p_pixbuf );
1437 p_sys->p_regions = NULL;
1441 if( p_sys->p_page->i_region_defs )
1442 free( p_sys->p_page->p_region_defs );
1443 free( p_sys->p_page );
1445 p_sys->p_page = NULL;
1448 static subpicture_t *render( decoder_t *p_dec )
1450 decoder_sys_t *p_sys = p_dec->p_sys;
1451 subpicture_t *p_spu;
1452 subpicture_region_t **pp_spu_region;
1453 int i, j, i_timeout = 0;
1455 /* Allocate the subpicture internal data. */
1456 p_spu = p_dec->pf_spu_buffer_new( p_dec );
1457 if( !p_spu ) return NULL;
1459 pp_spu_region = &p_spu->p_region;
1461 /* Loop on region definitions */
1464 msg_Dbg( p_dec, "rendering %i regions", p_sys->p_page->i_region_defs );
1467 for( i = 0; p_sys->p_page && ( i < p_sys->p_page->i_region_defs ); i++ )
1469 dvbsub_region_t *p_region;
1470 dvbsub_regiondef_t *p_regiondef;
1471 dvbsub_clut_t *p_clut;
1472 dvbsub_color_t *p_color;
1473 subpicture_region_t *p_spu_region;
1474 uint8_t *p_src, *p_dst;
1478 i_timeout = p_sys->p_page->i_timeout;
1480 p_regiondef = &p_sys->p_page->p_region_defs[i];
1483 msg_Dbg( p_dec, "rendering region %i (%i,%i)", i,
1484 p_regiondef->i_x, p_regiondef->i_y );
1487 /* Find associated region */
1488 for( p_region = p_sys->p_regions; p_region != NULL;
1489 p_region = p_region->p_next )
1491 if( p_regiondef->i_id == p_region->i_id ) break;
1496 msg_Dbg( p_dec, "region %i not found", p_regiondef->i_id );
1500 /* Find associated CLUT */
1501 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut->p_next )
1503 if( p_region->i_clut == p_clut->i_id ) break;
1507 msg_Dbg( p_dec, "clut %i not found", p_region->i_clut );
1511 /* FIXME: don't create a subpicture region with VLC_FOURCC YUVP
1512 * when it actually is a TEXT region */
1514 /* Create new SPU region */
1515 memset( &fmt, 0, sizeof(video_format_t) );
1516 fmt.i_chroma = VLC_FOURCC('Y','U','V','P');
1517 fmt.i_aspect = 0; /* 0 means use aspect ratio of background video */
1518 fmt.i_width = fmt.i_visible_width = p_region->i_width;
1519 fmt.i_height = fmt.i_visible_height = p_region->i_height;
1520 fmt.i_x_offset = fmt.i_y_offset = 0;
1521 p_spu_region = p_spu->pf_create_region( VLC_OBJECT(p_dec), &fmt );
1524 msg_Err( p_dec, "cannot allocate SPU region" );
1527 p_spu_region->i_x = p_regiondef->i_x;
1528 p_spu_region->i_y = p_regiondef->i_y;
1529 p_spu_region->i_align = p_sys->i_spu_position;
1530 *pp_spu_region = p_spu_region;
1531 pp_spu_region = &p_spu_region->p_next;
1534 fmt.p_palette->i_entries = ( p_region->i_depth == 1 ) ? 4 :
1535 ( ( p_region->i_depth == 2 ) ? 16 : 256 );
1536 p_color = ( p_region->i_depth == 1 ) ? p_clut->c_2b :
1537 ( ( p_region->i_depth == 2 ) ? p_clut->c_4b : p_clut->c_8b );
1538 for( j = 0; j < fmt.p_palette->i_entries; j++ )
1540 fmt.p_palette->palette[j][0] = p_color[j].Y;
1541 fmt.p_palette->palette[j][1] = p_color[j].Cb; /* U == Cb */
1542 fmt.p_palette->palette[j][2] = p_color[j].Cr; /* V == Cr */
1543 fmt.p_palette->palette[j][3] = 0xff - p_color[j].T;
1546 p_src = p_region->p_pixbuf;
1547 p_dst = p_spu_region->picture.Y_PIXELS;
1548 i_pitch = p_spu_region->picture.Y_PITCH;
1550 /* Copy pixel buffer */
1551 for( j = 0; j < p_region->i_height; j++ )
1553 memcpy( p_dst, p_src, p_region->i_width );
1554 p_src += p_region->i_width;
1558 /* Check subtitles encoded as strings of characters
1559 * (since there are not rendered in the pixbuffer) */
1560 for( j = 0; j < p_region->i_object_defs; j++ )
1562 dvbsub_objectdef_t *p_object_def = &p_region->p_object_defs[j];
1564 if( ( p_object_def->i_type != 1 ) || !p_object_def->psz_text )
1567 /* Create new SPU region */
1568 memset( &fmt, 0, sizeof(video_format_t) );
1569 fmt.i_chroma = VLC_FOURCC('T','E','X','T');
1570 fmt.i_aspect = VOUT_ASPECT_FACTOR;
1571 fmt.i_width = fmt.i_visible_width = p_region->i_width;
1572 fmt.i_height = fmt.i_visible_height = p_region->i_height;
1573 fmt.i_x_offset = fmt.i_y_offset = 0;
1574 p_spu_region = p_spu->pf_create_region( VLC_OBJECT(p_dec), &fmt );
1577 msg_Err( p_dec, "cannot allocate SPU region" );
1581 p_spu_region->psz_text = strdup( p_object_def->psz_text );
1582 p_spu_region->i_x = p_regiondef->i_x + p_object_def->i_x;
1583 p_spu_region->i_y = p_regiondef->i_y + p_object_def->i_y;
1584 p_spu_region->i_align = p_sys->i_spu_position;
1585 *pp_spu_region = p_spu_region;
1586 pp_spu_region = &p_spu_region->p_next;
1590 /* Set the pf_render callback */
1591 p_spu->i_start = (mtime_t) p_sys->i_pts;
1592 //p_spu->i_stop = (mtime_t) 0;
1593 p_spu->b_ephemer = true;
1594 p_spu->b_pausable = true;
1595 //p_spu->b_fade = true;
1596 //p_spu->i_stop = p_spu->i_start + (mtime_t) (i_timeout * 1000000);
1598 /* Correct positioning of SPU */
1599 p_spu->b_absolute = p_sys->b_absolute;
1600 p_spu->i_flags = p_sys->i_spu_position;
1601 p_spu->i_x = p_sys->i_spu_x;
1602 p_spu->i_y = p_sys->i_spu_y;
1603 p_spu->i_original_picture_width = 720;
1604 p_spu->i_original_picture_height = 576;
1606 if( p_sys->p_display )
1608 p_spu->i_original_picture_width = p_sys->p_display->i_width;
1609 p_spu->i_original_picture_height = p_sys->p_display->i_height;
1611 if( p_sys->p_display->b_windowed )
1613 /* TODO: check that this actually works */
1614 p_spu->i_original_picture_width = p_sys->p_display->i_max_x - p_sys->p_display->i_x;
1615 p_spu->i_original_picture_height = p_sys->p_display->i_max_y - p_sys->p_display->i_y;
1616 p_spu->i_x += p_sys->p_display->i_x;
1617 p_spu->i_y += p_sys->p_display->i_y;
1624 /*****************************************************************************
1625 * encoder_sys_t : encoder descriptor
1626 *****************************************************************************/
1627 typedef struct encoder_region_t
1634 struct encoder_sys_t
1636 unsigned int i_page_ver;
1637 unsigned int i_region_ver;
1638 unsigned int i_clut_ver;
1641 encoder_region_t *p_regions;
1645 /* subpicture positioning */
1650 static void encode_page_composition( encoder_t *, bs_t *, subpicture_t * );
1651 static void encode_clut( encoder_t *, bs_t *, subpicture_t * );
1652 static void encode_region_composition( encoder_t *, bs_t *, subpicture_t * );
1653 static void encode_object( encoder_t *, bs_t *, subpicture_t * );
1655 /*****************************************************************************
1656 * OpenEncoder: probe the encoder and return score
1657 *****************************************************************************/
1658 static int OpenEncoder( vlc_object_t *p_this )
1660 encoder_t *p_enc = (encoder_t *)p_this;
1661 encoder_sys_t *p_sys;
1664 if( ( p_enc->fmt_out.i_codec != VLC_FOURCC('d','v','b','s') ) &&
1667 return VLC_EGENERIC;
1670 /* Allocate the memory needed to store the decoder's structure */
1671 if( ( p_sys = (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
1673 msg_Err( p_enc, "out of memory" );
1676 p_enc->p_sys = p_sys;
1678 p_enc->pf_encode_sub = Encode;
1679 p_enc->fmt_out.i_codec = VLC_FOURCC('d','v','b','s');
1680 p_enc->fmt_out.subs.dvb.i_id = 1 << 16 | 1;
1682 config_ChainParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
1684 p_sys->i_page_ver = 0;
1685 p_sys->i_region_ver = 0;
1686 p_sys->i_clut_ver = 0;
1687 p_sys->i_regions = 0;
1688 p_sys->p_regions = 0;
1690 var_Create( p_this, ENC_CFG_PREFIX "x", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
1691 var_Get( p_this, ENC_CFG_PREFIX "x", &val );
1692 p_sys->i_offset_x = val.i_int;
1693 var_Create( p_this, ENC_CFG_PREFIX "y", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
1694 var_Get( p_this, ENC_CFG_PREFIX "y", &val );
1695 p_sys->i_offset_y = val.i_int;
1700 /* FIXME: this routine is a hack to convert VLC_FOURCC('Y','U','V','A')
1701 * into VLC_FOURCC('Y','U','V','P')
1703 static subpicture_t *YuvaYuvp( subpicture_t *p_subpic )
1705 subpicture_region_t *p_region = NULL;
1707 if( !p_subpic ) return NULL;
1709 for( p_region = p_subpic->p_region; p_region; p_region = p_region->p_next )
1711 video_format_t *p_fmt = &p_region->fmt;
1712 int i = 0, j = 0, n = 0, p = 0;
1713 int i_max_entries = 256;
1715 #ifdef RANDOM_DITHERING
1716 int i_seed = 0xdeadbeef; /* random seed */
1720 int i_pixels = p_region->picture.p[0].i_visible_lines
1721 * p_region->picture.p[0].i_pitch;
1722 int i_iterator = p_region->picture.p[0].i_visible_lines * 3 / 4
1723 * p_region->picture.p[0].i_pitch
1724 + p_region->picture.p[0].i_pitch * 1 / 3;
1725 int i_tolerance = 0;
1728 msg_Dbg( p_enc, "YuvaYuvp: i_pixels=%d, i_iterator=%d", i_pixels, i_iterator );
1730 p_fmt->i_chroma = VLC_FOURCC('Y','U','V','P');
1731 p_fmt->p_palette = (video_palette_t *) malloc( sizeof( video_palette_t ) );
1732 if( !p_fmt->p_palette ) break;
1733 p_fmt->p_palette->i_entries = 0;
1735 /* Find best iterator using Euclide’s algorithm */
1736 for( ; i_iterator > 1 ; i_iterator-- )
1755 /* Count colors, build best palette */
1756 for( i_tolerance = 0; i_tolerance < 128; i_tolerance++ )
1758 bool b_success = true;
1759 p_fmt->p_palette->i_entries = 0;
1761 for( i = 0; i < i_pixels ; )
1764 y = p_region->picture.p[0].p_pixels[i];
1765 u = p_region->picture.p[1].p_pixels[i];
1766 v = p_region->picture.p[2].p_pixels[i];
1767 a = p_region->picture.p[3].p_pixels[i];
1768 for( j = 0; j < p_fmt->p_palette->i_entries; j++ )
1770 if( abs((int)p_fmt->p_palette->palette[j][0] - (int)y) <= i_tolerance &&
1771 abs((int)p_fmt->p_palette->palette[j][1] - (int)u) <= i_tolerance &&
1772 abs((int)p_fmt->p_palette->palette[j][2] - (int)v) <= i_tolerance &&
1773 abs((int)p_fmt->p_palette->palette[j][3] - (int)a) <= i_tolerance / 2 )
1778 if( j == p_fmt->p_palette->i_entries )
1780 p_fmt->p_palette->palette[j][0] = y;
1781 p_fmt->p_palette->palette[j][1] = u;
1782 p_fmt->p_palette->palette[j][2] = v;
1783 p_fmt->p_palette->palette[j][3] = a;
1784 p_fmt->p_palette->i_entries++;
1786 if( p_fmt->p_palette->i_entries >= i_max_entries )
1805 msg_Dbg( p_enc, "best palette has %d colors", p_fmt->p_palette->i_entries );
1808 #ifndef RANDOM_DITHERING
1809 pi_delta = malloc( ( p_region->picture.p[0].i_pitch + 1 )
1810 * sizeof(int) * 4 );
1811 for( i = 0; i < (p_region->picture.p[0].i_pitch + 1) * 4 ; i++ )
1817 /* Fill image with our new colours */
1818 for( p = 0; p < p_region->picture.p[0].i_visible_lines ; p++ )
1820 int i_ydelta = 0, i_udelta = 0, i_vdelta = 0, i_adelta = 0;
1822 for( n = 0; n < p_region->picture.p[0].i_pitch ; n++ )
1824 int i_offset = p * p_region->picture.p[0].i_pitch + n;
1826 int i_mindist, i_best;
1828 y = (int)p_region->picture.p[0].p_pixels[i_offset];
1829 u = (int)p_region->picture.p[1].p_pixels[i_offset];
1830 v = (int)p_region->picture.p[2].p_pixels[i_offset];
1831 a = (int)p_region->picture.p[3].p_pixels[i_offset];
1833 /* Add dithering compensation */
1834 #ifdef RANDOM_DITHERING
1835 y += ((i_seed & 0xff) - 0x80) * i_tolerance / 0x80;
1836 u += (((i_seed >> 8) & 0xff) - 0x80) * i_tolerance / 0x80;
1837 v += (((i_seed >> 16) & 0xff) - 0x80) * i_tolerance / 0x80;
1838 a += (((i_seed >> 24) & 0xff) - 0x80) * i_tolerance / 0x80;
1840 y += i_ydelta + pi_delta[ n * 4 ];
1841 u += i_udelta + pi_delta[ n * 4 + 1 ];
1842 v += i_vdelta + pi_delta[ n * 4 + 2 ];
1843 a += i_adelta + pi_delta[ n * 4 + 3 ];
1846 /* Find best colour in palette */
1847 for( i_mindist = 99999999, i_best = 0, j = 0; j < p_fmt->p_palette->i_entries; j++ )
1851 i_dist += abs((int)p_fmt->p_palette->palette[j][0] - y);
1852 i_dist += abs((int)p_fmt->p_palette->palette[j][1] - u);
1853 i_dist += abs((int)p_fmt->p_palette->palette[j][2] - v);
1854 i_dist += 2 * abs((int)p_fmt->p_palette->palette[j][3] - a);
1856 if( i_dist < i_mindist )
1863 /* Set pixel to best color */
1864 p_region->picture.p[0].p_pixels[i_offset] = i_best;
1866 /* Update dithering state */
1867 #ifdef RANDOM_DITHERING
1868 i_seed = (i_seed * 0x1283837) ^ 0x789479 ^ (i_seed >> 13);
1870 i_ydelta = y - (int)p_fmt->p_palette->palette[i_best][0];
1871 i_udelta = u - (int)p_fmt->p_palette->palette[i_best][1];
1872 i_vdelta = v - (int)p_fmt->p_palette->palette[i_best][2];
1873 i_adelta = a - (int)p_fmt->p_palette->palette[i_best][3];
1874 pi_delta[ n * 4 ] = i_ydelta * 3 / 8;
1875 pi_delta[ n * 4 + 1 ] = i_udelta * 3 / 8;
1876 pi_delta[ n * 4 + 2 ] = i_vdelta * 3 / 8;
1877 pi_delta[ n * 4 + 3 ] = i_adelta * 3 / 8;
1878 i_ydelta = i_ydelta * 5 / 8;
1879 i_udelta = i_udelta * 5 / 8;
1880 i_vdelta = i_vdelta * 5 / 8;
1881 i_adelta = i_adelta * 5 / 8;
1885 #ifndef RANDOM_DITHERING
1890 for( i = p_fmt->p_palette->i_entries; i < i_max_entries; i++ )
1892 p_fmt->p_palette->palette[i][0] = 0;
1893 p_fmt->p_palette->palette[i][1] = 0;
1894 p_fmt->p_palette->palette[i][2] = 0;
1895 p_fmt->p_palette->palette[i][3] = 0;
1897 p_fmt->p_palette->i_entries = i_max_entries;
1899 msg_Dbg( p_enc, "best palette has %d colors", p_fmt->p_palette->i_entries );
1905 /****************************************************************************
1906 * Encode: the whole thing
1907 ****************************************************************************/
1908 static block_t *Encode( encoder_t *p_enc, subpicture_t *p_subpic )
1910 subpicture_t *p_temp = NULL;
1911 subpicture_region_t *p_region = NULL;
1912 bs_t bits, *s = &bits;
1915 if( !p_subpic || !p_subpic->p_region ) return NULL;
1917 /* FIXME: this is a hack to convert VLC_FOURCC('Y','U','V','A') into
1918 * VLC_FOURCC('Y','U','V','P')
1920 p_region = p_subpic->p_region;
1921 if( p_region->fmt.i_chroma == VLC_FOURCC('Y','U','V','A') )
1923 p_temp = YuvaYuvp( p_subpic );
1926 msg_Err( p_enc, "no picture in subpicture" );
1929 p_region = p_subpic->p_region;
1933 if( !p_region ) return NULL;
1935 if( ( p_region->fmt.i_chroma != VLC_FOURCC('T','E','X','T') ) &&
1936 ( p_region->fmt.i_chroma != VLC_FOURCC('Y','U','V','P') ) )
1939 memset( &psz_fourcc, 0, sizeof( psz_fourcc ) );
1940 vlc_fourcc_to_char( p_region->fmt.i_chroma, &psz_fourcc );
1941 msg_Err( p_enc, "chroma %4s not supported", psz_fourcc );
1945 if( p_region->fmt.p_palette )
1947 switch( p_region->fmt.p_palette->i_entries )
1955 msg_Err( p_enc, "subpicture palette (%d) not handled",
1956 p_region->fmt.p_palette->i_entries );
1963 msg_Dbg( p_enc, "encoding subpicture" );
1965 p_block = block_New( p_enc, 64000 );
1966 bs_init( s, p_block->p_buffer, p_block->i_buffer );
1968 bs_write( s, 8, 0x20 ); /* Data identifier */
1969 bs_write( s, 8, 0x0 ); /* Subtitle stream id */
1971 encode_page_composition( p_enc, s, p_subpic );
1972 encode_region_composition( p_enc, s, p_subpic );
1973 encode_clut( p_enc, s, p_subpic );
1974 encode_object( p_enc, s, p_subpic );
1976 /* End of display */
1977 bs_write( s, 8, 0x0f ); /* Sync byte */
1978 bs_write( s, 8, DVBSUB_ST_ENDOFDISPLAY ); /* Segment type */
1979 bs_write( s, 16, 1 ); /* Page id */
1980 bs_write( s, 16, 0 ); /* Segment length */
1982 bs_write( s, 8, 0xff );/* End marker */
1983 p_block->i_buffer = bs_pos( s ) / 8;
1984 p_block->i_pts = p_block->i_dts = p_subpic->i_start;
1985 if( !p_subpic->b_ephemer && ( p_subpic->i_stop > p_subpic->i_start ) )
1987 block_t *p_block_stop;
1989 p_block->i_length = p_subpic->i_stop - p_subpic->i_start;
1991 /* Send another (empty) subtitle to signal the end of display */
1992 p_block_stop = block_New( p_enc, 64000 );
1993 bs_init( s, p_block_stop->p_buffer, p_block_stop->i_buffer );
1994 bs_write( s, 8, 0x20 ); /* Data identifier */
1995 bs_write( s, 8, 0x0 ); /* Subtitle stream id */
1996 encode_page_composition( p_enc, s, 0 );
1997 bs_write( s, 8, 0x0f ); /* Sync byte */
1998 bs_write( s, 8, DVBSUB_ST_ENDOFDISPLAY ); /* Segment type */
1999 bs_write( s, 16, 1 ); /* Page id */
2000 bs_write( s, 16, 0 ); /* Segment length */
2001 bs_write( s, 8, 0xff );/* End marker */
2002 p_block_stop->i_buffer = bs_pos( s ) / 8;
2003 p_block_stop->i_pts = p_block_stop->i_dts = p_subpic->i_stop;
2004 block_ChainAppend( &p_block, p_block_stop );
2005 p_block_stop->i_length = 100000; /* p_subpic->i_stop - p_subpic->i_start; */
2008 msg_Dbg( p_enc, "subpicture encoded properly" );
2013 /*****************************************************************************
2014 * CloseEncoder: encoder destruction
2015 *****************************************************************************/
2016 static void CloseEncoder( vlc_object_t *p_this )
2018 encoder_t *p_enc = (encoder_t *)p_this;
2019 encoder_sys_t *p_sys = p_enc->p_sys;
2021 var_Destroy( p_this , ENC_CFG_PREFIX "x" );
2022 var_Destroy( p_this , ENC_CFG_PREFIX "y" );
2023 var_Destroy( p_this , ENC_CFG_PREFIX "timeout" );
2025 if( p_sys->i_regions ) free( p_sys->p_regions );
2029 static void encode_page_composition( encoder_t *p_enc, bs_t *s,
2030 subpicture_t *p_subpic )
2032 encoder_sys_t *p_sys = p_enc->p_sys;
2033 subpicture_region_t *p_region;
2034 bool b_mode_change = false;
2035 int i_regions, i_timeout;
2037 bs_write( s, 8, 0x0f ); /* Sync byte */
2038 bs_write( s, 8, DVBSUB_ST_PAGE_COMPOSITION ); /* Segment type */
2039 bs_write( s, 16, 1 ); /* Page id */
2041 for( i_regions = 0, p_region = p_subpic ? p_subpic->p_region : 0;
2042 p_region; p_region = p_region->p_next, i_regions++ )
2044 if( i_regions >= p_sys->i_regions )
2046 encoder_region_t region;
2047 region.i_width = region.i_height = 0;
2049 realloc( p_sys->p_regions, sizeof(encoder_region_t) *
2050 (p_sys->i_regions + 1) );
2051 p_sys->p_regions[p_sys->i_regions++] = region;
2054 if( ( p_sys->p_regions[i_regions].i_width <
2055 (int)p_region->fmt.i_visible_width ) ||
2056 ( p_sys->p_regions[i_regions].i_width >
2057 (int)p_region->fmt.i_visible_width ) )
2059 b_mode_change = true;
2060 msg_Dbg( p_enc, "region %i width change: %i -> %i",
2061 i_regions, p_sys->p_regions[i_regions].i_width,
2062 p_region->fmt.i_visible_width );
2063 p_sys->p_regions[i_regions].i_width =
2064 p_region->fmt.i_visible_width;
2066 if( p_sys->p_regions[i_regions].i_height <
2067 (int)p_region->fmt.i_visible_height )
2069 b_mode_change = true;
2070 msg_Dbg( p_enc, "region %i height change: %i -> %i",
2071 i_regions, p_sys->p_regions[i_regions].i_height,
2072 p_region->fmt.i_visible_height );
2073 p_sys->p_regions[i_regions].i_height =
2074 p_region->fmt.i_visible_height;
2078 bs_write( s, 16, i_regions * 6 + 2 ); /* Segment length */
2081 if( p_subpic && !p_subpic->b_ephemer &&
2082 ( p_subpic->i_stop > p_subpic->i_start ) )
2084 i_timeout = (p_subpic->i_stop - p_subpic->i_start) / 1000000;
2087 bs_write( s, 8, i_timeout ); /* Timeout */
2088 bs_write( s, 4, p_sys->i_page_ver++ );
2089 bs_write( s, 2, b_mode_change ?
2090 DVBSUB_PCS_STATE_CHANGE : DVBSUB_PCS_STATE_ACQUISITION );
2091 bs_write( s, 2, 0 ); /* Reserved */
2093 for( i_regions = 0, p_region = p_subpic ? p_subpic->p_region : 0;
2094 p_region; p_region = p_region->p_next, i_regions++ )
2096 bs_write( s, 8, i_regions );
2097 bs_write( s, 8, 0 ); /* Reserved */
2098 if( (p_sys->i_offset_x > 0) && (p_sys->i_offset_y > 0) )
2100 bs_write( s, 16, p_sys->i_offset_x ); /* override x position */
2101 bs_write( s, 16, p_sys->i_offset_y ); /* override y position */
2105 bs_write( s, 16, p_subpic->i_x + p_region->i_x );
2106 bs_write( s, 16, p_subpic->i_y + p_region->i_y );
2111 static void encode_clut( encoder_t *p_enc, bs_t *s, subpicture_t *p_subpic )
2113 encoder_sys_t *p_sys = p_enc->p_sys;
2114 subpicture_region_t *p_region = p_subpic->p_region;
2115 video_palette_t *p_pal, pal;
2119 if( !p_region ) return;
2121 if( p_region->fmt.i_chroma == VLC_FOURCC('Y','U','V','P') )
2123 p_pal = p_region->fmt.p_palette;
2128 for( i = 0; i < 4; i++ )
2130 pal.palette[i][0] = 0;
2131 pal.palette[i][1] = 0;
2132 pal.palette[i][2] = 0;
2133 pal.palette[i][3] = 0;
2138 bs_write( s, 8, 0x0f ); /* Sync byte */
2139 bs_write( s, 8, DVBSUB_ST_CLUT_DEFINITION ); /* Segment type */
2140 bs_write( s, 16, 1 ); /* Page id */
2142 bs_write( s, 16, p_pal->i_entries * 6 + 2 ); /* Segment length */
2143 bs_write( s, 8, 1 ); /* Clut id */
2144 bs_write( s, 4, p_sys->i_clut_ver++ );
2145 bs_write( s, 4, 0 ); /* Reserved */
2147 for( i = 0; i < p_pal->i_entries; i++ )
2149 bs_write( s, 8, i ); /* Clut entry id */
2150 bs_write( s, 1, p_pal->i_entries == 4 ); /* 2bit/entry flag */
2151 bs_write( s, 1, p_pal->i_entries == 16 ); /* 4bit/entry flag */
2152 bs_write( s, 1, p_pal->i_entries == 256 ); /* 8bit/entry flag */
2153 bs_write( s, 4, 0 ); /* Reserved */
2154 bs_write( s, 1, 1 ); /* Full range flag */
2155 bs_write( s, 8, p_pal->palette[i][3] ? /* Y value */
2156 (p_pal->palette[i][0] ? p_pal->palette[i][0] : 16) : 0 );
2157 bs_write( s, 8, p_pal->palette[i][1] ); /* Cr value */
2158 bs_write( s, 8, p_pal->palette[i][2] ); /* Cb value */
2159 bs_write( s, 8, 0xff - p_pal->palette[i][3] ); /* T value */
2163 static void encode_region_composition( encoder_t *p_enc, bs_t *s,
2164 subpicture_t *p_subpic )
2166 encoder_sys_t *p_sys = p_enc->p_sys;
2167 subpicture_region_t *p_region;
2170 for( i_region = 0, p_region = p_subpic->p_region; p_region;
2171 p_region = p_region->p_next, i_region++ )
2173 int i_entries = 4, i_depth = 0x1, i_bg = 0;
2175 ( p_region->fmt.i_chroma == VLC_FOURCC('T','E','X','T') );
2179 video_palette_t *p_pal = p_region->fmt.p_palette;
2183 msg_Err( p_enc, "subpicture has no palette - ignoring it" );
2187 i_entries = p_pal->i_entries;
2188 i_depth = i_entries == 4 ? 0x1 : i_entries == 16 ? 0x2 : 0x3;
2190 for( i_bg = 0; i_bg < p_pal->i_entries; i_bg++ )
2192 if( !p_pal->palette[i_bg][3] ) break;
2196 bs_write( s, 8, 0x0f ); /* Sync byte */
2197 bs_write( s, 8, DVBSUB_ST_REGION_COMPOSITION ); /* Segment type */
2198 bs_write( s, 16, 1 ); /* Page id */
2200 bs_write( s, 16, 10 + 6 + (b_text ? 2 : 0) ); /* Segment length */
2201 bs_write( s, 8, i_region );
2202 bs_write( s, 4, p_sys->i_region_ver++ );
2204 /* Region attributes */
2205 bs_write( s, 1, i_bg < i_entries ); /* Fill */
2206 bs_write( s, 3, 0 ); /* Reserved */
2207 bs_write( s, 16, p_sys->p_regions[i_region].i_width );
2208 bs_write( s, 16, p_sys->p_regions[i_region].i_height );
2209 bs_write( s, 3, i_depth ); /* Region level of compatibility */
2210 bs_write( s, 3, i_depth ); /* Region depth */
2211 bs_write( s, 2, 0 ); /* Reserved */
2212 bs_write( s, 8, 1 ); /* Clut id */
2213 bs_write( s, 8, i_bg ); /* region 8bit pixel code */
2214 bs_write( s, 4, i_bg ); /* region 4bit pixel code */
2215 bs_write( s, 2, i_bg ); /* region 2bit pixel code */
2216 bs_write( s, 2, 0 ); /* Reserved */
2218 /* In our implementation we only have 1 object per region */
2219 bs_write( s, 16, i_region );
2220 bs_write( s, 2, b_text ? DVBSUB_OT_BASIC_CHAR:DVBSUB_OT_BASIC_BITMAP );
2221 bs_write( s, 2, 0 ); /* object provider flag */
2222 bs_write( s, 12, 0 );/* object horizontal position */
2223 bs_write( s, 4, 0 ); /* Reserved */
2224 bs_write( s, 12, 0 );/* object vertical position */
2228 bs_write( s, 8, 1 ); /* foreground pixel code */
2229 bs_write( s, 8, 0 ); /* background pixel code */
2234 static void encode_pixel_data( encoder_t *p_enc, bs_t *s,
2235 subpicture_region_t *p_region,
2238 static void encode_object( encoder_t *p_enc, bs_t *s, subpicture_t *p_subpic )
2240 encoder_sys_t *p_sys = p_enc->p_sys;
2241 subpicture_region_t *p_region;
2244 int i_length_pos, i_update_pos, i_pixel_data_pos;
2246 for( i_region = 0, p_region = p_subpic->p_region; p_region;
2247 p_region = p_region->p_next, i_region++ )
2249 bs_write( s, 8, 0x0f ); /* Sync byte */
2250 bs_write( s, 8, DVBSUB_ST_OBJECT_DATA ); /* Segment type */
2251 bs_write( s, 16, 1 ); /* Page id */
2253 i_length_pos = bs_pos( s );
2254 bs_write( s, 16, 0 ); /* Segment length */
2255 bs_write( s, 16, i_region ); /* Object id */
2256 bs_write( s, 4, p_sys->i_region_ver++ );
2258 /* object coding method */
2259 switch( p_region->fmt.i_chroma )
2261 case VLC_FOURCC( 'Y','U','V','P' ):
2262 bs_write( s, 2, 0 );
2264 case VLC_FOURCC( 'T','E','X','T' ):
2265 bs_write( s, 2, 1 );
2268 msg_Err( p_enc, "FOURCC %d not supported by encoder.", p_region->fmt.i_chroma );
2272 bs_write( s, 1, 0 ); /* non modifying color flag */
2273 bs_write( s, 1, 0 ); /* Reserved */
2275 if( p_region->fmt.i_chroma == VLC_FOURCC( 'T','E','X','T' ) )
2279 if( !p_region->psz_text ) continue;
2281 i_size = __MIN( strlen( p_region->psz_text ), 256 );
2283 bs_write( s, 8, i_size ); /* number of characters in string */
2284 for( i = 0; i < i_size; i++ )
2286 bs_write( s, 16, p_region->psz_text[i] );
2289 /* Update segment length */
2290 SetWBE( &s->p_start[i_length_pos/8],
2291 (bs_pos(s) - i_length_pos)/8 -2 );
2295 /* Coding of a bitmap object */
2296 i_update_pos = bs_pos( s );
2297 bs_write( s, 16, 0 ); /* topfield data block length */
2298 bs_write( s, 16, 0 ); /* bottomfield data block length */
2301 i_pixel_data_pos = bs_pos( s );
2302 encode_pixel_data( p_enc, s, p_region, true );
2303 i_pixel_data_pos = ( bs_pos( s ) - i_pixel_data_pos ) / 8;
2304 SetWBE( &s->p_start[i_update_pos/8], i_pixel_data_pos );
2307 i_pixel_data_pos = bs_pos( s );
2308 encode_pixel_data( p_enc, s, p_region, false );
2309 i_pixel_data_pos = ( bs_pos( s ) - i_pixel_data_pos ) / 8;
2310 SetWBE( &s->p_start[i_update_pos/8+2], i_pixel_data_pos );
2312 /* Stuffing for word alignment */
2314 if( bs_pos( s ) % 16 ) bs_write( s, 8, 0 );
2316 /* Update segment length */
2317 SetWBE( &s->p_start[i_length_pos/8], (bs_pos(s) - i_length_pos)/8 -2 );
2321 static void encode_pixel_line_2bp( bs_t *s, subpicture_region_t *p_region,
2323 static void encode_pixel_line_4bp( bs_t *s, subpicture_region_t *p_region,
2325 static void encode_pixel_line_8bp( bs_t *s, subpicture_region_t *p_region,
2327 static void encode_pixel_data( encoder_t *p_enc, bs_t *s,
2328 subpicture_region_t *p_region,
2331 unsigned int i_line;
2334 if( p_region->fmt.i_chroma != VLC_FOURCC('Y','U','V','P') ) return;
2336 /* Encode line by line */
2337 for( i_line = !b_top; i_line < p_region->fmt.i_visible_height;
2340 switch( p_region->fmt.p_palette->i_entries )
2346 bs_write( s, 8, 0x10 ); /* 2 bit/pixel code string */
2347 encode_pixel_line_2bp( s, p_region, i_line );
2351 bs_write( s, 8, 0x11 ); /* 4 bit/pixel code string */
2352 encode_pixel_line_4bp( s, p_region, i_line );
2356 bs_write( s, 8, 0x12 ); /* 8 bit/pixel code string */
2357 encode_pixel_line_8bp( s, p_region, i_line );
2361 msg_Err( p_enc, "subpicture palette (%i) not handled",
2362 p_region->fmt.p_palette->i_entries );
2366 bs_write( s, 8, 0xf0 ); /* End of object line code */
2370 static void encode_pixel_line_2bp( bs_t *s, subpicture_region_t *p_region,
2373 unsigned int i, i_length = 0;
2374 int i_pitch = p_region->picture.p->i_pitch;
2375 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2376 int i_last_pixel = p_data[0];
2378 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2380 if( ( i != p_region->fmt.i_visible_width ) &&
2381 ( p_data[i] == i_last_pixel ) && ( i_length != 284 ) )
2387 if( ( i_length == 1 ) || ( i_length == 11 ) || ( i_length == 28 ) )
2389 /* 2bit/pixel code */
2391 bs_write( s, 2, i_last_pixel );
2394 bs_write( s, 2, 0 );
2395 bs_write( s, 1, 0 );
2396 bs_write( s, 1, 1 ); /* pseudo color 0 */
2405 bs_write( s, 2, i_last_pixel );
2406 bs_write( s, 2, i_last_pixel );
2410 bs_write( s, 2, 0 );
2411 bs_write( s, 1, 0 );
2412 bs_write( s, 1, 0 );
2413 bs_write( s, 2, 1 ); /* 2 * pseudo color 0 */
2416 else if( i_length > 2 )
2418 bs_write( s, 2, 0 );
2419 if( i_length <= 10 )
2421 bs_write( s, 1, 1 );
2422 bs_write( s, 3, i_length - 3 );
2423 bs_write( s, 2, i_last_pixel );
2427 bs_write( s, 1, 0 );
2428 bs_write( s, 1, 0 );
2430 if( i_length <= 27 )
2432 bs_write( s, 2, 2 );
2433 bs_write( s, 4, i_length - 12 );
2434 bs_write( s, 2, i_last_pixel );
2438 bs_write( s, 2, 3 );
2439 bs_write( s, 8, i_length - 29 );
2440 bs_write( s, 2, i_last_pixel );
2445 if( i == p_region->fmt.i_visible_width ) break;
2447 i_last_pixel = p_data[i];
2452 bs_write( s, 2, 0 );
2453 bs_write( s, 1, 0 );
2454 bs_write( s, 1, 0 );
2455 bs_write( s, 2, 0 );
2461 static void encode_pixel_line_4bp( bs_t *s, subpicture_region_t *p_region,
2464 unsigned int i, i_length = 0;
2465 int i_pitch = p_region->picture.p->i_pitch;
2466 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2467 int i_last_pixel = p_data[0];
2469 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2471 if( i != p_region->fmt.i_visible_width &&
2472 p_data[i] == i_last_pixel && i_length != 280 )
2478 if( ( i_length == 1 ) ||
2479 ( ( i_length == 3 ) && i_last_pixel ) ||
2482 /* 4bit/pixel code */
2484 bs_write( s, 4, i_last_pixel );
2487 bs_write( s, 4, 0 );
2488 bs_write( s, 1, 1 );
2489 bs_write( s, 1, 1 );
2490 bs_write( s, 2, 0 ); /* pseudo color 0 */
2499 bs_write( s, 4, i_last_pixel );
2500 bs_write( s, 4, i_last_pixel );
2504 bs_write( s, 4, 0 );
2505 bs_write( s, 1, 1 );
2506 bs_write( s, 1, 1 );
2507 bs_write( s, 2, 1 ); /* 2 * pseudo color 0 */
2510 else if( !i_last_pixel && ( i_length >= 3 ) && ( i_length <= 9 ) )
2512 bs_write( s, 4, 0 );
2513 bs_write( s, 1, 0 );
2514 bs_write( s, 3, i_length - 2 ); /* (i_length - 2) * color 0 */
2516 else if( i_length > 2 )
2518 bs_write( s, 4, 0 );
2519 bs_write( s, 1, 1 );
2523 bs_write( s, 1, 0 );
2524 bs_write( s, 2, i_length - 4 );
2525 bs_write( s, 4, i_last_pixel );
2529 bs_write( s, 1, 1 );
2531 if( i_length <= 24 )
2533 bs_write( s, 2, 2 );
2534 bs_write( s, 4, i_length - 9 );
2535 bs_write( s, 4, i_last_pixel );
2539 bs_write( s, 2, 3 );
2540 bs_write( s, 8, i_length - 25 );
2541 bs_write( s, 4, i_last_pixel );
2546 if( i == p_region->fmt.i_visible_width ) break;
2548 i_last_pixel = p_data[i];
2553 bs_write( s, 8, 0 );
2559 static void encode_pixel_line_8bp( bs_t *s, subpicture_region_t *p_region,
2562 unsigned int i, i_length = 0;
2563 int i_pitch = p_region->picture.p->i_pitch;
2564 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2565 int i_last_pixel = p_data[0];
2567 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2569 if( ( i != p_region->fmt.i_visible_width ) &&
2570 ( p_data[i] == i_last_pixel ) && ( i_length != 127 ) )
2576 if( ( i_length == 1 ) && i_last_pixel )
2578 /* 8bit/pixel code */
2579 bs_write( s, 8, i_last_pixel );
2581 else if( ( i_length == 2 ) && i_last_pixel )
2583 /* 8bit/pixel code */
2584 bs_write( s, 8, i_last_pixel );
2585 bs_write( s, 8, i_last_pixel );
2587 else if( i_length <= 127 )
2589 bs_write( s, 8, 0 );
2593 bs_write( s, 1, 0 );
2594 bs_write( s, 7, i_length ); /* pseudo color 0 */
2598 bs_write( s, 1, 1 );
2599 bs_write( s, 7, i_length );
2600 bs_write( s, 8, i_last_pixel );
2604 if( i == p_region->fmt.i_visible_width ) break;
2606 i_last_pixel = p_data[i];
2611 bs_write( s, 8, 0 );
2612 bs_write( s, 8, 0 );