"f, UdZddlmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZeeeZeed efZd Zeed <d eed edeeddffdZGdd eZdS),Array to bytes and bytes to array transform.)Any GeneratorListOptionalSequenceTupleUnion)Format)DataMeta)Record)IntX) TransformArrayXN GREEDY_DIMSarrayndimsreturnNc#`Kt|D]}t|V|d}dS)z8Determine dimensions from multi-dimensional array value.rN)rangelen)rr_s N/mounts/lovelace/software/anaconda3/lib/python3.11/site-packages/plum/array.py iterate_dimsrsB 5\\%jjac eZdZUdZeed<eeefed< ddede ede e ddffd Z e defd Z e defd Z dd ed ede ede edeeeeff dZ dd ed edede edeeeeff dZ ddeedeede ede eddf dZ ddeedeedede eddf dZxZS)rr __ndims____df__Nfmtdimsnamerct|ttfsJd}|t}nt|tr|}nt|t rT|}|j\}}t|trt|dks|dtd|d}nCtd|D}td|DsJt|}|j }|(t|tr|D] } | d}n|| z}nd}|j pd} t|D]} d| d} || }t|| ||f|_||_||_ dS) Nr rzqinvalid 'dims', when providing an array transform for `dims`, it must have a single, fixed (non-greedy) dimensionc3<K|]}|dnt|VdSr)int.0ds r z"ArrayX.__init__..Es0LL! s1vvLLLLLLrc3*K|]}|dn|dkVdS)NTrr's rr*z"ArrayX.__init__..Fs.IIqytta!eIIIIIIrrzList[]) isinstancer rrrrr tupler TypeErrorall __nbytes____hint__rsuper__init__r) selfr!r"r#num_dims locked_dims dims_dims_fmtnbytesdimhintr __class__s rr5zArrayX.__init__"s #)455555 <%KK d # # !KK f % % !K"kOIty%00 y>>Q&&Q<'J !|HH LLtLLLLLKII[IIIII I I I4yyH #  +u-- &""C{!%cMFF|$ux # #A"4???DD <D t$$$!3& ! rc|jdS)zArray element format.r r r6s rr!z ArrayX.fmte{1~rc|jdS)zArray dimensions.rr@rAs rr"z ArrayX.dimsjrBrbufferoffsetdumpc<|t|||||S|j\}}|t|tr|}d|ddvrt dnrt|t r||||\}}|f}n?t|tr||||\}}ntdg}|j } |^} } | At|} || kr+||||\} }| | || k+nx| r?t| D].}t||||| \} }| | /n7t| D]'}||||\} }| | (||fS)Nr 9array unpack does not support greedy secondary dimensionszunexpected implementation error) r__unpack_and_dump__r r.r/r0r __unpack__ RuntimeErrorappendrr)r6rDrErFr"x_dimsr!r<rrLthis_dim item_dimsenditemrs rrJzArrayX.__unpack__os  --dFFD$OO Ok  <&%(( F48###S$ FD)) F$//EE VvFF++ F%00FF ff##DEEE#9  f++C3,,"~~ffdCC ft 3,, 8__  %00vvtYWW ft  8__  "~~ffdCC ft f}rc &|j\}}|t|tr|}d|ddvrtdnt|tr3||||d|\}}|f}nWt|tr0||||d|\}}ntd|g}|j } |^} } | ht|} d} |j }|| krI||||d| d |\}}| || dz } || kIn| rWt| D]F} t ||||d| d  | \}}| |GnW|j }t| D]@} ||||d| d |\}}| |A|sg|_||fS) Nr rHlen()accessr!--dims--invalid 'dims' r[r-rU)r r.r/r0rrJ add_recordrrKrLrr#rrIvalue)r6rDrErFr"rMr!r<rrLrNrOrPifmt_namerQs rrIzArrayX.__unpack_and_dump__sk  <&%(( A48###S$ FD)) A$//FDOO7O$O$O VvFF++ A%00FDOO:6O$R$R   ff ##?V#?#?@@@#9  f++CAxH3,,"~~FDOO8q888O$R$R   ft Q 3,, 8__  %99&&$//Q/*J*JI   ft   xH8__  "~~FDOO8q888O$R$R   ft  DJf}rr[piecesc| t|||||dS|j\}}|t|tr|^}}nt|t r)t |g}}||||nit|tr>t t||j }|^}}||||ntd||^}} t |} n#t$rtddwxYw|| |krtd|r%|D] } t|| |||!dS|D]} || ||dS)NrWz'invalid array value, retrying with dump) r__pack_and_dump__r r.r/rr__pack__rrrKr0) r6r[r^rFr"rMr!rNrO actual_lengthrQs rrazArrayX.__pack__s    $ $T5&$ E E E E E+KFC|fe,, E+1(Hyy-- E*-e**biHOOHfd;;;;//E eT^!D!DEED+/(HyOOD&$7777''C'C'CDDD(,$9 U #E  U U U IJJPTT U# (A(A IJJJ 5!IIDOOD$iHHHHII"55DLLvt444455s 1DDc N|j\}}|t|tr|^}}nt|tr>t |g}}||||d|n~t|trStt||j }|^}}||||d|ntd||^}} t |} nD#t$r7||_ td|dndd|dnd t|zd dwxYw| sg|_ || }|rZtt||D];\} } t|| ||d | d  |)rUr[z ] )rUr[separatez length, got iterable of length )r r.r/rrrarZrrrrKr0r[strziprr`r#) r6r[r^rFr"rMr!rNrOrbr\rQr]s rr`zArrayX.__pack_and_dump__sk  <&%(( A'-$99FD)) A&)%jj")fdooW&o&Q&QFF++ A\%@@AA'+$9&$///"P"P ##?V#?#?@@@$( Hy JJMM   DJ&%-662&&'/S3x==5H&&&     DJ  $H  uX66  4(($x1xxx(H(H)  xHuX66  4 &$//Qx/"P"P H $ $=(33 D Dx1xxx{CCCCuX}==uXYY?OPP  4+q+++41= MMM=JMM  % $s 9D AE )NNr)__name__ __module__ __qualname____doc__r&__annotations__r DimsArgr rrhr5propertyr!r"bytesrDimsrrrJrIrrar` __classcell__)r>s@rrrs66NNN '6/ """" #'" A!A! A!wA!sm A!  A!A!A!A!A!A!FVXgX"&# 4444v 4 tn 4 tCy#~  4444v $ AAAA A tn A tCy#~  AAAAN"&# .5.5}.5U .5v .5 tn .5  .5.5.5.5j $ JJ}JU J J tn J  JJJJJJJJr)rmtypingrrrrrr r _typingr datar rFrr&r transformrrrrorrnrrr,rrrxsE322IIIIIIIIIIIIIIIIII   h$ % T ciT46PGGGGGYGGGGGr