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>
14 * This program is free software; you can redistribute it and/or modify
15 * it under the terms of the GNU General Public License as published by
16 * the Free Software Foundation; either version 2 of the License, or
17 * (at your option) any later version.
19 * This program is distributed in the hope that it will be useful,
20 * but WITHOUT ANY WARRANTY; without even the implied warranty of
21 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
22 * GNU General Public License for more details.
24 * You should have received a copy of the GNU General Public License
25 * along with this program; if not, write to the Free Software
26 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
27 *****************************************************************************/
28 /*****************************************************************************
32 * DVB subtitles coded as strings of characters are not handled correctly.
33 * The character codes in the string should actually be indexes refering to a
34 * character table identified in the subtitle descriptor.
35 *****************************************************************************/
38 #include <vlc/decoder.h>
43 /* #define DEBUG_DVBSUB 1 */
45 #define POSX_TEXT N_("Decoding X coordinate")
46 #define POSX_LONGTEXT N_("X coordinate of the rendered subtitle")
48 #define POSY_TEXT N_("Decoding Y coordinate")
49 #define POSY_LONGTEXT N_("Y coordinate of the rendered subtitle")
51 #define POS_TEXT N_("Subpicture position")
52 #define POS_LONGTEXT N_( \
53 "You can enforce the subpicture position on the video " \
54 "(0=center, 1=left, 2=right, 4=top, 8=bottom, you can " \
55 "also use combinations of these values, e.g. 6=top-right).")
57 #define ENC_POSX_TEXT N_("Encoding X coordinate")
58 #define ENC_POSX_LONGTEXT N_("X coordinate of the encoded subtitle" )
59 #define ENC_POSY_TEXT N_("Encoding Y coordinate")
60 #define ENC_POSY_LONGTEXT N_("Y coordinate of the encoded subtitle" )
62 static int pi_pos_values[] = { 0, 1, 2, 4, 8, 5, 6, 9, 10 };
63 static char *ppsz_pos_descriptions[] =
64 { N_("Center"), N_("Left"), N_("Right"), N_("Top"), N_("Bottom"),
65 N_("Top-Left"), N_("Top-Right"), N_("Bottom-Left"), N_("Bottom-Right") };
67 /*****************************************************************************
69 *****************************************************************************/
70 static int Open ( vlc_object_t * );
71 static void Close( vlc_object_t * );
72 static subpicture_t *Decode( decoder_t *, block_t ** );
74 static int OpenEncoder ( vlc_object_t * );
75 static void CloseEncoder( vlc_object_t * );
76 static block_t *Encode ( encoder_t *, subpicture_t * );
79 # define DVBSUB_CFG_PREFIX "dvbsub-"
80 set_description( _("DVB subtitles decoder") );
81 set_capability( "decoder", 50 );
82 set_category( CAT_INPUT );
83 set_subcategory( SUBCAT_INPUT_SCODEC );
84 set_callbacks( Open, Close );
86 add_integer( DVBSUB_CFG_PREFIX "position", 8, NULL, POS_TEXT, POS_LONGTEXT, VLC_TRUE );
87 change_integer_list( pi_pos_values, ppsz_pos_descriptions, 0 );
88 add_integer( DVBSUB_CFG_PREFIX "x", -1, NULL, POSX_TEXT, POSX_LONGTEXT, VLC_FALSE );
89 add_integer( DVBSUB_CFG_PREFIX "y", -1, NULL, POSY_TEXT, POSY_LONGTEXT, VLC_FALSE );
91 # define ENC_CFG_PREFIX "sout-dvbsub-"
93 set_description( _("DVB subtitles encoder") );
94 set_capability( "encoder", 100 );
95 set_callbacks( OpenEncoder, CloseEncoder );
97 add_integer( ENC_CFG_PREFIX "x", -1, NULL, ENC_POSX_TEXT, ENC_POSX_LONGTEXT, VLC_FALSE );
98 add_integer( ENC_CFG_PREFIX "y", -1, NULL, ENC_POSY_TEXT, ENC_POSY_LONGTEXT, VLC_FALSE );
99 add_suppressed_integer( ENC_CFG_PREFIX "timeout" ); /* Suppressed since 0.8.5 */
102 static const char *ppsz_enc_options[] = { "x", "y", NULL };
104 /****************************************************************************
106 ****************************************************************************
107 * Those structures refer closely to the ETSI 300 743 Object model
108 ****************************************************************************/
110 /* The object definition gives the position of the object in a region */
111 typedef struct dvbsub_objectdef_s
119 char *psz_text; /* for string of characters objects */
121 } dvbsub_objectdef_t;
123 /* The entry in the palette CLUT */
134 typedef struct dvbsub_clut_s
138 dvbsub_color_t c_2b[4];
139 dvbsub_color_t c_4b[16];
140 dvbsub_color_t c_8b[256];
142 struct dvbsub_clut_s *p_next;
146 /* The Region is an aera on the image
147 * with a list of the object definitions associated and a CLUT */
148 typedef struct dvbsub_region_s
163 dvbsub_objectdef_t *p_object_defs;
165 struct dvbsub_region_s *p_next;
169 /* The object definition gives the position of the object in a region */
170 typedef struct dvbsub_regiondef_s
176 } dvbsub_regiondef_t;
178 /* The page defines the list of regions */
182 int i_timeout; /* in seconds */
187 dvbsub_regiondef_t *p_region_defs;
195 /* Decoder internal data */
200 vlc_bool_t b_absolute;
206 dvbsub_page_t *p_page;
207 dvbsub_region_t *p_regions;
208 dvbsub_clut_t *p_cluts;
209 dvbsub_clut_t default_clut;
213 /* List of different SEGMENT TYPES */
214 /* According to EN 300-743, table 2 */
215 #define DVBSUB_ST_PAGE_COMPOSITION 0x10
216 #define DVBSUB_ST_REGION_COMPOSITION 0x11
217 #define DVBSUB_ST_CLUT_DEFINITION 0x12
218 #define DVBSUB_ST_OBJECT_DATA 0x13
219 #define DVBSUB_ST_ENDOFDISPLAY 0x80
220 #define DVBSUB_ST_STUFFING 0xff
221 /* List of different OBJECT TYPES */
222 /* According to EN 300-743, table 6 */
223 #define DVBSUB_OT_BASIC_BITMAP 0x00
224 #define DVBSUB_OT_BASIC_CHAR 0x01
225 #define DVBSUB_OT_COMPOSITE_STRING 0x02
226 /* Pixel DATA TYPES */
227 /* According to EN 300-743, table 9 */
228 #define DVBSUB_DT_2BP_CODE_STRING 0x10
229 #define DVBSUB_DT_4BP_CODE_STRING 0x11
230 #define DVBSUB_DT_8BP_CODE_STRING 0x12
231 #define DVBSUB_DT_24_TABLE_DATA 0x20
232 #define DVBSUB_DT_28_TABLE_DATA 0x21
233 #define DVBSUB_DT_48_TABLE_DATA 0x22
234 #define DVBSUB_DT_END_LINE 0xf0
235 /* List of different Page Composition Segment state */
236 /* According to EN 300-743, 7.2.1 table 3 */
237 #define DVBSUB_PCS_STATE_ACQUISITION 0x01
238 #define DVBSUB_PCS_STATE_CHANGE 0x02
240 /*****************************************************************************
242 *****************************************************************************/
243 static void decode_segment( decoder_t *, bs_t * );
244 static void decode_page_composition( decoder_t *, bs_t * );
245 static void decode_region_composition( decoder_t *, bs_t * );
246 static void decode_object( decoder_t *, bs_t * );
247 static void decode_clut( decoder_t *, bs_t * );
248 static void free_all( decoder_t * );
250 static void default_clut_init( decoder_t * );
252 static subpicture_t *render( decoder_t * );
254 /*****************************************************************************
255 * Open: probe the decoder and return score
256 *****************************************************************************
257 * Tries to launch a decoder and return score so that the interface is able
259 *****************************************************************************/
260 static int Open( vlc_object_t *p_this )
262 decoder_t *p_dec = (decoder_t *) p_this;
263 decoder_sys_t *p_sys;
267 if( p_dec->fmt_in.i_codec != VLC_FOURCC('d','v','b','s') )
272 p_dec->pf_decode_sub = Decode;
273 p_sys = p_dec->p_sys = malloc( sizeof(decoder_sys_t) );
274 memset( p_sys, 0, sizeof(decoder_sys_t) );
276 p_sys->i_pts = (mtime_t) 0;
277 p_sys->i_id = p_dec->fmt_in.subs.dvb.i_id & 0xFFFF;
278 p_sys->i_ancillary_id = p_dec->fmt_in.subs.dvb.i_id >> 16;
280 p_sys->p_regions = NULL;
281 p_sys->p_cluts = NULL;
282 p_sys->p_page = NULL;
284 var_Create( p_this, DVBSUB_CFG_PREFIX "position",
285 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
286 var_Get( p_this, DVBSUB_CFG_PREFIX "position", &val );
287 p_sys->i_spu_position = val.i_int;
288 var_Create( p_this, DVBSUB_CFG_PREFIX "x",
289 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
290 var_Get( p_this, DVBSUB_CFG_PREFIX "x", &val );
292 var_Create( p_this, DVBSUB_CFG_PREFIX "y",
293 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
294 var_Get( p_this, DVBSUB_CFG_PREFIX "y", &val );
297 /* Check if subpicture position was overridden */
298 p_sys->b_absolute = VLC_TRUE;
299 p_sys->i_spu_x = p_sys->i_spu_y = 0;
301 if( i_posx >= 0 && i_posy >= 0 )
303 p_sys->b_absolute = VLC_FALSE;
304 p_sys->i_spu_x = i_posx;
305 p_sys->i_spu_y = i_posy;
308 es_format_Init( &p_dec->fmt_out, SPU_ES, VLC_FOURCC( 'd','v','b','s' ) );
310 default_clut_init( p_dec );
315 /*****************************************************************************
317 *****************************************************************************/
318 static void Close( vlc_object_t *p_this )
320 decoder_t *p_dec = (decoder_t*) p_this;
321 decoder_sys_t *p_sys = p_dec->p_sys;
323 var_Destroy( p_this, DVBSUB_CFG_PREFIX "x" );
324 var_Destroy( p_this, DVBSUB_CFG_PREFIX "y" );
325 var_Destroy( p_this, DVBSUB_CFG_PREFIX "position" );
331 /*****************************************************************************
333 *****************************************************************************/
334 static subpicture_t *Decode( decoder_t *p_dec, block_t **pp_block )
336 decoder_sys_t *p_sys = p_dec->p_sys;
338 subpicture_t *p_spu = NULL;
340 if( pp_block == NULL || *pp_block == NULL ) return NULL;
344 p_sys->i_pts = p_block->i_pts;
345 if( p_sys->i_pts <= 0 )
348 /* Some DVB channels send stuffing segments in non-dated packets so
349 * don't complain too loudly. */
350 msg_Warn( p_dec, "non dated subtitle" );
352 block_Release( p_block );
356 bs_init( &p_sys->bs, p_block->p_buffer, p_block->i_buffer );
358 if( bs_read( &p_sys->bs, 8 ) != 0x20 ) /* Data identifier */
360 msg_Dbg( p_dec, "invalid data identifier" );
361 block_Release( p_block );
365 if( bs_read( &p_sys->bs, 8 ) ) /* Subtitle stream id */
367 msg_Dbg( p_dec, "invalid subtitle stream id" );
368 block_Release( p_block );
373 msg_Dbg( p_dec, "subtitle packet received: "I64Fd, p_sys->i_pts );
376 p_sys->b_page = VLC_FALSE;
377 while( bs_show( &p_sys->bs, 8 ) == 0x0f ) /* Sync byte */
379 decode_segment( p_dec, &p_sys->bs );
382 if( bs_read( &p_sys->bs, 8 ) != 0xff ) /* End marker */
384 msg_Warn( p_dec, "end marker not found (corrupted subtitle ?)" );
385 block_Release( p_block );
389 /* Check if the page is to be displayed */
390 if( p_sys->p_page && p_sys->b_page ) p_spu = render( p_dec );
392 block_Release( p_block );
397 /* following functions are local */
399 /*****************************************************************************
400 * default_clut_init: default clut as defined in EN 300-743 section 10
401 *****************************************************************************/
402 static void default_clut_init( decoder_t *p_dec )
404 decoder_sys_t *p_sys = p_dec->p_sys;
407 #define RGB_TO_Y(r, g, b) ((int16_t) 77 * r + 150 * g + 29 * b) / 256;
408 #define RGB_TO_U(r, g, b) ((int16_t) -44 * r - 87 * g + 131 * b) / 256;
409 #define RGB_TO_V(r, g, b) ((int16_t) 131 * r - 110 * g - 21 * b) / 256;
412 for( i = 0; i < 4; i++ )
414 uint8_t R = 0, G = 0, B = 0, T = 0;
416 if( !(i & 0x2) && !(i & 0x1) ) T = 0xFF;
417 else if( !(i & 0x2) && (i & 0x1) ) R = G = B = 0xFF;
418 else if( (i & 0x2) && !(i & 0x1) ) R = G = B = 0;
419 else R = G = B = 0x7F;
421 p_sys->default_clut.c_2b[i].Y = RGB_TO_Y(R,G,B);
422 p_sys->default_clut.c_2b[i].Cr = RGB_TO_U(R,G,B);
423 p_sys->default_clut.c_2b[i].Cb = RGB_TO_V(R,G,B);
424 p_sys->default_clut.c_2b[i].T = T;
427 /* 16 entries CLUT */
428 for( i = 0; i < 16; i++ )
430 uint8_t R = 0, G = 0, B = 0, T = 0;
434 if( !(i & 0x4) && !(i & 0x2) && !(i & 0x1) )
440 R = (i & 0x1) ? 0xFF : 0;
441 G = (i & 0x2) ? 0xFF : 0;
442 B = (i & 0x4) ? 0xFF : 0;
447 R = (i & 0x1) ? 0x7F : 0;
448 G = (i & 0x2) ? 0x7F : 0;
449 B = (i & 0x4) ? 0x7F : 0;
452 p_sys->default_clut.c_4b[i].Y = RGB_TO_Y(R,G,B);
453 p_sys->default_clut.c_4b[i].Cr = RGB_TO_U(R,G,B);
454 p_sys->default_clut.c_4b[i].Cb = RGB_TO_V(R,G,B);
455 p_sys->default_clut.c_4b[i].T = T;
458 /* 256 entries CLUT */
459 memset( p_sys->default_clut.c_8b, 0xFF, 256 * sizeof(dvbsub_color_t) );
462 static void decode_segment( decoder_t *p_dec, bs_t *s )
464 decoder_sys_t *p_sys = p_dec->p_sys;
469 /* sync_byte (already checked) */
473 i_type = bs_read( s, 8 );
476 i_page_id = bs_read( s, 16 );
479 i_size = bs_show( s, 16 );
481 if( i_page_id != p_sys->i_id && i_page_id != p_sys->i_ancillary_id )
484 msg_Dbg( p_dec, "subtitle skipped (page id: %i, %i)",
485 i_page_id, p_sys->i_id );
487 bs_skip( s, 8 * ( 2 + i_size ) );
491 if( p_sys->i_ancillary_id != p_sys->i_id &&
492 i_type == DVBSUB_ST_PAGE_COMPOSITION &&
493 i_page_id == p_sys->i_ancillary_id )
496 msg_Dbg( p_dec, "skipped invalid ancillary subtitle packet" );
498 bs_skip( s, 8 * ( 2 + i_size ) );
503 if( i_page_id == p_sys->i_id )
504 msg_Dbg( p_dec, "segment (id: %i)", i_page_id );
506 msg_Dbg( p_dec, "ancillary segment (id: %i)", i_page_id );
511 case DVBSUB_ST_PAGE_COMPOSITION:
513 msg_Dbg( p_dec, "decode_page_composition" );
515 decode_page_composition( p_dec, s );
518 case DVBSUB_ST_REGION_COMPOSITION:
520 msg_Dbg( p_dec, "decode_region_composition" );
522 decode_region_composition( p_dec, s );
525 case DVBSUB_ST_CLUT_DEFINITION:
527 msg_Dbg( p_dec, "decode_clut" );
529 decode_clut( p_dec, s );
532 case DVBSUB_ST_OBJECT_DATA:
534 msg_Dbg( p_dec, "decode_object" );
536 decode_object( p_dec, s );
539 case DVBSUB_ST_ENDOFDISPLAY:
541 msg_Dbg( p_dec, "end of display" );
543 bs_skip( s, 8 * ( 2 + i_size ) );
546 case DVBSUB_ST_STUFFING:
548 msg_Dbg( p_dec, "skip stuffing" );
550 bs_skip( s, 8 * ( 2 + i_size ) );
554 msg_Warn( p_dec, "unsupported segment type: (%04x)", i_type );
555 bs_skip( s, 8 * ( 2 + i_size ) );
560 static void decode_clut( decoder_t *p_dec, bs_t *s )
562 decoder_sys_t *p_sys = p_dec->p_sys;
563 uint16_t i_segment_length;
564 uint16_t i_processed_length;
565 dvbsub_clut_t *p_clut, *p_next;
568 i_segment_length = bs_read( s, 16 );
569 i_id = bs_read( s, 8 );
570 i_version = bs_read( s, 4 );
572 /* Check if we already have this clut */
573 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut->p_next )
575 if( p_clut->i_id == i_id ) break;
578 /* Check version number */
579 if( p_clut && p_clut->i_version == i_version )
582 bs_skip( s, 8 * i_segment_length - 12 );
589 msg_Dbg( p_dec, "new clut: %i", i_id );
591 p_clut = malloc( sizeof(dvbsub_clut_t) );
592 p_clut->p_next = p_sys->p_cluts;
593 p_sys->p_cluts = p_clut;
596 /* Initialize to default clut */
597 p_next = p_clut->p_next;
598 *p_clut = p_sys->default_clut;
599 p_clut->p_next = p_next;
601 /* We don't have this version of the CLUT: Parse it */
602 p_clut->i_version = i_version;
604 bs_skip( s, 4 ); /* Reserved bits */
605 i_processed_length = 2;
606 while( i_processed_length < i_segment_length )
608 uint8_t y, cb, cr, t;
612 i_id = bs_read( s, 8 );
613 i_type = bs_read( s, 3 );
617 if( bs_read( s, 1 ) )
620 cr = bs_read( s, 8 );
621 cb = bs_read( s, 8 );
623 i_processed_length += 6;
627 y = bs_read( s, 6 ) << 2;
628 cr = bs_read( s, 4 ) << 4;
629 cb = bs_read( s, 4 ) << 4;
630 t = bs_read( s, 2 ) << 6;
631 i_processed_length += 4;
634 /* We are not entirely compliant here as full transparency is indicated
635 * with a luma value of zero, not a transparency value of 0xff
636 * (full transparency would actually be 0xff + 1). */
643 /* According to EN 300-743 section 7.2.3 note 1, type should
644 * not have more than 1 bit set to one, but some streams don't
645 * respect this note. */
646 if( i_type & 0x04 && i_id < 4 )
648 p_clut->c_2b[i_id].Y = y;
649 p_clut->c_2b[i_id].Cr = cr;
650 p_clut->c_2b[i_id].Cb = cb;
651 p_clut->c_2b[i_id].T = t;
653 if( i_type & 0x02 && i_id < 16 )
655 p_clut->c_4b[i_id].Y = y;
656 p_clut->c_4b[i_id].Cr = cr;
657 p_clut->c_4b[i_id].Cb = cb;
658 p_clut->c_4b[i_id].T = t;
662 p_clut->c_8b[i_id].Y = y;
663 p_clut->c_8b[i_id].Cr = cr;
664 p_clut->c_8b[i_id].Cb = cb;
665 p_clut->c_8b[i_id].T = t;
670 static void decode_page_composition( decoder_t *p_dec, bs_t *s )
672 decoder_sys_t *p_sys = p_dec->p_sys;
673 int i_version, i_state, i_segment_length, i_timeout, i;
675 /* A page is composed by 0 or more region */
676 i_segment_length = bs_read( s, 16 );
677 i_timeout = bs_read( s, 8 );
678 i_version = bs_read( s, 4 );
679 i_state = bs_read( s, 2 );
680 bs_skip( s, 2 ); /* Reserved */
682 if( i_state == DVBSUB_PCS_STATE_CHANGE )
684 /* End of an epoch, reset decoder buffer */
686 msg_Dbg( p_dec, "page composition mode change" );
690 else if( !p_sys->p_page && i_state != DVBSUB_PCS_STATE_ACQUISITION &&
691 i_state != DVBSUB_PCS_STATE_CHANGE )
693 /* Not a full PCS, we need to wait for one */
694 msg_Dbg( p_dec, "didn't receive an acquisition page yet" );
697 /* Try to start decoding even without an acquisition page */
698 bs_skip( s, 8 * (i_segment_length - 2) );
704 if( i_state == DVBSUB_PCS_STATE_ACQUISITION )
705 msg_Dbg( p_dec, "acquisition page composition" );
708 /* Check version number */
709 if( p_sys->p_page && p_sys->p_page->i_version == i_version )
711 bs_skip( s, 8 * (i_segment_length - 2) );
714 else if( p_sys->p_page )
716 if( p_sys->p_page->i_region_defs )
717 free( p_sys->p_page->p_region_defs );
718 p_sys->p_page->i_region_defs = 0;
724 msg_Dbg( p_dec, "new page" );
726 /* Allocate a new page */
727 p_sys->p_page = malloc( sizeof(dvbsub_page_t) );
730 p_sys->p_page->i_version = i_version;
731 p_sys->p_page->i_timeout = i_timeout;
732 p_sys->b_page = VLC_TRUE;
734 /* Number of regions */
735 p_sys->p_page->i_region_defs = (i_segment_length - 2) / 6;
737 if( p_sys->p_page->i_region_defs == 0 ) return;
739 p_sys->p_page->p_region_defs =
740 malloc( p_sys->p_page->i_region_defs * sizeof(dvbsub_region_t) );
741 for( i = 0; i < p_sys->p_page->i_region_defs; i++ )
743 p_sys->p_page->p_region_defs[i].i_id = bs_read( s, 8 );
744 bs_skip( s, 8 ); /* Reserved */
745 p_sys->p_page->p_region_defs[i].i_x = bs_read( s, 16 );
746 p_sys->p_page->p_region_defs[i].i_y = bs_read( s, 16 );
749 msg_Dbg( p_dec, "page_composition, region %i (%i,%i)",
750 i, p_sys->p_page->p_region_defs[i].i_x,
751 p_sys->p_page->p_region_defs[i].i_y );
756 static void decode_region_composition( decoder_t *p_dec, bs_t *s )
758 decoder_sys_t *p_sys = p_dec->p_sys;
759 dvbsub_region_t *p_region, **pp_region = &p_sys->p_regions;
760 int i_segment_length, i_processed_length, i_id, i_version;
761 int i_width, i_height, i_level_comp, i_depth, i_clut;
762 int i_8_bg, i_4_bg, i_2_bg;
765 i_segment_length = bs_read( s, 16 );
766 i_id = bs_read( s, 8 );
767 i_version = bs_read( s, 4 );
769 /* Check if we already have this region */
770 for( p_region = p_sys->p_regions; p_region != NULL;
771 p_region = p_region->p_next )
773 pp_region = &p_region->p_next;
774 if( p_region->i_id == i_id ) break;
777 /* Check version number */
778 if( p_region && p_region->i_version == i_version )
780 bs_skip( s, 8 * (i_segment_length - 1) - 4 );
787 msg_Dbg( p_dec, "new region: %i", i_id );
789 p_region = *pp_region = malloc( sizeof(dvbsub_region_t) );
790 memset( p_region, 0, sizeof(dvbsub_region_t) );
791 p_region->p_object_defs = NULL;
792 p_region->p_pixbuf = NULL;
793 p_region->p_next = NULL;
796 /* Region attributes */
797 p_region->i_id = i_id;
798 p_region->i_version = i_version;
799 b_fill = bs_read( s, 1 );
800 bs_skip( s, 3 ); /* Reserved */
802 i_width = bs_read( s, 16 );
803 i_height = bs_read( s, 16 );
805 msg_Dbg( p_dec, " width=%d height=%d", i_width, i_height );
807 i_level_comp = bs_read( s, 3 );
808 i_depth = bs_read( s, 3 );
809 bs_skip( s, 2 ); /* Reserved */
810 i_clut = bs_read( s, 8 );
812 i_8_bg = bs_read( s, 8 );
813 i_4_bg = bs_read( s, 4 );
814 i_2_bg = bs_read( s, 2 );
815 bs_skip( s, 2 ); /* Reserved */
817 /* Free old object defs */
818 while( p_region->i_object_defs )
820 int i = p_region->i_object_defs - 1;
821 if( p_region->p_object_defs[i].psz_text )
822 free( p_region->p_object_defs[i].psz_text );
823 if( !i ) free( p_region->p_object_defs );
825 p_region->i_object_defs--;
827 p_region->p_object_defs = NULL;
829 /* Extra sanity checks */
830 if( p_region->i_width != i_width || p_region->i_height != i_height )
832 if( p_region->p_pixbuf )
834 msg_Dbg( p_dec, "region size changed (%dx%d->%dx%d)",
835 p_region->i_width, p_region->i_height, i_width, i_height );
836 free( p_region->p_pixbuf );
839 p_region->p_pixbuf = malloc( i_height * i_width );
840 p_region->i_depth = 0;
843 if( p_region->i_depth && (p_region->i_depth != i_depth ||
844 p_region->i_level_comp != i_level_comp || p_region->i_clut != i_clut) )
846 msg_Dbg( p_dec, "region parameters changed (not allowed)" );
849 /* Erase background of region */
852 int i_background = (p_region->i_depth == 1) ? i_2_bg :
853 (p_region->i_depth == 2) ? i_4_bg : i_8_bg;
854 memset( p_region->p_pixbuf, i_background, i_width * i_height );
857 p_region->i_width = i_width;
858 p_region->i_height = i_height;
859 p_region->i_level_comp = i_level_comp;
860 p_region->i_depth = i_depth;
861 p_region->i_clut = i_clut;
863 /* List of objects in the region */
864 i_processed_length = 10;
865 while( i_processed_length < i_segment_length )
867 dvbsub_objectdef_t *p_obj;
869 /* We create a new object */
870 p_region->i_object_defs++;
871 p_region->p_object_defs =
872 realloc( p_region->p_object_defs,
873 sizeof(dvbsub_objectdef_t) * p_region->i_object_defs );
875 /* We parse object properties */
876 p_obj = &p_region->p_object_defs[p_region->i_object_defs - 1];
877 p_obj->i_id = bs_read( s, 16 );
878 p_obj->i_type = bs_read( s, 2 );
879 bs_skip( s, 2 ); /* Provider */
880 p_obj->i_x = bs_read( s, 12 );
881 bs_skip( s, 4 ); /* Reserved */
882 p_obj->i_y = bs_read( s, 12 );
885 i_processed_length += 6;
887 if( p_obj->i_type == DVBSUB_OT_BASIC_CHAR ||
888 p_obj->i_type == DVBSUB_OT_COMPOSITE_STRING )
890 p_obj->i_fg_pc = bs_read( s, 8 );
891 p_obj->i_bg_pc = bs_read( s, 8 );
892 i_processed_length += 2;
897 static void dvbsub_render_pdata( decoder_t *, dvbsub_region_t *, int, int,
899 static void dvbsub_pdata2bpp( bs_t *, uint8_t *, int, int * );
900 static void dvbsub_pdata4bpp( bs_t *, uint8_t *, int, int * );
901 static void dvbsub_pdata8bpp( bs_t *, uint8_t *, int, int * );
903 static void decode_object( decoder_t *p_dec, bs_t *s )
905 decoder_sys_t *p_sys = p_dec->p_sys;
906 dvbsub_region_t *p_region;
907 int i_segment_length, i_coding_method, i_version, i_id, i;
908 vlc_bool_t b_non_modify_color;
910 /* ETSI 300-743 paragraph 7.2.4
911 * sync_byte, segment_type and page_id have already been processed.
913 i_segment_length = bs_read( s, 16 );
914 i_id = bs_read( s, 16 );
915 i_version = bs_read( s, 4 );
916 i_coding_method = bs_read( s, 2 );
918 if( i_coding_method > 1 )
920 msg_Dbg( p_dec, "unknown DVB subtitling coding %d is not handled!", i_coding_method );
921 bs_skip( s, 8 * (i_segment_length - 2) - 6 );
925 /* Check if the object needs to be rendered in at least one
927 for( p_region = p_sys->p_regions; p_region != NULL;
928 p_region = p_region->p_next )
930 for( i = 0; i < p_region->i_object_defs; i++ )
931 if( p_region->p_object_defs[i].i_id == i_id ) break;
933 if( i != p_region->i_object_defs ) break;
937 bs_skip( s, 8 * (i_segment_length - 2) - 6 );
942 msg_Dbg( p_dec, "new object: %i", i_id );
945 b_non_modify_color = bs_read( s, 1 );
946 bs_skip( s, 1 ); /* Reserved */
948 if( i_coding_method == 0x00 )
950 int i_topfield, i_bottomfield;
951 uint8_t *p_topfield, *p_bottomfield;
953 i_topfield = bs_read( s, 16 );
954 i_bottomfield = bs_read( s, 16 );
955 p_topfield = s->p_start + bs_pos( s ) / 8;
956 p_bottomfield = p_topfield + i_topfield;
958 bs_skip( s, 8 * (i_segment_length - 7) );
961 if( i_segment_length < i_topfield + i_bottomfield + 7 ||
962 p_topfield + i_topfield + i_bottomfield > s->p_end )
964 msg_Dbg( p_dec, "corrupted object data" );
968 for( p_region = p_sys->p_regions; p_region != NULL;
969 p_region = p_region->p_next )
971 for( i = 0; i < p_region->i_object_defs; i++ )
973 if( p_region->p_object_defs[i].i_id != i_id ) continue;
975 dvbsub_render_pdata( p_dec, p_region,
976 p_region->p_object_defs[i].i_x,
977 p_region->p_object_defs[i].i_y,
978 p_topfield, i_topfield );
982 dvbsub_render_pdata( p_dec, p_region,
983 p_region->p_object_defs[i].i_x,
984 p_region->p_object_defs[i].i_y + 1,
985 p_bottomfield, i_bottomfield );
989 /* Duplicate the top field */
990 dvbsub_render_pdata( p_dec, p_region,
991 p_region->p_object_defs[i].i_x,
992 p_region->p_object_defs[i].i_y + 1,
993 p_topfield, i_topfield );
1000 /* DVB subtitling as characters */
1001 int i_number_of_codes = bs_read( s, 8 );
1002 uint8_t* p_start = s->p_start + bs_pos( s ) / 8;
1005 if( i_segment_length < i_number_of_codes*2 + 4 ||
1006 p_start + i_number_of_codes*2 > s->p_end )
1008 msg_Dbg( p_dec, "corrupted object data" );
1012 for( p_region = p_sys->p_regions; p_region != NULL;
1013 p_region = p_region->p_next )
1015 for( i = 0; i < p_region->i_object_defs; i++ )
1019 if( p_region->p_object_defs[i].i_id != i_id ) continue;
1021 p_region->p_object_defs[i].psz_text =
1022 realloc( p_region->p_object_defs[i].psz_text,
1023 i_number_of_codes + 1 );
1025 for( j = 0; j < i_number_of_codes; j++ )
1027 p_region->p_object_defs[i].psz_text[j] = bs_read( s, 16 );
1029 p_region->p_object_defs[i].psz_text[j] = 0;
1035 msg_Dbg( p_dec, "end object: %i", i_id );
1039 static void dvbsub_render_pdata( decoder_t *p_dec, dvbsub_region_t *p_region,
1041 uint8_t *p_field, int i_field )
1048 if( !p_region->p_pixbuf )
1050 msg_Err( p_dec, "region %i has no pixel buffer!", p_region->i_id );
1053 if( i_y < 0 || i_x < 0 || i_y >= p_region->i_height ||
1054 i_x >= p_region->i_width )
1056 msg_Dbg( p_dec, "invalid offset (%i,%i)", i_x, i_y );
1060 p_pixbuf = p_region->p_pixbuf + i_y * p_region->i_width;
1061 bs_init( &bs, p_field, i_field );
1063 while( !bs_eof( &bs ) )
1066 if( i_y >= p_region->i_height ) return;
1068 switch( bs_read( &bs, 8 ) )
1071 dvbsub_pdata2bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1076 dvbsub_pdata4bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1081 dvbsub_pdata8bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1088 /* We don't use map tables */
1091 case 0xf0: /* End of line code */
1092 p_pixbuf += 2*p_region->i_width;
1093 i_offset = 0; i_y += 2;
1099 static void dvbsub_pdata2bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1101 vlc_bool_t b_stop = 0;
1103 while( !b_stop && !bs_eof( s ) )
1105 int i_count = 0, i_color = 0;
1107 if( (i_color = bs_read( s, 2 )) != 0x00 )
1113 if( bs_read( s, 1 ) == 0x01 ) // Switch1
1115 i_count = 3 + bs_read( s, 3 );
1116 i_color = bs_read( s, 2 );
1120 if( bs_read( s, 1 ) == 0x00 ) //Switch2
1122 switch( bs_read( s, 2 ) ) //Switch3
1131 i_count = 12 + bs_read( s, 4 );
1132 i_color = bs_read( s, 2 );
1135 i_count = 29 + bs_read( s, 8 );
1136 i_color = bs_read( s, 2 );
1144 /* 1 pixel color 0 */
1150 if( !i_count ) continue;
1153 if( i_count + *pi_off > i_width ) break;
1155 if( i_count == 1 ) p[*pi_off] = i_color;
1156 else memset( p + *pi_off, i_color, i_count );
1158 (*pi_off) += i_count;
1164 static void dvbsub_pdata4bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1166 vlc_bool_t b_stop = 0;
1168 while( !b_stop && !bs_eof( s ) )
1170 int i_count = 0, i_color = 0;
1172 if( (i_color = bs_read( s, 4 )) != 0x00 )
1179 if( bs_read( s, 1 ) == 0x00 ) // Switch1
1181 if( bs_show( s, 3 ) != 0x00 )
1183 i_count = 2 + bs_read( s, 3 );
1193 if( bs_read( s, 1 ) == 0x00) //Switch2
1195 i_count = 4 + bs_read( s, 2 );
1196 i_color = bs_read( s, 4 );
1200 switch ( bs_read( s, 2 ) ) //Switch3
1209 i_count = 9 + bs_read( s, 4 );
1210 i_color = bs_read( s, 4 );
1213 i_count= 25 + bs_read( s, 8 );
1214 i_color = bs_read( s, 4 );
1221 if( !i_count ) continue;
1224 if( i_count + *pi_off > i_width ) break;
1226 if( i_count == 1 ) p[*pi_off] = i_color;
1227 else memset( p + *pi_off, i_color, i_count );
1229 (*pi_off) += i_count;
1235 static void dvbsub_pdata8bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1237 vlc_bool_t b_stop = 0;
1239 while( !b_stop && !bs_eof( s ) )
1241 int i_count = 0, i_color = 0;
1243 if( (i_color = bs_read( s, 8 )) != 0x00 )
1250 if( bs_read( s, 1 ) == 0x00 ) // Switch1
1252 if( bs_show( s, 7 ) != 0x00 )
1254 i_count = bs_read( s, 7 );
1264 i_count = bs_read( s, 7 );
1265 i_color = bs_read( s, 8 );
1269 if( !i_count ) continue;
1272 if( i_count + *pi_off > i_width ) break;
1274 if( i_count == 1 ) p[*pi_off] = i_color;
1275 else memset( p + *pi_off, i_color, i_count );
1277 (*pi_off) += i_count;
1283 static void free_all( decoder_t *p_dec )
1285 decoder_sys_t *p_sys = p_dec->p_sys;
1286 dvbsub_region_t *p_reg, *p_reg_next;
1287 dvbsub_clut_t *p_clut, *p_clut_next;
1289 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut_next )
1291 p_clut_next = p_clut->p_next;
1294 p_sys->p_cluts = NULL;
1296 for( p_reg = p_sys->p_regions; p_reg != NULL; p_reg = p_reg_next )
1300 p_reg_next = p_reg->p_next;
1301 for( i = 0; i < p_reg->i_object_defs; i++ )
1302 if( p_reg->p_object_defs[i].psz_text )
1303 free( p_reg->p_object_defs[i].psz_text );
1304 if( p_reg->i_object_defs ) free( p_reg->p_object_defs );
1305 if( p_reg->p_pixbuf ) free( p_reg->p_pixbuf );
1308 p_sys->p_regions = NULL;
1312 if( p_sys->p_page->i_region_defs )
1313 free( p_sys->p_page->p_region_defs );
1314 free( p_sys->p_page );
1316 p_sys->p_page = NULL;
1319 static subpicture_t *render( decoder_t *p_dec )
1321 decoder_sys_t *p_sys = p_dec->p_sys;
1322 subpicture_t *p_spu;
1323 subpicture_region_t **pp_spu_region;
1324 int i, j, i_timeout = 0;
1326 /* Allocate the subpicture internal data. */
1327 p_spu = p_dec->pf_spu_buffer_new( p_dec );
1328 if( !p_spu ) return NULL;
1330 pp_spu_region = &p_spu->p_region;
1332 /* Loop on region definitions */
1335 msg_Dbg( p_dec, "rendering %i regions", p_sys->p_page->i_region_defs );
1338 for( i = 0; p_sys->p_page && i < p_sys->p_page->i_region_defs; i++ )
1340 dvbsub_region_t *p_region;
1341 dvbsub_regiondef_t *p_regiondef;
1342 dvbsub_clut_t *p_clut;
1343 dvbsub_color_t *p_color;
1344 subpicture_region_t *p_spu_region;
1345 uint8_t *p_src, *p_dst;
1349 i_timeout = p_sys->p_page->i_timeout;
1351 p_regiondef = &p_sys->p_page->p_region_defs[i];
1354 msg_Dbg( p_dec, "rendering region %i (%i,%i)", i,
1355 p_regiondef->i_x, p_regiondef->i_y );
1358 /* Find associated region */
1359 for( p_region = p_sys->p_regions; p_region != NULL;
1360 p_region = p_region->p_next )
1362 if( p_regiondef->i_id == p_region->i_id ) break;
1367 msg_Dbg( p_dec, "region %i not found", p_regiondef->i_id );
1371 /* Find associated CLUT */
1372 for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut->p_next )
1374 if( p_region->i_clut == p_clut->i_id ) break;
1378 msg_Dbg( p_dec, "clut %i not found", p_region->i_clut );
1382 /* Create new SPU region */
1383 memset( &fmt, 0, sizeof(video_format_t) );
1384 fmt.i_chroma = VLC_FOURCC('Y','U','V','P');
1385 fmt.i_aspect = 0; /* 0 means use aspect ratio of background video */
1386 fmt.i_width = fmt.i_visible_width = p_region->i_width;
1387 fmt.i_height = fmt.i_visible_height = p_region->i_height;
1388 fmt.i_x_offset = fmt.i_y_offset = 0;
1389 p_spu_region = p_spu->pf_create_region( VLC_OBJECT(p_dec), &fmt );
1392 msg_Err( p_dec, "cannot allocate SPU region" );
1395 p_spu_region->i_x = p_regiondef->i_x;
1396 p_spu_region->i_y = p_regiondef->i_y;
1397 *pp_spu_region = p_spu_region;
1398 pp_spu_region = &p_spu_region->p_next;
1401 fmt.p_palette->i_entries = p_region->i_depth == 1 ? 4 :
1402 p_region->i_depth == 2 ? 16 : 256;
1403 p_color = (p_region->i_depth == 1) ? p_clut->c_2b :
1404 (p_region->i_depth == 2) ? p_clut->c_4b : p_clut->c_8b;
1405 for( j = 0; j < fmt.p_palette->i_entries; j++ )
1407 fmt.p_palette->palette[j][0] = p_color[j].Y;
1408 fmt.p_palette->palette[j][1] = p_color[j].Cr;
1409 fmt.p_palette->palette[j][2] = p_color[j].Cb;
1410 fmt.p_palette->palette[j][3] = 0xff - p_color[j].T;
1413 p_src = p_region->p_pixbuf;
1414 p_dst = p_spu_region->picture.Y_PIXELS;
1415 i_pitch = p_spu_region->picture.Y_PITCH;
1417 /* Copy pixel buffer */
1418 for( j = 0; j < p_region->i_height; j++ )
1420 memcpy( p_dst, p_src, p_region->i_width );
1421 p_src += p_region->i_width;
1425 /* Check subtitles encoded as strings of characters
1426 * (since there are not rendered in the pixbuffer) */
1427 for( j = 0; j < p_region->i_object_defs; j++ )
1429 dvbsub_objectdef_t *p_object_def = &p_region->p_object_defs[j];
1431 if( p_object_def->i_type != 1 || !p_object_def->psz_text )
1434 /* Create new SPU region */
1435 memset( &fmt, 0, sizeof(video_format_t) );
1436 fmt.i_chroma = VLC_FOURCC('T','E','X','T');
1437 fmt.i_aspect = VOUT_ASPECT_FACTOR;
1438 fmt.i_width = fmt.i_visible_width = p_region->i_width;
1439 fmt.i_height = fmt.i_visible_height = p_region->i_height;
1440 fmt.i_x_offset = fmt.i_y_offset = 0;
1441 p_spu_region = p_spu->pf_create_region( VLC_OBJECT(p_dec), &fmt );
1444 msg_Err( p_dec, "cannot allocate SPU region" );
1448 p_spu_region->psz_text = strdup( p_object_def->psz_text );
1449 p_spu_region->i_x = p_regiondef->i_x + p_object_def->i_x;
1450 p_spu_region->i_y = p_regiondef->i_y + p_object_def->i_y;
1451 *pp_spu_region = p_spu_region;
1452 pp_spu_region = &p_spu_region->p_next;
1456 /* Set the pf_render callback */
1457 p_spu->i_start = p_sys->i_pts;
1458 p_spu->i_stop = p_spu->i_start + (mtime_t) (i_timeout * 1000000);
1459 p_spu->b_ephemer = VLC_TRUE;
1460 p_spu->b_fade = VLC_TRUE;
1462 /* Correct positioning of SPU */
1463 p_spu->b_absolute = p_sys->b_absolute;
1464 p_spu->i_flags = p_sys->i_spu_position;
1465 p_spu->i_x = p_sys->i_spu_x;
1466 p_spu->i_y = p_sys->i_spu_y;
1471 /*****************************************************************************
1472 * encoder_sys_t : encoder descriptor
1473 *****************************************************************************/
1474 typedef struct encoder_region_t
1481 struct encoder_sys_t
1483 unsigned int i_page_ver;
1484 unsigned int i_region_ver;
1485 unsigned int i_clut_ver;
1488 encoder_region_t *p_regions;
1492 /* subpicture positioning */
1497 static void encode_page_composition( encoder_t *, bs_t *, subpicture_t * );
1498 static void encode_clut( encoder_t *, bs_t *, subpicture_t * );
1499 static void encode_region_composition( encoder_t *, bs_t *, subpicture_t * );
1500 static void encode_object( encoder_t *, bs_t *, subpicture_t * );
1502 /*****************************************************************************
1503 * OpenEncoder: probe the encoder and return score
1504 *****************************************************************************/
1505 static int OpenEncoder( vlc_object_t *p_this )
1507 encoder_t *p_enc = (encoder_t *)p_this;
1508 encoder_sys_t *p_sys;
1511 if( p_enc->fmt_out.i_codec != VLC_FOURCC('d','v','b','s') &&
1514 return VLC_EGENERIC;
1517 /* Allocate the memory needed to store the decoder's structure */
1518 if( ( p_sys = (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
1520 msg_Err( p_enc, "out of memory" );
1523 p_enc->p_sys = p_sys;
1525 p_enc->pf_encode_sub = Encode;
1526 p_enc->fmt_out.i_codec = VLC_FOURCC('d','v','b','s');
1527 p_enc->fmt_out.subs.dvb.i_id = 1 << 16 | 1;
1529 sout_CfgParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
1531 p_sys->i_page_ver = 0;
1532 p_sys->i_region_ver = 0;
1533 p_sys->i_clut_ver = 0;
1534 p_sys->i_regions = 0;
1535 p_sys->p_regions = 0;
1537 var_Create( p_this, ENC_CFG_PREFIX "x", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
1538 var_Get( p_this, ENC_CFG_PREFIX "x", &val );
1539 p_sys->i_offset_x = val.i_int;
1540 var_Create( p_this, ENC_CFG_PREFIX "y", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
1541 var_Get( p_this, ENC_CFG_PREFIX "y", &val );
1542 p_sys->i_offset_y = val.i_int;
1547 /* FIXME: this routine is a hack to convert VLC_FOURCC('Y','U','V','A')
1548 * into VLC_FOURCC('Y','U','V','P')
1550 static subpicture_t *YuvaYuvp( encoder_t *p_enc, subpicture_t *p_subpic )
1552 subpicture_region_t *p_region = NULL;
1554 if( !p_subpic ) return NULL;
1556 for( p_region = p_subpic->p_region; p_region; p_region = p_region->p_next )
1558 video_format_t *p_fmt = &p_region->fmt;
1559 int i = 0, j = 0, n = 0, p = 0;
1560 int i_max_entries = 256;
1562 #ifdef RANDOM_DITHERING
1563 int i_seed = 0xdeadbeef; /* random seed */
1567 int i_pixels = p_region->picture.p[0].i_visible_lines
1568 * p_region->picture.p[0].i_pitch;
1569 int i_iterator = p_region->picture.p[0].i_visible_lines * 3 / 4
1570 * p_region->picture.p[0].i_pitch
1571 + p_region->picture.p[0].i_pitch * 1 / 3;
1572 int i_tolerance = 0;
1575 msg_Dbg( p_enc, "YuvaYuvp: i_pixels=%d, i_iterator=%d", i_pixels, i_iterator );
1577 p_fmt->i_chroma = VLC_FOURCC('Y','U','V','P');
1578 p_fmt->p_palette = (video_palette_t *) malloc( sizeof( video_palette_t ) );
1579 if( !p_fmt->p_palette ) break;
1580 p_fmt->p_palette->i_entries = 0;
1582 /* Find best iterator using Euclide’s algorithm */
1583 for( ; i_iterator > 1 ; i_iterator-- )
1602 /* Count colors, build best palette */
1603 for( i_tolerance = 0; i_tolerance < 128; i_tolerance++ )
1605 vlc_bool_t b_success = VLC_TRUE;
1606 p_fmt->p_palette->i_entries = 0;
1608 for( i = 0; i < i_pixels ; )
1611 y = p_region->picture.p[0].p_pixels[i];
1612 u = p_region->picture.p[1].p_pixels[i];
1613 v = p_region->picture.p[2].p_pixels[i];
1614 a = p_region->picture.p[3].p_pixels[i];
1615 for( j = 0; j < p_fmt->p_palette->i_entries; j++ )
1617 if( abs((int)p_fmt->p_palette->palette[j][0] - (int)y) <= i_tolerance &&
1618 abs((int)p_fmt->p_palette->palette[j][1] - (int)u) <= i_tolerance &&
1619 abs((int)p_fmt->p_palette->palette[j][2] - (int)v) <= i_tolerance &&
1620 abs((int)p_fmt->p_palette->palette[j][3] - (int)a) <= i_tolerance / 2 )
1625 if( j == p_fmt->p_palette->i_entries )
1627 p_fmt->p_palette->palette[j][0] = y;
1628 p_fmt->p_palette->palette[j][1] = u;
1629 p_fmt->p_palette->palette[j][2] = v;
1630 p_fmt->p_palette->palette[j][3] = a;
1631 p_fmt->p_palette->i_entries++;
1633 if( p_fmt->p_palette->i_entries >= i_max_entries )
1635 b_success = VLC_FALSE;
1652 msg_Dbg( p_enc, "best palette has %d colors", p_fmt->p_palette->i_entries );
1655 #ifndef RANDOM_DITHERING
1656 pi_delta = malloc( ( p_region->picture.p[0].i_pitch + 1 )
1657 * sizeof(int) * 4 );
1658 for( i = 0; i < (p_region->picture.p[0].i_pitch + 1) * 4 ; i++ )
1664 /* Fill image with our new colours */
1665 for( p = 0; p < p_region->picture.p[0].i_visible_lines ; p++ )
1667 int i_ydelta = 0, i_udelta = 0, i_vdelta = 0, i_adelta = 0;
1669 for( n = 0; n < p_region->picture.p[0].i_pitch ; n++ )
1671 int i_offset = p * p_region->picture.p[0].i_pitch + n;
1673 int i_mindist, i_best;
1675 y = (int)p_region->picture.p[0].p_pixels[i_offset];
1676 u = (int)p_region->picture.p[1].p_pixels[i_offset];
1677 v = (int)p_region->picture.p[2].p_pixels[i_offset];
1678 a = (int)p_region->picture.p[3].p_pixels[i_offset];
1680 /* Add dithering compensation */
1681 #ifdef RANDOM_DITHERING
1682 y += ((i_seed & 0xff) - 0x80) * i_tolerance / 0x80;
1683 u += (((i_seed >> 8) & 0xff) - 0x80) * i_tolerance / 0x80;
1684 v += (((i_seed >> 16) & 0xff) - 0x80) * i_tolerance / 0x80;
1685 a += (((i_seed >> 24) & 0xff) - 0x80) * i_tolerance / 0x80;
1687 y += i_ydelta + pi_delta[ n * 4 ];
1688 u += i_udelta + pi_delta[ n * 4 + 1 ];
1689 v += i_vdelta + pi_delta[ n * 4 + 2 ];
1690 a += i_adelta + pi_delta[ n * 4 + 3 ];
1693 /* Find best colour in palette */
1694 for( i_mindist = 99999999, i_best = 0, j = 0; j < p_fmt->p_palette->i_entries; j++ )
1698 i_dist += abs((int)p_fmt->p_palette->palette[j][0] - y);
1699 i_dist += abs((int)p_fmt->p_palette->palette[j][1] - u);
1700 i_dist += abs((int)p_fmt->p_palette->palette[j][2] - v);
1701 i_dist += 2 * abs((int)p_fmt->p_palette->palette[j][3] - a);
1703 if( i_dist < i_mindist )
1710 /* Set pixel to best color */
1711 p_region->picture.p[0].p_pixels[i_offset] = i_best;
1713 /* Update dithering state */
1714 #ifdef RANDOM_DITHERING
1715 i_seed = (i_seed * 0x1283837) ^ 0x789479 ^ (i_seed >> 13);
1717 i_ydelta = y - (int)p_fmt->p_palette->palette[i_best][0];
1718 i_udelta = u - (int)p_fmt->p_palette->palette[i_best][1];
1719 i_vdelta = v - (int)p_fmt->p_palette->palette[i_best][2];
1720 i_adelta = a - (int)p_fmt->p_palette->palette[i_best][3];
1721 pi_delta[ n * 4 ] = i_ydelta * 3 / 8;
1722 pi_delta[ n * 4 + 1 ] = i_udelta * 3 / 8;
1723 pi_delta[ n * 4 + 2 ] = i_vdelta * 3 / 8;
1724 pi_delta[ n * 4 + 3 ] = i_adelta * 3 / 8;
1725 i_ydelta = i_ydelta * 5 / 8;
1726 i_udelta = i_udelta * 5 / 8;
1727 i_vdelta = i_vdelta * 5 / 8;
1728 i_adelta = i_adelta * 5 / 8;
1732 #ifndef RANDOM_DITHERING
1737 for( i = p_fmt->p_palette->i_entries; i < i_max_entries; i++ )
1739 p_fmt->p_palette->palette[i][0] = 0;
1740 p_fmt->p_palette->palette[i][1] = 0;
1741 p_fmt->p_palette->palette[i][2] = 0;
1742 p_fmt->p_palette->palette[i][3] = 0;
1744 p_fmt->p_palette->i_entries = i_max_entries;
1746 msg_Dbg( p_enc, "best palette has %d colors", p_fmt->p_palette->i_entries );
1752 /****************************************************************************
1753 * Encode: the whole thing
1754 ****************************************************************************/
1755 static block_t *Encode( encoder_t *p_enc, subpicture_t *p_subpic )
1757 subpicture_t *p_temp = NULL;
1758 subpicture_region_t *p_region = NULL;
1759 bs_t bits, *s = &bits;
1762 if( !p_subpic || !p_subpic->p_region ) return NULL;
1764 /* FIXME: this is a hack to convert VLC_FOURCC('Y','U','V','A') into
1765 * VLC_FOURCC('Y','U','V','P')
1767 p_region = p_subpic->p_region;
1768 if( p_region->fmt.i_chroma == VLC_FOURCC('Y','U','V','A') )
1770 p_temp = YuvaYuvp( p_enc, p_subpic );
1773 msg_Err( p_enc, "no picture in subpicture" );
1776 p_region = p_subpic->p_region;
1780 if( !p_region ) return NULL;
1782 if( p_region->fmt.i_chroma != VLC_FOURCC('T','E','X','T') &&
1783 p_region->fmt.i_chroma != VLC_FOURCC('Y','U','V','P') ) return NULL;
1785 if( p_region->fmt.p_palette )
1787 switch( p_region->fmt.p_palette->i_entries )
1795 msg_Err( p_enc, "subpicture palette (%d) not handled",
1796 p_region->fmt.p_palette->i_entries );
1803 msg_Dbg( p_enc, "encoding subpicture" );
1805 p_block = block_New( p_enc, 64000 );
1806 bs_init( s, p_block->p_buffer, p_block->i_buffer );
1808 bs_write( s, 8, 0x20 ); /* Data identifier */
1809 bs_write( s, 8, 0x0 ); /* Subtitle stream id */
1811 encode_page_composition( p_enc, s, p_subpic );
1812 encode_region_composition( p_enc, s, p_subpic );
1813 encode_clut( p_enc, s, p_subpic );
1814 encode_object( p_enc, s, p_subpic );
1816 /* End of display */
1817 bs_write( s, 8, 0x0f ); /* Sync byte */
1818 bs_write( s, 8, DVBSUB_ST_ENDOFDISPLAY ); /* Segment type */
1819 bs_write( s, 16, 1 ); /* Page id */
1820 bs_write( s, 16, 0 ); /* Segment length */
1822 bs_write( s, 8, 0xff );/* End marker */
1823 p_block->i_buffer = bs_pos( s ) / 8;
1824 p_block->i_pts = p_block->i_dts = p_subpic->i_start;
1825 if( !p_subpic->b_ephemer && p_subpic->i_stop > p_subpic->i_start )
1827 block_t *p_block_stop;
1829 p_block->i_length = p_subpic->i_stop - p_subpic->i_start;
1831 /* Send another (empty) subtitle to signal the end of display */
1832 p_block_stop = block_New( p_enc, 64000 );
1833 bs_init( s, p_block_stop->p_buffer, p_block_stop->i_buffer );
1834 bs_write( s, 8, 0x20 ); /* Data identifier */
1835 bs_write( s, 8, 0x0 ); /* Subtitle stream id */
1836 encode_page_composition( p_enc, s, 0 );
1837 bs_write( s, 8, 0x0f ); /* Sync byte */
1838 bs_write( s, 8, DVBSUB_ST_ENDOFDISPLAY ); /* Segment type */
1839 bs_write( s, 16, 1 ); /* Page id */
1840 bs_write( s, 16, 0 ); /* Segment length */
1841 bs_write( s, 8, 0xff );/* End marker */
1842 p_block_stop->i_buffer = bs_pos( s ) / 8;
1843 p_block_stop->i_pts = p_block_stop->i_dts = p_subpic->i_stop;
1844 block_ChainAppend( &p_block, p_block_stop );
1845 p_block_stop->i_length = 100000; /* p_subpic->i_stop - p_subpic->i_start; */
1848 msg_Dbg( p_enc, "subpicture encoded properly" );
1853 /*****************************************************************************
1854 * CloseEncoder: encoder destruction
1855 *****************************************************************************/
1856 static void CloseEncoder( vlc_object_t *p_this )
1858 encoder_t *p_enc = (encoder_t *)p_this;
1859 encoder_sys_t *p_sys = p_enc->p_sys;
1861 var_Destroy( p_this , ENC_CFG_PREFIX "x" );
1862 var_Destroy( p_this , ENC_CFG_PREFIX "y" );
1863 var_Destroy( p_this , ENC_CFG_PREFIX "timeout" );
1865 if( p_sys->i_regions ) free( p_sys->p_regions );
1869 static void encode_page_composition( encoder_t *p_enc, bs_t *s,
1870 subpicture_t *p_subpic )
1872 encoder_sys_t *p_sys = p_enc->p_sys;
1873 subpicture_region_t *p_region;
1874 vlc_bool_t b_mode_change = VLC_FALSE;
1875 int i_regions, i_timeout;
1877 bs_write( s, 8, 0x0f ); /* Sync byte */
1878 bs_write( s, 8, DVBSUB_ST_PAGE_COMPOSITION ); /* Segment type */
1879 bs_write( s, 16, 1 ); /* Page id */
1881 for( i_regions = 0, p_region = p_subpic ? p_subpic->p_region : 0;
1882 p_region; p_region = p_region->p_next, i_regions++ )
1884 if( i_regions >= p_sys->i_regions )
1886 encoder_region_t region;
1887 region.i_width = region.i_height = 0;
1889 realloc( p_sys->p_regions, sizeof(encoder_region_t) *
1890 (p_sys->i_regions + 1) );
1891 p_sys->p_regions[p_sys->i_regions++] = region;
1894 if( ( p_sys->p_regions[i_regions].i_width <
1895 (int)p_region->fmt.i_visible_width ) ||
1896 ( p_sys->p_regions[i_regions].i_width >
1897 (int)p_region->fmt.i_visible_width ) )
1899 b_mode_change = VLC_TRUE;
1900 msg_Dbg( p_enc, "region %i width change: %i -> %i",
1901 i_regions, p_sys->p_regions[i_regions].i_width,
1902 p_region->fmt.i_visible_width );
1903 p_sys->p_regions[i_regions].i_width =
1904 p_region->fmt.i_visible_width;
1906 if( p_sys->p_regions[i_regions].i_height <
1907 (int)p_region->fmt.i_visible_height )
1909 b_mode_change = VLC_TRUE;
1910 msg_Dbg( p_enc, "region %i height change: %i -> %i",
1911 i_regions, p_sys->p_regions[i_regions].i_height,
1912 p_region->fmt.i_visible_height );
1913 p_sys->p_regions[i_regions].i_height =
1914 p_region->fmt.i_visible_height;
1918 bs_write( s, 16, i_regions * 6 + 2 ); /* Segment length */
1921 if( p_subpic && !p_subpic->b_ephemer &&
1922 p_subpic->i_stop > p_subpic->i_start )
1924 i_timeout = (p_subpic->i_stop - p_subpic->i_start) / 1000000;
1927 bs_write( s, 8, i_timeout ); /* Timeout */
1928 bs_write( s, 4, p_sys->i_page_ver++ );
1929 bs_write( s, 2, b_mode_change ?
1930 DVBSUB_PCS_STATE_CHANGE : DVBSUB_PCS_STATE_ACQUISITION );
1931 bs_write( s, 2, 0 ); /* Reserved */
1933 for( i_regions = 0, p_region = p_subpic ? p_subpic->p_region : 0;
1934 p_region; p_region = p_region->p_next, i_regions++ )
1936 bs_write( s, 8, i_regions );
1937 bs_write( s, 8, 0 ); /* Reserved */
1938 if( (p_sys->i_offset_x > 0) && (p_sys->i_offset_y > 0) )
1940 bs_write( s, 16, p_sys->i_offset_x ); /* override x position */
1941 bs_write( s, 16, p_sys->i_offset_y ); /* override y position */
1945 bs_write( s, 16, p_region->i_x );
1946 bs_write( s, 16, p_region->i_y );
1951 static void encode_clut( encoder_t *p_enc, bs_t *s, subpicture_t *p_subpic )
1953 encoder_sys_t *p_sys = p_enc->p_sys;
1954 subpicture_region_t *p_region = p_subpic->p_region;
1955 video_palette_t *p_pal, pal;
1959 if( !p_region ) return;
1961 if( p_region->fmt.i_chroma == VLC_FOURCC('Y','U','V','P') )
1963 p_pal = p_region->fmt.p_palette;
1968 for( i = 0; i < 4; i++ )
1970 pal.palette[i][0] = 0;
1971 pal.palette[i][1] = 0;
1972 pal.palette[i][2] = 0;
1973 pal.palette[i][3] = 0;
1978 bs_write( s, 8, 0x0f ); /* Sync byte */
1979 bs_write( s, 8, DVBSUB_ST_CLUT_DEFINITION ); /* Segment type */
1980 bs_write( s, 16, 1 ); /* Page id */
1982 bs_write( s, 16, p_pal->i_entries * 6 + 2 ); /* Segment length */
1983 bs_write( s, 8, 1 ); /* Clut id */
1984 bs_write( s, 4, p_sys->i_clut_ver++ );
1985 bs_write( s, 4, 0 ); /* Reserved */
1987 for( i = 0; i < p_pal->i_entries; i++ )
1989 bs_write( s, 8, i ); /* Clut entry id */
1990 bs_write( s, 1, p_pal->i_entries == 4 ); /* 2bit/entry flag */
1991 bs_write( s, 1, p_pal->i_entries == 16 ); /* 4bit/entry flag */
1992 bs_write( s, 1, p_pal->i_entries == 256 ); /* 8bit/entry flag */
1993 bs_write( s, 4, 0 ); /* Reserved */
1994 bs_write( s, 1, 1 ); /* Full range flag */
1995 bs_write( s, 8, p_pal->palette[i][3] ? /* Y value */
1996 (p_pal->palette[i][0] ? p_pal->palette[i][0] : 16) : 0 );
1997 bs_write( s, 8, p_pal->palette[i][1] ); /* Cr value */
1998 bs_write( s, 8, p_pal->palette[i][2] ); /* Cb value */
1999 bs_write( s, 8, 0xff - p_pal->palette[i][3] ); /* T value */
2003 static void encode_region_composition( encoder_t *p_enc, bs_t *s,
2004 subpicture_t *p_subpic )
2006 encoder_sys_t *p_sys = p_enc->p_sys;
2007 subpicture_region_t *p_region;
2010 for( i_region = 0, p_region = p_subpic->p_region; p_region;
2011 p_region = p_region->p_next, i_region++ )
2013 int i_entries = 4, i_depth = 0x1, i_bg = 0;
2015 p_region->fmt.i_chroma == VLC_FOURCC('T','E','X','T');
2019 video_palette_t *p_pal = p_region->fmt.p_palette;
2023 msg_Err( p_enc, "subpicture has no palette - ignoring it" );
2027 i_entries = p_pal->i_entries;
2028 i_depth = i_entries == 4 ? 0x1 : i_entries == 16 ? 0x2 : 0x3;
2030 for( i_bg = 0; i_bg < p_pal->i_entries; i_bg++ )
2032 if( !p_pal->palette[i_bg][3] ) break;
2036 bs_write( s, 8, 0x0f ); /* Sync byte */
2037 bs_write( s, 8, DVBSUB_ST_REGION_COMPOSITION ); /* Segment type */
2038 bs_write( s, 16, 1 ); /* Page id */
2040 bs_write( s, 16, 10 + 6 + (b_text ? 2 : 0) ); /* Segment length */
2041 bs_write( s, 8, i_region );
2042 bs_write( s, 4, p_sys->i_region_ver++ );
2044 /* Region attributes */
2045 bs_write( s, 1, i_bg < i_entries ); /* Fill */
2046 bs_write( s, 3, 0 ); /* Reserved */
2047 bs_write( s, 16, p_sys->p_regions[i_region].i_width );
2048 bs_write( s, 16, p_sys->p_regions[i_region].i_height );
2049 bs_write( s, 3, i_depth ); /* Region level of compatibility */
2050 bs_write( s, 3, i_depth ); /* Region depth */
2051 bs_write( s, 2, 0 ); /* Reserved */
2052 bs_write( s, 8, 1 ); /* Clut id */
2053 bs_write( s, 8, i_bg ); /* region 8bit pixel code */
2054 bs_write( s, 4, i_bg ); /* region 4bit pixel code */
2055 bs_write( s, 2, i_bg ); /* region 2bit pixel code */
2056 bs_write( s, 2, 0 ); /* Reserved */
2058 /* In our implementation we only have 1 object per region */
2059 bs_write( s, 16, i_region );
2060 bs_write( s, 2, b_text ? DVBSUB_OT_BASIC_CHAR:DVBSUB_OT_BASIC_BITMAP );
2061 bs_write( s, 2, 0 ); /* object provider flag */
2062 bs_write( s, 12, 0 );/* object horizontal position */
2063 bs_write( s, 4, 0 ); /* Reserved */
2064 bs_write( s, 12, 0 );/* object vertical position */
2068 bs_write( s, 8, 1 ); /* foreground pixel code */
2069 bs_write( s, 8, 0 ); /* background pixel code */
2074 static void encode_pixel_data( encoder_t *p_enc, bs_t *s,
2075 subpicture_region_t *p_region,
2078 static void encode_object( encoder_t *p_enc, bs_t *s, subpicture_t *p_subpic )
2080 encoder_sys_t *p_sys = p_enc->p_sys;
2081 subpicture_region_t *p_region;
2084 int i_length_pos, i_update_pos, i_pixel_data_pos;
2086 for( i_region = 0, p_region = p_subpic->p_region; p_region;
2087 p_region = p_region->p_next, i_region++ )
2089 bs_write( s, 8, 0x0f ); /* Sync byte */
2090 bs_write( s, 8, DVBSUB_ST_OBJECT_DATA ); /* Segment type */
2091 bs_write( s, 16, 1 ); /* Page id */
2093 i_length_pos = bs_pos( s );
2094 bs_write( s, 16, 0 ); /* Segment length */
2095 bs_write( s, 16, i_region ); /* Object id */
2096 bs_write( s, 4, p_sys->i_region_ver++ );
2098 /* object coding method */
2099 switch( p_region->fmt.i_chroma )
2101 case VLC_FOURCC( 'Y','U','V','P' ):
2102 bs_write( s, 2, 0 );
2104 case VLC_FOURCC( 'T','E','X','T' ):
2105 bs_write( s, 2, 1 );
2108 msg_Err( p_enc, "FOURCC %d not supported by encoder.", p_region->fmt.i_chroma );
2112 bs_write( s, 1, 0 ); /* non modifying color flag */
2113 bs_write( s, 1, 0 ); /* Reserved */
2115 if( p_region->fmt.i_chroma == VLC_FOURCC( 'T','E','X','T' ) )
2119 if( !p_region->psz_text ) continue;
2121 i_size = __MIN( strlen( p_region->psz_text ), 256 );
2123 bs_write( s, 8, i_size ); /* number of characters in string */
2124 for( i = 0; i < i_size; i++ )
2126 bs_write( s, 16, p_region->psz_text[i] );
2129 /* Update segment length */
2130 SetWBE( &s->p_start[i_length_pos/8],
2131 (bs_pos(s) - i_length_pos)/8 -2 );
2135 /* Coding of a bitmap object */
2136 i_update_pos = bs_pos( s );
2137 bs_write( s, 16, 0 ); /* topfield data block length */
2138 bs_write( s, 16, 0 ); /* bottomfield data block length */
2141 i_pixel_data_pos = bs_pos( s );
2142 encode_pixel_data( p_enc, s, p_region, VLC_TRUE );
2143 i_pixel_data_pos = ( bs_pos( s ) - i_pixel_data_pos ) / 8;
2144 SetWBE( &s->p_start[i_update_pos/8], i_pixel_data_pos );
2147 i_pixel_data_pos = bs_pos( s );
2148 encode_pixel_data( p_enc, s, p_region, VLC_FALSE );
2149 i_pixel_data_pos = ( bs_pos( s ) - i_pixel_data_pos ) / 8;
2150 SetWBE( &s->p_start[i_update_pos/8+2], i_pixel_data_pos );
2152 /* Stuffing for word alignment */
2154 if( bs_pos( s ) % 16 ) bs_write( s, 8, 0 );
2156 /* Update segment length */
2157 SetWBE( &s->p_start[i_length_pos/8], (bs_pos(s) - i_length_pos)/8 -2 );
2161 static void encode_pixel_line_2bp( encoder_t *p_enc, bs_t *s,
2162 subpicture_region_t *p_region,
2164 static void encode_pixel_line_4bp( encoder_t *p_enc, bs_t *s,
2165 subpicture_region_t *p_region,
2167 static void encode_pixel_line_8bp( encoder_t *p_enc, bs_t *s,
2168 subpicture_region_t *p_region,
2170 static void encode_pixel_data( encoder_t *p_enc, bs_t *s,
2171 subpicture_region_t *p_region,
2174 unsigned int i_line;
2177 if( p_region->fmt.i_chroma != VLC_FOURCC('Y','U','V','P') ) return;
2179 /* Encode line by line */
2180 for( i_line = !b_top; i_line < p_region->fmt.i_visible_height;
2183 switch( p_region->fmt.p_palette->i_entries )
2189 bs_write( s, 8, 0x10 ); /* 2 bit/pixel code string */
2190 encode_pixel_line_2bp( p_enc, s, p_region, i_line );
2194 bs_write( s, 8, 0x11 ); /* 4 bit/pixel code string */
2195 encode_pixel_line_4bp( p_enc, s, p_region, i_line );
2199 bs_write( s, 8, 0x12 ); /* 8 bit/pixel code string */
2200 encode_pixel_line_8bp( p_enc, s, p_region, i_line );
2204 msg_Err( p_enc, "subpicture palette (%i) not handled",
2205 p_region->fmt.p_palette->i_entries );
2209 bs_write( s, 8, 0xf0 ); /* End of object line code */
2213 static void encode_pixel_line_2bp( encoder_t *p_enc, bs_t *s,
2214 subpicture_region_t *p_region,
2217 unsigned int i, i_length = 0;
2218 int i_pitch = p_region->picture.p->i_pitch;
2219 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2220 int i_last_pixel = p_data[0];
2222 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2224 if( i != p_region->fmt.i_visible_width &&
2225 p_data[i] == i_last_pixel && i_length != 284 )
2231 if( i_length == 1 || i_length == 11 || i_length == 28 )
2233 /* 2bit/pixel code */
2234 if( i_last_pixel ) bs_write( s, 2, i_last_pixel );
2237 bs_write( s, 2, 0 );
2238 bs_write( s, 1, 0 );
2239 bs_write( s, 1, 1 ); /* pseudo color 0 */
2248 bs_write( s, 2, i_last_pixel );
2249 bs_write( s, 2, i_last_pixel );
2253 bs_write( s, 2, 0 );
2254 bs_write( s, 1, 0 );
2255 bs_write( s, 1, 0 );
2256 bs_write( s, 2, 1 ); /* 2 * pseudo color 0 */
2259 else if( i_length > 2 )
2261 bs_write( s, 2, 0 );
2262 if( i_length <= 10 )
2264 bs_write( s, 1, 1 );
2265 bs_write( s, 3, i_length - 3 );
2266 bs_write( s, 2, i_last_pixel );
2270 bs_write( s, 1, 0 );
2271 bs_write( s, 1, 0 );
2273 if( i_length <= 27 )
2275 bs_write( s, 2, 2 );
2276 bs_write( s, 4, i_length - 12 );
2277 bs_write( s, 2, i_last_pixel );
2281 bs_write( s, 2, 3 );
2282 bs_write( s, 8, i_length - 29 );
2283 bs_write( s, 2, i_last_pixel );
2288 if( i == p_region->fmt.i_visible_width ) break;
2290 i_last_pixel = p_data[i];
2295 bs_write( s, 2, 0 );
2296 bs_write( s, 1, 0 );
2297 bs_write( s, 1, 0 );
2298 bs_write( s, 2, 0 );
2304 static void encode_pixel_line_4bp( encoder_t *p_enc, bs_t *s,
2305 subpicture_region_t *p_region,
2308 unsigned int i, i_length = 0;
2309 int i_pitch = p_region->picture.p->i_pitch;
2310 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2311 int i_last_pixel = p_data[0];
2313 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2315 if( i != p_region->fmt.i_visible_width &&
2316 p_data[i] == i_last_pixel && i_length != 280 )
2322 if( i_length == 1 || (i_length == 3 && i_last_pixel) || i_length == 8 )
2324 /* 4bit/pixel code */
2325 if( i_last_pixel ) bs_write( s, 4, i_last_pixel );
2328 bs_write( s, 4, 0 );
2329 bs_write( s, 1, 1 );
2330 bs_write( s, 1, 1 );
2331 bs_write( s, 2, 0 ); /* pseudo color 0 */
2340 bs_write( s, 4, i_last_pixel );
2341 bs_write( s, 4, i_last_pixel );
2345 bs_write( s, 4, 0 );
2346 bs_write( s, 1, 1 );
2347 bs_write( s, 1, 1 );
2348 bs_write( s, 2, 1 ); /* 2 * pseudo color 0 */
2351 else if( !i_last_pixel && i_length >= 3 && i_length <= 9 )
2353 bs_write( s, 4, 0 );
2354 bs_write( s, 1, 0 );
2355 bs_write( s, 3, i_length - 2 ); /* (i_length - 2) * color 0 */
2357 else if( i_length > 2 )
2359 bs_write( s, 4, 0 );
2360 bs_write( s, 1, 1 );
2364 bs_write( s, 1, 0 );
2365 bs_write( s, 2, i_length - 4 );
2366 bs_write( s, 4, i_last_pixel );
2370 bs_write( s, 1, 1 );
2372 if( i_length <= 24 )
2374 bs_write( s, 2, 2 );
2375 bs_write( s, 4, i_length - 9 );
2376 bs_write( s, 4, i_last_pixel );
2380 bs_write( s, 2, 3 );
2381 bs_write( s, 8, i_length - 25 );
2382 bs_write( s, 4, i_last_pixel );
2387 if( i == p_region->fmt.i_visible_width ) break;
2389 i_last_pixel = p_data[i];
2394 bs_write( s, 8, 0 );
2400 static void encode_pixel_line_8bp( encoder_t *p_enc, bs_t *s,
2401 subpicture_region_t *p_region,
2404 unsigned int i, i_length = 0;
2405 int i_pitch = p_region->picture.p->i_pitch;
2406 uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2407 int i_last_pixel = p_data[0];
2409 for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2411 if( i != p_region->fmt.i_visible_width &&
2412 p_data[i] == i_last_pixel && i_length != 127 )
2418 if( i_length == 1 && i_last_pixel )
2420 /* 8bit/pixel code */
2421 bs_write( s, 8, i_last_pixel );
2423 else if( i_length == 2 && i_last_pixel )
2425 /* 8bit/pixel code */
2426 bs_write( s, 8, i_last_pixel );
2427 bs_write( s, 8, i_last_pixel );
2429 else if( i_length <= 127 )
2431 bs_write( s, 8, 0 );
2435 bs_write( s, 1, 0 );
2436 bs_write( s, 7, i_length ); /* pseudo color 0 */
2440 bs_write( s, 1, 1 );
2441 bs_write( s, 7, i_length );
2442 bs_write( s, 8, i_last_pixel );
2446 if( i == p_region->fmt.i_visible_width ) break;
2448 i_last_pixel = p_data[i];
2453 bs_write( s, 8, 0 );
2454 bs_write( s, 8, 0 );