[libsigrok.git] / src / input / wav.c

/*
 * This file is part of the libsigrok project.
 *
 * Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <ctype.h>
#include <string.h>
#include <stdint.h>
#include <libsigrok/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               (1 * 1024 * 1024 * sizeof(float))

/* 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    1024

#define WAVE_FORMAT_PCM_         0x0001
#define WAVE_FORMAT_IEEE_FLOAT_  0x0003
#define WAVE_FORMAT_EXTENSIBLE_  0xfffe

struct context {
	gboolean started;
	int fmt_code;
	uint64_t samplerate;
	int samplesize;
	int num_channels;
	int unitsize;
	gboolean found_data;
	gboolean create_channels;
	GSList *prev_sr_channels;
};

static int parse_wav_header(GString *buf, struct context *inc)
{
	uint64_t samplerate;
	unsigned int fmt_code, samplesize, num_channels, unitsize;

	if (buf->len < MIN_DATA_CHUNK_OFFSET)
		return SR_ERR_NA;

	fmt_code = RL16(buf->str + 20);
	samplerate = RL32(buf->str + 24);

	samplesize = RL16(buf->str + 32);
	num_channels = RL16(buf->str + 22);
	if (num_channels == 0)
		return SR_ERR;
	unitsize = samplesize / num_channels;
	if (unitsize != 1 && unitsize != 2 && unitsize != 4) {
		sr_err("Only 8, 16 or 32 bits per sample supported.");
		return SR_ERR_DATA;
	}

	if (fmt_code == WAVE_FORMAT_PCM_) {
	} else if (fmt_code == WAVE_FORMAT_IEEE_FLOAT_) {
		if (unitsize != 4) {
			sr_err("only 32-bit floats supported.");
			return SR_ERR_DATA;
		}
	} else if (fmt_code == WAVE_FORMAT_EXTENSIBLE_) {
		if (buf->len < 70)
			/* Not enough for extensible header and next chunk. */
			return SR_ERR_NA;

		if (RL16(buf->str + 16) != 40) {
			sr_err("WAV extensible format chunk must be 40 bytes.");
			return SR_ERR;
		}
		if (RL16(buf->str + 36) != 22) {
			sr_err("WAV extension must be 22 bytes.");
			return SR_ERR;
		}
		if (RL16(buf->str + 34) != RL16(buf->str + 38)) {
			sr_err("Reduced valid bits per sample not supported.");
			return SR_ERR_DATA;
		}
		/* Real format code is the first two bytes of the GUID. */
		fmt_code = RL16(buf->str + 44);
		if (fmt_code != WAVE_FORMAT_PCM_ && fmt_code != WAVE_FORMAT_IEEE_FLOAT_) {
			sr_err("Only PCM and floating point samples are supported.");
			return SR_ERR_DATA;
		}
		if (fmt_code == WAVE_FORMAT_IEEE_FLOAT_ && unitsize != 4) {
			sr_err("only 32-bit floats supported.");
			return SR_ERR_DATA;
		}
	} else {
		sr_err("Only PCM and floating point samples are supported.");
		return SR_ERR_DATA;
	}

	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(GHashTable *metadata, unsigned int *confidence)
{
	GString *buf;
	int ret;

	buf = g_hash_table_lookup(metadata, GINT_TO_POINTER(SR_INPUT_META_HEADER));
	if (strncmp(buf->str, "RIFF", 4))
		return SR_ERR;
	if (strncmp(buf->str + 8, "WAVE", 4))
		return SR_ERR;
	if (strncmp(buf->str + 12, "fmt ", 4))
		return SR_ERR;
	/*
	 * Only gets called when we already know this is a WAV file, so
	 * this parser can log error messages.
	 */
	if ((ret = parse_wav_header(buf, NULL)) != SR_OK)
		return ret;

	*confidence = 1;

	return SR_OK;
}

static int init(struct sr_input *in, GHashTable *options)
{
	struct context *inc;

	(void)options;

	in->sdi = g_malloc0(sizeof(struct sr_dev_inst));
	in->priv = g_malloc0(sizeof(struct context));
	inc = in->priv;

	inc->create_channels = TRUE;

	return SR_OK;
}

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 (!isalnum(buf->str[offset + i])
					&& !isblank(buf->str[offset + i]))
				/* Doesn't look like a chunk ID. */
				return -1;
		}
		/* Skip past this chunk. */
		offset += 8 + RL32(buf->str + offset + 4);
	}

	if (offset > MAX_DATA_CHUNK_OFFSET)
		return -1;

	return offset;
}

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 sr_analog_encoding encoding;
	struct sr_analog_meaning meaning;
	struct sr_analog_spec spec;
	struct context *inc;
	float *fdata;
	int total_samples, samplenum;
	char *s, *d;

	inc = in->priv;

	total_samples = num_samples * inc->num_channels;
	fdata = g_malloc0(total_samples * sizeof(float));
	s = in->buf->str + offset;
	d = (char *)fdata;

	for (samplenum = 0; samplenum < total_samples; samplenum++) {
		if (inc->fmt_code == WAVE_FORMAT_PCM_) {
			switch (inc->unitsize) {
			case 1:
				/* 8-bit PCM samples are unsigned. */
				fdata[samplenum] = *(uint8_t*)(s) / (float)255;
				break;
			case 2:
				fdata[samplenum] = RL16S(s) / (float)INT16_MAX;
				break;
			case 4:
				fdata[samplenum] = RL32S(s) / (float)INT32_MAX;
				break;
			}
		} else {
			/* BINARY32 float */
#ifdef WORDS_BIGENDIAN
			int i;
			for (i = 0; i < inc->unitsize; i++)
				d[i] = s[inc->unitsize - 1 - i];
#else
			memcpy(d, s, inc->unitsize);
#endif
		}
		s += inc->unitsize;
		d += inc->unitsize;
	}

	/* TODO: Use proper 'digits' value for this device (and its modes). */
	sr_analog_init(&analog, &encoding, &meaning, &spec, 2);
	packet.type = SR_DF_ANALOG;
	packet.payload = &analog;
	analog.num_samples = num_samples;
	analog.data = fdata;
	analog.meaning->channels = in->sdi->channels;
	analog.meaning->mq = 0;
	analog.meaning->mqflags = 0;
	analog.meaning->unit = 0;
	sr_session_send(in->sdi, &packet);
	g_free(fdata);
}

static int process_buffer(struct sr_input *in)
{
	struct context *inc;
	int offset, chunk_samples, total_samples, processed, max_chunk_samples;
	int num_samples, i;

	inc = in->priv;
	if (!inc->started) {
		std_session_send_df_header(in->sdi);
		(void)sr_session_send_meta(in->sdi, SR_CONF_SAMPLERATE,
			g_variant_new_uint64(inc->samplerate));
		inc->started = TRUE;
	}

	if (!inc->found_data) {
		/* Skip past size of 'fmt ' chunk. */
		i = 20 + RL32(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->samplesize;
	max_chunk_samples = CHUNK_SIZE / inc->samplesize;
	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->samplesize;
		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;
}

/*
 * Check the channel list for consistency across file re-import. See
 * the VCD input module for more details and motivation.
 */

static void keep_header_for_reread(const struct sr_input *in)
{
	struct context *inc;

	inc = in->priv;
	g_slist_free_full(inc->prev_sr_channels, sr_channel_free_cb);
	inc->prev_sr_channels = in->sdi->channels;
	in->sdi->channels = NULL;
}

static int check_header_in_reread(const struct sr_input *in)
{
	struct context *inc;

	if (!in)
		return FALSE;
	inc = in->priv;
	if (!inc)
		return FALSE;
	if (!inc->prev_sr_channels)
		return TRUE;

	if (sr_channel_lists_differ(inc->prev_sr_channels, in->sdi->channels)) {
		sr_err("Channel list change not supported for file re-read.");
		return FALSE;
	}
	g_slist_free_full(in->sdi->channels, sr_channel_free_cb);
	in->sdi->channels = inc->prev_sr_channels;
	inc->prev_sr_channels = NULL;

	return TRUE;
}

static int receive(struct sr_input *in, GString *buf)
{
	struct context *inc;
	int ret;
	char channelname[16];

	g_string_append_len(in->buf, buf->str, buf->len);

	if (in->buf->len < MIN_DATA_CHUNK_OFFSET) {
		/*
		 * Don't even try until there's enough room
		 * for the data segment to start.
		 */
		return SR_OK;
	}

	inc = in->priv;
	if (!in->sdi_ready) {
		if ((ret = parse_wav_header(in->buf, inc)) == SR_ERR_NA)
			/* Not enough data yet. */
			return SR_OK;
		else if (ret != SR_OK)
			return ret;

		if (inc->create_channels) {
			for (int i = 0; i < inc->num_channels; i++) {
				snprintf(channelname, sizeof(channelname), "CH%d", i + 1);
				sr_channel_new(in->sdi, i, SR_CHANNEL_ANALOG, TRUE, channelname);
			}
			if (!check_header_in_reread(in))
				return SR_ERR_DATA;
		}
		inc->create_channels = FALSE;

		/* sdi is ready, notify frontend. */
		in->sdi_ready = TRUE;
		return SR_OK;
	}

	ret = process_buffer(in);

	return ret;
}

static int end(struct sr_input *in)
{
	struct context *inc;
	int ret;

	if (in->sdi_ready)
		ret = process_buffer(in);
	else
		ret = SR_OK;

	inc = in->priv;
	if (inc->started)
		std_session_send_df_end(in->sdi);

	return ret;
}

static int reset(struct sr_input *in)
{
	struct context *inc;

	inc = in->priv;
	memset(inc, 0, sizeof(*inc));

	/*
	 * Create, and re-create channels for every iteration of file
	 * import. Other logic will enforce a consistent set of channels
	 * across re-import, or an appropriate error message when file
	 * properties should change.
	 */
	keep_header_for_reread(in);
	inc->create_channels = TRUE;

	g_string_truncate(in->buf, 0);

	return SR_OK;
}

SR_PRIV struct sr_input_module input_wav = {
	.id = "wav",
	.name = "WAV",
	.desc = "Microsoft WAV file format data",
	.exts = (const char*[]){"wav", NULL},
	.metadata = { SR_INPUT_META_HEADER | SR_INPUT_META_REQUIRED },
	.format_match = format_match,
	.init = init,
	.receive = receive,
	.end = end,
	.reset = reset,
};
Commit	Line	Data
1d36b4d2	1	/*
50985c20	2	* This file is part of the libsigrok project.
1d36b4d2 BV	3	*
	4	* Copyright (C) 2013 Bert Vermeulen <bert@biot.com>
	5	*
	6	* This program is free software: you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License as published by
	8	* the Free Software Foundation, either version 3 of the License, or
	9	* (at your option) any later version.
	10	*
	11	* This program is distributed in the hope that it will be useful,
	12	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	14	* GNU General Public License for more details.
	15	*
	16	* You should have received a copy of the GNU General Public License
	17	* along with this program. If not, see <http://www.gnu.org/licenses/>.
	18	*/
	19
6ec6c43b	20	#include <config.h>
1d36b4d2 BV	21	#include <sys/types.h>
	22	#include <sys/stat.h>
	23	#include <unistd.h>
	24	#include <fcntl.h>
cb41a838	25	#include <ctype.h>
1d36b4d2	26	#include <string.h>
962d4344	27	#include <stdint.h>
c1aae900	28	#include <libsigrok/libsigrok.h>
1d36b4d2 BV	29	#include "libsigrok-internal.h"
1d36b4d2 BV	30
3544f848	31	#define LOG_PREFIX "input/wav"
1d36b4d2	32
cb41a838	33	/* How many bytes at a time to process and send to the session bus. */
9a4fd01a	34	#define CHUNK_SIZE (1 * 1024 * 1024 * sizeof(float))
17bfaca6 BV	35
	36	/* Minimum size of header + 1 8-bit mono PCM sample. */
	37	#define MIN_DATA_CHUNK_OFFSET 45
	38
cb41a838	39	/* Expect to find the "data" chunk within this offset from the start. */
b944e336	40	#define MAX_DATA_CHUNK_OFFSET 1024
cb41a838	41
76598cda UH	42	#define WAVE_FORMAT_PCM_ 0x0001
	43	#define WAVE_FORMAT_IEEE_FLOAT_ 0x0003
	44	#define WAVE_FORMAT_EXTENSIBLE_ 0xfffe
1d36b4d2 BV	45
1d36b4d2 BV	46	struct context {
d0181813	47	gboolean started;
17bfaca6	48	int fmt_code;
1d36b4d2 BV	49	uint64_t samplerate;
	50	int samplesize;
	51	int num_channels;
cb41a838	52	int unitsize;
17bfaca6	53	gboolean found_data;
9e850040	54	gboolean create_channels;
2581d204	55	GSList *prev_sr_channels;
1d36b4d2 BV	56	};
1d36b4d2 BV	57
17bfaca6	58	static int parse_wav_header(GString buf, struct context inc)
1d36b4d2	59	{
17bfaca6	60	uint64_t samplerate;
cbd9e6e9	61	unsigned int fmt_code, samplesize, num_channels, unitsize;
1d36b4d2	62
cbd9e6e9	63	if (buf->len < MIN_DATA_CHUNK_OFFSET)
d0181813	64	return SR_ERR_NA;
1d36b4d2	65
962d4344 BV	66	fmt_code = RL16(buf->str + 20);
962d4344 BV	67	samplerate = RL32(buf->str + 24);
cbd9e6e9	68
962d4344 BV	69	samplesize = RL16(buf->str + 32);
962d4344 BV	70	num_channels = RL16(buf->str + 22);
cbd9e6e9 BV	71	if (num_channels == 0)
cbd9e6e9 BV	72	return SR_ERR;
17bfaca6	73	unitsize = samplesize / num_channels;
e8779db7 BV	74	if (unitsize != 1 && unitsize != 2 && unitsize != 4) {
	75	sr_err("Only 8, 16 or 32 bits per sample supported.");
	76	return SR_ERR_DATA;
	77	}
	78
76598cda UH	79	if (fmt_code == WAVE_FORMAT_PCM_) {
76598cda UH	80	} else if (fmt_code == WAVE_FORMAT_IEEE_FLOAT_) {
17bfaca6 BV	81	if (unitsize != 4) {
17bfaca6 BV	82	sr_err("only 32-bit floats supported.");
4f979115	83	return SR_ERR_DATA;
17bfaca6	84	}
76598cda	85	} else if (fmt_code == WAVE_FORMAT_EXTENSIBLE_) {
cbd9e6e9 BV	86	if (buf->len < 70)
cbd9e6e9 BV	87	/* Not enough for extensible header and next chunk. */
d0181813	88	return SR_ERR_NA;
cbd9e6e9 BV	89
	90	if (RL16(buf->str + 16) != 40) {
	91	sr_err("WAV extensible format chunk must be 40 bytes.");
	92	return SR_ERR;
	93	}
	94	if (RL16(buf->str + 36) != 22) {
	95	sr_err("WAV extension must be 22 bytes.");
	96	return SR_ERR;
	97	}
	98	if (RL16(buf->str + 34) != RL16(buf->str + 38)) {
	99	sr_err("Reduced valid bits per sample not supported.");
4f979115	100	return SR_ERR_DATA;
cbd9e6e9 BV	101	}
	102	/* Real format code is the first two bytes of the GUID. */
	103	fmt_code = RL16(buf->str + 44);
76598cda	104	if (fmt_code != WAVE_FORMAT_PCM_ && fmt_code != WAVE_FORMAT_IEEE_FLOAT_) {
cbd9e6e9	105	sr_err("Only PCM and floating point samples are supported.");
4f979115	106	return SR_ERR_DATA;
cbd9e6e9	107	}
76598cda	108	if (fmt_code == WAVE_FORMAT_IEEE_FLOAT_ && unitsize != 4) {
28d9df72 MC	109	sr_err("only 32-bit floats supported.");
	110	return SR_ERR_DATA;
	111	}
17bfaca6 BV	112	} else {
17bfaca6 BV	113	sr_err("Only PCM and floating point samples are supported.");
4f979115	114	return SR_ERR_DATA;
17bfaca6	115	}
1d36b4d2	116
17bfaca6 BV	117	if (inc) {
	118	inc->fmt_code = fmt_code;
	119	inc->samplerate = samplerate;
	120	inc->samplesize = samplesize;
	121	inc->num_channels = num_channels;
	122	inc->unitsize = unitsize;
	123	inc->found_data = FALSE;
	124	}
1d36b4d2 BV	125
	126	return SR_OK;
	127	}
	128
54ee427d	129	static int format_match(GHashTable metadata, unsigned int confidence)
1d36b4d2	130	{
17bfaca6	131	GString *buf;
4f979115	132	int ret;
1d36b4d2	133
17bfaca6 BV	134	buf = g_hash_table_lookup(metadata, GINT_TO_POINTER(SR_INPUT_META_HEADER));
17bfaca6 BV	135	if (strncmp(buf->str, "RIFF", 4))
4f979115	136	return SR_ERR;
17bfaca6	137	if (strncmp(buf->str + 8, "WAVE", 4))
4f979115	138	return SR_ERR;
17bfaca6	139	if (strncmp(buf->str + 12, "fmt ", 4))
4f979115	140	return SR_ERR;
17bfaca6 BV	141	/*
	142	* Only gets called when we already know this is a WAV file, so
	143	* this parser can log error messages.
	144	*/
4f979115 BV	145	if ((ret = parse_wav_header(buf, NULL)) != SR_OK)
4f979115 BV	146	return ret;
1d36b4d2	147
54ee427d GS	148	*confidence = 1;
54ee427d GS	149
4f979115	150	return SR_OK;
1d36b4d2 BV	151	}
1d36b4d2 BV	152
17bfaca6	153	static int init(struct sr_input in, GHashTable options)
1d36b4d2	154	{
9e850040 SA	155	struct context *inc;
9e850040 SA	156
17bfaca6	157	(void)options;
1d36b4d2	158
aac29cc1	159	in->sdi = g_malloc0(sizeof(struct sr_dev_inst));
d0181813	160	in->priv = g_malloc0(sizeof(struct context));
9e850040 SA	161	inc = in->priv;
	162
	163	inc->create_channels = TRUE;
1d36b4d2 BV	164
	165	return SR_OK;
	166	}
	167
17bfaca6	168	static int find_data_chunk(GString *buf, int initial_offset)
cb41a838	169	{
17bfaca6	170	unsigned int offset, i;
cb41a838 BV	171
cb41a838 BV	172	offset = initial_offset;
0c5f2abc	173	while (offset < MIN(MAX_DATA_CHUNK_OFFSET, buf->len)) {
17bfaca6	174	if (!memcmp(buf->str + offset, "data", 4))
cb41a838 BV	175	/* Skip into the samples. */
	176	return offset + 8;
	177	for (i = 0; i < 4; i++) {
94b138a3	178	if (!isalnum(buf->str[offset + i])
17bfaca6	179	&& !isblank(buf->str[offset + i]))
cb41a838 BV	180	/* Doesn't look like a chunk ID. */
	181	return -1;
	182	}
	183	/* Skip past this chunk. */
962d4344	184	offset += 8 + RL32(buf->str + offset + 4);
cb41a838 BV	185	}
cb41a838 BV	186
b944e336 SB	187	if (offset > MAX_DATA_CHUNK_OFFSET)
	188	return -1;
	189
cb41a838 BV	190	return offset;
	191	}
	192
17bfaca6 BV	193	static void send_chunk(const struct sr_input *in, int offset, int num_samples)
	194	{
	195	struct sr_datafeed_packet packet;
0f33aaef UH	196	struct sr_datafeed_analog analog;
	197	struct sr_analog_encoding encoding;
	198	struct sr_analog_meaning meaning;
	199	struct sr_analog_spec spec;
17bfaca6	200	struct context *inc;
2cff7a2b	201	float *fdata;
17bfaca6 BV	202	int total_samples, samplenum;
17bfaca6 BV	203	char s, d;
1d36b4d2	204
7db06394	205	inc = in->priv;
17bfaca6	206
5eb39a91 SA	207	total_samples = num_samples * inc->num_channels;
5eb39a91 SA	208	fdata = g_malloc0(total_samples * sizeof(float));
17bfaca6 BV	209	s = in->buf->str + offset;
17bfaca6 BV	210	d = (char *)fdata;
5eb39a91	211
17bfaca6	212	for (samplenum = 0; samplenum < total_samples; samplenum++) {
76598cda	213	if (inc->fmt_code == WAVE_FORMAT_PCM_) {
8b5aefc6	214	switch (inc->unitsize) {
17bfaca6 BV	215	case 1:
17bfaca6 BV	216	/* 8-bit PCM samples are unsigned. */
8b5aefc6	217	fdata[samplenum] = (uint8_t)(s) / (float)255;
17bfaca6 BV	218	break;
17bfaca6 BV	219	case 2:
8b5aefc6	220	fdata[samplenum] = RL16S(s) / (float)INT16_MAX;
17bfaca6 BV	221	break;
17bfaca6 BV	222	case 4:
8b5aefc6	223	fdata[samplenum] = RL32S(s) / (float)INT32_MAX;
17bfaca6 BV	224	break;
	225	}
	226	} else {
	227	/* BINARY32 float */
cb41a838	228	#ifdef WORDS_BIGENDIAN
28d9df72	229	int i;
17bfaca6	230	for (i = 0; i < inc->unitsize; i++)
28d9df72	231	d[i] = s[inc->unitsize - 1 - i];
cb41a838	232	#else
17bfaca6	233	memcpy(d, s, inc->unitsize);
cb41a838	234	#endif
1d36b4d2	235	}
17bfaca6 BV	236	s += inc->unitsize;
	237	d += inc->unitsize;
	238	}
0f33aaef	239
7dcaddd3 UH	240	/* TODO: Use proper 'digits' value for this device (and its modes). */
7dcaddd3 UH	241	sr_analog_init(&analog, &encoding, &meaning, &spec, 2);
0f33aaef	242	packet.type = SR_DF_ANALOG;
17bfaca6	243	packet.payload = &analog;
17bfaca6	244	analog.num_samples = num_samples;
17bfaca6	245	analog.data = fdata;
0f33aaef UH	246	analog.meaning->channels = in->sdi->channels;
	247	analog.meaning->mq = 0;
	248	analog.meaning->mqflags = 0;
	249	analog.meaning->unit = 0;
17bfaca6	250	sr_session_send(in->sdi, &packet);
2cff7a2b	251	g_free(fdata);
17bfaca6 BV	252	}
17bfaca6 BV	253
7066fd46	254	static int process_buffer(struct sr_input *in)
17bfaca6	255	{
7066fd46	256	struct context *inc;
d0181813	257	int offset, chunk_samples, total_samples, processed, max_chunk_samples;
7066fd46	258	int num_samples, i;
17bfaca6	259
d0181813	260	inc = in->priv;
d0181813	261	if (!inc->started) {
bee2b016	262	std_session_send_df_header(in->sdi);
f8a8d4bb GS	263	(void)sr_session_send_meta(in->sdi, SR_CONF_SAMPLERATE,
f8a8d4bb GS	264	g_variant_new_uint64(inc->samplerate));
d0181813	265	inc->started = TRUE;
17bfaca6	266	}
17bfaca6 BV	267
	268	if (!inc->found_data) {
	269	/* Skip past size of 'fmt ' chunk. */
962d4344	270	i = 20 + RL32(in->buf->str + 16);
17bfaca6 BV	271	offset = find_data_chunk(in->buf, i);
	272	if (offset < 0) {
	273	if (in->buf->len > MAX_DATA_CHUNK_OFFSET) {
	274	sr_err("Couldn't find data chunk.");
	275	return SR_ERR;
	276	}
	277	}
	278	inc->found_data = TRUE;
	279	} else
	280	offset = 0;
	281
	282	/* Round off up to the last channels * unitsize boundary. */
288f8ce2 SB	283	chunk_samples = (in->buf->len - offset) / inc->samplesize;
288f8ce2 SB	284	max_chunk_samples = CHUNK_SIZE / inc->samplesize;
17bfaca6 BV	285	processed = 0;
	286	total_samples = chunk_samples;
	287	while (processed < total_samples) {
	288	if (chunk_samples > max_chunk_samples)
	289	num_samples = max_chunk_samples;
	290	else
	291	num_samples = chunk_samples;
	292	send_chunk(in, offset, num_samples);
288f8ce2	293	offset += num_samples * inc->samplesize;
17bfaca6 BV	294	chunk_samples -= num_samples;
	295	processed += num_samples;
	296	}
	297
	298	if ((unsigned int)offset < in->buf->len) {
	299	/*
	300	* The incoming buffer wasn't processed completely. Stash
	301	* the leftover data for next time.
	302	*/
	303	g_string_erase(in->buf, 0, offset);
	304	} else
	305	g_string_truncate(in->buf, 0);
1d36b4d2 BV	306
	307	return SR_OK;
	308	}
	309
2581d204 GS	310	/*
	311	* Check the channel list for consistency across file re-import. See
	312	* the VCD input module for more details and motivation.
	313	*/
	314
	315	static void keep_header_for_reread(const struct sr_input *in)
	316	{
	317	struct context *inc;
	318
	319	inc = in->priv;
	320	g_slist_free_full(inc->prev_sr_channels, sr_channel_free_cb);
	321	inc->prev_sr_channels = in->sdi->channels;
	322	in->sdi->channels = NULL;
	323	}
	324
	325	static int check_header_in_reread(const struct sr_input *in)
	326	{
	327	struct context *inc;
	328
	329	if (!in)
	330	return FALSE;
	331	inc = in->priv;
	332	if (!inc)
	333	return FALSE;
	334	if (!inc->prev_sr_channels)
	335	return TRUE;
	336
	337	if (sr_channel_lists_differ(inc->prev_sr_channels, in->sdi->channels)) {
	338	sr_err("Channel list change not supported for file re-read.");
	339	return FALSE;
	340	}
	341	g_slist_free_full(in->sdi->channels, sr_channel_free_cb);
	342	in->sdi->channels = inc->prev_sr_channels;
	343	inc->prev_sr_channels = NULL;
	344
	345	return TRUE;
	346	}
	347
7066fd46 BV	348	static int receive(struct sr_input in, GString buf)
	349	{
	350	struct context *inc;
	351	int ret;
19d816c5	352	char channelname[16];
7066fd46 BV	353
	354	g_string_append_len(in->buf, buf->str, buf->len);
	355
	356	if (in->buf->len < MIN_DATA_CHUNK_OFFSET) {
	357	/*
	358	* Don't even try until there's enough room
	359	* for the data segment to start.
	360	*/
	361	return SR_OK;
	362	}
	363
	364	inc = in->priv;
	365	if (!in->sdi_ready) {
	366	if ((ret = parse_wav_header(in->buf, inc)) == SR_ERR_NA)
	367	/* Not enough data yet. */
	368	return SR_OK;
	369	else if (ret != SR_OK)
	370	return ret;
	371
9e850040 SA	372	if (inc->create_channels) {
	373	for (int i = 0; i < inc->num_channels; i++) {
	374	snprintf(channelname, sizeof(channelname), "CH%d", i + 1);
	375	sr_channel_new(in->sdi, i, SR_CHANNEL_ANALOG, TRUE, channelname);
	376	}
2581d204 GS	377	if (!check_header_in_reread(in))
2581d204 GS	378	return SR_ERR_DATA;
74c9a8d2	379	}
9e850040 SA	380	inc->create_channels = FALSE;
9e850040 SA	381
7066fd46 BV	382	/* sdi is ready, notify frontend. */
	383	in->sdi_ready = TRUE;
	384	return SR_OK;
	385	}
	386
	387	ret = process_buffer(in);
	388
18078d05	389	return ret;
7066fd46 BV	390	}
	391
	392	static int end(struct sr_input *in)
7db06394	393	{
d0181813	394	struct context *inc;
7066fd46 BV	395	int ret;
	396
	397	if (in->sdi_ready)
	398	ret = process_buffer(in);
	399	else
	400	ret = SR_OK;
0a4d68f7	401
d0181813	402	inc = in->priv;
3be42bc2	403	if (inc->started)
bee2b016	404	std_session_send_df_end(in->sdi);
7db06394	405
7066fd46	406	return ret;
7db06394 BV	407	}
7db06394 BV	408
ab4c27cf SA	409	static int reset(struct sr_input *in)
ab4c27cf SA	410	{
2581d204 GS	411	struct context *inc;
	412
	413	inc = in->priv;
	414	memset(inc, 0, sizeof(*inc));
9e850040 SA	415
9e850040 SA	416	/*
2581d204 GS	417	* Create, and re-create channels for every iteration of file
	418	* import. Other logic will enforce a consistent set of channels
	419	* across re-import, or an appropriate error message when file
	420	* properties should change.
9e850040	421	*/
2581d204 GS	422	keep_header_for_reread(in);
2581d204 GS	423	inc->create_channels = TRUE;
ab4c27cf	424
ab4c27cf SA	425	g_string_truncate(in->buf, 0);
	426
	427	return SR_OK;
	428	}
	429
d4c93774	430	SR_PRIV struct sr_input_module input_wav = {
1d36b4d2	431	.id = "wav",
17bfaca6	432	.name = "WAV",
b20eb520	433	.desc = "Microsoft WAV file format data",
c7bc82ff	434	.exts = (const char*[]){"wav", NULL},
17bfaca6	435	.metadata = { SR_INPUT_META_HEADER \| SR_INPUT_META_REQUIRED },
1d36b4d2 BV	436	.format_match = format_match,
1d36b4d2 BV	437	.init = init,
17bfaca6	438	.receive = receive,
7066fd46	439	.end = end,
ab4c27cf	440	.reset = reset,
1d36b4d2	441	};