1 /*****************************************************************************
2 * transrate.c: MPEG2 video transrating module
3 *****************************************************************************
4 * Copyright (C) 2003 VideoLAN
7 * Authors: Christophe Massiot <massiot@via.ecp.fr>
8 * Laurent Aimar <fenrir@via.ecp.fr>
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
23 *****************************************************************************/
25 /*****************************************************************************
27 *****************************************************************************/
36 #include <vlc/input.h>
38 #include "transrate.h"
40 /*****************************************************************************
42 *****************************************************************************/
43 static int Open ( vlc_object_t * );
44 static void Close ( vlc_object_t * );
46 static sout_stream_id_t *Add ( sout_stream_t *, es_format_t * );
47 static int Del ( sout_stream_t *, sout_stream_id_t * );
48 static int Send( sout_stream_t *, sout_stream_id_t *, block_t * );
50 static int transrate_video_process( sout_stream_t *, sout_stream_id_t *, block_t *, block_t ** );
52 /*****************************************************************************
54 *****************************************************************************/
56 set_description( _("MPEG2 video transrating stream output") );
57 set_capability( "sout stream", 50 );
58 add_shortcut( "transrate" );
59 set_callbacks( Open, Close );
62 struct sout_stream_sys_t
67 mtime_t i_shaping_delay;
73 /*****************************************************************************
75 *****************************************************************************/
76 static int Open( vlc_object_t *p_this )
78 sout_stream_t *p_stream = (sout_stream_t*)p_this;
79 sout_stream_sys_t *p_sys;
82 p_sys = malloc( sizeof( sout_stream_sys_t ) );
83 p_sys->p_out = sout_StreamNew( p_stream->p_sout, p_stream->psz_next );
85 p_sys->i_vbitrate = 0;
87 if( ( val = sout_cfg_find_value( p_stream->p_cfg, "vb" ) ) )
89 p_sys->i_vbitrate = atoi( val );
90 if( p_sys->i_vbitrate < 16000 )
92 p_sys->i_vbitrate *= 1000;
97 p_sys->i_vbitrate = 3000000;
100 p_sys->i_shaping_delay = 500000;
101 if( ( val = sout_cfg_find_value( p_stream->p_cfg, "shaping" ) ) )
103 p_sys->i_shaping_delay = (int64_t)atoi( val ) * 1000;
104 if( p_sys->i_shaping_delay <= 0 )
107 "invalid shaping ("I64Fd"ms) reseting to 500ms",
108 p_sys->i_shaping_delay / 1000 );
109 p_sys->i_shaping_delay = 500000;
113 p_sys->b_mpeg4_matrix = 0;
114 if( sout_cfg_find( p_stream->p_cfg, "mpeg4-matrix" ) )
116 p_sys->b_mpeg4_matrix = 1;
119 msg_Dbg( p_stream, "codec video %dkb/s max gop="I64Fd"us",
120 p_sys->i_vbitrate / 1024, p_sys->i_shaping_delay );
124 msg_Err( p_stream, "cannot create chain" );
128 p_stream->pf_add = Add;
129 p_stream->pf_del = Del;
130 p_stream->pf_send = Send;
132 p_stream->p_sys = p_sys;
137 /*****************************************************************************
139 *****************************************************************************/
140 static void Close( vlc_object_t * p_this )
142 sout_stream_t *p_stream = (sout_stream_t *)p_this;
143 sout_stream_sys_t *p_sys = p_stream->p_sys;
145 sout_StreamDelete( p_sys->p_out );
150 static sout_stream_id_t * Add( sout_stream_t *p_stream, es_format_t *p_fmt )
152 sout_stream_sys_t *p_sys = p_stream->p_sys;
153 sout_stream_id_t *id;
155 id = malloc( sizeof( sout_stream_id_t ) );
158 if( p_fmt->i_cat == VIDEO_ES
159 && p_fmt->i_codec == VLC_FOURCC('m', 'p', 'g', 'v') )
162 "creating video transrating for fcc=`%4.4s'",
163 (char*)&p_fmt->i_codec );
165 id->p_current_buffer = NULL;
166 id->p_next_gop = NULL;
167 id->i_next_gop_duration = 0;
168 id->i_next_gop_size = 0;
169 memset( &id->tr, 0, sizeof( transrate_t ) );
170 id->tr.bs.i_byte_in = id->tr.bs.i_byte_out = 0;
171 id->tr.mpeg4_matrix = p_sys->b_mpeg4_matrix;
173 /* open output stream */
174 id->id = p_sys->p_out->pf_add( p_sys->p_out, p_fmt );
175 id->b_transrate = VLC_TRUE;
179 msg_Dbg( p_stream, "not transrating a stream (fcc=`%4.4s')", (char*)&p_fmt->i_codec );
180 id->id = p_sys->p_out->pf_add( p_sys->p_out, p_fmt );
181 id->b_transrate = VLC_FALSE;
193 static int Del ( sout_stream_t *p_stream, sout_stream_id_t *id )
195 sout_stream_sys_t *p_sys = p_stream->p_sys;
199 p_sys->p_out->pf_del( p_sys->p_out, id->id );
206 static int Send( sout_stream_t *p_stream, sout_stream_id_t *id,
209 sout_stream_sys_t *p_sys = p_stream->p_sys;
211 if( id->b_transrate )
213 block_t *p_buffer_out;
214 /* be sure to have at least 8 bytes of padding (maybe only 4) */
215 p_buffer = block_Realloc( p_buffer, 0, p_buffer->i_buffer + 8 );
216 p_buffer->i_buffer -= 8;
217 memset( &p_buffer->p_buffer[p_buffer->i_buffer], 0, 8 );
219 transrate_video_process( p_stream, id, p_buffer, &p_buffer_out );
223 return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_buffer_out );
227 else if( id->id != NULL )
229 return p_sys->p_out->pf_send( p_sys->p_out, id->id, p_buffer );
233 block_Release( p_buffer );
238 static int transrate_video_process( sout_stream_t *p_stream,
239 sout_stream_id_t *id, block_t *in, block_t **out )
241 transrate_t *tr = &id->tr;
242 bs_transrate_t *bs = &tr->bs;
248 block_t * p_next = in->p_next;
249 int i_flags = in->i_flags;
252 block_ChainAppend( &id->p_next_gop, in );
253 id->i_next_gop_duration += in->i_length;
254 id->i_next_gop_size += in->i_buffer;
257 if( ((i_flags & BLOCK_FLAG_TYPE_I )
258 && id->i_next_gop_duration >= 300000)
259 || (id->i_next_gop_duration > p_stream->p_sys->i_shaping_delay) )
261 mtime_t i_bitrate = (mtime_t)id->i_next_gop_size * 8000
262 / (id->i_next_gop_duration / 1000);
263 mtime_t i_new_bitrate;
265 id->tr.i_total_input = id->i_next_gop_size;
266 id->tr.i_remaining_input = id->i_next_gop_size;
267 id->tr.i_wanted_output = (p_stream->p_sys->i_vbitrate)
268 * (id->i_next_gop_duration / 1000) / 8000;
269 id->tr.i_current_output = 0;
271 id->p_current_buffer = id->p_next_gop;
273 while ( id->p_current_buffer != NULL )
275 block_t * p_next = id->p_current_buffer->p_next;
276 if ( !p_stream->p_sys->b_mpeg4_matrix
277 && id->tr.i_wanted_output >= id->tr.i_total_input )
279 bs->i_byte_out += id->p_current_buffer->i_buffer;
280 id->p_current_buffer->p_next = NULL;
281 block_ChainAppend( out, id->p_current_buffer );
285 if ( process_frame( p_stream, id, id->p_current_buffer,
288 id->p_current_buffer->p_next = NULL;
289 block_ChainAppend( out, id->p_current_buffer );
290 if ( p_stream->p_sys->b_mpeg4_matrix )
291 id->tr.i_wanted_output = id->tr.i_total_input;
295 block_Release( id->p_current_buffer );
298 id->p_current_buffer = p_next;
301 if ( id->tr.i_wanted_output < id->tr.i_total_input )
303 i_new_bitrate = (mtime_t)tr->i_current_output * 8000
304 / (id->i_next_gop_duration / 1000);
305 if (i_new_bitrate > p_stream->p_sys->i_vbitrate + 300000)
306 msg_Err(p_stream, "%lld -> %lld d=%lld",
307 i_bitrate, i_new_bitrate,
308 id->i_next_gop_duration);
310 msg_Dbg(p_stream, "%lld -> %lld d=%lld",
311 i_bitrate, i_new_bitrate,
312 id->i_next_gop_duration);
315 id->p_next_gop = NULL;
316 id->i_next_gop_duration = 0;
317 id->i_next_gop_size = 0;