uki 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"dZd#dZd$dZd%dZGddZ y)&zQ An implementation of sourcemaps following `TC39 `_. ) annotations)IterableSequence) dataclassN)UnionT)frozencZeZdZUded<ded<ded<ded<ded<ed d Zd d Zy ) SourceMapintversionz Sequence[str]sourcessources_contentnamesMappingsmappingsc vtj|}||d|d|d|dt|dS)z#Deserialize a source map from JSON.r r sourcesContentrr)r r rrr)jsonloadsdeserialize_mappings)cls json_datadatas M/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/sourcemap.py from_jsonzSourceMap.from_json/sJ ::i D YY-.7m%d:&67  c|j|j|j|jt |j d}t j|S)zSerialize a source map to JSON.)r r rrr)r r rrserialize_mappingsrrdumps)selfrs rto_jsonzSourceMap.to_json;sF<<<<..&t}}5  D ::d rN)rstrreturnr )r#r")__name__ __module__ __qualname____annotations__ classmethodrr!rrr r %s8 ,       rr  AZaz09+/cttDcic]\}}|| }}}tdDcgc]}|j|dc}Scc}}wcc}w)N) enumerate VLQ_ALPHABETrangeget)dclookupis rmake_vlq_decode_tabler@SsJ&|4 5TQAqD 5& 5',Sz 2!&**Q  22 6 2s A Act|}tt|}t|tz}|t zdz }d}|t zr4|tz }tt|}||t z|zz}|t zr4|r| S|S)z%Decode a Base-64-VLQ into an integer.r*)iterVLQ_DECODE_TABLEnextbool VLQ_SIGN_MASKVLQ_VALUE_MASK VLQ_MORE_MASKVLQ_VALUE_BITWIDTH)encenc_iterr<signvalueshifts r decode_vlqrP[s #Y(tH~&! a- $ ~ ! #% % M Eh(A a. U **E M %"U"rc|dkr | dzdzn|dz}g} |tz}|tz}|dkD}|r |tz}|jt||s t |SI)z%Encode an integer into a Base-64-VLQ.rr*)rHrJrIappendr9bytes)rNbufr<mores r encode_vlqrVlsx"'UFq=1  % #  A   E 19D =aJJ|A  s rct|}t|} t|}t|}t|} t|}|||||fS#t$r|fcYSwxYw#t$r ||||fcYSwxYw)z)Decode a sequence of VLQs into a segment.)rCrP StopIteration)rKrLcolsource source_line source_colnames rdecode_segmentr^~s #Y(8#  !F8$+(#*0 h D fk:t 33  6M 0  Z //0s" A A AAA0/A0c2djd|DS)z)Encode a segment into a sequence of VLQs.rc32K|]}t|ywN)rV).0rNs r z!encode_segment..s5*U#5join)segs rencode_segmentrhs 55 55rc t|d}|jdDcgc].}|r(ttt|jdng0c}Scc}w)z%Decode a string of TC39 mapping data.ascii)encoding;,)rSsplitlistmapr^) mappings_strmappings_bytesmappings rrrsX8.$))$/  9@d3~w}}T2 34RG  s3AcTdjd|D}|jdS)z3Encode mappings into a string of TC39 mapping data.rlc3LK|]}djd|Dyw)rmc32K|]}t|ywra)rh)rbrgs rrcz/serialize_mappings...s4s#4rdNre)rbsegss rrcz%serialize_mappings..s$9=dii4t44s"$rj)rfdecode)rrKs rrrs. AI # G rc*eZdZdZdZdZdZddZy)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. cfd|_d|_d|_d|_d|_g|_d|_y)Nr) _last_col _last_source_last_source_line_last_source_col _last_name _mappings _cur_groupr s r__init__zMappingsGenerator.__init__s6DNDDDDODNDOrcjd|_g|_|jj|jy)zStart a new group (line).rN)r|rrrRrs r new_groupzMappingsGenerator.new_groups&DNDONN$//*rcNt|dk\sJ|j}| td|d|jz }|d|_t|dk(r|j |fy|d|j z }|d|_|d|j z }|d|_|d|jz }|d|_t|dk(r|j ||||fy|d|jz }|d|_t|dk(r|j |||||fyJd ) 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-zinvalid segment) lenr RuntimeErrorr|rRr}r~rr)r rggrouprYrZr[r\r]s r new_segmentzMappingsGenerator.new_segments7 s8q== OOE } H II a&4>> !CVDN 3x1} llC6 Vd'' 'FADa&4111K VDQ$///JFD 3x1} llCj9: q6DOO #D!fDO 3x1} llCj$?@ ###5rc|jS)z2Return the mapping as a list of segments per line.)rrs rrzMappingsGenerator.mappingss >>rN)r#r)r$r%r&__doc__rrrrr)rrrzrzs+ ($Trrz)rK Iterable[int]r#r )rNr r#rS)rKrr#Segment)rgrr#rS)rqr"r#r)rrr#r")!r __future__rcollections.abcrr dataclassesrrtypingrtupler rrr rGrIrHrJror:ordr9r@rDrPrVr^rhrrrzr)rrrsK#.!   #Jc3S()5c3S1H+II  HW% & $B  s3xSA &' 5S3s8a< ()* 5S3s8a< ()* 3xS 3 )*#"$4(6 DDr