theora.c 20.5 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$
Gildas Bazin's avatar
 
Gildas Bazin committed
6
 *
7
 * Authors: Gildas Bazin <gbazin@videolan.org>
Gildas Bazin's avatar
 
Gildas Bazin committed
8 9 10 11 12
 *
 * 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>
29
#include <vlc/input.h>
30
#include <vlc/sout.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
/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
85 86 87 88 89
#define ENC_QUALITY_TEXT N_("Encoding quality")
#define ENC_QUALITY_LONGTEXT N_( \
  "Allows you to specify a quality between 1 (low) and 10 (high), instead " \
  "of specifying a particular bitrate. This will produce a VBR stream." )

Gildas Bazin's avatar
 
Gildas Bazin committed
90
vlc_module_begin();
Gildas Bazin's avatar
 
Gildas Bazin committed
91
    set_description( _("Theora video decoder") );
Gildas Bazin's avatar
 
Gildas Bazin committed
92
    set_capability( "decoder", 100 );
Gildas Bazin's avatar
 
Gildas Bazin committed
93
    set_callbacks( OpenDecoder, CloseDecoder );
Gildas Bazin's avatar
 
Gildas Bazin committed
94
    add_shortcut( "theora" );
Gildas Bazin's avatar
 
Gildas Bazin committed
95 96 97 98

    add_submodule();
    set_description( _("Theora video packetizer") );
    set_capability( "packetizer", 100 );
Gildas Bazin's avatar
 
Gildas Bazin committed
99
    set_callbacks( OpenPacketizer, CloseDecoder );
Gildas Bazin's avatar
 
Gildas Bazin committed
100
    add_shortcut( "theora" );
Gildas Bazin's avatar
 
Gildas Bazin committed
101 102 103

    add_submodule();
    set_description( _("Theora video encoder") );
Gildas Bazin's avatar
 
Gildas Bazin committed
104
    set_capability( "encoder", 100 );
Gildas Bazin's avatar
 
Gildas Bazin committed
105 106
    set_callbacks( OpenEncoder, CloseEncoder );
    add_shortcut( "theora" );
107 108 109 110

#   define ENC_CFG_PREFIX "sout-theora-"
    add_integer( ENC_CFG_PREFIX "quality", 2, NULL, ENC_QUALITY_TEXT,
                 ENC_QUALITY_LONGTEXT, VLC_FALSE );
Gildas Bazin's avatar
 
Gildas Bazin committed
111 112
vlc_module_end();

113 114 115 116
static const char *ppsz_enc_options[] = {
    "quality", NULL
};

Gildas Bazin's avatar
 
Gildas Bazin committed
117 118 119 120 121
/*****************************************************************************
 * OpenDecoder: probe the decoder and return score
 *****************************************************************************/
static int OpenDecoder( vlc_object_t *p_this )
{
Gildas Bazin's avatar
 
Gildas Bazin committed
122
    decoder_t *p_dec = (decoder_t*)p_this;
Gildas Bazin's avatar
 
Gildas Bazin committed
123
    decoder_sys_t *p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
124

Gildas Bazin's avatar
 
Gildas Bazin committed
125
    if( p_dec->fmt_in.i_codec != VLC_FOURCC('t','h','e','o') )
Gildas Bazin's avatar
 
Gildas Bazin committed
126 127 128 129
    {
        return VLC_EGENERIC;
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
130
    /* Allocate the memory needed to store the decoder's structure */
Gildas Bazin's avatar
 
Gildas Bazin committed
131
    if( ( p_dec->p_sys = p_sys =
Gildas Bazin's avatar
 
Gildas Bazin committed
132 133 134 135 136 137 138
          (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
139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156
    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
157 158
    return VLC_SUCCESS;
}
Gildas Bazin's avatar
 
Gildas Bazin committed
159 160 161 162 163 164 165

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
166 167 168 169 170
    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
171 172 173 174 175

    return i_ret;
}

/****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
176
 * DecodeBlock: the whole thing
Gildas Bazin's avatar
 
Gildas Bazin committed
177 178 179
 ****************************************************************************
 * This function must be fed with ogg packets.
 ****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
180
static void *DecodeBlock( decoder_t *p_dec, block_t **pp_block )
Gildas Bazin's avatar
 
Gildas Bazin committed
181 182
{
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
183
    block_t *p_block;
Gildas Bazin's avatar
 
Gildas Bazin committed
184
    ogg_packet oggpacket;
Gildas Bazin's avatar
 
Gildas Bazin committed
185 186 187 188

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

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

Gildas Bazin's avatar
 
Gildas Bazin committed
190 191 192 193 194 195 196
    /* 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
197

Gildas Bazin's avatar
 
Gildas Bazin committed
198
    if( p_sys->i_headers == 0 )
Gildas Bazin's avatar
 
Gildas Bazin committed
199
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
200
        /* Take care of the initial Theora header */
Gildas Bazin's avatar
 
Gildas Bazin committed
201

Gildas Bazin's avatar
 
Gildas Bazin committed
202 203 204
        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
205
            msg_Err( p_dec, "This bitstream does not contain Theora "
Benjamin Pracht's avatar
Benjamin Pracht committed
206
                     "video data." );
Gildas Bazin's avatar
 
Gildas Bazin committed
207
            block_Release( p_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
208
            return NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
209 210
        }
        p_sys->i_headers++;
Gildas Bazin's avatar
 
Gildas Bazin committed
211

Gildas Bazin's avatar
 
Gildas Bazin committed
212 213 214
        /* 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
215

216
        if( p_sys->ti.aspect_denominator && p_sys->ti.aspect_numerator )
Gildas Bazin's avatar
 
Gildas Bazin committed
217
            p_dec->fmt_out.video.i_aspect = ((int64_t)VOUT_ASPECT_FACTOR) *
Gildas Bazin's avatar
 
Gildas Bazin committed
218
                p_sys->ti.aspect_numerator / p_sys->ti.aspect_denominator;
Gildas Bazin's avatar
 
Gildas Bazin committed
219
        else
Gildas Bazin's avatar
 
Gildas Bazin committed
220 221
            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
222

Gildas Bazin's avatar
 
Gildas Bazin committed
223
        msg_Dbg( p_dec, "%dx%d %.02f fps video, frame content "
Benjamin Pracht's avatar
Benjamin Pracht committed
224
                 "is %dx%d with offset (%d,%d).",
Gildas Bazin's avatar
 
Gildas Bazin committed
225 226 227 228
                 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
229

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

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

Gildas Bazin's avatar
 
Gildas Bazin committed
243 244
        ParseTheoraComments( p_dec );

Gildas Bazin's avatar
 
Gildas Bazin committed
245
        return ProcessPacket( p_dec, &oggpacket, pp_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
246 247
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
248
    if( p_sys->i_headers == 2 )
Gildas Bazin's avatar
 
Gildas Bazin committed
249
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
250 251 252 253 254
        /* 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
255
            msg_Err( p_dec, "3rd Theora header is corrupted." );
Gildas Bazin's avatar
 
Gildas Bazin committed
256
            return NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
257 258
        }
        p_sys->i_headers++;
Sam Hocevar's avatar
Sam Hocevar committed
259

Gildas Bazin's avatar
 
Gildas Bazin committed
260 261 262 263 264 265
        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
266
        return ProcessPacket( p_dec, &oggpacket, pp_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
267 268
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
269
    return ProcessPacket( p_dec, &oggpacket, pp_block );
Gildas Bazin's avatar
 
Gildas Bazin committed
270
}
Gildas Bazin's avatar
 
Gildas Bazin committed
271

Gildas Bazin's avatar
 
Gildas Bazin committed
272
/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
273
 * ProcessPacket: processes a theora packet.
Gildas Bazin's avatar
 
Gildas Bazin committed
274
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
275 276
static void *ProcessPacket( decoder_t *p_dec, ogg_packet *p_oggpacket,
                            block_t **pp_block )
Gildas Bazin's avatar
 
Gildas Bazin committed
277 278
{
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
279 280
    block_t *p_block = *pp_block;
    void *p_buf;
Gildas Bazin's avatar
 
Gildas Bazin committed
281

Gildas Bazin's avatar
 
Gildas Bazin committed
282
    /* Date management */
Gildas Bazin's avatar
 
Gildas Bazin committed
283
    if( p_block->i_pts > 0 && p_block->i_pts != p_sys->i_pts )
Gildas Bazin's avatar
 
Gildas Bazin committed
284
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
285
        p_sys->i_pts = p_block->i_pts;
Gildas Bazin's avatar
 
Gildas Bazin committed
286 287
    }

Gildas Bazin's avatar
 
Gildas Bazin committed
288 289
    if( p_sys->b_packetizer )
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
290 291 292 293 294 295 296 297 298
        /* 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
299 300 301
    }
    else
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
302 303 304 305
        if( p_sys->i_headers >= 3 )
            p_buf = DecodePacket( p_dec, p_oggpacket );
        else
            p_buf = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
306

Gildas Bazin's avatar
 
Gildas Bazin committed
307
        if( p_block )
Gildas Bazin's avatar
 
Gildas Bazin committed
308
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
309 310
            block_Release( p_block );
            *pp_block = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
311 312
        }
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
313 314

    /* Date management */
Gildas Bazin's avatar
 
Gildas Bazin committed
315 316
    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
317

Gildas Bazin's avatar
 
Gildas Bazin committed
318
    return p_buf;
Gildas Bazin's avatar
 
Gildas Bazin committed
319 320 321
}

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
322
 * DecodePacket: decodes a Theora packet.
Gildas Bazin's avatar
 
Gildas Bazin committed
323
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
324
static picture_t *DecodePacket( decoder_t *p_dec, ogg_packet *p_oggpacket )
Gildas Bazin's avatar
 
Gildas Bazin committed
325
{
Gildas Bazin's avatar
 
Gildas Bazin committed
326
    decoder_sys_t *p_sys = p_dec->p_sys;
Gildas Bazin's avatar
 
Gildas Bazin committed
327 328
    picture_t *p_pic;
    yuv_buffer yuv;
Gildas Bazin's avatar
 
Gildas Bazin committed
329

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

Gildas Bazin's avatar
 
Gildas Bazin committed
332 333
    /* Decode */
    theora_decode_YUVout( &p_sys->td, &yuv );
Gildas Bazin's avatar
 
Gildas Bazin committed
334

Gildas Bazin's avatar
 
Gildas Bazin committed
335 336 337
    /* 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
338

Gildas Bazin's avatar
 
Gildas Bazin committed
339
    theora_CopyPicture( p_dec, p_pic, &yuv );
Gildas Bazin's avatar
 
Gildas Bazin committed
340

Gildas Bazin's avatar
 
Gildas Bazin committed
341
    p_pic->date = p_sys->i_pts;
Gildas Bazin's avatar
 
Gildas Bazin committed
342

Gildas Bazin's avatar
 
Gildas Bazin committed
343
    return p_pic;
Gildas Bazin's avatar
 
Gildas Bazin committed
344 345 346
}

/*****************************************************************************
Gildas Bazin's avatar
 
Gildas Bazin committed
347
 * ParseTheoraComments: FIXME should be done in demuxer
Gildas Bazin's avatar
 
Gildas Bazin committed
348
 *****************************************************************************/
Gildas Bazin's avatar
 
Gildas Bazin committed
349
static void ParseTheoraComments( decoder_t *p_dec )
Gildas Bazin's avatar
 
Gildas Bazin committed
350
{
Gildas Bazin's avatar
 
Gildas Bazin committed
351
    input_thread_t *p_input = (input_thread_t *)p_dec->p_parent;
352

Gildas Bazin's avatar
 
Gildas Bazin committed
353 354 355
    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
356
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
357 358 359
        psz_comment = strdup( p_dec->p_sys->tc.user_comments[i] );
        if( !psz_comment )
        {
Benjamin Pracht's avatar
Benjamin Pracht committed
360
            msg_Warn( p_dec, "out of memory" );
Gildas Bazin's avatar
 
Gildas Bazin committed
361 362 363 364 365 366 367 368
            break;
        }
        psz_name = psz_comment;
        psz_value = strchr( psz_comment, '=' );
        if( psz_value )
        {
            *psz_value = '\0';
            psz_value++;
369 370
            input_Control( p_input, INPUT_ADD_INFO, _("Theora comment"),
                           psz_name, psz_value );
Gildas Bazin's avatar
 
Gildas Bazin committed
371 372 373 374 375 376 377
        }
        free( psz_comment );
        i++;
    }
}

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

Gildas Bazin's avatar
 
Gildas Bazin committed
385 386
    theora_info_clear( &p_sys->ti );
    theora_comment_clear( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
387

Gildas Bazin's avatar
 
Gildas Bazin committed
388
    free( p_sys );
Gildas Bazin's avatar
 
Gildas Bazin committed
389 390 391 392 393 394
}

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

    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
407 408 409 410 411 412 413 414
        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
415
        }
Gildas Bazin's avatar
 
Gildas Bazin committed
416 417

        p_src += (i_src_yoffset * i_src_stride + i_src_yoffset);
Gildas Bazin's avatar
 
Gildas Bazin committed
418 419 420

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

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

    /*
     * 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;
458 459
    vlc_value_t val;
    int i_quality;
Gildas Bazin's avatar
 
Gildas Bazin committed
460

461 462
    if( p_enc->fmt_out.i_codec != VLC_FOURCC('t','h','e','o') &&
        !p_enc->b_force )
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');
487
    p_enc->fmt_out.i_codec = VLC_FOURCC('t','h','e','o');
Gildas Bazin's avatar
 
Gildas Bazin committed
488

489 490 491 492 493 494 495
    sout_ParseCfg( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );

    var_Get( p_enc, ENC_CFG_PREFIX "quality", &val );
    i_quality = val.i_int;
    if( i_quality > 10 ) i_quality = 10;
    if( i_quality < 0 ) i_quality = 0;

Gildas Bazin's avatar
 
Gildas Bazin committed
496 497
    theora_info_init( &p_sys->ti );

Gildas Bazin's avatar
 
Gildas Bazin committed
498 499 500 501
    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;
502 503 504 505 506 507 508 509 510 511 512 513 514 515
    p_sys->ti.offset_x = 0 /*frame_x_offset*/;
    p_sys->ti.offset_y = 0/*frame_y_offset*/;

    if( !p_enc->fmt_in.video.i_frame_rate ||
        !p_enc->fmt_in.video.i_frame_rate_base )
    {
        p_sys->ti.fps_numerator = 25;
        p_sys->ti.fps_denominator = 1;
    }
    else
    {
        p_sys->ti.fps_numerator = p_enc->fmt_in.video.i_frame_rate;
        p_sys->ti.fps_denominator = p_enc->fmt_in.video.i_frame_rate_base;
    }
Gildas Bazin's avatar
 
Gildas Bazin committed
516 517 518 519 520 521 522 523 524 525 526 527

    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
528
    p_sys->ti.target_bitrate = p_enc->fmt_out.i_bitrate;
529
    p_sys->ti.quality = ((float)i_quality) * 6.3;
Gildas Bazin's avatar
 
Gildas Bazin committed
530 531 532 533 534 535

    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
536
    p_sys->ti.keyframe_data_target_bitrate = p_enc->fmt_out.i_bitrate * 1.5;
Gildas Bazin's avatar
 
Gildas Bazin committed
537 538 539 540 541 542
    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
543
    theora_comment_init( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
544

Gildas Bazin's avatar
 
Gildas Bazin committed
545
    p_sys->b_headers = VLC_FALSE;
Gildas Bazin's avatar
 
Gildas Bazin committed
546 547 548 549 550 551 552 553 554 555 556 557

    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
558
    block_t *p_chain = NULL;
Gildas Bazin's avatar
 
Gildas Bazin committed
559 560

    /* Create theora headers */
Gildas Bazin's avatar
 
Gildas Bazin committed
561
    if( !p_sys->b_headers )
Gildas Bazin's avatar
 
Gildas Bazin committed
562
    {
Gildas Bazin's avatar
 
Gildas Bazin committed
563
        ogg_packet oggpackets;
Gildas Bazin's avatar
 
Gildas Bazin committed
564
        int i;
Eric Petit's avatar
Eric Petit committed
565
        block_t *p_block;
Gildas Bazin's avatar
 
Gildas Bazin committed
566

Gildas Bazin's avatar
 
Gildas Bazin committed
567 568 569
        /* Ogg packet to block */
        for( i = 0; i < 3; i++ )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
570 571 572 573 574 575 576 577 578 579 580 581 582
            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
583
            p_block = block_New( p_enc, oggpackets.bytes );
Gildas Bazin's avatar
 
Gildas Bazin committed
584
            memcpy( p_block->p_buffer, oggpackets.packet, oggpackets.bytes );
Gildas Bazin's avatar
 
Gildas Bazin committed
585 586 587
            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
588

Gildas Bazin's avatar
 
Gildas Bazin committed
589 590 591 592
        p_sys->b_headers = VLC_TRUE;
    }

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

/****************************************************************************
 * 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
622 623 624 625 626
    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
627 628 629 630 631

    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
632
    memcpy( p_block->p_buffer, oggpacket.packet, oggpacket.bytes );
633
    p_block->i_dts = p_block->i_pts = p_pict->date;
Gildas Bazin's avatar
 
Gildas Bazin committed
634 635 636 637 638 639 640 641 642 643 644 645 646

    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 );
647
    theora_comment_clear( &p_sys->tc );
Gildas Bazin's avatar
 
Gildas Bazin committed
648 649 650

    free( p_sys );
}