B   ›\I ã@sÐdZddlmZmZmZddlmZedkr@ddlmZ e ƒZ n ddl m Z ddl Z ddlZdd lmZmZdd lmZmZd d „ZGd d„deƒZdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„ZdS)z/ Serialization support for compiled functions. é)Úprint_functionÚdivisionÚabsolute_importé)Ú PYVERSION)ér)Ú get_magic)Ú MAGIC_NUMBERN)Ú FunctionTypeÚ ModuleType)ÚbytecodeÚcompilercGs |j|ŽS)zM Global hook to rebuild a given class from its __reduce__ arguments. )Z_rebuild)ÚclsÚargs©rú.lib/python3.7/site-packages/numba/serialize.pyÚ_rebuild_reductionsrc@seZdZdd„Zdd„ZdS)Ú _ModuleRefcCs ||_dS)N)Úname)ÚselfrrrrÚ__init__$sz_ModuleRef.__init__cCs t|jffS)N)Ú_rebuild_moduler)rrrrÚ __reduce__'sz_ModuleRef.__reduce__N)Ú__name__Ú __module__Ú __qualname__rrrrrrr"srcCs"|dkrtdƒ‚t|ƒtj|S)Nzcannot import None)Ú ImportErrorÚ __import__ÚsysÚmodules)rrrrr+srcCsZtj |¡}t |¡}| ¡}x,| ¡D] \}}t|tƒr(t|j ƒ||<q(W|j |d<|S)z] Analyse *func* and return a dictionary of global values suitable for reduction. r) r ZFunctionIdentityZ from_functionZByteCodeZget_used_globalsÚitemsÚ isinstancer rrr)ÚfuncZfunc_idZbcÚglobsÚkÚvrrrÚ#_get_function_globals_for_reduction2s    r&cCs0|jrdd„|jDƒ}nd}t|jƒ||j|fS)z¬ Reduce a Python function and its globals to picklable components. If there are cell variables (i.e. references to a closure), their values will be frozen. cSsg|] }|j‘qSr)Ú cell_contents)Ú.0Zcellrrrú Jsz$_reduce_function..N)Ú __closure__Ú _reduce_codeÚ__code__r)r"r#ÚcellsrrrÚ_reduce_functionCsr.cCstjtt |¡fS)z7 Reduce a code object to picklable components. )ÚmarshalÚversionÚbc_magicÚdumps)Úcoderrrr+Osr+cs ‡fdd„S)z= A dummy function allowing us to build cell objects. csˆS)Nrr)ÚxrrÚYsz _dummy_closure..r)r4r)r4rÚ_dummy_closureUsr6cCsd|rtdd„|Dƒƒ}nd}t|Ž}|d}y t|ƒWntk rR|d=YnXt|||d|ƒS)zA Rebuild a function from its _reduce_function() results. css|]}t|ƒjdVqdS)rN)r6r*)r(r%rrrú `sz$_rebuild_function..rr)ÚtupleÚ _rebuild_coderrr )Z code_reducedÚglobalsrZ cell_valuesr-r3ÚmodnamerrrÚ_rebuild_function[s  r<cCs6tj|krtdtj|fƒ‚t|kr,tdƒ‚t |¡S)z@ Rebuild a code object from its _reduce_code() results. zHincompatible marshal version: interpreter has %r, marshalled code has %rzincompatible bytecode version)r/r0Ú RuntimeErrorr1Úloads)Zmarshal_versionZbytecode_magicZ marshalledrrrr9ms  r9)Ú__doc__Z __future__rrrZutilsrÚimprZ _get_magicr1Úimportlib.utilr r/rÚtypesr r Úr r rÚobjectrrr&r.r+r6r<r9rrrrÚs&