o 1&i@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)clsargsr?C:\wamp64\www\opt\env\Lib\site-packages\numba/core/serialize.py_rebuild_reductions r cCs>||f}zt|}W|Styt|t|<}Y|Sw)zUsed by `numba_unpickle` from _helperlib.c Parameters ---------- address : int bytedata : bytes hashed : bytes Returns ------- obj : object unpickled object )_unpickled_memoKeyErrorrloads)addressZbytedatahashedkeyobjrrr_numba_unpickles  rcCsPt}t}||dd}|||}Wd|S1s!wY|S)zISimilar to `pickle.dumps()`. Returns the serialized object in bytes. )protocolN) 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__]s z_CustomPickled.__init__cCstj|j|jffSr)r'rr)r*r+rrr_reduceaz_CustomPickled._reducecCs |||Srr)rr)r*rrrrds z_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__CustomPickledrs r9cCstt||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}}|jdi|S)ziCustomized object deserialization. This function is referenced internally by `custom_reduce()`. Nr)r)r*r)Zcustom_pickledrr*rrrr:sr:c Csjt'}t|}z||Wntjy!YWddSw WddS1s.wYdS)zCheck if *obj* can be serialized. Parameters ---------- obj : object Returns -------- can_serialize : bool NFT)rrrrpickle PicklingError)rZfoutrrrris_serialiables $r>cCstdt|d)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$eZdZeZ fddZZS)rcs"t||jvr t|t|Sr)r?rAr@superreducer_override)r+r __class__rrrFs zNumbaPickler.reducer_override)r0r1r2setrArF __classcell__rrrGrrsrcCs|Sr)r.)r8rrr_custom_reduce__custompickledsrKc@s@eZdZdZejddZeejddZddZ dd Z d S) ReduceMixinznA mixin class for objects that should be reduced by the NumbaPickler instead of the standard pickler. cCtrNotImplementedErrorr-rrr_reduce_statesszReduceMixin._reduce_statescKrMrrN)rkwargsrrrrszReduceMixin._rebuildcCs|jSrrGr-rrr _reduce_classszReduceMixin._reduce_classcCst||Sr)r;rRrPr-rrr __reduce__r/zReduceMixin.__reduce__N) r0r1r2r3abcabstractmethodrPr5rrRrSrrrrrLs   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,s zPickleCallableByPath.__init__cOs|j|i|SrrW)r+rrQrrr__call__szPickleCallableByPath.__call__cCst|j|jj|jjffSr)r?rrXr1r0r-rrrrSszPickleCallableByPath.__reduce__cCs|ttj||Sr)getattrsysmodules)rmodnameZfn_pathrrrrszPickleCallableByPath._rebuildN) r0r1r2r3r,rZrSr5rrrrrrVsrV)r3r\rTrcopyregr<ZnumbarZllvmliterr r rrr&r r'r7r9r;r:r>r@rDZ CloudPicklerrrKdispatch_tableABCrLrVrrrrs8