xKgldZddlZddlZddlZddlZddlZddlmZddlm Z dZ iZ dZ dZ dZejZGd d Zd Zd Zej eed ZdZdZdZdZGddej0ZdZeej6e<Gddej8ZGddZy)z/ Serialization support for compiled functions. N) cloudpickle)irc |j|S)zM Global hook to rebuild a given class from its __reduce__ arguments. )_rebuild)clsargss X/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/numba/core/serialize.py_rebuild_reductionr s 3<< c~||f} t|}|S#t$r"tj|xt|<}Y|SwxYw)zUsed by `numba_unpickle` from _helperlib.c Parameters ---------- address : int bytedata : bytes hashed : bytes Returns ------- obj : object unpickled object )_unpickled_memoKeyErrorrloads)addressbytedatahashedkeyobjs r _numba_unpicklersW F CAc" J A%0%6%6x%@@s JAs '<<ct}tj5}||d}|j||j }ddd|S#1swYSxYw)zISimilar to `pickle.dumps()`. Returns the serialized object in bytes. )protocolN) NumbaPicklerioBytesIOdumpgetvalue)rpicklerbufppickleds r dumpsr"4sOG  C! $ s ,,.  N  Ns ,AActj|\}}}g}t|}|D]H}t|tj r|j t|8|j |J|t||fSN) rriter isinstancerValueappendnexttuple) static_excexc_argsexc static_argslocinfo real_args exc_args_iterargs r runtime_build_excinfo_structr3@st + 1 1* =CgINM c288 $   T-0 1   S !  y!7 ++r c0eZdZdZdZdZdZedZy)_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`. ctorstatesc ||_||_yr$r6)selfr7r8s r __init__z_CustomPickled.__init__]s  r cRtj|j|jffSr$)r5rr7r8r:s r _reducez_CustomPickled._reduceas &&DKK(@@@r c|||Sr$)rr7r8s r rz_CustomPickled._rebuildds4  r N) __name__ __module__ __qualname____doc__ __slots__r;r> classmethodrr@r r r5r5Ps-!IA!!r r5c6t|\}}t||S)zPstandard unpickling for `_CustomPickled`. Uses `NumbaPickler` to load. )rr5) serializedr7r8s r _unpickle__CustomPickledrIis $LD& $ ''r cVt|j|jf}t|ffS)zNstandard pickling for `_CustomPickled`. Uses `NumbaPickler` to dump. )r"r7r8rI)cprHs r _pickle__CustomPickledrLrs( +,J #j] 22r c(tt||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_rebuildr5)rr8s r custom_reducerOs N379 99r cV|j|j}}|jdi|S)ziCustomized object deserialization. This function is referenced internally by `custom_reduce()`. r@)r7r8r)custom_pickledrr8s r rNrNs- !%%~'<'<C 3<< !& !!r ctj5}t|} |j| dddy#tj $r YdddywxYw#1swYyxYw)zCheck if *obj* can be serialized. Parameters ---------- obj : object Returns -------- can_serialize : bool NTF)rrrrpickle PicklingError)rfoutrs r is_serialiablerVs_ t$  LL  ##    s+ A=AAAAAA'cFtjdt|d)Nz Pickling of z is unsupported)rSrTtype)rs r _no_picklerYs   d3i[H IIr cDtjj||S)z1This is called on a type to disable pickling )rdisabled_typesadd)typs r disable_picklingr^s##C( Jr c.eZdZeZ fdZxZS)rcft||jvr t|t||Sr$)rXr[rYsuperreducer_override)r:r __class__s r rbzNumbaPickler.reducer_overrides, 9++ + sOw',,r )rArBrCsetr[rb __classcell__rcs@r rrsUN--r rc"|jSr$)r>)rKs r _custom_reduce__custompickledrhs ::<r cdeZdZdZej dZejdZdZ dZ y) ReduceMixinznA mixin class for objects that should be reduced by the NumbaPickler instead of the standard pickler. ctr$NotImplementedErrorr=s r _reduce_stateszReduceMixin._reduce_states!!r c tr$rl)rkwargss r rzReduceMixin._rebuildror c|jSr$rfr=s r _reduce_classzReduceMixin._reduce_classs ~~r cRt|j|jSr$)rOrsrnr=s r __reduce__zReduceMixin.__reduce__s!T//143F3F3HIIr N) rArBrCrDabcabstractmethodrnabstractclassmethodrrsrur@r r rjrjsF  "" ""  Jr rjc2eZdZdZdZdZdZedZy)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` c||_yr$_fn)r:fns r r;zPickleCallableByPath.__init__s r c&|j|i|Sr$r|)r:rrqs r __call__zPickleCallableByPath.__call__stxx(((r ct|j|jj|jjffSr$)rXrr}rBrAr=s r ruzPickleCallableByPath.__reduce__s0Dz""TXX%8%8$((:K:K$MMMr cH|ttj||Sr$)getattrsysmodules)rmodnamefn_paths r rzPickleCallableByPath._rebuilds73;;w/9::r N) rArBrCrDr;rrurFrr@r r rzrzs+)N;;r rz)rDrrvrcopyregrSnumbarllvmliterr r rr"r3rr5rIrLrOrNrVrYr^ CloudPicklerrrhdispatch_tableABCrjrzr@r r rs ,  , !!2(3~56:&"*J -;++ -/L N+J#''J2;;r