2 * Copyright (C) 2006 Andre Noll <maan@systemlinux.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
19 /** \file write_common.c common functions of para_audiod and para_write */
28 const char *writer_names
[] ={WRITER_NAMES
};
29 struct writer writers
[NUM_SUPPORTED_WRITERS
] = {WRITER_ARRAY
};
31 int wng_write(struct writer_node_group
*g
, char *buf
, size_t *loaded
)
33 int ret
, i
, need_more_writes
= 1;
34 size_t min_written
= 0;
36 while (need_more_writes
) {
38 FOR_EACH_WRITER_NODE(i
, g
) {
39 size_t w
= g
->written
[i
];
41 struct writer_node
*wn
= &g
->writer_nodes
[i
];
45 min_written
= PARA_MIN(min_written
, w
);
48 if (!g
->eof
&& (*loaded
< wn
->chunk_bytes
+ w
))
50 bytes_to_write
= PARA_MIN(wn
->chunk_bytes
,
52 ret
= wn
->writer
->write(buf
+ w
, bytes_to_write
, wn
);
55 if (ret
!= bytes_to_write
)
56 PARA_WARNING_LOG("short write: %d/%d\n", ret
,
62 *loaded
-= min_written
;
67 memmove(buf
, buf
+ min_written
, *loaded
);
68 FOR_EACH_WRITER_NODE(i
, g
)
69 g
->written
[i
] -= min_written
;
75 static void wng_post_select(struct sched
*s
, struct task
*t
)
77 struct writer_node_group
*g
= t
->private_data
;
79 size_t min_written
= 0;
81 FOR_EACH_WRITER_NODE(i
, g
) {
82 struct writer_node
*wn
= &g
->writer_nodes
[i
];
83 t
->ret
= wn
->task
.ret
;
89 min_written
= PARA_MIN(min_written
, t
->ret
);
91 *g
->loaded
-= min_written
;
92 if (!*g
->loaded
&& g
->eof
)
96 if (*g
->loaded
&& min_written
)
97 memmove(g
->buf
, g
->buf
+ min_written
, *g
->loaded
);
100 int wng_open(struct writer_node_group
*g
)
104 FOR_EACH_WRITER_NODE(i
, g
) {
105 struct writer_node
*wn
= &g
->writer_nodes
[i
];
106 ret
= wn
->writer
->open(wn
);
109 wn
->chunk_bytes
= ret
;
110 g
->max_chunk_bytes
= PARA_MAX(g
->max_chunk_bytes
, ret
);
112 PARA_DEBUG_LOG("pre_select: %p\n", &wn
->writer
->pre_select
);
113 PARA_DEBUG_LOG("post_select: %p\n", &wn
->writer
->post_select
);
114 wn
->task
.pre_select
= wn
->writer
->pre_select
;
115 wn
->task
.post_select
= wn
->writer
->post_select
;
116 wn
->task
.private_data
= wn
;
117 register_task(&wn
->task
);
119 register_task(&g
->task
);
124 void wng_destroy(struct writer_node_group
*g
)
129 free(g
->writer_nodes
);
133 void wng_close(struct writer_node_group
*g
)
137 FOR_EACH_WRITER_NODE(i
, g
) {
138 struct writer_node
*wn
= &g
->writer_nodes
[i
];
139 unregister_task(&wn
->task
);
140 wn
->writer
->close(wn
);
144 static void wng_error_handler(struct task
*t
)
146 struct writer_node_group
*g
= t
->private_data
;
148 PARA_INFO_LOG("%p: ret = %d\n", t
, t
->ret
);
154 struct writer_node_group
*wng_new(unsigned num_writers
)
156 struct writer_node_group
*g
= para_calloc(sizeof(struct writer_node_group
));
157 g
->num_writers
= num_writers
;
158 g
->writer_nodes
= para_calloc(num_writers
159 * sizeof(struct writer_node
));
160 g
->written
= para_calloc(num_writers
* sizeof(size_t));
161 g
->task
.private_data
= g
;
162 g
->task
.post_select
= wng_post_select
;
163 g
->task
.error_handler
= wng_error_handler
;
164 g
->task
.flags
= POST_ADD_TAIL
| POST_EOF_IS_ERROR
;
168 void init_supported_writers(void)
173 writers
[i
].init(&writers
[i
]);
176 int check_writer_arg(const char *arg
)
178 int i
, ret
= -E_WRITE_COMMON_SYNTAX
;
179 char *a
= para_strdup(arg
), *p
= strchr(a
, ':');
184 if (strcmp(writer_names
[i
], a
))
194 struct writer_node_group
*setup_default_wng(void)
196 struct writer_node_group
*wng
= wng_new(1);
197 enum writer_enum default_writer
;
199 if (NUM_SUPPORTED_WRITERS
== 1)
200 default_writer
= FILE_WRITE
;
203 wng
->writer_nodes
[0].writer
= &writers
[default_writer
];
204 PARA_INFO_LOG("using default writer: %s\n",
205 writer_names
[default_writer
]);