X-Git-Url: https://sigrok.org/gitweb/?a=blobdiff_plain;f=src%2Finput%2Fwav.c;h=5669e8a4453c456195c95af9d92432167f89cb75;hb=d65fcbcd4101c276e509746e1af73b3a95aa03fb;hp=1c3f050a050434ea040f64b7b1c974d520377bc1;hpb=155b680da482cea2381becb73c51cfb838bff31e;p=libsigrok.git diff --git a/src/input/wav.c b/src/input/wav.c index 1c3f050a..5669e8a4 100644 --- a/src/input/wav.c +++ b/src/input/wav.c @@ -21,184 +21,307 @@ #include #include #include +#include #include #include "libsigrok.h" #include "libsigrok-internal.h" #define LOG_PREFIX "input/wav" +/* How many bytes at a time to process and send to the session bus. */ #define CHUNK_SIZE 4096 +/* Minimum size of header + 1 8-bit mono PCM sample. */ +#define MIN_DATA_CHUNK_OFFSET 45 + +/* Expect to find the "data" chunk within this offset from the start. */ +#define MAX_DATA_CHUNK_OFFSET 256 + +#define WAVE_FORMAT_PCM 1 +#define WAVE_FORMAT_IEEE_FLOAT 3 + struct context { + int fmt_code; uint64_t samplerate; int samplesize; int num_channels; + int unitsize; + gboolean found_data; }; -static int get_wav_header(const char *filename, char *buf) +static int parse_wav_header(GString *buf, struct context *inc) { - struct stat st; - int fd, l; + uint64_t samplerate; + int fmt_code, samplesize, num_channels, unitsize; - l = strlen(filename); - if (l <= 4 || strcasecmp(filename + l - 4, ".wav")) + if (buf->len < MIN_DATA_CHUNK_OFFSET) { return SR_ERR; + } - if (stat(filename, &st) == -1) - return SR_ERR; - if (st.st_size <= 45) - /* Minimum size of header + 1 8-bit mono PCM sample. */ - return SR_ERR; + fmt_code = GUINT16_FROM_LE(*(uint16_t *)(buf->str + 20)); + samplerate = GUINT32_FROM_LE(*(uint32_t *)(buf->str + 24)); + samplesize = GUINT16_FROM_LE(*(uint16_t *)(buf->str + 32)); + num_channels = GUINT16_FROM_LE(*(uint16_t *)(buf->str + 22)); + /* TODO div0 */ + unitsize = samplesize / num_channels; - if ((fd = open(filename, O_RDONLY)) == -1) + if (fmt_code == WAVE_FORMAT_PCM) { + if (samplesize != 1 && samplesize != 2 && samplesize != 4) { + sr_err("only 8, 16 or 32 bits per sample supported."); + return SR_ERR; + } + } else if (fmt_code == WAVE_FORMAT_IEEE_FLOAT) { + if (unitsize != 4) { + sr_err("only 32-bit floats supported."); + return SR_ERR; + } + } else { + sr_err("Only PCM and floating point samples are supported."); return SR_ERR; + } - l = read(fd, buf, 40); - close(fd); - if (l != 40) - return SR_ERR; + if (inc) { + inc->fmt_code = fmt_code; + inc->samplerate = samplerate; + inc->samplesize = samplesize; + inc->num_channels = num_channels; + inc->unitsize = unitsize; + inc->found_data = FALSE; + } return SR_OK; } -static int format_match(const char *filename) +static int format_match(GHashTable *metadata) { - char buf[40]; + GString *buf; - if (get_wav_header(filename, buf) != SR_OK) + buf = g_hash_table_lookup(metadata, GINT_TO_POINTER(SR_INPUT_META_HEADER)); + if (strncmp(buf->str, "RIFF", 4)) return FALSE; - - if (strncmp(buf, "RIFF", 4)) + if (strncmp(buf->str + 8, "WAVE", 4)) return FALSE; - if (strncmp(buf + 8, "WAVE", 4)) + if (strncmp(buf->str + 12, "fmt ", 4)) return FALSE; - if (strncmp(buf + 12, "fmt ", 4)) - return FALSE; - if (GUINT16_FROM_LE(*(uint16_t *)(buf + 20)) != 1) - /* Not PCM. */ - return FALSE; - if (strncmp(buf + 36, "data", 4)) + /* + * Only gets called when we already know this is a WAV file, so + * this parser can log error messages. + */ + if (parse_wav_header(buf, NULL) != SR_OK) return FALSE; return TRUE; } -static int init(struct sr_input *in, const char *filename) +static int init(struct sr_input *in, GHashTable *options) { - struct sr_channel *ch; - struct context *ctx; - char buf[40], channelname[8]; - int i; + (void)options; - if (get_wav_header(filename, buf) != SR_OK) - return SR_ERR; - - if (!(ctx = g_try_malloc0(sizeof(struct context)))) - return SR_ERR_MALLOC; - - /* Create a virtual device. */ in->sdi = sr_dev_inst_new(0, SR_ST_ACTIVE, NULL, NULL, NULL); - in->sdi->priv = ctx; - - ctx->samplerate = GUINT32_FROM_LE(*(uint32_t *)(buf + 24)); - ctx->samplesize = GUINT16_FROM_LE(*(uint16_t *)(buf + 34)) / 8; - if (ctx->samplesize != 1 && ctx->samplesize != 2 && ctx->samplesize != 4) { - sr_err("only 8, 16 or 32 bits per sample supported."); - return SR_ERR; - } - if ((ctx->num_channels = GUINT16_FROM_LE(*(uint16_t *)(buf + 22))) > 20) { - sr_err("%d channels seems crazy.", ctx->num_channels); - return SR_ERR; - } + return SR_OK; +} - for (i = 0; i < ctx->num_channels; i++) { - snprintf(channelname, 8, "CH%d", i + 1); - if (!(ch = sr_channel_new(0, SR_CHANNEL_ANALOG, TRUE, channelname))) - return SR_ERR; - in->sdi->channels = g_slist_append(in->sdi->channels, ch); +static int find_data_chunk(GString *buf, int initial_offset) +{ + unsigned int offset, i; + + offset = initial_offset; + while(offset < MIN(MAX_DATA_CHUNK_OFFSET, buf->len)) { + if (!memcmp(buf->str + offset, "data", 4)) + /* Skip into the samples. */ + return offset + 8; + for (i = 0; i < 4; i++) { + if (!isalpha(buf->str[offset + i]) + && !isascii(buf->str[offset + i]) + && !isblank(buf->str[offset + i])) + /* Doesn't look like a chunk ID. */ + return -1; + } + /* Skip past this chunk. */ + offset += 8 + GUINT32_FROM_LE(*(uint32_t *)(buf->str + offset + 4)); } - return SR_OK; + return offset; } -static int loadfile(struct sr_input *in, const char *filename) +static int initial_receive(struct sr_input *in) { struct sr_datafeed_packet packet; struct sr_datafeed_meta meta; - struct sr_datafeed_analog analog; + struct sr_channel *ch; struct sr_config *src; - struct context *ctx; - float fdata[CHUNK_SIZE]; - uint64_t sample; - int num_samples, chunk_samples, s, c, fd, l; - char buf[CHUNK_SIZE]; + struct context *inc; + int i; + char channelname[8]; + + if (!in->buf) + /* Shouldn't happen. */ + return SR_ERR; + + inc = in->priv = g_malloc(sizeof(struct context)); + if (parse_wav_header(in->buf, inc) != SR_OK) + return SR_ERR; - ctx = in->sdi->priv; + for (i = 0; i < inc->num_channels; i++) { + snprintf(channelname, 8, "CH%d", i + 1); + ch = sr_channel_new(i, SR_CHANNEL_ANALOG, TRUE, channelname); + in->sdi->channels = g_slist_append(in->sdi->channels, ch); + } - /* Send header packet to the session bus. */ std_session_send_df_header(in->sdi, LOG_PREFIX); packet.type = SR_DF_META; packet.payload = &meta; - src = sr_config_new(SR_CONF_SAMPLERATE, - g_variant_new_uint64(ctx->samplerate)); + src = sr_config_new(SR_CONF_SAMPLERATE, g_variant_new_uint64(inc->samplerate)); meta.config = g_slist_append(NULL, src); sr_session_send(in->sdi, &packet); sr_config_free(src); - if ((fd = open(filename, O_RDONLY)) == -1) - return SR_ERR; + return SR_OK; +} - lseek(fd, 40, SEEK_SET); - l = read(fd, buf, 4); - num_samples = GUINT32_FROM_LE((uint32_t)*(buf)); - num_samples /= ctx->samplesize / ctx->num_channels; - while (TRUE) { - if ((l = read(fd, buf, CHUNK_SIZE)) < 1) - break; - chunk_samples = l / ctx->samplesize / ctx->num_channels; - for (s = 0; s < chunk_samples; s++) { - for (c = 0; c < ctx->num_channels; c++) { - sample = 0; - memcpy(&sample, buf + s * ctx->samplesize + c, ctx->samplesize); - switch (ctx->samplesize) { - case 1: - /* 8-bit PCM samples are unsigned. */ - fdata[s + c] = (uint8_t)sample / 255.0; - break; - case 2: - fdata[s + c] = GINT16_FROM_LE(sample) / 32767.0; - break; - case 4: - fdata[s + c] = GINT32_FROM_LE(sample) / 65535.0; - break; - } +static void send_chunk(const struct sr_input *in, int offset, int num_samples) +{ + struct sr_datafeed_packet packet; + struct sr_datafeed_analog analog; + struct context *inc; + float fdata[CHUNK_SIZE]; + uint64_t sample; + int total_samples, samplenum; + char *s, *d; + + inc = in->priv; + + s = in->buf->str + offset; + d = (char *)fdata; + memset(fdata, 0, CHUNK_SIZE); + total_samples = num_samples * inc->num_channels; + for (samplenum = 0; samplenum < total_samples; samplenum++) { + if (inc->fmt_code == WAVE_FORMAT_PCM) { + sample = 0; + memcpy(&sample, s, inc->unitsize); + switch (inc->samplesize) { + case 1: + /* 8-bit PCM samples are unsigned. */ + fdata[samplenum] = (uint8_t)sample / 255.0; + break; + case 2: + fdata[samplenum] = GINT16_FROM_LE(sample) / 32767.0; + break; + case 4: + fdata[samplenum] = GINT32_FROM_LE(sample) / 65535.0; + break; } + } else { + /* BINARY32 float */ +#ifdef WORDS_BIGENDIAN + for (i = 0; i < inc->unitsize; i++) + d[i] = s[inc->unitsize - i]; +#else + memcpy(d, s, inc->unitsize); +#endif } - packet.type = SR_DF_ANALOG; - packet.payload = &analog; - analog.channels = in->sdi->channels; - analog.num_samples = chunk_samples; - analog.mq = 0; - analog.unit = 0; - analog.data = fdata; + s += inc->unitsize; + d += inc->unitsize; + } + packet.type = SR_DF_ANALOG; + packet.payload = &analog; + analog.channels = in->sdi->channels; + analog.num_samples = num_samples; + analog.mq = 0; + analog.mqflags = 0; + analog.unit = 0; + analog.data = fdata; + sr_session_send(in->sdi, &packet); +} + +static int receive(const struct sr_input *in, GString *buf) +{ + struct sr_datafeed_packet packet; + struct context *inc; + int offset, chunk_samples, total_samples, processed, max_chunk_samples, num_samples, i; + + if (buf->len == 0) { + /* End of stream. */ + packet.type = SR_DF_END; sr_session_send(in->sdi, &packet); + return SR_OK; } - close(fd); - packet.type = SR_DF_END; - sr_session_send(in->sdi, &packet); + g_string_append_len(in->buf, buf->str, buf->len); + + if (!in->priv) { + if (initial_receive((struct sr_input *)in) != SR_OK) + return SR_ERR; + if (in->buf->len < MIN_DATA_CHUNK_OFFSET) { + /* + * Don't even get started until there's enough room + * for the data segment to start. + */ + return SR_OK; + } + } + inc = in->priv; + + if (!inc->found_data) { + /* Skip past size of 'fmt ' chunk. */ + i = 20 + GUINT32_FROM_LE(*(uint32_t *)(in->buf->str + 16)); + offset = find_data_chunk(in->buf, i); + if (offset < 0) { + if (in->buf->len > MAX_DATA_CHUNK_OFFSET) { + sr_err("Couldn't find data chunk."); + return SR_ERR; + } + } + inc->found_data = TRUE; + } else + offset = 0; + + /* Round off up to the last channels * unitsize boundary. */ + chunk_samples = (in->buf->len - offset) / inc->num_channels / inc->unitsize; + max_chunk_samples = CHUNK_SIZE / inc->num_channels / inc->unitsize; + processed = 0; + total_samples = chunk_samples; + while (processed < total_samples) { + if (chunk_samples > max_chunk_samples) + num_samples = max_chunk_samples; + else + num_samples = chunk_samples; + send_chunk(in, offset, num_samples); + offset += num_samples * inc->unitsize; + chunk_samples -= num_samples; + processed += num_samples; + } + + if ((unsigned int)offset < in->buf->len) { + /* + * The incoming buffer wasn't processed completely. Stash + * the leftover data for next time. + */ + g_string_erase(in->buf, 0, offset); + } else + g_string_truncate(in->buf, 0); return SR_OK; } +static int cleanup(struct sr_input *in) +{ + g_free(in->priv); + in->priv = NULL; + + return SR_OK; +} -SR_PRIV struct sr_input_format input_wav = { +SR_PRIV struct sr_input_module input_wav = { .id = "wav", - .description = "WAV file", + .name = "WAV", + .desc = "WAV file", + .metadata = { SR_INPUT_META_HEADER | SR_INPUT_META_REQUIRED }, .format_match = format_match, .init = init, - .loadfile = loadfile, + .receive = receive, + .cleanup = cleanup, };