]> git.sesse.net Git - vlc/blob - modules/gui/qt4/dialogs/sout.cpp
Qt4 - Mkv muxer in GUI... Same as previous from thresh...
[vlc] / modules / gui / qt4 / dialogs / sout.cpp
1 /*****************************************************************************
2  * sout.cpp : Stream output dialog ( old-style )
3  ****************************************************************************
4  * Copyright ( C ) 2006 the VideoLAN team
5  * Copyright (C) 2007 Société des arts technologiques
6  * Copyright (C) 2007 Savoir-faire Linux
7  *
8  * $Id$
9  *
10  * Authors: Clément Stenac <zorglub@videolan.org>
11  *          Jean-Baptiste Kempf <jb@videolan.org>
12  *          Jean-François Massol <jf.massol -at- gmail.com>
13  *          Pierre-Luc Beaudoin <pierre-luc.beaudoin@savoirfairelinux.com>
14  *
15  * This program is free software; you can redistribute it and/or modify
16  * it under the terms of the GNU General Public License as published by
17  * the Free Software Foundation; either version 2 of the License, or
18  * ( at your option ) any later version.
19  *
20  * This program is distributed in the hope that it will be useful,
21  * but WITHOUT ANY WARRANTY; without even the implied warranty of
22  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
23  * GNU General Public License for more details.
24  *
25  * You should have received a copy of the GNU General Public License
26  * along with this program; if not, write to the Free Software
27  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
28  *****************************************************************************/
29
30 #include "dialogs/sout.hpp"
31 #include <vlc_streaming.h>
32
33 #include <iostream>
34 #include <QString>
35 #include <QFileDialog>
36
37 SoutDialog::SoutDialog( QWidget *parent, intf_thread_t *_p_intf,
38                      bool _transcode_only ) : QVLCDialog( parent,  _p_intf )
39 {
40     setWindowTitle( qtr( "Stream output" ) );
41
42     b_transcode_only = _transcode_only;
43
44     /* UI stuff */
45     ui.setupUi( this );
46
47     ui.UDPEdit->hide(); ui.UDPLabel->hide();
48     ui.UDPPort->hide(); ui.UDPPortLabel->hide();
49 /* ADD HERE for new profiles */
50 #define ADD_PROFILE( name, shortname ) ui.profileBox->addItem( qtr( name ), QVariant( QString( shortname ) ) );
51     ADD_PROFILE( "Custom" , "Custom" )
52     ADD_PROFILE( "IPod (mp4/aac)", "IPod" )
53     ADD_PROFILE( "XBox", "XBox" )
54     ADD_PROFILE( "Windows (wmv/asf)", "Windows" )
55     ADD_PROFILE( "PSP", "PSP")
56     ADD_PROFILE( "GSM", "GSM" )
57
58 #define ADD_VCODEC( name, fourcc ) ui.vCodecBox->addItem( name, QVariant( fourcc ) );
59     ADD_VCODEC( "MPEG-1", "mp1v" )
60     ADD_VCODEC( "MPEG-2", "mp2v" )
61     ADD_VCODEC( "MPEG-4", "mp4v" )
62     ADD_VCODEC( "DIVX 1" , "DIV1" )
63     ADD_VCODEC( "DIVX 2" , "DIV1" )
64     ADD_VCODEC( "DIVX 3" , "DIV1" )
65     ADD_VCODEC( "H-263", "H263" )
66     ADD_VCODEC( "H-264", "h264" )
67     ADD_VCODEC( "WMV1", "WMV1" )
68     ADD_VCODEC( "WMV2" , "WMV2" )
69     ADD_VCODEC( "M-JPEG", "MJPG" )
70     ADD_VCODEC( "Theora", "theo" )
71
72 #define ADD_ACODEC( name, fourcc ) ui.aCodecBox->addItem( name, QVariant( fourcc ) );
73     ADD_ACODEC( "MPEG Audio", "mpga" )
74     ADD_ACODEC( "MP3", "mp3" )
75     ADD_ACODEC( "MPEG 4 Audio ( AAC )", "mp4a" )
76     ADD_ACODEC( "A52/AC-3", "a52" )
77     ADD_ACODEC( "Vorbis", "vorb" )
78     ADD_ACODEC( "Flac", "flac" )
79     ADD_ACODEC( "Speex", "spx" )
80     ADD_ACODEC( "WAV", "s16l" )
81     ADD_ACODEC( "WMA", "wma" )
82
83 #define ADD_SCALING( factor ) ui.vScaleBox->addItem( factor );
84     ADD_SCALING( "0.25" )
85     ADD_SCALING( "0.5" )
86     ADD_SCALING( "0.75" )
87     ADD_SCALING( "1" )
88     ADD_SCALING( "1.25" )
89     ADD_SCALING( "1.5" )
90     ADD_SCALING( "1.75" )
91     ADD_SCALING( "2" )
92
93     ui.mrlEdit->setToolTip ( qtr( "Stream output string.\n This is automatically generated "
94                                                 "when you change the above settings,\n but you can update it manually." ) ) ;
95
96 //     /* Connect everything to the updateMRL function */
97  #define CB( x ) CONNECT( ui.x, toggled( bool ), this, updateMRL() );
98  #define CT( x ) CONNECT( ui.x, textChanged( const QString ), this, updateMRL() );
99  #define CS( x ) CONNECT( ui.x, valueChanged( int ), this, updateMRL() );
100  #define CC( x ) CONNECT( ui.x, currentIndexChanged( int ), this, updateMRL() );
101 //     /* Output */
102     CB( fileOutput ); CB( HTTPOutput ); CB( localOutput );
103     CB( RTPOutput ); CB( MMSHOutput ); CB( rawInput ); CB( UDPOutput );
104     CT( fileEdit ); CT( HTTPEdit ); CT( RTPEdit ); CT( MMSHEdit ); CT( UDPEdit );
105     CT( IcecastEdit ); CT( IcecastMountpointEdit ); CT( IcecastNamePassEdit );
106     CS( HTTPPort ); CS( RTPPort ); CS( MMSHPort ); CS( UDPPort );
107 //     /* Transcode */
108     CC( vCodecBox ); CC( subsCodecBox ); CC( aCodecBox ) ;
109     CB( transcodeVideo ); CB( transcodeAudio ); CB( transcodeSubs );
110 //     CB( sOverlay );
111     CS( vBitrateSpin ); CS( aBitrateSpin ); CS( aChannelsSpin ); CC( vScaleBox );
112 //     /* Mux */
113     CB( PSMux ); CB( TSMux ); CB( MPEG1Mux ); CB( OggMux ); CB( ASFMux );
114     CB( MP4Mux ); CB( MOVMux ); CB( WAVMux ); CB( RAWMux ); CB( FLVMux );
115 //     /* Misc */
116     CB( soutAll ); CS( ttl ); CT( sapName ); CT( sapGroup );
117 //
118     CONNECT( ui.profileBox, activated( const QString & ), this, setOptions() );
119     CONNECT( ui.fileSelectButton, clicked() , this, fileBrowse()  );
120     CONNECT( ui.transcodeVideo, toggled( bool ), this, setVTranscodeOptions( bool ) );
121     CONNECT( ui.transcodeAudio, toggled( bool ), this, setATranscodeOptions( bool ) );
122     CONNECT( ui.transcodeSubs, toggled( bool ), this, setSTranscodeOptions( bool ) );
123     CONNECT( ui.rawInput, toggled( bool ), this, setRawOptions( bool ) );
124
125     okButton = new QPushButton( qtr( "&Stream" ) );
126     QPushButton *cancelButton = new QPushButton( qtr( "&Cancel" ) );
127
128     okButton->setDefault( true );
129     ui.acceptButtonBox->addButton( okButton, QDialogButtonBox::AcceptRole );
130     ui.acceptButtonBox->addButton( cancelButton, QDialogButtonBox::RejectRole );
131
132     BUTTONACT( okButton, ok() );
133     BUTTONACT( cancelButton, cancel() );
134
135     if( b_transcode_only ) toggleSout();
136 }
137
138 QString SoutDialog::getMrl()
139 {
140     return mrl;
141 }
142
143 void SoutDialog::fileBrowse()
144 {
145     ui.tabWidget->setTabEnabled( 0,false );
146     QString fileName = QFileDialog::getSaveFileName( this, qtr( "Save file" ), "",
147         qtr( "Containers (*.ps *.ts *.mpg *.ogg *.asf *.mp4 *.mov *.wav *.raw *.flv)" ) );
148     ui.fileEdit->setText( fileName );
149     updateMRL();
150 }
151
152 void SoutDialog::setVTranscodeOptions( bool b_trans )
153 {
154     ui.vCodecLabel->setEnabled( b_trans );
155     ui.vCodecBox->setEnabled( b_trans );
156     ui.vBitrateLabel->setEnabled( b_trans );
157     ui.vBitrateSpin->setEnabled( b_trans );
158     ui.vScaleLabel->setEnabled( b_trans );
159     ui.vScaleBox->setEnabled( b_trans );
160 }
161
162 void SoutDialog::setATranscodeOptions( bool b_trans )
163 {
164     ui.aCodecLabel->setEnabled( b_trans );
165     ui.aCodecBox->setEnabled( b_trans );
166     ui.aBitrateLabel->setEnabled( b_trans );
167     ui.aBitrateSpin->setEnabled( b_trans );
168     ui.aChannelsLabel->setEnabled( b_trans );
169     ui.aChannelsSpin->setEnabled( b_trans );
170 }
171
172 void SoutDialog::setSTranscodeOptions( bool b_trans )
173 {
174     ui.subsCodecBox->setEnabled( b_trans );
175     ui.subsOverlay->setEnabled( b_trans );
176 }
177
178 void SoutDialog::setRawOptions( bool b_raw )
179 {
180     if ( b_raw )
181     {
182         ui.tabWidget->setDisabled( true );
183     }
184     else
185     {
186         SoutDialog::setOptions();
187     }
188 }
189
190 void SoutDialog::setOptions()
191 {
192     QString profileString = ui.profileBox->itemData( ui.profileBox->currentIndex() ).toString();
193     msg_Dbg( p_intf, "Profile Used: %s",  qta( profileString ));
194     int index;
195
196 #define setProfile( muxName, hasVideo, vCodecName, hasAudio, aCodecName ) \
197     { \
198         ui.muxName ##Mux->setChecked( true ); \
199         \
200         ui.transcodeAudio->setChecked( hasAudio ); \
201         index = ui.aCodecBox->findText( vCodecName );  \
202         if( index >= 0 ) ui.aCodecBox->setCurrentIndex( index ); \
203         \
204         ui.transcodeVideo->setChecked( hasVideo ); \
205         index = ui.aCodecBox->findText( vCodecName );  \
206         if( index >=0 ) ui.vCodecBox->setCurrentIndex( index ); \
207     }
208
209     /* ADD HERE the profiles you want and need */
210     if( profileString == "IPod" ) setProfile( MP4, true, "mp4a", true, "mp4v" )
211     else if( profileString == "XBox" ) setProfile( ASF, true, "wma", true, "WMV2" )
212
213         /* If the profile is not a custom one, then disable the tabWidget */
214         if ( profileString == "Custom" )
215         {
216             ui.tabWidget->setEnabled( true );
217         }
218         else
219         {
220             ui.tabWidget->setDisabled( true );
221         }
222
223     /* Update the MRL !! */
224     updateMRL();
225 }
226
227 void SoutDialog::toggleSout()
228 {
229     //Toggle all the streaming options.
230     TOGGLEV( ui.HTTPOutput ) ; TOGGLEV( ui.RTPOutput ) ; TOGGLEV( ui.MMSHOutput ) ; TOGGLEV( ui.UDPOutput ) ;
231     TOGGLEV( ui.HTTPEdit ) ; TOGGLEV( ui.RTPEdit ) ; TOGGLEV( ui.MMSHEdit ) ; TOGGLEV( ui.UDPEdit ) ;
232     TOGGLEV( ui.HTTPLabel ) ; TOGGLEV( ui.RTPLabel ) ; TOGGLEV( ui.MMSHLabel ) ; TOGGLEV( ui.UDPLabel ) ;
233     TOGGLEV( ui.HTTPPortLabel ) ; TOGGLEV( ui.RTPPortLabel ) ; TOGGLEV( ui.MMSHPortLabel ) ; TOGGLEV( ui.UDPPortLabel )
234     TOGGLEV( ui.HTTPPort ) ; TOGGLEV( ui.RTPPort ) ; TOGGLEV( ui.MMSHPort ) ; TOGGLEV( ui.UDPPort ) ;
235
236     TOGGLEV( ui.sap ); TOGGLEV( ui.sapName );
237     TOGGLEV( ui.sapGroup ); TOGGLEV( ui.sapGroupLabel );
238     TOGGLEV( ui.ttlLabel ); TOGGLEV( ui.ttl );
239
240     if( b_transcode_only ) okButton->setText( "&Save" );
241     else okButton->setText( "&Stream" );
242
243     updateGeometry();
244 }
245
246 void SoutDialog::ok()
247 {
248     mrl = ui.mrlEdit->text();
249     accept();
250 }
251 void SoutDialog::cancel()
252 {
253     mrl = ui.mrlEdit->text();
254     reject();
255 }
256
257 void SoutDialog::updateMRL()
258 {
259     sout_gui_descr_t sout;
260     memset( &sout, 0, sizeof( sout_gui_descr_t ) );
261     unsigned int counter = 0;
262
263     sout.b_local = ui.localOutput->isChecked();
264     sout.b_file = ui.fileOutput->isChecked();
265     sout.b_http = ui.HTTPOutput->isChecked();
266     sout.b_mms = ui.MMSHOutput->isChecked();
267     sout.b_icecast = ui.IcecastOutput->isChecked();
268     sout.b_rtp = ui.RTPOutput->isChecked();
269     sout.b_udp = ui.UDPOutput->isChecked();
270     sout.b_sap = ui.sap->isChecked();
271     sout.b_all_es = ui.soutAll->isChecked();
272     sout.psz_vcodec = strdup( qtu( ui.vCodecBox->itemData( ui.vCodecBox->currentIndex() ).toString() ) );
273     sout.psz_acodec = strdup( qtu( ui.aCodecBox->itemData( ui.aCodecBox->currentIndex() ).toString() ) );
274     sout.psz_scodec = strdup( qtu( ui.subsCodecBox->itemData( ui.subsCodecBox->currentIndex() ).toString() ) );
275     sout.psz_file = strdup( qtu( ui.fileEdit->text() ) );
276     sout.psz_http = strdup( qtu( ui.HTTPEdit->text() ) );
277     sout.psz_mms = strdup( qtu( ui.MMSHEdit->text() ) );
278     sout.psz_rtp = strdup( qtu( ui.RTPEdit->text() ) );
279     sout.psz_udp = strdup( qtu( ui.UDPEdit->text() ) );
280     sout.psz_icecast = strdup( qtu( ui.IcecastEdit->text() ) );
281     sout.sa_icecast.psz_username = strdup( qtu( ui.IcecastNamePassEdit->text() ) );
282     sout.sa_icecast.psz_password = strdup( qtu( ui.IcecastNamePassEdit->text() ) );
283     sout.psz_icecast_mountpoint = strdup( qtu( ui.IcecastMountpointEdit->text() ) );
284     sout.i_http = ui.HTTPPort->value();
285     sout.i_mms = ui.MMSHPort->value();
286     sout.i_rtp = ui.RTPPort->value();
287     sout.i_udp = ui.UDPPort->value();
288     sout.i_icecast = ui.IcecastPort->value();
289     sout.i_ab = ui.aBitrateSpin->value();
290     sout.i_vb = ui.vBitrateSpin->value();
291     sout.i_channels = ui.aChannelsSpin->value();
292     sout.f_scale = atof( qta( ui.vScaleBox->currentText() ) );
293     sout.psz_group = strdup( qtu( ui.sapGroup->text() ) );
294     sout.psz_name = strdup( qtu( ui.sapName->text() ) );
295
296     if ( sout.b_local ) counter++ ;
297     if ( sout.b_file ) counter++ ;
298     if ( sout.b_http ) counter++ ;
299     if ( sout.b_mms ) counter++ ;
300     if ( sout.b_rtp ) counter++ ;
301     if ( sout.b_udp ) counter ++;
302     if ( sout.b_icecast ) counter ++; 
303
304 #define SMUX( x, txt ) if( ui.x->isChecked() ) sout.psz_mux = strdup( txt );
305     SMUX( PSMux, "ps" );
306     SMUX( TSMux, "ts" );
307     SMUX( MPEG1Mux, "mpeg" );
308     SMUX( OggMux, "ogg" );
309     SMUX( ASFMux, "asf" );
310     SMUX( MP4Mux, "mp4" );
311     SMUX( MOVMux, "mov" );
312     SMUX( WAVMux, "wav" );
313     SMUX( RAWMux, "raw" );
314     SMUX( FLVMux, "flv" );
315     SMUX( MKVMux, "mkv" );
316
317     bool trans = false;
318     bool more = false;
319
320     if ( ui.transcodeVideo->isChecked() || ui.transcodeAudio->isChecked() )
321     {
322         if ( ui.transcodeVideo->isChecked() )
323         {
324             mrl = ":sout=#transcode{";
325             mrl.append( "vcodec=" );
326             mrl.append( sout.psz_vcodec );
327             mrl.append( "," );
328             mrl.append( "vb=" );
329             mrl.append( QString::number( sout.i_vb,10 ) );
330             mrl.append( "," );
331             mrl.append( "scale=" );
332             mrl.append( QString::number( sout.f_scale ) );
333             trans = true;
334         }
335
336         if ( ui.transcodeAudio->isChecked() )
337         {
338             if ( trans )
339             {
340                 mrl.append( "," );
341             }
342             else
343             {
344                 mrl = ":sout=#transcode{";
345             }
346             mrl.append( "acodec=" );
347             mrl.append( sout.psz_acodec );
348             mrl.append( "," );
349             mrl.append( "ab=" );
350             mrl.append( QString::number( sout.i_ab,10 ) );
351             mrl.append( "," );
352             mrl.append( "channels=" );
353             mrl.append( QString::number( sout.i_channels,10 ) );
354             trans = true;
355         }
356         mrl.append( "}" );
357     }
358
359     if ( sout.b_local || sout.b_file || sout.b_http || sout.b_mms || sout.b_rtp || sout.b_udp )
360     {
361
362 #define ISMORE() if ( more ) mrl.append( "," );
363 #define ATLEASTONE() if ( counter ) mrl.append( "dst=" ); 
364
365 #define CHECKMUX() \
366        if( sout.psz_mux ) \
367        {                  \
368          mrl.append( ",mux=");\
369          mrl.append( sout.psz_mux ); \
370        }
371
372         if ( trans )
373         {
374             mrl.append( ":" );
375         }
376         else
377         {
378             mrl = ":sout=#";
379         }
380
381         if ( counter )
382         {
383             mrl.append( "duplicate{" );
384         }
385
386         if ( sout.b_local )
387         {
388             ISMORE();
389             ATLEASTONE()
390                 mrl.append( "display" );
391             more = true;
392         }
393
394         if ( sout.b_file )
395         {
396             ISMORE();
397             ATLEASTONE()
398                 mrl.append( "std{access=file" );
399             CHECKMUX();
400             mrl.append( ",dst=" );
401             mrl.append( sout.psz_file );
402             mrl.append( "}" );
403             more = true;
404         }
405
406         if ( sout.b_http )
407         {
408             ISMORE();
409             ATLEASTONE()
410                 mrl.append( "std{access=http" );
411             CHECKMUX();
412             mrl.append( ",dst=" );
413             mrl.append( sout.psz_http );
414             mrl.append( ":" );
415             mrl.append( QString::number( sout.i_http,10 ) );
416             mrl.append( "}" );
417             more = true;
418         }
419
420         if ( sout.b_mms )
421         {
422             ISMORE();
423             ATLEASTONE()
424                 mrl.append( "std{access=mmsh" );
425             CHECKMUX();
426             mrl.append( ",dst=" );
427             mrl.append( sout.psz_mms );
428             mrl.append( ":" );
429             mrl.append( QString::number( sout.i_mms,10 ) );
430             mrl.append( "}" );
431             more = true;
432         }
433
434         if ( sout.b_rtp )
435         {
436             ISMORE();
437             ATLEASTONE()
438                 mrl.append( "rtp{" );
439             CHECKMUX();
440             mrl.append( ",dst=" );
441             mrl.append( sout.psz_rtp );
442             mrl.append( ":" );
443             mrl.append( QString::number( sout.i_rtp,10 ) );
444             mrl.append( "}" );
445             more = true;
446         }
447
448         if ( sout.b_udp )
449         {
450             ISMORE();
451             ATLEASTONE()
452             mrl.append( "std{access=udp" );
453             CHECKMUX();
454             mrl.append( ",dst=" );
455             mrl.append( sout.psz_udp );
456             mrl.append( ":" );
457             mrl.append( QString::number( sout.i_udp,10 ) );
458             if ( sout.b_sap )
459             {
460                 mrl.append( ",sap," );
461                 mrl.append( "group=\"" );
462                 mrl.append( sout.psz_group );
463                 mrl.append( "\"," );
464                 mrl.append( "name=\"" );
465                 mrl.append( sout.psz_name );
466                 mrl.append( "\"" );
467             }
468             mrl.append( "}" );
469             more = true;
470         }
471
472         if( sout.b_icecast )
473         {
474             // TODO
475         }
476         
477         if ( counter )
478         {
479             mrl.append( "}" );
480         }
481     }
482
483 #undef CHECKMUX
484
485     if ( sout.b_all_es )
486         mrl.append( ":sout-all" );
487
488     ui.mrlEdit->setText( mrl );
489     free( sout.psz_acodec ); free( sout.psz_vcodec ); free( sout.psz_scodec );
490     free( sout.psz_file );free( sout.psz_http ); free( sout.psz_mms );
491     free( sout.psz_rtp ); free( sout.psz_udp ); free( sout.psz_mux );
492     free( sout.psz_name ); free( sout.psz_group );
493     free( sout.psz_icecast ); free( sout.psz_icecast_mountpoint );
494     free( sout.sa_icecast.psz_password ); free( sout.sa_icecast.psz_username );
495 }