1 /* $Id: encvorbis.c,v 1.6 2005/03/05 15:08:32 titer Exp $
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. */
9 #include "vorbis/vorbisenc.h"
11 #define OGGVORBIS_FRAME_SIZE 1024
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
* );
17 hb_work_object_t hb_encvorbis
=
20 "Vorbis encoder (libvorbis)",
26 struct hb_work_private_s
35 unsigned long input_samples
;
40 int out_discrete_channels
;
42 int64_t prev_blocksize
;
45 int encvorbisInit( hb_work_object_t
* w
, hb_job_t
* job
)
49 struct ovectl_ratemanage2_arg ctl_rate_arg
;
51 hb_work_private_t
* pv
= calloc( 1, sizeof( hb_work_private_t
) );
53 pv
->out_discrete_channels
= HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT(w
->amixdown
);
57 hb_log( "encvorbis: opening libvorbis" );
59 /* 28kbps/channel seems to be the minimum for 6ch vorbis. */
60 int min_bitrate
= 28 * pv
->out_discrete_channels
;
61 if (pv
->out_discrete_channels
> 2 && job
->abitrate
< min_bitrate
)
63 hb_log( "encvorbis: Selected bitrate (%d kbps) too low for %d channel audio.", job
->abitrate
, pv
->out_discrete_channels
);
64 hb_log( "encvorbis: Resetting bitrate to %d kbps", min_bitrate
);
65 job
->abitrate
= min_bitrate
;
69 vorbis_info_init( &pv
->vi
);
70 if( vorbis_encode_setup_managed( &pv
->vi
, pv
->out_discrete_channels
,
71 job
->arate
, -1, 1000 * job
->abitrate
, -1 ) )
73 hb_error( "encvorbis: vorbis_encode_setup_managed failed.\n" );
78 if( vorbis_encode_ctl( &pv
->vi
, OV_ECTL_RATEMANAGE2_GET
, &ctl_rate_arg
) )
80 hb_log( "encvorbis: vorbis_encode_ctl( ratemanage2_get ) failed" );
83 ctl_rate_arg
.bitrate_average_kbps
= 1000 * job
->abitrate
;
84 ctl_rate_arg
.management_active
= 1;
86 if( vorbis_encode_ctl( &pv
->vi
, OV_ECTL_RATEMANAGE2_SET
, &ctl_rate_arg
) ||
87 vorbis_encode_setup_init( &pv
->vi
) )
89 hb_error( "encvorbis: vorbis_encode_ctl( ratemanage2_set ) OR vorbis_encode_setup_init failed.\n" );
95 vorbis_comment_init( &pv
->vc
);
96 vorbis_comment_add_tag( &pv
->vc
, "Encoder", "HandBrake");
97 vorbis_comment_add_tag( &pv
->vc
, "LANGUAGE", w
->config
->vorbis
.language
);
99 /* set up the analysis state and auxiliary encoding storage */
100 vorbis_analysis_init( &pv
->vd
, &pv
->vi
);
101 vorbis_block_init( &pv
->vd
, &pv
->vb
);
103 /* get the 3 headers */
104 vorbis_analysis_headerout( &pv
->vd
, &pv
->vc
,
105 &header
[0], &header
[1], &header
[2] );
106 for( i
= 0; i
< 3; i
++ )
108 memcpy( w
->config
->vorbis
.headers
[i
], &header
[i
],
109 sizeof( ogg_packet
) );
110 memcpy( w
->config
->vorbis
.headers
[i
] + sizeof( ogg_packet
),
111 header
[i
].packet
, header
[i
].bytes
);
114 pv
->input_samples
= pv
->out_discrete_channels
* OGGVORBIS_FRAME_SIZE
;
115 pv
->buf
= malloc( pv
->input_samples
* sizeof( float ) );
117 pv
->list
= hb_list_init();
119 switch (pv
->out_discrete_channels
) {
121 pv
->channel_map
[0] = 0;
124 pv
->channel_map
[0] = 0;
125 pv
->channel_map
[1] = 2;
126 pv
->channel_map
[2] = 1;
127 pv
->channel_map
[3] = 4;
128 pv
->channel_map
[4] = 5;
129 pv
->channel_map
[5] = 3;
132 hb_log("encvorbis.c: Unable to correctly proccess %d channels, assuming stereo.", pv
->out_discrete_channels
);
135 pv
->channel_map
[0] = 0;
136 pv
->channel_map
[1] = 1;
143 /***********************************************************************
145 ***********************************************************************
147 **********************************************************************/
148 void encvorbisClose( hb_work_object_t
* w
)
150 hb_work_private_t
* pv
= w
->private_data
;
152 vorbis_block_clear( &pv
->vb
);
153 vorbis_dsp_clear( &pv
->vd
);
154 vorbis_comment_clear( &pv
->vc
);
155 vorbis_info_clear( &pv
->vi
);
158 hb_list_empty( &pv
->list
);
162 w
->private_data
= NULL
;
165 /***********************************************************************
167 ***********************************************************************
169 **********************************************************************/
170 static hb_buffer_t
* Flush( hb_work_object_t
* w
)
172 hb_work_private_t
* pv
= w
->private_data
;
174 int64_t blocksize
= 0;
176 if( vorbis_analysis_blockout( &pv
->vd
, &pv
->vb
) == 1 )
180 vorbis_analysis( &pv
->vb
, NULL
);
181 vorbis_bitrate_addblock( &pv
->vb
);
183 if( vorbis_bitrate_flushpacket( &pv
->vd
, &op
) )
185 buf
= hb_buffer_init( sizeof( ogg_packet
) + op
.bytes
);
186 memcpy( buf
->data
, &op
, sizeof( ogg_packet
) );
187 memcpy( buf
->data
+ sizeof( ogg_packet
), op
.packet
,
189 blocksize
= vorbis_packet_blocksize(&pv
->vi
, &op
);
190 buf
->frametype
= HB_FRAME_AUDIO
;
191 buf
->start
= (int64_t)(vorbis_granule_time(&pv
->vd
, op
.granulepos
) * 90000);
192 buf
->stop
= (int64_t)(vorbis_granule_time(&pv
->vd
, (pv
->prev_blocksize
+ blocksize
)/4 + op
.granulepos
) * 90000);
193 /* The stop time isn't accurate for the first ~3 packets, as the actual blocksize depends on the previous _and_ current packets. */
194 pv
->prev_blocksize
= blocksize
;
202 /***********************************************************************
204 ***********************************************************************
206 **********************************************************************/
207 static hb_buffer_t
* Encode( hb_work_object_t
* w
)
209 hb_work_private_t
* pv
= w
->private_data
;
214 /* Try to extract more data */
215 if( ( buf
= Flush( w
) ) )
220 if( hb_list_bytes( pv
->list
) < pv
->input_samples
* sizeof( float ) )
225 /* Process more samples */
226 hb_list_getbytes( pv
->list
, pv
->buf
, pv
->input_samples
* sizeof( float ),
228 buffer
= vorbis_analysis_buffer( &pv
->vd
, OGGVORBIS_FRAME_SIZE
);
229 for( i
= 0; i
< OGGVORBIS_FRAME_SIZE
; i
++ )
231 for( j
= 0; j
< pv
->out_discrete_channels
; j
++)
233 buffer
[j
][i
] = ((float *) pv
->buf
)[(pv
->out_discrete_channels
* i
+ pv
->channel_map
[j
])] / 32768.f
;
236 vorbis_analysis_wrote( &pv
->vd
, OGGVORBIS_FRAME_SIZE
);
238 /* Try to extract again */
242 /***********************************************************************
244 ***********************************************************************
246 **********************************************************************/
247 int encvorbisWork( hb_work_object_t
* w
, hb_buffer_t
** buf_in
,
248 hb_buffer_t
** buf_out
)
250 hb_work_private_t
* pv
= w
->private_data
;
253 hb_list_add( pv
->list
, *buf_in
);
256 *buf_out
= buf
= Encode( w
);
260 buf
->next
= Encode( w
);