src/polyp/simple.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of polypaudio.
   5
   6   polypaudio 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   polypaudio 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 polypaudio; 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 <stdio.h>
  27 #include <string.h>
  28 #include <assert.h>
  29 #include <stdlib.h>
  30
  31 #include "simple.h"
  32 #include "polypaudio.h"
  33 #include "mainloop.h"
  34 #include <polypcore/native-common.h>
  35 #include <polypcore/xmalloc.h>
  36 #include <polypcore/log.h>
  37
  38 struct pa_simple {
  39     pa_mainloop *mainloop;
  40     pa_context *context;
  41     pa_stream *stream;
  42     pa_stream_direction_t direction;
  43
  44     int dead;
  45
  46     void *read_data;
  47     size_t read_index, read_length;
  48     pa_usec_t latency;
  49 };
  50
  51 static void read_callback(pa_stream *s, const void*data, size_t length, void *userdata);
  52
  53 static int check_error(pa_simple *p, int *rerror) {
  54     pa_context_state_t cst;
  55     pa_stream_state_t sst;
  56     assert(p);
  57
  58     if ((cst = pa_context_get_state(p->context)) == PA_CONTEXT_FAILED)
  59         goto fail;
  60
  61     assert(cst != PA_CONTEXT_TERMINATED);
  62
  63     if (p->stream) {
  64         if ((sst = pa_stream_get_state(p->stream)) == PA_STREAM_FAILED)
  65             goto fail;
  66
  67         assert(sst != PA_STREAM_TERMINATED);
  68     }
  69
  70     return 0;
  71
  72 fail:
  73     if (rerror)
  74         *rerror = pa_context_errno(p->context);
  75
  76     p->dead = 1;
  77
  78     return -1;
  79 }
  80
  81 static int iterate(pa_simple *p, int block, int *rerror) {
  82     assert(p && p->context && p->mainloop);
  83
  84     if (check_error(p, rerror) < 0)
  85         return -1;
  86
  87     if (!block && !pa_context_is_pending(p->context))
  88         return 0;
  89
  90     do {
  91         if (pa_mainloop_iterate(p->mainloop, 1, NULL) < 0) {
  92             if (rerror)
  93                 *rerror = PA_ERROR_INTERNAL;
  94             return -1;
  95         }
  96
  97         if (check_error(p, rerror) < 0)
  98             return -1;
  99
 100     } while (pa_context_is_pending(p->context));
 101
 102
 103     while (pa_mainloop_deferred_pending(p->mainloop)) {
 104
 105         if (pa_mainloop_iterate(p->mainloop, 0, NULL) < 0) {
 106             if (rerror)
 107                 *rerror = PA_ERROR_INTERNAL;
 108             return -1;
 109         }
 110
 111         if (check_error(p, rerror) < 0)
 112             return -1;
 113     }
 114
 115     return 0;
 116 }
 117
 118 pa_simple* pa_simple_new(
 119     const char *server,
 120     const char *name,
 121     pa_stream_direction_t dir,
 122     const char *dev,
 123     const char *stream_name,
 124     const pa_sample_spec *ss,
 125     const pa_buffer_attr *attr,
 126     int *rerror) {
 127
 128     pa_simple *p;
 129     int error = PA_ERROR_INTERNAL;
 130     assert(ss && (dir == PA_STREAM_PLAYBACK || dir == PA_STREAM_RECORD));
 131
 132     p = pa_xmalloc(sizeof(pa_simple));
 133     p->context = NULL;
 134     p->stream = NULL;
 135     p->mainloop = pa_mainloop_new();
 136     assert(p->mainloop);
 137     p->dead = 0;
 138     p->direction = dir;
 139     p->read_data = NULL;
 140     p->read_index = p->read_length = 0;
 141     p->latency = 0;
 142
 143     if (!(p->context = pa_context_new(pa_mainloop_get_api(p->mainloop), name)))
 144         goto fail;
 145
 146     pa_context_connect(p->context, server, 1, NULL);
 147
 148     /* Wait until the context is ready */
 149     while (pa_context_get_state(p->context) != PA_CONTEXT_READY) {
 150         if (iterate(p, 1, &error) < 0)
 151             goto fail;
 152     }
 153
 154     if (!(p->stream = pa_stream_new(p->context, stream_name, ss, NULL)))
 155         goto fail;
 156
 157     if (dir == PA_STREAM_PLAYBACK)
 158         pa_stream_connect_playback(p->stream, dev, attr, 0, NULL);
 159     else
 160         pa_stream_connect_record(p->stream, dev, attr, 0);
 161
 162     /* Wait until the stream is ready */
 163     while (pa_stream_get_state(p->stream) != PA_STREAM_READY) {
 164         if (iterate(p, 1, &error) < 0)
 165             goto fail;
 166     }
 167
 168     pa_stream_set_read_callback(p->stream, read_callback, p);
 169
 170     return p;
 171
 172 fail:
 173     if (rerror)
 174         *rerror = error;
 175     pa_simple_free(p);
 176     return NULL;
 177 }
 178
 179 void pa_simple_free(pa_simple *s) {
 180     assert(s);
 181
 182     pa_xfree(s->read_data);
 183
 184     if (s->stream)
 185         pa_stream_unref(s->stream);
 186
 187     if (s->context)
 188         pa_context_unref(s->context);
 189
 190     if (s->mainloop)
 191         pa_mainloop_free(s->mainloop);
 192
 193     pa_xfree(s);
 194 }
 195
 196 int pa_simple_write(pa_simple *p, const void*data, size_t length, int *rerror) {
 197     assert(p && data && p->direction == PA_STREAM_PLAYBACK);
 198
 199     if (p->dead) {
 200         if (rerror)
 201             *rerror = pa_context_errno(p->context);
 202
 203         return -1;
 204     }
 205
 206     while (length > 0) {
 207         size_t l;
 208
 209         while (!(l = pa_stream_writable_size(p->stream)))
 210             if (iterate(p, 1, rerror) < 0)
 211                 return -1;
 212
 213         if (l > length)
 214             l = length;
 215
 216         pa_stream_write(p->stream, data, l, NULL, 0);
 217         data = (const uint8_t*) data + l;
 218         length -= l;
 219     }
 220
 221     /* Make sure that no data is pending for write */
 222     if (iterate(p, 0, rerror) < 0)
 223         return -1;
 224
 225     return 0;
 226 }
 227
 228 static void read_callback(pa_stream *s, const void*data, size_t length, void *userdata) {
 229     pa_simple *p = userdata;
 230     assert(s && data && length && p);
 231
 232     if (p->read_data) {
 233         pa_log(__FILE__": Buffer overflow, dropping incoming memory blocks.\n");
 234         pa_xfree(p->read_data);
 235     }
 236
 237     p->read_data = pa_xmemdup(data, p->read_length = length);
 238     p->read_index = 0;
 239 }
 240
 241 int pa_simple_read(pa_simple *p, void*data, size_t length, int *rerror) {
 242     assert(p && data && p->direction == PA_STREAM_RECORD);
 243
 244     if (p->dead) {
 245         if (rerror)
 246             *rerror = pa_context_errno(p->context);
 247
 248         return -1;
 249     }
 250
 251     while (length > 0) {
 252         if (p->read_data) {
 253             size_t l = length;
 254
 255             if (p->read_length <= l)
 256                 l = p->read_length;
 257
 258             memcpy(data, (uint8_t*) p->read_data+p->read_index, l);
 259
 260             data = (uint8_t*) data + l;
 261             length -= l;
 262
 263             p->read_index += l;
 264             p->read_length -= l;
 265
 266             if (!p->read_length) {
 267                 pa_xfree(p->read_data);
 268                 p->read_data = NULL;
 269                 p->read_index = 0;
 270             }
 271
 272             if (!length)
 273                 return 0;
 274
 275             assert(!p->read_data);
 276         }
 277
 278         if (iterate(p, 1, rerror) < 0)
 279             return -1;
 280     }
 281
 282     return 0;
 283 }
 284
 285 static void drain_or_flush_complete(pa_stream *s, int success, void *userdata) {
 286     pa_simple *p = userdata;
 287     assert(s && p);
 288     if (!success)
 289         p->dead = 1;
 290 }
 291
 292 int pa_simple_drain(pa_simple *p, int *rerror) {
 293     pa_operation *o;
 294     assert(p && p->direction == PA_STREAM_PLAYBACK);
 295
 296     if (p->dead) {
 297         if (rerror)
 298             *rerror = pa_context_errno(p->context);
 299
 300         return -1;
 301     }
 302
 303     o = pa_stream_drain(p->stream, drain_or_flush_complete, p);
 304
 305     while (pa_operation_get_state(o) == PA_OPERATION_RUNNING) {
 306         if (iterate(p, 1, rerror) < 0) {
 307             pa_operation_cancel(o);
 308             pa_operation_unref(o);
 309             return -1;
 310         }
 311     }
 312
 313     pa_operation_unref(o);
 314
 315     if (p->dead && rerror)
 316         *rerror = pa_context_errno(p->context);
 317
 318     return p->dead ? -1 : 0;
 319 }
 320
 321 static void latency_complete(pa_stream *s, const pa_latency_info *l, void *userdata) {
 322     pa_simple *p = userdata;
 323     assert(s && p);
 324
 325     if (!l)
 326         p->dead = 1;
 327     else {
 328         int negative = 0;
 329         p->latency = pa_stream_get_latency(s, l, &negative);
 330         if (negative)
 331             p->latency = 0;
 332     }
 333 }
 334
 335 pa_usec_t pa_simple_get_playback_latency(pa_simple *p, int *rerror) {
 336     pa_operation *o;
 337     assert(p && p->direction == PA_STREAM_PLAYBACK);
 338
 339     if (p->dead) {
 340         if (rerror)
 341             *rerror = pa_context_errno(p->context);
 342
 343         return (pa_usec_t) -1;
 344     }
 345
 346     p->latency = 0;
 347     o = pa_stream_get_latency_info(p->stream, latency_complete, p);
 348
 349     while (pa_operation_get_state(o) == PA_OPERATION_RUNNING) {
 350
 351         if (iterate(p, 1, rerror) < 0) {
 352             pa_operation_cancel(o);
 353             pa_operation_unref(o);
 354             return -1;
 355         }
 356     }
 357
 358     pa_operation_unref(o);
 359
 360     if (p->dead && rerror)
 361         *rerror = pa_context_errno(p->context);
 362
 363     return p->dead ? (pa_usec_t) -1 : p->latency;
 364 }
 365
 366 int pa_simple_flush(pa_simple *p, int *rerror) {
 367     pa_operation *o;
 368     assert(p && p->direction == PA_STREAM_PLAYBACK);
 369
 370     if (p->dead) {
 371         if (rerror)
 372             *rerror = pa_context_errno(p->context);
 373
 374         return -1;
 375     }
 376
 377     o = pa_stream_flush(p->stream, drain_or_flush_complete, p);
 378
 379     while (pa_operation_get_state(o) == PA_OPERATION_RUNNING) {
 380         if (iterate(p, 1, rerror) < 0) {
 381             pa_operation_cancel(o);
 382             pa_operation_unref(o);
 383             return -1;
 384         }
 385     }
 386
 387     pa_operation_unref(o);
 388
 389     if (p->dead && rerror)
 390         *rerror = pa_context_errno(p->context);
 391
 392     return p->dead ? -1 : 0;
 393 }