U hd@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddZ iZ ddZ d d Z d d ZejZGd ddZddZddZeeeddZddZddZddZddZGdddejZdd Zeeje<Gd!d"d"ejZGd#d$d$ZdS)%z/ Serialization support for compiled functions. N) cloudpickle)ircGs |j|S)zM Global hook to rebuild a given class from its __reduce__ arguments. )_rebuild)clsargsrM/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/core/serialize.py_rebuild_reductionsr cCs@||f}z t|}Wn&tk r:t|t|<}YnX|S)zUsed by `numba_unpickle` from _helperlib.c Parameters ---------- address : int bytedata : bytes hashed : bytes Returns ------- obj : object unpickled object )_unpickled_memoKeyErrorrloads)addressZbytedatahashedkeyobjrrr_numba_unpickles  rc Cs:t}t$}||dd}|||}W5QRX|S)zISimilar to `pickle.dumps()`. Returns the serialized object in bytes. )protocol) NumbaPicklerioBytesIOdumpgetvalue)rpicklerbufpZpickledrrrdumps4s    rcCsZt|\}}}g}t|}|D]*}t|tjr@|t|q ||q |t||fSN) rr iter isinstancerValueappendnexttuple)Z static_excZexc_argsexcZ static_argsZlocinfoZ real_argsZ exc_args_iterargrrrruntime_build_excinfo_struct@s  r&c@s0eZdZdZdZddZddZeddZd S) _CustomPickledaA wrapper for objects that must be pickled with `NumbaPickler`. Standard `pickle` will pick up the implementation registered via `copyreg`. This will spawn a `NumbaPickler` instance to serialize the data. `NumbaPickler` overrides the handling of this type so as not to spawn a new pickler for the object when it is already being pickled by a `NumbaPickler`. ctorstatescCs||_||_dSrr()selfr)r*rrr__init__]sz_CustomPickled.__init__cCstj|j|jffSr)r'rr)r*r+rrr_reduceasz_CustomPickled._reducecCs |||Srr)rr)r*rrrrdsz_CustomPickled._rebuildN) __name__ __module__ __qualname____doc__ __slots__r,r. classmethodrrrrrr'Ps  r'cCst|\}}t||S)zPstandard unpickling for `_CustomPickled`. Uses `NumbaPickler` to load. )r r') serializedr)r*rrr_unpickle__CustomPickledis r6cCst|j|jf}t|ffS)zNstandard pickling for `_CustomPickled`. Uses `NumbaPickler` to dump. )rr)r*r6)cpr5rrr_pickle__CustomPickledrsr8cCstt||ffS)aFor customizing object serialization in `__reduce__`. Object states provided here are used as keyword arguments to the `._rebuild()` class method. Parameters ---------- states : dict Dictionary of object states to be serialized. Returns ------- result : tuple This tuple conforms to the return type requirement for `__reduce__`. )custom_rebuildr')rr*rrr custom_reducesr:cCs|j|j}}|jf|S)ziCustomized object deserialization. This function is referenced internally by `custom_reduce()`. )r)r*r)Zcustom_pickledrr*rrrr9sr9c Cs`tN}t|}z||Wn"tjk rBYW5QRdSXW5QRdSW5QRXdS)zCheck if *obj* can be serialized. Parameters ---------- obj : object Returns -------- can_serialize : bool FTN)rrrrpickle PicklingError)rZfoutrrrris_serialiables r=cCstdt|ddS)Nz Pickling of z is unsupported)r;r<type)rrrr _no_picklesr?cCstj||S)z1This is called on a type to disable pickling )rdisabled_typesadd)typrrrdisable_picklings rCcs"eZdZeZfddZZS)rcs"t||jkrt|t|Sr)r>r@r?superreducer_override)r+r __class__rrrEszNumbaPickler.reducer_override)r/r0r1setr@rE __classcell__rrrFrrsrcCs|Sr)r.)r7rrr_custom_reduce__custompickledsrJc@s<eZdZdZejddZejddZddZ dd Z d S) ReduceMixinznA mixin class for objects that should be reduced by the NumbaPickler instead of the standard pickler. cCstdSrNotImplementedErrorr-rrr_reduce_statesszReduceMixin._reduce_statescKstdSrrL)rkwargsrrrrszReduceMixin._rebuildcCs|jSrrFr-rrr _reduce_classszReduceMixin._reduce_classcCst||Sr)r:rPrNr-rrr __reduce__szReduceMixin.__reduce__N) r/r0r1r2abcabstractmethodrNabstractclassmethodrrPrQrrrrrKs  rKc@s4eZdZdZddZddZddZedd Zd S) PickleCallableByPathaWrap a callable object to be pickled by path to workaround limitation in pickling due to non-pickleable objects in function non-locals. Note: - Do not use this as a decorator. - Wrapped object must be a global that exist in its parent module and it can be imported by `from the_module import the_object`. Usage: >>> def my_fn(x): >>> ... >>> wrapped_fn = PickleCallableByPath(my_fn) >>> # refer to `wrapped_fn` instead of `my_fn` cCs ||_dSr_fn)r+fnrrrr,szPickleCallableByPath.__init__cOs |j||SrrV)r+rrOrrr__call__szPickleCallableByPath.__call__cCst|j|jj|jjffSr)r>rrWr0r/r-rrrrQszPickleCallableByPath.__reduce__cCs|ttj||Sr)getattrsysmodules)rmodnameZfn_pathrrrrszPickleCallableByPath._rebuildN) r/r0r1r2r,rYrQr4rrrrrrUs rU)r2r[rRrcopyregr;ZnumbarZllvmliterr r rrr&r r'r6r8r:r9r=r?rCZ CloudPicklerrrJdispatch_tableABCrKrUrrrrs6