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

   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
   7  * $Id$
   8  *
   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>
  14  *
  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.
  19  *
  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.
  24  *
  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 /*****************************************************************************
  30  * Preamble
  31  *
  32  * FIXME:
  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.
  36  *
  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  *****************************************************************************/
  42 #ifdef HAVE_CONFIG_H
  43 # include "config.h"
  44 #endif
  45
  46 #include <vlc/vlc.h>
  47 #include <vlc_vout.h>
  48 #include <vlc_codec.h>
  49 #include <vlc_sout.h>
  50
  51 #include "vlc_bits.h"
  52
  53 /* #define DEBUG_DVBSUB 1 */
  54
  55 #define POSX_TEXT N_("Decoding X coordinate")
  56 #define POSX_LONGTEXT N_("X coordinate of the rendered subtitle")
  57
  58 #define POSY_TEXT N_("Decoding Y coordinate")
  59 #define POSY_LONGTEXT N_("Y coordinate of the rendered subtitle")
  60
  61 #define POS_TEXT N_("Subpicture position")
  62 #define POS_LONGTEXT N_( \
  63   "You can enforce the subpicture position on the video " \
  64   "(0=center, 1=left, 2=right, 4=top, 8=bottom, you can " \
  65   "also use combinations of these values, e.g. 6=top-right).")
  66
  67 #define ENC_POSX_TEXT N_("Encoding X coordinate")
  68 #define ENC_POSX_LONGTEXT N_("X coordinate of the encoded subtitle" )
  69 #define ENC_POSY_TEXT N_("Encoding Y coordinate")
  70 #define ENC_POSY_LONGTEXT N_("Y coordinate of the encoded subtitle" )
  71
  72 static int pi_pos_values[] = { 0, 1, 2, 4, 8, 5, 6, 9, 10 };
  73 static const char *ppsz_pos_descriptions[] =
  74 { N_("Center"), N_("Left"), N_("Right"), N_("Top"), N_("Bottom"),
  75   N_("Top-Left"), N_("Top-Right"), N_("Bottom-Left"), N_("Bottom-Right") };
  76
  77 /*****************************************************************************
  78  * Module descriptor.
  79  *****************************************************************************/
  80 static int  Open ( vlc_object_t * );
  81 static void Close( vlc_object_t * );
  82 static subpicture_t *Decode( decoder_t *, block_t ** );
  83
  84 static int OpenEncoder  ( vlc_object_t * );
  85 static void CloseEncoder( vlc_object_t * );
  86 static block_t *Encode  ( encoder_t *, subpicture_t * );
  87
  88 vlc_module_begin();
  89 #   define DVBSUB_CFG_PREFIX "dvbsub-"
  90     set_description( _("DVB subtitles decoder") );
  91     set_capability( "decoder", 50 );
  92     set_category( CAT_INPUT );
  93     set_subcategory( SUBCAT_INPUT_SCODEC );
  94     set_callbacks( Open, Close );
  95
  96     add_integer( DVBSUB_CFG_PREFIX "position", 8, NULL, POS_TEXT, POS_LONGTEXT, VLC_TRUE );
  97         change_integer_list( pi_pos_values, ppsz_pos_descriptions, 0 );
  98     add_integer( DVBSUB_CFG_PREFIX "x", -1, NULL, POSX_TEXT, POSX_LONGTEXT, VLC_FALSE );
  99     add_integer( DVBSUB_CFG_PREFIX "y", -1, NULL, POSY_TEXT, POSY_LONGTEXT, VLC_FALSE );
 100
 101 #   define ENC_CFG_PREFIX "sout-dvbsub-"
 102     add_submodule();
 103     set_description( _("DVB subtitles encoder") );
 104     set_capability( "encoder", 100 );
 105     set_callbacks( OpenEncoder, CloseEncoder );
 106
 107     add_integer( ENC_CFG_PREFIX "x", -1, NULL, ENC_POSX_TEXT, ENC_POSX_LONGTEXT, VLC_FALSE );
 108     add_integer( ENC_CFG_PREFIX "y", -1, NULL, ENC_POSY_TEXT, ENC_POSY_LONGTEXT, VLC_FALSE );
 109     add_obsolete_integer( ENC_CFG_PREFIX "timeout" ); /* Suppressed since 0.8.5 */
 110 vlc_module_end();
 111
 112 static const char *ppsz_enc_options[] = { "x", "y", NULL };
 113
 114 /****************************************************************************
 115  * Local structures
 116  ****************************************************************************
 117  * Those structures refer closely to the ETSI 300 743 Object model
 118  ****************************************************************************/
 119
 120 /* The object definition gives the position of the object in a region [7.2.5] */
 121 typedef struct dvbsub_objectdef_s
 122 {
 123     int i_id;
 124     int i_type;
 125     int i_x;
 126     int i_y;
 127     int i_fg_pc;
 128     int i_bg_pc;
 129     char *psz_text; /* for string of characters objects */
 130
 131 } dvbsub_objectdef_t;
 132
 133 /* The entry in the palette CLUT */
 134 typedef struct
 135 {
 136     uint8_t                 Y;
 137     uint8_t                 Cr;
 138     uint8_t                 Cb;
 139     uint8_t                 T;
 140
 141 } dvbsub_color_t;
 142
 143 /* The displays dimensions [7.2.1] */
 144 typedef struct dvbsub_display_s
 145 {
 146     uint8_t                 i_id;
 147     uint8_t                 i_version;
 148
 149     int                     i_width;
 150     int                     i_height;
 151
 152     vlc_bool_t              b_windowed;
 153     int                     i_x;
 154     int                     i_y;
 155     int                     i_max_x;
 156     int                     i_max_y;
 157
 158 } dvbsub_display_t;
 159
 160 /* [7.2.4] */
 161 typedef struct dvbsub_clut_s
 162 {
 163     uint8_t                 i_id;
 164     uint8_t                 i_version;
 165     dvbsub_color_t          c_2b[4];
 166     dvbsub_color_t          c_4b[16];
 167     dvbsub_color_t          c_8b[256];
 168
 169     struct dvbsub_clut_s    *p_next;
 170
 171 } dvbsub_clut_t;
 172
 173 /* The Region is an aera on the image [7.2.3]
 174  * with a list of the object definitions associated and a CLUT */
 175 typedef struct dvbsub_region_s
 176 {
 177     int i_id;
 178     int i_version;
 179     int i_x;
 180     int i_y;
 181     int i_width;
 182     int i_height;
 183     int i_level_comp;
 184     int i_depth;
 185     int i_clut;
 186
 187     uint8_t *p_pixbuf;
 188
 189     int                    i_object_defs;
 190     dvbsub_objectdef_t     *p_object_defs;
 191
 192     struct dvbsub_region_s *p_next;
 193
 194 } dvbsub_region_t;
 195
 196 /* The object definition gives the position of the object in a region */
 197 typedef struct dvbsub_regiondef_s
 198 {
 199     int i_id;
 200     int i_x;
 201     int i_y;
 202
 203 } dvbsub_regiondef_t;
 204
 205 /* The page defines the list of regions [7.2.2] */
 206 typedef struct
 207 {
 208     int i_id;
 209     int i_timeout; /* in seconds */
 210     int i_state;
 211     int i_version;
 212
 213     int                i_region_defs;
 214     dvbsub_regiondef_t *p_region_defs;
 215
 216 } dvbsub_page_t;
 217
 218 struct decoder_sys_t
 219 {
 220     bs_t            bs;
 221
 222     /* Decoder internal data */
 223     int             i_id;
 224     int             i_ancillary_id;
 225     mtime_t         i_pts;
 226
 227     vlc_bool_t      b_absolute;
 228     int             i_spu_position;
 229     int             i_spu_x;
 230     int             i_spu_y;
 231
 232     vlc_bool_t      b_page;
 233     dvbsub_page_t   *p_page;
 234     dvbsub_region_t *p_regions;
 235     dvbsub_clut_t   *p_cluts;
 236     dvbsub_display_t *p_display;
 237     dvbsub_clut_t    default_clut;
 238 };
 239
 240
 241 /* List of different SEGMENT TYPES */
 242 /* According to EN 300-743, table 2 */
 243 #define DVBSUB_ST_PAGE_COMPOSITION      0x10
 244 #define DVBSUB_ST_REGION_COMPOSITION    0x11
 245 #define DVBSUB_ST_CLUT_DEFINITION       0x12
 246 #define DVBSUB_ST_OBJECT_DATA           0x13
 247 #define DVBSUB_ST_DISPLAY_DEFINITION    0x14
 248 #define DVBSUB_ST_ENDOFDISPLAY          0x80
 249 #define DVBSUB_ST_STUFFING              0xff
 250 /* List of different OBJECT TYPES */
 251 /* According to EN 300-743, table 6 */
 252 #define DVBSUB_OT_BASIC_BITMAP          0x00
 253 #define DVBSUB_OT_BASIC_CHAR            0x01
 254 #define DVBSUB_OT_COMPOSITE_STRING      0x02
 255 /* Pixel DATA TYPES */
 256 /* According to EN 300-743, table 9 */
 257 #define DVBSUB_DT_2BP_CODE_STRING       0x10
 258 #define DVBSUB_DT_4BP_CODE_STRING       0x11
 259 #define DVBSUB_DT_8BP_CODE_STRING       0x12
 260 #define DVBSUB_DT_24_TABLE_DATA         0x20
 261 #define DVBSUB_DT_28_TABLE_DATA         0x21
 262 #define DVBSUB_DT_48_TABLE_DATA         0x22
 263 #define DVBSUB_DT_END_LINE              0xf0
 264 /* List of different Page Composition Segment state */
 265 /* According to EN 300-743, 7.2.1 table 3 */
 266 #define DVBSUB_PCS_STATE_ACQUISITION    0x01
 267 #define DVBSUB_PCS_STATE_CHANGE         0x02
 268
 269 /*****************************************************************************
 270  * Local prototypes
 271  *****************************************************************************/
 272 static void decode_segment( decoder_t *, bs_t * );
 273 static void decode_page_composition( decoder_t *, bs_t * );
 274 static void decode_region_composition( decoder_t *, bs_t * );
 275 static void decode_object( decoder_t *, bs_t * );
 276 static void decode_display_definition( decoder_t *, bs_t * );
 277 static void decode_clut( decoder_t *, bs_t * );
 278 static void free_all( decoder_t * );
 279
 280 static void default_clut_init( decoder_t * );
 281
 282 static subpicture_t *render( decoder_t * );
 283
 284 /*****************************************************************************
 285  * Open: probe the decoder and return score
 286  *****************************************************************************
 287  * Tries to launch a decoder and return score so that the interface is able
 288  * to chose.
 289  *****************************************************************************/
 290 static int Open( vlc_object_t *p_this )
 291 {
 292     decoder_t     *p_dec = (decoder_t *) p_this;
 293     decoder_sys_t *p_sys;
 294     vlc_value_t    val;
 295     int i_posx, i_posy;
 296
 297     if( p_dec->fmt_in.i_codec != VLC_FOURCC('d','v','b','s') )
 298     {
 299         return VLC_EGENERIC;
 300     }
 301
 302     p_dec->pf_decode_sub = Decode;
 303     p_sys = p_dec->p_sys = malloc( sizeof(decoder_sys_t) );
 304     if( !p_sys )
 305     {
 306         msg_Err( p_dec, "out of memory" );
 307         return VLC_ENOMEM;
 308     }
 309     memset( p_sys, 0, sizeof(decoder_sys_t) );
 310
 311     p_sys->i_pts          = (mtime_t) 0;
 312     p_sys->i_id           = p_dec->fmt_in.subs.dvb.i_id & 0xFFFF;
 313     p_sys->i_ancillary_id = p_dec->fmt_in.subs.dvb.i_id >> 16;
 314
 315     p_sys->p_regions      = NULL;
 316     p_sys->p_cluts        = NULL;
 317     p_sys->p_page         = NULL;
 318     p_sys->p_display      = NULL;
 319
 320     var_Create( p_this, DVBSUB_CFG_PREFIX "position",
 321                 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
 322     var_Get( p_this, DVBSUB_CFG_PREFIX "position", &val );
 323     p_sys->i_spu_position = val.i_int;
 324     var_Create( p_this, DVBSUB_CFG_PREFIX "x",
 325                 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
 326     var_Get( p_this, DVBSUB_CFG_PREFIX "x", &val );
 327     i_posx = val.i_int;
 328     var_Create( p_this, DVBSUB_CFG_PREFIX "y",
 329                 VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
 330     var_Get( p_this, DVBSUB_CFG_PREFIX "y", &val );
 331     i_posy = val.i_int;
 332
 333     /* Check if subpicture position was overridden */
 334     p_sys->b_absolute = VLC_FALSE;
 335     p_sys->i_spu_x = p_sys->i_spu_y = 0;
 336
 337     if( ( i_posx >= 0 ) && ( i_posy >= 0 ) )
 338     {
 339         p_sys->b_absolute = VLC_TRUE;
 340         p_sys->i_spu_x = i_posx;
 341         p_sys->i_spu_y = i_posy;
 342     }
 343
 344     es_format_Init( &p_dec->fmt_out, SPU_ES, VLC_FOURCC( 'd','v','b','s' ) );
 345
 346     default_clut_init( p_dec );
 347
 348     return VLC_SUCCESS;
 349 }
 350
 351 /*****************************************************************************
 352  * Close:
 353  *****************************************************************************/
 354 static void Close( vlc_object_t *p_this )
 355 {
 356     decoder_t     *p_dec = (decoder_t*) p_this;
 357     decoder_sys_t *p_sys = p_dec->p_sys;
 358
 359     var_Destroy( p_this, DVBSUB_CFG_PREFIX "x" );
 360     var_Destroy( p_this, DVBSUB_CFG_PREFIX "y" );
 361     var_Destroy( p_this, DVBSUB_CFG_PREFIX "position" );
 362
 363     free_all( p_dec );
 364     free( p_sys );
 365 }
 366
 367 /*****************************************************************************
 368  * Decode:
 369  *****************************************************************************/
 370 static subpicture_t *Decode( decoder_t *p_dec, block_t **pp_block )
 371 {
 372     decoder_sys_t *p_sys = p_dec->p_sys;
 373     block_t       *p_block;
 374     subpicture_t  *p_spu = NULL;
 375
 376     if( ( pp_block == NULL ) || ( *pp_block == NULL ) ) return NULL;
 377     p_block = *pp_block;
 378     *pp_block = NULL;
 379
 380     p_sys->i_pts = p_block->i_pts;
 381     if( p_sys->i_pts <= 0 )
 382     {
 383 #ifdef DEBUG_DVBSUB
 384         /* Some DVB channels send stuffing segments in non-dated packets so
 385          * don't complain too loudly. */
 386         msg_Warn( p_dec, "non dated subtitle" );
 387 #endif
 388         block_Release( p_block );
 389         return NULL;
 390     }
 391
 392     bs_init( &p_sys->bs, p_block->p_buffer, p_block->i_buffer );
 393
 394     if( bs_read( &p_sys->bs, 8 ) != 0x20 ) /* Data identifier */
 395     {
 396         msg_Dbg( p_dec, "invalid data identifier" );
 397         block_Release( p_block );
 398         return NULL;
 399     }
 400
 401     if( bs_read( &p_sys->bs, 8 ) ) /* Subtitle stream id */
 402     {
 403         msg_Dbg( p_dec, "invalid subtitle stream id" );
 404         block_Release( p_block );
 405         return NULL;
 406     }
 407
 408 #ifdef DEBUG_DVBSUB
 409     msg_Dbg( p_dec, "subtitle packet received: "I64Fd, p_sys->i_pts );
 410 #endif
 411
 412     p_sys->b_page = VLC_FALSE;
 413     while( bs_show( &p_sys->bs, 8 ) == 0x0f ) /* Sync byte */
 414     {
 415         decode_segment( p_dec, &p_sys->bs );
 416     }
 417
 418     if( bs_read( &p_sys->bs, 8 ) != 0xff ) /* End marker */
 419     {
 420         msg_Warn( p_dec, "end marker not found (corrupted subtitle ?)" );
 421         block_Release( p_block );
 422         return NULL;
 423     }
 424
 425     /* Check if the page is to be displayed */
 426     if( p_sys->p_page && p_sys->b_page )
 427         p_spu = render( p_dec );
 428
 429     block_Release( p_block );
 430
 431     return p_spu;
 432 }
 433
 434 /* following functions are local */
 435
 436 /*****************************************************************************
 437  * default_clut_init: default clut as defined in EN 300-743 section 10
 438  *****************************************************************************/
 439 static void default_clut_init( decoder_t *p_dec )
 440 {
 441     decoder_sys_t *p_sys = p_dec->p_sys;
 442     uint8_t i;
 443
 444 #define RGB_TO_Y(r, g, b) ((int16_t) 77 * r + 150 * g + 29 * b) / 256;
 445 #define RGB_TO_U(r, g, b) ((int16_t) -44 * r - 87 * g + 131 * b) / 256;
 446 #define RGB_TO_V(r, g, b) ((int16_t) 131 * r - 110 * g - 21 * b) / 256;
 447
 448     /* 4 entries CLUT */
 449     for( i = 0; i < 4; i++ )
 450     {
 451         uint8_t R = 0, G = 0, B = 0, T = 0;
 452
 453         if( !(i & 0x2) && !(i & 0x1) ) T = 0xFF;
 454         else if( !(i & 0x2) && (i & 0x1) ) R = G = B = 0xFF;
 455         else if( (i & 0x2) && !(i & 0x1) ) R = G = B = 0;
 456         else R = G = B = 0x7F;
 457
 458         p_sys->default_clut.c_2b[i].Y = RGB_TO_Y(R,G,B);
 459         p_sys->default_clut.c_2b[i].Cb = RGB_TO_V(R,G,B);
 460         p_sys->default_clut.c_2b[i].Cr = RGB_TO_U(R,G,B);
 461         p_sys->default_clut.c_2b[i].T = T;
 462     }
 463
 464     /* 16 entries CLUT */
 465     for( i = 0; i < 16; i++ )
 466     {
 467         uint8_t R = 0, G = 0, B = 0, T = 0;
 468
 469         if( !(i & 0x8) )
 470         {
 471             if( !(i & 0x4) && !(i & 0x2) && !(i & 0x1) )
 472             {
 473                 T = 0xFF;
 474             }
 475             else
 476             {
 477                 R = (i & 0x1) ? 0xFF : 0;
 478                 G = (i & 0x2) ? 0xFF : 0;
 479                 B = (i & 0x4) ? 0xFF : 0;
 480             }
 481         }
 482         else
 483         {
 484             R = (i & 0x1) ? 0x7F : 0;
 485             G = (i & 0x2) ? 0x7F : 0;
 486             B = (i & 0x4) ? 0x7F : 0;
 487         }
 488
 489         p_sys->default_clut.c_4b[i].Y = RGB_TO_Y(R,G,B);
 490         p_sys->default_clut.c_4b[i].Cr = RGB_TO_V(R,G,B);
 491         p_sys->default_clut.c_4b[i].Cb = RGB_TO_U(R,G,B);
 492         p_sys->default_clut.c_4b[i].T = T;
 493     }
 494
 495     /* 256 entries CLUT */
 496     memset( p_sys->default_clut.c_8b, 0xFF, 256 * sizeof(dvbsub_color_t) );
 497 }
 498
 499 static void decode_segment( decoder_t *p_dec, bs_t *s )
 500 {
 501     decoder_sys_t *p_sys = p_dec->p_sys;
 502     int i_type;
 503     int i_page_id;
 504     int i_size;
 505
 506     /* sync_byte (already checked) */
 507     bs_skip( s, 8 );
 508
 509     /* segment type */
 510     i_type = bs_read( s, 8 );
 511
 512     /* page id */
 513     i_page_id = bs_read( s, 16 );
 514
 515     /* segment size */
 516     i_size = bs_show( s, 16 );
 517
 518     if( ( i_page_id != p_sys->i_id ) &&
 519         ( i_page_id != p_sys->i_ancillary_id ) )
 520     {
 521 #ifdef DEBUG_DVBSUB
 522         msg_Dbg( p_dec, "subtitle skipped (page id: %i, %i)",
 523                  i_page_id, p_sys->i_id );
 524 #endif
 525         bs_skip( s,  8 * ( 2 + i_size ) );
 526         return;
 527     }
 528
 529     if( ( p_sys->i_ancillary_id != p_sys->i_id ) &&
 530         ( i_type == DVBSUB_ST_PAGE_COMPOSITION ) &&
 531         ( i_page_id == p_sys->i_ancillary_id ) )
 532     {
 533 #ifdef DEBUG_DVBSUB
 534         msg_Dbg( p_dec, "skipped invalid ancillary subtitle packet" );
 535 #endif
 536         bs_skip( s,  8 * ( 2 + i_size ) );
 537         return;
 538     }
 539
 540 #ifdef DEBUG_DVBSUB
 541     if( i_page_id == p_sys->i_id )
 542         msg_Dbg( p_dec, "segment (id: %i)", i_page_id );
 543     else
 544         msg_Dbg( p_dec, "ancillary segment (id: %i)", i_page_id );
 545 #endif
 546
 547     switch( i_type )
 548     {
 549     case DVBSUB_ST_PAGE_COMPOSITION:
 550 #ifdef DEBUG_DVBSUB
 551         msg_Dbg( p_dec, "decode_page_composition" );
 552 #endif
 553         decode_page_composition( p_dec, s );
 554         break;
 555
 556     case DVBSUB_ST_REGION_COMPOSITION:
 557 #ifdef DEBUG_DVBSUB
 558         msg_Dbg( p_dec, "decode_region_composition" );
 559 #endif
 560         decode_region_composition( p_dec, s );
 561         break;
 562
 563     case DVBSUB_ST_CLUT_DEFINITION:
 564 #ifdef DEBUG_DVBSUB
 565         msg_Dbg( p_dec, "decode_clut" );
 566 #endif
 567         decode_clut( p_dec, s );
 568         break;
 569
 570     case DVBSUB_ST_OBJECT_DATA:
 571 #ifdef DEBUG_DVBSUB
 572         msg_Dbg( p_dec, "decode_object" );
 573 #endif
 574         decode_object( p_dec, s );
 575         break;
 576
 577     case DVBSUB_ST_DISPLAY_DEFINITION:
 578 #ifdef DEBUG_DVBSUB
 579         msg_Dbg( p_dec, "decode_display_definition" );
 580 #endif
 581         decode_display_definition( p_dec, s );
 582         break;
 583
 584     case DVBSUB_ST_ENDOFDISPLAY:
 585 #ifdef DEBUG_DVBSUB
 586         msg_Dbg( p_dec, "end of display" );
 587 #endif
 588         bs_skip( s,  8 * ( 2 + i_size ) );
 589         break;
 590
 591     case DVBSUB_ST_STUFFING:
 592 #ifdef DEBUG_DVBSUB
 593         msg_Dbg( p_dec, "skip stuffing" );
 594 #endif
 595         bs_skip( s,  8 * ( 2 + i_size ) );
 596         break;
 597
 598     default:
 599         msg_Warn( p_dec, "unsupported segment type: (%04x)", i_type );
 600         bs_skip( s,  8 * ( 2 + i_size ) );
 601         break;
 602     }
 603 }
 604
 605 static void decode_clut( decoder_t *p_dec, bs_t *s )
 606 {
 607     decoder_sys_t *p_sys = p_dec->p_sys;
 608     uint16_t      i_segment_length;
 609     uint16_t      i_processed_length;
 610     dvbsub_clut_t *p_clut, *p_next;
 611     int           i_id, i_version;
 612
 613     i_segment_length = bs_read( s, 16 );
 614     i_id             = bs_read( s, 8 );
 615     i_version        = bs_read( s, 4 );
 616
 617     /* Check if we already have this clut */
 618     for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut->p_next )
 619     {
 620         if( p_clut->i_id == i_id ) break;
 621     }
 622
 623     /* Check version number */
 624     if( p_clut && ( p_clut->i_version == i_version ) )
 625     {
 626         /* Nothing to do */
 627         bs_skip( s, 8 * i_segment_length - 12 );
 628         return;
 629     }
 630
 631     if( !p_clut )
 632     {
 633 #ifdef DEBUG_DVBSUB
 634         msg_Dbg( p_dec, "new clut: %i", i_id );
 635 #endif
 636         p_clut = malloc( sizeof(dvbsub_clut_t) );
 637         p_clut->p_next = p_sys->p_cluts;
 638         p_sys->p_cluts = p_clut;
 639     }
 640
 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;
 645
 646     /* We don't have this version of the CLUT: Parse it */
 647     p_clut->i_version = i_version;
 648     p_clut->i_id = i_id;
 649     bs_skip( s, 4 ); /* Reserved bits */
 650     i_processed_length = 2;
 651     while( i_processed_length < i_segment_length )
 652     {
 653         uint8_t y, cb, cr, t;
 654         uint8_t i_id;
 655         uint8_t i_type;
 656
 657         i_id = bs_read( s, 8 );
 658         i_type = bs_read( s, 3 );
 659
 660         bs_skip( s, 4 );
 661
 662         if( bs_read( s, 1 ) )
 663         {
 664             y  = bs_read( s, 8 );
 665             cr = bs_read( s, 8 );
 666             cb = bs_read( s, 8 );
 667             t  = bs_read( s, 8 );
 668             i_processed_length += 6;
 669         }
 670         else
 671         {
 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;
 677         }
 678
 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). */
 682         if( y == 0 )
 683         {
 684             cr = cb = 0;
 685             t  = 0xff;
 686         }
 687
 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 ) )
 692         {
 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;
 697         }
 698         if( ( i_type & 0x02 ) && ( i_id < 16 ) )
 699         {
 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;
 704         }
 705         if( i_type & 0x01 )
 706         {
 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;
 711         }
 712     }
 713 }
 714
 715 static void decode_page_composition( decoder_t *p_dec, bs_t *s )
 716 {
 717     decoder_sys_t *p_sys = p_dec->p_sys;
 718     int i_version, i_state, i_segment_length, i_timeout, i;
 719
 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 */
 726
 727     if( i_state == DVBSUB_PCS_STATE_CHANGE )
 728     {
 729         /* End of an epoch, reset decoder buffer */
 730 #ifdef DEBUG_DVBSUB
 731         msg_Dbg( p_dec, "page composition mode change" );
 732 #endif
 733         free_all( p_dec );
 734     }
 735     else if( !p_sys->p_page && ( i_state != DVBSUB_PCS_STATE_ACQUISITION ) &&
 736              ( i_state != DVBSUB_PCS_STATE_CHANGE ) )
 737     {
 738         /* Not a full PCS, we need to wait for one */
 739         msg_Dbg( p_dec, "didn't receive an acquisition page yet" );
 740
 741 #if 0
 742         /* Try to start decoding even without an acquisition page */
 743         bs_skip( s,  8 * (i_segment_length - 2) );
 744         return;
 745 #endif
 746     }
 747
 748 #ifdef DEBUG_DVBSUB
 749     if( i_state == DVBSUB_PCS_STATE_ACQUISITION )
 750         msg_Dbg( p_dec, "acquisition page composition" );
 751 #endif
 752
 753     /* Check version number */
 754     if( p_sys->p_page && ( p_sys->p_page->i_version == i_version ) )
 755     {
 756         bs_skip( s,  8 * (i_segment_length - 2) );
 757         return;
 758     }
 759     else if( p_sys->p_page )
 760     {
 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;
 765     }
 766
 767     if( !p_sys->p_page )
 768     {
 769 #ifdef DEBUG_DVBSUB
 770         msg_Dbg( p_dec, "new page" );
 771 #endif
 772         /* Allocate a new page */
 773         p_sys->p_page = malloc( sizeof(dvbsub_page_t) );
 774     }
 775
 776     p_sys->p_page->i_version = i_version;
 777     p_sys->p_page->i_timeout = i_timeout;
 778     p_sys->b_page = VLC_TRUE;
 779
 780     /* Number of regions */
 781     p_sys->p_page->i_region_defs = (i_segment_length - 2) / 6;
 782
 783     if( p_sys->p_page->i_region_defs == 0 ) return;
 784
 785     p_sys->p_page->p_region_defs =
 786         malloc( p_sys->p_page->i_region_defs * sizeof(dvbsub_region_t) );
 787     if( p_sys->p_page->p_region_defs )
 788     {
 789         for( i = 0; i < p_sys->p_page->i_region_defs; i++ )
 790         {
 791             p_sys->p_page->p_region_defs[i].i_id = bs_read( s, 8 );
 792             bs_skip( s, 8 ); /* Reserved */
 793             p_sys->p_page->p_region_defs[i].i_x = bs_read( s, 16 );
 794             p_sys->p_page->p_region_defs[i].i_y = bs_read( s, 16 );
 795
 796 #ifdef DEBUG_DVBSUB
 797             msg_Dbg( p_dec, "page_composition, region %i (%i,%i)",
 798                     i, p_sys->p_page->p_region_defs[i].i_x,
 799                     p_sys->p_page->p_region_defs[i].i_y );
 800 #endif
 801         }
 802     }
 803 }
 804
 805 static void decode_region_composition( decoder_t *p_dec, bs_t *s )
 806 {
 807     decoder_sys_t *p_sys = p_dec->p_sys;
 808     dvbsub_region_t *p_region, **pp_region = &p_sys->p_regions;
 809     int i_segment_length, i_processed_length, i_id, i_version;
 810     int i_width, i_height, i_level_comp, i_depth, i_clut;
 811     int i_8_bg, i_4_bg, i_2_bg;
 812     vlc_bool_t b_fill;
 813
 814     i_segment_length = bs_read( s, 16 );
 815     i_id = bs_read( s, 8 );
 816     i_version = bs_read( s, 4 );
 817
 818     /* Check if we already have this region */
 819     for( p_region = p_sys->p_regions; p_region != NULL;
 820          p_region = p_region->p_next )
 821     {
 822         pp_region = &p_region->p_next;
 823         if( p_region->i_id == i_id ) break;
 824     }
 825
 826     /* Check version number */
 827     if( p_region && ( p_region->i_version == i_version ) )
 828     {
 829         bs_skip( s, 8 * (i_segment_length - 1) - 4 );
 830         return;
 831     }
 832
 833     if( !p_region )
 834     {
 835 #ifdef DEBUG_DVBSUB
 836         msg_Dbg( p_dec, "new region: %i", i_id );
 837 #endif
 838         p_region = *pp_region = malloc( sizeof(dvbsub_region_t) );
 839         if( p_region )
 840         {
 841             memset( p_region, 0, sizeof(dvbsub_region_t) );
 842             p_region->p_object_defs = NULL;
 843             p_region->p_pixbuf = NULL;
 844             p_region->p_next = NULL;
 845         }
 846     }
 847
 848     /* Region attributes */
 849     p_region->i_id = i_id;
 850     p_region->i_version = i_version;
 851     b_fill = bs_read( s, 1 );
 852     bs_skip( s, 3 ); /* Reserved */
 853
 854     i_width = bs_read( s, 16 );
 855     i_height = bs_read( s, 16 );
 856 #ifdef DEBUG_DVBSUB
 857     msg_Dbg( p_dec, " width=%d height=%d", i_width, i_height );
 858 #endif
 859     i_level_comp = bs_read( s, 3 );
 860     i_depth = bs_read( s, 3 );
 861     bs_skip( s, 2 ); /* Reserved */
 862     i_clut = bs_read( s, 8 );
 863
 864     i_8_bg = bs_read( s, 8 );
 865     i_4_bg = bs_read( s, 4 );
 866     i_2_bg = bs_read( s, 2 );
 867     bs_skip( s, 2 ); /* Reserved */
 868
 869     /* Free old object defs */
 870     while( p_region->i_object_defs )
 871     {
 872         int i = p_region->i_object_defs - 1;
 873         if( p_region->p_object_defs[i].psz_text )
 874             free( p_region->p_object_defs[i].psz_text );
 875         if( !i )
 876             free( p_region->p_object_defs );
 877
 878         p_region->i_object_defs--;
 879     }
 880     p_region->p_object_defs = NULL;
 881
 882     /* Extra sanity checks */
 883     if( ( p_region->i_width != i_width ) ||
 884         ( p_region->i_height != i_height ) )
 885     {
 886         if( p_region->p_pixbuf )
 887         {
 888             msg_Dbg( p_dec, "region size changed (%dx%d->%dx%d)",
 889                      p_region->i_width, p_region->i_height, i_width, i_height );
 890             free( p_region->p_pixbuf );
 891         }
 892
 893         p_region->p_pixbuf = malloc( i_height * i_width );
 894         p_region->i_depth = 0;
 895         b_fill = VLC_TRUE;
 896     }
 897     if( p_region->i_depth &&
 898         ( ( p_region->i_depth != i_depth ) ||
 899           ( p_region->i_level_comp != i_level_comp ) ||
 900           ( p_region->i_clut != i_clut) ) )
 901     {
 902         msg_Dbg( p_dec, "region parameters changed (not allowed)" );
 903     }
 904
 905     /* Erase background of region */
 906     if( b_fill )
 907     {
 908         int i_background = ( p_region->i_depth == 1 ) ? i_2_bg :
 909             ( ( p_region->i_depth == 2 ) ? i_4_bg : i_8_bg );
 910         memset( p_region->p_pixbuf, i_background, i_width * i_height );
 911     }
 912
 913     p_region->i_width = i_width;
 914     p_region->i_height = i_height;
 915     p_region->i_level_comp = i_level_comp;
 916     p_region->i_depth = i_depth;
 917     p_region->i_clut = i_clut;
 918
 919     /* List of objects in the region */
 920     i_processed_length = 10;
 921     while( i_processed_length < i_segment_length )
 922     {
 923         dvbsub_objectdef_t *p_obj;
 924
 925         /* We create a new object */
 926         p_region->i_object_defs++;
 927         p_region->p_object_defs =
 928             realloc( p_region->p_object_defs,
 929                      sizeof(dvbsub_objectdef_t) * p_region->i_object_defs );
 930
 931         /* We parse object properties */
 932         p_obj = &p_region->p_object_defs[p_region->i_object_defs - 1];
 933         p_obj->i_id         = bs_read( s, 16 );
 934         p_obj->i_type       = bs_read( s, 2 );
 935         bs_skip( s, 2 ); /* Provider */
 936         p_obj->i_x          = bs_read( s, 12 );
 937         bs_skip( s, 4 ); /* Reserved */
 938         p_obj->i_y          = bs_read( s, 12 );
 939         p_obj->psz_text     = 0;
 940
 941         i_processed_length += 6;
 942
 943         if( ( p_obj->i_type == DVBSUB_OT_BASIC_CHAR ) ||
 944             ( p_obj->i_type == DVBSUB_OT_COMPOSITE_STRING ) )
 945         {
 946             p_obj->i_fg_pc =  bs_read( s, 8 );
 947             p_obj->i_bg_pc =  bs_read( s, 8 );
 948             i_processed_length += 2;
 949         }
 950     }
 951 }
 952
 953 /* ETSI 300 743 [7.2.1] */
 954 static void decode_display_definition( decoder_t *p_dec, bs_t *s )
 955 {
 956     decoder_sys_t *p_sys = p_dec->p_sys;
 957     uint16_t      i_segment_length;
 958     uint16_t      i_processed_length = 40;
 959     dvbsub_display_t *p_display;
 960     dvbsub_display_t *p_old = p_sys->p_display;
 961     int           i_version;
 962
 963     i_segment_length = bs_read( s, 16 );
 964     i_version        = bs_read( s, 4 );
 965
 966     /* Check version number */
 967     if( p_old && ( p_old->i_version == i_version ) )
 968     {
 969         /* The definition did not change */
 970         bs_skip( s, 8*i_segment_length - 4 );
 971         return;
 972     }
 973
 974 #ifdef DEBUG_DVBSUB
 975     msg_Dbg( p_dec, "new display definition: %i", i_version );
 976 #endif
 977     p_display = malloc( sizeof(dvbsub_display_t) );
 978     if( p_display )
 979     {
 980         /* We don't have this version of the display definition: Parse it */
 981         p_display->i_version = i_version;
 982         p_display->b_windowed = bs_read( s, 1 );
 983         bs_skip( s, 3 ); /* Reserved bits */
 984         p_display->i_width = bs_read( s, 16 )+1;
 985         p_display->i_height = bs_read( s, 16 )+1;
 986
 987         if( p_display->b_windowed )
 988         {
 989 #ifdef DEBUG_DVBSUB
 990             msg_Dbg( p_dec, "display definition with offsets (windowed)" );
 991 #endif
 992             /* Coordinates are measured from the top left corner */
 993             p_display->i_x     = bs_read( s, 16 );
 994             p_display->i_max_x = bs_read( s, 16 );
 995             p_display->i_y     = bs_read( s, 16 );
 996             p_display->i_max_y = bs_read( s, 16 );
 997             i_processed_length += 64;
 998         }
 999     }
1000
1001     p_sys->p_display = p_display;
1002     if( p_old ) free( p_old );
1003
1004     if( i_processed_length != i_segment_length*8 )
1005     {
1006         msg_Err( p_dec, "processed length %d != segment length %d",
1007                  i_processed_length, i_segment_length );
1008     }
1009
1010 #ifdef DEBUG_DVBSUB
1011     msg_Dbg( p_dec, "version: %d, width: %d, height: %d",
1012              p_display->i_version, p_display->i_width, p_display->i_height );
1013     if( p_display->b_windowed )
1014         msg_Dbg( p_dec, "xmin: %d, xmax: %d, ymin: %d, ymax: %d",
1015                  p_display->i_x, p_display->i_max_x, p_display->i_y, p_display->i_max_y );
1016 #endif
1017 }
1018
1019 static void dvbsub_render_pdata( decoder_t *, dvbsub_region_t *, int, int,
1020                                  uint8_t *, int );
1021 static void dvbsub_pdata2bpp( bs_t *, uint8_t *, int, int * );
1022 static void dvbsub_pdata4bpp( bs_t *, uint8_t *, int, int * );
1023 static void dvbsub_pdata8bpp( bs_t *, uint8_t *, int, int * );
1024
1025 static void decode_object( decoder_t *p_dec, bs_t *s )
1026 {
1027     decoder_sys_t *p_sys = p_dec->p_sys;
1028     dvbsub_region_t *p_region;
1029     int i_segment_length, i_coding_method, i_version, i_id, i;
1030     vlc_bool_t b_non_modify_color;
1031
1032     /* ETSI 300-743 paragraph 7.2.4
1033      * sync_byte, segment_type and page_id have already been processed.
1034      */
1035     i_segment_length = bs_read( s, 16 );
1036     i_id             = bs_read( s, 16 );
1037     i_version        = bs_read( s, 4 );
1038     i_coding_method  = bs_read( s, 2 );
1039
1040     if( i_coding_method > 1 )
1041     {
1042         msg_Dbg( p_dec, "unknown DVB subtitling coding %d is not handled!", i_coding_method );
1043         bs_skip( s, 8 * (i_segment_length - 2) - 6 );
1044         return;
1045     }
1046
1047     /* Check if the object needs to be rendered in at least one
1048      * of the regions */
1049     for( p_region = p_sys->p_regions; p_region != NULL;
1050          p_region = p_region->p_next )
1051     {
1052         for( i = 0; i < p_region->i_object_defs; i++ )
1053             if( p_region->p_object_defs[i].i_id == i_id ) break;
1054
1055         if( i != p_region->i_object_defs ) break;
1056     }
1057     if( !p_region )
1058     {
1059         bs_skip( s, 8 * (i_segment_length - 2) - 6 );
1060         return;
1061     }
1062
1063 #ifdef DEBUG_DVBSUB
1064     msg_Dbg( p_dec, "new object: %i", i_id );
1065 #endif
1066
1067     b_non_modify_color = bs_read( s, 1 );
1068     bs_skip( s, 1 ); /* Reserved */
1069
1070     if( i_coding_method == 0x00 )
1071     {
1072         int i_topfield, i_bottomfield;
1073         uint8_t *p_topfield, *p_bottomfield;
1074
1075         i_topfield    = bs_read( s, 16 );
1076         i_bottomfield = bs_read( s, 16 );
1077         p_topfield    = s->p_start + bs_pos( s ) / 8;
1078         p_bottomfield = p_topfield + i_topfield;
1079
1080         bs_skip( s, 8 * (i_segment_length - 7) );
1081
1082         /* Sanity check */
1083         if( ( i_segment_length < ( i_topfield + i_bottomfield + 7 ) ) ||
1084             ( ( p_topfield + i_topfield + i_bottomfield ) > s->p_end ) )
1085         {
1086             msg_Dbg( p_dec, "corrupted object data" );
1087             return;
1088         }
1089
1090         for( p_region = p_sys->p_regions; p_region != NULL;
1091              p_region = p_region->p_next )
1092         {
1093             for( i = 0; i < p_region->i_object_defs; i++ )
1094             {
1095                 if( p_region->p_object_defs[i].i_id != i_id ) continue;
1096
1097                 dvbsub_render_pdata( p_dec, p_region,
1098                                      p_region->p_object_defs[i].i_x,
1099                                      p_region->p_object_defs[i].i_y,
1100                                      p_topfield, i_topfield );
1101
1102                 if( i_bottomfield )
1103                 {
1104                     dvbsub_render_pdata( p_dec, p_region,
1105                                          p_region->p_object_defs[i].i_x,
1106                                          p_region->p_object_defs[i].i_y + 1,
1107                                          p_bottomfield, i_bottomfield );
1108                 }
1109                 else
1110                 {
1111                     /* Duplicate the top field */
1112                     dvbsub_render_pdata( p_dec, p_region,
1113                                          p_region->p_object_defs[i].i_x,
1114                                          p_region->p_object_defs[i].i_y + 1,
1115                                          p_topfield, i_topfield );
1116                 }
1117             }
1118         }
1119     }
1120     else
1121     {
1122         /* DVB subtitling as characters */
1123         int i_number_of_codes = bs_read( s, 8 );
1124         uint8_t* p_start = s->p_start + bs_pos( s ) / 8;
1125
1126         /* Sanity check */
1127         if( ( i_segment_length < ( i_number_of_codes*2 + 4 ) ) ||
1128             ( ( p_start + i_number_of_codes*2 ) > s->p_end ) )
1129         {
1130             msg_Dbg( p_dec, "corrupted object data" );
1131             return;
1132         }
1133
1134         for( p_region = p_sys->p_regions; p_region != NULL;
1135              p_region = p_region->p_next )
1136         {
1137             for( i = 0; i < p_region->i_object_defs; i++ )
1138             {
1139                 int j;
1140
1141                 if( p_region->p_object_defs[i].i_id != i_id ) continue;
1142
1143                 p_region->p_object_defs[i].psz_text =
1144                     realloc( p_region->p_object_defs[i].psz_text,
1145                              i_number_of_codes + 1 );
1146
1147                 /* FIXME 16bits -> char ??? See Preamble */
1148                 for( j = 0; j < i_number_of_codes; j++ )
1149                 {
1150                     p_region->p_object_defs[i].psz_text[j] = (char)(bs_read( s, 16 ) & 0xFF);
1151                 }
1152                 /* Null terminate the string */
1153                 p_region->p_object_defs[i].psz_text[j] = 0;
1154             }
1155         }
1156     }
1157
1158 #ifdef DEBUG_DVBSUB
1159     msg_Dbg( p_dec, "end object: %i", i_id );
1160 #endif
1161 }
1162
1163 static void dvbsub_render_pdata( decoder_t *p_dec, dvbsub_region_t *p_region,
1164                                  int i_x, int i_y,
1165                                  uint8_t *p_field, int i_field )
1166 {
1167     uint8_t *p_pixbuf;
1168     int i_offset = 0;
1169     bs_t bs;
1170
1171     /* Sanity check */
1172     if( !p_region->p_pixbuf )
1173     {
1174         msg_Err( p_dec, "region %i has no pixel buffer!", p_region->i_id );
1175         return;
1176     }
1177     if( i_y < 0 || i_x < 0 || i_y >= p_region->i_height ||
1178         i_x >= p_region->i_width )
1179     {
1180         msg_Dbg( p_dec, "invalid offset (%i,%i)", i_x, i_y );
1181         return;
1182     }
1183
1184     p_pixbuf = p_region->p_pixbuf + i_y * p_region->i_width;
1185     bs_init( &bs, p_field, i_field );
1186
1187     while( !bs_eof( &bs ) )
1188     {
1189         /* Sanity check */
1190         if( i_y >= p_region->i_height ) return;
1191
1192         switch( bs_read( &bs, 8 ) )
1193         {
1194         case 0x10:
1195             dvbsub_pdata2bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1196                               &i_offset );
1197             break;
1198
1199         case 0x11:
1200             dvbsub_pdata4bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1201                               &i_offset );
1202             break;
1203
1204         case 0x12:
1205             dvbsub_pdata8bpp( &bs, p_pixbuf + i_x, p_region->i_width - i_x,
1206                               &i_offset );
1207             break;
1208
1209         case 0x20:
1210         case 0x21:
1211         case 0x22:
1212             /* We don't use map tables */
1213             break;
1214
1215         case 0xf0: /* End of line code */
1216             p_pixbuf += 2*p_region->i_width;
1217             i_offset = 0; i_y += 2;
1218             break;
1219         }
1220     }
1221 }
1222
1223 static void dvbsub_pdata2bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1224 {
1225     vlc_bool_t b_stop = VLC_FALSE;
1226
1227     while( !b_stop && !bs_eof( s ) )
1228     {
1229         int i_count = 0, i_color = 0;
1230
1231         i_color = bs_read( s, 2 );
1232         if( i_color != 0x00 )
1233         {
1234             i_count = 1;
1235         }
1236         else
1237         {
1238             if( bs_read( s, 1 ) == 0x01 )         // Switch1
1239             {
1240                 i_count = 3 + bs_read( s, 3 );
1241                 i_color = bs_read( s, 2 );
1242             }
1243             else
1244             {
1245                 if( bs_read( s, 1 ) == 0x00 )     //Switch2
1246                 {
1247                     switch( bs_read( s, 2 ) )     //Switch3
1248                     {
1249                     case 0x00:
1250                         b_stop = VLC_TRUE;
1251                         break;
1252                     case 0x01:
1253                         i_count = 2;
1254                         break;
1255                     case 0x02:
1256                         i_count =  12 + bs_read( s, 4 );
1257                         i_color = bs_read( s, 2 );
1258                         break;
1259                     case 0x03:
1260                         i_count =  29 + bs_read( s, 8 );
1261                         i_color = bs_read( s, 2 );
1262                         break;
1263                     default:
1264                         break;
1265                     }
1266                 }
1267                 else
1268                 {
1269                     /* 1 pixel color 0 */
1270                     i_count = 1;
1271                 }
1272             }
1273         }
1274
1275         if( !i_count ) continue;
1276
1277         /* Sanity check */
1278         if( ( i_count + *pi_off ) > i_width ) break;
1279
1280         if( i_count == 1 ) p[*pi_off] = i_color;
1281         else memset( ( p + *pi_off ), i_color, i_count );
1282
1283         (*pi_off) += i_count;
1284     }
1285
1286     bs_align( s );
1287 }
1288
1289 static void dvbsub_pdata4bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1290 {
1291     vlc_bool_t b_stop = VLC_FALSE;
1292
1293     while( !b_stop && !bs_eof( s ) )
1294     {
1295         int i_count = 0, i_color = 0;
1296
1297         i_color = bs_read( s, 4 );
1298         if( i_color != 0x00 )
1299         {
1300             /* Add 1 pixel */
1301             i_count = 1;
1302         }
1303         else
1304         {
1305             if( bs_read( s, 1 ) == 0x00 )           // Switch1
1306             {
1307                 if( bs_show( s, 3 ) != 0x00 )
1308                 {
1309                     i_count = 2 + bs_read( s, 3 );
1310                 }
1311                 else
1312                 {
1313                     bs_skip( s, 3 );
1314                     b_stop = VLC_TRUE;
1315                 }
1316             }
1317             else
1318             {
1319                 if( bs_read( s, 1 ) == 0x00)        //Switch2
1320                 {
1321                     i_count =  4 + bs_read( s, 2 );
1322                     i_color = bs_read( s, 4 );
1323                 }
1324                 else
1325                 {
1326                     switch ( bs_read( s, 2 ) )     //Switch3
1327                     {
1328                     case 0x0:
1329                         i_count = 1;
1330                         break;
1331                     case 0x1:
1332                         i_count = 2;
1333                         break;
1334                     case 0x2:
1335                         i_count = 9 + bs_read( s, 4 );
1336                         i_color = bs_read( s, 4 );
1337                         break;
1338                     case 0x3:
1339                         i_count= 25 + bs_read( s, 8 );
1340                         i_color = bs_read( s, 4 );
1341                         break;
1342                     }
1343                 }
1344             }
1345         }
1346
1347         if( !i_count ) continue;
1348
1349         /* Sanity check */
1350         if( ( i_count + *pi_off ) > i_width ) break;
1351
1352         if( i_count == 1 ) p[*pi_off] = i_color;
1353         else memset( ( p + *pi_off ), i_color, i_count );
1354
1355         (*pi_off) += i_count;
1356     }
1357
1358     bs_align( s );
1359 }
1360
1361 static void dvbsub_pdata8bpp( bs_t *s, uint8_t *p, int i_width, int *pi_off )
1362 {
1363     vlc_bool_t b_stop = VLC_FALSE;
1364
1365     while( !b_stop && !bs_eof( s ) )
1366     {
1367         int i_count = 0, i_color = 0;
1368
1369         i_color = bs_read( s, 8 );
1370         if( i_color != 0x00 )
1371         {
1372             /* Add 1 pixel */
1373             i_count = 1;
1374         }
1375         else
1376         {
1377             if( bs_read( s, 1 ) == 0x00 )           // Switch1
1378             {
1379                 if( bs_show( s, 7 ) != 0x00 )
1380                 {
1381                     i_count = bs_read( s, 7 );
1382                 }
1383                 else
1384                 {
1385                     bs_skip( s, 7 );
1386                     b_stop = VLC_TRUE;
1387                 }
1388             }
1389             else
1390             {
1391                 i_count = bs_read( s, 7 );
1392                 i_color = bs_read( s, 8 );
1393             }
1394         }
1395
1396         if( !i_count ) continue;
1397
1398         /* Sanity check */
1399         if( ( i_count + *pi_off ) > i_width ) break;
1400
1401         if( i_count == 1 ) p[*pi_off] = i_color;
1402         else memset( ( p + *pi_off ), i_color, i_count );
1403
1404         (*pi_off) += i_count;
1405     }
1406
1407     bs_align( s );
1408 }
1409
1410 static void free_all( decoder_t *p_dec )
1411 {
1412     decoder_sys_t *p_sys = p_dec->p_sys;
1413     dvbsub_region_t *p_reg, *p_reg_next;
1414     dvbsub_clut_t *p_clut, *p_clut_next;
1415
1416     if( p_sys->p_display ) free( p_sys->p_display );
1417
1418     for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut_next )
1419     {
1420         p_clut_next = p_clut->p_next;
1421         free( p_clut );
1422     }
1423     p_sys->p_cluts = NULL;
1424
1425     for( p_reg = p_sys->p_regions; p_reg != NULL; p_reg = p_reg_next )
1426     {
1427         int i;
1428
1429         p_reg_next = p_reg->p_next;
1430         for( i = 0; i < p_reg->i_object_defs; i++ )
1431             if( p_reg->p_object_defs[i].psz_text )
1432                 free( p_reg->p_object_defs[i].psz_text );
1433         if( p_reg->i_object_defs ) free( p_reg->p_object_defs );
1434         if( p_reg->p_pixbuf ) free( p_reg->p_pixbuf );
1435         free( p_reg );
1436     }
1437     p_sys->p_regions = NULL;
1438
1439     if( p_sys->p_page )
1440     {
1441         if( p_sys->p_page->i_region_defs )
1442             free( p_sys->p_page->p_region_defs );
1443         free( p_sys->p_page );
1444     }
1445     p_sys->p_page = NULL;
1446 }
1447
1448 static subpicture_t *render( decoder_t *p_dec )
1449 {
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;
1454
1455     /* Allocate the subpicture internal data. */
1456     p_spu = p_dec->pf_spu_buffer_new( p_dec );
1457     if( !p_spu ) return NULL;
1458
1459     pp_spu_region = &p_spu->p_region;
1460
1461     /* Loop on region definitions */
1462 #ifdef DEBUG_DVBSUB
1463     if( p_sys->p_page )
1464         msg_Dbg( p_dec, "rendering %i regions", p_sys->p_page->i_region_defs );
1465 #endif
1466
1467     for( i = 0; p_sys->p_page && ( i < p_sys->p_page->i_region_defs ); i++ )
1468     {
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;
1475         video_format_t fmt;
1476         int i_pitch;
1477
1478         i_timeout = p_sys->p_page->i_timeout;
1479
1480         p_regiondef = &p_sys->p_page->p_region_defs[i];
1481
1482 #ifdef DEBUG_DVBSUB
1483         msg_Dbg( p_dec, "rendering region %i (%i,%i)", i,
1484                  p_regiondef->i_x, p_regiondef->i_y );
1485 #endif
1486
1487         /* Find associated region */
1488         for( p_region = p_sys->p_regions; p_region != NULL;
1489              p_region = p_region->p_next )
1490         {
1491             if( p_regiondef->i_id == p_region->i_id ) break;
1492         }
1493
1494         if( !p_region )
1495         {
1496             msg_Dbg( p_dec, "region %i not found", p_regiondef->i_id );
1497             continue;
1498         }
1499
1500         /* Find associated CLUT */
1501         for( p_clut = p_sys->p_cluts; p_clut != NULL; p_clut = p_clut->p_next )
1502         {
1503             if( p_region->i_clut == p_clut->i_id ) break;
1504         }
1505         if( !p_clut )
1506         {
1507             msg_Dbg( p_dec, "clut %i not found", p_region->i_clut );
1508             continue;
1509         }
1510
1511         /* FIXME: don't create a subpicture region with VLC_FOURCC YUVP
1512          * when it actually is a TEXT region */
1513
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 );
1522         if( !p_spu_region )
1523         {
1524             msg_Err( p_dec, "cannot allocate SPU region" );
1525             continue;
1526         }
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;
1532
1533         /* Build palette */
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++ )
1539         {
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;
1544         }
1545
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;
1549
1550         /* Copy pixel buffer */
1551         for( j = 0; j < p_region->i_height; j++ )
1552         {
1553             memcpy( p_dst, p_src, p_region->i_width );
1554             p_src += p_region->i_width;
1555             p_dst += i_pitch;
1556         }
1557
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++ )
1561         {
1562             dvbsub_objectdef_t *p_object_def = &p_region->p_object_defs[j];
1563
1564             if( ( p_object_def->i_type != 1 ) || !p_object_def->psz_text )
1565                 continue;
1566
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 );
1575             if( !p_region )
1576             {
1577                 msg_Err( p_dec, "cannot allocate SPU region" );
1578                 continue;
1579             }
1580
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;
1587         }
1588     }
1589
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 = VLC_TRUE;
1594     p_spu->b_pausable = VLC_TRUE;
1595     //p_spu->b_fade = VLC_TRUE;
1596     //p_spu->i_stop = p_spu->i_start + (mtime_t) (i_timeout * 1000000);
1597
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;
1605
1606     if( p_sys->p_display )
1607     {
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;
1610
1611         if( p_sys->p_display->b_windowed )
1612         {
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;
1618         }
1619     }
1620
1621     return p_spu;
1622 }
1623
1624 /*****************************************************************************
1625  * encoder_sys_t : encoder descriptor
1626  *****************************************************************************/
1627 typedef struct encoder_region_t
1628 {
1629     int i_width;
1630     int i_height;
1631
1632 } encoder_region_t;
1633
1634 struct encoder_sys_t
1635 {
1636     unsigned int i_page_ver;
1637     unsigned int i_region_ver;
1638     unsigned int i_clut_ver;
1639
1640     int i_regions;
1641     encoder_region_t *p_regions;
1642
1643     mtime_t i_pts;
1644
1645     /* subpicture positioning */
1646     int i_offset_x;
1647     int i_offset_y;
1648 };
1649
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 * );
1654
1655 /*****************************************************************************
1656  * OpenEncoder: probe the encoder and return score
1657  *****************************************************************************/
1658 static int OpenEncoder( vlc_object_t *p_this )
1659 {
1660     encoder_t *p_enc = (encoder_t *)p_this;
1661     encoder_sys_t *p_sys;
1662     vlc_value_t val;
1663
1664     if( ( p_enc->fmt_out.i_codec != VLC_FOURCC('d','v','b','s') ) &&
1665         !p_enc->b_force )
1666     {
1667         return VLC_EGENERIC;
1668     }
1669
1670     /* Allocate the memory needed to store the decoder's structure */
1671     if( ( p_sys = (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
1672     {
1673         msg_Err( p_enc, "out of memory" );
1674         return VLC_ENOMEM;
1675     }
1676     p_enc->p_sys = p_sys;
1677
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;
1681
1682     config_ChainParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
1683
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;
1689
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;
1696
1697     return VLC_SUCCESS;
1698 }
1699
1700 /* FIXME: this routine is a hack to convert VLC_FOURCC('Y','U','V','A')
1701  *        into VLC_FOURCC('Y','U','V','P')
1702  */
1703 static subpicture_t *YuvaYuvp( encoder_t *p_enc, subpicture_t *p_subpic )
1704 {
1705     subpicture_region_t *p_region = NULL;
1706
1707     if( !p_subpic ) return NULL;
1708
1709     for( p_region = p_subpic->p_region; p_region; p_region = p_region->p_next )
1710     {
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;
1714
1715 #ifdef RANDOM_DITHERING
1716         int i_seed = 0xdeadbeef; /* random seed */
1717 #else
1718         int *pi_delta;
1719 #endif
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;
1726
1727 #ifdef DEBUG_DVBSUB
1728         msg_Dbg( p_enc, "YuvaYuvp: i_pixels=%d, i_iterator=%d", i_pixels, i_iterator );
1729 #endif
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;
1734
1735         /* Find best iterator using Euclide’s algorithm */
1736         for( ; i_iterator > 1 ; i_iterator-- )
1737         {
1738             int a = i_pixels;
1739             int b = i_iterator;
1740             int c;
1741
1742             while( b )
1743             {
1744                 c = a % b;
1745                 a = b;
1746                 b = c;
1747             }
1748
1749             if( a == 1 )
1750             {
1751                 break;
1752             }
1753         }
1754
1755         /* Count colors, build best palette */
1756         for( i_tolerance = 0; i_tolerance < 128; i_tolerance++ )
1757         {
1758             vlc_bool_t b_success = VLC_TRUE;
1759             p_fmt->p_palette->i_entries = 0;
1760
1761             for( i = 0; i < i_pixels ; )
1762             {
1763                 uint8_t y, u, v, a;
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++ )
1769                 {
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 )
1774                     {
1775                         break;
1776                     }
1777                 }
1778                 if( j == p_fmt->p_palette->i_entries )
1779                 {
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++;
1785                 }
1786                 if( p_fmt->p_palette->i_entries >= i_max_entries )
1787                 {
1788                     b_success = VLC_FALSE;
1789                     break;
1790                 }
1791                 i += i_iterator;
1792                 if( i > i_pixels )
1793                 {
1794                     i -= i_pixels;
1795                 }
1796             }
1797
1798             if( b_success )
1799             {
1800                 break;
1801             }
1802         }
1803
1804 #ifdef DEBUG_DVBSUB
1805         msg_Dbg( p_enc, "best palette has %d colors", p_fmt->p_palette->i_entries );
1806 #endif
1807
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++ )
1812         {
1813             pi_delta[ i ] = 0;
1814         }
1815 #endif
1816
1817         /* Fill image with our new colours */
1818         for( p = 0; p < p_region->picture.p[0].i_visible_lines ; p++ )
1819         {
1820             int i_ydelta = 0, i_udelta = 0, i_vdelta = 0, i_adelta = 0;
1821
1822             for( n = 0; n < p_region->picture.p[0].i_pitch ; n++ )
1823             {
1824                 int i_offset = p * p_region->picture.p[0].i_pitch + n;
1825                 int y, u, v, a;
1826                 int i_mindist, i_best;
1827
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];
1832
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;
1839 #else
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 ];
1844 #endif
1845
1846                 /* Find best colour in palette */
1847                 for( i_mindist = 99999999, i_best = 0, j = 0; j < p_fmt->p_palette->i_entries; j++ )
1848                 {
1849                     int i_dist = 0;
1850
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);
1855
1856                     if( i_dist < i_mindist )
1857                     {
1858                         i_mindist = i_dist;
1859                         i_best = j;
1860                     }
1861                 }
1862
1863                 /* Set pixel to best color */
1864                 p_region->picture.p[0].p_pixels[i_offset] = i_best;
1865
1866                 /* Update dithering state */
1867 #ifdef RANDOM_DITHERING
1868                 i_seed = (i_seed * 0x1283837) ^ 0x789479 ^ (i_seed >> 13);
1869 #else
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;
1882 #endif
1883             }
1884         }
1885 #ifndef RANDOM_DITHERING
1886         free( pi_delta );
1887 #endif
1888
1889         /* pad palette */
1890         for( i = p_fmt->p_palette->i_entries; i < i_max_entries; i++ )
1891         {
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;
1896         }
1897         p_fmt->p_palette->i_entries = i_max_entries;
1898 #ifdef DEBUG_DVBSUB
1899         msg_Dbg( p_enc, "best palette has %d colors", p_fmt->p_palette->i_entries );
1900 #endif
1901     }
1902     return p_subpic;
1903 } /* End of hack */
1904
1905 /****************************************************************************
1906  * Encode: the whole thing
1907  ****************************************************************************/
1908 static block_t *Encode( encoder_t *p_enc, subpicture_t *p_subpic )
1909 {
1910     subpicture_t *p_temp = NULL;
1911     subpicture_region_t *p_region = NULL;
1912     bs_t bits, *s = &bits;
1913     block_t *p_block;
1914
1915     if( !p_subpic || !p_subpic->p_region ) return NULL;
1916
1917     /* FIXME: this is a hack to convert VLC_FOURCC('Y','U','V','A') into
1918      *  VLC_FOURCC('Y','U','V','P')
1919      */
1920     p_region = p_subpic->p_region;
1921     if( p_region->fmt.i_chroma == VLC_FOURCC('Y','U','V','A') )
1922     {
1923         p_temp = YuvaYuvp( p_enc, p_subpic );
1924         if( !p_temp )
1925         {
1926             msg_Err( p_enc, "no picture in subpicture" );
1927             return NULL;
1928         }
1929         p_region = p_subpic->p_region;
1930     }
1931
1932     /* Sanity check */
1933     if( !p_region ) return NULL;
1934
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') ) )
1937     {
1938         char psz_fourcc[5];
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 );
1942         return NULL;
1943     }
1944
1945     if( p_region->fmt.p_palette )
1946     {
1947         switch( p_region->fmt.p_palette->i_entries )
1948         {
1949             case 0:
1950             case 4:
1951             case 16:
1952             case 256:
1953                 break;
1954             default:
1955                 msg_Err( p_enc, "subpicture palette (%d) not handled",
1956                             p_region->fmt.p_palette->i_entries );
1957                 return NULL;
1958         }
1959     }
1960     /* End of hack */
1961
1962 #ifdef DEBUG_DVBSUB
1963     msg_Dbg( p_enc, "encoding subpicture" );
1964 #endif
1965     p_block = block_New( p_enc, 64000 );
1966     bs_init( s, p_block->p_buffer, p_block->i_buffer );
1967
1968     bs_write( s, 8, 0x20 ); /* Data identifier */
1969     bs_write( s, 8, 0x0 );  /* Subtitle stream id */
1970
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 );
1975
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 */
1981
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 ) )
1986     {
1987         block_t *p_block_stop;
1988
1989         p_block->i_length = p_subpic->i_stop - p_subpic->i_start;
1990
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; */
2006     }
2007 #ifdef DEBUG_DVBSUB
2008     msg_Dbg( p_enc, "subpicture encoded properly" );
2009 #endif
2010     return p_block;
2011 }
2012
2013 /*****************************************************************************
2014  * CloseEncoder: encoder destruction
2015  *****************************************************************************/
2016 static void CloseEncoder( vlc_object_t *p_this )
2017 {
2018     encoder_t *p_enc = (encoder_t *)p_this;
2019     encoder_sys_t *p_sys = p_enc->p_sys;
2020
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" );
2024
2025     if( p_sys->i_regions ) free( p_sys->p_regions );
2026     free( p_sys );
2027 }
2028
2029 static void encode_page_composition( encoder_t *p_enc, bs_t *s,
2030                                      subpicture_t *p_subpic )
2031 {
2032     encoder_sys_t *p_sys = p_enc->p_sys;
2033     subpicture_region_t *p_region;
2034     vlc_bool_t b_mode_change = VLC_FALSE;
2035     int i_regions, i_timeout;
2036
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 */
2040
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++ )
2043     {
2044         if( i_regions >= p_sys->i_regions )
2045         {
2046             encoder_region_t region;
2047             region.i_width = region.i_height = 0;
2048             p_sys->p_regions =
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;
2052         }
2053
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 ) )
2058         {
2059             b_mode_change = VLC_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;
2065         }
2066         if( p_sys->p_regions[i_regions].i_height <
2067              (int)p_region->fmt.i_visible_height )
2068         {
2069             b_mode_change = VLC_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;
2075         }
2076     }
2077
2078     bs_write( s, 16, i_regions * 6 + 2 ); /* Segment length */
2079
2080     i_timeout = 0;
2081     if( p_subpic && !p_subpic->b_ephemer &&
2082         ( p_subpic->i_stop > p_subpic->i_start ) )
2083     {
2084         i_timeout = (p_subpic->i_stop - p_subpic->i_start) / 1000000;
2085     }
2086
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 */
2092
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++ )
2095     {
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) )
2099         {
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 */
2102         }
2103         else
2104         {
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 );
2107         }
2108     }
2109 }
2110
2111 static void encode_clut( encoder_t *p_enc, bs_t *s, subpicture_t *p_subpic )
2112 {
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;
2116     int i;
2117
2118     /* Sanity check */
2119     if( !p_region ) return;
2120
2121     if( p_region->fmt.i_chroma == VLC_FOURCC('Y','U','V','P') )
2122     {
2123         p_pal = p_region->fmt.p_palette;
2124     }
2125     else
2126     {
2127         pal.i_entries = 4;
2128         for( i = 0; i < 4; i++ )
2129         {
2130             pal.palette[i][0] = 0;
2131             pal.palette[i][1] = 0;
2132             pal.palette[i][2] = 0;
2133             pal.palette[i][3] = 0;
2134         }
2135         p_pal = &pal;
2136     }
2137
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 */
2141
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 */
2146
2147     for( i = 0; i < p_pal->i_entries; i++ )
2148     {
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 */
2160     }
2161 }
2162
2163 static void encode_region_composition( encoder_t *p_enc, bs_t *s,
2164                                        subpicture_t *p_subpic )
2165 {
2166     encoder_sys_t *p_sys = p_enc->p_sys;
2167     subpicture_region_t *p_region;
2168     int i_region;
2169
2170     for( i_region = 0, p_region = p_subpic->p_region; p_region;
2171          p_region = p_region->p_next, i_region++ )
2172     {
2173         int i_entries = 4, i_depth = 0x1, i_bg = 0;
2174         vlc_bool_t b_text =
2175             ( p_region->fmt.i_chroma == VLC_FOURCC('T','E','X','T') );
2176
2177         if( !b_text )
2178         {
2179             video_palette_t *p_pal = p_region->fmt.p_palette;
2180
2181             if( !p_pal )
2182             {
2183                 msg_Err( p_enc, "subpicture has no palette - ignoring it" );
2184                 break;
2185             }
2186
2187             i_entries = p_pal->i_entries;
2188             i_depth = i_entries == 4 ? 0x1 : i_entries == 16 ? 0x2 : 0x3;
2189
2190             for( i_bg = 0; i_bg < p_pal->i_entries; i_bg++ )
2191             {
2192                 if( !p_pal->palette[i_bg][3] ) break;
2193             }
2194         }
2195
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 */
2199
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++ );
2203
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 */
2217
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 */
2225
2226         if( b_text )
2227         {
2228             bs_write( s, 8, 1 ); /* foreground pixel code */
2229             bs_write( s, 8, 0 ); /* background pixel code */
2230         }
2231     }
2232 }
2233
2234 static void encode_pixel_data( encoder_t *p_enc, bs_t *s,
2235                                subpicture_region_t *p_region,
2236                                vlc_bool_t b_top );
2237
2238 static void encode_object( encoder_t *p_enc, bs_t *s, subpicture_t *p_subpic )
2239 {
2240     encoder_sys_t *p_sys = p_enc->p_sys;
2241     subpicture_region_t *p_region;
2242     int i_region;
2243
2244     int i_length_pos, i_update_pos, i_pixel_data_pos;
2245
2246     for( i_region = 0, p_region = p_subpic->p_region; p_region;
2247          p_region = p_region->p_next, i_region++ )
2248     {
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 */
2252
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++ );
2257
2258         /* object coding method */
2259         switch( p_region->fmt.i_chroma )
2260         {
2261         case VLC_FOURCC( 'Y','U','V','P' ):
2262             bs_write( s, 2, 0 );
2263             break;
2264         case VLC_FOURCC( 'T','E','X','T' ):
2265             bs_write( s, 2, 1 );
2266             break;
2267         default:
2268             msg_Err( p_enc, "FOURCC %d not supported by encoder.", p_region->fmt.i_chroma );
2269             continue;
2270         }
2271
2272         bs_write( s, 1, 0 ); /* non modifying color flag */
2273         bs_write( s, 1, 0 ); /* Reserved */
2274
2275         if( p_region->fmt.i_chroma == VLC_FOURCC( 'T','E','X','T' ) )
2276         {
2277             int i_size, i;
2278
2279             if( !p_region->psz_text ) continue;
2280
2281             i_size = __MIN( strlen( p_region->psz_text ), 256 );
2282
2283             bs_write( s, 8, i_size ); /* number of characters in string */
2284             for( i = 0; i < i_size; i++ )
2285             {
2286                 bs_write( s, 16, p_region->psz_text[i] );
2287             }
2288
2289             /* Update segment length */
2290             SetWBE( &s->p_start[i_length_pos/8],
2291                     (bs_pos(s) - i_length_pos)/8 -2 );
2292             continue;
2293         }
2294
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 */
2299
2300         /* Top field */
2301         i_pixel_data_pos = bs_pos( s );
2302         encode_pixel_data( p_enc, s, p_region, VLC_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 );
2305
2306         /* Bottom field */
2307         i_pixel_data_pos = bs_pos( s );
2308         encode_pixel_data( p_enc, s, p_region, VLC_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 );
2311
2312         /* Stuffing for word alignment */
2313         bs_align_0( s );
2314         if( bs_pos( s ) % 16 ) bs_write( s, 8, 0 );
2315
2316         /* Update segment length */
2317         SetWBE( &s->p_start[i_length_pos/8], (bs_pos(s) - i_length_pos)/8 -2 );
2318     }
2319 }
2320
2321 static void encode_pixel_line_2bp( encoder_t *p_enc, bs_t *s,
2322                                    subpicture_region_t *p_region,
2323                                    int i_line );
2324 static void encode_pixel_line_4bp( encoder_t *p_enc, bs_t *s,
2325                                    subpicture_region_t *p_region,
2326                                    int i_line );
2327 static void encode_pixel_line_8bp( encoder_t *p_enc, bs_t *s,
2328                                    subpicture_region_t *p_region,
2329                                    int i_line );
2330 static void encode_pixel_data( encoder_t *p_enc, bs_t *s,
2331                                subpicture_region_t *p_region,
2332                                vlc_bool_t b_top )
2333 {
2334     unsigned int i_line;
2335
2336     /* Sanity check */
2337     if( p_region->fmt.i_chroma != VLC_FOURCC('Y','U','V','P') ) return;
2338
2339     /* Encode line by line */
2340     for( i_line = !b_top; i_line < p_region->fmt.i_visible_height;
2341          i_line += 2 )
2342     {
2343         switch( p_region->fmt.p_palette->i_entries )
2344         {
2345         case 0:
2346             break;
2347
2348         case 4:
2349             bs_write( s, 8, 0x10 ); /* 2 bit/pixel code string */
2350             encode_pixel_line_2bp( p_enc, s, p_region, i_line );
2351             break;
2352
2353         case 16:
2354             bs_write( s, 8, 0x11 ); /* 4 bit/pixel code string */
2355             encode_pixel_line_4bp( p_enc, s, p_region, i_line );
2356             break;
2357
2358         case 256:
2359             bs_write( s, 8, 0x12 ); /* 8 bit/pixel code string */
2360             encode_pixel_line_8bp( p_enc, s, p_region, i_line );
2361             break;
2362
2363         default:
2364             msg_Err( p_enc, "subpicture palette (%i) not handled",
2365                      p_region->fmt.p_palette->i_entries );
2366             break;
2367         }
2368
2369         bs_write( s, 8, 0xf0 ); /* End of object line code */
2370     }
2371 }
2372
2373 static void encode_pixel_line_2bp( encoder_t *p_enc, bs_t *s,
2374                                    subpicture_region_t *p_region,
2375                                    int i_line )
2376 {
2377     unsigned int i, i_length = 0;
2378     int i_pitch = p_region->picture.p->i_pitch;
2379     uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2380     int i_last_pixel = p_data[0];
2381
2382     for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2383     {
2384         if( ( i != p_region->fmt.i_visible_width ) &&
2385             ( p_data[i] == i_last_pixel ) && ( i_length != 284 ) )
2386         {
2387             i_length++;
2388             continue;
2389         }
2390
2391         if( ( i_length == 1 ) || ( i_length == 11 ) || ( i_length == 28 ) )
2392         {
2393             /* 2bit/pixel code */
2394             if( i_last_pixel )
2395                 bs_write( s, 2, i_last_pixel );
2396             else
2397             {
2398                 bs_write( s, 2, 0 );
2399                 bs_write( s, 1, 0 );
2400                 bs_write( s, 1, 1 ); /* pseudo color 0 */
2401             }
2402             i_length--;
2403         }
2404
2405         if( i_length == 2 )
2406         {
2407             if( i_last_pixel )
2408             {
2409                 bs_write( s, 2, i_last_pixel );
2410                 bs_write( s, 2, i_last_pixel );
2411             }
2412             else
2413             {
2414                 bs_write( s, 2, 0 );
2415                 bs_write( s, 1, 0 );
2416                 bs_write( s, 1, 0 );
2417                 bs_write( s, 2, 1 ); /* 2 * pseudo color 0 */
2418             }
2419         }
2420         else if( i_length > 2 )
2421         {
2422             bs_write( s, 2, 0 );
2423             if( i_length <= 10 )
2424             {
2425                 bs_write( s, 1, 1 );
2426                 bs_write( s, 3, i_length - 3 );
2427                 bs_write( s, 2, i_last_pixel );
2428             }
2429             else
2430             {
2431                 bs_write( s, 1, 0 );
2432                 bs_write( s, 1, 0 );
2433
2434                 if( i_length <= 27 )
2435                 {
2436                     bs_write( s, 2, 2 );
2437                     bs_write( s, 4, i_length - 12 );
2438                     bs_write( s, 2, i_last_pixel );
2439                 }
2440                 else
2441                 {
2442                     bs_write( s, 2, 3 );
2443                     bs_write( s, 8, i_length - 29 );
2444                     bs_write( s, 2, i_last_pixel );
2445                 }
2446             }
2447         }
2448
2449         if( i == p_region->fmt.i_visible_width ) break;
2450
2451         i_last_pixel = p_data[i];
2452         i_length = 1;
2453     }
2454
2455     /* Stop */
2456     bs_write( s, 2, 0 );
2457     bs_write( s, 1, 0 );
2458     bs_write( s, 1, 0 );
2459     bs_write( s, 2, 0 );
2460
2461     /* Stuffing */
2462     bs_align_0( s );
2463 }
2464
2465 static void encode_pixel_line_4bp( encoder_t *p_enc, bs_t *s,
2466                                    subpicture_region_t *p_region,
2467                                    int i_line )
2468 {
2469     unsigned int i, i_length = 0;
2470     int i_pitch = p_region->picture.p->i_pitch;
2471     uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2472     int i_last_pixel = p_data[0];
2473
2474     for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2475     {
2476         if( i != p_region->fmt.i_visible_width &&
2477             p_data[i] == i_last_pixel && i_length != 280 )
2478         {
2479             i_length++;
2480             continue;
2481         }
2482
2483         if( ( i_length == 1 ) ||
2484             ( ( i_length == 3 ) && i_last_pixel ) ||
2485             ( i_length == 8 ) )
2486         {
2487             /* 4bit/pixel code */
2488             if( i_last_pixel )
2489                 bs_write( s, 4, i_last_pixel );
2490             else
2491             {
2492                 bs_write( s, 4, 0 );
2493                 bs_write( s, 1, 1 );
2494                 bs_write( s, 1, 1 );
2495                 bs_write( s, 2, 0 ); /* pseudo color 0 */
2496             }
2497             i_length--;
2498         }
2499
2500         if( i_length == 2 )
2501         {
2502             if( i_last_pixel )
2503             {
2504                 bs_write( s, 4, i_last_pixel );
2505                 bs_write( s, 4, i_last_pixel );
2506             }
2507             else
2508             {
2509                 bs_write( s, 4, 0 );
2510                 bs_write( s, 1, 1 );
2511                 bs_write( s, 1, 1 );
2512                 bs_write( s, 2, 1 ); /* 2 * pseudo color 0 */
2513             }
2514         }
2515         else if( !i_last_pixel && ( i_length >= 3 ) && ( i_length <= 9 ) )
2516         {
2517             bs_write( s, 4, 0 );
2518             bs_write( s, 1, 0 );
2519             bs_write( s, 3, i_length - 2 ); /* (i_length - 2) * color 0 */
2520         }
2521         else if( i_length > 2 )
2522         {
2523             bs_write( s, 4, 0 );
2524             bs_write( s, 1, 1 );
2525
2526             if( i_length <= 7 )
2527             {
2528                 bs_write( s, 1, 0 );
2529                 bs_write( s, 2, i_length - 4 );
2530                 bs_write( s, 4, i_last_pixel );
2531             }
2532             else
2533             {
2534                 bs_write( s, 1, 1 );
2535
2536                 if( i_length <= 24 )
2537                 {
2538                     bs_write( s, 2, 2 );
2539                     bs_write( s, 4, i_length - 9 );
2540                     bs_write( s, 4, i_last_pixel );
2541                 }
2542                 else
2543                 {
2544                     bs_write( s, 2, 3 );
2545                     bs_write( s, 8, i_length - 25 );
2546                     bs_write( s, 4, i_last_pixel );
2547                 }
2548             }
2549         }
2550
2551         if( i == p_region->fmt.i_visible_width ) break;
2552
2553         i_last_pixel = p_data[i];
2554         i_length = 1;
2555     }
2556
2557     /* Stop */
2558     bs_write( s, 8, 0 );
2559
2560     /* Stuffing */
2561     bs_align_0( s );
2562 }
2563
2564 static void encode_pixel_line_8bp( encoder_t *p_enc, bs_t *s,
2565                                    subpicture_region_t *p_region,
2566                                    int i_line )
2567 {
2568     unsigned int i, i_length = 0;
2569     int i_pitch = p_region->picture.p->i_pitch;
2570     uint8_t *p_data = &p_region->picture.p->p_pixels[ i_pitch * i_line ];
2571     int i_last_pixel = p_data[0];
2572
2573     for( i = 0; i <= p_region->fmt.i_visible_width; i++ )
2574     {
2575         if( ( i != p_region->fmt.i_visible_width ) &&
2576             ( p_data[i] == i_last_pixel ) && ( i_length != 127 ) )
2577         {
2578             i_length++;
2579             continue;
2580         }
2581
2582         if( ( i_length == 1 ) && i_last_pixel )
2583         {
2584             /* 8bit/pixel code */
2585             bs_write( s, 8, i_last_pixel );
2586         }
2587         else if( ( i_length == 2 ) && i_last_pixel )
2588         {
2589             /* 8bit/pixel code */
2590             bs_write( s, 8, i_last_pixel );
2591             bs_write( s, 8, i_last_pixel );
2592         }
2593         else if( i_length <= 127 )
2594         {
2595             bs_write( s, 8, 0 );
2596
2597             if( !i_last_pixel )
2598             {
2599                 bs_write( s, 1, 0 );
2600                 bs_write( s, 7, i_length ); /* pseudo color 0 */
2601             }
2602             else
2603             {
2604                 bs_write( s, 1, 1 );
2605                 bs_write( s, 7, i_length );
2606                 bs_write( s, 8, i_last_pixel );
2607             }
2608         }
2609
2610         if( i == p_region->fmt.i_visible_width ) break;
2611
2612         i_last_pixel = p_data[i];
2613         i_length = 1;
2614     }
2615
2616     /* Stop */
2617     bs_write( s, 8, 0 );
2618     bs_write( s, 8, 0 );
2619
2620     /* Stuffing */
2621     bs_align_0( s );
2622 }