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