modules/stream_out/transcode/spu.c

   1 /*****************************************************************************
   2  * spu.c: transcoding stream output module (spu)
   3  *****************************************************************************
   4  * Copyright (C) 2003-2009 VLC authors and VideoLAN
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Gildas Bazin <gbazin@videolan.org>
   9  *          Jean-Paul Saman <jpsaman #_at_# m2x dot nl>
  10  *          Antoine Cellerier <dionoea at videolan dot org>
  11  *
  12  * This program is free software; you can redistribute it and/or modify it
  13  * under the terms of the GNU Lesser General Public License as published by
  14  * the Free Software Foundation; either version 2.1 of the License, or
  15  * (at your option) any later version.
  16  *
  17  * This program is distributed in the hope that it will be useful,
  18  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  20  * GNU Lesser General Public License for more details.
  21  *
  22  * You should have received a copy of the GNU Lesser General Public License
  23  * along with this program; if not, write to the Free Software Foundation,
  24  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  25  *****************************************************************************/
  26
  27 /*****************************************************************************
  28  * Preamble
  29  *****************************************************************************/
  30
  31 #include "transcode.h"
  32
  33 #include <vlc_meta.h>
  34 #include <vlc_spu.h>
  35 #include <vlc_modules.h>
  36 #include <assert.h>
  37
  38 static subpicture_t *spu_new_buffer( decoder_t *p_dec,
  39                                      const subpicture_updater_t *p_upd )
  40 {
  41     VLC_UNUSED( p_dec );
  42     subpicture_t *p_subpicture = subpicture_New( p_upd );
  43     if( likely(p_subpicture != NULL) )
  44         p_subpicture->b_subtitle = true;
  45     return p_subpicture;
  46 }
  47
  48 static int decoder_queue_sub( decoder_t *p_dec, subpicture_t *p_spu )
  49 {
  50     sout_stream_id_sys_t *id = p_dec->p_queue_ctx;
  51
  52     vlc_mutex_lock(&id->fifo.lock);
  53     *id->fifo.spu.last = p_spu;
  54     id->fifo.spu.last = &p_spu->p_next;
  55     vlc_mutex_unlock(&id->fifo.lock);
  56     return 0;
  57 }
  58
  59 static subpicture_t *transcode_dequeue_all_subs( sout_stream_id_sys_t *id )
  60 {
  61     vlc_mutex_lock(&id->fifo.lock);
  62     subpicture_t *p_subpics = id->fifo.spu.first;
  63     id->fifo.spu.first = NULL;
  64     id->fifo.spu.last = &id->fifo.spu.first;
  65     vlc_mutex_unlock(&id->fifo.lock);
  66
  67     return p_subpics;
  68 }
  69
  70 static int transcode_spu_new( sout_stream_t *p_stream, sout_stream_id_sys_t *id )
  71 {
  72     sout_stream_sys_t *p_sys = p_stream->p_sys;
  73
  74     /*
  75      * Open decoder
  76      */
  77
  78     /* Initialization of decoder structures */
  79     id->p_decoder->pf_decode = NULL;
  80     id->p_decoder->pf_spu_buffer_new = spu_new_buffer;
  81     id->p_decoder->pf_queue_sub = decoder_queue_sub;
  82     id->p_decoder->p_queue_ctx = id;
  83     /* id->p_decoder->p_cfg = p_sys->p_spu_cfg; */
  84
  85     id->p_decoder->p_module =
  86         module_need_var( id->p_decoder, "spu decoder", "codec" );
  87
  88     if( !id->p_decoder->p_module )
  89     {
  90         msg_Err( p_stream, "cannot find spu decoder" );
  91         return VLC_EGENERIC;
  92     }
  93
  94     if( !p_sys->b_soverlay )
  95     {
  96         /* Open encoder */
  97         /* Initialization of encoder format structures */
  98         es_format_Init( &id->p_encoder->fmt_in, id->p_decoder->fmt_in.i_cat,
  99                         id->p_decoder->fmt_in.i_codec );
 100
 101         id->p_encoder->p_cfg = p_sys->p_spu_cfg;
 102
 103         id->p_encoder->p_module =
 104             module_need( id->p_encoder, "encoder", p_sys->psz_senc, true );
 105
 106         if( !id->p_encoder->p_module )
 107         {
 108             module_unneed( id->p_decoder, id->p_decoder->p_module );
 109             msg_Err( p_stream, "cannot find spu encoder (%s)", p_sys->psz_senc );
 110             return VLC_EGENERIC;
 111         }
 112     }
 113
 114     if( !p_sys->p_spu )
 115         p_sys->p_spu = spu_Create( p_stream, NULL );
 116
 117     return VLC_SUCCESS;
 118 }
 119
 120 void transcode_spu_close( sout_stream_t *p_stream, sout_stream_id_sys_t *id)
 121 {
 122     sout_stream_sys_t *p_sys = p_stream->p_sys;
 123     /* Close decoder */
 124     if( id->p_decoder->p_module )
 125         module_unneed( id->p_decoder, id->p_decoder->p_module );
 126     if( id->p_decoder->p_description )
 127         vlc_meta_Delete( id->p_decoder->p_description );
 128
 129     /* Close encoder */
 130     if( id->p_encoder->p_module )
 131         module_unneed( id->p_encoder, id->p_encoder->p_module );
 132
 133     if( p_sys->p_spu )
 134     {
 135         spu_Destroy( p_sys->p_spu );
 136         p_sys->p_spu = NULL;
 137     }
 138 }
 139
 140 int transcode_spu_process( sout_stream_t *p_stream,
 141                                   sout_stream_id_sys_t *id,
 142                                   block_t *in, block_t **out )
 143 {
 144     sout_stream_sys_t *p_sys = p_stream->p_sys;
 145     *out = NULL;
 146     bool b_error = false;
 147
 148     int ret = id->p_decoder->pf_decode( id->p_decoder, in );
 149     if( ret != VLCDEC_SUCCESS )
 150         return VLC_EGENERIC;
 151
 152     subpicture_t *p_subpics = transcode_dequeue_all_subs( id );
 153     if( p_subpics == NULL )
 154         return VLC_SUCCESS;
 155     do
 156     {
 157         subpicture_t *p_subpic = p_subpics;
 158         p_subpics = p_subpics->p_next;
 159         p_subpic->p_next = NULL;
 160
 161         if( b_error )
 162         {
 163             subpicture_Delete( p_subpic );
 164             continue;
 165         }
 166
 167         if( p_sys->b_master_sync && p_sys->i_master_drift )
 168         {
 169             p_subpic->i_start -= p_sys->i_master_drift;
 170             if( p_subpic->i_stop ) p_subpic->i_stop -= p_sys->i_master_drift;
 171         }
 172
 173         if( p_sys->b_soverlay )
 174             spu_PutSubpicture( p_sys->p_spu, p_subpic );
 175         else
 176         {
 177             block_t *p_block;
 178
 179             es_format_t fmt;
 180             es_format_Init( &fmt, VIDEO_ES, VLC_CODEC_TEXT );
 181
 182             fmt.video.i_sar_num =
 183             fmt.video.i_visible_width =
 184             fmt.video.i_width = p_sys->i_spu_width;
 185
 186             fmt.video.i_sar_den =
 187             fmt.video.i_visible_height =
 188             fmt.video.i_height = p_sys->i_spu_height;
 189
 190             subpicture_Update( p_subpic, &fmt.video, &fmt.video, p_subpic->i_start );
 191             es_format_Clean( &fmt );
 192
 193             p_block = id->p_encoder->pf_encode_sub( id->p_encoder, p_subpic );
 194             subpicture_Delete( p_subpic );
 195             if( p_block )
 196                 block_ChainAppend( out, p_block );
 197             else
 198                 b_error = true;
 199         }
 200     } while( p_subpics );
 201
 202     return b_error ? VLC_EGENERIC : VLC_SUCCESS;
 203 }
 204
 205 bool transcode_spu_add( sout_stream_t *p_stream, const es_format_t *p_fmt,
 206                         sout_stream_id_sys_t *id )
 207 {
 208     sout_stream_sys_t *p_sys = p_stream->p_sys;
 209
 210     id->fifo.spu.first = NULL;
 211     id->fifo.spu.last = &id->fifo.spu.first;
 212
 213     if( p_sys->i_scodec )
 214     {
 215         msg_Dbg( p_stream, "creating subtitle transcoding from fcc=`%4.4s' "
 216                  "to fcc=`%4.4s'", (char*)&p_fmt->i_codec,
 217                  (char*)&p_sys->i_scodec );
 218
 219         /* Complete destination format */
 220         id->p_encoder->fmt_out.i_codec = p_sys->i_scodec;
 221
 222         /* build decoder -> filter -> encoder */
 223         if( transcode_spu_new( p_stream, id ) )
 224         {
 225             msg_Err( p_stream, "cannot create subtitle chain" );
 226             return false;
 227         }
 228
 229         /* open output stream */
 230         id->id = sout_StreamIdAdd( p_stream->p_next, &id->p_encoder->fmt_out );
 231         id->b_transcode = true;
 232
 233         if( !id->id )
 234         {
 235             transcode_spu_close( p_stream, id );
 236             return false;
 237         }
 238     }
 239     else
 240     {
 241         assert( p_sys->b_soverlay );
 242         msg_Dbg( p_stream, "subtitle (fcc=`%4.4s') overlaying",
 243                  (char*)&p_fmt->i_codec );
 244
 245         id->b_transcode = true;
 246
 247         /* Build decoder -> filter -> overlaying chain */
 248         if( transcode_spu_new( p_stream, id ) )
 249         {
 250             msg_Err( p_stream, "cannot create subtitle chain" );
 251             return false;
 252         }
 253     }
 254
 255     return true;
 256 }