libhb/encvorbis.c

   1 /* $Id: encvorbis.c,v 1.6 2005/03/05 15:08:32 titer Exp $
   2
   3    This file is part of the HandBrake source code.
   4    Homepage: <http://handbrake.m0k.org/>.
   5    It may be used under the terms of the GNU General Public License. */
   6
   7 #include "hb.h"
   8
   9 #include "vorbis/vorbisenc.h"
  10
  11 #define OGGVORBIS_FRAME_SIZE 1024
  12
  13 int  encvorbisInit( hb_work_object_t *, hb_job_t * );
  14 int  encvorbisWork( hb_work_object_t *, hb_buffer_t **, hb_buffer_t ** );
  15 void encvorbisClose( hb_work_object_t * );
  16
  17 hb_work_object_t hb_encvorbis =
  18 {
  19     WORK_ENCVORBIS,
  20     "Vorbis encoder (libvorbis)",
  21     encvorbisInit,
  22     encvorbisWork,
  23     encvorbisClose
  24 };
  25
  26 struct hb_work_private_s
  27 {
  28     hb_job_t   * job;
  29
  30     vorbis_info        vi;
  31     vorbis_comment     vc;
  32     vorbis_dsp_state   vd;
  33     vorbis_block       vb;
  34
  35     unsigned long   input_samples;
  36     uint8_t       * buf;
  37     uint64_t        pts;
  38
  39     hb_list_t     * list;
  40     int           channelsused;
  41     int           channel_map[6];
  42 };
  43
  44 int encvorbisInit( hb_work_object_t * w, hb_job_t * job )
  45 {
  46     int i;
  47     ogg_packet header[3];
  48
  49     hb_work_private_t * pv = calloc( 1, sizeof( hb_work_private_t ) );
  50     w->private_data = pv;
  51     pv->channelsused = w->config->vorbis.channelsused;
  52
  53     pv->job   = job;
  54
  55     hb_log( "encvorbis: opening libvorbis" );
  56
  57     /* init */
  58     vorbis_info_init( &pv->vi );
  59     if( vorbis_encode_setup_managed( &pv->vi, pv->channelsused,
  60           job->arate, -1, 1000 * job->abitrate, -1 ) ||
  61         vorbis_encode_ctl( &pv->vi, OV_ECTL_RATEMANAGE_AVG, NULL ) ||
  62           vorbis_encode_setup_init( &pv->vi ) )
  63     {
  64         hb_log( "encvorbis: vorbis_encode_setup_managed failed" );
  65     }
  66
  67     /* add a comment */
  68     vorbis_comment_init( &pv->vc );
  69     vorbis_comment_add_tag( &pv->vc, "Encoder", "HandBrake");
  70     vorbis_comment_add_tag( &pv->vc, "LANGUAGE", w->config->vorbis.language);
  71
  72     /* set up the analysis state and auxiliary encoding storage */
  73     vorbis_analysis_init( &pv->vd, &pv->vi);
  74     vorbis_block_init( &pv->vd, &pv->vb);
  75
  76     /* get the 3 headers */
  77     vorbis_analysis_headerout( &pv->vd, &pv->vc,
  78                                &header[0], &header[1], &header[2] );
  79     for( i = 0; i < 3; i++ )
  80     {
  81         memcpy( w->config->vorbis.headers[i], &header[i],
  82                 sizeof( ogg_packet ) );
  83         memcpy( w->config->vorbis.headers[i] + sizeof( ogg_packet ),
  84                 header[i].packet, header[i].bytes );
  85     }
  86
  87     pv->input_samples = pv->channelsused * OGGVORBIS_FRAME_SIZE;
  88     pv->buf = malloc( pv->input_samples * sizeof( float ) );
  89
  90     pv->list = hb_list_init();
  91
  92     switch (pv->channelsused) {
  93         case 1:
  94             pv->channel_map[0] = 0;
  95             break;
  96         case 6:
  97             pv->channel_map[0] = 0;
  98             pv->channel_map[1] = 2;
  99             pv->channel_map[2] = 1;
 100             pv->channel_map[3] = 4;
 101             pv->channel_map[4] = 5;
 102             pv->channel_map[5] = 3;
 103             break;
 104         default:
 105             hb_log("encvorbis.c: Unable to correctly proccess %d channels, assuming stereo.", pv->channelsused);
 106         case 2:
 107             // Assume stereo
 108             pv->channel_map[0] = 0;
 109             pv->channel_map[1] = 1;
 110             break;
 111     }
 112
 113     return 0;
 114 }
 115
 116 /***********************************************************************
 117  * Close
 118  ***********************************************************************
 119  *
 120  **********************************************************************/
 121 void encvorbisClose( hb_work_object_t * w )
 122 {
 123     hb_work_private_t * pv = w->private_data;
 124
 125     vorbis_block_clear( &pv->vb );
 126     vorbis_dsp_clear( &pv->vd );
 127     vorbis_comment_clear( &pv->vc );
 128     vorbis_info_clear( &pv->vi );
 129
 130     hb_list_empty( &pv->list );
 131
 132     free( pv->buf );
 133     free( pv );
 134     w->private_data = NULL;
 135 }
 136
 137 /***********************************************************************
 138  * Flush
 139  ***********************************************************************
 140  *
 141  **********************************************************************/
 142 static hb_buffer_t * Flush( hb_work_object_t * w )
 143 {
 144     hb_work_private_t * pv = w->private_data;
 145     hb_buffer_t * buf;
 146
 147     if( vorbis_analysis_blockout( &pv->vd, &pv->vb ) == 1 )
 148     {
 149         ogg_packet op;
 150
 151         vorbis_analysis( &pv->vb, NULL );
 152         vorbis_bitrate_addblock( &pv->vb );
 153
 154         if( vorbis_bitrate_flushpacket( &pv->vd, &op ) )
 155         {
 156             buf = hb_buffer_init( sizeof( ogg_packet ) + op.bytes );
 157             memcpy( buf->data, &op, sizeof( ogg_packet ) );
 158             memcpy( buf->data + sizeof( ogg_packet ), op.packet,
 159                     op.bytes );
 160             buf->key   = 1;
 161             buf->start = pv->pts; /* No exact, but who cares - the OGM
 162                                     muxer doesn't use it */
 163             buf->stop  = buf->start +
 164                 90000 * OGGVORBIS_FRAME_SIZE + pv->job->arate;
 165
 166             return buf;
 167         }
 168     }
 169
 170     return NULL;
 171 }
 172
 173 /***********************************************************************
 174  * Encode
 175  ***********************************************************************
 176  *
 177  **********************************************************************/
 178 static hb_buffer_t * Encode( hb_work_object_t * w )
 179 {
 180     hb_work_private_t * pv = w->private_data;
 181     hb_buffer_t * buf;
 182     float ** buffer;
 183     int i, j;
 184
 185     /* Try to extract more data */
 186     if( ( buf = Flush( w ) ) )
 187     {
 188         return buf;
 189     }
 190
 191     if( hb_list_bytes( pv->list ) < pv->input_samples * sizeof( float ) )
 192     {
 193         return NULL;
 194     }
 195
 196     /* Process more samples */
 197     hb_list_getbytes( pv->list, pv->buf, pv->input_samples * sizeof( float ),
 198                       &pv->pts, NULL );
 199     buffer = vorbis_analysis_buffer( &pv->vd, OGGVORBIS_FRAME_SIZE );
 200     for( i = 0; i < OGGVORBIS_FRAME_SIZE; i++ )
 201     {
 202         for( j = 0; j < pv->channelsused; j++)
 203         {
 204             buffer[j][i] = ((float *) pv->buf)[(pv->channelsused * i + pv->channel_map[j])] / 32768.f;
 205         }
 206     }
 207     vorbis_analysis_wrote( &pv->vd, OGGVORBIS_FRAME_SIZE );
 208
 209     /* Try to extract again */
 210     return Flush( w );
 211 }
 212
 213 /***********************************************************************
 214  * Work
 215  ***********************************************************************
 216  *
 217  **********************************************************************/
 218 int encvorbisWork( hb_work_object_t * w, hb_buffer_t ** buf_in,
 219                    hb_buffer_t ** buf_out )
 220 {
 221     hb_work_private_t * pv = w->private_data;
 222     hb_buffer_t * buf;
 223
 224     hb_list_add( pv->list, *buf_in );
 225     *buf_in = NULL;
 226
 227     *buf_out = buf = Encode( w );
 228
 229     while( buf )
 230     {
 231         buf->next = Encode( w );
 232         buf       = buf->next;
 233     }
 234
 235     return HB_WORK_OK;
 236 }