git.sesse.net Git - vlc/blob - modules/access/alsa.c

   1 /*****************************************************************************
   2  * alsa.c: ALSA capture module for VLC
   3  *****************************************************************************
   4  * Copyright (C) 2012 Rémi Denis-Courmont
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU Lesser General Public License as published by
   8  * the Free Software Foundation; either version 2.1 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14  * GNU Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  19  *****************************************************************************/
  20
  21 #ifdef HAVE_CONFIG_H
  22 # include <config.h>
  23 #endif
  24
  25 #include <assert.h>
  26 #include <sys/types.h>
  27 #include <poll.h>
  28 #include <alsa/asoundlib.h>
  29
  30 #include <vlc_common.h>
  31 #include <vlc_demux.h>
  32 #include <vlc_aout.h>
  33 #include <vlc_plugin.h>
  34
  35 #define HELP_TEXT N_( \
  36     "Pass alsa:// to open the default ALSA capture device, " \
  37     "or alsa://SOURCE to open a specific device named SOURCE.")
  38 #define STEREO_TEXT N_("Stereo")
  39 #define RATE_TEXT N_("Sample rate")
  40
  41 static int Open (vlc_object_t *);
  42 static void Close (vlc_object_t *);
  43
  44 static const int rate_values[] = { 192000, 176400,
  45     96000, 88200, 48000, 44100,
  46     32000, 22050, 24000, 16000,
  47     11025, 8000, 4000
  48 };
  49 static const const char *rate_names[] = { N_("192000 Hz"), N_("176400 Hz"),
  50     N_("96000 Hz"), N_("88200 Hz"), N_("48000 Hz"), N_("44100 Hz"),
  51     N_("32000 Hz"), N_("22050 Hz"), N_("24000 Hz"), N_("16000 Hz"),
  52     N_("11025 Hz"), N_("8000 Hz"), N_("4000 Hz")
  53 };
  54
  55 vlc_module_begin ()
  56     set_shortname (N_("ALSA"))
  57     set_description (N_("ALSA audio capture"))
  58     set_capability ("access_demux", 0)
  59     set_category (CAT_INPUT)
  60     set_subcategory (SUBCAT_INPUT_ACCESS)
  61     set_help (HELP_TEXT)
  62
  63     add_obsolete_string ("alsa-format") /* since 2.1.0 */
  64     add_bool ("alsa-stereo", true, STEREO_TEXT, STEREO_TEXT, true)
  65     add_integer ("alsa-samplerate", 48000, RATE_TEXT, RATE_TEXT, true)
  66         change_integer_list (rate_values, rate_names)
  67
  68     add_shortcut ("alsa")
  69     set_callbacks (Open, Close)
  70 vlc_module_end ()
  71
  72 /** Helper for ALSA -> VLC debugging output */
  73 /** XXX: duplicated from ALSA output */
  74 static void Dump (vlc_object_t *obj, const char *msg,
  75                   int (*cb)(void *, snd_output_t *), void *p)
  76 {
  77     snd_output_t *output;
  78     char *str;
  79
  80     if (unlikely(snd_output_buffer_open (&output)))
  81         return;
  82
  83     int val = cb (p, output);
  84     if (val)
  85     {
  86         msg_Warn (obj, "cannot get info: %s", snd_strerror (val));
  87         return;
  88     }
  89
  90     size_t len = snd_output_buffer_string (output, &str);
  91     if (len > 0 && str[len - 1])
  92         len--; /* strip trailing newline */
  93     msg_Dbg (obj, "%s%.*s", msg, (int)len, str);
  94     snd_output_close (output);
  95 }
  96 #define Dump(o, m, cb, p) \
  97         Dump(VLC_OBJECT(o), m, (int (*)(void *, snd_output_t *))(cb), p)
  98
  99 static void DumpDevice (vlc_object_t *obj, snd_pcm_t *pcm)
 100 {
 101     snd_pcm_info_t *info;
 102
 103     Dump (obj, " ", snd_pcm_dump, pcm);
 104     snd_pcm_info_alloca (&info);
 105     if (snd_pcm_info (pcm, info) == 0)
 106     {
 107         msg_Dbg (obj, " device name   : %s", snd_pcm_info_get_name (info));
 108         msg_Dbg (obj, " device ID     : %s", snd_pcm_info_get_id (info));
 109         msg_Dbg (obj, " subdevice name: %s",
 110                 snd_pcm_info_get_subdevice_name (info));
 111     }
 112 }
 113
 114 static void DumpDeviceStatus (vlc_object_t *obj, snd_pcm_t *pcm)
 115 {
 116     snd_pcm_status_t *status;
 117
 118     snd_pcm_status_alloca (&status);
 119     snd_pcm_status (pcm, status);
 120     Dump (obj, "current status:\n", snd_pcm_status_dump, status);
 121 }
 122 #define DumpDeviceStatus(o, p) DumpDeviceStatus(VLC_OBJECT(o), p)
 123
 124
 125 struct demux_sys_t
 126 {
 127     snd_pcm_t *pcm;
 128     es_out_id_t *es;
 129     vlc_thread_t thread;
 130
 131     mtime_t start;
 132     mtime_t caching;
 133     snd_pcm_uframes_t period_size;
 134     unsigned rate;
 135 };
 136
 137 static void Poll (snd_pcm_t *pcm, int canc)
 138 {
 139     int n = snd_pcm_poll_descriptors_count (pcm);
 140     struct pollfd ufd[n];
 141     unsigned short revents;
 142
 143     snd_pcm_poll_descriptors (pcm, ufd, n);
 144     do
 145     {
 146         vlc_restorecancel (canc);
 147         poll (ufd, n, -1);
 148         canc = vlc_savecancel ();
 149         snd_pcm_poll_descriptors_revents (pcm, ufd, n, &revents);
 150     }
 151     while (!revents);
 152 }
 153
 154 static void *Thread (void *data)
 155 {
 156     demux_t *demux = data;
 157     demux_sys_t *sys = demux->p_sys;
 158     snd_pcm_t *pcm = sys->pcm;
 159     size_t bytes;
 160     int canc, val;
 161
 162     canc = vlc_savecancel ();
 163     bytes = snd_pcm_frames_to_bytes (pcm, sys->period_size);
 164     val = snd_pcm_start (pcm);
 165     if (val)
 166     {
 167         msg_Err (demux, "cannot prepare device: %s", snd_strerror (val));
 168         return NULL;
 169     }
 170
 171     for (;;)
 172     {
 173         block_t *block = block_Alloc (bytes);
 174         if (unlikely(block == NULL))
 175             break;
 176
 177         /* Wait for data */
 178         Poll (pcm, canc);
 179
 180         /* Read data */
 181         snd_pcm_sframes_t frames, delay;
 182         mtime_t pts;
 183
 184         frames = snd_pcm_readi (pcm, block->p_buffer, sys->period_size);
 185         pts = mdate ();
 186         if (frames < 0)
 187         {
 188             if (frames == -EAGAIN)
 189                 continue;
 190
 191             val = snd_pcm_recover (pcm, frames, 1);
 192             if (val == 0)
 193             {
 194                 msg_Warn (demux, "cannot read samples: %s",
 195                           snd_strerror (frames));
 196                 continue;
 197             }
 198             msg_Err (demux, "cannot recover record stream: %s",
 199                      snd_strerror (val));
 200             DumpDeviceStatus (demux, pcm);
 201             break;
 202         }
 203
 204         /* Compute time stamp */
 205         if (snd_pcm_delay (pcm, &delay))
 206             delay = 0;
 207         delay += frames;
 208         pts -= (CLOCK_FREQ * delay) / sys->rate;
 209
 210         block->i_buffer = snd_pcm_frames_to_bytes (pcm, frames);
 211         block->i_nb_samples = frames;
 212         block->i_pts = pts;
 213         block->i_length = (CLOCK_FREQ * frames) / sys->rate;
 214
 215         es_out_Control (demux->out, ES_OUT_SET_PCR, block->i_pts);
 216         es_out_Send (demux->out, sys->es, block);
 217     }
 218     return NULL;
 219 }
 220
 221 static int Control (demux_t *demux, int query, va_list ap)
 222 {
 223     demux_sys_t *sys = demux->p_sys;
 224
 225     switch (query)
 226     {
 227         case DEMUX_GET_TIME:
 228             *va_arg (ap, int64_t *) = mdate () - sys->start;
 229             break;
 230
 231         case DEMUX_GET_PTS_DELAY:
 232             *va_arg (ap, int64_t *) = sys->caching;
 233             break;
 234
 235         //case DEMUX_SET_NEXT_DEMUX_TIME: still needed?
 236
 237         case DEMUX_HAS_UNSUPPORTED_META:
 238         case DEMUX_CAN_RECORD:
 239         case DEMUX_CAN_PAUSE:
 240         case DEMUX_CAN_CONTROL_PACE:
 241         case DEMUX_CAN_CONTROL_RATE:
 242         case DEMUX_CAN_SEEK:
 243             *va_arg (ap, bool *) = false;
 244             break;;
 245
 246         default:
 247             return VLC_EGENERIC;
 248     }
 249
 250     return VLC_SUCCESS;
 251 }
 252
 253 static const vlc_fourcc_t formats[] = {
 254     [SND_PCM_FORMAT_S8]                 = VLC_CODEC_S8,
 255     [SND_PCM_FORMAT_U8]                 = VLC_CODEC_U8,
 256     [SND_PCM_FORMAT_S16_LE]             = VLC_CODEC_S16L,
 257     [SND_PCM_FORMAT_S16_BE]             = VLC_CODEC_S16B,
 258     [SND_PCM_FORMAT_U16_LE]             = VLC_CODEC_U16L,
 259     [SND_PCM_FORMAT_U16_BE]             = VLC_CODEC_U16B,
 260     [SND_PCM_FORMAT_S24_LE]             = VLC_CODEC_S24L32,
 261     [SND_PCM_FORMAT_S24_BE]             = VLC_CODEC_S24B32,
 262     [SND_PCM_FORMAT_U24_LE]             = VLC_CODEC_U32L, // TODO: replay gain
 263     [SND_PCM_FORMAT_U24_BE]             = VLC_CODEC_U32B, // ^
 264     [SND_PCM_FORMAT_S32_LE]             = VLC_CODEC_S32L,
 265     [SND_PCM_FORMAT_S32_BE]             = VLC_CODEC_S32B,
 266     [SND_PCM_FORMAT_U32_LE]             = VLC_CODEC_U32L,
 267     [SND_PCM_FORMAT_U32_BE]             = VLC_CODEC_U32B,
 268     [SND_PCM_FORMAT_FLOAT_LE]           = VLC_CODEC_F32L,
 269     [SND_PCM_FORMAT_FLOAT_BE]           = VLC_CODEC_F32B,
 270     [SND_PCM_FORMAT_FLOAT64_LE]         = VLC_CODEC_F32L,
 271     [SND_PCM_FORMAT_FLOAT64_BE]         = VLC_CODEC_F32B,
 272   //[SND_PCM_FORMAT_IEC958_SUBFRAME_LE] = VLC_CODEC_SPDIFL,
 273   //[SND_PCM_FORMAT_IEC958_SUBFRAME_BE] = VLC_CODEC_SPDIFB,
 274     [SND_PCM_FORMAT_MU_LAW]             = VLC_CODEC_MULAW,
 275     [SND_PCM_FORMAT_A_LAW]              = VLC_CODEC_ALAW,
 276   //[SND_PCM_FORMAT_IMA_ADPCM]          = VLC_CODEC_ADPCM_?, // XXX: which one?
 277     [SND_PCM_FORMAT_MPEG]               = VLC_CODEC_MPGA,
 278     [SND_PCM_FORMAT_GSM]                = VLC_CODEC_GSM,
 279   //[SND_PCM_FORMAT_SPECIAL]            = VLC_CODEC_?
 280     [SND_PCM_FORMAT_S24_3LE]            = VLC_CODEC_S24L,
 281     [SND_PCM_FORMAT_S24_3BE]            = VLC_CODEC_S24B,
 282     [SND_PCM_FORMAT_U24_3LE]            = VLC_CODEC_U24L,
 283     [SND_PCM_FORMAT_U24_3BE]            = VLC_CODEC_U24B,
 284     [SND_PCM_FORMAT_S20_3LE]            = VLC_CODEC_S24L, // TODO: replay gain
 285     [SND_PCM_FORMAT_S20_3BE]            = VLC_CODEC_S24B, // ^
 286     [SND_PCM_FORMAT_U20_3LE]            = VLC_CODEC_U24L, // ^
 287     [SND_PCM_FORMAT_U20_3BE]            = VLC_CODEC_U24B, // ^
 288     [SND_PCM_FORMAT_S18_3LE]            = VLC_CODEC_S24L, // ^
 289     [SND_PCM_FORMAT_S18_3BE]            = VLC_CODEC_S24B, // ^
 290     [SND_PCM_FORMAT_U18_3LE]            = VLC_CODEC_U24L, // ^
 291     [SND_PCM_FORMAT_U18_3BE]            = VLC_CODEC_U24B, // ^
 292 };
 293
 294 #ifdef WORDS_BIGENDIAN
 295 # define C(f) f##BE, f##LE
 296 #else
 297 # define C(f) f##LE, f##BE
 298 #endif
 299
 300 /* Formats in order of decreasing preference */
 301 static const uint8_t choices[] = {
 302     C(SND_PCM_FORMAT_FLOAT_),
 303     C(SND_PCM_FORMAT_S32_),
 304     C(SND_PCM_FORMAT_U32_),
 305     C(SND_PCM_FORMAT_S16_),
 306     C(SND_PCM_FORMAT_U16_),
 307     C(SND_PCM_FORMAT_FLOAT64_),
 308     C(SND_PCM_FORMAT_S24_3),
 309     C(SND_PCM_FORMAT_U24_3),
 310     SND_PCM_FORMAT_MPEG,
 311     SND_PCM_FORMAT_GSM,
 312     SND_PCM_FORMAT_MU_LAW,
 313     SND_PCM_FORMAT_A_LAW,
 314     SND_PCM_FORMAT_S8,
 315     SND_PCM_FORMAT_U8,
 316 };
 317
 318 static uint16_t channel_maps[] = {
 319     AOUT_CHAN_CENTER, AOUT_CHANS_2_0, AOUT_CHANS_3_0 /* ? */,
 320     AOUT_CHANS_4_0, AOUT_CHANS_5_0 /* ? */, AOUT_CHANS_5_1,
 321     /* TODO: support 7-8 channels - need channels reodering */
 322 };
 323
 324 static int Open (vlc_object_t *obj)
 325 {
 326     demux_t *demux = (demux_t *)obj;
 327     demux_sys_t *sys = malloc (sizeof (*sys));
 328
 329     static_assert (sizeof (formats) / sizeof (formats[0]) ==
 330                    SND_PCM_FORMAT_LAST + 1, "unknown formats");
 331
 332     if (unlikely(sys == NULL))
 333         return VLC_ENOMEM;
 334
 335     /* Open the device */
 336     const char *device = demux->psz_location;
 337     if (device == NULL || !device[0])
 338         device = "default";
 339
 340     const int mode = SND_PCM_NONBLOCK
 341                  /*| SND_PCM_NO_AUTO_RESAMPLE*/
 342                    | SND_PCM_NO_AUTO_CHANNELS
 343                  /*| SND_PCM_NO_AUTO_FORMAT*/;
 344     snd_pcm_t *pcm;
 345     int val = snd_pcm_open (&pcm, device, SND_PCM_STREAM_CAPTURE, mode);
 346     if (val != 0)
 347     {
 348         msg_Err (demux, "cannot open ALSA device \"%s\": %s", device,
 349                  snd_strerror (val));
 350         free (sys);
 351         return VLC_EGENERIC;
 352     }
 353     sys->pcm = pcm;
 354     msg_Dbg (demux, "using ALSA device: %s", device);
 355     DumpDevice (VLC_OBJECT(demux), pcm);
 356
 357     /* Negotiate capture parameters */
 358     snd_pcm_hw_params_t *hw;
 359     es_format_t fmt;
 360     unsigned param;
 361     int dir;
 362
 363     snd_pcm_hw_params_alloca (&hw);
 364     snd_pcm_hw_params_any (pcm, hw);
 365     Dump (demux, "initial hardware setup:\n", snd_pcm_hw_params_dump, hw);
 366
 367     val = snd_pcm_hw_params_set_rate_resample (pcm, hw, 0);
 368     if (val)
 369     {
 370         msg_Err (demux, "cannot disable resampling: %s", snd_strerror (val));
 371         goto error;
 372     }
 373
 374     val = snd_pcm_hw_params_set_access (pcm, hw,
 375                                         SND_PCM_ACCESS_RW_INTERLEAVED);
 376     if (val)
 377     {
 378         msg_Err (demux, "cannot set access mode: %s", snd_strerror (val));
 379         goto error;
 380     }
 381
 382     snd_pcm_format_t format = SND_PCM_FORMAT_UNKNOWN;
 383     for (size_t i = 0; i < sizeof (choices) / sizeof (choices[0]); i++)
 384         if (snd_pcm_hw_params_test_format (pcm, hw, choices[i]) == 0)
 385         {
 386             val = snd_pcm_hw_params_set_format (pcm, hw, choices[i]);
 387             if (val)
 388             {
 389                 msg_Err (demux, "cannot set sample format: %s",
 390                          snd_strerror (val));
 391                 goto error;
 392             }
 393             format = choices[i];
 394             break;
 395         }
 396
 397     if (format == SND_PCM_FORMAT_UNKNOWN)
 398     {
 399         msg_Err (demux, "no supported sample format");
 400         goto error;
 401     }
 402
 403     assert ((size_t)format < (sizeof (formats) / sizeof (formats[0])));
 404     es_format_Init (&fmt, AUDIO_ES, formats[format]);
 405     fmt.audio.i_format = fmt.i_codec;
 406
 407     param = 1 + var_InheritBool (demux, "alsa-stereo");
 408     val = snd_pcm_hw_params_set_channels_max (pcm, hw, &param);
 409     if (val)
 410     {
 411         msg_Err (demux, "cannot restrict channels count: %s",
 412                  snd_strerror (val));
 413         goto error;
 414     }
 415     val = snd_pcm_hw_params_set_channels_last (pcm, hw, &param);
 416     if (val)
 417     {
 418         msg_Err (demux, "cannot set channels count: %s", snd_strerror (val));
 419         goto error;
 420     }
 421     assert (param > 0);
 422     assert (param < (sizeof (channel_maps) / sizeof (channel_maps[0])));
 423     fmt.audio.i_channels = param;
 424     fmt.audio.i_original_channels =
 425     fmt.audio.i_physical_channels = channel_maps[param - 1];
 426
 427     param = var_InheritInteger (demux, "alsa-samplerate");
 428     val = snd_pcm_hw_params_set_rate_max (pcm, hw, &param, NULL);
 429     if (val)
 430     {
 431         msg_Err (demux, "cannot restrict rate to %u Hz or less: %s", 192000,
 432                  snd_strerror (val));
 433         goto error;
 434     }
 435     val = snd_pcm_hw_params_set_rate_last (pcm, hw, &param, &dir);
 436     if (val)
 437     {
 438         msg_Err (demux, "cannot set sample rate: %s", snd_strerror (val));
 439         goto error;
 440     }
 441     if (dir)
 442         msg_Warn (demux, "sample rate is not integral");
 443     fmt.audio.i_rate = param;
 444     sys->rate = param;
 445
 446     sys->start = mdate ();
 447     sys->caching = INT64_C(1000) * var_InheritInteger (demux, "live-caching");
 448     param = sys->caching;
 449     val = snd_pcm_hw_params_set_buffer_time_near (pcm, hw, &param, NULL);
 450     if (val)
 451     {
 452         msg_Err (demux, "cannot set buffer duration: %s", snd_strerror (val));
 453         goto error;
 454     }
 455
 456     param /= 4;
 457     val = snd_pcm_hw_params_set_period_time_near (pcm, hw, &param, NULL);
 458     if (val)
 459     {
 460         msg_Err (demux, "cannot set period: %s", snd_strerror (val));
 461         goto error;
 462     }
 463
 464     val = snd_pcm_hw_params_get_period_size (hw, &sys->period_size, &dir);
 465     if (val)
 466     {
 467         msg_Err (demux, "cannot get period size: %s", snd_strerror (val));
 468         goto error;
 469     }
 470     if (dir > 0)
 471         sys->period_size++;
 472
 473     /* Commit hardware parameters */
 474     val = snd_pcm_hw_params (pcm, hw);
 475     if (val)
 476     {
 477         msg_Err (demux, "cannot commit hardware parameters: %s",
 478                  snd_strerror (val));
 479         goto error;
 480     }
 481     Dump (demux, "final HW setup:\n", snd_pcm_hw_params_dump, hw);
 482
 483     /* Kick recording */
 484     aout_FormatPrepare (&fmt.audio);
 485     sys->es = es_out_Add (demux->out, &fmt);
 486
 487     if (vlc_clone (&sys->thread, Thread, demux, VLC_THREAD_PRIORITY_INPUT))
 488     {
 489         es_out_Del (demux->out, sys->es);
 490         goto error;
 491     }
 492
 493     demux->p_sys = sys;
 494     demux->pf_demux = NULL;
 495     demux->pf_control = Control;
 496     return VLC_SUCCESS;
 497 error:
 498     snd_pcm_close (pcm);
 499     free (sys);
 500     return VLC_EGENERIC;
 501 }
 502
 503 static void Close (vlc_object_t *obj)
 504 {
 505     demux_t *demux = (demux_t *)obj;
 506     demux_sys_t *sys = demux->p_sys;
 507
 508     vlc_cancel (sys->thread);
 509     vlc_join (sys->thread, NULL);
 510
 511     snd_pcm_close (sys->pcm);
 512     free (sys);
 513 }