X-Git-Url: http://git.tuebingen.mpg.de/?a=blobdiff_plain;f=play.c;h=d98a4baaaf19dfdfa4a634a93972f22c01558715;hb=924db525a63d5aac65c09f2e8c6b72a733854414;hp=38203b401c710f4629285ad81b74c67d4eb9c8f0;hpb=cda80f7d3b0a8adf778028ac5a33496b7123ef8b;p=paraslash.git

diff --git a/play.c b/play.c
index 38203b40..d98a4baa 100644
--- a/play.c
+++ b/play.c
@@ -33,18 +33,42 @@
 
 #define FORMAT SND_PCM_FORMAT_S16_LE
 
-static snd_pcm_t *handle;
+struct private_alsa_data {
+	snd_pcm_t *handle;
+	size_t bytes_per_frame;
+};
+
+struct writer_node {
+	struct writer *writer;
+	void *private_data;
+	int chunk_bytes;
+};
+
+struct writer {
+	int (*open)(struct writer_node *);
+	int (*write)(char *data, size_t nbytes, struct writer_node *);
+	void (*close)(struct writer_node *);
+	void (*shutdown)(struct writer_node *);
+};
+
+#define NUM_WRITERS 1
+static struct writer writers[NUM_WRITERS];
+#define FOR_EACH_WRITER(i) for (i = 0; i < NUM_WRITERS, i++)
+static struct writer_node *writer_nodes;
+
+
 static unsigned char *audiobuf;
-static size_t bytes_per_frame;
 static struct timeval *start_time;
 static struct gengetopt_args_info conf;
 
 INIT_PLAY_ERRLISTS;
 
-void para_log(__a_unused int ll, const char* fmt,...)
+void para_log(int ll, const char* fmt,...)
 {
 	va_list argp;
 
+	if (ll < conf.loglevel_arg)
+		return;
 	va_start(argp, fmt);
 	vfprintf(stderr, fmt, argp);
 	va_end(argp);
@@ -74,7 +98,7 @@ static int read_wav_header(void)
  *
  * Install PCM software and hardware configuration. Exit on errors.
  */
-static int alsa_init(void)
+static int alsa_open(struct writer_node *w)
 {
 	snd_pcm_hw_params_t *hwparams;
 	snd_pcm_sw_params_t *swparams;
@@ -85,68 +109,73 @@ static int alsa_init(void)
 	snd_pcm_info_t *info;
 	snd_output_t *log;
 	snd_pcm_uframes_t period_size;
+	struct private_alsa_data *pad = para_malloc(sizeof(struct private_alsa_data));
+	w->private_data = pad;
 
 	snd_pcm_info_alloca(&info);
 	if (snd_output_stdio_attach(&log, stderr, 0) < 0)
 		return -E_ALSA_LOG;
-	err = snd_pcm_open(&handle, conf.device_arg,
+	err = snd_pcm_open(&pad->handle, conf.device_arg,
 		SND_PCM_STREAM_PLAYBACK, 0);
 	if (err < 0)
 		return -E_PCM_OPEN;
-	if ((err = snd_pcm_info(handle, info)) < 0)
+	if ((err = snd_pcm_info(pad->handle, info)) < 0)
 		return -E_SND_PCM_INFO;
 
 	snd_pcm_hw_params_alloca(&hwparams);
 	snd_pcm_sw_params_alloca(&swparams);
-	if (snd_pcm_hw_params_any(handle, hwparams) < 0)
+	if (snd_pcm_hw_params_any(pad->handle, hwparams) < 0)
 		return -E_BROKEN_CONF;
-	if (snd_pcm_hw_params_set_access(handle, hwparams,
+	if (snd_pcm_hw_params_set_access(pad->handle, hwparams,
 			SND_PCM_ACCESS_RW_INTERLEAVED) < 0)
 		return -E_ACCESS_TYPE;
-	if (snd_pcm_hw_params_set_format(handle, hwparams, FORMAT) < 0)
+	if (snd_pcm_hw_params_set_format(pad->handle, hwparams, FORMAT) < 0)
 		return -E_SAMPLE_FORMAT;
-	if (snd_pcm_hw_params_set_channels(handle, hwparams, conf.channels_arg) < 0)
+	if (snd_pcm_hw_params_set_channels(pad->handle, hwparams,
+			conf.channels_arg) < 0)
 		return -E_CHANNEL_COUNT;
-	if (snd_pcm_hw_params_set_rate_near(handle, hwparams,
+	if (snd_pcm_hw_params_set_rate_near(pad->handle, hwparams,
 			(unsigned int*) &conf.sample_rate_arg, 0) < 0)
 		return -E_SET_RATE;
 	err = snd_pcm_hw_params_get_buffer_time_max(hwparams, &buffer_time, 0);
 	if (err < 0 || !buffer_time)
 		return -E_GET_BUFFER_TIME;
 	PARA_DEBUG_LOG("buffer time: %d\n", buffer_time);
-	if (snd_pcm_hw_params_set_buffer_time_near(handle, hwparams,
+	if (snd_pcm_hw_params_set_buffer_time_near(pad->handle, hwparams,
 			&buffer_time, 0) < 0)
 		return -E_SET_BUFFER_TIME;
-	if (snd_pcm_hw_params(handle, hwparams) < 0)
+	if (snd_pcm_hw_params(pad->handle, hwparams) < 0)
 		return -E_HW_PARAMS;
 	snd_pcm_hw_params_get_period_size(hwparams, &period_size, 0);
 	snd_pcm_hw_params_get_buffer_size(hwparams, &buffer_size);
-	PARA_DEBUG_LOG("buffer size: %lu, period_size: %lu\n", buffer_size, period_size);
+	PARA_DEBUG_LOG("buffer size: %lu, period_size: %lu\n", buffer_size,
+		period_size);
 	if (period_size == buffer_size)
 		return -E_BAD_PERIOD;
-	snd_pcm_sw_params_current(handle, swparams);
+	snd_pcm_sw_params_current(pad->handle, swparams);
 	err = snd_pcm_sw_params_get_xfer_align(swparams, &xfer_align);
 	if (err < 0 || !xfer_align)
 		return -E_GET_XFER;
-//	snd_pcm_sw_params_set_sleep_min(handle, swparams, 0);
-	snd_pcm_sw_params_set_avail_min(handle, swparams, period_size);
+	snd_pcm_sw_params_set_avail_min(pad->handle, swparams, period_size);
 	/* round to closest transfer boundary */
 	start_threshold = (buffer_size / xfer_align) * xfer_align;
 	if (start_threshold < 1)
 		start_threshold = 1;
-	if (snd_pcm_sw_params_set_start_threshold(handle, swparams,
+	if (snd_pcm_sw_params_set_start_threshold(pad->handle, swparams,
 			start_threshold) < 0)
 		return -E_START_THRESHOLD;
 	stop_threshold = buffer_size;
-	if (snd_pcm_sw_params_set_stop_threshold(handle, swparams,
+	if (snd_pcm_sw_params_set_stop_threshold(pad->handle, swparams,
 			stop_threshold) < 0)
 		return -E_STOP_THRESHOLD;
-	if (snd_pcm_sw_params_set_xfer_align(handle, swparams, xfer_align) < 0)
+	if (snd_pcm_sw_params_set_xfer_align(pad->handle, swparams,
+			xfer_align) < 0)
 		return -E_SET_XFER;
-	if (snd_pcm_sw_params(handle, swparams) < 0)
+	if (snd_pcm_sw_params(pad->handle, swparams) < 0)
 		return -E_SW_PARAMS;
-	bytes_per_frame = snd_pcm_format_physical_width(FORMAT) * conf.channels_arg / 8;
-	return period_size * bytes_per_frame;
+	pad->bytes_per_frame = snd_pcm_format_physical_width(FORMAT)
+		* conf.channels_arg / 8;
+	return period_size * pad->bytes_per_frame;
 }
 
 /**
@@ -156,37 +185,51 @@ static int alsa_init(void)
  *
  * \return Number of bytes written, -E_ALSA_WRITE on errors.
  */
-static int alsa_write(u_char *data, size_t nbytes)
+static int alsa_write(char *data, size_t nbytes, struct writer_node *wn)
 {
-	size_t frames = nbytes / bytes_per_frame;
+	struct private_alsa_data *pad = wn->private_data;
+	size_t frames = nbytes / pad->bytes_per_frame;
+	unsigned char *d = data;
 	snd_pcm_sframes_t r, result = 0;
+
 	while (frames > 0) {
 		/* write interleaved frames */
-		r = snd_pcm_writei(handle, data, frames);
+		r = snd_pcm_writei(pad->handle, d, frames);
 		if (r < 0)
 			PARA_ERROR_LOG("write error: %s\n", snd_strerror(r));
 		if (r == -EAGAIN || (r >= 0 && r < frames))
-			snd_pcm_wait(handle, 1);
+			snd_pcm_wait(pad->handle, 1);
 		else if (r == -EPIPE)
-			snd_pcm_prepare(handle);
+			snd_pcm_prepare(pad->handle);
 		else if (r < 0)
 			return -E_ALSA_WRITE;
 		if (r > 0) {
 			result += r;
 			frames -= r;
-			data += r * bytes_per_frame;
+			d += r * pad->bytes_per_frame;
 		}
 	}
-	return result * bytes_per_frame;
+	return result * pad->bytes_per_frame;
 }
 
-static void alsa_shutdown(void)
+static void alsa_close(struct writer_node *wn)
 {
-	snd_pcm_drain(handle);
-	snd_pcm_close(handle);
+	struct private_alsa_data *pad = wn->private_data;
+	snd_pcm_drain(pad->handle);
+	snd_pcm_close(pad->handle);
 	snd_config_update_free_global();
+	free(pad);
+}
+
+void alsa_writer_init(struct writer *w)
+{
+	w->open = alsa_open;
+	w->write = alsa_write;
+	w->close = alsa_close;
+	w->shutdown = NULL; /* nothing to do */
 }
 
+
 /**
  * check if current time is later than start_time
  * \param diff pointer to write remaining time to
@@ -222,6 +265,22 @@ static void do_initial_delay(struct timeval *delay)
 	while (start_time_in_future(delay));
 }
 
+static int read_stdin(char *buf, size_t bytes_to_load, size_t *loaded)
+{
+	ssize_t ret;
+
+	while (*loaded < bytes_to_load) {
+		ret = read(STDIN_FILENO, buf + *loaded, bytes_to_load - *loaded);
+		if (ret <= 0) {
+			if (ret < 0)
+				ret = -E_READ_STDIN;
+			return ret;
+		}
+		*loaded += ret;
+	}
+	return 1;
+}
+
 /**
  * play raw pcm data
  * \param loaded number of bytes already loaded
@@ -234,50 +293,86 @@ static void do_initial_delay(struct timeval *delay)
  */
 static int play_pcm(size_t loaded)
 {
-	size_t bufsize, written = 0, prebuf_size;
-	unsigned char *p;
+	size_t bufsize, min_written = 0, prebuf_size, bytes_to_load;
 	struct timeval delay;
-	int chunk_bytes, ret = alsa_init();
+	int eof = 0, i, max_chunk_bytes = 0, ret, not_yet_started = 1, need_more_writes;
+	struct writer_node *wn;
+	size_t *written = para_calloc(1 * sizeof(size_t));
 
-	if (ret < 0)
-		goto out;
-	chunk_bytes = ret;
-	bufsize = (conf.bufsize_arg * 1024 / chunk_bytes) * chunk_bytes;
+	for (i = 0; i < 1; i++) {
+		wn = &writer_nodes[i];
+		ret = wn->writer->open(wn);
+		if (ret < 0)
+			goto out;
+		wn->chunk_bytes = ret;
+		max_chunk_bytes = PARA_MAX(max_chunk_bytes, ret);
+	}
+	PARA_INFO_LOG("max chunk_bytes: %d\n", max_chunk_bytes);
+	bufsize = (conf.bufsize_arg * 1024 / max_chunk_bytes) * max_chunk_bytes;
 	audiobuf = para_realloc(audiobuf, bufsize);
 	prebuf_size = conf.prebuffer_arg * bufsize / 100;
+	bytes_to_load =  PARA_MAX(prebuf_size, max_chunk_bytes);
+	ret = read_stdin(audiobuf, bytes_to_load, &loaded);
+	if (ret <= 0 || loaded < bytes_to_load) {
+		if (ret >= 0)
+			ret = -E_PREMATURE_END;
+		goto out;
+	}
+	if (not_yet_started && start_time && start_time_in_future(&delay))
+		do_initial_delay(&delay);
+	not_yet_started = 0;
 again:
-	if (!written) {
-		if (loaded < prebuf_size)
-			goto read;
-		if (start_time && start_time_in_future(&delay)) {
-			do_initial_delay(&delay);
-			start_time = NULL;
+	need_more_writes = 1;
+	while (need_more_writes) {
+		need_more_writes = 0;
+		for (i = 0; i < 1; i++) {
+			unsigned char *p = audiobuf + written[i];
+			int bytes_to_write;
+			wn = &writer_nodes[i];
+			if (!i)
+				min_written = written[i];
+			else
+				min_written = PARA_MIN(min_written, written[i]);
+			if (loaded == written[i])
+				continue;
+			if (!eof && (loaded < wn->chunk_bytes + written[i]))
+				continue;
+			bytes_to_write = PARA_MIN(wn->chunk_bytes,
+				loaded - written[i]);
+			ret = wn->writer->write(p, bytes_to_write, wn);
+			if (ret < 0)
+				goto out;
+			if (ret != bytes_to_write)
+				PARA_WARNING_LOG("short write: %d/%d\n", ret,
+					bytes_to_write);
+			written[i] += ret;
+			need_more_writes = 1;
 		}
 	}
-	p = audiobuf;
-	while (loaded >= chunk_bytes) {
-		ret = alsa_write(p, chunk_bytes);
-		if (ret < 0)
-			goto out;
-		p += ret;
-		written += ret;
-		loaded -= ret;
-	}
-	if (loaded && p != audiobuf)
-		memmove(audiobuf, p, loaded);
-read:
-	ret = read(STDIN_FILENO, audiobuf + loaded, bufsize - loaded);
-	if (ret < 0) {
-		ret = -E_READ_STDIN;
+	loaded -= min_written;
+	ret = 0;
+	if (eof)
+		goto out;
+	if (loaded >= bufsize) {
+		ret = -E_PLAY_OVERRUN;
 		goto out;
 	}
-	if (ret) {
-		loaded += ret;
-		goto again;
-	}
-	ret = 1;
+	memmove(audiobuf, audiobuf + min_written, loaded);
+	for (i = 0; i < 1; i++)
+		written[i] -= min_written;
+	bytes_to_load = PARA_MIN(max_chunk_bytes, bufsize);
+	ret = read_stdin(audiobuf, bytes_to_load, &loaded);
+	if (ret < 0)
+		goto out;
+	if (!ret)
+		eof = 1;
+	goto again;
 out:
-	alsa_shutdown();
+	free(written);
+	for (i = 0; i < 1; i++) {
+		wn = &writer_nodes[i];
+		wn->writer->close(wn);
+	}
 	return ret;
 }
 
@@ -300,16 +395,23 @@ static size_t check_wave(void)
 int main(int argc, char *argv[])
 {
 	struct timeval tv;
-	int ret;
+	int ret = -E_PLAY_SYNTAX;
 
 	cmdline_parser(argc, argv, &conf);
+	if (conf.prebuffer_arg < 0 || conf.prebuffer_arg > 100)
+		goto out;
 	if (conf.start_time_given) {
-		ret = -E_PLAY_SYNTAX;
 		if (sscanf(conf.start_time_arg, "%lu:%lu",
 				&tv.tv_sec, &tv.tv_usec) != 2)
 			goto out;
 		start_time = &tv;
 	}
+	/* call init for each supported writer */
+	alsa_writer_init(&writers[0]);
+	/* one for each given writer */
+	writer_nodes = para_calloc(2 * sizeof(struct writer_node));
+	writer_nodes[0].writer = &writers[0]; /* alsa */
+
 	audiobuf = para_malloc(WAV_HEADER_LEN);
 	ret = read_wav_header();
 	if (ret < 0)
@@ -317,6 +419,7 @@ int main(int argc, char *argv[])
 	ret = play_pcm(check_wave());
 out:
 	free(audiobuf);
+	free(writer_nodes);
 	if (ret < 0)
 		PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
 	return ret;