code.delx.au - pulseaudio/blob - src/pulsecore/sound-file-stream.c

   1 /***
   2   This file is part of PulseAudio.
   3
   4   Copyright 2004-2008 Lennart Poettering
   5
   6   PulseAudio is free software; you can redistribute it and/or modify
   7   it under the terms of the GNU Lesser General Public License as published
   8   by the Free Software Foundation; either version 2 of the License,
   9   or (at your option) any later version.
  10
  11   PulseAudio is distributed in the hope that it will be useful, but
  12   WITHOUT ANY WARRANTY; without even the implied warranty of
  13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14   General Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public License
  17   along with PulseAudio; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <stdlib.h>
  27 #include <stdio.h>
  28 #include <string.h>
  29 #include <unistd.h>
  30 #include <fcntl.h>
  31 #include <errno.h>
  32
  33 #include <sndfile.h>
  34
  35 #include <pulse/xmalloc.h>
  36 #include <pulse/util.h>
  37
  38 #include <pulsecore/core-error.h>
  39 #include <pulsecore/sink-input.h>
  40 #include <pulsecore/log.h>
  41 #include <pulsecore/thread-mq.h>
  42 #include <pulsecore/core-util.h>
  43 #include <pulsecore/sample-util.h>
  44
  45 #include "sound-file-stream.h"
  46
  47 #define MEMBLOCKQ_MAXLENGTH (16*1024*1024)
  48
  49 typedef struct file_stream {
  50     pa_msgobject parent;
  51     pa_core *core;
  52     pa_sink_input *sink_input;
  53
  54     SNDFILE *sndfile;
  55     sf_count_t (*readf_function)(SNDFILE *sndfile, void *ptr, sf_count_t frames);
  56
  57     /* We need this memblockq here to easily fulfill rewind requests
  58      * (even beyond the file start!) */
  59     pa_memblockq *memblockq;
  60 } file_stream;
  61
  62 enum {
  63     FILE_STREAM_MESSAGE_UNLINK
  64 };
  65
  66 PA_DECLARE_CLASS(file_stream);
  67 #define FILE_STREAM(o) (file_stream_cast(o))
  68 static PA_DEFINE_CHECK_TYPE(file_stream, pa_msgobject);
  69
  70 /* Called from main context */
  71 static void file_stream_unlink(file_stream *u) {
  72     pa_assert(u);
  73
  74     if (!u->sink_input)
  75         return;
  76
  77     pa_sink_input_unlink(u->sink_input);
  78     pa_sink_input_unref(u->sink_input);
  79     u->sink_input = NULL;
  80
  81     /* Make sure we don't decrease the ref count twice. */
  82     file_stream_unref(u);
  83 }
  84
  85 /* Called from main context */
  86 static void file_stream_free(pa_object *o) {
  87     file_stream *u = FILE_STREAM(o);
  88     pa_assert(u);
  89
  90     if (u->memblockq)
  91         pa_memblockq_free(u->memblockq);
  92
  93     if (u->sndfile)
  94         sf_close(u->sndfile);
  95
  96     pa_xfree(u);
  97 }
  98
  99 /* Called from main context */
 100 static int file_stream_process_msg(pa_msgobject *o, int code, void*userdata, int64_t offset, pa_memchunk *chunk) {
 101     file_stream *u = FILE_STREAM(o);
 102     file_stream_assert_ref(u);
 103
 104     switch (code) {
 105         case FILE_STREAM_MESSAGE_UNLINK:
 106             file_stream_unlink(u);
 107             break;
 108     }
 109
 110     return 0;
 111 }
 112
 113 /* Called from main context */
 114 static void sink_input_kill_cb(pa_sink_input *i) {
 115     file_stream *u;
 116
 117     pa_sink_input_assert_ref(i);
 118     u = FILE_STREAM(i->userdata);
 119     file_stream_assert_ref(u);
 120
 121     file_stream_unlink(u);
 122 }
 123
 124 /* Called from IO thread context */
 125 static void sink_input_state_change_cb(pa_sink_input *i, pa_sink_input_state_t state) {
 126     file_stream *u;
 127
 128     pa_sink_input_assert_ref(i);
 129     u = FILE_STREAM(i->userdata);
 130     file_stream_assert_ref(u);
 131
 132     /* If we are added for the first time, ask for a rewinding so that
 133      * we are heard right-away. */
 134     if (PA_SINK_INPUT_IS_LINKED(state) &&
 135         i->thread_info.state == PA_SINK_INPUT_INIT)
 136         pa_sink_input_request_rewind(i, 0, FALSE, TRUE);
 137 }
 138
 139 /* Called from IO thread context */
 140 static int sink_input_pop_cb(pa_sink_input *i, size_t length, pa_memchunk *chunk) {
 141     file_stream *u;
 142
 143     pa_sink_input_assert_ref(i);
 144     pa_assert(chunk);
 145     u = FILE_STREAM(i->userdata);
 146     file_stream_assert_ref(u);
 147
 148     if (!u->memblockq)
 149         return -1;
 150
 151     for (;;) {
 152         pa_memchunk tchunk;
 153         size_t fs;
 154         void *p;
 155         sf_count_t n;
 156
 157         if (pa_memblockq_peek(u->memblockq, chunk) >= 0) {
 158             chunk->length = PA_MIN(chunk->length, length);
 159             pa_memblockq_drop(u->memblockq, chunk->length);
 160             return 0;
 161         }
 162
 163         if (!u->sndfile)
 164             break;
 165
 166         tchunk.memblock = pa_memblock_new(i->sink->core->mempool, length);
 167         tchunk.index = 0;
 168
 169         p = pa_memblock_acquire(tchunk.memblock);
 170
 171         if (u->readf_function) {
 172             fs = pa_frame_size(&i->sample_spec);
 173             n = u->readf_function(u->sndfile, p, length/fs);
 174         } else {
 175             fs = 1;
 176             n = sf_read_raw(u->sndfile, p, length);
 177         }
 178
 179         pa_memblock_release(tchunk.memblock);
 180
 181         if (n <= 0) {
 182             pa_memblock_unref(tchunk.memblock);
 183
 184             sf_close(u->sndfile);
 185             u->sndfile = NULL;
 186             break;
 187         }
 188
 189         tchunk.length = n * fs;
 190
 191         pa_memblockq_push(u->memblockq, &tchunk);
 192         pa_memblock_unref(tchunk.memblock);
 193     }
 194
 195     if (pa_sink_input_safe_to_remove(i)) {
 196         pa_memblockq_free(u->memblockq);
 197         u->memblockq = NULL;
 198
 199         pa_asyncmsgq_post(pa_thread_mq_get()->outq, PA_MSGOBJECT(u), FILE_STREAM_MESSAGE_UNLINK, NULL, 0, NULL, NULL);
 200     }
 201
 202     return -1;
 203  }
 204
 205 static void sink_input_process_rewind_cb(pa_sink_input *i, size_t nbytes) {
 206     file_stream *u;
 207
 208     pa_sink_input_assert_ref(i);
 209     pa_assert(nbytes > 0);
 210     u = FILE_STREAM(i->userdata);
 211     file_stream_assert_ref(u);
 212
 213     pa_log("backwards %lu", (unsigned long) nbytes);
 214
 215     if (!u->memblockq)
 216         return;
 217
 218     pa_memblockq_rewind(u->memblockq, nbytes);
 219 }
 220
 221 static void sink_input_update_max_rewind_cb(pa_sink_input *i, size_t nbytes) {
 222     file_stream *u;
 223
 224     pa_sink_input_assert_ref(i);
 225     u = FILE_STREAM(i->userdata);
 226     file_stream_assert_ref(u);
 227
 228     if (!u->memblockq)
 229         return;
 230
 231     pa_memblockq_set_maxrewind(u->memblockq, nbytes);
 232 }
 233
 234 int pa_play_file(
 235         pa_sink *sink,
 236         const char *fname,
 237         const pa_cvolume *volume) {
 238
 239     file_stream *u = NULL;
 240     SF_INFO sfinfo;
 241     pa_sample_spec ss;
 242     pa_sink_input_new_data data;
 243     int fd;
 244
 245     pa_assert(sink);
 246     pa_assert(fname);
 247
 248     u = pa_msgobject_new(file_stream);
 249     u->parent.parent.free = file_stream_free;
 250     u->parent.process_msg = file_stream_process_msg;
 251     u->core = sink->core;
 252     u->sink_input = NULL;
 253     u->sndfile = NULL;
 254     u->readf_function = NULL;
 255     u->memblockq = NULL;
 256
 257     memset(&sfinfo, 0, sizeof(sfinfo));
 258
 259     if ((fd = open(fname, O_RDONLY
 260 #ifdef O_NOCTTY
 261                    |O_NOCTTY
 262 #endif
 263                    )) < 0) {
 264         pa_log("Failed to open file %s: %s", fname, pa_cstrerror(errno));
 265         goto fail;
 266     }
 267
 268     /* FIXME: For now we just use posix_fadvise to avoid page faults
 269      * when accessing the file data. Eventually we should move the
 270      * file reader into the main event loop and pass the data over the
 271      * asyncmsgq. */
 272
 273 #ifdef HAVE_POSIX_FADVISE
 274     if (posix_fadvise(fd, 0, 0, POSIX_FADV_SEQUENTIAL) < 0) {
 275         pa_log_warn("POSIX_FADV_SEQUENTIAL failed: %s", pa_cstrerror(errno));
 276         goto fail;
 277     } else
 278         pa_log_debug("POSIX_FADV_SEQUENTIAL succeeded.");
 279
 280     if (posix_fadvise(fd, 0, 0, POSIX_FADV_WILLNEED) < 0) {
 281         pa_log_warn("POSIX_FADV_WILLNEED failed: %s", pa_cstrerror(errno));
 282         goto fail;
 283     } else
 284         pa_log_debug("POSIX_FADV_WILLNEED succeeded.");
 285 #endif
 286
 287     if (!(u->sndfile = sf_open_fd(fd, SFM_READ, &sfinfo, 1))) {
 288         pa_log("Failed to open file %s", fname);
 289         pa_close(fd);
 290         goto fail;
 291     }
 292
 293     switch (sfinfo.format & 0xFF) {
 294         case SF_FORMAT_PCM_16:
 295         case SF_FORMAT_PCM_U8:
 296         case SF_FORMAT_PCM_S8:
 297             ss.format = PA_SAMPLE_S16NE;
 298             u->readf_function = (sf_count_t (*)(SNDFILE *sndfile, void *ptr, sf_count_t frames)) sf_readf_short;
 299             break;
 300
 301         case SF_FORMAT_ULAW:
 302             ss.format = PA_SAMPLE_ULAW;
 303             break;
 304
 305         case SF_FORMAT_ALAW:
 306             ss.format = PA_SAMPLE_ALAW;
 307             break;
 308
 309         case SF_FORMAT_FLOAT:
 310         default:
 311             ss.format = PA_SAMPLE_FLOAT32NE;
 312             u->readf_function = (sf_count_t (*)(SNDFILE *sndfile, void *ptr, sf_count_t frames)) sf_readf_float;
 313             break;
 314     }
 315
 316     ss.rate = sfinfo.samplerate;
 317     ss.channels = sfinfo.channels;
 318
 319     if (!pa_sample_spec_valid(&ss)) {
 320         pa_log("Unsupported sample format in file %s", fname);
 321         goto fail;
 322     }
 323
 324     pa_sink_input_new_data_init(&data);
 325     data.sink = sink;
 326     data.driver = __FILE__;
 327     pa_sink_input_new_data_set_sample_spec(&data, &ss);
 328     pa_sink_input_new_data_set_volume(&data, volume);
 329     pa_proplist_sets(data.proplist, PA_PROP_MEDIA_NAME, pa_path_get_filename(fname));
 330     pa_proplist_sets(data.proplist, PA_PROP_MEDIA_FILENAME, fname);
 331
 332     u->sink_input = pa_sink_input_new(sink->core, &data, 0);
 333     pa_sink_input_new_data_done(&data);
 334
 335     if (!u->sink_input)
 336         goto fail;
 337
 338     u->sink_input->pop = sink_input_pop_cb;
 339     u->sink_input->process_rewind = sink_input_process_rewind_cb;
 340     u->sink_input->update_max_rewind = sink_input_update_max_rewind_cb;
 341     u->sink_input->kill = sink_input_kill_cb;
 342     u->sink_input->state_change = sink_input_state_change_cb;
 343     u->sink_input->userdata = u;
 344
 345     u->memblockq = pa_memblockq_new(0, MEMBLOCKQ_MAXLENGTH, 0, pa_frame_size(&ss), 1, 1, 0, NULL);
 346
 347     pa_sink_input_put(u->sink_input);
 348
 349     /* The reference to u is dangling here, because we want to keep
 350      * this stream around until it is fully played. */
 351
 352     return 0;
 353
 354 fail:
 355     if (u)
 356         file_stream_unref(u);
 357
 358     return -1;
 359 }