1 /* $Id$
2 * Substantial penalty for early withdrawal.
3 */
5 #include <unistd.h>
6 #include <ruby.h>
7 #include <math.h>
8 #include "../../src/rrd_tool.h"
10 typedef struct string_arr_t {
11 int len;
12 char **strings;
13 } string_arr;
15 VALUE mRRD;
16 VALUE rb_eRRDError;
18 typedef int (
19 *RRDFUNC) (
20 int argc,
21 char **argv);
23 typedef rrd_info_t *(
24 *RRDINFOFUNC) (
25 int argc,
26 char **argv);
28 #define RRD_CHECK_ERROR \
29 if (rrd_test_error()) \
30 rb_raise(rb_eRRDError, rrd_get_error()); \
31 rrd_clear_error();
33 string_arr string_arr_new(
34 VALUE rb_strings)
35 {
36 string_arr a;
37 char buf[64];
38 int i;
40 Check_Type(rb_strings, T_ARRAY);
41 a.len = RARRAY(rb_strings)->len + 1;
43 a.strings = malloc(a.len * sizeof(char *));
44 a.strings[0] = "dummy"; /* first element is a dummy element */
46 for (i = 0; i < a.len - 1; i++) {
47 VALUE v = rb_ary_entry(rb_strings, i);
49 switch (TYPE(v)) {
50 case T_STRING:
51 a.strings[i + 1] = strdup(STR2CSTR(v));
52 break;
53 case T_FIXNUM:
54 snprintf(buf, 63, "%d", FIX2INT(v));
55 a.strings[i + 1] = strdup(buf);
56 break;
57 default:
58 rb_raise(rb_eTypeError,
59 "invalid argument - %s, expected T_STRING or T_FIXNUM on index %d",
60 rb_class2name(CLASS_OF(v)), i);
61 break;
62 }
63 }
65 return a;
66 }
68 void string_arr_delete(
69 string_arr a)
70 {
71 int i;
73 /* skip dummy first entry */
74 for (i = 1; i < a.len; i++) {
75 free(a.strings[i]);
76 }
78 free(a.strings);
79 }
81 void reset_rrd_state(
82 )
83 {
84 optind = 0;
85 opterr = 0;
86 rrd_clear_error();
87 }
89 /* Simple Calls */
91 VALUE rrd_call(
92 RRDFUNC func,
93 VALUE args)
94 {
95 string_arr a;
97 a = string_arr_new(args);
98 reset_rrd_state();
99 func(a.len, a.strings);
100 string_arr_delete(a);
102 RRD_CHECK_ERROR return Qnil;
103 }
105 VALUE rb_rrd_create(
106 VALUE self,
107 VALUE args)
108 {
109 return rrd_call(rrd_create, args);
110 }
112 VALUE rb_rrd_dump(
113 VALUE self,
114 VALUE args)
115 {
116 return rrd_call(rrd_dump, args);
117 }
119 VALUE rb_rrd_resize(
120 VALUE self,
121 VALUE args)
122 {
123 return rrd_call(rrd_resize, args);
124 }
126 VALUE rb_rrd_restore(
127 VALUE self,
128 VALUE args)
129 {
130 return rrd_call(rrd_restore, args);
131 }
133 VALUE rb_rrd_tune(
134 VALUE self,
135 VALUE args)
136 {
137 return rrd_call(rrd_tune, args);
138 }
140 VALUE rb_rrd_update(
141 VALUE self,
142 VALUE args)
143 {
144 return rrd_call(rrd_update, args);
145 }
148 /* Calls Returning Data via the Info Interface */
150 VALUE rb_rrd_infocall(
151 RRDINFOFUNC func,
152 VALUE args)
153 {
154 string_arr a;
155 rrd_info_t *p, *data;
156 VALUE result;
158 a = string_arr_new(args);
159 data = func(a.len, a.strings);
160 string_arr_delete(a);
162 RRD_CHECK_ERROR result = rb_hash_new();
164 p = data;
165 while (data) {
166 VALUE key = rb_str_new2(data->key);
168 switch (data->type) {
169 case RD_I_VAL:
170 if (isnan(data->value.u_val)) {
171 rb_hash_aset(result, key, Qnil);
172 } else {
173 rb_hash_aset(result, key, rb_float_new(data->value.u_val));
174 }
175 break;
176 case RD_I_CNT:
177 rb_hash_aset(result, key, INT2FIX(data->value.u_cnt));
178 break;
179 case RD_I_STR:
180 rb_hash_aset(result, key, rb_str_new2(data->value.u_str));
181 break;
182 case RD_I_INT:
183 rb_hash_aset(result, key, INT2FIX(data->value.u_int));
184 break;
185 case RD_I_BLO:
186 rb_hash_aset(result, key,
187 rb_str_new((char *)data->value.u_blo.ptr,
188 data->value.u_blo.size));
189 break;
190 }
191 data = data->next;
192 }
193 rrd_info_free(p);
194 return result;
195 }
197 VALUE rb_rrd_info(
198 VALUE self,
199 VALUE args)
200 {
201 return rb_rrd_infocall(rrd_info, args);
202 }
204 VALUE rb_rrd_updatev(
205 VALUE self,
206 VALUE args)
207 {
208 return rb_rrd_infocall(rrd_update_v, args);
209 }
211 VALUE rb_rrd_graphv(
212 VALUE self,
213 VALUE args)
214 {
215 return rb_rrd_infocall(rrd_graph_v, args);
216 }
219 /* Other Calls */
221 VALUE rb_rrd_fetch(
222 VALUE self,
223 VALUE args)
224 {
225 string_arr a;
226 unsigned long i, j, k, step, ds_cnt;
227 rrd_value_t *raw_data;
228 char **raw_names;
229 VALUE data, names, result;
230 time_t start, end;
232 a = string_arr_new(args);
233 reset_rrd_state();
234 rrd_fetch(a.len, a.strings, &start, &end, &step, &ds_cnt, &raw_names,
235 &raw_data);
236 string_arr_delete(a);
238 RRD_CHECK_ERROR names = rb_ary_new();
240 for (i = 0; i < ds_cnt; i++) {
241 rb_ary_push(names, rb_str_new2(raw_names[i]));
242 rrd_freemem(raw_names[i]);
243 }
244 rrd_freemem(raw_names);
246 k = 0;
247 data = rb_ary_new();
248 for (i = start; i <= end; i += step) {
249 VALUE line = rb_ary_new2(ds_cnt);
251 for (j = 0; j < ds_cnt; j++) {
252 rb_ary_store(line, j, rb_float_new(raw_data[k]));
253 k++;
254 }
255 rb_ary_push(data, line);
256 }
257 rrd_freemem(raw_data);
259 result = rb_ary_new2(5);
260 rb_ary_store(result, 0, INT2NUM(start));
261 rb_ary_store(result, 1, INT2NUM(end));
262 rb_ary_store(result, 2, names);
263 rb_ary_store(result, 3, data);
264 rb_ary_store(result, 4, INT2FIX(step));
265 return result;
266 }
268 VALUE rb_rrd_graph(
269 VALUE self,
270 VALUE args)
271 {
272 string_arr a;
273 char **calcpr, **p;
274 VALUE result, print_results;
275 int xsize, ysize;
276 double ymin, ymax;
278 a = string_arr_new(args);
279 reset_rrd_state();
280 rrd_graph(a.len, a.strings, &calcpr, &xsize, &ysize, NULL, &ymin, &ymax);
281 string_arr_delete(a);
283 RRD_CHECK_ERROR result = rb_ary_new2(3);
285 print_results = rb_ary_new();
286 p = calcpr;
287 for (p = calcpr; p && *p; p++) {
288 rb_ary_push(print_results, rb_str_new2(*p));
289 rrd_freemem(*p);
290 }
291 rrd_freemem(calcpr);
292 rb_ary_store(result, 0, print_results);
293 rb_ary_store(result, 1, INT2FIX(xsize));
294 rb_ary_store(result, 2, INT2FIX(ysize));
295 return result;
296 }
299 VALUE rb_rrd_last(
300 VALUE self,
301 VALUE args)
302 {
303 string_arr a;
304 time_t last;
306 a = string_arr_new(args);
307 reset_rrd_state();
308 last = rrd_last(a.len, a.strings);
309 string_arr_delete(a);
311 RRD_CHECK_ERROR
312 return rb_funcall(rb_cTime, rb_intern("at"), 1, UINT2NUM(last));
313 }
315 void Init_RRD(
316 )
317 {
318 mRRD = rb_define_module("RRD");
319 rb_eRRDError = rb_define_class("RRDError", rb_eStandardError);
321 rb_define_module_function(mRRD, "create", rb_rrd_create, -2);
322 rb_define_module_function(mRRD, "dump", rb_rrd_dump, -2);
323 rb_define_module_function(mRRD, "fetch", rb_rrd_fetch, -2);
324 rb_define_module_function(mRRD, "graph", rb_rrd_graph, -2);
325 rb_define_module_function(mRRD, "last", rb_rrd_last, -2);
326 rb_define_module_function(mRRD, "resize", rb_rrd_resize, -2);
327 rb_define_module_function(mRRD, "restore", rb_rrd_restore, -2);
328 rb_define_module_function(mRRD, "tune", rb_rrd_tune, -2);
329 rb_define_module_function(mRRD, "update", rb_rrd_update, -2);
330 rb_define_module_function(mRRD, "info", rb_rrd_info, -2);
331 rb_define_module_function(mRRD, "updatev", rb_rrd_updatev, -2);
332 rb_define_module_function(mRRD, "graphv", rb_rrd_graphv, -2);
333 }