No more speed penalty for settrace(gettrace()). #397.

1 files changed, 4 insertions, 0 deletions

diff --git a/coverage/ctracer/tracer.c b/coverage/ctracer/tracer.c
index 23472575..e5f39d09 100644
--- a/coverage/ctracer/tracer.c
+++ b/coverage/ctracer/tracer.c
@@ -903,6 +903,10 @@ CTracer_call(CTracer *self, PyObject *args, PyObject *kwds)
     /* Clean up. */
     frame->f_lineno = orig_lineno;
 
+    /* For better speed, install ourselves the C way so that future calls go
+       directly to CTracer_trace, without this intermediate function. */
+    PyEval_SetTrace((Py_tracefunc)CTracer_trace, (PyObject*)self);
+
 done:
     return ret;
 }