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

   1 /*****************************************************************************
   2  * zvbi.c : VBI and Teletext PES demux and decoder using libzvbi
   3  *****************************************************************************
   4  * Copyright (C) 2007, M2X
   5  * $Id$
   6  *
   7  * Authors: Derk-Jan Hartman <djhartman at m2x dot nl> for M2X
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23 /*****************************************************************************
  24  *
  25  * information on teletext format can be found here :
  26  * http://pdc.ro.nu/teletext.html
  27  *
  28  *****************************************************************************/
  29
  30 /* This module implements:
  31  * ETSI EN 301 775: VBI data in PES
  32  * ETSI EN 300 472: EBU Teletext data in PES
  33  * ETSI EN 300 706: Enhanced Teletext (libzvbi)
  34  * ETSI EN 300 231: Video Programme System [VPS] (libzvbi)
  35  * ETSI EN 300 294: 625-line Wide Screen Signaling [WSS] (libzvbi)
  36  * EIA-608 Revision A: Closed Captioning [CC] (libzvbi)
  37  */
  38
  39 #include <vlc/vlc.h>
  40 #include <assert.h>
  41 #include <stdint.h>
  42 #include <libzvbi.h>
  43
  44 #include "vlc_vout.h"
  45 #include "vlc_bits.h"
  46 #include "vlc_codec.h"
  47
  48 typedef enum {
  49     DATA_UNIT_EBU_TELETEXT_NON_SUBTITLE     = 0x02,
  50     DATA_UNIT_EBU_TELETEXT_SUBTITLE         = 0x03,
  51     DATA_UNIT_EBU_TELETEXT_INVERTED         = 0x0C,
  52
  53     DATA_UNIT_ZVBI_WSS_CPR1204              = 0xB4,
  54     DATA_UNIT_ZVBI_CLOSED_CAPTION_525       = 0xB5,
  55     DATA_UNIT_ZVBI_MONOCHROME_SAMPLES_525   = 0xB6,
  56
  57     DATA_UNIT_VPS                           = 0xC3,
  58     DATA_UNIT_WSS                           = 0xC4,
  59     DATA_UNIT_CLOSED_CAPTION                = 0xC5,
  60     DATA_UNIT_MONOCHROME_SAMPLES            = 0xC6,
  61
  62     DATA_UNIT_STUFFING                      = 0xFF,
  63 } data_unit_id;
  64
  65 /*****************************************************************************
  66  * Module descriptor.
  67  *****************************************************************************/
  68 static int  Open ( vlc_object_t * );
  69 static void Close( vlc_object_t * );
  70 static subpicture_t *Decode( decoder_t *, block_t ** );
  71
  72 #define PAGE_TEXT N_("Teletext page")
  73 #define PAGE_LONGTEXT N_("Open the indicated Teletext page." \
  74         "Default page is index 100")
  75
  76 vlc_module_begin();
  77     set_description( _("VBI and Teletext decoder") );
  78     set_shortname( "VBI & Teletext" );
  79     set_capability( "decoder", 51 );
  80     set_category( CAT_INPUT );
  81     set_subcategory( SUBCAT_INPUT_SCODEC );
  82     set_callbacks( Open, Close );
  83
  84     add_integer( "vbi-page", 100, NULL,
  85                  PAGE_TEXT, PAGE_LONGTEXT, VLC_FALSE );
  86 vlc_module_end();
  87
  88 /****************************************************************************
  89  * Local structures
  90  ****************************************************************************/
  91
  92 struct decoder_sys_t
  93 {
  94    vbi_decoder *           p_vbi_dec;
  95    vbi_dvb_demux *         p_dvb_demux;
  96    unsigned int            i_wanted_page;
  97    unsigned int            i_last_page;
  98    vlc_bool_t              b_update;
  99 }
 100
 101 static void event_handler( vbi_event *ev, void *user_data );
 102 static int RequestPage( vlc_object_t *p_this, char const *psz_cmd,
 103                         vlc_value_t oldval, vlc_value_t newval, void *p_data );
 104
 105 /*****************************************************************************
 106  * Open: probe the decoder and return score
 107  *****************************************************************************
 108  * Tries to launch a decoder and return score so that the interface is able
 109  * to chose.
 110  *****************************************************************************/
 111 static int Open( vlc_object_t *p_this )
 112 {
 113     decoder_t     *p_dec = (decoder_t *) p_this;
 114     decoder_sys_t *p_sys = NULL;
 115     vlc_value_t    val;
 116     int            i;
 117
 118     if( p_dec->fmt_in.i_codec != VLC_FOURCC('t','e','l','x') )
 119     {
 120         return VLC_EGENERIC;
 121     }
 122
 123     p_dec->pf_decode_sub = Decode;
 124     p_sys = p_dec->p_sys = malloc( sizeof(decoder_sys_t) );
 125     if( p_sys == NULL )
 126     {
 127         msg_Err( p_dec, "out of memory" );
 128         return VLC_ENOMEM;
 129     }
 130     memset( p_sys, 0, sizeof(decoder_sys_t) );
 131
 132     p_sys->p_vbi_dec = vbi_decoder_new();
 133     p_sys->p_dvb_demux = vbi_dvb_pes_demux_new( NULL, NULL );
 134
 135     if( (p_sys->p_vbi_dec == NULL) || (p_sys->p_dvb_demux == NULL) )
 136     {
 137         msg_Err( p_dec, "VBI decoder/demux could not be created." );
 138         return VLC_ENOMEM;
 139     }
 140
 141     vbi_event_handler_register( p_sys->p_vbi_dec, VBI_EVENT_TTX_PAGE |
 142                                 VBI_EVENT_CAPTION | VBI_EVENT_NETWORK |
 143                                 VBI_EVENT_ASPECT | VBI_EVENT_PROG_INFO,
 144                                 event_handler, p_dec );
 145
 146     /* Create the var on vlc_global. */
 147     p_sys->i_wanted_page = var_CreateGetInteger( p_dec->p_libvlc_global,
 148                                                  "vbi-page" );
 149     var_AddCallback( p_dec->p_libvlc_global, "vbi-page",
 150                      RequestPage, p_sys );
 151     return VLC_SUCCESS;
 152 }
 153
 154 /*****************************************************************************
 155  * Close:
 156  *****************************************************************************/
 157 static void Close( vlc_object_t *p_this )
 158 {
 159     decoder_t     *p_dec = (decoder_t*) p_this;
 160     decoder_sys_t *p_sys = p_dec->p_sys;
 161
 162     if( p_sys->p_vbi_dec ) vbi_decoder_delete( p_sys->p_vbi_dec );
 163     if( p_sys->p_dvb_demux ) vbi_dvb_demux_delete( p_sys->p_dvb_demux );
 164     free( p_sys );
 165 }
 166
 167 #define MAX_SLICES 32
 168
 169 /*****************************************************************************
 170  * Decode:
 171  *****************************************************************************/
 172 static subpicture_t *Decode( decoder_t *p_dec, block_t **pp_block )
 173 {
 174     decoder_sys_t   *p_sys = p_dec->p_sys;
 175     block_t         *p_block;
 176     subpicture_t    *p_spu = NULL;
 177     video_format_t  fmt;
 178     vlc_bool_t      b_cached = VLC_FALSE;
 179     vbi_page        p_page;
 180     uint8_t         *p_pos;
 181     unsigned int    i_left;
 182
 183     if( (pp_block == NULL) || (*pp_block == NULL) )
 184         return NULL;
 185     p_block = *pp_block;
 186     *pp_block = NULL;
 187
 188     p_pos = p_block->p_buffer;
 189     i_left = p_block->i_buffer;
 190
 191     while( i_left > 0 )
 192     {
 193         vbi_sliced      p_sliced[MAX_SLICES];
 194         unsigned int    i_lines = 0;
 195         int64_t         i_pts;
 196
 197         i_lines = vbi_dvb_demux_cor( p_sys->p_dvb_demux, p_sliced,
 198                                      MAX_SLICES, &i_pts, &p_pos, &i_left );
 199
 200         if( i_lines > 0 )
 201             vbi_decode( p_sys->p_vbi_dec, p_sliced, i_lines, i_pts / 90000.0 );
 202     }
 203
 204 stop_decode:
 205     /* Try to see if the page we want is in the cache yet */
 206     b_cached = vbi_fetch_vt_page( p_sys->p_vbi_dec, &p_page,
 207                                   vbi_dec2bcd( p_sys->i_wanted_page ),
 208                                   VBI_ANY_SUBNO, VBI_WST_LEVEL_3p5,
 209                                   25, FALSE );
 210
 211     if( !b_cached )
 212         goto error;
 213
 214     if( (p_sys->i_wanted_page == p_sys->i_last_page) &&
 215          (p_sys->b_update != VLC_TRUE) )
 216         goto error;
 217
 218     p_sys->i_last_page = p_sys->i_wanted_page;
 219     p_sys->b_update = VLC_FALSE;
 220     msg_Dbg( p_dec, "we now have page: %d ready for display",
 221              p_sys->i_wanted_page );
 222
 223     /* If there is a page or sub to render, then we do that here */
 224     /* Create the subpicture unit */
 225     p_spu = p_dec->pf_spu_buffer_new( p_dec );
 226     if( !p_spu )
 227     {
 228         msg_Warn( p_dec, "can't get spu buffer" );
 229         goto error;
 230     }
 231
 232     /* Create a new subpicture region */
 233     memset( &fmt, 0, sizeof(video_format_t) );
 234     fmt.i_chroma = VLC_FOURCC('R','G','B','A');
 235     fmt.i_aspect = VOUT_ASPECT_FACTOR;
 236     fmt.i_sar_num = fmt.i_sar_den = 1;
 237     fmt.i_width = fmt.i_visible_width = p_page.columns * 12;
 238     fmt.i_height = fmt.i_visible_height = p_page.rows * 10;
 239     fmt.i_bits_per_pixel = 32;
 240     fmt.i_x_offset = fmt.i_y_offset = 0;
 241
 242     p_spu->p_region = p_spu->pf_create_region( VLC_OBJECT(p_dec), &fmt );
 243     if( p_spu->p_region == NULL )
 244     {
 245         msg_Err( p_dec, "cannot allocate SPU region" );
 246         goto error;
 247     }
 248
 249     p_spu->p_region->i_x = 0;
 250     p_spu->p_region->i_y = 0;
 251
 252     /* Normal text subs, easy markup */
 253     p_spu->i_flags = SUBPICTURE_ALIGN_TOP;
 254
 255     p_spu->i_start = p_block->i_pts;
 256     p_spu->i_stop = 0;
 257     p_spu->b_ephemer = VLC_TRUE;
 258     p_spu->b_absolute = VLC_FALSE;
 259     p_spu->b_pausable = VLC_TRUE;
 260     p_spu->i_original_picture_width = p_page.columns * 12;
 261     p_spu->i_original_picture_height = p_page.rows * 10;
 262
 263     vbi_draw_vt_page( &p_page, VBI_PIXFMT_RGBA32_LE,
 264                       p_spu->p_region->picture.p->p_pixels, 1, 1 );
 265     p_spu->p_region->picture.p->i_lines = p_page.rows * 10;
 266     p_spu->p_region->picture.p->i_pitch = p_page.columns * 12 * 4;
 267
 268 #if 0
 269     unsigned int i_total, i_textsize = 7000;
 270     char p_text[7000];
 271
 272     i_total =  vbi_print_page_region( &p_page, p_text, i_textsize, "ISO-8859-1", 0, 0, 0, 0, p_page.columns, p_page.rows );
 273     p_text[i_total] = '\0';
 274
 275     msg_Dbg( p_dec, "page %x-%x\n%s", p_page.pgno, p_page.subno, p_text );
 276 #endif
 277
 278     //if( p_page != NULL)
 279         vbi_unref_page( &p_page );
 280
 281     block_Release( p_block );
 282     return p_spu;
 283
 284 error:
 285     //if( p_page != NULL)
 286         vbi_unref_page( &p_page );
 287     if ( p_spu != NULL )
 288     {
 289         p_dec->pf_spu_buffer_del( p_dec, p_spu );
 290         p_spu = NULL;
 291     }
 292
 293     block_Release( p_block );
 294     return NULL;
 295 }
 296
 297 static void event_handler( vbi_event *ev, void *user_data)
 298 {
 299     decoder_t *p_dec        = (decoder_t *)user_data;
 300     decoder_sys_t *p_sys    = p_dec->p_sys;
 301
 302     if( ev->type == VBI_EVENT_TTX_PAGE )
 303     {
 304         /* msg_Dbg( p_dec, "Page %03x.%02x ",
 305                     ev->ev.ttx_page.pgno,
 306                     ev->ev.ttx_page.subno & 0xFF);
 307         */
 308         if( p_sys->i_last_page == vbi_bcd2dec( ev->ev.ttx_page.pgno ) )
 309             p_sys->b_update = VLC_TRUE;
 310
 311         if( ev->ev.ttx_page.clock_update )
 312             msg_Dbg( p_dec, "clock" );
 313         if( ev->ev.ttx_page.header_update )
 314             msg_Dbg( p_dec, "header" );
 315     }
 316     else if( ev->type == VBI_EVENT_CAPTION )
 317         msg_Dbg( p_dec, "Caption line: %x", ev->ev.caption.pgno );
 318     else if( ev->type == VBI_EVENT_NETWORK )
 319         msg_Dbg( p_dec, "Network change" );
 320     else if( ev->type == VBI_EVENT_ASPECT )
 321         msg_Dbg( p_dec, "Aspect update" );
 322     else if( ev->type == VBI_EVENT_NETWORK )
 323         msg_Dbg( p_dec, "Program info received" );
 324 }
 325
 326 static int RequestPage( vlc_object_t *p_this, char const *psz_cmd,
 327                         vlc_value_t oldval, vlc_value_t newval, void *p_data )
 328 {
 329     decoder_sys_t   *p_sys = p_data;
 330
 331     if( (newval.i_int > 0) && (newval.i_int < 999) )
 332         p_sys->i_wanted_page = newval.i_int;
 333
 334     return VLC_SUCCESS;
 335 }