4 This file is part of PulseAudio.
6 Copyright 2004-2006 Lennart Poettering
8 PulseAudio is free software; you can redistribute it and/or modify
9 it under the terms of the GNU Lesser General Public License as published
10 by the Free Software Foundation; either version 2 of the License,
11 or (at your option) any later version.
13 PulseAudio is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with PulseAudio; if not, write to the Free Software
20 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
31 #include <pulse/timeval.h>
32 #include <pulse/xmalloc.h>
34 #include <pulsecore/module.h>
35 #include <pulsecore/llist.h>
36 #include <pulsecore/sink.h>
37 #include <pulsecore/sink-input.h>
38 #include <pulsecore/memblockq.h>
39 #include <pulsecore/log.h>
40 #include <pulsecore/core-util.h>
41 #include <pulsecore/modargs.h>
42 #include <pulsecore/namereg.h>
44 #include "module-combine-symdef.h"
46 PA_MODULE_AUTHOR("Lennart Poettering")
47 PA_MODULE_DESCRIPTION("Combine multiple sinks to one")
48 PA_MODULE_VERSION(PACKAGE_VERSION
)
50 "sink_name=<name for the sink> "
51 "master=<master sink> "
52 "slaves=<slave sinks> "
53 "adjust_time=<seconds> "
54 "resample_method=<method> "
55 "format=<sample format> "
56 "channels=<number of channels> "
58 "channel_map=<channel map> ")
60 #define DEFAULT_SINK_NAME "combined"
61 #define MEMBLOCKQ_MAXLENGTH (1024*170)
62 #define RENDER_SIZE (1024*10)
64 #define DEFAULT_ADJUST_TIME 20
66 static const char* const valid_modargs
[] = {
80 struct userdata
*userdata
;
81 pa_sink_input
*sink_input
;
83 pa_memblockq
*memblockq
;
84 pa_usec_t total_latency
;
85 PA_LLIST_FIELDS(struct output
);
93 struct output
*master
;
94 pa_time_event
*time_event
;
97 PA_LLIST_HEAD(struct output
, outputs
);
100 static void output_free(struct output
*o
);
101 static void clear_up(struct userdata
*u
);
103 static void update_usage(struct userdata
*u
) {
104 pa_module_set_used(u
->module
, u
->sink
? pa_sink_used_by(u
->sink
) : 0);
107 static void adjust_rates(struct userdata
*u
) {
109 pa_usec_t max_sink_latency
= 0, min_total_latency
= (pa_usec_t
) -1, target_latency
;
111 assert(u
&& u
->sink
);
113 for (o
= u
->outputs
; o
; o
= o
->next
) {
114 uint32_t sink_latency
= o
->sink_input
->sink
? pa_sink_get_latency(o
->sink_input
->sink
) : 0;
116 o
->total_latency
= sink_latency
+ pa_sink_input_get_latency(o
->sink_input
);
118 if (sink_latency
> max_sink_latency
)
119 max_sink_latency
= sink_latency
;
121 if (o
->total_latency
< min_total_latency
)
122 min_total_latency
= o
->total_latency
;
125 assert(min_total_latency
!= (pa_usec_t
) -1);
127 target_latency
= max_sink_latency
> min_total_latency
? max_sink_latency
: min_total_latency
;
129 pa_log_info("[%s] target latency is %0.0f usec.", u
->sink
->name
, (float) target_latency
);
131 base_rate
= u
->sink
->sample_spec
.rate
;
133 for (o
= u
->outputs
; o
; o
= o
->next
) {
134 uint32_t r
= base_rate
;
136 if (o
->total_latency
< target_latency
)
137 r
-= (uint32_t) (((((double) target_latency
- o
->total_latency
))/u
->adjust_time
)*r
/ 1000000);
138 else if (o
->total_latency
> target_latency
)
139 r
+= (uint32_t) (((((double) o
->total_latency
- target_latency
))/u
->adjust_time
)*r
/ 1000000);
141 if (r
< (uint32_t) (base_rate
*0.9) || r
> (uint32_t) (base_rate
*1.1))
142 pa_log_warn("[%s] sample rates too different, not adjusting (%u vs. %u).", o
->sink_input
->name
, base_rate
, r
);
144 pa_log_info("[%s] new rate is %u Hz; ratio is %0.3f; latency is %0.0f usec.", o
->sink_input
->name
, r
, (double) r
/ base_rate
, (float) o
->total_latency
);
145 pa_sink_input_set_rate(o
->sink_input
, r
);
150 static void request_memblock(struct userdata
*u
) {
153 assert(u
&& u
->sink
);
157 if (pa_sink_render(u
->sink
, RENDER_SIZE
, &chunk
) < 0)
160 for (o
= u
->outputs
; o
; o
= o
->next
)
161 pa_memblockq_push_align(o
->memblockq
, &chunk
);
163 pa_memblock_unref(chunk
.memblock
);
166 static void time_callback(pa_mainloop_api
*a
, pa_time_event
* e
, PA_GCC_UNUSED
const struct timeval
*tv
, void *userdata
) {
167 struct userdata
*u
= userdata
;
169 assert(u
&& a
&& u
->time_event
== e
);
174 n
.tv_sec
+= u
->adjust_time
;
175 u
->sink
->core
->mainloop
->time_restart(e
, &n
);
178 static int sink_input_peek_cb(pa_sink_input
*i
, pa_memchunk
*chunk
) {
179 struct output
*o
= i
->userdata
;
180 assert(i
&& o
&& o
->sink_input
&& chunk
);
182 if (pa_memblockq_peek(o
->memblockq
, chunk
) >= 0)
186 request_memblock(o
->userdata
);
188 return pa_memblockq_peek(o
->memblockq
, chunk
);
191 static void sink_input_drop_cb(pa_sink_input
*i
, const pa_memchunk
*chunk
, size_t length
) {
192 struct output
*o
= i
->userdata
;
193 assert(i
&& o
&& o
->sink_input
&& chunk
&& length
);
195 pa_memblockq_drop(o
->memblockq
, chunk
, length
);
196 o
->counter
+= length
;
199 static void sink_input_kill_cb(pa_sink_input
*i
) {
200 struct output
*o
= i
->userdata
;
201 assert(i
&& o
&& o
->sink_input
);
202 pa_module_unload_request(o
->userdata
->module
);
203 clear_up(o
->userdata
);
206 static pa_usec_t
sink_input_get_latency_cb(pa_sink_input
*i
) {
207 struct output
*o
= i
->userdata
;
208 assert(i
&& o
&& o
->sink_input
);
210 return pa_bytes_to_usec(pa_memblockq_get_length(o
->memblockq
), &i
->sample_spec
);
213 static pa_usec_t
sink_get_latency_cb(pa_sink
*s
) {
214 struct userdata
*u
= s
->userdata
;
215 assert(s
&& u
&& u
->sink
&& u
->master
);
218 pa_sink_input_get_latency(u
->master
->sink_input
) +
219 pa_sink_get_latency(u
->master
->sink_input
->sink
);
222 static void sink_notify(pa_sink
*s
) {
230 for (o
= u
->outputs
; o
; o
= o
->next
)
231 pa_sink_notify(o
->sink_input
->sink
);
234 static struct output
*output_new(struct userdata
*u
, pa_sink
*sink
, int resample_method
) {
235 struct output
*o
= NULL
;
237 pa_sink_input_new_data data
;
239 assert(u
&& sink
&& u
->sink
);
241 o
= pa_xmalloc(sizeof(struct output
));
245 o
->memblockq
= pa_memblockq_new(
249 pa_frame_size(&u
->sink
->sample_spec
),
254 snprintf(t
, sizeof(t
), "Output stream #%u of sink %s", u
->n_outputs
+1, u
->sink
->name
);
256 pa_sink_input_new_data_init(&data
);
258 data
.driver
= __FILE__
;
260 pa_sink_input_new_data_set_sample_spec(&data
, &u
->sink
->sample_spec
);
261 pa_sink_input_new_data_set_channel_map(&data
, &u
->sink
->channel_map
);
262 data
.module
= u
->module
;
264 if (!(o
->sink_input
= pa_sink_input_new(u
->core
, &data
, PA_SINK_INPUT_VARIABLE_RATE
)))
267 o
->sink_input
->get_latency
= sink_input_get_latency_cb
;
268 o
->sink_input
->peek
= sink_input_peek_cb
;
269 o
->sink_input
->drop
= sink_input_drop_cb
;
270 o
->sink_input
->kill
= sink_input_kill_cb
;
271 o
->sink_input
->userdata
= o
;
273 PA_LLIST_PREPEND(struct output
, u
->outputs
, o
);
281 pa_sink_input_disconnect(o
->sink_input
);
282 pa_sink_input_unref(o
->sink_input
);
286 pa_memblockq_free(o
->memblockq
);
294 static void output_free(struct output
*o
) {
296 PA_LLIST_REMOVE(struct output
, o
->userdata
->outputs
, o
);
297 o
->userdata
->n_outputs
--;
298 pa_memblockq_free(o
->memblockq
);
299 pa_sink_input_disconnect(o
->sink_input
);
300 pa_sink_input_unref(o
->sink_input
);
304 static void clear_up(struct userdata
*u
) {
309 u
->core
->mainloop
->time_free(u
->time_event
);
310 u
->time_event
= NULL
;
313 while ((o
= u
->outputs
))
319 pa_sink_disconnect(u
->sink
);
320 pa_sink_unref(u
->sink
);
325 int pa__init(pa_core
*c
, pa_module
*m
) {
327 pa_modargs
*ma
= NULL
;
328 const char *master_name
, *slaves
, *rm
;
329 pa_sink
*master_sink
;
331 const char*split_state
;
333 int resample_method
= -1;
339 if (!(ma
= pa_modargs_new(m
->argument
, valid_modargs
))) {
340 pa_log("failed to parse module arguments");
344 if ((rm
= pa_modargs_get_value(ma
, "resample_method", NULL
))) {
345 if ((resample_method
= pa_parse_resample_method(rm
)) < 0) {
346 pa_log("invalid resample method '%s'", rm
);
351 u
= pa_xnew(struct userdata
, 1);
358 u
->time_event
= NULL
;
359 u
->adjust_time
= DEFAULT_ADJUST_TIME
;
360 PA_LLIST_HEAD_INIT(struct output
, u
->outputs
);
362 if (pa_modargs_get_value_u32(ma
, "adjust_time", &u
->adjust_time
) < 0) {
363 pa_log("failed to parse adjust_time value");
367 if (!(master_name
= pa_modargs_get_value(ma
, "master", NULL
)) || !(slaves
= pa_modargs_get_value(ma
, "slaves", NULL
))) {
368 pa_log("no master or slave sinks specified");
372 if (!(master_sink
= pa_namereg_get(c
, master_name
, PA_NAMEREG_SINK
, 1))) {
373 pa_log("invalid master sink '%s'", master_name
);
377 ss
= master_sink
->sample_spec
;
378 if ((pa_modargs_get_sample_spec(ma
, &ss
) < 0)) {
379 pa_log("invalid sample specification.");
383 if (ss
.channels
== master_sink
->sample_spec
.channels
)
384 map
= master_sink
->channel_map
;
386 pa_channel_map_init_auto(&map
, ss
.channels
, PA_CHANNEL_MAP_DEFAULT
);
388 if ((pa_modargs_get_channel_map(ma
, &map
) < 0)) {
389 pa_log("invalid channel map.");
393 if (ss
.channels
!= map
.channels
) {
394 pa_log("channel map and sample specification don't match.");
398 if (!(u
->sink
= pa_sink_new(c
, __FILE__
, pa_modargs_get_value(ma
, "sink_name", DEFAULT_SINK_NAME
), 0, &ss
, &map
))) {
399 pa_log("failed to create sink");
403 pa_sink_set_owner(u
->sink
, m
);
404 pa_sink_set_description(u
->sink
, "Combined Sink");
405 u
->sink
->get_latency
= sink_get_latency_cb
;
406 u
->sink
->notify
= sink_notify
;
407 u
->sink
->userdata
= u
;
409 if (!(u
->master
= output_new(u
, master_sink
, resample_method
))) {
410 pa_log("failed to create master sink input on sink '%s'.", u
->sink
->name
);
415 while ((n
= pa_split(slaves
, ",", &split_state
))) {
418 if (!(slave_sink
= pa_namereg_get(c
, n
, PA_NAMEREG_SINK
, 1))) {
419 pa_log("invalid slave sink '%s'", n
);
425 if (!output_new(u
, slave_sink
, resample_method
)) {
426 pa_log("failed to create slave sink input on sink '%s'.", slave_sink
->name
);
431 if (u
->n_outputs
<= 1)
432 pa_log_warn("WARNING: no slave sinks specified.");
434 if (u
->adjust_time
> 0) {
435 pa_gettimeofday(&tv
);
436 tv
.tv_sec
+= u
->adjust_time
;
437 u
->time_event
= c
->mainloop
->time_new(c
->mainloop
, &tv
, time_callback
, u
);
453 void pa__done(pa_core
*c
, pa_module
*m
) {
457 if (!(u
= m
->userdata
))