3j7PddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z m Z ddgZejdk(rdnd Zd ZGd d eZGd deZGddeZGddeZGddeZGddeZGddeZGddeZddefdZdZy)N) BaseHandlerregister unregister) b64decode b64encoderegister_handlersunregister_handlerslittle<>cH|jj}|dk(rtS|S)z%translate equals sign to native order=)dtype byteordernative_byteorder)arrrs A/DATA/.local/lib/python3.12/site-packages/jsonpickle/ext/numpy.py get_byteorderrs$ ##I(C/ >Y>ceZdZdZdZy)NumpyBaseHandlerct|dr|j|d<yt|}d}|j|r|t |d}||d<y)Ntostringrz(numpy.record, )hasattrrstr startswithlen)selfrdataprefixs r flatten_dtypezNumpyBaseHandler.flatten_dtypesQ 5* %!NN,DMJE&F'c&kB/!DMrc|d}|jdrtj|}tj|S)Nr){[)rast literal_evalnpr)r r!rs r restore_dtypezNumpyBaseHandler.restore_dtype"s7W    J '$$U+ExxrN)__name__ __module__ __qualname__r#r*rrrrs "rrceZdZdZdZy)NumpyDTypeHandlerc*|j|||SN)r#r objr!s rflattenzNumpyDTypeHandler.flatten*s 3% rc$|j|Sr2)r*r r!s rrestorezNumpyDTypeHandler.restore.s!!$''rNr+r,r-r5r8r.rrr0r0)s (rr0ceZdZdZdZy)NumpyGenericHandlerc|j|jjd||jj |j d|d<|S)NNFresetvalue)r#r newbyteordercontextr5tolistr3s rr5zNumpyGenericHandler.flatten3sI 39911#6= ,,SZZ\,GW  rc|jj|dd}|j|j|S)Nr@Fr>)rBr8r*type)r r!r@s rr8zNumpyGenericHandler.restore8s; $$T']%$@!!$',,U33rNr9r.rrr;r;2s  4rr;ceZdZdZdZy)NumpyDatetimeHandlerzEExtend NumpyGenericHandler to handle nanosecond-resolution datetime64c|jj|dd}|d}|jdr!|j|j |dS|j|j |S)Nr@Fr>rz[ns]ns)rBr8endswithr*rE)r r!r@rs rr8zNumpyDatetimeHandler.restore@sn $$T']%$@W  >>& !%%d+00= =!!$',,U33rN)r+r,r-__doc__r8r.rrrGrG=s O4rrGc(eZdZdZfdZdZxZS)UnpickleableNumpyGenericHandlerz From issue #381, this is used for simplifying the output of numpy arrays when unpicklable=False (the default is True). c|jjs+|jj|jdStt | ||S)NFr>)rB unpicklabler5rCsuperr;)r r4r! __class__s rr5z'UnpickleableNumpyGenericHandler.flattenNsE||''<<'' E'B B,d;CF Frctr2)NotImplementedErrorr7s rr8z'UnpickleableNumpyGenericHandler.restoreTs!!r)r+r,r-rKr5r8 __classcell__rQs@rrMrMHs G "rrMc(eZdZdZdZdZdZdZy)NumpyNDArrayHandlerz>Stores arrays as text representation, without regard for viewsc@|jjdurd|d<yy)NF writeable)flagsrYr3s r flatten_flagsz!NumpyNDArrayHandler.flatten_flags[s# 99  % ' %D  (rcL|jddsd|j_yy)NrYTF)getrZrY)r r!rs r restore_flagsz!NumpyNDArrayHandler.restore_flags_s!xx T*"'CII +rc|j|jjd||j|||jj |j d|d<d|jvr|j|d<|S)Nr=Fr>valuesrshape)r#rrAr[rBr5rCrar3s rr5zNumpyNDArrayHandler.flattencss 39911#6= 3%--cjjl%-HX  > IIDM rc"|jj|dd}tj||j ||j dd}|j dd}||j |}|j|||S)Nr`Fr>orderCrrcra)rBr8r)arrayr*r]reshaper^)r r!r`rras rr8zNumpyNDArrayHandler.restorems%%d8nE%Bhh $,,T2$((7C:P $'  ++e$C 4% rN)r+r,r-rKr[r^r5r8r.rrrWrWXsH&( rrWcDeZdZdZdefdZdZdZfdZfdZ xZ S)NumpyNDArrayHandlerBinaryzstores arrays with size greater than 'size_threshold' as (optionally) compressed base64 Notes ----- This would be easier to implement using np.save/np.load, but that would be less language-agnostic c ||_||_y)a :param size_threshold: nonnegative int or None valid values for 'size_threshold' are all nonnegative integers and None if size_threshold is None, values are always stored as nested lists :param compression: a compression module or None valid values for 'compression' are {zlib, bz2, None} if compression is None, no compression is applied N)size_threshold compression)r rlrms r__init__z"NumpyNDArrayHandlerBinary.__init__s-&rcX|jj}|dk7rt||d<yy)N|r)rrr)r r4r!rs rflatten_byteorderz+NumpyNDArrayHandlerBinary.flatten_byteorders-II''   -c 2D  rcn|jdd}|r!|jj||_yy)Nr)r]rrA)r r!rrs rrestore_byteorderz+NumpyNDArrayHandlerBinary.restore_byteorders0HH[$/  ..y9CI rc|j|j|jk\rt| ||}|S|jt k(r2t j|jj}n1t|dr|jd}n|jd}|jr|jj|}t||d<|j |d<|j#|jj%d||j'|||j)|||j*j,sd|d<|S) encode numpy to jsontobytesa)rcr`rar=Frc)rlsizerPr5robjectjsondumpsrCencoderrvrrmcompressrrar#rArqr[rZ c_contiguous)r r4r!bufrQs rr5z!NumpyNDArrayHandlerBinary.flattens'    &$*=*=*I7?3-DH CyyF"jj.557i(kkk,lll-&&//4&s^DNIIDM   syy55c:D A  " "3 -   sD )99)) #W  rc  |d}t|trt| |}|St|tt fr)t j|g|j|}|S|j|}t|}|jr|jj|}|tk(rqtj|j}t j|||j!dd}|j!dd}|j|j#|}nXt j$|||j!d|j!ddj'}|j)|||j+|||S) decode numpy from jsonr`)rrcrdreraN)bufferrrarc) isinstancelistrPr8intfloatr)rfr*rrm decompressrzr{loadsdecoder]rgndarraycopyrsr^)r r!r`rrrrarQs rr8z!NumpyNDArrayHandlerBinary.restoresYh fd #'/$'C8 7e -((F84+=+=d+CDC2 -&&t,EF#C&&11#6CJJL1hhvU$((7C:PQ$/$++e,Cjj((7+((7C0  $&  &&tS1   tS ) r) r+r,r-rKzlibrnrqrsr5r8rTrUs@rririzs.')d '3 : (T!!rric>eZdZdZddeffd ZfdZfdZxZS)NumpyNDArrayHandlerViewaPickles references inside ndarrays, or array-views Notes ----- The current implementation has some restrictions. 'base' arrays, or arrays which are viewed by other arrays, must be f-or-c-contiguous. This is not such a large restriction in practice, because all numpy array creation is c-contiguous by default. Relaxing this restriction would be nice though; especially if it can be done without bloating the design too much. Furthermore, ndarrays which are views of array-like objects implementing __array_interface__, but which are not themselves nd-arrays, are deepcopied with a warning (by default), as we cannot guarantee whatever custom logic such classes implement is correctly reproduced. warnrjc4t|||||_y)aO :param mode: {'warn', 'raise', 'ignore'} How to react when encountering array-like objects whose references we cannot safely serialize :param size_threshold: nonnegative int or None valid values for 'size_threshold' are all nonnegative integers and None if size_threshold is None, values are always stored as nested lists :param compression: a compression module or None valid values for 'compression' are {zlib, bz2, None} if compression is None, no compression is applied N)rPrnmode)r rrlrmrQs rrnz NumpyNDArrayHandlerView.__init__s 5 rc0|j}|C|jjr-t|||}|jj sd|d<|St |tjrG|jjr0|jj |d|d<|jj|jjz }|r||d<|jj s|j|d<|j|d <|j|jj!d ||j#||t%|d k7r"t%|t%|k7rd nd}|r||d <|j&|j&|j(k\rtt*||||S|j,dk(rd}t/j0|n|j,dk(r d}t3|t||j5|}|S)ruNrxrcFr>baseoffsetstridesrar=rpSrrzndarray is defined by reference to an object we do not know how to serialize. A deep copy is serialized instead, breaking memory aliasing.raisezMndarray is defined by reference to an object we do not know how to serialize.)rrZforcrPr5rrr)rrBctypesr!rrar#rrAr[rrlryrirwarningsr ValueErrorr)r r4r!rrrmsgrQs rr5zNumpyNDArrayHandlerView.flattensxx sDI& !yyF"'  c"g%1!o%7?388:t4D rc J|jdd}|t| |}|S|jj|d}t |t j st j|}|jjsJdt j |j|j|j|jdd|jd|jd d |jd d }|j|||S) rrNFr>z8Current implementation assumes base is C or F contiguousrrprarrr)rrrarr)r]rPr8rBrr)rrfrZrr!r*rAr^)r r!rrrQs rr8zNumpyNDArrayHandlerView.restoreEsxx% <'/$'C* %<<''E':DdBJJ/xx~  JI J**yy((.;;DHH[RUTJ RXXbhh  ) )+<4H RXXbmm $ . .0AM R]]0trs  88'  5 6--833? {$(((4*44.4 "&9 " *Dl 3l^v7vt=6 2r