B \<@sddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z Gdddej ZeZd d Zd d Zd dZddZddZddZddZe jredZeejege_eZeZneZeZdS))absolute_importN)wraps)chain)configc@seZdZdZddZdS)TLSzGUse a subclass to properly initialize the TLS variables in all threads.cCsd|_d|_dS)NFr)tracingindent)selfr ,lib/python3.7/site-packages/numba/tracing.py__init__sz TLS.__init__N)__name__ __module__ __qualname____doc__r r r r r r srcCst|dd}t|dd}t|dd}d}|r4|j}nHt|jrZ|jddkrZ|djj}n"t|jr||jddkr||dj}|rg}|r|dkr|||d |r|||d ||d |}|dfS) z7Return function meta-data in a tuple. (name, type)rNr__self__rr cls__main__.)getattrrlenargs __class__appendjoin)funcspecrmodulenamer ZcnameZqnamer r r find_function_infos*           r!cCs8d}t|}t||kr0|d|d|dS|SdS)Ni@z...)reprr)valueZMAX_SIZEsr r r chop3s  r&c st|jr.tt|jt|j d|j|tt|jdt||fdd|jD}dd|t|dD}fddtD}d ddt |||D}dt j d|d |d g}d t j d|g} || fS) Ncsg|]}d||fqS)z%s=%r)pop).0a)valuesr r Bsz!create_events..cSsg|] }t|qSr )str)r(r)r r r r+Cscsg|]}d||fqS)z%s=%rr )r(k)r*r r r+Dsz, cSsg|] }|r|qSr r )r(fr r r r+Esz>>  ()z<< ) dictdefaultsziprrupdatelistsortedkeysrrtlsr ) fnamerrkwdsZ positionalZ anonymouskeywordsZparamsenterleaver )r*r create_events;s" "r?c OsR|dd}dd}t|r$|dp&d}|rtt|rx*t|tjD]\}}t||||qJWxht|tjD]\}}t |f|d|iqvWn8t|rx,t|ddD]\}}t||||qWt |st |t t fkr||St |tkrJd \}} } |jr||j}|jr,||j} |jr>||j} t|| | S|SdS) zFunction decorator to trace a function's entry and exit. *args: categories in which to trace this function. Example usage: @trace def function(...):... @trace('mycategory') def function(...):... recursiveFcsdtdfdd}d}dd}ttkrLt}djn0ttkrlt}dnttkr|tt |t |S)Ntracec s|tjrtjr||St|\}}z@dt_t|||\}}zd|tj d7_ y"zdt_||}Wddt_XWnt \}}} | d|j } | dkr| | | d| |j|jr| d| d d d |jD| d n | d Yn$X|dk rH| d| t|Wdtj d8_ d|XWddt_X|S)NTrrFz => exception thrown raise rrr0z, css|]}t|VqdS)N)r&)r(vr r r |sz>dotrace..decorator..wrapper..r1z()z -> )Z isEnabledForloggingINFOr9rr!r?inforr sysexc_inforrrrr&) rr;r:Zftyper=r>resulttyper$ tracebackZmname)rloggerrr r wrapper_sF           z+dotrace..decorator..wrappercSs|S)Nr )xr r r sz,dotrace..decorator..T) rD getLoggerrJ classmethod__get____func__ staticmethodpropertyNotImplementedErrorinspectZ getargspecr)rrMrIZrewrapr )rrLrr decorator[s .     zdotrace..decoratorrNcSst|pt|S)N)rW isfunctionZismethod)rNr r r rOs zdotrace..)NNN)getrrVrWZismoduleZ getmembersrYsetattrZisclassdotracecallablerJrQrTrUfgetfsetfdel) rr;r@rXarg0nr.cZpgetZpsetZpdelr r r r\Ls2 D       r\cOsDdd}t|r|dpd}t|s4t|ttfkr<||S|SdS)z)Just a no-op in case tracing is disabled.cSs|S)Nr )rr r r rXsznotrace..decoratorrN)rr]rJrQrT)rr;rXrar r r notraces rdcCs.dtjd|g}td}|d|dS)Nz== r/rAr)r9r rDrPrFr)msgrLr r r doevents rfcCsdS)Nr )rer r r noeventsrgrA) Z __future__rrDrGtypesZ threadingrW functoolsr itertoolsrrrZlocalrr9r!r&r?r\rdrfrgZTRACErPrLZsetLevelrEZ StreamHandlerZhandlersrAZeventr r r r s2    s