5a849868e0ec2998009512e2e1f2d8e7dde43539
[paraslash.git] / prebuffer_filter.c
1 /*
2 * Copyright (C) 2009-2012 Andre Noll <maan@systemlinux.org>
3 *
4 * Licensed under the GPL v2. For licencing details see COPYING.
5 */
6
7 /** \file prebuffer_filter.c Paraslash's prebuffering filter. */
8
9 #include <regex.h>
10
11 #include "para.h"
12 #include "prebuffer_filter.cmdline.h"
13 #include "list.h"
14 #include "sched.h"
15 #include "ggo.h"
16 #include "buffer_tree.h"
17 #include "filter.h"
18 #include "string.h"
19 #include "error.h"
20
21 /** Data specific to the prebuffer filter. */
22 struct private_prebuffer_data {
23 /** The configuration data for this instance of the filter. */
24 struct prebuffer_filter_args_info *conf;
25 /** Number of bytes prebuffered or -1 if no longer prebuffering. */
26 int prebuffered;
27 /** End of prebuffering period. */
28 struct timeval barrier;
29 };
30
31 static void prebuffer_pre_select(struct sched *s, struct task *t)
32 {
33 struct filter_node *fn = container_of(t, struct filter_node, task);
34 struct btr_node *btrn = fn->btrn;
35 size_t iqs = btr_get_input_queue_size(btrn);
36 struct private_prebuffer_data *ppd = fn->private_data;
37 struct prebuffer_filter_args_info *conf = ppd->conf;
38 struct timeval diff;
39
40 t->error = 0;
41 if (iqs == 0)
42 return;
43 if (ppd->barrier.tv_sec == 0) {
44 struct timeval tv;
45 PARA_INFO_LOG("prebuffer period %dms\n",
46 conf->duration_arg);
47 ms2tv(conf->duration_arg, &tv);
48 tv_add(&tv, now, &ppd->barrier);
49 }
50 if (tv_diff(&ppd->barrier, now, &diff) < 0)
51 return sched_min_delay(s);
52 sched_request_timeout(&diff, s);
53 }
54
55 static void prebuffer_close(struct filter_node *fn)
56 {
57 free(fn->private_data);
58 }
59
60 static void prebuffer_post_select(__a_unused struct sched *s, struct task *t)
61 {
62 struct filter_node *fn = container_of(t, struct filter_node, task);
63 struct btr_node *btrn = fn->btrn;
64 size_t iqs = btr_get_input_queue_size(btrn);
65 struct private_prebuffer_data *ppd = fn->private_data;
66 struct prebuffer_filter_args_info *conf = ppd->conf;
67
68 t->error = 0;
69 if (ppd->barrier.tv_sec == 0)
70 return;
71 if (tv_diff(now, &ppd->barrier, NULL) < 0)
72 return;
73 if (iqs < conf->size_arg)
74 return;
75 btr_splice_out_node(btrn);
76 t->error = -E_PREBUFFER_SUCCESS;
77 }
78
79 static int prebuffer_parse_config(int argc, char **argv, void **config)
80 {
81 struct prebuffer_filter_args_info *conf = para_calloc(sizeof(*conf));
82 int ret = -E_PREBUFFER_SYNTAX;
83
84 prebuffer_filter_cmdline_parser(argc, argv, conf);
85 ret = -ERRNO_TO_PARA_ERROR(EINVAL);
86 if (conf->duration_arg < 0)
87 goto err;
88 if (conf->size_arg < 0)
89 goto err;
90 PARA_NOTICE_LOG("prebuffering %ims, %i bytes\n", conf->duration_arg,
91 conf->size_arg);
92 *config = conf;
93 return 1;
94 err:
95 free(conf);
96 return ret;
97 }
98
99 static void prebuffer_open(struct filter_node *fn)
100 {
101 struct private_prebuffer_data *ppd = para_calloc(sizeof(*ppd));
102
103 ppd->conf = fn->conf;
104 fn->private_data = ppd;
105 }
106
107 static void prebuffer_free_config(void *conf)
108 {
109 prebuffer_filter_cmdline_parser_free(conf);
110 }
111
112 /**
113 * The init function of the prebuffer filter.
114 *
115 * \param f Pointer to the struct to initialize.
116 */
117 void prebuffer_filter_init(struct filter *f)
118 {
119 struct prebuffer_filter_args_info dummy;
120
121 prebuffer_filter_cmdline_parser_init(&dummy);
122 f->open = prebuffer_open;
123 f->close = prebuffer_close;
124 f->parse_config = prebuffer_parse_config;
125 f->free_config = prebuffer_free_config;
126 f->pre_select = prebuffer_pre_select;
127 f->post_select = prebuffer_post_select;
128 f->help = (struct ggo_help) {
129 .short_help = prebuffer_filter_args_info_help,
130 .detailed_help = prebuffer_filter_args_info_detailed_help
131 };
132 }