src/tests/sync-playback.c

   1 /***
   2   This file is part of PulseAudio.
   3
   4   PulseAudio is free software; you can redistribute it and/or modify
   5   it under the terms of the GNU Lesser General Public License as published
   6   by the Free Software Foundation; either version 2.1 of the License,
   7   or (at your option) any later version.
   8
   9   PulseAudio is distributed in the hope that it will be useful, but
  10   WITHOUT ANY WARRANTY; without even the implied warranty of
  11   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12   General Public License for more details.
  13
  14   You should have received a copy of the GNU Lesser General Public License
  15   along with PulseAudio; if not, write to the Free Software
  16   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  17   USA.
  18 ***/
  19
  20 #ifdef HAVE_CONFIG_H
  21 #include <config.h>
  22 #endif
  23
  24 #include <signal.h>
  25 #include <errno.h>
  26 #include <unistd.h>
  27 #include <assert.h>
  28 #include <stdio.h>
  29 #include <stdlib.h>
  30 #include <math.h>
  31
  32 #include <check.h>
  33
  34 #include <pulse/pulseaudio.h>
  35 #include <pulse/mainloop.h>
  36
  37 #define NSTREAMS 4
  38 #define SINE_HZ 440
  39 #define SAMPLE_HZ 8000
  40
  41 static pa_context *context = NULL;
  42 static pa_stream *streams[NSTREAMS];
  43 static pa_mainloop_api *mainloop_api = NULL;
  44 static const char *bname = NULL;
  45
  46 static float data[SAMPLE_HZ]; /* one second space */
  47
  48 static int n_streams_ready = 0;
  49
  50 static const pa_sample_spec sample_spec = {
  51     .format = PA_SAMPLE_FLOAT32,
  52     .rate = SAMPLE_HZ,
  53     .channels = 1
  54 };
  55
  56 static const pa_buffer_attr buffer_attr = {
  57     .maxlength = SAMPLE_HZ*sizeof(float)*NSTREAMS, /* exactly space for the entire play time */
  58     .tlength = (uint32_t) -1,
  59     .prebuf = 0, /* Setting prebuf to 0 guarantees us the streams will run synchronously, no matter what */
  60     .minreq = (uint32_t) -1,
  61     .fragsize = 0
  62 };
  63
  64 static void nop_free_cb(void *p) {}
  65
  66 static void underflow_cb(struct pa_stream *s, void *userdata) {
  67     int i = (int) (long) userdata;
  68
  69     fprintf(stderr, "Stream %i finished\n", i);
  70
  71     if (++n_streams_ready >= 2*NSTREAMS) {
  72         fprintf(stderr, "We're done\n");
  73         mainloop_api->quit(mainloop_api, 0);
  74     }
  75 }
  76
  77 /* This routine is called whenever the stream state changes */
  78 static void stream_state_callback(pa_stream *s, void *userdata) {
  79     fail_unless(s != NULL);
  80
  81     switch (pa_stream_get_state(s)) {
  82         case PA_STREAM_UNCONNECTED:
  83         case PA_STREAM_CREATING:
  84         case PA_STREAM_TERMINATED:
  85             break;
  86
  87         case PA_STREAM_READY: {
  88
  89             int r, i = (int) (long) userdata;
  90
  91             fprintf(stderr, "Writing data to stream %i.\n", i);
  92
  93             r = pa_stream_write(s, data, sizeof(data), nop_free_cb, (int64_t) sizeof(data) * (int64_t) i, PA_SEEK_ABSOLUTE);
  94             fail_unless(r == 0);
  95
  96             /* Be notified when this stream is drained */
  97             pa_stream_set_underflow_callback(s, underflow_cb, userdata);
  98
  99             /* All streams have been set up, let's go! */
 100             if (++n_streams_ready >= NSTREAMS) {
 101                 fprintf(stderr, "Uncorking\n");
 102                 pa_operation_unref(pa_stream_cork(s, 0, NULL, NULL));
 103             }
 104
 105             break;
 106         }
 107
 108         default:
 109         case PA_STREAM_FAILED:
 110             fprintf(stderr, "Stream error: %s\n", pa_strerror(pa_context_errno(pa_stream_get_context(s))));
 111             fail();
 112     }
 113 }
 114
 115 /* This is called whenever the context status changes */
 116 static void context_state_callback(pa_context *c, void *userdata) {
 117     fail_unless(c != NULL);
 118
 119     switch (pa_context_get_state(c)) {
 120         case PA_CONTEXT_CONNECTING:
 121         case PA_CONTEXT_AUTHORIZING:
 122         case PA_CONTEXT_SETTING_NAME:
 123             break;
 124
 125         case PA_CONTEXT_READY: {
 126
 127             int i;
 128             fprintf(stderr, "Connection established.\n");
 129
 130             for (i = 0; i < NSTREAMS; i++) {
 131                 char name[64];
 132
 133                 fprintf(stderr, "Creating stream %i\n", i);
 134
 135                 snprintf(name, sizeof(name), "stream #%i", i);
 136
 137                 streams[i] = pa_stream_new(c, name, &sample_spec, NULL);
 138                 fail_unless(streams[i] != NULL);
 139                 pa_stream_set_state_callback(streams[i], stream_state_callback, (void*) (long) i);
 140                 pa_stream_connect_playback(streams[i], NULL, &buffer_attr, PA_STREAM_START_CORKED, NULL, i == 0 ? NULL : streams[0]);
 141             }
 142
 143             break;
 144         }
 145
 146         case PA_CONTEXT_TERMINATED:
 147             mainloop_api->quit(mainloop_api, 0);
 148             break;
 149
 150         case PA_CONTEXT_FAILED:
 151         default:
 152             fprintf(stderr, "Context error: %s\n", pa_strerror(pa_context_errno(c)));
 153             fail();
 154     }
 155 }
 156
 157 START_TEST (sync_playback_test) {
 158     pa_mainloop* m = NULL;
 159     int i, ret = 0;
 160
 161     for (i = 0; i < SAMPLE_HZ; i++)
 162         data[i] = (float) sin(((double) i/SAMPLE_HZ)*2*M_PI*SINE_HZ)/2;
 163
 164     for (i = 0; i < NSTREAMS; i++)
 165         streams[i] = NULL;
 166
 167     /* Set up a new main loop */
 168     m = pa_mainloop_new();
 169     fail_unless(m != NULL);
 170
 171     mainloop_api = pa_mainloop_get_api(m);
 172
 173     context = pa_context_new(mainloop_api, bname);
 174     fail_unless(context != NULL);
 175
 176     pa_context_set_state_callback(context, context_state_callback, NULL);
 177
 178     /* Connect the context */
 179     if (pa_context_connect(context, NULL, 0, NULL) < 0) {
 180         fprintf(stderr, "pa_context_connect() failed.\n");
 181         goto quit;
 182     }
 183
 184     if (pa_mainloop_run(m, &ret) < 0)
 185         fprintf(stderr, "pa_mainloop_run() failed.\n");
 186
 187 quit:
 188     pa_context_unref(context);
 189
 190     for (i = 0; i < NSTREAMS; i++)
 191         if (streams[i])
 192             pa_stream_unref(streams[i]);
 193
 194     pa_mainloop_free(m);
 195
 196     fail_unless(ret == 0);
 197 }
 198 END_TEST
 199
 200 int main(int argc, char *argv[]) {
 201     int failed = 0;
 202     Suite *s;
 203     TCase *tc;
 204     SRunner *sr;
 205
 206     bname = argv[0];
 207
 208     s = suite_create("Sync Playback");
 209     tc = tcase_create("syncplayback");
 210     tcase_add_test(tc, sync_playback_test);
 211     tcase_set_timeout(tc, 5 * 60);
 212     suite_add_tcase(s, tc);
 213
 214     sr = srunner_create(s);
 215     srunner_run_all(sr, CK_NORMAL);
 216     failed = srunner_ntests_failed(sr);
 217     srunner_free(sr);
 218
 219     return (failed == 0) ? EXIT_SUCCESS : EXIT_FAILURE;
 220 }