1 /*****************************************************************************
2 * es.c: Elementary stream output module
3 *****************************************************************************
4 * Copyright (C) 2003-2004 VLC authors and VideoLAN
5 * $Id: a03c40276d39776e92e5525be9f4ac9509aa35ce $
6 *
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 *
9 * This program is free software; you can redistribute it and/or modify it
10 * under the terms of the GNU Lesser General Public License as published by
11 * the Free Software Foundation; either version 2.1 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public License
20 * along with this program; if not, write to the Free Software Foundation,
21 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
23
24 /*****************************************************************************
25 * Preamble
26 *****************************************************************************/
27
28 #ifdef HAVE_CONFIG_H
29 # include "config.h"
30 #endif
31
32 #include <vlc_common.h>
33 #include <vlc_plugin.h>
34 #include <vlc_input.h>
35 #include <vlc_sout.h>
36 #include <vlc_dialog.h>
37 #include <vlc_memstream.h>
38
39 /*****************************************************************************
40 * Module descriptor
41 *****************************************************************************/
42 #define ACCESS_TEXT N_("Output access method")
43 #define ACCESS_LONGTEXT N_( \
44 "This is the default output access method that will be used." )
45
46 #define ACCESSA_TEXT N_("Audio output access method")
47 #define ACCESSA_LONGTEXT N_( \
48 "This is the output access method that will be used for audio." )
49 #define ACCESSV_TEXT N_("Video output access method")
50 #define ACCESSV_LONGTEXT N_( \
51 "This is the output access method that will be used for video." )
52
53 #define MUX_TEXT N_("Output muxer")
54 #define MUX_LONGTEXT N_( \
55 "This is the default muxer method that will be used." )
56 #define MUXA_TEXT N_("Audio output muxer")
57 #define MUXA_LONGTEXT N_( \
58 "This is the muxer that will be used for audio." )
59 #define MUXV_TEXT N_("Video output muxer")
60 #define MUXV_LONGTEXT N_( \
61 "This is the muxer that will be used for video." )
62
63 #define DEST_TEXT N_("Output URL")
64 #define DEST_LONGTEXT N_( \
65 "This is the default output URI." )
66 #define DESTA_TEXT N_("Audio output URL")
67 #define DESTA_LONGTEXT N_( \
68 "This is the output URI that will be used for audio." )
69 #define DESTV_TEXT N_("Video output URL")
70 #define DESTV_LONGTEXT N_( \
71 "This is the output URI that will be used for video." )
72
73 static int Open ( vlc_object_t * );
74 static void Close ( vlc_object_t * );
75
76 #define SOUT_CFG_PREFIX "sout-es-"
77
78 vlc_module_begin ()
79 set_shortname( "ES" )
80 set_description( N_("Elementary stream output") )
81 set_capability( "sout stream", 50 )
82 add_shortcut( "es" )
83 set_category( CAT_SOUT )
84 set_subcategory( SUBCAT_SOUT_STREAM )
85
86 set_section( N_("Generic"), NULL )
87 add_string( SOUT_CFG_PREFIX "access", "", ACCESS_TEXT,
88 ACCESS_LONGTEXT, true )
89 add_string( SOUT_CFG_PREFIX "mux", "", MUX_TEXT,
90 MUX_LONGTEXT, true )
91 add_string( SOUT_CFG_PREFIX "dst", "", DEST_TEXT,
92 DEST_LONGTEXT, true )
93
94 set_section( N_("Audio"), NULL )
95 add_string( SOUT_CFG_PREFIX "access-audio", "", ACCESSA_TEXT,
96 ACCESSA_LONGTEXT, true )
97 add_string( SOUT_CFG_PREFIX "mux-audio", "", MUXA_TEXT,
98 MUXA_LONGTEXT, true )
99 add_string( SOUT_CFG_PREFIX "dst-audio", "", DESTA_TEXT,
100 DESTA_LONGTEXT, true )
101
102 set_section( N_("Video"), NULL )
103 add_string( SOUT_CFG_PREFIX "access-video", "", ACCESSV_TEXT,
104 ACCESSV_LONGTEXT, true )
105 add_string( SOUT_CFG_PREFIX "mux-video", "", MUXV_TEXT,
106 MUXV_LONGTEXT, true )
107 add_string( SOUT_CFG_PREFIX "dst-video", "", DESTV_TEXT,
108 DESTV_LONGTEXT, true )
109
110 set_callbacks( Open, Close )
111 vlc_module_end ()
112
113 /*****************************************************************************
114 * Exported prototypes
115 *****************************************************************************/
116 static const char *const ppsz_sout_options[] = {
117 "access", "access-audio", "access-video",
118 "mux", "mux-audio", "mux-video",
119 "dst", "dst-audio", "dst-video",
120 NULL
121 };
122
123 static sout_stream_id_sys_t *Add( sout_stream_t *, const es_format_t * );
124 static void Del ( sout_stream_t *, sout_stream_id_sys_t * );
125 static int Send( sout_stream_t *, sout_stream_id_sys_t *, block_t* );
126
127 struct sout_stream_sys_t
128 {
129 int i_count_audio;
130 int i_count_video;
131 int i_count;
132
133 char *psz_mux;
134 char *psz_mux_audio;
135 char *psz_mux_video;
136
137 char *psz_access;
138 char *psz_access_audio;
139 char *psz_access_video;
140
141 char *psz_dst;
142 char *psz_dst_audio;
143 char *psz_dst_video;
144 };
145
146 /*****************************************************************************
147 * Open:
148 *****************************************************************************/
Open(vlc_object_t * p_this)149 static int Open( vlc_object_t *p_this )
150 {
151 sout_stream_t *p_stream = (sout_stream_t*)p_this;
152 sout_stream_sys_t *p_sys;
153
154 config_ChainParse( p_stream, SOUT_CFG_PREFIX, ppsz_sout_options, p_stream->p_cfg );
155 p_sys = malloc( sizeof( sout_stream_sys_t ) );
156
157 p_sys->i_count = 0;
158 p_sys->i_count_audio = 0;
159 p_sys->i_count_video = 0;
160
161 p_sys->psz_access = var_GetString( p_stream, SOUT_CFG_PREFIX "access" );
162 p_sys->psz_access_audio = var_GetString( p_stream, SOUT_CFG_PREFIX "access-audio" );
163 p_sys->psz_access_video = var_GetString( p_stream, SOUT_CFG_PREFIX "access-video" );
164
165 p_sys->psz_mux = var_GetString( p_stream, SOUT_CFG_PREFIX "mux" );
166 p_sys->psz_mux_audio = var_GetString( p_stream, SOUT_CFG_PREFIX "mux-audio" );
167 p_sys->psz_mux_video = var_GetString( p_stream, SOUT_CFG_PREFIX "mux-video" );
168
169 p_sys->psz_dst = var_GetString( p_stream, SOUT_CFG_PREFIX "dst" );
170 p_sys->psz_dst_audio = var_GetString( p_stream, SOUT_CFG_PREFIX "dst-audio" );
171 p_sys->psz_dst_video = var_GetString( p_stream, SOUT_CFG_PREFIX "dst-video" );
172
173 p_stream->pf_add = Add;
174 p_stream->pf_del = Del;
175 p_stream->pf_send = Send;
176
177 p_stream->p_sys = p_sys;
178
179 return VLC_SUCCESS;
180 }
181
182 /*****************************************************************************
183 * Close:
184 *****************************************************************************/
185
Close(vlc_object_t * p_this)186 static void Close( vlc_object_t * p_this )
187 {
188 sout_stream_t *p_stream = (sout_stream_t*)p_this;
189 sout_stream_sys_t *p_sys = p_stream->p_sys;
190
191 free( p_sys->psz_access );
192 free( p_sys->psz_access_audio );
193 free( p_sys->psz_access_video );
194
195 free( p_sys->psz_mux );
196 free( p_sys->psz_mux_audio );
197 free( p_sys->psz_mux_video );
198
199 free( p_sys->psz_dst );
200 free( p_sys->psz_dst_audio );
201 free( p_sys->psz_dst_video );
202
203 free( p_sys );
204 }
205
206 struct sout_stream_id_sys_t
207 {
208 sout_input_t *p_input;
209 sout_mux_t *p_mux;
210 };
211
es_print_url(const char * psz_fmt,vlc_fourcc_t i_fourcc,int i_count,const char * psz_access,const char * psz_mux)212 static char * es_print_url( const char *psz_fmt, vlc_fourcc_t i_fourcc, int i_count,
213 const char *psz_access, const char *psz_mux )
214 {
215 struct vlc_memstream stream;
216 unsigned char c;
217
218 if (vlc_memstream_open(&stream))
219 return NULL;
220
221 if( psz_fmt == NULL || !*psz_fmt )
222 psz_fmt = "stream-%n-%c.%m";
223
224 while ((c = *(psz_fmt++)) != '\0')
225 {
226 if (c != '%')
227 {
228 vlc_memstream_putc(&stream, c);
229 continue;
230 }
231
232 switch (c = *(psz_fmt++))
233 {
234 case 'n':
235 vlc_memstream_printf(&stream, "%d", i_count);
236 break;
237 case 'c':
238 vlc_memstream_printf(&stream, "%4.4s", (char *)&i_fourcc);
239 break;
240 case 'm':
241 vlc_memstream_puts(&stream, psz_mux);
242 break;
243 case 'a':
244 vlc_memstream_puts(&stream, psz_access);
245 break;
246 case '\0':
247 vlc_memstream_putc(&stream, '%');
248 goto out;
249 default:
250 vlc_memstream_printf(&stream, "%%%c", (int) c);
251 break;
252 }
253 }
254 out:
255 if (vlc_memstream_close(&stream))
256 return NULL;
257 return stream.ptr;
258 }
259
Add(sout_stream_t * p_stream,const es_format_t * p_fmt)260 static sout_stream_id_sys_t *Add( sout_stream_t *p_stream, const es_format_t *p_fmt )
261 {
262 sout_stream_sys_t *p_sys = p_stream->p_sys;
263 sout_stream_id_sys_t *id;
264
265 const char *psz_access;
266 const char *psz_mux;
267 char *psz_dst;
268
269 sout_access_out_t *p_access;
270 sout_mux_t *p_mux;
271
272 /* *** get access name *** */
273 if( p_fmt->i_cat == AUDIO_ES && p_sys->psz_access_audio && *p_sys->psz_access_audio )
274 {
275 psz_access = p_sys->psz_access_audio;
276 }
277 else if( p_fmt->i_cat == VIDEO_ES && p_sys->psz_access_video && *p_sys->psz_access_video )
278 {
279 psz_access = p_sys->psz_access_video;
280 }
281 else
282 {
283 psz_access = p_sys->psz_access;
284 }
285
286 /* *** get mux name *** */
287 if( p_fmt->i_cat == AUDIO_ES && p_sys->psz_mux_audio && *p_sys->psz_mux_audio )
288 {
289 psz_mux = p_sys->psz_mux_audio;
290 }
291 else if( p_fmt->i_cat == VIDEO_ES && p_sys->psz_mux_video && *p_sys->psz_mux_video )
292 {
293 psz_mux = p_sys->psz_mux_video;
294 }
295 else
296 {
297 psz_mux = p_sys->psz_mux;
298 }
299
300 /* Get url (%d expanded as a codec count, %c expanded as codec fcc ) */
301 if( p_fmt->i_cat == AUDIO_ES && p_sys->psz_dst_audio && *p_sys->psz_dst_audio )
302 {
303 psz_dst = es_print_url( p_sys->psz_dst_audio, p_fmt->i_codec,
304 p_sys->i_count_audio, psz_access, psz_mux );
305 }
306 else if( p_fmt->i_cat == VIDEO_ES && p_sys->psz_dst_video && *p_sys->psz_dst_video )
307 {
308 psz_dst = es_print_url( p_sys->psz_dst_video, p_fmt->i_codec,
309 p_sys->i_count_video, psz_access, psz_mux );
310 }
311 else
312 {
313 int i_count;
314 if( p_fmt->i_cat == VIDEO_ES )
315 {
316 i_count = p_sys->i_count_video;
317 }
318 else if( p_fmt->i_cat == AUDIO_ES )
319 {
320 i_count = p_sys->i_count_audio;
321 }
322 else
323 {
324 i_count = p_sys->i_count;
325 }
326
327 psz_dst = es_print_url( p_sys->psz_dst, p_fmt->i_codec,
328 i_count, psz_access, psz_mux );
329 }
330
331 p_sys->i_count++;
332 if( p_fmt->i_cat == VIDEO_ES )
333 {
334 p_sys->i_count_video++;
335 }
336 else if( p_fmt->i_cat == AUDIO_ES )
337 {
338 p_sys->i_count_audio++;
339 }
340 msg_Dbg( p_stream, "creating `%s/%s://%s'",
341 psz_access, psz_mux, psz_dst );
342
343 /* *** find and open appropriate access module *** */
344 p_access = sout_AccessOutNew( p_stream, psz_access, psz_dst );
345 if( p_access == NULL )
346 {
347 msg_Err( p_stream, "no suitable sout access module for `%s/%s://%s'",
348 psz_access, psz_mux, psz_dst );
349 vlc_dialog_display_error( p_stream, _("Streaming / Transcoding failed"),
350 _("There is no suitable stream-output access module for \"%s/%s://%s\"."),
351 psz_access, psz_mux, psz_dst );
352 free( psz_dst );
353 return( NULL );
354 }
355
356 /* *** find and open appropriate mux module *** */
357 p_mux = sout_MuxNew( p_stream->p_sout, psz_mux, p_access );
358 if( p_mux == NULL )
359 {
360 msg_Err( p_stream, "no suitable sout mux module for `%s/%s://%s'",
361 psz_access, psz_mux, psz_dst );
362 vlc_dialog_display_error( p_stream, _("Streaming / Transcoding failed"),
363 _("There is no suitable stream-output access module "\
364 "for \"%s/%s://%s\"."), psz_access, psz_mux, psz_dst );
365 sout_AccessOutDelete( p_access );
366 free( psz_dst );
367 return( NULL );
368 }
369 free( psz_dst );
370
371 id = malloc( sizeof( sout_stream_id_sys_t ) );
372 if( !id )
373 {
374 sout_MuxDelete( p_mux );
375 sout_AccessOutDelete( p_access );
376 return NULL;
377 }
378 id->p_mux = p_mux;
379 id->p_input = sout_MuxAddStream( p_mux, p_fmt );
380
381 if( id->p_input == NULL )
382 {
383 sout_MuxDelete( p_mux );
384 sout_AccessOutDelete( p_access );
385 free( id );
386 return NULL;
387 }
388
389 if( !sout_AccessOutCanControlPace( p_access ) )
390 p_stream->p_sout->i_out_pace_nocontrol++;
391
392 return id;
393 }
394
Del(sout_stream_t * p_stream,sout_stream_id_sys_t * id)395 static void Del( sout_stream_t *p_stream, sout_stream_id_sys_t *id )
396 {
397 VLC_UNUSED(p_stream);
398 sout_access_out_t *p_access = id->p_mux->p_access;
399
400 sout_MuxDeleteStream( id->p_mux, id->p_input );
401 sout_MuxDelete( id->p_mux );
402 if( !sout_AccessOutCanControlPace( p_access ) )
403 p_stream->p_sout->i_out_pace_nocontrol--;
404 sout_AccessOutDelete( p_access );
405
406 free( id );
407 }
408
Send(sout_stream_t * p_stream,sout_stream_id_sys_t * id,block_t * p_buffer)409 static int Send( sout_stream_t *p_stream, sout_stream_id_sys_t *id,
410 block_t *p_buffer )
411 {
412 VLC_UNUSED(p_stream);
413 return sout_MuxSendBuffer( id->p_mux, id->p_input, p_buffer );
414 }
415
416