bindings/ruby/main.c

   1 /* $Id$
   2  * Substantial penalty for early withdrawal.
   3  */
   4
   5 #include <unistd.h>
   6 #include <ruby.h>
   7 #include <math.h>
   8 #include "../../src/rrd_tool.h"
   9
  10 typedef struct string_arr_t {
  11     int       len;
  12     char    **strings;
  13 } string_arr;
  14
  15 VALUE     mRRD;
  16 VALUE     rb_eRRDError;
  17
  18 typedef int (
  19     *RRDFUNC) (
  20     int argc,
  21     char **argv);
  22
  23 typedef rrd_info_t *(
  24     *RRDINFOFUNC) (
  25     int argc,
  26     char **argv);
  27
  28 #define RRD_CHECK_ERROR  \
  29     if (rrd_test_error()) \
  30       rb_raise(rb_eRRDError, rrd_get_error()); \
  31     rrd_clear_error();
  32
  33 string_arr string_arr_new(
  34     VALUE rb_strings)
  35 {
  36     string_arr a;
  37     char      buf[64];
  38     int       i;
  39
  40     Check_Type(rb_strings, T_ARRAY);
  41     a.len = RARRAY(rb_strings)->len + 1;
  42
  43     a.strings = malloc(a.len * sizeof(char *));
  44     a.strings[0] = "dummy"; /* first element is a dummy element */
  45
  46     for (i = 0; i < a.len - 1; i++) {
  47         VALUE     v = rb_ary_entry(rb_strings, i);
  48
  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     }
  64
  65     return a;
  66 }
  67
  68 void string_arr_delete(
  69     string_arr a)
  70 {
  71     int       i;
  72
  73     /* skip dummy first entry */
  74     for (i = 1; i < a.len; i++) {
  75         free(a.strings[i]);
  76     }
  77
  78     free(a.strings);
  79 }
  80
  81 void reset_rrd_state(
  82     )
  83 {
  84     optind = 0;
  85     opterr = 0;
  86     rrd_clear_error();
  87 }
  88
  89 /* Simple Calls */
  90
  91 VALUE rrd_call(
  92     RRDFUNC func,
  93     VALUE args)
  94 {
  95     string_arr a;
  96
  97     a = string_arr_new(args);
  98     reset_rrd_state();
  99     func(a.len, a.strings);
 100     string_arr_delete(a);
 101
 102     RRD_CHECK_ERROR return Qnil;
 103 }
 104
 105 VALUE rb_rrd_create(
 106     VALUE self,
 107     VALUE args)
 108 {
 109     return rrd_call(rrd_create, args);
 110 }
 111
 112 VALUE rb_rrd_dump(
 113     VALUE self,
 114     VALUE args)
 115 {
 116     return rrd_call(rrd_dump, args);
 117 }
 118
 119 VALUE rb_rrd_resize(
 120     VALUE self,
 121     VALUE args)
 122 {
 123     return rrd_call(rrd_resize, args);
 124 }
 125
 126 VALUE rb_rrd_restore(
 127     VALUE self,
 128     VALUE args)
 129 {
 130     return rrd_call(rrd_restore, args);
 131 }
 132
 133 VALUE rb_rrd_tune(
 134     VALUE self,
 135     VALUE args)
 136 {
 137     return rrd_call(rrd_tune, args);
 138 }
 139
 140 VALUE rb_rrd_update(
 141     VALUE self,
 142     VALUE args)
 143 {
 144     return rrd_call(rrd_update, args);
 145 }
 146
 147
 148 /* Calls Returning Data via the Info Interface */
 149
 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;
 157
 158     a = string_arr_new(args);
 159     data = func(a.len, a.strings);
 160     string_arr_delete(a);
 161
 162     RRD_CHECK_ERROR result = rb_hash_new();
 163
 164     p = data;
 165     while (data) {
 166         VALUE     key = rb_str_new2(data->key);
 167
 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 }
 196
 197 VALUE rb_rrd_info(
 198     VALUE self,
 199     VALUE args)
 200 {
 201     return rb_rrd_infocall(rrd_info, args);
 202 }
 203
 204 VALUE rb_rrd_updatev(
 205     VALUE self,
 206     VALUE args)
 207 {
 208     return rb_rrd_infocall(rrd_update_v, args);
 209 }
 210
 211 VALUE rb_rrd_graphv(
 212     VALUE self,
 213     VALUE args)
 214 {
 215     return rb_rrd_infocall(rrd_graph_v, args);
 216 }
 217
 218
 219 /* Other Calls */
 220
 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;
 231
 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);
 237
 238     RRD_CHECK_ERROR names = rb_ary_new();
 239
 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);
 245
 246     k = 0;
 247     data = rb_ary_new();
 248     for (i = start; i <= end; i += step) {
 249         VALUE     line = rb_ary_new2(ds_cnt);
 250
 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);
 258
 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 }
 267
 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;
 277
 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);
 282
 283     RRD_CHECK_ERROR result = rb_ary_new2(3);
 284
 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 }
 297
 298
 299 VALUE rb_rrd_last(
 300     VALUE self,
 301     VALUE args)
 302 {
 303     string_arr a;
 304     time_t    last;
 305
 306     a = string_arr_new(args);
 307     reset_rrd_state();
 308     last = rrd_last(a.len, a.strings);
 309     string_arr_delete(a);
 310
 311     RRD_CHECK_ERROR
 312         return rb_funcall(rb_cTime, rb_intern("at"), 1, INT2FIX(last));
 313 }
 314
 315 void Init_RRD(
 316     )
 317 {
 318     mRRD = rb_define_module("RRD");
 319     rb_eRRDError = rb_define_class("RRDError", rb_eStandardError);
 320
 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 }