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