Vpf dZddlmZddlmZmZddlmZddlZddl m Z e e e e e e e e fe e e e e e ffZ eee ZedGd d Zd Zd Zd ZdZeeededd zeeededd zzeeededd zzededgzZdZeZd,dZd-dZd.d!Zd/d#Zd0d'Zd1d)ZGd*d+Z dS)2zQ An implementation of sourcemaps following `TC39 `_. ) annotations)IterableSequence) dataclassN)UnionT)frozencbeZdZUded<ded<ded<ded<ded<edd Zdd ZdS) SourceMapintversionz Sequence[str]sourcessources_contentnamesMappingsmappings json_datastrreturnc tj|}||d|d|d|dt|dS)z#Deserialize a source map from JSON.r r sourcesContentrr)r r rrr)jsonloadsdeserialize_mappings)clsrdatas R/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/sourcemap.py from_jsonzSourceMap.from_json/sZ :i D 3YY-.7m%d:&677    c|j|j|j|jt |jd}t j|S)zSerialize a source map to JSON.)r r rrr)r r rrserialize_mappingsrrdumps)selfrs rto_jsonzSourceMap.to_json;sD<<.&t}55   D :d  rN)rrrr )rr)__name__ __module__ __qualname____annotations__ classmethodrr#rrr r %s,,,       ;       rr  AZaz09+/cvdttDfdtdDS)Nci|]\}}|| Sr)r)).0dcs r z)make_vlq_decode_table..Ts 5 5 5TQAq 5 5 5rc<g|]}|dSN)get)r8ilookups r z)make_vlq_decode_table..Us' 2 2 2!&**Q   2 2 2r) enumerate VLQ_ALPHABETrange)r@s@rmake_vlq_decode_tablerFSs> 5 5Y|44 5 5 5& 2 2 2 2uSzz 2 2 22renc Iterable[int]rr c>t|}tt|}t|tz}|t zdz }d}|t zr>|tz }tt|}||t z|zz}|t z>|r| n|S)z%Decode a Base-64-VLQ into an integer.r*)iterVLQ_DECODE_TABLEnextbool VLQ_SIGN_MASKVLQ_VALUE_MASK VLQ_MORE_MASKVLQ_VALUE_BITWIDTH)rGenc_iterr9signvalueshifts r decode_vlqrW[s #YY(tH~~&! a- $ ~ ! #% % M+ Eh(A a. U **E M+  "%U"rrUbytesc|dkr | dzdzn|dz}g} |tz}|tz}|dk}|r |tz}|t||snJt |S)z%Encode an integer into a Base-64-VLQ.rr*)rPrRrQappendrDrX)rUbufr9mores r encode_vlqr]ls"'UFq=1   % #  A   E 19D =aJJ|A     srSegmentct|}t|} t|}n#t$r|fcYSwxYwt|}t|} t|}n#t$r ||||fcYSwxYw|||||fS)z)Decode a sequence of VLQs into a segment.)rKrW StopIteration)rGrScolsource source_line source_colnames rdecode_segmentrf~s #YY(8#  ! !FF  6MMM8$$+(##*0 h  DD 000  Z ////0 fk:t 33s 0 AA"A22BBsegc@dd|DS)z)Encode a segment into a sequence of VLQs.rc34K|]}t|VdSr=)r])r8rUs r z!encode_segment..s*55*U##555555rjoin)rgs rencode_segmentrms# 55555 5 55r mappings_strrrcbt|d}d|dDS)z%Decode a string of TC39 mapping data.ascii)encodingc g|];}|r5ttt|dng.sQ    9@Gd3~w}}T22 3 3444R   r;)rXrv)rnmappings_bytess rrrsC888.  #))$//   rrcjdd|D}|dS)z3Encode mappings into a string of TC39 mapping data.rxc3TK|]#}dd|DV$dS)rsc34K|]}t|VdSr=)rm)r8rgs rrjz/serialize_mappings...s*44s##444444rNrk)r8segss rrjz%serialize_mappings..sN9=dii44t44444rrp)rldecode)rrGs rr r sE AI  # G  rc,eZdZdZdZdZdZd dZdS) MappingsGeneratora"MappingsGenerator is a builder API for mappings. TC39 mapping data is inconvenient to emit directly: in an effort to compress data it encodes most indices using values _relative_ to the previous element. MappingsGenerator simplifies things by taking absolute indices everywhere. chd|_d|_d|_d|_d|_g|_d|_dS)Nr) _last_col _last_source_last_source_line_last_source_col _last_name _mappings _cur_groupr"s r__init__zMappingsGenerator.__init__s:DNDDDDODNDOOOrc`d|_g|_|j|jdS)zStart a new group (line).rN)rrrrZrs r new_groupzMappingsGenerator.new_groups.DNDON$/*****rct|dksJ|j}|td|d|jz }|d|_t|dkr||fdS|d|jz }|d|_|d|jz }|d|_|d|jz }|d|_t|dkr|||||fdS|d|jz }|d|_t|dkr||||||fdSJd ) a Start a new source mapping segment in the current group. Args: *seg: A segment as in TC39, but all indices are absolute. See https://tc39.es/source-map/#mappings-structure for details. Raises: RuntimeError: If no current group exists. r*Nz-No current group. Forgot to call new_group()?rr-Fzinvalid segment) lenr RuntimeErrorrrZrrrr)r"rggrouprarbrcrdres r new_segmentzMappingsGenerator.new_segmentsV s88q==== OE } H I II a&4> !CVDN 3xx1}} llC6 f Vd' 'FADa&411K VDQ$//JFD 3xx1}} llCj9::: f q6DO #D!fDO 3xx1}} llCj$?@@@ f#####rrrc|jS)z2Return the mapping as a list of segments per line.)rrs rrzMappingsGenerator.mappingss >rN)rr)r$r%r&__doc__rrrrr)rrrrs`+++ ($($($Trr)rGrHrr )rUr rrX)rGrHrr^)rgr^rrX)rnrrr)rrrr)!r __future__rcollections.abcrr dataclassesrrtypingrtupler r^rr rOrQrPrRrtrEordrDrFrLrWr]rfrmrr rr)rrrs#"""""........!!!!!!   #Jc3S()5c3S1H+II  HW% & $B  Dss3xxSA & &'' d55S33s88a< ( ())* d55S33s88a< ( ())* s3xxS 333 )(**####"$4444(6666 DDDDDDDDDDr