]> git.sesse.net Git - vlc/blob - modules/audio_output/directx.c
* ALL: Finally fixed the MSVC project files.
[vlc] / modules / audio_output / directx.c
1 /*****************************************************************************
2  * directx.c: Windows DirectX audio output method
3  *****************************************************************************
4  * Copyright (C) 2001 VideoLAN
5  * $Id: directx.c,v 1.15 2003/03/03 14:21:08 gbazin Exp $
6  *
7  * Authors: Gildas Bazin <gbazin@netcourrier.com>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 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 General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
22  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #include <errno.h>                                                 /* ENOMEM */
28 #include <fcntl.h>                                       /* open(), O_WRONLY */
29 #include <string.h>                                            /* strerror() */
30
31 #include <stdlib.h>                            /* calloc(), malloc(), free() */
32
33 #include <vlc/vlc.h>
34 #include <vlc/aout.h>
35 #include "aout_internal.h"
36
37 #include <windows.h>
38 #include <mmsystem.h>
39 #include <dsound.h>
40
41 #define FRAME_SIZE 2048              /* The size is in samples, not in bytes */
42 #define FRAMES_NUM 4
43
44 /* frame buffer status */
45 #define FRAME_QUEUED 0
46 #define FRAME_EMPTY 1
47
48 /*****************************************************************************
49  * DirectSound GUIDs.
50  * Defining them here allows us to get rid of the dxguid library during
51  * the linking stage.
52  *****************************************************************************/
53 #include <initguid.h>
54 DEFINE_GUID(IID_IDirectSoundNotify, 0xb0210783, 0x89cd, 0x11d0, 0xaf, 0x8, 0x0, 0xa0, 0xc9, 0x25, 0xcd, 0x16);
55
56 /*****************************************************************************
57  * Useful macros
58  *****************************************************************************/
59 #ifndef WAVE_FORMAT_IEEE_FLOAT
60 #   define WAVE_FORMAT_IEEE_FLOAT 0x0003
61 #endif
62
63 #ifndef WAVE_FORMAT_DOLBY_AC3_SPDIF
64 #   define WAVE_FORMAT_DOLBY_AC3_SPDIF 0x0092
65 #endif
66
67 #ifndef WAVE_FORMAT_EXTENSIBLE
68 #define  WAVE_FORMAT_EXTENSIBLE   0xFFFE
69 #endif
70
71 #ifndef SPEAKER_FRONT_LEFT
72 #   define SPEAKER_FRONT_LEFT             0x1
73 #   define SPEAKER_FRONT_RIGHT            0x2
74 #   define SPEAKER_FRONT_CENTER           0x4
75 #   define SPEAKER_LOW_FREQUENCY          0x8
76 #   define SPEAKER_BACK_LEFT              0x10
77 #   define SPEAKER_BACK_RIGHT             0x20
78 #   define SPEAKER_FRONT_LEFT_OF_CENTER   0x40
79 #   define SPEAKER_FRONT_RIGHT_OF_CENTER  0x80
80 #   define SPEAKER_BACK_CENTER            0x100
81 #   define SPEAKER_SIDE_LEFT              0x200
82 #   define SPEAKER_SIDE_RIGHT             0x400
83 #   define SPEAKER_TOP_CENTER             0x800
84 #   define SPEAKER_TOP_FRONT_LEFT         0x1000
85 #   define SPEAKER_TOP_FRONT_CENTER       0x2000
86 #   define SPEAKER_TOP_FRONT_RIGHT        0x4000
87 #   define SPEAKER_TOP_BACK_LEFT          0x8000
88 #   define SPEAKER_TOP_BACK_CENTER        0x10000
89 #   define SPEAKER_TOP_BACK_RIGHT         0x20000
90 #   define SPEAKER_RESERVED               0x80000000
91 #endif
92
93 #ifndef DSSPEAKER_HEADPHONE
94 #   define DSSPEAKER_HEADPHONE         0x00000001
95 #endif
96 #ifndef DSSPEAKER_MONO
97 #   define DSSPEAKER_MONO              0x00000002
98 #endif
99 #ifndef DSSPEAKER_QUAD
100 #   define DSSPEAKER_QUAD              0x00000003
101 #endif
102 #ifndef DSSPEAKER_STEREO
103 #   define DSSPEAKER_STEREO            0x00000004
104 #endif
105 #ifndef DSSPEAKER_SURROUND
106 #   define DSSPEAKER_SURROUND          0x00000005
107 #endif
108 #ifndef DSSPEAKER_5POINT1
109 #   define DSSPEAKER_5POINT1           0x00000006
110 #endif
111
112 #ifndef _WAVEFORMATEXTENSIBLE_
113 typedef struct {
114     WAVEFORMATEX    Format;
115     union {
116         WORD wValidBitsPerSample;       /* bits of precision  */
117         WORD wSamplesPerBlock;          /* valid if wBitsPerSample==0 */
118         WORD wReserved;                 /* If neither applies, set to zero. */
119     } Samples;
120     DWORD           dwChannelMask;      /* which channels are */
121                                         /* present in stream  */
122     GUID            SubFormat;
123 } WAVEFORMATEXTENSIBLE, *PWAVEFORMATEXTENSIBLE;
124 #endif
125
126 #ifndef KSDATAFORMAT_SUBTYPE_IEEE_FLOAT
127 DEFINE_GUID( KSDATAFORMAT_SUBTYPE_IEEE_FLOAT, WAVE_FORMAT_IEEE_FLOAT, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 );
128 #endif
129 #ifndef KSDATAFORMAT_SUBTYPE_PCM
130 DEFINE_GUID( KSDATAFORMAT_SUBTYPE_PCM, WAVE_FORMAT_PCM, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 );
131 #endif
132 #ifndef KSDATAFORMAT_SUBTYPE_DOLBY_AC3_SPDIF
133 DEFINE_GUID( KSDATAFORMAT_SUBTYPE_DOLBY_AC3_SPDIF, WAVE_FORMAT_DOLBY_AC3_SPDIF, 0x0000, 0x0010, 0x80, 0x00, 0x00, 0xaa, 0x00, 0x38, 0x9b, 0x71 );
134 #endif
135
136 /*****************************************************************************
137  * notification_thread_t: DirectX event thread
138  *****************************************************************************/
139 typedef struct notification_thread_t
140 {
141     VLC_COMMON_MEMBERS
142
143     aout_instance_t * p_aout;
144     int i_frame_status[FRAMES_NUM];           /* status of each frame buffer */
145     DSBPOSITIONNOTIFY p_events[FRAMES_NUM];      /* play notification events */
146     int i_frame_size;                         /* Size in bytes of one frame */
147
148     mtime_t start_date;
149
150 } notification_thread_t;
151
152 /*****************************************************************************
153  * aout_sys_t: directx audio output method descriptor
154  *****************************************************************************
155  * This structure is part of the audio output thread descriptor.
156  * It describes the direct sound specific properties of an audio device.
157  *****************************************************************************/
158 struct aout_sys_t
159 {
160     HINSTANCE           hdsound_dll;      /* handle of the opened dsound dll */
161     LPDIRECTSOUND       p_dsobject;              /* main Direct Sound object */
162     LPDIRECTSOUNDBUFFER p_dsbuffer;   /* the sound buffer we use (direct sound
163                                        * takes care of mixing all the
164                                        * secondary buffers into the primary) */
165
166     LPDIRECTSOUNDNOTIFY p_dsnotify;         /* the position notify interface */
167     notification_thread_t *p_notif;                  /* DirectSoundThread id */
168
169     int b_playing;                                         /* playing status */
170
171     int i_frame_size;                         /* Size in bytes of one frame */
172
173     vlc_bool_t b_chan_reorder;              /* do we need channel reordering */
174     int *pi_chan_table;
175     uint32_t i_channel_mask;
176 };
177
178 static const uint32_t pi_channels_in[] =
179     { AOUT_CHAN_LEFT, AOUT_CHAN_RIGHT,
180       AOUT_CHAN_REARLEFT, AOUT_CHAN_REARRIGHT,
181       AOUT_CHAN_CENTER, AOUT_CHAN_LFE };
182 static const uint32_t pi_channels_out[] =
183     { SPEAKER_FRONT_LEFT, SPEAKER_FRONT_RIGHT,
184       SPEAKER_BACK_LEFT, SPEAKER_BACK_RIGHT,
185       SPEAKER_FRONT_CENTER, SPEAKER_LOW_FREQUENCY };
186 static const uint32_t pi_channels_ordered[] =
187     { SPEAKER_FRONT_LEFT, SPEAKER_FRONT_RIGHT, SPEAKER_FRONT_CENTER,
188       SPEAKER_LOW_FREQUENCY,
189       SPEAKER_BACK_LEFT, SPEAKER_BACK_RIGHT };
190
191 /*****************************************************************************
192  * Local prototypes.
193  *****************************************************************************/
194 static int  OpenAudio  ( vlc_object_t * );
195 static void CloseAudio ( vlc_object_t * );
196 static void Play       ( aout_instance_t * );
197
198 /* local functions */
199 static void Probe             ( aout_instance_t * );
200 static int  InitDirectSound   ( aout_instance_t * );
201 static int  CreateDSBuffer    ( aout_instance_t *, int, int, int, int, int, vlc_bool_t );
202 static int  CreateDSBufferPCM ( aout_instance_t *, int*, int, int, int, vlc_bool_t );
203 static void DestroyDSBuffer   ( aout_instance_t * );
204 static void DirectSoundThread ( notification_thread_t * );
205 static int  FillBuffer        ( aout_instance_t *, int, aout_buffer_t * );
206
207 static void CheckReordering   ( aout_instance_t *, int );
208 static void InterleaveFloat32 ( float *, float *, int *, int );
209 static void InterleaveS16     ( int16_t *, int16_t *, int *, int );
210
211 /*****************************************************************************
212  * Module descriptor
213  *****************************************************************************/
214 vlc_module_begin();
215     set_description( _("DirectX audio module") );
216     set_capability( "audio output", 100 );
217     add_shortcut( "directx" );
218     set_callbacks( OpenAudio, CloseAudio );
219 vlc_module_end();
220
221 /*****************************************************************************
222  * OpenAudio: open the audio device
223  *****************************************************************************
224  * This function opens and setups Direct Sound.
225  *****************************************************************************/
226 static int OpenAudio( vlc_object_t *p_this )
227 {
228     aout_instance_t * p_aout = (aout_instance_t *)p_this;
229     vlc_value_t val;
230     int i;
231
232     msg_Dbg( p_aout, "OpenAudio" );
233
234    /* Allocate structure */
235     p_aout->output.p_sys = malloc( sizeof( aout_sys_t ) );
236     if( p_aout->output.p_sys == NULL )
237     {
238         msg_Err( p_aout, "out of memory" );
239         return VLC_EGENERIC;
240     }
241
242     /* Initialize some variables */
243     p_aout->output.p_sys->p_dsobject = NULL;
244     p_aout->output.p_sys->p_dsbuffer = NULL;
245     p_aout->output.p_sys->p_dsnotify = NULL;
246     p_aout->output.p_sys->p_notif = NULL;
247     p_aout->output.p_sys->b_playing = 0;
248     p_aout->output.p_sys->pi_chan_table = NULL;
249
250     p_aout->output.pf_play = Play;
251     aout_VolumeSoftInit( p_aout );
252
253     /* Initialise DirectSound */
254     if( InitDirectSound( p_aout ) )
255     {
256         msg_Err( p_aout, "cannot initialize DirectSound" );
257         goto error;
258     }
259
260     /* Now we need to setup our DirectSound play notification structure */
261     p_aout->output.p_sys->p_notif =
262         vlc_object_create( p_aout, sizeof(notification_thread_t) );
263     p_aout->output.p_sys->p_notif->p_aout = p_aout;
264
265     if( var_Type( p_aout, "audio-device" ) == 0 )
266     {
267         Probe( p_aout );
268     }
269
270     if( var_Get( p_aout, "audio-device", &val ) < 0 )
271     {
272         /* Probe() has failed. */
273         free( p_aout->output.p_sys );
274         return VLC_EGENERIC;
275     }
276
277     /* Then create the notification events */
278     for( i = 0; i < FRAMES_NUM; i++ )
279         p_aout->output.p_sys->p_notif->p_events[i].hEventNotify =
280             CreateEvent( NULL, FALSE, FALSE, NULL );
281
282     /* Open the device */
283     if( !strcmp( val.psz_string, N_("A/52 over S/PDIF") ) )
284     {
285         free( val.psz_string );
286         p_aout->output.output.i_format = VLC_FOURCC('s','p','d','i');
287
288         /* Calculate the frame size in bytes */
289         p_aout->output.i_nb_samples = A52_FRAME_NB;
290         p_aout->output.output.i_bytes_per_frame = AOUT_SPDIF_SIZE;
291         p_aout->output.output.i_frame_length = A52_FRAME_NB;
292         p_aout->output.p_sys->i_frame_size =
293             p_aout->output.output.i_bytes_per_frame;
294
295         if( CreateDSBuffer( p_aout, VLC_FOURCC('s','p','d','i'),
296                             p_aout->output.output.i_physical_channels,
297                             aout_FormatNbChannels( &p_aout->output.output ),
298                             p_aout->output.output.i_rate,
299                             p_aout->output.p_sys->i_frame_size, VLC_FALSE )
300             != VLC_SUCCESS )
301         {
302             msg_Err( p_aout, "cannot open waveout audio device" );
303             free( p_aout->output.p_sys );
304             return VLC_EGENERIC;
305         }
306
307         aout_VolumeNoneInit( p_aout );
308     }
309     else
310     {
311         if( !strcmp( val.psz_string, N_("5.1") ) )
312         {
313             p_aout->output.output.i_physical_channels
314                 = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT | AOUT_CHAN_CENTER
315                    | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT
316                    | AOUT_CHAN_LFE;
317         }
318         else if( !strcmp( val.psz_string, N_("2 Front 2 Rear") ) )
319         {
320             p_aout->output.output.i_physical_channels
321                 = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT
322                    | AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
323         }
324         else if( !strcmp( val.psz_string, "Mono" ) )
325         {
326             p_aout->output.output.i_physical_channels = AOUT_CHAN_CENTER;
327         }
328         else
329         {
330             p_aout->output.output.i_physical_channels
331                 = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
332         }
333         free( val.psz_string );
334
335         if( CreateDSBufferPCM( p_aout, &p_aout->output.output.i_format,
336                                p_aout->output.output.i_physical_channels,
337                                aout_FormatNbChannels( &p_aout->output.output ),
338                                p_aout->output.output.i_rate, VLC_FALSE )
339             != VLC_SUCCESS )
340         {
341             msg_Err( p_aout, "cannot open waveout audio device" );
342             free( p_aout->output.p_sys );
343             return VLC_EGENERIC;
344         }
345
346         /* Calculate the frame size in bytes */
347         p_aout->output.i_nb_samples = FRAME_SIZE;
348         aout_FormatPrepare( &p_aout->output.output );
349         p_aout->output.p_sys->i_frame_size =
350             FRAME_SIZE * p_aout->output.output.i_bytes_per_frame;
351
352         aout_VolumeSoftInit( p_aout );
353     }
354
355     /* then launch the notification thread */
356     msg_Dbg( p_aout, "creating DirectSoundThread" );
357     if( vlc_thread_create( p_aout->output.p_sys->p_notif,
358                            "DirectSound Notification Thread",
359                            DirectSoundThread,
360                            VLC_THREAD_PRIORITY_HIGHEST, VLC_FALSE ) )
361     {
362         msg_Err( p_aout, "cannot create DirectSoundThread" );
363         goto error;
364     }
365
366     vlc_object_attach( p_aout->output.p_sys->p_notif, p_aout );
367
368     return VLC_SUCCESS;
369
370  error:
371     CloseAudio( VLC_OBJECT(p_aout) );
372     return VLC_EGENERIC;
373 }
374
375 /*****************************************************************************
376  * Probe: probe the audio device for available formats and channels
377  *****************************************************************************/
378 static void Probe( aout_instance_t * p_aout )
379 {
380     vlc_value_t val;
381     int i_format;
382     unsigned int i_physical_channels;
383     DWORD ui_speaker_config;
384
385     var_Create( p_aout, "audio-device", VLC_VAR_STRING | VLC_VAR_HASCHOICE );
386
387     /* Test for 5.1 support */
388     i_physical_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT |
389                           AOUT_CHAN_CENTER | AOUT_CHAN_REARLEFT |
390                           AOUT_CHAN_REARRIGHT | AOUT_CHAN_LFE;
391     if( p_aout->output.output.i_physical_channels == i_physical_channels )
392     {
393         if( CreateDSBufferPCM( p_aout, &i_format, i_physical_channels, 6,
394                                p_aout->output.output.i_rate, VLC_TRUE )
395             == VLC_SUCCESS )
396         {
397             val.psz_string = N_("5.1");
398             var_Change( p_aout, "audio-device", VLC_VAR_ADDCHOICE, &val );
399             msg_Dbg( p_aout, "device supports 5.1 channels" );
400         }
401     }
402
403     /* Test for 2 Front 2 Rear support */
404     i_physical_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT |
405                           AOUT_CHAN_REARLEFT | AOUT_CHAN_REARRIGHT;
406     if( ( p_aout->output.output.i_physical_channels & i_physical_channels )
407         == i_physical_channels )
408     {
409         if( CreateDSBufferPCM( p_aout, &i_format, i_physical_channels, 4,
410                                p_aout->output.output.i_rate, VLC_TRUE )
411             == VLC_SUCCESS )
412         {
413             val.psz_string = N_("2 Front 2 Rear");
414             var_Change( p_aout, "audio-device", VLC_VAR_ADDCHOICE, &val );
415             msg_Dbg( p_aout, "device supports 4 channels" );
416         }
417     }
418
419     /* Test for stereo support */
420     i_physical_channels = AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
421     if( CreateDSBufferPCM( p_aout, &i_format, i_physical_channels, 2,
422                            p_aout->output.output.i_rate, VLC_TRUE )
423         == VLC_SUCCESS )
424     {
425         val.psz_string = N_("Stereo");
426         var_Change( p_aout, "audio-device", VLC_VAR_ADDCHOICE, &val );
427         msg_Dbg( p_aout, "device supports 2 channels" );
428     }
429
430     /* Test for mono support */
431     i_physical_channels = AOUT_CHAN_CENTER;
432     if( CreateDSBufferPCM( p_aout, &i_format, i_physical_channels, 1,
433                            p_aout->output.output.i_rate, VLC_TRUE )
434         == VLC_SUCCESS )
435     {
436         val.psz_string = N_("Mono");
437         var_Change( p_aout, "audio-device", VLC_VAR_ADDCHOICE, &val );        
438         msg_Dbg( p_aout, "device supports 1 channel" );
439     }
440
441     /* Check the speaker configuration to determine which channel config should
442      * be the default */
443     if FAILED( IDirectSound_GetSpeakerConfig( p_aout->output.p_sys->p_dsobject,
444                                               &ui_speaker_config ) )
445     {
446         ui_speaker_config = DSSPEAKER_STEREO;
447     }
448     switch( DSSPEAKER_CONFIG(ui_speaker_config) )
449     {
450     case DSSPEAKER_5POINT1:
451         val.psz_string = N_("5.1");
452         break;
453     case DSSPEAKER_QUAD:
454         val.psz_string = N_("2 Front 2 Rear");
455         break;
456     case DSSPEAKER_MONO:
457         val.psz_string = N_("Mono");
458         break;
459     case DSSPEAKER_SURROUND:
460     case DSSPEAKER_STEREO:
461     default:
462         val.psz_string = N_("Stereo");
463         break;
464     }
465     var_Set( p_aout, "audio-device", val );
466
467     /* Test for SPDIF support */
468     if ( AOUT_FMT_NON_LINEAR( &p_aout->output.output ) )
469     {
470         if( CreateDSBuffer( p_aout, VLC_FOURCC('s','p','d','i'),
471                             p_aout->output.output.i_physical_channels,
472                             aout_FormatNbChannels( &p_aout->output.output ),
473                             p_aout->output.output.i_rate,
474                             AOUT_SPDIF_SIZE, VLC_TRUE )
475             == VLC_SUCCESS )
476         {
477             msg_Dbg( p_aout, "device supports A/52 over S/PDIF" );
478             val.psz_string = N_("A/52 over S/PDIF");
479             var_Change( p_aout, "audio-device", VLC_VAR_ADDCHOICE, &val );
480             if( config_GetInt( p_aout, "spdif" ) )
481                 var_Set( p_aout, "audio-device", val );
482         }
483     }
484
485     var_AddCallback( p_aout, "audio-device", aout_ChannelsRestart, NULL );
486
487     val.b_bool = VLC_TRUE;
488     var_Set( p_aout, "intf-change", val );
489 }
490
491 /*****************************************************************************
492  * Play: we'll start playing the directsound buffer here because at least here
493  *       we know the first buffer has been put in the aout fifo and we also
494  *       know its date.
495  *****************************************************************************/
496 static void Play( aout_instance_t *p_aout )
497 {
498     if( !p_aout->output.p_sys->b_playing )
499     {
500         aout_buffer_t *p_buffer;
501
502         p_aout->output.p_sys->b_playing = 1;
503
504         /* get the playing date of the first aout buffer */
505         p_aout->output.p_sys->p_notif->start_date =
506             aout_FifoFirstDate( p_aout, &p_aout->output.fifo );
507
508         /* fill in the first samples */
509         p_buffer = aout_FifoPop( p_aout, &p_aout->output.fifo );
510         FillBuffer( p_aout, 0, p_buffer );
511
512         /* wake up the audio output thread */
513         SetEvent( p_aout->output.p_sys->p_notif->p_events[0].hEventNotify );
514     }
515 }
516
517 /*****************************************************************************
518  * CloseAudio: close the audio device
519  *****************************************************************************/
520 static void CloseAudio( vlc_object_t *p_this )
521 {
522     aout_instance_t * p_aout = (aout_instance_t *)p_this;
523
524     msg_Dbg( p_aout, "CloseAudio" );
525
526     /* kill the position notification thread, if any */
527     if( p_aout->output.p_sys->p_notif )
528     {
529         vlc_object_detach( p_aout->output.p_sys->p_notif );
530         if( p_aout->output.p_sys->p_notif->b_thread )
531         {
532             p_aout->output.p_sys->p_notif->b_die = 1;
533
534             if( !p_aout->output.p_sys->b_playing )
535                 /* wake up the audio thread */
536                 SetEvent(
537                     p_aout->output.p_sys->p_notif->p_events[0].hEventNotify );
538
539             vlc_thread_join( p_aout->output.p_sys->p_notif );
540         }
541         vlc_object_destroy( p_aout->output.p_sys->p_notif );
542     }
543
544     /* release the secondary buffer */
545     DestroyDSBuffer( p_aout );
546
547     /* finally release the DirectSound object */
548     if( p_aout->output.p_sys->p_dsobject )
549         IDirectSound_Release( p_aout->output.p_sys->p_dsobject );
550     
551     /* free DSOUND.DLL */
552     if( p_aout->output.p_sys->hdsound_dll )
553        FreeLibrary( p_aout->output.p_sys->hdsound_dll );
554
555     if( p_aout->output.p_sys->pi_chan_table )
556         free( p_aout->output.p_sys->pi_chan_table );
557
558     free( p_aout->output.p_sys );
559 }
560
561 /*****************************************************************************
562  * InitDirectSound: handle all the gory details of DirectSound initialisation
563  *****************************************************************************/
564 static int InitDirectSound( aout_instance_t *p_aout )
565 {
566     HRESULT (WINAPI *OurDirectSoundCreate)(LPGUID, LPDIRECTSOUND *, LPUNKNOWN);
567
568     p_aout->output.p_sys->hdsound_dll = LoadLibrary("DSOUND.DLL");
569     if( p_aout->output.p_sys->hdsound_dll == NULL )
570     {
571         msg_Warn( p_aout, "cannot open DSOUND.DLL" );
572         goto error;
573     }
574
575     OurDirectSoundCreate = (void *)GetProcAddress(
576                                            p_aout->output.p_sys->hdsound_dll,
577                                            "DirectSoundCreate" );
578     if( OurDirectSoundCreate == NULL )
579     {
580         msg_Warn( p_aout, "GetProcAddress FAILED" );
581         goto error;
582     }
583
584     /* Create the direct sound object */
585     if FAILED( OurDirectSoundCreate( NULL, &p_aout->output.p_sys->p_dsobject,
586                                      NULL ) )
587     {
588         msg_Warn( p_aout, "cannot create a direct sound device" );
589         goto error;
590     }
591
592     /* Set DirectSound Cooperative level, ie what control we want over Windows
593      * sound device. In our case, DSSCL_EXCLUSIVE means that we can modify the
594      * settings of the primary buffer, but also that only the sound of our
595      * application will be hearable when it will have the focus.
596      * !!! (this is not really working as intended yet because to set the
597      * cooperative level you need the window handle of your application, and
598      * I don't know of any easy way to get it. Especially since we might play
599      * sound without any video, and so what window handle should we use ???
600      * The hack for now is to use the Desktop window handle - it seems to be
601      * working */
602     if( IDirectSound_SetCooperativeLevel( p_aout->output.p_sys->p_dsobject,
603                                           GetDesktopWindow(),
604                                           DSSCL_EXCLUSIVE) )
605     {
606         msg_Warn( p_aout, "cannot set direct sound cooperative level" );
607     }
608
609     return VLC_SUCCESS;
610
611  error:
612     p_aout->output.p_sys->p_dsobject = NULL;
613     if( p_aout->output.p_sys->hdsound_dll )
614     {
615         FreeLibrary( p_aout->output.p_sys->hdsound_dll );
616         p_aout->output.p_sys->hdsound_dll = NULL;
617     }
618     return VLC_EGENERIC;
619
620 }
621
622 /*****************************************************************************
623  * CreateDSBuffer: Creates a direct sound buffer of the required format.
624  *****************************************************************************
625  * This function creates the buffer we'll use to play audio.
626  * In DirectSound there are two kinds of buffers:
627  * - the primary buffer: which is the actual buffer that the soundcard plays
628  * - the secondary buffer(s): these buffers are the one actually used by
629  *    applications and DirectSound takes care of mixing them into the primary.
630  *
631  * Once you create a secondary buffer, you cannot change its format anymore so
632  * you have to release the current one and create another.
633  *****************************************************************************/
634 static int CreateDSBuffer( aout_instance_t *p_aout, int i_format,
635                            int i_channels, int i_nb_channels, int i_rate,
636                            int i_bytes_per_frame, vlc_bool_t b_probe )
637 {
638     WAVEFORMATEXTENSIBLE waveformat;
639     DSBUFFERDESC         dsbdesc;
640     unsigned int         i;
641
642     /* First set the sound buffer format */
643     waveformat.dwChannelMask = 0;
644     for( i = 0; i < sizeof(pi_channels_in)/sizeof(uint32_t); i++ )
645     {
646         if( i_channels & pi_channels_in[i] )
647             waveformat.dwChannelMask |= pi_channels_out[i];
648     }
649
650     switch( i_format )
651     {
652     case VLC_FOURCC('s','p','d','i'):
653         i_nb_channels = 2;
654         /* To prevent channel re-ordering */
655         waveformat.dwChannelMask = SPEAKER_FRONT_LEFT | SPEAKER_FRONT_RIGHT;
656         waveformat.Format.wBitsPerSample = 16;
657         waveformat.Samples.wValidBitsPerSample =
658             waveformat.Format.wBitsPerSample;
659         waveformat.Format.wFormatTag = WAVE_FORMAT_DOLBY_AC3_SPDIF;
660         waveformat.SubFormat = KSDATAFORMAT_SUBTYPE_DOLBY_AC3_SPDIF;
661         break;
662
663     case VLC_FOURCC('f','l','3','2'):
664         waveformat.Format.wBitsPerSample = sizeof(float) * 8;
665         waveformat.Samples.wValidBitsPerSample =
666             waveformat.Format.wBitsPerSample;
667         waveformat.Format.wFormatTag = WAVE_FORMAT_IEEE_FLOAT;
668         waveformat.SubFormat = KSDATAFORMAT_SUBTYPE_IEEE_FLOAT;
669         break;
670
671     case VLC_FOURCC('s','1','6','l'):
672         waveformat.Format.wBitsPerSample = 16;
673         waveformat.Samples.wValidBitsPerSample =
674             waveformat.Format.wBitsPerSample;
675         waveformat.Format.wFormatTag = WAVE_FORMAT_PCM;
676         waveformat.SubFormat = KSDATAFORMAT_SUBTYPE_PCM;
677         break;
678     }
679
680     waveformat.Format.nChannels = i_nb_channels;
681     waveformat.Format.nSamplesPerSec = i_rate;
682     waveformat.Format.nBlockAlign =
683         waveformat.Format.wBitsPerSample / 8 * i_nb_channels;
684     waveformat.Format.nAvgBytesPerSec =
685         waveformat.Format.nSamplesPerSec * waveformat.Format.nBlockAlign;
686
687     /* Only use the new WAVE_FORMAT_EXTENSIBLE format for multichannel audio */
688     if( i_nb_channels <= 2 )
689     {
690         waveformat.Format.cbSize = 0;
691     }
692     else
693     {
694         waveformat.Format.wFormatTag = WAVE_FORMAT_EXTENSIBLE;
695         waveformat.Format.cbSize =
696             sizeof(WAVEFORMATEXTENSIBLE) - sizeof(WAVEFORMATEX);
697     }
698
699
700     /* Then fill in the direct sound descriptor */
701     memset(&dsbdesc, 0, sizeof(DSBUFFERDESC));
702     dsbdesc.dwSize = sizeof(DSBUFFERDESC);
703     dsbdesc.dwFlags = DSBCAPS_GETCURRENTPOSITION2/* Better position accuracy */
704                     | DSBCAPS_CTRLPOSITIONNOTIFY     /* We need notification */
705                     | DSBCAPS_GLOBALFOCUS;      /* Allows background playing */
706     dsbdesc.dwBufferBytes = FRAMES_NUM * i_bytes_per_frame;   /* buffer size */
707     dsbdesc.lpwfxFormat = (WAVEFORMATEX *)&waveformat;
708
709     if FAILED( IDirectSound_CreateSoundBuffer(
710                    p_aout->output.p_sys->p_dsobject, &dsbdesc,
711                    &p_aout->output.p_sys->p_dsbuffer, NULL) )
712     {
713         goto error;
714     }
715
716     /* Stop here if we were just probing */
717     if( b_probe )
718     {
719         IDirectSoundBuffer_Release( p_aout->output.p_sys->p_dsbuffer );
720         p_aout->output.p_sys->p_dsbuffer = NULL;
721         return VLC_SUCCESS;
722     }
723
724     /* Backup the size of a frame */
725     p_aout->output.p_sys->p_notif->i_frame_size = i_bytes_per_frame;
726
727     /* Now the secondary buffer is created, we need to setup its position
728      * notification */
729     for( i = 0; i < FRAMES_NUM; i++ )
730     {
731         p_aout->output.p_sys->p_notif->p_events[i].dwOffset = i *
732             p_aout->output.p_sys->p_notif->i_frame_size;
733
734         p_aout->output.p_sys->p_notif->i_frame_status[i] = FRAME_EMPTY;
735     }
736
737     /* Get the IDirectSoundNotify interface */
738     if FAILED( IDirectSoundBuffer_QueryInterface(
739                                 p_aout->output.p_sys->p_dsbuffer,
740                                 &IID_IDirectSoundNotify,
741                                 (LPVOID *)&p_aout->output.p_sys->p_dsnotify ) )
742     {
743         msg_Err( p_aout, "cannot get IDirectSoundNotify interface" );
744         goto error;
745     }
746
747     if FAILED( IDirectSoundNotify_SetNotificationPositions(
748                                     p_aout->output.p_sys->p_dsnotify,
749                                     FRAMES_NUM,
750                                     p_aout->output.p_sys->p_notif->p_events ) )
751     {
752         msg_Err( p_aout, "cannot set position notification" );
753         goto error;
754     }
755
756     p_aout->output.p_sys->i_channel_mask = waveformat.dwChannelMask;
757     CheckReordering( p_aout, i_nb_channels );
758
759     return VLC_SUCCESS;
760
761  error:
762     if( p_aout->output.p_sys->p_dsbuffer )
763     {
764         IDirectSoundBuffer_Release( p_aout->output.p_sys->p_dsbuffer );
765         p_aout->output.p_sys->p_dsbuffer = NULL;
766     }
767     if( p_aout->output.p_sys->p_dsnotify )
768     {
769         IDirectSoundBuffer_Release( p_aout->output.p_sys->p_dsbuffer );
770         p_aout->output.p_sys->p_dsnotify = NULL;
771     }
772     return VLC_EGENERIC;
773 }
774
775 /*****************************************************************************
776  * CreateDSBufferPCM: creates a PCM direct sound buffer.
777  *****************************************************************************
778  * We first try to create a WAVE_FORMAT_IEEE_FLOAT buffer if supported by
779  * the hardware, otherwise we create a WAVE_FORMAT_PCM buffer.
780  ****************************************************************************/
781 static int CreateDSBufferPCM( aout_instance_t *p_aout, int *i_format,
782                               int i_channels, int i_nb_channels, int i_rate,
783                               vlc_bool_t b_probe )
784 {
785     if( CreateDSBuffer( p_aout, VLC_FOURCC('f','l','3','2'),
786                         i_channels, i_nb_channels, i_rate,
787                         FRAME_SIZE * 4 * i_nb_channels, b_probe )
788         != VLC_SUCCESS )
789     {
790         if ( CreateDSBuffer( p_aout, VLC_FOURCC('s','1','6','l'),
791                              i_channels, i_nb_channels, i_rate,
792                              FRAME_SIZE * 2 * i_nb_channels, b_probe )
793              != VLC_SUCCESS )
794         {
795             return VLC_EGENERIC;
796         }
797         else
798         {
799             *i_format = VLC_FOURCC('s','1','6','l');
800             return VLC_SUCCESS;
801         }
802     }
803     else
804     {
805         *i_format = VLC_FOURCC('f','l','3','2');
806         return VLC_SUCCESS;
807     }
808 }
809
810 /*****************************************************************************
811  * DestroyDSBuffer
812  *****************************************************************************
813  * This function destroys the secondary buffer.
814  *****************************************************************************/
815 static void DestroyDSBuffer( aout_instance_t *p_aout )
816 {
817     if( p_aout->output.p_sys->p_dsnotify )
818     {
819         IDirectSoundNotify_Release( p_aout->output.p_sys->p_dsnotify );
820         p_aout->output.p_sys->p_dsnotify = NULL;
821     }
822
823     if( p_aout->output.p_sys->p_dsbuffer )
824     {
825         IDirectSoundBuffer_Release( p_aout->output.p_sys->p_dsbuffer );
826         p_aout->output.p_sys->p_dsbuffer = NULL;
827     }
828 }
829
830 /*****************************************************************************
831  * FillBuffer: Fill in one of the direct sound frame buffers.
832  *****************************************************************************
833  * Returns VLC_SUCCESS on success.
834  *****************************************************************************/
835 static int FillBuffer( aout_instance_t *p_aout, int i_frame,
836                        aout_buffer_t *p_buffer )
837 {
838     notification_thread_t *p_notif = p_aout->output.p_sys->p_notif;
839     void *p_write_position, *p_wrap_around;
840     long l_bytes1, l_bytes2;
841     HRESULT dsresult;
842
843     /* Before copying anything, we have to lock the buffer */
844     dsresult = IDirectSoundBuffer_Lock(
845                 p_aout->output.p_sys->p_dsbuffer,               /* DS buffer */
846                 i_frame * p_notif->i_frame_size,             /* Start offset */
847                 p_notif->i_frame_size,                    /* Number of bytes */
848                 &p_write_position,                  /* Address of lock start */
849                 &l_bytes1,       /* Count of bytes locked before wrap around */
850                 &p_wrap_around,            /* Buffer adress (if wrap around) */
851                 &l_bytes2,               /* Count of bytes after wrap around */
852                 0 );                                                /* Flags */
853     if( dsresult == DSERR_BUFFERLOST )
854     {
855         IDirectSoundBuffer_Restore( p_aout->output.p_sys->p_dsbuffer );
856         dsresult = IDirectSoundBuffer_Lock(
857                                p_aout->output.p_sys->p_dsbuffer,
858                                i_frame * p_notif->i_frame_size,
859                                p_notif->i_frame_size,
860                                &p_write_position,
861                                &l_bytes1,
862                                &p_wrap_around,
863                                &l_bytes2,
864                                0 );
865     }
866     if( dsresult != DS_OK )
867     {
868         msg_Warn( p_notif, "cannot lock buffer" );
869         if( p_buffer ) aout_BufferFree( p_buffer );
870         return VLC_EGENERIC;
871     }
872
873     if( p_buffer == NULL )
874     {
875         memset( p_write_position, 0, l_bytes1 );
876     }
877     else if( p_aout->output.p_sys->b_chan_reorder )
878     {
879         /* Do the channel reordering here */
880
881         if( p_aout->output.output.i_format ==  VLC_FOURCC('s','1','6','l') )
882           InterleaveS16( (int16_t *)p_buffer->p_buffer,
883                          (int16_t *)p_write_position,
884                          p_aout->output.p_sys->pi_chan_table,
885                          aout_FormatNbChannels( &p_aout->output.output ) );
886         else
887           InterleaveFloat32( (float *)p_buffer->p_buffer,
888                              (float *)p_write_position,
889                              p_aout->output.p_sys->pi_chan_table,
890                              aout_FormatNbChannels( &p_aout->output.output ) );
891     }
892     else
893     {
894         p_aout->p_vlc->pf_memcpy( p_write_position, p_buffer->p_buffer,
895                                   l_bytes1 );
896         aout_BufferFree( p_buffer );
897     }
898
899     /* Now the data has been copied, unlock the buffer */
900     IDirectSoundBuffer_Unlock( p_aout->output.p_sys->p_dsbuffer,
901                                p_write_position, l_bytes1,
902                                p_wrap_around, l_bytes2 );
903
904     return VLC_SUCCESS;
905 }
906
907 /*****************************************************************************
908  * DirectSoundThread: this thread will capture play notification events. 
909  *****************************************************************************
910  * We use this thread to emulate a callback mechanism. The thread probes for
911  * event notification and fills up the DS secondary buffer when needed.
912  *****************************************************************************/
913 static void DirectSoundThread( notification_thread_t *p_notif )
914 {
915     HANDLE  notification_events[FRAMES_NUM];
916     HRESULT dsresult;
917     aout_instance_t *p_aout = p_notif->p_aout;
918     int i, i_which_frame, i_last_frame, i_next_frame;
919     mtime_t mtime;
920     vlc_bool_t b_sleek;
921
922     for( i = 0; i < FRAMES_NUM; i++ )
923         notification_events[i] = p_notif->p_events[i].hEventNotify;
924
925     /* We don't want any resampling when using S/PDIF output */
926     b_sleek = p_aout->output.output.i_format == VLC_FOURCC('s','p','d','i');
927
928     /* Tell the main thread that we are ready */
929     vlc_thread_ready( p_notif );
930
931     msg_Dbg( p_notif, "DirectSoundThread ready" );
932
933     /* Wait here until Play() is called */
934     WaitForSingleObject( notification_events[0], INFINITE );
935
936     if( !p_notif->b_die )
937     {
938         mwait( p_notif->start_date - AOUT_PTS_TOLERANCE / 2 );
939
940         /* start playing the buffer */
941         dsresult = IDirectSoundBuffer_Play( p_aout->output.p_sys->p_dsbuffer,
942                                         0,                         /* Unused */
943                                         0,                         /* Unused */
944                                         DSBPLAY_LOOPING );          /* Flags */
945         if( dsresult == DSERR_BUFFERLOST )
946         {
947             IDirectSoundBuffer_Restore( p_aout->output.p_sys->p_dsbuffer );
948             dsresult = IDirectSoundBuffer_Play(
949                                             p_aout->output.p_sys->p_dsbuffer,
950                                             0,                     /* Unused */
951                                             0,                     /* Unused */
952                                             DSBPLAY_LOOPING );      /* Flags */
953         }
954         if( dsresult != DS_OK )
955         {
956             msg_Err( p_aout, "cannot start playing buffer" );
957         }
958     }
959
960     while( !p_notif->b_die )
961     {
962         aout_buffer_t *p_buffer;
963         long l_latency;
964
965         /* wait for the position notification */
966         i_which_frame = WaitForMultipleObjects( FRAMES_NUM,
967                                                 notification_events, 0,
968                                                 INFINITE ) - WAIT_OBJECT_0;
969
970         if( p_notif->b_die )
971             break;
972
973         mtime = mdate();
974
975         /* We take into account the current latency */
976         if SUCCEEDED( IDirectSoundBuffer_GetCurrentPosition(
977                         p_aout->output.p_sys->p_dsbuffer,
978                         &l_latency, NULL ) )
979         {
980             if( l_latency > (i_which_frame * FRAME_SIZE)
981                   && l_latency < ((i_which_frame+1) * FRAME_SIZE) )
982             {
983                 l_latency = - ( l_latency /
984                                 p_aout->output.output.i_bytes_per_frame %
985                                 FRAME_SIZE );
986             }
987             else
988             {
989                 l_latency = FRAME_SIZE - ( l_latency /
990                                       p_aout->output.output.i_bytes_per_frame %
991                                       FRAME_SIZE );
992             }
993         }
994         else
995         {
996             l_latency = 0;
997         }
998
999         /* Mark last frame as empty */
1000         i_last_frame = (i_which_frame + FRAMES_NUM -1) % FRAMES_NUM;
1001         i_next_frame = (i_which_frame + 1) % FRAMES_NUM;
1002         p_notif->i_frame_status[i_last_frame] = FRAME_EMPTY;
1003
1004         /* Try to fill in as many frame buffers as possible */
1005         for( i = i_next_frame; (i % FRAMES_NUM) != i_which_frame; i++ )
1006         {
1007
1008             /* Check if frame buf is already filled */
1009             if( p_notif->i_frame_status[i % FRAMES_NUM] == FRAME_QUEUED )
1010                 continue;
1011
1012             p_buffer = aout_OutputNextBuffer( p_aout,
1013                 mtime + 1000000 / p_aout->output.output.i_rate *
1014                 ((i - i_next_frame + 1) * FRAME_SIZE + l_latency), b_sleek );
1015
1016             /* If there is no audio data available and we have some buffered
1017              * already, then just wait for the next time */
1018             if( !p_buffer && (i != i_next_frame) )
1019             {
1020                 //msg_Err( p_aout, "only %i frame buffers filled!",
1021                 //         i - i_next_frame );
1022                 break;
1023             }
1024
1025             if( FillBuffer( p_aout, (i%FRAMES_NUM), p_buffer )
1026                 != VLC_SUCCESS )
1027                 break;
1028
1029             /* Mark the frame buffer as QUEUED */
1030             p_notif->i_frame_status[i%FRAMES_NUM] = FRAME_QUEUED;
1031         }
1032
1033     }
1034
1035     /* make sure the buffer isn't playing */
1036     IDirectSoundBuffer_Stop( p_aout->output.p_sys->p_dsbuffer );
1037
1038     /* free the events */
1039     for( i = 0; i < FRAMES_NUM; i++ )
1040         CloseHandle( notification_events[i] );
1041
1042     msg_Dbg( p_notif, "DirectSoundThread exiting" );
1043 }
1044
1045 /*****************************************************************************
1046  * CheckReordering: Check if we need to do some channel re-ordering (the ac3
1047  *                  channel order is different from the one chosen by
1048  *                  Microsoft).
1049  *****************************************************************************/
1050 static void CheckReordering( aout_instance_t *p_aout, int i_nb_channels )
1051 {
1052     int i, j, k, l;
1053
1054 #define i_channel_mask p_aout->output.p_sys->i_channel_mask
1055 #define pi_chan_table p_aout->output.p_sys->pi_chan_table
1056
1057     p_aout->output.p_sys->b_chan_reorder = VLC_FALSE;
1058
1059     pi_chan_table = malloc( i_nb_channels * sizeof(int) );
1060     if( !pi_chan_table )
1061     {
1062         return;
1063     }
1064
1065     for( i = 0, j = 0;
1066          i < (int)(sizeof(pi_channels_out)/sizeof(uint32_t)); i++ )
1067     {
1068         if( i_channel_mask & pi_channels_out[i] )
1069         {
1070             for( k = 0, l = 0;
1071                  pi_channels_out[i] != pi_channels_ordered[k]; k++ )
1072             {
1073                 if( i_channel_mask & pi_channels_ordered[k] )
1074                 {
1075                     l++;
1076                 }
1077             }
1078
1079             pi_chan_table[j] = l;
1080
1081             j++;
1082         }
1083     }
1084
1085     for( i = 0; i < i_nb_channels; i++ )
1086     {
1087         if( pi_chan_table[i] != i )
1088         {
1089             p_aout->output.p_sys->b_chan_reorder = VLC_TRUE;
1090         }
1091     }
1092
1093     if( p_aout->output.p_sys->b_chan_reorder )
1094     {
1095         msg_Dbg( p_aout, "channel reordering needed" );
1096     }
1097
1098 #undef pi_chan_table
1099 #undef waveformat
1100 }
1101
1102 /*****************************************************************************
1103  * InterleaveFloat32/S16: change the channel order to the Microsoft one.
1104  *****************************************************************************/
1105 static void InterleaveFloat32( float *p_buf, float *p_buf_out,
1106                                int *pi_chan_table, int i_nb_channels )
1107 {
1108     int i, j;
1109
1110     for( i = 0; i < FRAME_SIZE; i++ )
1111     {
1112         for( j = 0; j < i_nb_channels; j++ )
1113         {
1114             p_buf_out[i*i_nb_channels + pi_chan_table[j]] =
1115                 p_buf[i*i_nb_channels + j];
1116         }
1117     }
1118 }
1119
1120 static void InterleaveS16( int16_t *p_buf, int16_t *p_buf_out,
1121                            int *pi_chan_table, int i_nb_channels )
1122 {
1123     int i, j;
1124
1125     for( i = 0; i < FRAME_SIZE; i++ )
1126     {
1127         for( j = 0; j < i_nb_channels; j++ )
1128         {
1129             p_buf_out[ i*i_nb_channels + pi_chan_table[j]] =
1130                 p_buf[i*i_nb_channels + j];
1131         }
1132     }
1133 }