U |e@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)clsargsrQ/var/www/website-v5/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_excexc_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 r7cCst|j|jf}t|ffS)zNstandard pickling for `_CustomPickled`. Uses `NumbaPickler` to dump. )rr*r+r7)cpr6rrr_pickle__CustomPickledrsr9cCstt||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+rrrr:sr:c 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 rDcs"eZdZeZfddZZS)rcs"t||jkrt|t|Sr)r?rAr@superreducer_override)r,r __class__rrrFszNumbaPickler.reducer_override)r0r1r2setrArF __classcell__rrrGrrsrcCs|Sr)r/)r8rrr_custom_reduce__custompickledsrKc@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_statescKstdSrrM)rkwargsrrrrszReduceMixin._rebuildcCs|jSrrGr.rrr _reduce_classszReduceMixin._reduce_classcCst||Sr)r;rQrOr.rrr __reduce__szReduceMixin.__reduce__N) r0r1r2r3abcabstractmethodrOabstractclassmethodrrQrRrrrrrLs  rLc@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||SrrW)r,rrPrrr__call__szPickleCallableByPath.__call__cCst|j|jj|jjffSr)r?rrXr1r0r.rrrrRszPickleCallableByPath.__reduce__cCs|ttj||Sr)getattrsysmodules)rmodnameZfn_pathrrrrszPickleCallableByPath._rebuildN) r0r1r2r3r-rZrRr5rrrrrrVs rV)r3r\rSrcopyregr<numbarllvmliterr r rrr'r r(r7r9r;r:r>r@rD CloudPicklerrrKdispatch_tableABCrLrVrrrrs6