theora.c 20.3 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.24 2004/01/29 17:51:07 zorglub Exp $
Gildas Bazin's avatar
 
Gildas Bazin committed
6 7 8 9 10 11 12
 *
 * 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.
Sam Hocevar's avatar
Sam Hocevar committed
13
 *
Gildas Bazin's avatar
 
Gildas Bazin committed
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
 * 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
 *****************************************************************************/
#include <vlc/vlc.h>
#include <vlc/decoder.h>
Gildas Bazin's avatar
 
Gildas Bazin committed
29
#include "input_ext-plugins.h"
30
#include "vlc_playlist.h"
Gildas Bazin's avatar
 
Gildas Bazin committed
31 32 33 34 35 36

#include <ogg/ogg.h>

#include <theora/theora.h>

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
37
 * decoder_sys_t : theora decoder descriptor
Gildas Bazin's avatar
 
Gildas Bazin committed
38
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
39
struct decoder_sys_t
Gildas Bazin's avatar
 
Gildas Bazin committed
40
{
Gildas Bazin's avatar
 
Gildas Bazin committed
41 42 43
    /* Module mode */
    vlc_bool_t b_packetizer;

Gildas Bazin's avatar
 
Gildas Bazin committed
44
    /*
Gildas Bazin's avatar
 
Gildas Bazin committed
45
     * Input properties
Gildas Bazin's avatar
 
Gildas Bazin committed
46
     */
Gildas Bazin's avatar
 
Gildas Bazin committed
47
    int i_headers;
Gildas Bazin's avatar
 
Gildas Bazin committed
48 49 50 51 52

    /*
     * Theora properties
     */
    theora_info      ti;                        /* theora bitstream settings */
Gildas Bazin's avatar
 
Gildas Bazin committed
53
    theora_comment   tc;                            /* theora comment header */
Gildas Bazin's avatar
 
Gildas Bazin committed
54 55
    theora_state     td;                   /* theora bitstream user comments */

Gildas Bazin's avatar
 
Gildas Bazin committed
56 57 58 59 60
    /*
     * Common properties
     */
    mtime_t i_pts;
};
Gildas Bazin's avatar
 
Gildas Bazin committed
61 62 63 64

/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
65 66 67
static int  OpenDecoder   ( vlc_object_t * );
static int  OpenPacketizer( vlc_object_t * );
static void CloseDecoder  ( vlc_object_t * );
Gildas Bazin's avatar
 
Gildas Bazin committed
68

Gildas Bazin's avatar
 
Gildas Bazin committed
69 70
static void *DecodeBlock  ( decoder_t *, block_t ** );
static void *ProcessPacket ( decoder_t *, ogg_packet *, block_t ** );
Gildas Bazin's avatar
 
Gildas Bazin committed
71

Gildas Bazin's avatar
 
Gildas Bazin committed
72
static picture_t *DecodePacket( decoder_t *, ogg_packet * );
Gildas Bazin's avatar
 
Gildas Bazin committed
73

Gildas Bazin's avatar
 
Gildas Bazin committed
74 75
static void ParseTheoraComments( decoder_t * );
static void theora_CopyPicture( decoder_t *, picture_t *, yuv_buffer * );
76

Gildas Bazin's avatar
 
Gildas Bazin committed
77 78 79 80 81
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
82 83 84 85
/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
vlc_module_begin();
Gildas Bazin's avatar
 
Gildas Bazin committed
86
    set_description( _("Theora video decoder") );
Gildas Bazin's avatar
 
Gildas Bazin committed
87
    set_capability( "decoder", 100 );
Gildas Bazin's avatar
 
Gildas Bazin committed
88
    set_callbacks( OpenDecoder, CloseDecoder );
Gildas Bazin's avatar
 
Gildas Bazin committed
89
    add_shortcut( "theora" );
Gildas Bazin's avatar
 
Gildas Bazin committed
90 91 92 93

    add_submodule();
    set_description( _("Theora video packetizer") );
    set_capability( "packetizer", 100 );
Gildas Bazin's avatar
 
Gildas Bazin committed
94
    set_callbacks( OpenPacketizer, CloseDecoder );
Gildas Bazin's avatar
 
Gildas Bazin committed
95
    add_shortcut( "theora" );
Gildas Bazin's avatar
 
Gildas Bazin committed
96 97 98

    add_submodule();
    set_description( _("Theora video encoder") );
Gildas Bazin's avatar
 
Gildas Bazin committed
99
    set_capability( "encoder", 100 );
Gildas Bazin's avatar
 
Gildas Bazin committed
100 101
    set_callbacks( OpenEncoder, CloseEncoder );
    add_shortcut( "theora" );
Gildas Bazin's avatar
 
Gildas Bazin committed
102 103 104 105 106 107 108
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
109
    decoder_t *p_dec = (decoder_t*)p_this;
Gildas Bazin's avatar
 
Gildas Bazin committed
110
    decoder_sys_t *p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
111

Gildas Bazin's avatar
 
Gildas Bazin committed
112
    if( p_dec->fmt_in.i_codec != VLC_FOURCC('t','h','e','o') )
Gildas Bazin's avatar
 
Gildas Bazin committed
113 114 115 116
    {
        return VLC_EGENERIC;
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
117
    /* Allocate the memory needed to store the decoder's structure */
Gildas Bazin's avatar
 
Gildas Bazin committed
118
    if( ( p_dec->p_sys = p_sys =
Gildas Bazin's avatar
 
Gildas Bazin committed
119 120 121 122 123 124 125
          (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
126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143
    p_sys->i_pts = 0;

    /* Set output properties */
    p_dec->fmt_out.i_cat = VIDEO_ES;
    p_dec->fmt_out.i_codec = VLC_FOURCC('I','4','2','0');

    /* Set callbacks */
    p_dec->pf_decode_video = (picture_t *(*)(decoder_t *, block_t **))
        DecodeBlock;
    p_dec->pf_packetize    = (block_t *(*)(decoder_t *, block_t **))
        DecodeBlock;

    /* Init supporting Theora structures needed in header parsing */
    theora_comment_init( &p_sys->tc );
    theora_info_init( &p_sys->ti );

    p_sys->i_headers = 0;

Gildas Bazin's avatar
 
Gildas Bazin committed
144 145
    return VLC_SUCCESS;
}
Gildas Bazin's avatar
 
Gildas Bazin committed
146 147 148 149 150 151 152

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

    int i_ret = OpenDecoder( p_this );

Gildas Bazin's avatar
 
Gildas Bazin committed
153 154 155 156 157
    if( i_ret == VLC_SUCCESS )
    {
        p_dec->p_sys->b_packetizer = VLC_TRUE;
        p_dec->fmt_out.i_codec = VLC_FOURCC( 't', 'h', 'e', 'o' );
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
158 159 160 161 162

    return i_ret;
}

/****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
163
 * DecodeBlock: the whole thing
Gildas Bazin's avatar
 
Gildas Bazin committed
164 165 166
 ****************************************************************************
 * This function must be fed with ogg packets.
 ****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
167
static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
Gildas Bazin's avatar
 
Gildas Bazin committed
168 169
{
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
170
    block_t *p_block;
Gildas Bazin's avatar
 
Gildas Bazin committed
171
    ogg_packet oggpacket;
Gildas Bazin's avatar
 
Gildas Bazin committed
172 173 174 175

    if( !pp_block || !*pp_block ) return NULL;

    p_block = *pp_block;
Gildas Bazin's avatar
 
Gildas Bazin committed
176

Gildas Bazin's avatar
 
Gildas Bazin committed
177 178 179 180 181 182 183
    /* 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
184

Gildas Bazin's avatar
 
Gildas Bazin committed
185
    if( p_sys->i_headers == 0 )
Gildas Bazin's avatar
 
Gildas Bazin committed
186
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
187
        /* Take care of the initial Theora header */
Gildas Bazin's avatar
 
Gildas Bazin committed
188

Gildas Bazin's avatar
 
Gildas Bazin committed
189 190 191
        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 )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
192
            msg_Err( p_dec, "This bitstream does not contain Theora "
Benjamin Pracht's avatar
Benjamin Pracht committed
193
                     "video data." );
Gildas Bazin's avatar
 
Gildas Bazin committed
194
            block_Release( p_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
195
            return NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
196 197
        }
        p_sys->i_headers++;
Gildas Bazin's avatar
 
Gildas Bazin committed
198

Gildas Bazin's avatar
 
Gildas Bazin committed
199 200 201
        /* Set output properties */
        p_dec->fmt_out.video.i_width = p_sys->ti.width;
        p_dec->fmt_out.video.i_height = p_sys->ti.height;
Sam Hocevar's avatar
Sam Hocevar committed
202

Gildas Bazin's avatar
 
Gildas Bazin committed
203
        if( p_sys->ti.aspect_denominator )
Gildas Bazin's avatar
 
Gildas Bazin committed
204
            p_dec->fmt_out.video.i_aspect = ((int64_t)VOUT_ASPECT_FACTOR) *
Gildas Bazin's avatar
 
Gildas Bazin committed
205
                p_sys->ti.aspect_numerator / p_sys->ti.aspect_denominator;
Gildas Bazin's avatar
 
Gildas Bazin committed
206
        else
Gildas Bazin's avatar
 
Gildas Bazin committed
207 208
            p_dec->fmt_out.video.i_aspect = VOUT_ASPECT_FACTOR *
                p_sys->ti.frame_width / p_sys->ti.frame_height;
Gildas Bazin's avatar
 
Gildas Bazin committed
209

Gildas Bazin's avatar
 
Gildas Bazin committed
210
        msg_Dbg( p_dec, "%dx%d %.02f fps video, frame content "
Benjamin Pracht's avatar
Benjamin Pracht committed
211
                 "is %dx%d with offset (%d,%d).",
Gildas Bazin's avatar
 
Gildas Bazin committed
212 213 214 215
                 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
216

Gildas Bazin's avatar
 
Gildas Bazin committed
217
        return ProcessPacket( p_dec, &oggpacket, pp_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
218 219 220
    }

    if( p_sys->i_headers == 1 )
Gildas Bazin's avatar
 
Gildas Bazin committed
221
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
222 223 224
        /* The next packet in order is the comments header */
        if( theora_decode_header( &p_sys->ti, &p_sys->tc, &oggpacket ) < 0 )
        {
Benjamin Pracht's avatar
Benjamin Pracht committed
225
            msg_Err( p_dec, "2nd Theora header is corrupted." );
Gildas Bazin's avatar
 
Gildas Bazin committed
226
            return NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
227 228
        }
        p_sys->i_headers++;
Sam Hocevar's avatar
Sam Hocevar committed
229

Gildas Bazin's avatar
 
Gildas Bazin committed
230 231
        ParseTheoraComments( p_dec );

Gildas Bazin's avatar
 
Gildas Bazin committed
232
        return ProcessPacket( p_dec, &oggpacket, pp_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
233 234
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
235
    if( p_sys->i_headers == 2 )
Gildas Bazin's avatar
 
Gildas Bazin committed
236
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
237 238 239 240 241
        /* 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 )
        {
Benjamin Pracht's avatar
Benjamin Pracht committed
242
            msg_Err( p_dec, "3rd Theora header is corrupted." );
Gildas Bazin's avatar
 
Gildas Bazin committed
243
            return NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
244 245
        }
        p_sys->i_headers++;
Sam Hocevar's avatar
Sam Hocevar committed
246

Gildas Bazin's avatar
 
Gildas Bazin committed
247 248 249 250 251 252
        if( !p_sys->b_packetizer )
        {
            /* We have all the headers, initialize decoder */
            theora_decode_init( &p_sys->td, &p_sys->ti );
        }

Gildas Bazin's avatar
 
Gildas Bazin committed
253
        return ProcessPacket( p_dec, &oggpacket, pp_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
254 255
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
256
    return ProcessPacket( p_dec, &oggpacket, pp_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
257
}
Gildas Bazin's avatar
 
Gildas Bazin committed
258

Gildas Bazin's avatar
 
Gildas Bazin committed
259
/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
260
 * ProcessPacket: processes a theora packet.
Gildas Bazin's avatar
 
Gildas Bazin committed
261
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
262 263
static void *ProcessPacket( decoder_t *p_dec, ogg_packet *p_oggpacket,
                            block_t **pp_block )
Gildas Bazin's avatar
 
Gildas Bazin committed
264 265
{
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
266 267
    block_t *p_block = *pp_block;
    void *p_buf;
Gildas Bazin's avatar
 
Gildas Bazin committed
268

Gildas Bazin's avatar
 
Gildas Bazin committed
269
    /* Date management */
Gildas Bazin's avatar
 
Gildas Bazin committed
270
    if( p_block->i_pts > 0 && p_block->i_pts != p_sys->i_pts )
Gildas Bazin's avatar
 
Gildas Bazin committed
271
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
272
        p_sys->i_pts = p_block->i_pts;
Gildas Bazin's avatar
 
Gildas Bazin committed
273 274
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
275 276
    if( p_sys->b_packetizer )
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
277 278 279 280 281 282 283 284 285
        /* Date management */
        p_block->i_dts = p_block->i_pts = p_sys->i_pts;

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

        p_buf = p_block;
Gildas Bazin's avatar
 
Gildas Bazin committed
286 287 288
    }
    else
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
289 290 291 292
        if( p_sys->i_headers >= 3 )
            p_buf = DecodePacket( p_dec, p_oggpacket );
        else
            p_buf = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
293

Gildas Bazin's avatar
 
Gildas Bazin committed
294
        if( p_block )
Gildas Bazin's avatar
 
Gildas Bazin committed
295
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
296 297
            block_Release( p_block );
            *pp_block = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
298 299
        }
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
300 301

    /* Date management */
Gildas Bazin's avatar
 
Gildas Bazin committed
302 303
    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
304

Gildas Bazin's avatar
 
Gildas Bazin committed
305
    return p_buf;
Gildas Bazin's avatar
 
Gildas Bazin committed
306 307 308
}

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
309
 * DecodePacket: decodes a Theora packet.
Gildas Bazin's avatar
 
Gildas Bazin committed
310
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
311
static picture_t *DecodePacket( decoder_t *p_dec, ogg_packet *p_oggpacket )
Gildas Bazin's avatar
 
Gildas Bazin committed
312
{
Gildas Bazin's avatar
 
Gildas Bazin committed
313
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
314 315
    picture_t *p_pic;
    yuv_buffer yuv;
Gildas Bazin's avatar
 
Gildas Bazin committed
316

Gildas Bazin's avatar
 
Gildas Bazin committed
317
    theora_decode_packetin( &p_sys->td, p_oggpacket );
Gildas Bazin's avatar
 
Gildas Bazin committed
318

Gildas Bazin's avatar
 
Gildas Bazin committed
319 320
    /* Decode */
    theora_decode_YUVout( &p_sys->td, &yuv );
Gildas Bazin's avatar
 
Gildas Bazin committed
321

Gildas Bazin's avatar
 
Gildas Bazin committed
322 323 324
    /* Get a new picture */
    p_pic = p_dec->pf_vout_buffer_new( p_dec );
    if( !p_pic ) return NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
325

Gildas Bazin's avatar
 
Gildas Bazin committed
326
    theora_CopyPicture( p_dec, p_pic, &yuv );
Gildas Bazin's avatar
 
Gildas Bazin committed
327

Gildas Bazin's avatar
 
Gildas Bazin committed
328
    p_pic->date = p_sys->i_pts;
Gildas Bazin's avatar
 
Gildas Bazin committed
329

Gildas Bazin's avatar
 
Gildas Bazin committed
330
    return p_pic;
Gildas Bazin's avatar
 
Gildas Bazin committed
331 332 333
}

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
334
 * ParseTheoraComments: FIXME should be done in demuxer
Gildas Bazin's avatar
 
Gildas Bazin committed
335
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
336
static void ParseTheoraComments( decoder_t *p_dec )
Gildas Bazin's avatar
 
Gildas Bazin committed
337
{
Gildas Bazin's avatar
 
Gildas Bazin committed
338 339
    input_thread_t *p_input = (input_thread_t *)p_dec->p_parent;
    input_info_category_t *p_cat =
Sam Hocevar's avatar
Sam Hocevar committed
340
        input_InfoCategory( p_input, _("Theora comment") );
341 342
    playlist_t *p_playlist = vlc_object_find( p_dec, VLC_OBJECT_PLAYLIST,
                                              FIND_ANYWHERE );
343
    playlist_item_t *p_item;
Gildas Bazin's avatar
 
Gildas Bazin committed
344 345 346
    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
347
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
348 349 350
        psz_comment = strdup( p_dec->p_sys->tc.user_comments[i] );
        if( !psz_comment )
        {
Benjamin Pracht's avatar
Benjamin Pracht committed
351
            msg_Warn( p_dec, "out of memory" );
Gildas Bazin's avatar
 
Gildas Bazin committed
352 353 354 355 356 357 358 359 360
            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 );
361 362 363 364 365 366 367 368 369 370 371 372
            vlc_mutex_lock( &p_playlist->object_lock );
            p_item = playlist_ItemGetByPos( p_playlist, -1 );
            vlc_mutex_unlock( &p_playlist->object_lock );
            if( !p_item)
            {
                msg_Err(p_dec, "unable to find item" );
                return;
            }
            vlc_mutex_lock( &p_item->lock );
            playlist_ItemAddInfo( p_item, _("Theora comment") ,
                                  psz_name, psz_value );
            vlc_mutex_unlock( &p_item->lock );
Gildas Bazin's avatar
 
Gildas Bazin committed
373 374 375 376
        }
        free( psz_comment );
        i++;
    }
377
    if( p_playlist) vlc_object_release( p_playlist );
Gildas Bazin's avatar
 
Gildas Bazin committed
378 379 380
}

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
381
 * CloseDecoder: theora decoder destruction
Gildas Bazin's avatar
 
Gildas Bazin committed
382
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
383
static void CloseDecoder( vlc_object_t *p_this )
Gildas Bazin's avatar
 
Gildas Bazin committed
384
{
Gildas Bazin's avatar
 
Gildas Bazin committed
385
    decoder_t *p_dec = (decoder_t *)p_this;
Gildas Bazin's avatar
 
Gildas Bazin committed
386
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
387

Gildas Bazin's avatar
 
Gildas Bazin committed
388 389
    theora_info_clear( &p_sys->ti );
    theora_comment_clear( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
390

Gildas Bazin's avatar
 
Gildas Bazin committed
391
    free( p_sys );
Gildas Bazin's avatar
 
Gildas Bazin committed
392 393 394 395 396 397
}

/*****************************************************************************
 * theora_CopyPicture: copy a picture from theora internal buffers to a
 *                     picture_t structure.
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
398
static void theora_CopyPicture( decoder_t *p_dec, picture_t *p_pic,
Gildas Bazin's avatar
 
Gildas Bazin committed
399 400 401
                                yuv_buffer *yuv )
{
    int i_plane, i_line, i_width, i_dst_stride, i_src_stride;
Gildas Bazin's avatar
 
Gildas Bazin committed
402
    int i_src_xoffset, i_src_yoffset;
Sam Hocevar's avatar
Sam Hocevar committed
403
    uint8_t *p_dst, *p_src;
Gildas Bazin's avatar
 
Gildas Bazin committed
404 405 406 407 408 409

    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
410 411 412 413 414 415 416 417
        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;
Sam Hocevar's avatar
Sam Hocevar committed
418
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
419 420

        p_src += (i_src_yoffset * i_src_stride + i_src_yoffset);
Gildas Bazin's avatar
 
Gildas Bazin committed
421 422 423

        for( i_line = 0; i_line < p_pic->p[i_plane].i_lines; i_line++ )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
424
            p_dec->p_vlc->pf_memcpy( p_dst, p_src, i_width );
Gildas Bazin's avatar
 
Gildas Bazin committed
425 426 427 428 429
            p_src += i_src_stride;
            p_dst += i_dst_stride;
        }
    }
}
Gildas Bazin's avatar
 
Gildas Bazin committed
430 431 432 433 434 435 436 437 438

/*****************************************************************************
 * encoder_sys_t : theora encoder descriptor
 *****************************************************************************/
struct encoder_sys_t
{
    /*
     * Input properties
     */
Gildas Bazin's avatar
 
Gildas Bazin committed
439
    vlc_bool_t b_headers;
Gildas Bazin's avatar
 
Gildas Bazin committed
440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461

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

    /*
     * 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;

Gildas Bazin's avatar
 
Gildas Bazin committed
462
    if( p_enc->fmt_out.i_codec != VLC_FOURCC('t','h','e','o') )
Gildas Bazin's avatar
 
Gildas Bazin committed
463 464 465 466
    {
        return VLC_EGENERIC;
    }

467 468 469 470
    if( p_enc->fmt_in.video.i_width % 16 ||
        p_enc->fmt_in.video.i_height % 16 )
    {
        msg_Err( p_enc, "Theora video encoding requires dimensions which are "
Benjamin Pracht's avatar
Benjamin Pracht committed
471
                 "multiples of 16. Which is not the case here (%dx%d).",
472 473 474 475
                 p_enc->fmt_in.video.i_width, p_enc->fmt_in.video.i_height );
        return VLC_EGENERIC;
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
476 477 478 479 480 481 482 483 484 485
    /* 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;
Gildas Bazin's avatar
 
Gildas Bazin committed
486
    p_enc->fmt_in.i_codec = VLC_FOURCC('I','4','2','0');
Gildas Bazin's avatar
 
Gildas Bazin committed
487 488 489 490 491 492 493 494 495

#define frame_x_offset 0
#define frame_y_offset 0
#define video_hzn 25
#define video_hzd 1
#define video_q 5

    theora_info_init( &p_sys->ti );

Gildas Bazin's avatar
 
Gildas Bazin committed
496 497 498 499
    p_sys->ti.width = p_enc->fmt_in.video.i_width;
    p_sys->ti.height = p_enc->fmt_in.video.i_height;
    p_sys->ti.frame_width = p_enc->fmt_in.video.i_width;
    p_sys->ti.frame_height = p_enc->fmt_in.video.i_height;
Gildas Bazin's avatar
 
Gildas Bazin committed
500 501 502 503
    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;
Gildas Bazin's avatar
 
Gildas Bazin committed
504 505 506 507 508 509 510 511 512 513 514 515

    if( p_enc->fmt_in.video.i_aspect )
    {
        p_sys->ti.aspect_numerator = p_enc->fmt_in.video.i_aspect;
        p_sys->ti.aspect_denominator = VOUT_ASPECT_FACTOR;
    }
    else
    {
        p_sys->ti.aspect_numerator = 4;
        p_sys->ti.aspect_denominator = 3;
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
516
    p_sys->ti.target_bitrate = p_enc->fmt_out.i_bitrate;
Gildas Bazin's avatar
 
Gildas Bazin committed
517 518 519 520 521 522 523
    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;
Gildas Bazin's avatar
 
Gildas Bazin committed
524
    p_sys->ti.keyframe_data_target_bitrate = p_enc->fmt_out.i_bitrate * 1.5;
Gildas Bazin's avatar
 
Gildas Bazin committed
525 526 527 528 529 530
    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
531
    theora_comment_init( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
532

Gildas Bazin's avatar
 
Gildas Bazin committed
533
    p_sys->b_headers = VLC_FALSE;
Gildas Bazin's avatar
 
Gildas Bazin committed
534 535 536 537 538 539 540 541 542 543 544 545

    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;
Gildas Bazin's avatar
 
Gildas Bazin committed
546
    block_t *p_chain = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
547 548

    /* Create theora headers */
Gildas Bazin's avatar
 
Gildas Bazin committed
549
    if( !p_sys->b_headers )
Gildas Bazin's avatar
 
Gildas Bazin committed
550
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
551
        ogg_packet oggpackets;
Gildas Bazin's avatar
 
Gildas Bazin committed
552
        int i;
Eric Petit's avatar
Eric Petit committed
553
        block_t *p_block;
Gildas Bazin's avatar
 
Gildas Bazin committed
554

Gildas Bazin's avatar
 
Gildas Bazin committed
555 556 557
        /* Ogg packet to block */
        for( i = 0; i < 3; i++ )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
558 559 560 561 562 563 564 565 566 567 568 569 570
            switch( i )
            {
            case 0:
                theora_encode_header( &p_sys->td, &oggpackets );
                break;
            case 1:
                theora_encode_comment( &p_sys->tc, &oggpackets );
                break;
            case 2:
                theora_encode_tables( &p_sys->td, &oggpackets );
                break;
            }

Eric Petit's avatar
Eric Petit committed
571
            p_block = block_New( p_enc, oggpackets.bytes );
Gildas Bazin's avatar
 
Gildas Bazin committed
572
            memcpy( p_block->p_buffer, oggpackets.packet, oggpackets.bytes );
Gildas Bazin's avatar
 
Gildas Bazin committed
573 574 575
            p_block->i_dts = p_block->i_pts = p_block->i_length = 0;
            block_ChainAppend( &p_chain, p_block );
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
576

Gildas Bazin's avatar
 
Gildas Bazin committed
577 578 579 580
        p_sys->b_headers = VLC_TRUE;
    }

    return p_chain;
Gildas Bazin's avatar
 
Gildas Bazin committed
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
}

/****************************************************************************
 * 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;

Gildas Bazin's avatar
 
Gildas Bazin committed
610 611 612 613 614
    if( theora_encode_YUVin( &p_sys->td, &yuv ) < 0 )
    {
        msg_Warn( p_enc, "failed encoding a frame" );
        return NULL;
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
615 616 617 618 619

    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
620
    memcpy( p_block->p_buffer, oggpacket.packet, oggpacket.bytes );
Gildas Bazin's avatar
 
Gildas Bazin committed
621
    p_block->i_dts = p_block->i_pts = p_pict->date;;
Gildas Bazin's avatar
 
Gildas Bazin committed
622 623 624 625 626 627 628 629 630 631 632 633 634

    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 );
635
    theora_comment_clear( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
636 637 638

    free( p_sys );
}