Messages in this thread | | | Date | Mon, 31 May 2010 23:23:15 +0200 | From | Frederic Weisbecker <> | Subject | Re: [PATCH] trace-event-python: give field dict to unhandled callback |
| |
On Mon, May 31, 2010 at 11:12:09PM +0200, Pierre Tardy wrote: > trace_unhandled() callback does not allow to access event fields, this > patch resolves the problem. > > It can also been used as a more pythonic and flexible way for script > writters to demux event types > > This will for example greatly simplify pytimechart event demux. > > Signed-off-by: Pierre Tardy <tardyp@gmail.com> > --- > tools/perf/scripts/python/check-perf-trace.py | 3 +- > .../util/scripting-engines/trace-event-python.c | 50 +++++++++++++------ > 2 files changed, 35 insertions(+), 18 deletions(-) > > diff --git a/tools/perf/scripts/python/check-perf-trace.py b/tools/perf/scripts/python/check-perf-trace.py > index 964d934..d9f7893 100644 > --- a/tools/perf/scripts/python/check-perf-trace.py > +++ b/tools/perf/scripts/python/check-perf-trace.py > @@ -51,8 +51,7 @@ def kmem__kmalloc(event_name, context, common_cpu, > > flag_str("kmem__kmalloc", "gfp_flags", gfp_flags)), > > -def trace_unhandled(event_name, context, common_cpu, common_secs, common_nsecs, > - common_pid, common_comm): > +def trace_unhandled(event_name, context, event_fields_dict): > try: > unhandled[event_name] += 1 > except TypeError: > diff --git a/tools/perf/util/scripting-engines/trace-event-python.c b/tools/perf/util/scripting-engines/trace-event-python.c > index 81f39ca..33a6325 100644 > --- a/tools/perf/util/scripting-engines/trace-event-python.c > +++ b/tools/perf/util/scripting-engines/trace-event-python.c > @@ -208,7 +208,7 @@ static void python_process_event(int cpu, void *data, > int size __unused, > unsigned long long nsecs, char *comm) > { > - PyObject *handler, *retval, *context, *t, *obj; > + PyObject *handler, *retval, *context, *t, *obj, *dict = NULL; > static char handler_name[256]; > struct format_field *field; > unsigned long long val; > @@ -232,6 +232,14 @@ static void python_process_event(int cpu, void *data, > > sprintf(handler_name, "%s__%s", event->system, event->name); > > + handler = PyDict_GetItemString(main_dict, handler_name); > + if (handler && !PyCallable_Check(handler)) > + handler = NULL; > + if (!handler) { > + dict = PyDict_New(); > + if (!dict) > + Py_FatalError("couldn't create Python dict"); > + } > s = nsecs / NSECS_PER_SEC; > ns = nsecs - s * NSECS_PER_SEC; > > @@ -242,12 +250,20 @@ static void python_process_event(int cpu, void *data, > PyTuple_SetItem(t, n++, PyString_FromString(handler_name)); > PyTuple_SetItem(t, n++, > PyCObject_FromVoidPtr(scripting_context, NULL)); > - PyTuple_SetItem(t, n++, PyInt_FromLong(cpu)); > - PyTuple_SetItem(t, n++, PyInt_FromLong(s)); > - PyTuple_SetItem(t, n++, PyInt_FromLong(ns)); > - PyTuple_SetItem(t, n++, PyInt_FromLong(pid)); > - PyTuple_SetItem(t, n++, PyString_FromString(comm)); > > + if (handler) { > + PyTuple_SetItem(t, n++, PyInt_FromLong(cpu)); > + PyTuple_SetItem(t, n++, PyInt_FromLong(s)); > + PyTuple_SetItem(t, n++, PyInt_FromLong(ns)); > + PyTuple_SetItem(t, n++, PyInt_FromLong(pid)); > + PyTuple_SetItem(t, n++, PyString_FromString(comm)); > + } else { > + PyDict_SetItemString(dict, "common_cpu", PyInt_FromLong(cpu)); > + PyDict_SetItemString(dict, "common_s", PyInt_FromLong(s)); > + PyDict_SetItemString(dict, "common_ns", PyInt_FromLong(ns)); > + PyDict_SetItemString(dict, "common_pid", PyInt_FromLong(pid)); > + PyDict_SetItemString(dict, "common_comm", PyString_FromString(comm)); > + } > for (field = event->format.fields; field; field = field->next) { > if (field->flags & FIELD_IS_STRING) { > int offset; > @@ -272,27 +288,31 @@ static void python_process_event(int cpu, void *data, > obj = PyLong_FromUnsignedLongLong(val); > } > } > - PyTuple_SetItem(t, n++, obj); > + if (handler) > + PyTuple_SetItem(t, n++, obj); > + else > + PyDict_SetItemString(dict, field->name, obj); > + > } > + if (!handler) > + PyTuple_SetItem(t, n++, dict); > > if (_PyTuple_Resize(&t, n) == -1) > Py_FatalError("error resizing Python tuple"); > > - handler = PyDict_GetItemString(main_dict, handler_name); > - if (handler && PyCallable_Check(handler)) { > + if (handler) { > retval = PyObject_CallObject(handler, t); > if (retval == NULL) > handler_call_die(handler_name); > } else { > handler = PyDict_GetItemString(main_dict, "trace_unhandled"); > if (handler && PyCallable_Check(handler)) { > - if (_PyTuple_Resize(&t, N_COMMON_FIELDS) == -1) > - Py_FatalError("error resizing Python tuple"); > > retval = PyObject_CallObject(handler, t); > if (retval == NULL) > handler_call_die("trace_unhandled"); > } > + Py_DECREF(dict);
Cool!
Acked-by: Frederic Weisbecker <fweisbec@gmail.com>
| |