avdevice/pulse_audio_enc: fix stream index
[ffmpeg.git] / libavdevice / pulse_audio_enc.c
1 /*
2  * Copyright (c) 2013 Lukasz Marek <lukasz.m.luki@gmail.com>
3  *
4  * This file is part of FFmpeg.
5  *
6  * FFmpeg is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * FFmpeg 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 GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with FFmpeg; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19  */
20
21 #include <pulse/simple.h>
22 #include <pulse/error.h>
23 #include "libavformat/avformat.h"
24 #include "libavformat/internal.h"
25 #include "libavutil/opt.h"
26 #include "libavutil/time.h"
27 #include "libavutil/log.h"
28 #include "pulse_audio_common.h"
29
30 typedef struct PulseData {
31     AVClass *class;
32     const char *server;
33     const char *name;
34     const char *stream_name;
35     const char *device;
36     pa_simple *pa;
37     unsigned int stream_index;
38 } PulseData;
39
40 static av_cold int pulse_write_header(AVFormatContext *h)
41 {
42     PulseData *s = h->priv_data;
43     AVStream *st = NULL;
44     int ret;
45     pa_sample_spec ss;
46     pa_buffer_attr attr = { -1, -1, -1, -1, -1 };
47     const char *stream_name = s->stream_name;
48
49     for (unsigned i = 0; i < h->nb_streams; i++) {
50         if (h->streams[i]->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
51             st = h->streams[i];
52             s->stream_index = i;
53             break;
54         }
55     }
56
57     if (!st) {
58         av_log(s, AV_LOG_ERROR, "No audio stream found.\n");
59         return AVERROR(EINVAL);
60     }
61
62     if (!stream_name)
63         stream_name = h->filename;
64
65     ss.format = codec_id_to_pulse_format(st->codec->codec_id);
66     ss.rate = st->codec->sample_rate;
67     ss.channels = st->codec->channels;
68
69     s->pa = pa_simple_new(s->server,                 // Server
70                           s->name,                   // Application name
71                           PA_STREAM_PLAYBACK,
72                           s->device,                 // Device
73                           stream_name,               // Description of a stream
74                           &ss,                       // Sample format
75                           NULL,                      // Use default channel map
76                           &attr,                     // Buffering attributes
77                           &ret);                     // Result
78
79     if (!s->pa) {
80         av_log(s, AV_LOG_ERROR, "pa_simple_new failed: %s\n", pa_strerror(ret));
81         return AVERROR(EIO);
82     }
83
84     avpriv_set_pts_info(st, 64, 1, 1000000);  /* 64 bits pts in us */
85
86     return 0;
87 }
88
89 static av_cold int pulse_write_trailer(AVFormatContext *h)
90 {
91     PulseData *s = h->priv_data;
92     pa_simple_flush(s->pa, NULL);
93     pa_simple_free(s->pa);
94     s->pa = NULL;
95     return 0;
96 }
97
98 static int pulse_write_packet(AVFormatContext *h, AVPacket *pkt)
99 {
100     PulseData *s = h->priv_data;
101     int size     = pkt->size;
102     uint8_t *buf = pkt->data;
103     int error;
104
105     if (s->stream_index != pkt->stream_index)
106         return 0;
107
108     if ((error = pa_simple_write(s->pa, buf, size, &error))) {
109         av_log(s, AV_LOG_ERROR, "pa_simple_write failed: %s\n", pa_strerror(error));
110         return AVERROR(EIO);
111     }
112
113     return 0;
114 }
115
116 static void pulse_get_output_timestamp(AVFormatContext *h, int stream, int64_t *dts, int64_t *wall)
117 {
118     PulseData *s = h->priv_data;
119     pa_usec_t latency = pa_simple_get_latency(s->pa, NULL);
120     *wall = av_gettime();
121     *dts = h->streams[s->stream_index]->cur_dts - latency;
122 }
123
124 #define OFFSET(a) offsetof(PulseData, a)
125 #define E AV_OPT_FLAG_ENCODING_PARAM
126
127 static const AVOption options[] = {
128     { "server",        "set PulseAudio server",  OFFSET(server),      AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E },
129     { "name",          "set application name",   OFFSET(name),        AV_OPT_TYPE_STRING, {.str = LIBAVFORMAT_IDENT},  0, 0, E },
130     { "stream_name",   "set stream description", OFFSET(stream_name), AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E },
131     { "device",        "set device name",        OFFSET(device),      AV_OPT_TYPE_STRING, {.str = NULL}, 0, 0, E },
132     { NULL }
133 };
134
135 static const AVClass pulse_muxer_class = {
136     .class_name     = "Pulse muxer",
137     .item_name      = av_default_item_name,
138     .option         = options,
139     .version        = LIBAVUTIL_VERSION_INT,
140 };
141
142 AVOutputFormat ff_pulse_muxer = {
143     .name           = "pulse",
144     .long_name      = NULL_IF_CONFIG_SMALL("Pulse audio output"),
145     .priv_data_size = sizeof(PulseData),
146     .audio_codec    = AV_NE(AV_CODEC_ID_PCM_S16BE, AV_CODEC_ID_PCM_S16LE),
147     .video_codec    = AV_CODEC_ID_NONE,
148     .write_header   = pulse_write_header,
149     .write_packet   = pulse_write_packet,
150     .write_trailer  = pulse_write_trailer,
151     .get_output_timestamp = pulse_get_output_timestamp,
152     .flags          = AVFMT_NOFILE,
153     .priv_class     = &pulse_muxer_class,
154 };