Merge rkern#10 from SyamGadde/line_profiler

Enable tracing for all threads.

Author: ahippo

_line_profiler.pyx

@@ -1,5 +1,6 @@
 from python25 cimport PyFrameObject, PyObject, PyStringObject
 
+import threading
 
 cdef extern from "frameobject.h":
     ctypedef int (*Py_tracefunc)(object self, PyFrameObject *py_frame, int what, PyObject *arg)
@@ -108,15 +109,15 @@ cdef class LineProfiler:
     cdef public dict code_map
     cdef public dict last_time
     cdef public double timer_unit
-    cdef public long enable_count
+    cdef public object threaddata
     cdef public bint profile_all
 
     def __init__(self, *functions):
         self.functions = []
         self.code_map = {}
         self.last_time = {}
         self.timer_unit = hpTimerUnit()
-        self.enable_count = 0
+        self.threaddata = threading.local()
         self.profile_all = False
         for func in functions:
             self.add_function(func)
@@ -144,6 +145,14 @@ cdef class LineProfiler:
         """
         self.profile_all = False
 
+    property enable_count:
+        def __get__(self):
+            if not hasattr(self.threaddata, 'enable_count'):
+                self.threaddata.enable_count = 0
+            return self.threaddata.enable_count
+        def __set__(self, value):
+            self.threaddata.enable_count = value
+
     def enable_by_count(self):
         """ Enable the profiler if it hasn't been enabled before.
         """