00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00028 #include <pulse/simple.h>
00029 #include <pulse/rtclock.h>
00030 #include <pulse/error.h>
00031
00032 #include "libavformat/avformat.h"
00033 #include "libavformat/internal.h"
00034 #include "libavutil/opt.h"
00035
00036 #define DEFAULT_CODEC_ID AV_NE(AV_CODEC_ID_PCM_S16BE, AV_CODEC_ID_PCM_S16LE)
00037
00038 typedef struct PulseData {
00039 AVClass *class;
00040 char *server;
00041 char *name;
00042 char *stream_name;
00043 int sample_rate;
00044 int channels;
00045 int frame_size;
00046 int fragment_size;
00047 pa_simple *s;
00048 int64_t pts;
00049 int64_t frame_duration;
00050 } PulseData;
00051
00052 static pa_sample_format_t codec_id_to_pulse_format(int codec_id) {
00053 switch (codec_id) {
00054 case AV_CODEC_ID_PCM_U8: return PA_SAMPLE_U8;
00055 case AV_CODEC_ID_PCM_ALAW: return PA_SAMPLE_ALAW;
00056 case AV_CODEC_ID_PCM_MULAW: return PA_SAMPLE_ULAW;
00057 case AV_CODEC_ID_PCM_S16LE: return PA_SAMPLE_S16LE;
00058 case AV_CODEC_ID_PCM_S16BE: return PA_SAMPLE_S16BE;
00059 case AV_CODEC_ID_PCM_F32LE: return PA_SAMPLE_FLOAT32LE;
00060 case AV_CODEC_ID_PCM_F32BE: return PA_SAMPLE_FLOAT32BE;
00061 case AV_CODEC_ID_PCM_S32LE: return PA_SAMPLE_S32LE;
00062 case AV_CODEC_ID_PCM_S32BE: return PA_SAMPLE_S32BE;
00063 case AV_CODEC_ID_PCM_S24LE: return PA_SAMPLE_S24LE;
00064 case AV_CODEC_ID_PCM_S24BE: return PA_SAMPLE_S24BE;
00065 default: return PA_SAMPLE_INVALID;
00066 }
00067 }
00068
00069 static av_cold int pulse_read_header(AVFormatContext *s)
00070 {
00071 PulseData *pd = s->priv_data;
00072 AVStream *st;
00073 char *device = NULL;
00074 int ret;
00075 enum AVCodecID codec_id =
00076 s->audio_codec_id == AV_CODEC_ID_NONE ? DEFAULT_CODEC_ID : s->audio_codec_id;
00077 const pa_sample_spec ss = { codec_id_to_pulse_format(codec_id),
00078 pd->sample_rate,
00079 pd->channels };
00080
00081 pa_buffer_attr attr = { -1 };
00082
00083 st = avformat_new_stream(s, NULL);
00084
00085 if (!st) {
00086 av_log(s, AV_LOG_ERROR, "Cannot add stream\n");
00087 return AVERROR(ENOMEM);
00088 }
00089
00090 attr.fragsize = pd->fragment_size;
00091
00092 if (strcmp(s->filename, "default"))
00093 device = s->filename;
00094
00095 pd->s = pa_simple_new(pd->server, pd->name,
00096 PA_STREAM_RECORD,
00097 device, pd->stream_name, &ss,
00098 NULL, &attr, &ret);
00099
00100 if (!pd->s) {
00101 av_log(s, AV_LOG_ERROR, "pa_simple_new failed: %s\n",
00102 pa_strerror(ret));
00103 return AVERROR(EIO);
00104 }
00105
00106 st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
00107 st->codec->codec_id = codec_id;
00108 st->codec->sample_rate = pd->sample_rate;
00109 st->codec->channels = pd->channels;
00110 avpriv_set_pts_info(st, 64, 1, 1000000);
00111
00112 pd->pts = AV_NOPTS_VALUE;
00113 pd->frame_duration = (pd->frame_size * 1000000LL * 8) /
00114 (pd->sample_rate * pd->channels * av_get_bits_per_sample(codec_id));
00115
00116 return 0;
00117 }
00118
00119 static int pulse_read_packet(AVFormatContext *s, AVPacket *pkt)
00120 {
00121 PulseData *pd = s->priv_data;
00122 int res;
00123 pa_usec_t latency;
00124
00125 if (av_new_packet(pkt, pd->frame_size) < 0) {
00126 return AVERROR(ENOMEM);
00127 }
00128
00129 if ((pa_simple_read(pd->s, pkt->data, pkt->size, &res)) < 0) {
00130 av_log(s, AV_LOG_ERROR, "pa_simple_read failed: %s\n",
00131 pa_strerror(res));
00132 av_free_packet(pkt);
00133 return AVERROR(EIO);
00134 }
00135
00136 if ((latency = pa_simple_get_latency(pd->s, &res)) == (pa_usec_t) -1) {
00137 av_log(s, AV_LOG_ERROR, "pa_simple_get_latency() failed: %s\n",
00138 pa_strerror(res));
00139 return AVERROR(EIO);
00140 }
00141
00142 if (pd->pts == AV_NOPTS_VALUE) {
00143 pd->pts = -latency;
00144 }
00145
00146 pkt->pts = pd->pts;
00147
00148 pd->pts += pd->frame_duration;
00149
00150 return 0;
00151 }
00152
00153 static av_cold int pulse_close(AVFormatContext *s)
00154 {
00155 PulseData *pd = s->priv_data;
00156 pa_simple_free(pd->s);
00157 return 0;
00158 }
00159
00160 #define OFFSET(a) offsetof(PulseData, a)
00161 #define D AV_OPT_FLAG_DECODING_PARAM
00162
00163 static const AVOption options[] = {
00164 { "server", "pulse server name", OFFSET(server), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, D },
00165 { "name", "application name", OFFSET(name), AV_OPT_TYPE_STRING, {.str = LIBAVFORMAT_IDENT}, 0, 0, D },
00166 { "stream_name", "stream description", OFFSET(stream_name), AV_OPT_TYPE_STRING, {.str = "record"}, 0, 0, D },
00167 { "sample_rate", "sample rate in Hz", OFFSET(sample_rate), AV_OPT_TYPE_INT, {.i64 = 48000}, 1, INT_MAX, D },
00168 { "channels", "number of audio channels", OFFSET(channels), AV_OPT_TYPE_INT, {.i64 = 2}, 1, INT_MAX, D },
00169 { "frame_size", "number of bytes per frame", OFFSET(frame_size), AV_OPT_TYPE_INT, {.i64 = 1024}, 1, INT_MAX, D },
00170 { "fragment_size", "buffering size, affects latency and cpu usage", OFFSET(fragment_size), AV_OPT_TYPE_INT, {.i64 = -1}, -1, INT_MAX, D },
00171 { NULL },
00172 };
00173
00174 static const AVClass pulse_demuxer_class = {
00175 .class_name = "Pulse demuxer",
00176 .item_name = av_default_item_name,
00177 .option = options,
00178 .version = LIBAVUTIL_VERSION_INT,
00179 };
00180
00181 AVInputFormat ff_pulse_demuxer = {
00182 .name = "pulse",
00183 .long_name = NULL_IF_CONFIG_SMALL("Pulse audio input"),
00184 .priv_data_size = sizeof(PulseData),
00185 .read_header = pulse_read_header,
00186 .read_packet = pulse_read_packet,
00187 .read_close = pulse_close,
00188 .flags = AVFMT_NOFILE,
00189 .priv_class = &pulse_demuxer_class,
00190 };
