theora.c 27.3 KB
Newer Older
gbazin's avatar
 
gbazin committed
1 2 3
/*****************************************************************************
 * theora.c: theora decoder module making use of libtheora.
 *****************************************************************************
4
 * Copyright (C) 1999-2001 the VideoLAN team
5
 * $Id$
gbazin's avatar
 
gbazin committed
6
 *
7
 * Authors: Gildas Bazin <gbazin@videolan.org>
gbazin's avatar
 
gbazin 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
 *
gbazin's avatar
 
gbazin committed
14 15 16 17 18 19 20
 * 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
dionoea's avatar
dionoea committed
21
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
gbazin's avatar
 
gbazin committed
22 23 24 25 26
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
27 28 29 30
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

gbazin's avatar
 
gbazin committed
31
#include <vlc/vlc.h>
zorglub's avatar
zorglub committed
32 33 34
#include <vlc_codec.h>
#include <vlc_vout.h>
#include <vlc_sout.h>
zorglub's avatar
zorglub committed
35
#include <vlc_input.h>
gbazin's avatar
 
gbazin committed
36 37 38 39 40
#include <ogg/ogg.h>

#include <theora/theora.h>

/*****************************************************************************
gbazin's avatar
 
gbazin committed
41
 * decoder_sys_t : theora decoder descriptor
gbazin's avatar
 
gbazin committed
42
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
43
struct decoder_sys_t
gbazin's avatar
 
gbazin committed
44
{
gbazin's avatar
 
gbazin committed
45
    /* Module mode */
46
    bool b_packetizer;
gbazin's avatar
 
gbazin committed
47

gbazin's avatar
 
gbazin committed
48
    /*
gbazin's avatar
 
gbazin committed
49
     * Input properties
gbazin's avatar
 
gbazin committed
50
     */
gbazin's avatar
 
gbazin committed
51
    int i_headers;
gbazin's avatar
 
gbazin committed
52 53 54 55 56

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

60 61 62
    /*
     * Decoding properties
     */
63
    bool b_decoded_first_keyframe;
64

gbazin's avatar
 
gbazin committed
65 66 67 68 69
    /*
     * Common properties
     */
    mtime_t i_pts;
};
gbazin's avatar
 
gbazin committed
70 71 72 73

/*****************************************************************************
 * Local prototypes
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
74 75 76
static int  OpenDecoder   ( vlc_object_t * );
static int  OpenPacketizer( vlc_object_t * );
static void CloseDecoder  ( vlc_object_t * );
gbazin's avatar
 
gbazin committed
77

gbazin's avatar
 
gbazin committed
78
static void *DecodeBlock  ( decoder_t *, block_t ** );
79
static int  ProcessHeaders( decoder_t * );
gbazin's avatar
 
gbazin committed
80
static void *ProcessPacket ( decoder_t *, ogg_packet *, block_t ** );
gbazin's avatar
 
gbazin committed
81

gbazin's avatar
 
gbazin committed
82
static picture_t *DecodePacket( decoder_t *, ogg_packet * );
gbazin's avatar
 
gbazin committed
83

gbazin's avatar
 
gbazin committed
84 85
static void ParseTheoraComments( decoder_t * );
static void theora_CopyPicture( decoder_t *, picture_t *, yuv_buffer * );
86

gbazin's avatar
 
gbazin committed
87 88 89 90
static int  OpenEncoder( vlc_object_t *p_this );
static void CloseEncoder( vlc_object_t *p_this );
static block_t *Encode( encoder_t *p_enc, picture_t *p_pict );

gbazin's avatar
 
gbazin committed
91 92 93
/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
94 95
#define ENC_QUALITY_TEXT N_("Encoding quality")
#define ENC_QUALITY_LONGTEXT N_( \
96
  "Enforce a quality between 1 (low) and 10 (high), instead " \
97 98
  "of specifying a particular bitrate. This will produce a VBR stream." )

gbazin's avatar
 
gbazin committed
99
vlc_module_begin();
zorglub's avatar
zorglub committed
100 101
    set_category( CAT_INPUT );
    set_subcategory( SUBCAT_INPUT_VCODEC );
102
    set_shortname( "Theora" );
gbazin's avatar
 
gbazin committed
103
    set_description( _("Theora video decoder") );
gbazin's avatar
 
gbazin committed
104
    set_capability( "decoder", 100 );
gbazin's avatar
 
gbazin committed
105
    set_callbacks( OpenDecoder, CloseDecoder );
gbazin's avatar
 
gbazin committed
106
    add_shortcut( "theora" );
gbazin's avatar
 
gbazin committed
107 108 109 110

    add_submodule();
    set_description( _("Theora video packetizer") );
    set_capability( "packetizer", 100 );
gbazin's avatar
 
gbazin committed
111
    set_callbacks( OpenPacketizer, CloseDecoder );
gbazin's avatar
 
gbazin committed
112 113 114

    add_submodule();
    set_description( _("Theora video encoder") );
115
    set_capability( "encoder", 150 );
gbazin's avatar
 
gbazin committed
116
    set_callbacks( OpenEncoder, CloseEncoder );
117 118 119

#   define ENC_CFG_PREFIX "sout-theora-"
    add_integer( ENC_CFG_PREFIX "quality", 2, NULL, ENC_QUALITY_TEXT,
120
                 ENC_QUALITY_LONGTEXT, false );
gbazin's avatar
 
gbazin committed
121 122
vlc_module_end();

123 124 125 126
static const char *ppsz_enc_options[] = {
    "quality", NULL
};

gbazin's avatar
 
gbazin committed
127 128 129 130 131
/*****************************************************************************
 * OpenDecoder: probe the decoder and return score
 *****************************************************************************/
static int OpenDecoder( vlc_object_t *p_this )
{
gbazin's avatar
 
gbazin committed
132
    decoder_t *p_dec = (decoder_t*)p_this;
gbazin's avatar
 
gbazin committed
133
    decoder_sys_t *p_sys;
gbazin's avatar
 
gbazin committed
134

gbazin's avatar
 
gbazin committed
135
    if( p_dec->fmt_in.i_codec != VLC_FOURCC('t','h','e','o') )
gbazin's avatar
 
gbazin committed
136 137 138 139
    {
        return VLC_EGENERIC;
    }

gbazin's avatar
 
gbazin committed
140
    /* Allocate the memory needed to store the decoder's structure */
gbazin's avatar
 
gbazin committed
141
    if( ( p_dec->p_sys = p_sys =
gbazin's avatar
 
gbazin committed
142 143 144 145 146
          (decoder_sys_t *)malloc(sizeof(decoder_sys_t)) ) == NULL )
    {
        msg_Err( p_dec, "out of memory" );
        return VLC_EGENERIC;
    }
147
    p_dec->p_sys->b_packetizer = false;
gbazin's avatar
 
gbazin committed
148

gbazin's avatar
 
gbazin committed
149
    p_sys->i_pts = 0;
150
    p_sys->b_decoded_first_keyframe = false;
gbazin's avatar
 
gbazin committed
151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167

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

gbazin's avatar
 
gbazin committed
168 169
    return VLC_SUCCESS;
}
gbazin's avatar
 
gbazin committed
170 171 172 173 174 175 176

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

    int i_ret = OpenDecoder( p_this );

gbazin's avatar
 
gbazin committed
177 178
    if( i_ret == VLC_SUCCESS )
    {
179
        p_dec->p_sys->b_packetizer = true;
gbazin's avatar
 
gbazin committed
180 181
        p_dec->fmt_out.i_codec = VLC_FOURCC( 't', 'h', 'e', 'o' );
    }
gbazin's avatar
 
gbazin committed
182 183 184 185 186

    return i_ret;
}

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

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

    p_block = *pp_block;
gbazin's avatar
 
gbazin committed
200

gbazin's avatar
 
gbazin committed
201 202 203 204 205 206 207
    /* 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;
gbazin's avatar
 
gbazin committed
208

209 210
    /* Check for headers */
    if( p_sys->i_headers == 0 && p_dec->fmt_in.i_extra )
gbazin's avatar
 
gbazin committed
211
    {
212 213 214 215 216 217 218 219 220 221 222
        /* Headers already available as extra data */
        p_sys->i_headers = 3;
    }
    else if( oggpacket.bytes && p_sys->i_headers < 3 )
    {
        /* Backup headers as extra data */
        uint8_t *p_extra;

        p_dec->fmt_in.p_extra =
            realloc( p_dec->fmt_in.p_extra, p_dec->fmt_in.i_extra +
                     oggpacket.bytes + 2 );
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
223
        p_extra = ((uint8_t *)p_dec->fmt_in.p_extra) + p_dec->fmt_in.i_extra;
224 225 226 227 228 229 230 231 232 233
        *(p_extra++) = oggpacket.bytes >> 8;
        *(p_extra++) = oggpacket.bytes & 0xFF;

        memcpy( p_extra, oggpacket.packet, oggpacket.bytes );
        p_dec->fmt_in.i_extra += oggpacket.bytes + 2;

        block_Release( *pp_block );
        p_sys->i_headers++;
        return NULL;
    }
gbazin's avatar
 
gbazin committed
234

235 236 237
    if( p_sys->i_headers == 3 )
    {
        if( ProcessHeaders( p_dec ) != VLC_SUCCESS )
gbazin's avatar
 
gbazin committed
238
        {
239 240 241
            p_sys->i_headers = 0;
            p_dec->fmt_in.i_extra = 0;
            block_Release( *pp_block );
gbazin's avatar
 
gbazin committed
242
            return NULL;
gbazin's avatar
 
gbazin committed
243
        }
244 245
        else p_sys->i_headers++;
    }
gbazin's avatar
 
gbazin committed
246

247 248
    return ProcessPacket( p_dec, &oggpacket, pp_block );
}
Sam Hocevar's avatar
Sam Hocevar committed
249

250 251 252 253 254 255 256 257 258
/*****************************************************************************
 * ProcessHeaders: process Theora headers.
 *****************************************************************************/
static int ProcessHeaders( decoder_t *p_dec )
{
    decoder_sys_t *p_sys = p_dec->p_sys;
    ogg_packet oggpacket;
    uint8_t *p_extra;
    int i_extra;
gbazin's avatar
 
gbazin committed
259

260
    if( !p_dec->fmt_in.i_extra ) return VLC_EGENERIC;
gbazin's avatar
 
gbazin committed
261

262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
    oggpacket.granulepos = -1;
    oggpacket.b_o_s = 1; /* yes this actually is a b_o_s packet :) */
    oggpacket.e_o_s = 0;
    oggpacket.packetno = 0;
    p_extra = p_dec->fmt_in.p_extra;
    i_extra = p_dec->fmt_in.i_extra;

    /* Take care of the initial Vorbis header */
    oggpacket.bytes = *(p_extra++) << 8;
    oggpacket.bytes |= (*(p_extra++) & 0xFF);
    oggpacket.packet = p_extra;
    p_extra += oggpacket.bytes;
    i_extra -= (oggpacket.bytes + 2);
    if( i_extra < 0 )
    {
        msg_Err( p_dec, "header data corrupted");
        return VLC_EGENERIC;
gbazin's avatar
 
gbazin committed
279 280
    }

281
    if( theora_decode_header( &p_sys->ti, &p_sys->tc, &oggpacket ) < 0 )
gbazin's avatar
 
gbazin committed
282
    {
283
        msg_Err( p_dec, "this bitstream does not contain Theora video data" );
284 285
        return VLC_EGENERIC;
    }
Sam Hocevar's avatar
Sam Hocevar committed
286

287 288 289
    /* 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;
290 291 292 293 294
    if( p_sys->ti.frame_width && p_sys->ti.frame_height )
    {
        p_dec->fmt_out.video.i_width = p_sys->ti.frame_width;
        p_dec->fmt_out.video.i_height = p_sys->ti.frame_height;
    }
gbazin's avatar
 
gbazin committed
295

296 297 298
    if( p_sys->ti.aspect_denominator && p_sys->ti.aspect_numerator )
    {
        p_dec->fmt_out.video.i_aspect = ((int64_t)VOUT_ASPECT_FACTOR) *
299 300
            ( p_sys->ti.aspect_numerator * p_dec->fmt_out.video.i_width ) /
            ( p_sys->ti.aspect_denominator * p_dec->fmt_out.video.i_height );
gbazin's avatar
 
gbazin committed
301
    }
302 303 304 305 306 307
    else
    {
        p_dec->fmt_out.video.i_aspect = VOUT_ASPECT_FACTOR *
            p_sys->ti.frame_width / p_sys->ti.frame_height;
    }

308 309 310 311 312 313
    if( p_sys->ti.fps_numerator > 0 && p_sys->ti.fps_denominator > 0 )
    {
        p_dec->fmt_out.video.i_frame_rate = p_sys->ti.fps_numerator;
        p_dec->fmt_out.video.i_frame_rate_base = p_sys->ti.fps_denominator;
    }

314 315 316 317 318 319
    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 );
gbazin's avatar
 
gbazin committed
320

321 322 323 324 325 326 327 328 329 330 331 332 333 334 335
    /* Sanity check that seems necessary for some corrupted files */
    if( p_sys->ti.width < p_sys->ti.frame_width ||
        p_sys->ti.height < p_sys->ti.frame_height )
    {
        msg_Warn( p_dec, "trying to correct invalid theora header "
                  "(frame size (%dx%d) is smaller than frame content (%d,%d))",
                  p_sys->ti.width, p_sys->ti.height,
                  p_sys->ti.frame_width, p_sys->ti.frame_height );

        if( p_sys->ti.width < p_sys->ti.frame_width )
            p_sys->ti.width = p_sys->ti.frame_width;
        if( p_sys->ti.height < p_sys->ti.frame_height )
            p_sys->ti.height = p_sys->ti.frame_height;
    }

336 337 338 339 340 341 342 343
    /* The next packet in order is the comments header */
    oggpacket.b_o_s = 0;
    oggpacket.bytes = *(p_extra++) << 8;
    oggpacket.bytes |= (*(p_extra++) & 0xFF);
    oggpacket.packet = p_extra;
    p_extra += oggpacket.bytes;
    i_extra -= (oggpacket.bytes + 2);
    if( i_extra < 0 )
gbazin's avatar
 
gbazin committed
344
    {
345 346 347
        msg_Err( p_dec, "header data corrupted");
        return VLC_EGENERIC;
    }
Sam Hocevar's avatar
Sam Hocevar committed
348

349 350 351 352 353 354 355 356
    /* 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;
    }

    ParseTheoraComments( p_dec );
gbazin's avatar
 
gbazin committed
357

358 359 360 361 362 363 364 365 366 367 368
    /* 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. */
    oggpacket.bytes = *(p_extra++) << 8;
    oggpacket.bytes |= (*(p_extra++) & 0xFF);
    oggpacket.packet = p_extra;
    i_extra -= (oggpacket.bytes + 2);
    if( i_extra < 0 )
    {
        msg_Err( p_dec, "header data corrupted");
        return VLC_EGENERIC;
gbazin's avatar
 
gbazin committed
369 370
    }

371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394
    /* 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;
    }

    if( !p_sys->b_packetizer )
    {
        /* We have all the headers, initialize decoder */
        theora_decode_init( &p_sys->td, &p_sys->ti );
    }
    else
    {
        p_dec->fmt_out.i_extra = p_dec->fmt_in.i_extra;
        p_dec->fmt_out.p_extra =
            realloc( p_dec->fmt_out.p_extra, p_dec->fmt_out.i_extra );
        memcpy( p_dec->fmt_out.p_extra,
                p_dec->fmt_in.p_extra, p_dec->fmt_out.i_extra );
    }

    return VLC_SUCCESS;
gbazin's avatar
 
gbazin committed
395
}
gbazin's avatar
 
gbazin committed
396

gbazin's avatar
 
gbazin committed
397
/*****************************************************************************
gbazin's avatar
 
gbazin committed
398
 * ProcessPacket: processes a theora packet.
gbazin's avatar
 
gbazin committed
399
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
400 401
static void *ProcessPacket( decoder_t *p_dec, ogg_packet *p_oggpacket,
                            block_t **pp_block )
gbazin's avatar
 
gbazin committed
402 403
{
    decoder_sys_t *p_sys = p_dec->p_sys;
gbazin's avatar
 
gbazin committed
404 405
    block_t *p_block = *pp_block;
    void *p_buf;
gbazin's avatar
 
gbazin committed
406

Laurent Aimar's avatar
Laurent Aimar committed
407
    if( ( p_block->i_flags&(BLOCK_FLAG_DISCONTINUITY|BLOCK_FLAG_CORRUPTED) ) != 0 )
408 409 410 411 412 413 414
    {
        /* Don't send the the first packet after a discontinuity to
         * theora_decode, otherwise we get purple/green display artifacts
         * appearing in the video output */
        return NULL;
    }

gbazin's avatar
 
gbazin committed
415
    /* Date management */
gbazin's avatar
 
gbazin committed
416
    if( p_block->i_pts > 0 && p_block->i_pts != p_sys->i_pts )
gbazin's avatar
 
gbazin committed
417
    {
gbazin's avatar
 
gbazin committed
418
        p_sys->i_pts = p_block->i_pts;
gbazin's avatar
 
gbazin committed
419 420
    }

421 422
    *pp_block = NULL; /* To avoid being fed the same packet again */

gbazin's avatar
 
gbazin committed
423 424
    if( p_sys->b_packetizer )
    {
gbazin's avatar
 
gbazin committed
425 426 427 428 429 430 431 432 433
        /* 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;
gbazin's avatar
 
gbazin committed
434 435 436
    }
    else
    {
gbazin's avatar
 
gbazin committed
437 438 439 440
        if( p_sys->i_headers >= 3 )
            p_buf = DecodePacket( p_dec, p_oggpacket );
        else
            p_buf = NULL;
gbazin's avatar
 
gbazin committed
441

442
        if( p_block ) block_Release( p_block );
gbazin's avatar
 
gbazin committed
443
    }
gbazin's avatar
 
gbazin committed
444 445

    /* Date management */
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
446
    p_sys->i_pts += ( INT64_C(1000000) * p_sys->ti.fps_denominator /
gbazin's avatar
 
gbazin committed
447
                      p_sys->ti.fps_numerator ); /* 1 frame per packet */
gbazin's avatar
 
gbazin committed
448

gbazin's avatar
 
gbazin committed
449
    return p_buf;
gbazin's avatar
 
gbazin committed
450 451 452
}

/*****************************************************************************
gbazin's avatar
 
gbazin committed
453
 * DecodePacket: decodes a Theora packet.
gbazin's avatar
 
gbazin committed
454
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
455
static picture_t *DecodePacket( decoder_t *p_dec, ogg_packet *p_oggpacket )
gbazin's avatar
 
gbazin committed
456
{
gbazin's avatar
 
gbazin committed
457
    decoder_sys_t *p_sys = p_dec->p_sys;
gbazin's avatar
 
gbazin committed
458 459
    picture_t *p_pic;
    yuv_buffer yuv;
gbazin's avatar
 
gbazin committed
460

gbazin's avatar
 
gbazin committed
461
    theora_decode_packetin( &p_sys->td, p_oggpacket );
gbazin's avatar
 
gbazin committed
462

463 464 465
    /* Check for keyframe */
    if( !(p_oggpacket->packet[0] & 0x80) /* data packet */ &&
        !(p_oggpacket->packet[0] & 0x40) /* intra frame */ )
466
        p_sys->b_decoded_first_keyframe = true;
467 468 469 470 471 472 473 474 475 476

    /* If we haven't seen a single keyframe yet, don't let Theora decode
     * anything, otherwise we'll get display artifacts.  (This is impossible
     * in the general case, but can happen if e.g. we play a network stream
     * using a timed URL, such that the server doesn't start the video with a
     * keyframe). */
    if( p_sys->b_decoded_first_keyframe )
        theora_decode_YUVout( &p_sys->td, &yuv );
    else
        return NULL;
gbazin's avatar
 
gbazin committed
477

gbazin's avatar
 
gbazin committed
478 479 480
    /* Get a new picture */
    p_pic = p_dec->pf_vout_buffer_new( p_dec );
    if( !p_pic ) return NULL;
gbazin's avatar
 
gbazin committed
481

gbazin's avatar
 
gbazin committed
482
    theora_CopyPicture( p_dec, p_pic, &yuv );
gbazin's avatar
 
gbazin committed
483

gbazin's avatar
 
gbazin committed
484
    p_pic->date = p_sys->i_pts;
gbazin's avatar
 
gbazin committed
485

gbazin's avatar
 
gbazin committed
486
    return p_pic;
gbazin's avatar
 
gbazin committed
487 488 489
}

/*****************************************************************************
gbazin's avatar
 
gbazin committed
490
 * ParseTheoraComments: FIXME should be done in demuxer
gbazin's avatar
 
gbazin committed
491
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
492
static void ParseTheoraComments( decoder_t *p_dec )
gbazin's avatar
 
gbazin committed
493
{
gbazin's avatar
 
gbazin committed
494 495
    input_thread_t *p_input = (input_thread_t *)p_dec->p_parent;
    char *psz_name, *psz_value, *psz_comment;
496 497 498 499
    int i = 0;

    if( p_input->i_object_type != VLC_OBJECT_INPUT ) return;

gbazin's avatar
 
gbazin committed
500
    while ( i < p_dec->p_sys->tc.comments )
gbazin's avatar
 
gbazin committed
501
    {
gbazin's avatar
 
gbazin committed
502 503 504
        psz_comment = strdup( p_dec->p_sys->tc.user_comments[i] );
        if( !psz_comment )
        {
bigben's avatar
bigben committed
505
            msg_Warn( p_dec, "out of memory" );
gbazin's avatar
 
gbazin committed
506 507 508 509 510 511 512 513
            break;
        }
        psz_name = psz_comment;
        psz_value = strchr( psz_comment, '=' );
        if( psz_value )
        {
            *psz_value = '\0';
            psz_value++;
514
            input_Control( p_input, INPUT_ADD_INFO, _("Theora comment"),
515
                           psz_name, "%s", psz_value );
gbazin's avatar
 
gbazin committed
516 517 518 519 520 521 522
        }
        free( psz_comment );
        i++;
    }
}

/*****************************************************************************
gbazin's avatar
 
gbazin committed
523
 * CloseDecoder: theora decoder destruction
gbazin's avatar
 
gbazin committed
524
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
525
static void CloseDecoder( vlc_object_t *p_this )
gbazin's avatar
 
gbazin committed
526
{
gbazin's avatar
 
gbazin committed
527
    decoder_t *p_dec = (decoder_t *)p_this;
gbazin's avatar
 
gbazin committed
528
    decoder_sys_t *p_sys = p_dec->p_sys;
gbazin's avatar
 
gbazin committed
529

gbazin's avatar
 
gbazin committed
530 531
    theora_info_clear( &p_sys->ti );
    theora_comment_clear( &p_sys->tc );
gbazin's avatar
 
gbazin committed
532

gbazin's avatar
 
gbazin committed
533
    free( p_sys );
gbazin's avatar
 
gbazin committed
534 535 536 537 538 539
}

/*****************************************************************************
 * theora_CopyPicture: copy a picture from theora internal buffers to a
 *                     picture_t structure.
 *****************************************************************************/
gbazin's avatar
 
gbazin committed
540
static void theora_CopyPicture( decoder_t *p_dec, picture_t *p_pic,
gbazin's avatar
 
gbazin committed
541 542 543
                                yuv_buffer *yuv )
{
    int i_plane, i_line, i_width, i_dst_stride, i_src_stride;
gbazin's avatar
 
gbazin committed
544
    int i_src_xoffset, i_src_yoffset;
Sam Hocevar's avatar
Sam Hocevar committed
545
    uint8_t *p_dst, *p_src;
gbazin's avatar
 
gbazin committed
546 547 548 549 550 551

    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;
gbazin's avatar
 
gbazin committed
552 553 554 555 556 557 558 559
        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
560
        }
gbazin's avatar
 
gbazin committed
561

562
        p_src += (i_src_yoffset * i_src_stride + i_src_xoffset);
gbazin's avatar
 
gbazin committed
563

564
        for( i_line = 0; i_line < p_pic->p[i_plane].i_visible_lines; i_line++ )
gbazin's avatar
 
gbazin committed
565
        {
566 567
            vlc_memcpy( p_dst, p_src + i_src_xoffset,
                        i_plane ? yuv->uv_width : yuv->y_width );
gbazin's avatar
 
gbazin committed
568 569 570 571 572
            p_src += i_src_stride;
            p_dst += i_dst_stride;
        }
    }
}
gbazin's avatar
 
gbazin committed
573 574 575 576 577 578 579 580 581

/*****************************************************************************
 * encoder_sys_t : theora encoder descriptor
 *****************************************************************************/
struct encoder_sys_t
{
    /*
     * Input properties
     */
582
    bool b_headers;
gbazin's avatar
 
gbazin committed
583 584 585 586 587 588 589 590

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

591
    int i_width, i_height;
gbazin's avatar
 
gbazin committed
592 593 594 595 596 597 598 599 600
};

/*****************************************************************************
 * 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;
601
    ogg_packet header;
602
    uint8_t *p_extra;
603
    vlc_value_t val;
604
    int i_quality, i;
gbazin's avatar
 
gbazin committed
605

606 607
    if( p_enc->fmt_out.i_codec != VLC_FOURCC('t','h','e','o') &&
        !p_enc->b_force )
gbazin's avatar
 
gbazin committed
608 609 610 611 612 613 614 615 616 617 618 619 620
    {
        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_encode_video = Encode;
gbazin's avatar
 
gbazin committed
621
    p_enc->fmt_in.i_codec = VLC_FOURCC('I','4','2','0');
622
    p_enc->fmt_out.i_codec = VLC_FOURCC('t','h','e','o');
gbazin's avatar
 
gbazin committed
623

624
    config_ChainParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
625 626 627 628 629 630

    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;

gbazin's avatar
 
gbazin committed
631 632
    theora_info_init( &p_sys->ti );

gbazin's avatar
 
gbazin committed
633 634
    p_sys->ti.width = p_enc->fmt_in.video.i_width;
    p_sys->ti.height = p_enc->fmt_in.video.i_height;
635 636 637 638 639 640 641 642 643 644 645 646 647

    if( p_sys->ti.width % 16 || p_sys->ti.height % 16 )
    {
        /* Pictures from the transcoder should always have a pitch
         * which is a multiple of 16 */
        p_sys->ti.width = (p_sys->ti.width + 15) >> 4 << 4;
        p_sys->ti.height = (p_sys->ti.height + 15) >> 4 << 4;

        msg_Dbg( p_enc, "padding video from %dx%d to %dx%d",
                 p_enc->fmt_in.video.i_width, p_enc->fmt_in.video.i_height,
                 p_sys->ti.width, p_sys->ti.height );
    }

gbazin's avatar
 
gbazin committed
648 649
    p_sys->ti.frame_width = p_enc->fmt_in.video.i_width;
    p_sys->ti.frame_height = p_enc->fmt_in.video.i_height;
650
    p_sys->ti.offset_x = 0 /*frame_x_offset*/;
651
    p_sys->ti.offset_y = 0 /*frame_y_offset*/;
652

653 654 655
    p_sys->i_width = p_sys->ti.width;
    p_sys->i_height = p_sys->ti.height;

656 657 658 659 660 661 662 663 664 665 666
    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;
    }
gbazin's avatar
 
gbazin committed
667 668 669

    if( p_enc->fmt_in.video.i_aspect )
    {
670 671
        uint64_t i_num, i_den;
        unsigned i_dst_num, i_dst_den;
672 673 674

        i_num = p_enc->fmt_in.video.i_aspect * (int64_t)p_sys->ti.height;
        i_den = VOUT_ASPECT_FACTOR * p_sys->ti.width;
675
        vlc_ureduce( &i_dst_num, &i_dst_den, i_num, i_den, 0 );
676 677
        p_sys->ti.aspect_numerator = i_dst_num;
        p_sys->ti.aspect_denominator = i_dst_den;
gbazin's avatar
 
gbazin committed
678 679 680 681 682 683 684
    }
    else
    {
        p_sys->ti.aspect_numerator = 4;
        p_sys->ti.aspect_denominator = 3;
    }

gbazin's avatar
 
gbazin committed
685
    p_sys->ti.target_bitrate = p_enc->fmt_out.i_bitrate;
686
    p_sys->ti.quality = ((float)i_quality) * 6.3;
gbazin's avatar
 
gbazin committed
687 688 689 690 691 692

    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;
gbazin's avatar
 
gbazin committed
693
    p_sys->ti.keyframe_data_target_bitrate = p_enc->fmt_out.i_bitrate * 1.5;
gbazin's avatar
 
gbazin committed
694 695 696 697 698 699
    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 );
gbazin's avatar
 
gbazin committed
700
    theora_comment_init( &p_sys->tc );
gbazin's avatar
 
gbazin committed
701

702
    /* Create and store headers */
703
    p_enc->fmt_out.i_extra = 3 * 2;
704
    for( i = 0; i < 3; i++ )
gbazin's avatar
 
gbazin committed
705
    {
706 707 708 709 710 711 712 713 714 715 716 717 718 719 720
        if( i == 0 ) theora_encode_header( &p_sys->td, &header );
        else if( i == 1 ) theora_encode_comment( &p_sys->tc, &header );
        else if( i == 2 ) theora_encode_tables( &p_sys->td, &header );

        p_enc->fmt_out.p_extra =
            realloc( p_enc->fmt_out.p_extra,
                     p_enc->fmt_out.i_extra + header.bytes );
        p_extra = p_enc->fmt_out.p_extra;
        p_extra += p_enc->fmt_out.i_extra + (i-3)*2;
        p_enc->fmt_out.i_extra += header.bytes;

        *(p_extra++) = header.bytes >> 8;
        *(p_extra++) = header.bytes & 0xFF;

        memcpy( p_extra, header.packet, header.bytes );
gbazin's avatar
 
gbazin committed
721 722
    }

723
    return VLC_SUCCESS;
gbazin's avatar
 
gbazin committed
724 725 726 727 728 729 730 731 732 733 734 735 736
}

/****************************************************************************
 * 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;
737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790
    int i;

    /* Sanity check */
    if( p_pict->p[0].i_pitch < (int)p_sys->i_width ||
        p_pict->p[0].i_lines < (int)p_sys->i_height )
    {
        msg_Warn( p_enc, "frame is smaller than encoding size"
                  "(%ix%i->%ix%i) -> dropping frame",
                  p_pict->p[0].i_pitch, p_pict->p[0].i_lines,
                  p_sys->i_width, p_sys->i_height );
        return NULL;
    }

    /* Fill padding */
    if( p_pict->p[0].i_visible_pitch < (int)p_sys->i_width )
    {
        for( i = 0; i < p_sys->i_height; i++ )
        {
            memset( p_pict->p[0].p_pixels + i * p_pict->p[0].i_pitch +
                    p_pict->p[0].i_visible_pitch,
                    *( p_pict->p[0].p_pixels + i * p_pict->p[0].i_pitch +
                       p_pict->p[0].i_visible_pitch - 1 ),
                    p_sys->i_width - p_pict->p[0].i_visible_pitch );
        }
        for( i = 0; i < p_sys->i_height / 2; i++ )
        {
            memset( p_pict->p[1].p_pixels + i * p_pict->p[1].i_pitch +
                    p_pict->p[1].i_visible_pitch,
                    *( p_pict->p[1].p_pixels + i * p_pict->p[1].i_pitch +
                       p_pict->p[1].i_visible_pitch - 1 ),
                    p_sys->i_width / 2 - p_pict->p[1].i_visible_pitch );
            memset( p_pict->p[2].p_pixels + i * p_pict->p[2].i_pitch +
                    p_pict->p[2].i_visible_pitch,
                    *( p_pict->p[2].p_pixels + i * p_pict->p[2].i_pitch +
                       p_pict->p[2].i_visible_pitch - 1 ),
                    p_sys->i_width / 2 - p_pict->p[2].i_visible_pitch );
        }
    }

    if( p_pict->p[0].i_visible_lines < (int)p_sys->i_height )
    {
        for( i = p_pict->p[0].i_visible_lines; i < p_sys->i_height; i++ )
        {
            memset( p_pict->p[0].p_pixels + i * p_pict->p[0].i_pitch, 0,
                    p_sys->i_width );
        }
        for( i = p_pict->p[1].i_visible_lines; i < p_sys->i_height / 2; i++ )
        {
            memset( p_pict->p[1].p_pixels + i * p_pict->p[1].i_pitch, 0x80,
                    p_sys->i_width / 2 );
            memset( p_pict->p[2].p_pixels + i * p_pict->p[2].i_pitch, 0x80,
                    p_sys->i_width / 2 );
        }
    }
gbazin's avatar
 
gbazin committed
791 792 793 794

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

795 796
    yuv.y_width  = p_sys->i_width;
    yuv.y_height = p_sys->i_height;
gbazin's avatar
 
gbazin committed
797 798
    yuv.y_stride = p_pict->p[0].i_pitch;

799 800
    yuv.uv_width  = p_sys->i_width / 2;
    yuv.uv_height = p_sys->i_height / 2;
gbazin's avatar
 
gbazin committed
801 802 803 804 805 806
    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;

gbazin's avatar
 
gbazin committed
807 808 809 810 811
    if( theora_encode_YUVin( &p_sys->td, &yuv ) < 0 )
    {
        msg_Warn( p_enc, "failed encoding a frame" );
        return NULL;
    }
gbazin's avatar
 
gbazin committed
812 813 814 815 816

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

    /* Ogg packet to block */
    p_block = block_New( p_enc, oggpacket.bytes );
gbazin's avatar
 
gbazin committed
817
    memcpy( p_block->p_buffer, oggpacket.packet, oggpacket.bytes );
818
    p_block->i_dts = p_block->i_pts = p_pict->date;
gbazin's avatar
 
gbazin committed
819 820 821 822 823 824 825 826 827 828 829 830 831

    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 );
832
    theora_comment_clear( &p_sys->tc );
gbazin's avatar
 
gbazin committed
833 834 835

    free( p_sys );
}