Commit | Line | Data |
---|---|---|
2ba45a60 DM |
1 | /* |
2 | * Pulseaudio common | |
3 | * Copyright (c) 2014 Lukasz Marek | |
4 | * Copyright (c) 2011 Luca Barbato <lu_zero@gentoo.org> | |
5 | * | |
6 | * This file is part of FFmpeg. | |
7 | * | |
8 | * FFmpeg is free software; you can redistribute it and/or | |
9 | * modify it under the terms of the GNU Lesser General Public | |
10 | * License as published by the Free Software Foundation; either | |
11 | * version 2.1 of the License, or (at your option) any later version. | |
12 | * | |
13 | * FFmpeg is distributed in the hope that it will be useful, | |
14 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
15 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
16 | * Lesser General Public License for more details. | |
17 | * | |
18 | * You should have received a copy of the GNU Lesser General Public | |
19 | * License along with FFmpeg; if not, write to the Free Software | |
20 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
21 | */ | |
22 | ||
23 | #include "pulse_audio_common.h" | |
24 | #include "libavutil/attributes.h" | |
25 | #include "libavutil/avstring.h" | |
26 | #include "libavutil/mem.h" | |
27 | #include "libavutil/avassert.h" | |
28 | ||
29 | pa_sample_format_t av_cold ff_codec_id_to_pulse_format(enum AVCodecID codec_id) | |
30 | { | |
31 | switch (codec_id) { | |
32 | case AV_CODEC_ID_PCM_U8: return PA_SAMPLE_U8; | |
33 | case AV_CODEC_ID_PCM_ALAW: return PA_SAMPLE_ALAW; | |
34 | case AV_CODEC_ID_PCM_MULAW: return PA_SAMPLE_ULAW; | |
35 | case AV_CODEC_ID_PCM_S16LE: return PA_SAMPLE_S16LE; | |
36 | case AV_CODEC_ID_PCM_S16BE: return PA_SAMPLE_S16BE; | |
37 | case AV_CODEC_ID_PCM_F32LE: return PA_SAMPLE_FLOAT32LE; | |
38 | case AV_CODEC_ID_PCM_F32BE: return PA_SAMPLE_FLOAT32BE; | |
39 | case AV_CODEC_ID_PCM_S32LE: return PA_SAMPLE_S32LE; | |
40 | case AV_CODEC_ID_PCM_S32BE: return PA_SAMPLE_S32BE; | |
41 | case AV_CODEC_ID_PCM_S24LE: return PA_SAMPLE_S24LE; | |
42 | case AV_CODEC_ID_PCM_S24BE: return PA_SAMPLE_S24BE; | |
43 | default: return PA_SAMPLE_INVALID; | |
44 | } | |
45 | } | |
46 | ||
47 | enum PulseAudioContextState { | |
48 | PULSE_CONTEXT_INITIALIZING, | |
49 | PULSE_CONTEXT_READY, | |
50 | PULSE_CONTEXT_FINISHED | |
51 | }; | |
52 | ||
53 | typedef struct PulseAudioDeviceList { | |
54 | AVDeviceInfoList *devices; | |
55 | int error_code; | |
56 | int output; | |
57 | char *default_device; | |
58 | } PulseAudioDeviceList; | |
59 | ||
60 | static void pa_state_cb(pa_context *c, void *userdata) | |
61 | { | |
62 | enum PulseAudioContextState *context_state = userdata; | |
63 | ||
64 | switch (pa_context_get_state(c)) { | |
65 | case PA_CONTEXT_FAILED: | |
66 | case PA_CONTEXT_TERMINATED: | |
67 | *context_state = PULSE_CONTEXT_FINISHED; | |
68 | break; | |
69 | case PA_CONTEXT_READY: | |
70 | *context_state = PULSE_CONTEXT_READY; | |
71 | break; | |
72 | default: | |
73 | break; | |
74 | } | |
75 | } | |
76 | ||
77 | void ff_pulse_audio_disconnect_context(pa_mainloop **pa_ml, pa_context **pa_ctx) | |
78 | { | |
79 | av_assert0(pa_ml); | |
80 | av_assert0(pa_ctx); | |
81 | ||
82 | if (*pa_ctx) { | |
83 | pa_context_set_state_callback(*pa_ctx, NULL, NULL); | |
84 | pa_context_disconnect(*pa_ctx); | |
85 | pa_context_unref(*pa_ctx); | |
86 | } | |
87 | if (*pa_ml) | |
88 | pa_mainloop_free(*pa_ml); | |
89 | *pa_ml = NULL; | |
90 | *pa_ctx = NULL; | |
91 | } | |
92 | ||
93 | int ff_pulse_audio_connect_context(pa_mainloop **pa_ml, pa_context **pa_ctx, | |
94 | const char *server, const char *description) | |
95 | { | |
96 | int ret; | |
97 | pa_mainloop_api *pa_mlapi = NULL; | |
98 | enum PulseAudioContextState context_state = PULSE_CONTEXT_INITIALIZING; | |
99 | ||
100 | av_assert0(pa_ml); | |
101 | av_assert0(pa_ctx); | |
102 | ||
103 | *pa_ml = NULL; | |
104 | *pa_ctx = NULL; | |
105 | ||
106 | if (!(*pa_ml = pa_mainloop_new())) | |
107 | return AVERROR(ENOMEM); | |
108 | if (!(pa_mlapi = pa_mainloop_get_api(*pa_ml))) { | |
109 | ret = AVERROR_EXTERNAL; | |
110 | goto fail; | |
111 | } | |
112 | if (!(*pa_ctx = pa_context_new(pa_mlapi, description))) { | |
113 | ret = AVERROR(ENOMEM); | |
114 | goto fail; | |
115 | } | |
116 | pa_context_set_state_callback(*pa_ctx, pa_state_cb, &context_state); | |
117 | if (pa_context_connect(*pa_ctx, server, 0, NULL) < 0) { | |
118 | ret = AVERROR_EXTERNAL; | |
119 | goto fail; | |
120 | } | |
121 | ||
122 | while (context_state == PULSE_CONTEXT_INITIALIZING) | |
123 | pa_mainloop_iterate(*pa_ml, 1, NULL); | |
124 | if (context_state == PULSE_CONTEXT_FINISHED) { | |
125 | ret = AVERROR_EXTERNAL; | |
126 | goto fail; | |
127 | } | |
128 | return 0; | |
129 | ||
130 | fail: | |
131 | ff_pulse_audio_disconnect_context(pa_ml, pa_ctx); | |
132 | return ret; | |
133 | } | |
134 | ||
135 | static void pulse_add_detected_device(PulseAudioDeviceList *info, | |
136 | const char *name, const char *description) | |
137 | { | |
138 | int ret; | |
139 | AVDeviceInfo *new_device = NULL; | |
140 | ||
141 | if (info->error_code) | |
142 | return; | |
143 | ||
144 | new_device = av_mallocz(sizeof(AVDeviceInfo)); | |
145 | if (!new_device) { | |
146 | info->error_code = AVERROR(ENOMEM); | |
147 | return; | |
148 | } | |
149 | ||
150 | new_device->device_description = av_strdup(description); | |
151 | new_device->device_name = av_strdup(name); | |
152 | ||
153 | if (!new_device->device_description || !new_device->device_name) { | |
154 | info->error_code = AVERROR(ENOMEM); | |
155 | goto fail; | |
156 | } | |
157 | ||
158 | if ((ret = av_dynarray_add_nofree(&info->devices->devices, | |
159 | &info->devices->nb_devices, new_device)) < 0) { | |
160 | info->error_code = ret; | |
161 | goto fail; | |
162 | } | |
163 | return; | |
164 | ||
165 | fail: | |
f6fa7814 DM |
166 | av_freep(&new_device->device_description); |
167 | av_freep(&new_device->device_name); | |
2ba45a60 DM |
168 | av_free(new_device); |
169 | ||
170 | } | |
171 | ||
172 | static void pulse_audio_source_device_cb(pa_context *c, const pa_source_info *dev, | |
173 | int eol, void *userdata) | |
174 | { | |
175 | if (!eol) | |
176 | pulse_add_detected_device(userdata, dev->name, dev->description); | |
177 | } | |
178 | ||
179 | static void pulse_audio_sink_device_cb(pa_context *c, const pa_sink_info *dev, | |
180 | int eol, void *userdata) | |
181 | { | |
182 | if (!eol) | |
183 | pulse_add_detected_device(userdata, dev->name, dev->description); | |
184 | } | |
185 | ||
186 | static void pulse_server_info_cb(pa_context *c, const pa_server_info *i, void *userdata) | |
187 | { | |
188 | PulseAudioDeviceList *info = userdata; | |
189 | if (info->output) | |
190 | info->default_device = av_strdup(i->default_sink_name); | |
191 | else | |
192 | info->default_device = av_strdup(i->default_source_name); | |
193 | if (!info->default_device) | |
194 | info->error_code = AVERROR(ENOMEM); | |
195 | } | |
196 | ||
197 | int ff_pulse_audio_get_devices(AVDeviceInfoList *devices, const char *server, int output) | |
198 | { | |
199 | pa_mainloop *pa_ml = NULL; | |
200 | pa_operation *pa_op = NULL; | |
201 | pa_context *pa_ctx = NULL; | |
202 | enum pa_operation_state op_state; | |
203 | PulseAudioDeviceList dev_list = { 0 }; | |
204 | int i; | |
205 | ||
206 | dev_list.output = output; | |
207 | dev_list.devices = devices; | |
208 | if (!devices) | |
209 | return AVERROR(EINVAL); | |
210 | devices->nb_devices = 0; | |
211 | devices->devices = NULL; | |
212 | ||
213 | if ((dev_list.error_code = ff_pulse_audio_connect_context(&pa_ml, &pa_ctx, server, "Query devices")) < 0) | |
214 | goto fail; | |
215 | ||
216 | if (output) | |
217 | pa_op = pa_context_get_sink_info_list(pa_ctx, pulse_audio_sink_device_cb, &dev_list); | |
218 | else | |
219 | pa_op = pa_context_get_source_info_list(pa_ctx, pulse_audio_source_device_cb, &dev_list); | |
220 | while ((op_state = pa_operation_get_state(pa_op)) == PA_OPERATION_RUNNING) | |
221 | pa_mainloop_iterate(pa_ml, 1, NULL); | |
222 | if (op_state != PA_OPERATION_DONE) | |
223 | dev_list.error_code = AVERROR_EXTERNAL; | |
224 | pa_operation_unref(pa_op); | |
225 | if (dev_list.error_code < 0) | |
226 | goto fail; | |
227 | ||
228 | pa_op = pa_context_get_server_info(pa_ctx, pulse_server_info_cb, &dev_list); | |
229 | while ((op_state = pa_operation_get_state(pa_op)) == PA_OPERATION_RUNNING) | |
230 | pa_mainloop_iterate(pa_ml, 1, NULL); | |
231 | if (op_state != PA_OPERATION_DONE) | |
232 | dev_list.error_code = AVERROR_EXTERNAL; | |
233 | pa_operation_unref(pa_op); | |
234 | if (dev_list.error_code < 0) | |
235 | goto fail; | |
236 | ||
237 | devices->default_device = -1; | |
238 | for (i = 0; i < devices->nb_devices; i++) { | |
239 | if (!strcmp(devices->devices[i]->device_name, dev_list.default_device)) { | |
240 | devices->default_device = i; | |
241 | break; | |
242 | } | |
243 | } | |
244 | ||
245 | fail: | |
246 | av_free(dev_list.default_device); | |
247 | ff_pulse_audio_disconnect_context(&pa_ml, &pa_ctx); | |
248 | return dev_list.error_code; | |
249 | } |