hardware/alsa/api.c

   1 /*
   2  * This file is part of the libsigrok project.
   3  *
   4  * Copyright (C) 2011 Daniel Ribeiro <drwyrm@gmail.com>
   5  * Copyright (C) 2012 Uwe Hermann <uwe@hermann-uwe.de>
   6  * Copyright (C) 2012 Alexandru Gagniuc <mr.nuke.me@gmail.com>
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 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 General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  21  */
  22
  23 #include <stdlib.h>
  24 #include <unistd.h>
  25 #include <string.h>
  26 #include "libsigrok.h"
  27 #include "libsigrok-internal.h"
  28 #include "protocol.h"
  29
  30 static const int32_t hwcaps[] = {
  31         SR_CONF_SAMPLERATE,
  32         SR_CONF_LIMIT_SAMPLES,
  33         SR_CONF_CONTINUOUS,
  34 };
  35
  36 SR_PRIV struct sr_dev_driver alsa_driver_info;
  37 static struct sr_dev_driver *di = &alsa_driver_info;
  38
  39 static void clear_helper(void *priv)
  40 {
  41         struct dev_context *devc;
  42
  43         devc = priv;
  44
  45         snd_pcm_hw_params_free(devc->hw_params);
  46         devc->hw_params = NULL;
  47
  48         g_free((void *)devc->samplerates);
  49         devc->samplerates = NULL;
  50 }
  51
  52 static int dev_clear(void)
  53 {
  54         return std_dev_clear(di, clear_helper);
  55 }
  56
  57 static int init(struct sr_context *sr_ctx)
  58 {
  59         return std_init(sr_ctx, di, LOG_PREFIX);
  60 }
  61
  62 static GSList *scan(GSList *options)
  63 {
  64         return alsa_scan(options, di);
  65 }
  66
  67 static GSList *dev_list(void)
  68 {
  69         return ((struct drv_context *)(di->priv))->instances;
  70 }
  71
  72 static int dev_open(struct sr_dev_inst *sdi)
  73 {
  74         struct dev_context *devc;
  75         int ret;
  76
  77         devc = sdi->priv;
  78
  79         if (!(devc->hwdev)) {
  80                 sr_err("devc->hwdev was NULL.");
  81                 return SR_ERR_BUG;
  82         }
  83
  84         sr_dbg("Opening audio device '%s' for stream capture.", devc->hwdev);
  85         ret = snd_pcm_open(&devc->capture_handle, devc->hwdev,
  86                            SND_PCM_STREAM_CAPTURE, 0);
  87         if (ret < 0) {
  88                 sr_err("Can't open audio device: %s.", snd_strerror(ret));
  89                 return SR_ERR;
  90         }
  91
  92         sr_dbg("Initializing hardware parameter structure.");
  93         ret = snd_pcm_hw_params_any(devc->capture_handle, devc->hw_params);
  94         if (ret < 0) {
  95                 sr_err("Can't initialize hardware parameter structure: %s.",
  96                        snd_strerror(ret));
  97                 return SR_ERR;
  98         }
  99
 100         sdi->status = SR_ST_ACTIVE;
 101
 102         return SR_OK;
 103 }
 104
 105 static int dev_close(struct sr_dev_inst *sdi)
 106 {
 107         int ret;
 108         struct dev_context *devc;
 109
 110         devc = sdi->priv;
 111
 112         if (devc->capture_handle) {
 113                 sr_dbg("Closing PCM device.");
 114                 if ((ret = snd_pcm_close(devc->capture_handle)) < 0)
 115                         sr_err("Failed to close device: %s.",
 116                                snd_strerror(ret));
 117                 devc->capture_handle = NULL;
 118                 sdi->status = SR_ST_INACTIVE;
 119         } else {
 120                 sr_dbg("No capture handle, no need to close audio device.");
 121         }
 122
 123         return SR_OK;
 124 }
 125
 126 static int cleanup(void)
 127 {
 128         return dev_clear();
 129 }
 130
 131 static int config_get(int id, GVariant **data, const struct sr_dev_inst *sdi)
 132 {
 133         struct dev_context *devc;
 134
 135         switch (id) {
 136         case SR_CONF_SAMPLERATE:
 137                 devc = sdi->priv;
 138                 *data = g_variant_new_uint64(devc->cur_samplerate);
 139                 break;
 140         default:
 141                 return SR_ERR_NA;
 142         }
 143
 144         return SR_OK;
 145 }
 146
 147 static int config_set(int id, GVariant *data, const struct sr_dev_inst *sdi)
 148 {
 149         struct dev_context *devc;
 150
 151         if (sdi->status != SR_ST_ACTIVE)
 152                 return SR_ERR_DEV_CLOSED;
 153
 154         devc = sdi->priv;
 155
 156         switch (id) {
 157         case SR_CONF_SAMPLERATE:
 158                 alsa_set_samplerate(sdi, g_variant_get_uint64(data));
 159                 break;
 160         case SR_CONF_LIMIT_SAMPLES:
 161                 devc->limit_samples = g_variant_get_uint64(data);
 162                 break;
 163         default:
 164                 return SR_ERR_NA;
 165         }
 166
 167         return SR_OK;
 168 }
 169
 170 static int config_list(int key, GVariant **data, const struct sr_dev_inst *sdi)
 171 {
 172         struct dev_context *devc;
 173         GVariant *gvar;
 174         GVariantBuilder gvb;
 175         int i;
 176
 177         switch (key) {
 178         case SR_CONF_DEVICE_OPTIONS:
 179                 *data = g_variant_new_fixed_array(G_VARIANT_TYPE_INT32,
 180                                 hwcaps, ARRAY_SIZE(hwcaps), sizeof(int32_t));
 181                 break;
 182         case SR_CONF_SAMPLERATE:
 183                 if (!sdi || !sdi->priv)
 184                         return SR_ERR_ARG;
 185                 devc = sdi->priv;
 186                 if (!devc->samplerates) {
 187                         sr_err("Instance did not contain a samplerate list.");
 188                         return SR_ERR_ARG;
 189                 }
 190                 for (i = 0; devc->samplerates[i]; i++)
 191                         ;
 192                 g_variant_builder_init(&gvb, G_VARIANT_TYPE("a{sv}"));
 193                 gvar = g_variant_new_fixed_array(G_VARIANT_TYPE("t"),
 194                                 devc->samplerates, i, sizeof(uint64_t));
 195                 g_variant_builder_add(&gvb, "{sv}", "samplerates", gvar);
 196                 *data = g_variant_builder_end(&gvb);
 197                 break;
 198         default:
 199                 return SR_ERR_NA;
 200         }
 201
 202         return SR_OK;
 203 }
 204
 205 static int dev_acquisition_start(const struct sr_dev_inst *sdi, void *cb_data)
 206 {
 207         struct dev_context *devc;
 208         int count, ret;
 209         char *endianness;
 210
 211         if (sdi->status != SR_ST_ACTIVE)
 212                 return SR_ERR_DEV_CLOSED;
 213
 214         devc = sdi->priv;
 215         devc->cb_data = cb_data;
 216         devc->num_samples = 0;
 217
 218         sr_dbg("Setting audio access type to RW/interleaved.");
 219         ret = snd_pcm_hw_params_set_access(devc->capture_handle,
 220                         devc->hw_params, SND_PCM_ACCESS_RW_INTERLEAVED);
 221         if (ret < 0) {
 222                 sr_err("Can't set audio access type: %s.", snd_strerror(ret));
 223                 return SR_ERR;
 224         }
 225
 226         /* FIXME: Hardcoded for 16bits. */
 227         if (SND_PCM_FORMAT_S16 == SND_PCM_FORMAT_S16_LE)
 228                 endianness = "little endian";
 229         else
 230                 endianness = "big endian";
 231         sr_dbg("Setting audio sample format to signed 16bit (%s).", endianness);
 232         ret = snd_pcm_hw_params_set_format(devc->capture_handle,
 233                                            devc->hw_params,
 234                                            SND_PCM_FORMAT_S16);
 235         if (ret < 0) {
 236                 sr_err("Can't set audio sample format: %s.", snd_strerror(ret));
 237                 return SR_ERR;
 238         }
 239
 240         sr_dbg("Setting audio samplerate to %" PRIu64 "Hz.",
 241                devc->cur_samplerate);
 242         ret = snd_pcm_hw_params_set_rate(devc->capture_handle, devc->hw_params,
 243                                          (unsigned int)devc->cur_samplerate, 0);
 244         if (ret < 0) {
 245                 sr_err("Can't set audio sample rate: %s.", snd_strerror(ret));
 246                 return SR_ERR;
 247         }
 248
 249         sr_dbg("Setting audio channel count to %d.", devc->num_probes);
 250         ret = snd_pcm_hw_params_set_channels(devc->capture_handle,
 251                                              devc->hw_params, devc->num_probes);
 252         if (ret < 0) {
 253                 sr_err("Can't set channel count: %s.", snd_strerror(ret));
 254                 return SR_ERR;
 255         }
 256
 257         sr_dbg("Setting audio parameters.");
 258         ret = snd_pcm_hw_params(devc->capture_handle, devc->hw_params);
 259         if (ret < 0) {
 260                 sr_err("Can't set parameters: %s.", snd_strerror(ret));
 261                 return SR_ERR;
 262         }
 263
 264         sr_dbg("Preparing audio interface for use.");
 265         ret = snd_pcm_prepare(devc->capture_handle);
 266         if (ret < 0) {
 267                 sr_err("Can't prepare audio interface for use: %s.",
 268                        snd_strerror(ret));
 269                 return SR_ERR;
 270         }
 271
 272         count = snd_pcm_poll_descriptors_count(devc->capture_handle);
 273         if (count < 1) {
 274                 sr_err("Unable to obtain poll descriptors count.");
 275                 return SR_ERR;
 276         }
 277
 278         if (!(devc->ufds = g_try_malloc(count * sizeof(struct pollfd)))) {
 279                 sr_err("Failed to malloc ufds.");
 280                 return SR_ERR_MALLOC;
 281         }
 282
 283         sr_spew("Getting %d poll descriptors.", count);
 284         ret = snd_pcm_poll_descriptors(devc->capture_handle, devc->ufds, count);
 285         if (ret < 0) {
 286                 sr_err("Unable to obtain poll descriptors: %s.",
 287                        snd_strerror(ret));
 288                 g_free(devc->ufds);
 289                 return SR_ERR;
 290         }
 291
 292         /* Send header packet to the session bus. */
 293         std_session_send_df_header(cb_data, LOG_PREFIX);
 294
 295         /* Poll every 10ms, or whenever some data comes in. */
 296         sr_source_add(devc->ufds[0].fd, devc->ufds[0].events, 10,
 297                       alsa_receive_data, (void *)sdi);
 298
 299         // g_free(devc->ufds); /* FIXME */
 300
 301         return SR_OK;
 302 }
 303
 304 static int dev_acquisition_stop(struct sr_dev_inst *sdi, void *cb_data)
 305 {
 306         struct sr_datafeed_packet packet;
 307         struct dev_context *devc;
 308
 309         devc = sdi->priv;
 310         devc->cb_data = cb_data;
 311
 312         sr_source_remove(devc->ufds[0].fd);
 313
 314         /* Send end packet to the session bus. */
 315         sr_dbg("Sending SR_DF_END packet.");
 316         packet.type = SR_DF_END;
 317         sr_session_send(cb_data, &packet);
 318
 319         return SR_OK;
 320 }
 321
 322 SR_PRIV struct sr_dev_driver alsa_driver_info = {
 323         .name = "alsa",
 324         .longname = "ALSA driver",
 325         .api_version = 1,
 326         .init = init,
 327         .cleanup = cleanup,
 328         .scan = scan,
 329         .dev_list = dev_list,
 330         .dev_clear = dev_clear,
 331         .config_get = config_get,
 332         .config_set = config_set,
 333         .config_list = config_list,
 334         .dev_open = dev_open,
 335         .dev_close = dev_close,
 336         .dev_acquisition_start = dev_acquisition_start,
 337         .dev_acquisition_stop = dev_acquisition_stop,
 338         .priv = NULL,
 339 };