modules/access/pulse.c

   1 /**
   2  * \file pulse.c
   3  * \brief PulseAudio input plugin for vlc
   4  */
   5 /*****************************************************************************
   6  * Copyright (C) 2011 Rémi Denis-Courmont
   7  *
   8  * This program is free software; you can redistribute it and/or modify it
   9  * under the terms of the GNU Lesser General Public License as published by
  10  * the Free Software Foundation; either version 2.1 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16  * GNU Lesser General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU Lesser General Public License
  19  * along with this program; if not, write to the Free Software Foundation,
  20  * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  21  *****************************************************************************/
  22
  23 #ifdef HAVE_CONFIG_H
  24 # include "config.h"
  25 #endif
  26
  27 #include <vlc_common.h>
  28 #include <vlc_aout.h>
  29 #include <vlc_demux.h>
  30 #include <vlc_plugin.h>
  31 #include <pulse/pulseaudio.h>
  32 #include "audio_output/vlcpulse.h"
  33
  34 #define HELP_TEXT N_( \
  35     "Pass pulse:// to open the default PulseAudio source, " \
  36     "or pulse://SOURCE to open a specific source named SOURCE.")
  37
  38 static int Open(vlc_object_t *);
  39 static void Close(vlc_object_t *);
  40
  41 vlc_module_begin ()
  42     set_shortname (N_("PulseAudio"))
  43     set_description (N_("PulseAudio input"))
  44     set_capability ("access_demux", 0)
  45     set_category (CAT_INPUT)
  46     set_subcategory (SUBCAT_INPUT_ACCESS)
  47     set_help (HELP_TEXT)
  48
  49     add_shortcut ("pulse", "pulseaudio", "pa")
  50     set_callbacks (Open, Close)
  51 vlc_module_end ()
  52
  53 struct demux_sys_t
  54 {
  55     pa_stream *stream; /**< PulseAudio playback stream object */
  56     pa_context *context; /**< PulseAudio connection context */
  57     pa_threaded_mainloop *mainloop; /**< PulseAudio thread */
  58
  59     es_out_id_t *es;
  60     bool discontinuity; /**< The next block will not follow the last one */
  61     unsigned framesize; /**< Byte size of a sample */
  62     mtime_t caching; /**< Caching value */
  63 };
  64
  65 /* Stream helpers */
  66 static void stream_state_cb(pa_stream *s, void *userdata)
  67 {
  68     pa_threaded_mainloop *mainloop = userdata;
  69
  70     switch (pa_stream_get_state(s)) {
  71         case PA_STREAM_READY:
  72         case PA_STREAM_FAILED:
  73         case PA_STREAM_TERMINATED:
  74             pa_threaded_mainloop_signal(mainloop, 0);
  75         default:
  76             break;
  77     }
  78 }
  79
  80 static void stream_success_cb (pa_stream *s, int success, void *userdata)
  81 {
  82     pa_threaded_mainloop *mainloop = userdata;
  83
  84     pa_threaded_mainloop_signal(mainloop, 0);
  85     (void) s;
  86     (void) success;
  87 }
  88
  89 static void stream_buffer_attr_cb(pa_stream *s, void *userdata)
  90 {
  91     demux_t *demux = userdata;
  92     const struct pa_buffer_attr *pba = pa_stream_get_buffer_attr(s);
  93
  94     msg_Dbg(demux, "using buffer metrics: maxlength=%"PRIu32", "
  95             "fragsize=%"PRIu32, pba->maxlength, pba->fragsize);
  96 }
  97
  98 static void stream_moved_cb(pa_stream *s, void *userdata)
  99 {
 100     demux_t *demux = userdata;
 101     uint32_t idx = pa_stream_get_device_index(s);
 102
 103     msg_Dbg(demux, "connected to source %"PRIu32": %s", idx,
 104                   pa_stream_get_device_name(s));
 105     stream_buffer_attr_cb(s, userdata);
 106 }
 107
 108 static void stream_overflow_cb(pa_stream *s, void *userdata)
 109 {
 110     demux_t *demux = userdata;
 111
 112     msg_Err(demux, "overflow");
 113     (void) s;
 114 }
 115
 116 static void stream_started_cb(pa_stream *s, void *userdata)
 117 {
 118     demux_t *demux = userdata;
 119
 120     msg_Dbg(demux, "started");
 121     (void) s;
 122 }
 123
 124 static void stream_suspended_cb(pa_stream *s, void *userdata)
 125 {
 126     demux_t *demux = userdata;
 127
 128     msg_Dbg(demux, "suspended");
 129     (void) s;
 130 }
 131
 132 static void stream_underflow_cb(pa_stream *s, void *userdata)
 133 {
 134     demux_t *demux = userdata;
 135
 136     msg_Dbg(demux, "underflow");
 137     (void) s;
 138 }
 139
 140 static int stream_wait(pa_stream *stream, pa_threaded_mainloop *mainloop)
 141 {
 142     pa_stream_state_t state;
 143
 144     while ((state = pa_stream_get_state(stream)) != PA_STREAM_READY) {
 145         if (state == PA_STREAM_FAILED || state == PA_STREAM_TERMINATED)
 146             return -1;
 147         pa_threaded_mainloop_wait(mainloop);
 148     }
 149     return 0;
 150 }
 151
 152 static void stream_read_cb(pa_stream *s, size_t length, void *userdata)
 153 {
 154     demux_t *demux = userdata;
 155     demux_sys_t *sys = demux->p_sys;
 156     const void *ptr;
 157     unsigned samples = length / sys->framesize;
 158
 159     if (pa_stream_peek(s, &ptr, &length) < 0) {
 160         vlc_pa_error(demux, "cannot peek stream", sys->context);
 161         return;
 162     }
 163
 164     mtime_t pts = mdate();
 165     pa_usec_t latency;
 166     int negative;
 167
 168     if (pa_stream_get_latency(s, &latency, &negative) < 0) {
 169         vlc_pa_error(demux, "cannot determine latency", sys->context);
 170         return;
 171     }
 172     if (negative)
 173         pts += latency;
 174     else
 175         pts -= latency;
 176
 177     es_out_Control(demux->out, ES_OUT_SET_PCR, pts);
 178     if (unlikely(sys->es == NULL))
 179         goto race;
 180
 181     block_t *block = block_Alloc(length);
 182     if (likely(block != NULL)) {
 183         memcpy(block->p_buffer, ptr, length);
 184         block->i_nb_samples = samples;
 185         block->i_dts = block->i_pts = pts;
 186         if (sys->discontinuity) {
 187             block->i_flags |= BLOCK_FLAG_DISCONTINUITY;
 188             sys->discontinuity = false;
 189         }
 190
 191         es_out_Send(demux->out, sys->es, block);
 192     } else
 193         sys->discontinuity = true;
 194 race:
 195     pa_stream_drop(s);
 196 }
 197
 198 static int Control(demux_t *demux, int query, va_list ap)
 199 {
 200     demux_sys_t *sys = demux->p_sys;
 201
 202     switch (query)
 203     {
 204         case DEMUX_GET_TIME:
 205         {
 206             pa_usec_t us;
 207
 208             if (pa_stream_get_time(sys->stream, &us) < 0)
 209                 return VLC_EGENERIC;
 210             *(va_arg(ap, int64_t *)) = us;
 211             break;
 212         }
 213
 214         //case DEMUX_SET_NEXT_DEMUX_TIME: TODO
 215         //case DEMUX_GET_META TODO
 216
 217         case DEMUX_GET_PTS_DELAY:
 218             *(va_arg(ap, int64_t *)) = sys->caching;
 219             break;
 220
 221         case DEMUX_HAS_UNSUPPORTED_META:
 222         case DEMUX_CAN_RECORD:
 223         case DEMUX_CAN_PAUSE:
 224         case DEMUX_CAN_CONTROL_PACE:
 225         case DEMUX_CAN_CONTROL_RATE:
 226         case DEMUX_CAN_SEEK:
 227             *(va_arg(ap, bool *)) = false;
 228             break;
 229
 230         default:
 231             return VLC_EGENERIC;
 232     }
 233
 234     return VLC_SUCCESS;
 235 }
 236
 237 /** PulseAudio sample (PCM) format to VLC codec */
 238 static const vlc_fourcc_t fourccs[] = {
 239     [PA_SAMPLE_U8] =        VLC_CODEC_U8,
 240     [PA_SAMPLE_ALAW] =      VLC_CODEC_ALAW,
 241     [PA_SAMPLE_ULAW] =      VLC_CODEC_MULAW,
 242     [PA_SAMPLE_S16LE] =     VLC_CODEC_S16L,
 243     [PA_SAMPLE_S16BE] =     VLC_CODEC_S16B,
 244     [PA_SAMPLE_FLOAT32LE] = VLC_CODEC_F32L,
 245     [PA_SAMPLE_FLOAT32BE] = VLC_CODEC_F32B,
 246     [PA_SAMPLE_S32LE] =     VLC_CODEC_S32L,
 247     [PA_SAMPLE_S32BE] =     VLC_CODEC_S32B,
 248     [PA_SAMPLE_S24LE] =     VLC_CODEC_S24L,
 249     [PA_SAMPLE_S24BE] =     VLC_CODEC_S24B,
 250     [PA_SAMPLE_S24_32LE] =  VLC_CODEC_S24L32,
 251     [PA_SAMPLE_S24_32BE] =  VLC_CODEC_S24B32,
 252 };
 253
 254 static int Open(vlc_object_t *obj)
 255 {
 256     demux_t *demux = (demux_t *)obj;
 257
 258     demux_sys_t *sys = vlc_malloc(obj, sizeof (*sys));
 259     if (unlikely(sys == NULL))
 260         return VLC_ENOMEM;
 261
 262     sys->context = vlc_pa_connect(obj, &sys->mainloop);
 263     if (sys->context == NULL)
 264         return VLC_EGENERIC;
 265
 266     sys->stream = NULL;
 267     sys->es = NULL;
 268     sys->discontinuity = false;
 269     sys->caching = INT64_C(1000) * var_InheritInteger(obj, "live-caching");
 270     demux->p_sys = sys;
 271
 272     /* Stream parameters */
 273     struct pa_sample_spec ss;
 274     ss.format = PA_SAMPLE_S16NE;
 275     ss.rate = 48000;
 276     ss.channels = 2;
 277     assert(pa_sample_spec_valid(&ss));
 278
 279     struct pa_channel_map map;
 280     map.channels = 2;
 281     map.map[0] = PA_CHANNEL_POSITION_FRONT_LEFT;
 282     map.map[1] = PA_CHANNEL_POSITION_FRONT_RIGHT;
 283     assert(pa_channel_map_valid(&map));
 284
 285     const pa_stream_flags_t flags = PA_STREAM_INTERPOLATE_TIMING
 286                                   | PA_STREAM_AUTO_TIMING_UPDATE
 287                                   | PA_STREAM_ADJUST_LATENCY
 288                                   | PA_STREAM_FIX_FORMAT
 289                                   | PA_STREAM_FIX_RATE
 290                                   /*| PA_STREAM_FIX_CHANNELS*/;
 291
 292     const char *dev = NULL;
 293     if (demux->psz_location != NULL && demux->psz_location[0] != '\0')
 294         dev = demux->psz_location;
 295
 296     struct pa_buffer_attr attr = {
 297         .maxlength = -1,
 298         .fragsize = pa_usec_to_bytes(sys->caching, &ss) / 2,
 299     };
 300
 301     es_format_t fmt;
 302
 303     /* Create record stream */
 304     pa_stream *s;
 305     pa_operation *op;
 306
 307     pa_threaded_mainloop_lock(sys->mainloop);
 308     s = pa_stream_new(sys->context, "audio stream", &ss, &map);
 309     if (s == NULL)
 310         goto error;
 311
 312     sys->stream = s;
 313     pa_stream_set_state_callback(s, stream_state_cb, sys->mainloop);
 314     pa_stream_set_read_callback(s, stream_read_cb, demux);
 315     pa_stream_set_buffer_attr_callback(s, stream_buffer_attr_cb, demux);
 316     pa_stream_set_moved_callback(s, stream_moved_cb, demux);
 317     pa_stream_set_overflow_callback(s, stream_overflow_cb, demux);
 318     pa_stream_set_started_callback(s, stream_started_cb, demux);
 319     pa_stream_set_suspended_callback(s, stream_suspended_cb, demux);
 320     pa_stream_set_underflow_callback(s, stream_underflow_cb, demux);
 321
 322     if (pa_stream_connect_record(s, dev, &attr, flags) < 0
 323      || stream_wait(s, sys->mainloop)) {
 324         vlc_pa_error(obj, "cannot connect record stream", sys->context);
 325         goto error;
 326     }
 327
 328     /* The ES should be initialized before stream_read_cb(), but how? */
 329     const struct pa_sample_spec *pss = pa_stream_get_sample_spec(s);
 330     if ((unsigned)pss->format >= sizeof (fourccs) / sizeof (fourccs[0])) {
 331         msg_Err(obj, "unknown PulseAudio sample format %u",
 332                 (unsigned)pss->format);
 333         goto error;
 334     }
 335
 336     vlc_fourcc_t format = fourccs[pss->format];
 337     if (format == 0) { /* FIXME: should renegotiate something else */
 338         msg_Err(obj, "unsupported PulseAudio sample format %u",
 339                 (unsigned)pss->format);
 340         goto error;
 341     }
 342
 343     es_format_Init(&fmt, AUDIO_ES, format);
 344     fmt.audio.i_physical_channels = fmt.audio.i_original_channels =
 345         AOUT_CHAN_LEFT | AOUT_CHAN_RIGHT;
 346     fmt.audio.i_channels = ss.channels;
 347     fmt.audio.i_rate = pss->rate;
 348     fmt.audio.i_bitspersample = aout_BitsPerSample(format);
 349     fmt.audio.i_blockalign = fmt.audio.i_bitspersample * ss.channels / 8;
 350     fmt.i_bitrate = fmt.audio.i_bitspersample * ss.channels * pss->rate;
 351     sys->framesize = fmt.audio.i_blockalign;
 352     sys->es = es_out_Add (demux->out, &fmt);
 353
 354     /* Update the buffer attributes according to actual format */
 355     attr.fragsize = pa_usec_to_bytes(sys->caching, pss) / 2;
 356     op = pa_stream_set_buffer_attr(s, &attr, stream_success_cb, sys->mainloop);
 357     if (likely(op != NULL)) {
 358         while (pa_operation_get_state(op) == PA_OPERATION_RUNNING)
 359             pa_threaded_mainloop_wait(sys->mainloop);
 360         pa_operation_unref(op);
 361     }
 362     stream_buffer_attr_cb(s, demux);
 363     pa_threaded_mainloop_unlock(sys->mainloop);
 364
 365     demux->pf_demux = NULL;
 366     demux->pf_control = Control;
 367     return VLC_SUCCESS;
 368
 369 error:
 370     pa_threaded_mainloop_unlock(sys->mainloop);
 371     Close(obj);
 372     return VLC_EGENERIC;
 373 }
 374
 375 static void Close (vlc_object_t *obj)
 376 {
 377     demux_t *demux = (demux_t *)obj;
 378     demux_sys_t *sys = demux->p_sys;
 379     pa_stream *s = sys->stream;
 380
 381     if (likely(s != NULL)) {
 382         pa_threaded_mainloop_lock(sys->mainloop);
 383         pa_stream_disconnect(s);
 384         pa_stream_set_state_callback(s, NULL, NULL);
 385         pa_stream_set_read_callback(s, NULL, NULL);
 386         pa_stream_set_buffer_attr_callback(s, NULL, NULL);
 387         pa_stream_set_moved_callback(s, NULL, NULL);
 388         pa_stream_set_overflow_callback(s, NULL, NULL);
 389         pa_stream_set_started_callback(s, NULL, NULL);
 390         pa_stream_set_suspended_callback(s, NULL, NULL);
 391         pa_stream_set_underflow_callback(s, NULL, NULL);
 392         pa_stream_unref(s);
 393         pa_threaded_mainloop_unlock(sys->mainloop);
 394     }
 395
 396     vlc_pa_disconnect(obj, sys->context, sys->mainloop);
 397 }