xref: /linux/tools/perf/scripts/python/Perf-Trace-Util/Context.c (revision ac3167257b9fe16c9426c2087ead1c9f1b0992b1)
1  /*
2   * Context.c.  Python interfaces for perf script.
3   *
4   * Copyright (C) 2010 Tom Zanussi <tzanussi@gmail.com>
5   *
6   *  This program is free software; you can redistribute it and/or modify
7   *  it under the terms of the GNU General Public License as published by
8   *  the Free Software Foundation; either version 2 of the License, or
9   *  (at your option) any later version.
10   *
11   *  This program is distributed in the hope that it will be useful,
12   *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13   *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14   *  GNU General Public License for more details.
15   *
16   *  You should have received a copy of the GNU General Public License
17   *  along with this program; if not, write to the Free Software
18   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19   *
20   */
21  
22  #include <Python.h>
23  #include "../../../perf.h"
24  #include "../../../util/trace-event.h"
25  
26  #if PY_MAJOR_VERSION < 3
27  #define _PyCapsule_GetPointer(arg1, arg2) \
28    PyCObject_AsVoidPtr(arg1)
29  
30  PyMODINIT_FUNC initperf_trace_context(void);
31  #else
32  #define _PyCapsule_GetPointer(arg1, arg2) \
33    PyCapsule_GetPointer((arg1), (arg2))
34  
35  PyMODINIT_FUNC PyInit_perf_trace_context(void);
36  #endif
37  
38  static PyObject *perf_trace_context_common_pc(PyObject *obj, PyObject *args)
39  {
40  	static struct scripting_context *scripting_context;
41  	PyObject *context;
42  	int retval;
43  
44  	if (!PyArg_ParseTuple(args, "O", &context))
45  		return NULL;
46  
47  	scripting_context = _PyCapsule_GetPointer(context, NULL);
48  	retval = common_pc(scripting_context);
49  
50  	return Py_BuildValue("i", retval);
51  }
52  
53  static PyObject *perf_trace_context_common_flags(PyObject *obj,
54  						 PyObject *args)
55  {
56  	static struct scripting_context *scripting_context;
57  	PyObject *context;
58  	int retval;
59  
60  	if (!PyArg_ParseTuple(args, "O", &context))
61  		return NULL;
62  
63  	scripting_context = _PyCapsule_GetPointer(context, NULL);
64  	retval = common_flags(scripting_context);
65  
66  	return Py_BuildValue("i", retval);
67  }
68  
69  static PyObject *perf_trace_context_common_lock_depth(PyObject *obj,
70  						      PyObject *args)
71  {
72  	static struct scripting_context *scripting_context;
73  	PyObject *context;
74  	int retval;
75  
76  	if (!PyArg_ParseTuple(args, "O", &context))
77  		return NULL;
78  
79  	scripting_context = _PyCapsule_GetPointer(context, NULL);
80  	retval = common_lock_depth(scripting_context);
81  
82  	return Py_BuildValue("i", retval);
83  }
84  
85  static PyMethodDef ContextMethods[] = {
86  	{ "common_pc", perf_trace_context_common_pc, METH_VARARGS,
87  	  "Get the common preempt count event field value."},
88  	{ "common_flags", perf_trace_context_common_flags, METH_VARARGS,
89  	  "Get the common flags event field value."},
90  	{ "common_lock_depth", perf_trace_context_common_lock_depth,
91  	  METH_VARARGS,	"Get the common lock depth event field value."},
92  	{ NULL, NULL, 0, NULL}
93  };
94  
95  #if PY_MAJOR_VERSION < 3
96  PyMODINIT_FUNC initperf_trace_context(void)
97  {
98  	(void) Py_InitModule("perf_trace_context", ContextMethods);
99  }
100  #else
101  PyMODINIT_FUNC PyInit_perf_trace_context(void)
102  {
103  	static struct PyModuleDef moduledef = {
104  		PyModuleDef_HEAD_INIT,
105  		"perf_trace_context",	/* m_name */
106  		"",			/* m_doc */
107  		-1,			/* m_size */
108  		ContextMethods,		/* m_methods */
109  		NULL,			/* m_reload */
110  		NULL,			/* m_traverse */
111  		NULL,			/* m_clear */
112  		NULL,			/* m_free */
113  	};
114  	return PyModule_Create(&moduledef);
115  }
116  #endif
117