VALUE rb_rrd_graph(
VALUE self,
VALUE args)
{
string_arr a;
char **calcpr, **p;
VALUE result, print_results;
int xsize, ysize;
double ymin, ymax;
a = string_arr_new(args);
reset_rrd_state();
rrd_graph(a.len, a.strings, &calcpr, &xsize, &ysize, NULL, &ymin, &ymax);
string_arr_delete(a);
RRD_CHECK_ERROR result = rb_ary_new2(3);
print_results = rb_ary_new();
p = calcpr;
for (p = calcpr; p && *p; p++) {
rb_ary_push(print_results, rb_str_new2(*p));
rrd_freemem(*p);
}
rrd_freemem(calcpr);
rb_ary_store(result, 0, print_results);
rb_ary_store(result, 1, INT2FIX(xsize));
rb_ary_store(result, 2, INT2FIX(ysize));
return result;
}
static int ntop_rrd_fetch(lua_State* L) {
int argc = lua_gettop(L) + 1;
char **argv = make_argv("fetch", L);
unsigned long i, j, step, ds_cnt;
rrd_value_t *data, *p;
char **names;
time_t t, start, end;
ntop->rrdLock(__FILE__, __LINE__);
reset_rrd_state();
rrd_fetch(argc, argv, &start, &end, &step, &ds_cnt, &names, &data);
free(argv);
if (rrd_test_error()) luaL_error(L, rrd_get_error());
lua_pushnumber(L, (lua_Number) start);
lua_pushnumber(L, (lua_Number) step);
/* fprintf(stderr, "%lu, %lu, %lu, %lu\n", start, end, step, num_points); */
/* create the ds names array */
lua_newtable(L);
for (i=0; i<ds_cnt; i++) {
lua_pushstring(L, names[i]);
lua_rawseti(L, -2, i+1);
rrd_freemem(names[i]);
}
rrd_freemem(names);
/* create the data points array */
lua_newtable(L);
p = data;
for (t=start, i=0; t<end; t+=step, i++) {
lua_newtable(L);
for (j=0; j<ds_cnt; j++) {
/*fprintf(stderr, "Point #%lu\n", j+1); */
lua_pushnumber(L, (lua_Number) *p++);
lua_rawseti(L, -2, j+1);
}
lua_rawseti(L, -2, i+1);
}
rrd_freemem(data);
ntop->rrdUnlock(__FILE__, __LINE__);
/* return the end as the last value */
lua_pushnumber(L, (lua_Number) end);
return 5;
}
VALUE rb_rrd_fetch(
VALUE self,
VALUE args)
{
string_arr a;
unsigned long i, j, k, step, ds_cnt;
rrd_value_t *raw_data;
char **raw_names;
VALUE data, names, result;
time_t start, end;
a = string_arr_new(args);
reset_rrd_state();
rrd_fetch(a.len, a.strings, &start, &end, &step, &ds_cnt, &raw_names,
&raw_data);
string_arr_delete(a);
RRD_CHECK_ERROR names = rb_ary_new();
for (i = 0; i < ds_cnt; i++) {
rb_ary_push(names, rb_str_new2(raw_names[i]));
rrd_freemem(raw_names[i]);
}
rrd_freemem(raw_names);
k = 0;
data = rb_ary_new();
for (i = start; i <= end; i += step) {
VALUE line = rb_ary_new2(ds_cnt);
for (j = 0; j < ds_cnt; j++) {
rb_ary_store(line, j, rb_float_new(raw_data[k]));
k++;
}
rb_ary_push(data, line);
}
rrd_freemem(raw_data);
result = rb_ary_new2(5);
rb_ary_store(result, 0, INT2NUM(start));
rb_ary_store(result, 1, INT2NUM(end));
rb_ary_store(result, 2, names);
rb_ary_store(result, 3, data);
rb_ary_store(result, 4, INT2FIX(step));
return result;
}
static PyObject *PyRRD_graph(
PyObject UNUSED(*self),
PyObject * args)
{
PyObject *r;
char **argv, **calcpr;
int argc, xsize, ysize, i;
double ymin, ymax;
if (create_args("graph", args, &argc, &argv) < 0)
return NULL;
if (rrd_graph(argc, argv, &calcpr, &xsize, &ysize, NULL, &ymin, &ymax) ==
-1) {
PyErr_SetString(ErrorObject, rrd_get_error());
rrd_clear_error();
r = NULL;
} else {
r = PyTuple_New(3);
PyTuple_SET_ITEM(r, 0, PyInt_FromLong((long) xsize));
PyTuple_SET_ITEM(r, 1, PyInt_FromLong((long) ysize));
if (calcpr) {
PyObject *e, *t;
e = PyList_New(0);
PyTuple_SET_ITEM(r, 2, e);
for (i = 0; calcpr[i]; i++) {
t = PyString_FromString(calcpr[i]);
PyList_Append(e, t);
Py_DECREF(t);
rrd_freemem(calcpr[i]);
}
rrd_freemem(calcpr);
} else {
Py_INCREF(Py_None);
PyTuple_SET_ITEM(r, 2, Py_None);
}
}
destroy_args(&argv);
return r;
}
static PyObject *PyRRD_xport(
PyObject UNUSED(*self),
PyObject * args)
{
PyObject *r;
int argc, xsize;
char **argv, **legend_v;
time_t start, end;
unsigned long step, col_cnt;
rrd_value_t *data, *datai;
if (create_args("xport", args, &argc, &argv) < 0)
return NULL;
if (rrd_xport(argc, argv, &xsize, &start, &end,
&step, &col_cnt, &legend_v, &data) == -1) {
PyErr_SetString(ErrorObject, rrd_get_error());
rrd_clear_error();
r = NULL;
} else {
PyObject *meta_dict, *data_list, *legend_list, *t;
unsigned long i, j;
rrd_value_t dv;
unsigned long row_cnt = ((end - start) / step);
r = PyDict_New();
meta_dict = PyDict_New();
legend_list = PyList_New(col_cnt);
data_list = PyList_New(row_cnt);
PyDict_SetItem(r, PyString_FromString("meta"), meta_dict);
PyDict_SetItem(r, PyString_FromString("data"), data_list);
datai = data;
PyDict_SetItem(meta_dict, PyString_FromString("start"), PyInt_FromLong((long) start));
PyDict_SetItem(meta_dict, PyString_FromString("end"), PyInt_FromLong((long) end));
PyDict_SetItem(meta_dict, PyString_FromString("step"), PyInt_FromLong((long) step));
PyDict_SetItem(meta_dict, PyString_FromString("rows"), PyInt_FromLong((long) row_cnt));
PyDict_SetItem(meta_dict, PyString_FromString("columns"), PyInt_FromLong((long) col_cnt));
PyDict_SetItem(meta_dict, PyString_FromString("legend"), legend_list);
for (i = 0; i < col_cnt; i++) {
PyList_SET_ITEM(legend_list, i, PyString_FromString(legend_v[i]));
}
for (i = 0; i < row_cnt; i++) {
t = PyTuple_New(col_cnt);
PyList_SET_ITEM(data_list, i, t);
for (j = 0; j < col_cnt; j++) {
dv = *(datai++);
if (isnan(dv)) {
PyTuple_SET_ITEM(t, j, Py_None);
Py_INCREF(Py_None);
} else {
PyTuple_SET_ITEM(t, j, PyFloat_FromDouble((double) dv));
}
}
}
for (i = 0; i < col_cnt; i++) {
rrd_freemem(legend_v[i]);
}
rrd_freemem(legend_v);
rrd_freemem(data);
}
destroy_args(&argv);
return r;
}
static PyObject *PyRRD_fetch(
PyObject UNUSED(*self),
PyObject * args)
{
PyObject *r;
rrd_value_t *data, *datai;
unsigned long step, ds_cnt;
time_t start, end;
int argc;
char **argv, **ds_namv;
if (create_args("fetch", args, &argc, &argv) < 0)
return NULL;
if (rrd_fetch(argc, argv, &start, &end, &step,
&ds_cnt, &ds_namv, &data) == -1) {
PyErr_SetString(ErrorObject, rrd_get_error());
rrd_clear_error();
r = NULL;
} else {
/* Return :
((start, end, step), (name1, name2, ...), [(data1, data2, ..), ...]) */
PyObject *range_tup, *dsnam_tup, *data_list, *t;
unsigned long i, j, row;
rrd_value_t dv;
row = (end - start) / step;
r = PyTuple_New(3);
range_tup = PyTuple_New(3);
dsnam_tup = PyTuple_New(ds_cnt);
data_list = PyList_New(row);
PyTuple_SET_ITEM(r, 0, range_tup);
PyTuple_SET_ITEM(r, 1, dsnam_tup);
PyTuple_SET_ITEM(r, 2, data_list);
datai = data;
PyTuple_SET_ITEM(range_tup, 0, PyInt_FromLong((long) start));
PyTuple_SET_ITEM(range_tup, 1, PyInt_FromLong((long) end));
PyTuple_SET_ITEM(range_tup, 2, PyInt_FromLong((long) step));
for (i = 0; i < ds_cnt; i++)
PyTuple_SET_ITEM(dsnam_tup, i, PyString_FromString(ds_namv[i]));
for (i = 0; i < row; i++) {
t = PyTuple_New(ds_cnt);
PyList_SET_ITEM(data_list, i, t);
for (j = 0; j < ds_cnt; j++) {
dv = *(datai++);
if (isnan(dv)) {
PyTuple_SET_ITEM(t, j, Py_None);
Py_INCREF(Py_None);
} else {
PyTuple_SET_ITEM(t, j, PyFloat_FromDouble((double) dv));
}
}
}
for (i = 0; i < ds_cnt; i++)
rrd_freemem(ds_namv[i]);
rrd_freemem(ds_namv); /* rrdtool don't use PyMem_Malloc :) */
rrd_freemem(data);
}
destroy_args(&argv);
return r;
}
