theora.c 22.2 KB
Newer Older
Gildas Bazin's avatar
 
Gildas Bazin committed
1 2 3 4
/*****************************************************************************
 * theora.c: theora decoder module making use of libtheora.
 *****************************************************************************
 * Copyright (C) 1999-2001 VideoLAN
5
 * $Id: theora.c,v 1.11 2003/10/09 11:48:41 gbazin Exp $
Gildas Bazin's avatar
 
Gildas Bazin committed
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
 *
 * Authors: Gildas Bazin <gbazin@netcourrier.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
27 28 29
#include <stdlib.h>                                      /* malloc(), free() */
#include <string.h>                                    /* memcpy(), memset() */

Gildas Bazin's avatar
 
Gildas Bazin committed
30 31 32
#include <vlc/vlc.h>
#include <vlc/vout.h>
#include <vlc/decoder.h>
Gildas Bazin's avatar
 
Gildas Bazin committed
33 34 35
#include <vlc/input.h>
#include <vlc/sout.h>
#include <input_ext-dec.h>
Gildas Bazin's avatar
 
Gildas Bazin committed
36 37 38 39 40 41

#include <ogg/ogg.h>

#include <theora/theora.h>

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
42
 * decoder_sys_t : theora decoder descriptor
Gildas Bazin's avatar
 
Gildas Bazin committed
43
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
44
struct decoder_sys_t
Gildas Bazin's avatar
 
Gildas Bazin committed
45
{
Gildas Bazin's avatar
 
Gildas Bazin committed
46 47 48
    /* Module mode */
    vlc_bool_t b_packetizer;

Gildas Bazin's avatar
 
Gildas Bazin committed
49
    /*
Gildas Bazin's avatar
 
Gildas Bazin committed
50
     * Input properties
Gildas Bazin's avatar
 
Gildas Bazin committed
51
     */
Gildas Bazin's avatar
 
Gildas Bazin committed
52
    int i_headers;
Gildas Bazin's avatar
 
Gildas Bazin committed
53 54 55 56 57

    /*
     * Theora properties
     */
    theora_info      ti;                        /* theora bitstream settings */
Gildas Bazin's avatar
 
Gildas Bazin committed
58
    theora_comment   tc;                            /* theora comment header */
Gildas Bazin's avatar
 
Gildas Bazin committed
59 60 61
    theora_state     td;                   /* theora bitstream user comments */

    /*
Gildas Bazin's avatar
 
Gildas Bazin committed
62
     * Output properties
Gildas Bazin's avatar
 
Gildas Bazin committed
63
     */
Gildas Bazin's avatar
 
Gildas Bazin committed
64
    vout_thread_t *p_vout;
Gildas Bazin's avatar
 
Gildas Bazin committed
65 66

    /*
Gildas Bazin's avatar
 
Gildas Bazin committed
67
     * Packetizer output properties
Gildas Bazin's avatar
 
Gildas Bazin committed
68
     */
Gildas Bazin's avatar
 
Gildas Bazin committed
69 70
    sout_packetizer_input_t *p_sout_input;
    sout_format_t           sout_format;
Gildas Bazin's avatar
 
Gildas Bazin committed
71

Gildas Bazin's avatar
 
Gildas Bazin committed
72 73 74 75 76
    /*
     * Common properties
     */
    mtime_t i_pts;
};
Gildas Bazin's avatar
 
Gildas Bazin committed
77 78 79 80

/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
81 82 83 84 85 86
static int OpenDecoder   ( vlc_object_t * );
static int OpenPacketizer( vlc_object_t * );

static int InitDecoder   ( decoder_t * );
static int RunDecoder    ( decoder_t *, block_t * );
static int EndDecoder    ( decoder_t * );
Gildas Bazin's avatar
 
Gildas Bazin committed
87

Gildas Bazin's avatar
 
Gildas Bazin committed
88 89 90
static int ProcessPacket ( decoder_t *, ogg_packet *, mtime_t );
static int DecodePacket  ( decoder_t *, ogg_packet * );
static int SendPacket    ( decoder_t *, ogg_packet * );
Gildas Bazin's avatar
 
Gildas Bazin committed
91

Gildas Bazin's avatar
 
Gildas Bazin committed
92 93
static void ParseTheoraComments( decoder_t * );
static void theora_CopyPicture( decoder_t *, picture_t *, yuv_buffer * );
94

Gildas Bazin's avatar
 
Gildas Bazin committed
95 96 97 98 99
static int  OpenEncoder( vlc_object_t *p_this );
static void CloseEncoder( vlc_object_t *p_this );
static block_t *Headers( encoder_t *p_enc );
static block_t *Encode( encoder_t *p_enc, picture_t *p_pict );

Gildas Bazin's avatar
 
Gildas Bazin committed
100 101 102 103
/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
vlc_module_begin();
Gildas Bazin's avatar
 
Gildas Bazin committed
104
    set_description( _("Theora video decoder") );
Gildas Bazin's avatar
 
Gildas Bazin committed
105 106 107
    set_capability( "decoder", 100 );
    set_callbacks( OpenDecoder, NULL );
    add_shortcut( "theora" );
Gildas Bazin's avatar
 
Gildas Bazin committed
108 109 110 111 112 113

    add_submodule();
    set_description( _("Theora video packetizer") );
    set_capability( "packetizer", 100 );
    set_callbacks( OpenPacketizer, NULL );
    add_shortcut( "theora" );
Gildas Bazin's avatar
 
Gildas Bazin committed
114 115 116 117 118 119

    add_submodule();
    set_description( _("Theora video encoder") );
    set_capability( "video encoder", 100 );
    set_callbacks( OpenEncoder, CloseEncoder );
    add_shortcut( "theora" );
Gildas Bazin's avatar
 
Gildas Bazin committed
120 121 122 123 124 125 126
vlc_module_end();

/*****************************************************************************
 * OpenDecoder: probe the decoder and return score
 *****************************************************************************/
static int OpenDecoder( vlc_object_t *p_this )
{
Gildas Bazin's avatar
 
Gildas Bazin committed
127
    decoder_t *p_dec = (decoder_t*)p_this;
Gildas Bazin's avatar
 
Gildas Bazin committed
128

Gildas Bazin's avatar
 
Gildas Bazin committed
129
    if( p_dec->p_fifo->i_fourcc != VLC_FOURCC('t','h','e','o') )
Gildas Bazin's avatar
 
Gildas Bazin committed
130 131 132 133
    {
        return VLC_EGENERIC;
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
134 135 136 137 138 139 140 141 142 143 144 145 146
    p_dec->pf_init = InitDecoder;
    p_dec->pf_decode = RunDecoder;
    p_dec->pf_end = EndDecoder;

    /* Allocate the memory needed to store the decoder's structure */
    if( ( p_dec->p_sys =
          (decoder_sys_t *)malloc(sizeof(decoder_sys_t)) ) == NULL )
    {
        msg_Err( p_dec, "out of memory" );
        return VLC_EGENERIC;
    }
    p_dec->p_sys->b_packetizer = VLC_FALSE;

Gildas Bazin's avatar
 
Gildas Bazin committed
147 148
    return VLC_SUCCESS;
}
Gildas Bazin's avatar
 
Gildas Bazin committed
149 150 151 152 153 154 155 156 157 158 159 160

static int OpenPacketizer( vlc_object_t *p_this )
{
    decoder_t *p_dec = (decoder_t*)p_this;

    int i_ret = OpenDecoder( p_this );

    if( i_ret == VLC_SUCCESS ) p_dec->p_sys->b_packetizer = VLC_TRUE;

    return i_ret;
}

Gildas Bazin's avatar
 
Gildas Bazin committed
161
/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
162
 * InitDecoder: Initalize the decoder
Gildas Bazin's avatar
 
Gildas Bazin committed
163
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
164
static int InitDecoder( decoder_t *p_dec )
Gildas Bazin's avatar
 
Gildas Bazin committed
165
{
Gildas Bazin's avatar
 
Gildas Bazin committed
166
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
167

Gildas Bazin's avatar
 
Gildas Bazin committed
168
    p_sys->i_pts = 0;
Gildas Bazin's avatar
 
Gildas Bazin committed
169

Gildas Bazin's avatar
 
Gildas Bazin committed
170 171 172 173 174 175 176 177
    p_sys->p_sout_input = NULL;
    p_sys->sout_format.i_cat = VIDEO_ES;
    p_sys->sout_format.i_fourcc = VLC_FOURCC( 't', 'h', 'e', 'o' );
    p_sys->sout_format.i_width  = 0;
    p_sys->sout_format.i_height = 0;
    p_sys->sout_format.i_bitrate     = 0;
    p_sys->sout_format.i_extra_data  = 0;
    p_sys->sout_format.p_extra_data  = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
178

Gildas Bazin's avatar
 
Gildas Bazin committed
179
    /* Init supporting Theora structures needed in header parsing */
Gildas Bazin's avatar
 
Gildas Bazin committed
180 181
    theora_comment_init( &p_sys->tc );
    theora_info_init( &p_sys->ti );
Gildas Bazin's avatar
 
Gildas Bazin committed
182

Gildas Bazin's avatar
 
Gildas Bazin committed
183
    p_sys->i_headers = 0;
Gildas Bazin's avatar
 
Gildas Bazin committed
184

Gildas Bazin's avatar
 
Gildas Bazin committed
185 186
    return VLC_SUCCESS;
}
Gildas Bazin's avatar
 
Gildas Bazin committed
187

Gildas Bazin's avatar
 
Gildas Bazin committed
188 189 190 191 192 193 194 195 196 197
/****************************************************************************
 * RunDecoder: the whole thing
 ****************************************************************************
 * This function must be fed with ogg packets.
 ****************************************************************************/
static int RunDecoder( decoder_t *p_dec, block_t *p_block )
{
    decoder_sys_t *p_sys = p_dec->p_sys;
    ogg_packet oggpacket;
    int i_ret;
Gildas Bazin's avatar
 
Gildas Bazin committed
198

Gildas Bazin's avatar
 
Gildas Bazin committed
199 200 201 202 203 204 205
    /* Block to Ogg packet */
    oggpacket.packet = p_block->p_buffer;
    oggpacket.bytes = p_block->i_buffer;
    oggpacket.granulepos = p_block->i_dts;
    oggpacket.b_o_s = 0;
    oggpacket.e_o_s = 0;
    oggpacket.packetno = 0;
Gildas Bazin's avatar
 
Gildas Bazin committed
206

Gildas Bazin's avatar
 
Gildas Bazin committed
207
    if( p_sys->i_headers == 0 )
Gildas Bazin's avatar
 
Gildas Bazin committed
208
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
209
        /* Take care of the initial Theora header */
Gildas Bazin's avatar
 
Gildas Bazin committed
210

Gildas Bazin's avatar
 
Gildas Bazin committed
211 212 213 214 215 216 217 218 219
        oggpacket.b_o_s = 1; /* yes this actually is a b_o_s packet :) */
        if( theora_decode_header( &p_sys->ti, &p_sys->tc, &oggpacket ) < 0 )
        {
            msg_Err( p_dec->p_fifo, "This bitstream does not contain Theora "
                     "video data" );
            block_Release( p_block );
            return VLC_EGENERIC;
        }
        p_sys->i_headers++;
Gildas Bazin's avatar
 
Gildas Bazin committed
220

Gildas Bazin's avatar
 
Gildas Bazin committed
221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262
 
        if( p_sys->b_packetizer )
        {
            /* add a input for the stream ouput */
            p_sys->sout_format.i_width  = p_sys->ti.width;
            p_sys->sout_format.i_height = p_sys->ti.height;

            p_sys->p_sout_input =
                sout_InputNew( p_dec, &p_sys->sout_format );

            if( !p_sys->p_sout_input )
            {
                msg_Err( p_dec, "cannot add a new stream" );
                block_Release( p_block );
                return VLC_EGENERIC;
            }
        }
        else
        {
            /* Initialize video output */
            int i_chroma, i_aspect;

            if( p_sys->ti.aspect_denominator )
                i_aspect = VOUT_ASPECT_FACTOR * p_sys->ti.aspect_numerator /
                    p_sys->ti.aspect_denominator;
            else
                i_aspect = VOUT_ASPECT_FACTOR *
                    p_sys->ti.frame_width / p_sys->ti.frame_height;

            i_chroma = VLC_FOURCC('Y','V','1','2');

            p_sys->p_vout =
                vout_Request( p_dec, NULL,
                              p_sys->ti.frame_width, p_sys->ti.frame_height,
                              i_chroma, i_aspect );
            if( p_sys->p_vout == NULL )
            {
                msg_Err( p_dec, "failed to create video output" );
                block_Release( p_block );
                return VLC_EGENERIC;
            }
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
263

Gildas Bazin's avatar
 
Gildas Bazin committed
264 265 266 267 268 269
        msg_Dbg( p_dec, "%dx%d %.02f fps video, frame content "
                 "is %dx%d with offset (%d,%d)",
                 p_sys->ti.width, p_sys->ti.height,
                 (double)p_sys->ti.fps_numerator/p_sys->ti.fps_denominator,
                 p_sys->ti.frame_width, p_sys->ti.frame_height,
                 p_sys->ti.offset_x, p_sys->ti.offset_y );
Gildas Bazin's avatar
 
Gildas Bazin committed
270

Gildas Bazin's avatar
 
Gildas Bazin committed
271 272 273 274 275 276 277 278 279 280 281 282 283 284
        if( p_sys->b_packetizer )
        {
            i_ret = SendPacket( p_dec, &oggpacket );
            block_Release( p_block );
            return i_ret;
        }
        else
        {
            block_Release( p_block );
            return VLC_SUCCESS;
        }
    }

    if( p_sys->i_headers == 1 )
Gildas Bazin's avatar
 
Gildas Bazin committed
285
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306
        /* The next packet in order is the comments header */
        if( theora_decode_header( &p_sys->ti, &p_sys->tc, &oggpacket ) < 0 )
        {
            msg_Err( p_dec, "2nd Theora header is corrupted" );
            return VLC_EGENERIC;
        }
        p_sys->i_headers++;
    
        ParseTheoraComments( p_dec );

        if( p_sys->b_packetizer )
        {
            i_ret = SendPacket( p_dec, &oggpacket );
            block_Release( p_block );
            return i_ret;
        }
        else
        {
            block_Release( p_block );
            return VLC_SUCCESS;
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
307 308
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
309
    if( p_sys->i_headers == 2 )
Gildas Bazin's avatar
 
Gildas Bazin committed
310
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337
        /* The next packet in order is the codebooks header
           We need to watch out that this packet is not missing as a
           missing or corrupted header is fatal. */
        if( theora_decode_header( &p_sys->ti, &p_sys->tc, &oggpacket ) < 0 )
        {
            msg_Err( p_dec, "3rd Theora header is corrupted" );
            return VLC_EGENERIC;
        }
        p_sys->i_headers++;
    
        if( !p_sys->b_packetizer )
        {
            /* We have all the headers, initialize decoder */
            theora_decode_init( &p_sys->td, &p_sys->ti );
        }

        if( p_sys->b_packetizer )
        {
            i_ret = SendPacket( p_dec, &oggpacket );
            block_Release( p_block );
            return i_ret;
        }
        else
        {
            block_Release( p_block );
            return VLC_SUCCESS;
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
338 339
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
340 341 342 343
    i_ret = ProcessPacket( p_dec, &oggpacket, p_block->i_pts );
    block_Release( p_block );
    return i_ret;
}
Gildas Bazin's avatar
 
Gildas Bazin committed
344

Gildas Bazin's avatar
 
Gildas Bazin committed
345 346 347 348 349 350 351
/*****************************************************************************
 * ProcessPacket: processes a Vorbis packet.
 *****************************************************************************/
static int ProcessPacket( decoder_t *p_dec, ogg_packet *p_oggpacket,
                          mtime_t i_pts )
{
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
352

Gildas Bazin's avatar
 
Gildas Bazin committed
353 354
    /* Date management */
    if( i_pts > 0 && i_pts != p_sys->i_pts )
Gildas Bazin's avatar
 
Gildas Bazin committed
355
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
356
        p_sys->i_pts = i_pts;
Gildas Bazin's avatar
 
Gildas Bazin committed
357 358
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
359 360 361 362 363 364 365 366
    if( p_sys->b_packetizer )
    {
        return SendPacket( p_dec, p_oggpacket );
    }
    else
    {
        return DecodePacket( p_dec, p_oggpacket );
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
367 368 369 370 371
}

/*****************************************************************************
 * DecodePacket: decodes a Theora packet.
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
372
static int DecodePacket( decoder_t *p_dec, ogg_packet *p_oggpacket )
Gildas Bazin's avatar
 
Gildas Bazin committed
373 374 375 376
{
    picture_t *p_pic;
    yuv_buffer yuv;

Gildas Bazin's avatar
 
Gildas Bazin committed
377
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
378

Gildas Bazin's avatar
 
Gildas Bazin committed
379
    theora_decode_packetin( &p_sys->td, p_oggpacket );
Gildas Bazin's avatar
 
Gildas Bazin committed
380 381

    /* Decode */
Gildas Bazin's avatar
 
Gildas Bazin committed
382
    theora_decode_YUVout( &p_sys->td, &yuv );
Gildas Bazin's avatar
 
Gildas Bazin committed
383 384

    /* Get a new picture */
Gildas Bazin's avatar
 
Gildas Bazin committed
385
    while( !(p_pic = vout_CreatePicture( p_sys->p_vout, 0, 0, 0 ) ) )
Gildas Bazin's avatar
 
Gildas Bazin committed
386 387 388
    {
        if( p_dec->p_fifo->b_die || p_dec->p_fifo->b_error )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
389
            return VLC_EGENERIC;
Gildas Bazin's avatar
 
Gildas Bazin committed
390 391 392
        }
        msleep( VOUT_OUTMEM_SLEEP );
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
393
    if( !p_pic ) return VLC_EGENERIC;
Gildas Bazin's avatar
 
Gildas Bazin committed
394 395 396

    theora_CopyPicture( p_dec, p_pic, &yuv );

Gildas Bazin's avatar
 
Gildas Bazin committed
397 398 399 400
    vout_DatePicture( p_sys->p_vout, p_pic, p_sys->i_pts );
    vout_DisplayPicture( p_sys->p_vout, p_pic );

    /* Date management */
Gildas Bazin's avatar
 
Gildas Bazin committed
401 402
    p_sys->i_pts += ( I64C(1000000) * p_sys->ti.fps_denominator /
                      p_sys->ti.fps_numerator ); /* 1 frame per packet */
Gildas Bazin's avatar
 
Gildas Bazin committed
403 404

    return VLC_SUCCESS;
Gildas Bazin's avatar
 
Gildas Bazin committed
405 406 407
}

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
408
 * SendPacket: send an ogg packet to the stream output.
Gildas Bazin's avatar
 
Gildas Bazin committed
409
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
410
static int SendPacket( decoder_t *p_dec, ogg_packet *p_oggpacket )
Gildas Bazin's avatar
 
Gildas Bazin committed
411
{
Gildas Bazin's avatar
 
Gildas Bazin committed
412 413 414 415
    decoder_sys_t *p_sys = p_dec->p_sys;

    sout_buffer_t *p_sout_buffer =
        sout_BufferNew( p_sys->p_sout_input->p_sout, p_oggpacket->bytes );
Gildas Bazin's avatar
 
Gildas Bazin committed
416

Gildas Bazin's avatar
 
Gildas Bazin committed
417
    if( !p_sout_buffer ) return VLC_EGENERIC;
Gildas Bazin's avatar
 
Gildas Bazin committed
418

Gildas Bazin's avatar
 
Gildas Bazin committed
419 420 421
    p_dec->p_vlc->pf_memcpy( p_sout_buffer->p_buffer,
                             p_oggpacket->packet,
                             p_oggpacket->bytes );
Gildas Bazin's avatar
 
Gildas Bazin committed
422

Gildas Bazin's avatar
 
Gildas Bazin committed
423 424
    /* Date management */
    p_sout_buffer->i_dts = p_sout_buffer->i_pts = p_sys->i_pts;
Gildas Bazin's avatar
 
Gildas Bazin committed
425 426
    p_sys->i_pts += ( I64C(1000000) * p_sys->ti.fps_denominator /
                      p_sys->ti.fps_numerator ); /* 1 frame per packet */
Gildas Bazin's avatar
 
Gildas Bazin committed
427 428 429 430 431 432 433

    if( p_sys->i_headers >= 3 )
        p_sout_buffer->i_length = p_sys->i_pts - p_sout_buffer->i_pts;
    else
        p_sout_buffer->i_length = 0;

    sout_InputSendBuffer( p_sys->p_sout_input, p_sout_buffer );
Gildas Bazin's avatar
 
Gildas Bazin committed
434 435 436 437 438

    return VLC_SUCCESS;
}

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
439
 * ParseTheoraComments: FIXME should be done in demuxer
Gildas Bazin's avatar
 
Gildas Bazin committed
440
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
441
static void ParseTheoraComments( decoder_t *p_dec )
Gildas Bazin's avatar
 
Gildas Bazin committed
442
{
Gildas Bazin's avatar
 
Gildas Bazin committed
443 444 445 446 447 448
    input_thread_t *p_input = (input_thread_t *)p_dec->p_parent;
    input_info_category_t *p_cat =
        input_InfoCategory( p_input, _("Theora Comment") );
    int i = 0;
    char *psz_name, *psz_value, *psz_comment;
    while ( i < p_dec->p_sys->tc.comments )
Gildas Bazin's avatar
 
Gildas Bazin committed
449
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474
        psz_comment = strdup( p_dec->p_sys->tc.user_comments[i] );
        if( !psz_comment )
        {
            msg_Warn( p_dec, "Out of memory" );
            break;
        }
        psz_name = psz_comment;
        psz_value = strchr( psz_comment, '=' );
        if( psz_value )
        {
            *psz_value = '\0';
            psz_value++;
            input_AddInfo( p_cat, psz_name, psz_value );
        }
        free( psz_comment );
        i++;
    }
}

/*****************************************************************************
 * EndDecoder: theora decoder destruction
 *****************************************************************************/
static int EndDecoder( decoder_t *p_dec )
{
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
475

Gildas Bazin's avatar
 
Gildas Bazin committed
476 477
    if( !p_sys->b_packetizer )
        vout_Request( p_dec, p_sys->p_vout, 0, 0, 0, 0 );
Gildas Bazin's avatar
 
Gildas Bazin committed
478

Gildas Bazin's avatar
 
Gildas Bazin committed
479 480 481 482 483
    if( p_sys->p_sout_input != NULL )
    {
        sout_InputDelete( p_sys->p_sout_input );
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
484 485
    theora_info_clear( &p_sys->ti );
    theora_comment_clear( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
486

Gildas Bazin's avatar
 
Gildas Bazin committed
487 488 489
    free( p_sys );

    return VLC_SUCCESS;
Gildas Bazin's avatar
 
Gildas Bazin committed
490 491 492 493 494 495
}

/*****************************************************************************
 * theora_CopyPicture: copy a picture from theora internal buffers to a
 *                     picture_t structure.
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
496
static void theora_CopyPicture( decoder_t *p_dec, picture_t *p_pic,
Gildas Bazin's avatar
 
Gildas Bazin committed
497 498 499
                                yuv_buffer *yuv )
{
    int i_plane, i_line, i_width, i_dst_stride, i_src_stride;
Gildas Bazin's avatar
 
Gildas Bazin committed
500
    int i_src_xoffset, i_src_yoffset;
Gildas Bazin's avatar
 
Gildas Bazin committed
501 502 503 504 505 506 507
    u8  *p_dst, *p_src;

    for( i_plane = 0; i_plane < p_pic->i_planes; i_plane++ )
    {
        p_dst = p_pic->p[i_plane].p_pixels;
        p_src = i_plane ? (i_plane - 1 ? yuv->v : yuv->u ) : yuv->y;
        i_width = p_pic->p[i_plane].i_visible_pitch;
Gildas Bazin's avatar
 
Gildas Bazin committed
508 509 510 511 512 513 514 515 516 517 518
        i_dst_stride  = p_pic->p[i_plane].i_pitch;
        i_src_stride  = i_plane ? yuv->uv_stride : yuv->y_stride;
        i_src_xoffset = p_dec->p_sys->ti.offset_x;
        i_src_yoffset = p_dec->p_sys->ti.offset_y;
        if( i_plane )
        {
            i_src_xoffset /= 2;
            i_src_yoffset /= 2;
        } 

        p_src += (i_src_yoffset * i_src_stride + i_src_yoffset);
Gildas Bazin's avatar
 
Gildas Bazin committed
519 520 521

        for( i_line = 0; i_line < p_pic->p[i_plane].i_lines; i_line++ )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
522
            p_dec->p_vlc->pf_memcpy( p_dst, p_src, i_width );
Gildas Bazin's avatar
 
Gildas Bazin committed
523 524 525 526 527
            p_src += i_src_stride;
            p_dst += i_dst_stride;
        }
    }
}
Gildas Bazin's avatar
 
Gildas Bazin committed
528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618

/*****************************************************************************
 * encoder_sys_t : theora encoder descriptor
 *****************************************************************************/
struct encoder_sys_t
{
    /*
     * Input properties
     */
    int i_headers;

    /*
     * Theora properties
     */
    theora_info      ti;                        /* theora bitstream settings */
    theora_comment   tc;                            /* theora comment header */
    theora_state     td;                   /* theora bitstream user comments */

    /*
     * Packetizer output properties
     */
    sout_packetizer_input_t *p_sout_input;
    sout_format_t           sout_format;

    /*
     * Common properties
     */
    mtime_t i_pts;
};

/*****************************************************************************
 * OpenEncoder: probe the encoder and return score
 *****************************************************************************/
static int OpenEncoder( vlc_object_t *p_this )
{
    encoder_t *p_enc = (encoder_t *)p_this;
    encoder_sys_t *p_sys = p_enc->p_sys;

    if( p_enc->i_fourcc != VLC_FOURCC('t','h','e','o') )
    {
        return VLC_EGENERIC;
    }

    /* Allocate the memory needed to store the decoder's structure */
    if( ( p_sys = (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
    {
        msg_Err( p_enc, "out of memory" );
        return VLC_EGENERIC;
    }
    p_enc->p_sys = p_sys;

    p_enc->pf_header = Headers;
    p_enc->pf_encode_video = Encode;
    p_enc->format.video.i_chroma = VLC_FOURCC('I','4','2','0');

#define frame_x_offset 0
#define frame_y_offset 0
#define video_hzn 25
#define video_hzd 1
#define video_an 4
#define video_ad 3
#define video_q 5

    theora_info_init( &p_sys->ti );

    p_sys->ti.width = p_enc->format.video.i_width;
    p_sys->ti.height = p_enc->format.video.i_height;
    p_sys->ti.frame_width = p_enc->format.video.i_width;
    p_sys->ti.frame_height = p_enc->format.video.i_height;
    p_sys->ti.offset_x = frame_x_offset;
    p_sys->ti.offset_y = frame_y_offset;
    p_sys->ti.fps_numerator = video_hzn;
    p_sys->ti.fps_denominator = video_hzd;
    p_sys->ti.aspect_numerator = video_an;
    p_sys->ti.aspect_denominator = video_ad;
    p_sys->ti.colorspace = not_specified;
    p_sys->ti.target_bitrate = p_enc->i_bitrate;
    p_sys->ti.quality = video_q;

    p_sys->ti.dropframes_p = 0;
    p_sys->ti.quick_p = 1;
    p_sys->ti.keyframe_auto_p = 1;
    p_sys->ti.keyframe_frequency = 64;
    p_sys->ti.keyframe_frequency_force = 64;
    p_sys->ti.keyframe_data_target_bitrate = p_enc->i_bitrate * 1.5;
    p_sys->ti.keyframe_auto_threshold = 80;
    p_sys->ti.keyframe_mindistance = 8;
    p_sys->ti.noise_sensitivity = 1;

    theora_encode_init( &p_sys->td, &p_sys->ti );
    theora_info_clear( &p_sys->ti );
Gildas Bazin's avatar
 
Gildas Bazin committed
619
    theora_comment_init( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649

    p_sys->i_headers = 0;

    return VLC_SUCCESS;
}

/****************************************************************************
 * Encode: the whole thing
 ****************************************************************************
 * This function spits out ogg packets.
 ****************************************************************************/
static block_t *Headers( encoder_t *p_enc )
{
    encoder_sys_t *p_sys = p_enc->p_sys;
    ogg_packet oggpacket;
    block_t *p_block;

    /* Create theora headers */
    switch( p_sys->i_headers )
    {
    case 0:
        theora_encode_header( &p_sys->td, &oggpacket );
        break;
    case 1:
        theora_encode_comment( &p_sys->tc, &oggpacket );
        break;
    case 2:
        theora_encode_tables( &p_sys->td, &oggpacket );
        break;
    default:
Gildas Bazin's avatar
 
Gildas Bazin committed
650
        break;
Gildas Bazin's avatar
 
Gildas Bazin committed
651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697
    }

    p_sys->i_headers++;
    if( p_sys->i_headers > 3 ) return NULL;

    /* Ogg packet to block */
    p_block = block_New( p_enc, oggpacket.bytes );
    p_block->p_buffer = oggpacket.packet;
    p_block->i_buffer = oggpacket.bytes;
    p_block->i_dts = oggpacket.granulepos;

    return p_block;
}

/****************************************************************************
 * Encode: the whole thing
 ****************************************************************************
 * This function spits out ogg packets.
 ****************************************************************************/
static block_t *Encode( encoder_t *p_enc, picture_t *p_pict )
{
    encoder_sys_t *p_sys = p_enc->p_sys;
    ogg_packet oggpacket;
    block_t *p_block;
    yuv_buffer yuv;

    /* Theora is a one-frame-in, one-frame-out system. Submit a frame
     * for compression and pull out the packet. */

    yuv.y_width  = p_pict->p[0].i_visible_pitch;
    yuv.y_height = p_pict->p[0].i_lines;
    yuv.y_stride = p_pict->p[0].i_pitch;

    yuv.uv_width  = p_pict->p[1].i_visible_pitch;
    yuv.uv_height = p_pict->p[1].i_lines;
    yuv.uv_stride = p_pict->p[1].i_pitch;

    yuv.y = p_pict->p[0].p_pixels;
    yuv.u = p_pict->p[1].p_pixels;
    yuv.v = p_pict->p[2].p_pixels;

    theora_encode_YUVin( &p_sys->td, &yuv );

    theora_encode_packetout( &p_sys->td, 0, &oggpacket );

    /* Ogg packet to block */
    p_block = block_New( p_enc, oggpacket.bytes );
Gildas Bazin's avatar
 
Gildas Bazin committed
698 699
    memcpy( p_block->p_buffer, oggpacket.packet, oggpacket.bytes );
    p_block->i_dts = p_pict->date;//oggpacket.granulepos;
Gildas Bazin's avatar
 
Gildas Bazin committed
700 701 702 703 704 705 706 707 708 709 710 711 712

    return p_block;
}

/*****************************************************************************
 * CloseEncoder: theora encoder destruction
 *****************************************************************************/
static void CloseEncoder( vlc_object_t *p_this )
{
    encoder_t *p_enc = (encoder_t *)p_this;
    encoder_sys_t *p_sys = p_enc->p_sys;

    theora_info_clear( &p_sys->ti );
713
    theora_comment_clear( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
714 715 716

    free( p_sys );
}