trx/rx.c

231 lines
4.8 KiB
C
Raw Normal View History

2009-09-05 00:04:44 +02:00
#include <netdb.h>
#include <string.h>
#include <alsa/asoundlib.h>
#include <celt/celt.h>
2009-09-11 17:40:27 +02:00
#include <ortp/ortp.h>
2009-09-05 00:04:44 +02:00
#include <sys/socket.h>
#include <sys/types.h>
#include "defaults.h"
#include "device.h"
#include "sched.h"
2009-09-11 17:40:27 +02:00
static unsigned int verbose = DEFAULT_VERBOSE;
2009-09-05 00:04:44 +02:00
2009-09-11 17:40:27 +02:00
static RtpSession* create_rtp_recv(const char *addr_desc, const int port,
unsigned int jitter)
2009-09-05 00:04:44 +02:00
{
2009-09-11 17:40:27 +02:00
RtpSession *session;
session = rtp_session_new(RTP_SESSION_RECVONLY);
rtp_session_set_scheduling_mode(session, 0);
rtp_session_set_blocking_mode(session, 0);
rtp_session_set_local_addr(session, addr_desc, port);
rtp_session_set_connected_mode(session, FALSE);
rtp_session_enable_adaptive_jitter_compensation(session, TRUE);
rtp_session_set_jitter_compensation(session, jitter); /* ms */
if (rtp_session_set_payload_type(session, 0) != 0)
abort();
if (rtp_session_signal_connect(session, "timestamp_jump",
(RtpCallback)rtp_session_resync, 0) != 0)
{
abort();
2009-09-05 00:04:44 +02:00
}
2009-09-11 17:40:27 +02:00
return session;
2009-09-05 00:04:44 +02:00
}
2009-09-11 17:40:27 +02:00
static int play_one_frame(void *packet,
size_t len,
CELTDecoder *decoder,
snd_pcm_t *snd,
const unsigned int channels,
const snd_pcm_uframes_t samples)
2009-09-05 00:04:44 +02:00
{
int r;
float *pcm;
snd_pcm_sframes_t f;
2009-09-11 17:40:27 +02:00
pcm = alloca(sizeof(float) * samples * channels);
2009-09-05 00:04:44 +02:00
2009-09-11 17:40:27 +02:00
if (packet == NULL) {
2009-09-05 00:04:44 +02:00
r = celt_decode_float(decoder, NULL, 0, pcm);
} else {
2009-09-11 17:40:27 +02:00
r = celt_decode_float(decoder, packet, len, pcm);
2009-09-05 00:04:44 +02:00
}
if (r != 0) {
fputs("Error in celt_decode\n", stderr);
return -1;
}
2009-09-11 17:40:27 +02:00
f = snd_pcm_writei(snd, pcm, samples);
2009-09-05 00:04:44 +02:00
if (f < 0) {
aerror("snd_pcm_writei", f);
return -1;
}
2009-09-11 17:40:27 +02:00
if (f < samples)
2009-09-05 00:04:44 +02:00
fprintf(stderr, "Short write %ld\n", f);
return 0;
}
2009-09-11 17:40:27 +02:00
static int run_rx(RtpSession *session,
CELTDecoder *decoder,
snd_pcm_t *snd,
const unsigned int channels,
const snd_pcm_uframes_t frame)
2009-09-05 00:04:44 +02:00
{
2009-09-11 17:40:27 +02:00
int ts = 0;
2009-09-05 00:04:44 +02:00
for (;;) {
2009-09-11 17:40:27 +02:00
int r, have_more;
char buf[32768];
void *packet;
r = rtp_session_recv_with_ts(session, buf,
sizeof(buf), ts, &have_more);
assert(r >= 0);
assert(have_more == 0);
if (r == 0) {
packet = NULL;
if (verbose > 1)
fputc('#', stderr);
} else {
packet = buf;
if (verbose > 1)
fputc('.', stderr);
2009-09-05 00:04:44 +02:00
}
2009-09-11 17:40:27 +02:00
r = play_one_frame(packet, r, decoder, snd, channels, frame);
if (r == -1)
2009-09-05 00:04:44 +02:00
return -1;
2009-09-11 17:40:27 +02:00
ts += frame;
}
2009-09-05 00:04:44 +02:00
}
static void usage(FILE *fd)
{
fprintf(fd, "Usage: rx [<parameters>]\n");
fprintf(fd, "\nAudio device (ALSA) parameters:\n");
2009-09-11 17:40:27 +02:00
fprintf(fd, " -d <dev> Device name (default '%s')\n",
2009-09-05 00:04:44 +02:00
DEFAULT_DEVICE);
2009-09-11 17:40:27 +02:00
fprintf(fd, " -m <ms> Buffer time (milliseconds, default %d)\n",
2009-09-05 00:04:44 +02:00
DEFAULT_BUFFER);
fprintf(fd, "\nNetwork parameters:\n");
2009-09-11 17:40:27 +02:00
fprintf(fd, " -h <addr> IP address to listen on (default %s)\n",
DEFAULT_ADDR);
fprintf(fd, " -p <port> UDP port number (default %d)\n",
DEFAULT_PORT);
fprintf(fd, " -j <ms> Jitter buffer (milliseconds, default %d)\n",
DEFAULT_JITTER);
fprintf(fd, "\nEncoding parameters (must match sender):\n");
fprintf(fd, " -r <rate> Sample rate (default %d)\n",
DEFAULT_RATE);
fprintf(fd, " -c <n> Number of channels (default %d)\n",
DEFAULT_CHANNELS);
fprintf(fd, " -f <bytes> Frame size (default %d)\n",
DEFAULT_FRAME);
fprintf(fd, "\nDisplay parameters:\n");
fprintf(fd, " -v <n> Verbosity level (default %d)\n",
DEFAULT_VERBOSE);
2009-09-05 00:04:44 +02:00
}
int main(int argc, char *argv[])
{
2009-09-11 17:40:27 +02:00
int r;
2009-09-05 00:04:44 +02:00
snd_pcm_t *snd;
CELTMode *mode;
CELTDecoder *decoder;
2009-09-11 17:40:27 +02:00
RtpSession *session;
2009-09-05 00:04:44 +02:00
/* command-line options */
const char *device = DEFAULT_DEVICE,
2009-09-11 17:40:27 +02:00
*addr = DEFAULT_ADDR;
unsigned int buffer = DEFAULT_BUFFER,
rate = DEFAULT_RATE,
frame = DEFAULT_FRAME,
jitter = DEFAULT_JITTER,
channels = DEFAULT_CHANNELS,
port = DEFAULT_PORT;
2009-09-05 00:04:44 +02:00
for (;;) {
int c;
2009-09-11 17:40:27 +02:00
c = getopt(argc, argv, "d:h:j:m:p:v:");
2009-09-05 00:04:44 +02:00
if (c == -1)
break;
switch (c) {
case 'd':
device = optarg;
break;
2009-09-11 17:40:27 +02:00
case 'h':
addr = optarg;
break;
case 'j':
jitter = atoi(optarg);
break;
2009-09-05 00:04:44 +02:00
case 'm':
buffer = atoi(optarg);
break;
case 'p':
2009-09-11 17:40:27 +02:00
port = atoi(optarg);
break;
case 'v':
verbose = atoi(optarg);
2009-09-05 00:04:44 +02:00
break;
default:
usage(stderr);
return -1;
}
}
2009-09-11 17:40:27 +02:00
mode = celt_mode_create(rate, channels, frame, NULL);
2009-09-05 00:04:44 +02:00
if (mode == NULL) {
fputs("celt_mode_create failed\n", stderr);
return -1;
}
decoder = celt_decoder_create(mode);
if (decoder == NULL) {
fputs("celt_decoder_create failed\n", stderr);
return -1;
}
2009-09-11 17:40:27 +02:00
if (go_realtime() != 0)
2009-09-05 00:04:44 +02:00
return -1;
2009-09-11 17:40:27 +02:00
ortp_init();
ortp_scheduler_init();
session = create_rtp_recv(addr, port, jitter);
assert(session != NULL);
2009-09-05 00:04:44 +02:00
r = snd_pcm_open(&snd, device, SND_PCM_STREAM_PLAYBACK, 0);
if (r < 0) {
aerror("snd_pcm_open", r);
return -1;
}
2009-09-11 17:40:27 +02:00
if (set_alsa_hw(snd, rate, channels, buffer * 1000) == -1)
2009-09-05 00:04:44 +02:00
return -1;
if (set_alsa_sw(snd) == -1)
return -1;
2009-09-11 17:40:27 +02:00
r = run_rx(session, decoder, snd, channels, frame);
2009-09-05 00:04:44 +02:00
if (snd_pcm_close(snd) < 0)
abort();
2009-09-11 17:40:27 +02:00
rtp_session_destroy(session);
ortp_exit();
ortp_global_stats_display();
2009-09-05 00:04:44 +02:00
celt_decoder_destroy(decoder);
celt_mode_destroy(mode);
return r;
}