o Pj)@slUdZddlZddlmZddlmZmZmZmZm Z ddddd Z ee e fe d <Gd d d e Z d-dede de de ee e ffddZdede de de de de ee e ff ddZdee defddZde de defddZde de e e ffddZde de de fd d!Zd"ee de de defd#d$Zd%e d&e de fd'd(ZGd)d*d*ZGd+d,d,ZdS).zSparse Bit Set encoding/decoding for IFT (Incremental Font Transfer). Implements the sparse bit set format defined in the W3C IFT specification: https://w3c.github.io/IFT/Overview.html#sparse-bit-set-decoding N)deque)DictIterableOptionalSetTuple  _BF_MAX_HEIGHTc@s eZdZdS)SparseBitSetDecodeErrorN)__name__ __module__ __qualname__rrS/home/thesage/.local/lib/python3.10/site-packages/fontTools/misc/iftSparseBitSet.pyrsrdatabiasmaxValuereturncCsV|stdt|d\}}t|}||kr#td|d|d|t|||||S)aMDecode a sparse bit set from binary data. Args: data: bytes-like object containing the sparse bit set encoding. bias: integer added to each decoded value. Returns: A tuple (values, bytesConsumed) where values is a set of integers and bytesConsumed is the number of bytes read from data. z Empty datarzHeight z exceeds max z for branch factor )r _decodeHeaderr _decodeImpl)rrr branchFactorheight maxHeightrrrdecodes r"rr cCsf|dkr tdfSt||}t}t}|d|r|\}} |} | dur-td| dkrd|| d} || } ||} | |krDqt||| d|}| dkrUd} | |krc|t | |dq|| } || } t | d}|dkrwn4| |kr|||}||kr| n!|dkr| |n||}|||| df| d|>M} qm|s|| fS)Nr)rr#zUnexpected end of dataTr)set_InputBitStreamrappendpopleftnextrminupdaterange_trailingZerosclearadd bytesConsumed)rrr rr bitStreamresultqueuestartdepthbitsexpnodeSize fillStartfillEnd nextNodeSizebitIndexval startDeltarrrr1sV         (rvaluescCstt|}|s tddS|d}t|}d}dD]"}t||}|t|kr'qt|||}|dus9t|t|kr;|}q|durGtd||S)zEncode a set of integers as a sparse bit set. Tries all branching factors and returns the shortest encoding. Args: values: iterable of non-negative integers. Returns: bytes containing the sparse bit set encoding. r rNr zCannot encode max value )sortedr$ _encodeHeader _treeHeightr _encodeWithBflen ValueError)r> valuesSortedrvalueSetbestrr encodedrrrencodejs"     rJcCs$ddddd}t|d>||BgS)Nrr#r r )bytes)rr branchFactorToIdrrrrAsrA headerBytecCs.|d@}ddddd}|d?d@}|||fS)NrKr rrr)rr#r rKrr)rNididToBranchFactorr rrrrs  rcCs,d}|}||kr||9}|d7}||ks|S)z|dkr t|dSig}|D]"}||}||}||dvr$d|d|<|d|d|>O<qtd|D]0}|d}i} |D]\}} ||} ||}| | vrVd| | <| | d|>O<qB|| q6t|dtdtdtffdd } t|} ||}tddd|dfg}|r|\}}}}|d|}d|krt |krnn|| |dnd} ||dkr| ||||dkr| dq| | | dkr||dkr||d|}| }|rt |d }|||}|||}||d}|||d||f|d|>M}|s|st||| S) Nrr#r?lohircst|t|SN)bisect bisect_right bisect_left)rRrSrFrr rangeCountsz!_encodeWithBf..rangeCountr)rAr+itemsr&r@int_OutputBitStreamrr'rDgetwriter,toBytes)rGrr layersv nodeIndexbitPos_ prevLayernewLayerbitmask parentIndexrYstream subtreeSizer2r4 rangeStartrangeEndlayerIdx childSizer5r; childIndex childStartchildEndrrXrrCs^     .       rCr<maxBitscCs<|dkr|Sd}|d@dkr|dL}|d7}|d@dks|SNrr#r)r<rrcountrrrr,s  r,c@sBeZdZdZdedefddZdeefddZdefd d Z d S) r%zBReads bit nodes from a byte array, starting after the header byte.rrcCs||_||_d|_d|_dS)Nr#r)rr byteIndexsubIndex)selfrrrrr__init__s z_InputBitStream.__init__rcCs,|jdvr:|jt|jkrdSd|j>d}|j|j|j?|@}|j|j7_|jdkr8d|_|jd7_|S|jdkrX|jt|jkrIdS|j|j}|jd7_|S|jdkr|jdt|jkridS|j}|j}||||dd>B||dd>B||dd >B}|jd 7_|SdS) Nr rr#rrrrKr r r)rrurDrrv)rwmaskr<birrrr(s2     8z_InputBitStream.nextcCs|j|jdkr dSdSrs)rurvrwrrrr/sz_InputBitStream.bytesConsumedN) rrr__doc__rLr[rxrr(r/rrrrr%s r%c@s>eZdZdZdefddZdeddfdd Zdefd d ZdS) r\z#Writes bit nodes into a byte array.rcCs||_t|_d|_dS)Nr)r bytearrayrrv)rwrrrrrxs z_OutputBitStream.__init__valuerNcCs|jdvr;d|j>d}||M}|jdkr|jd|jd||j>O<|j|j7_|jdkr9d|_dSdS|jdkrJ|j|d@dS|jdkrw|j|d@|j|d?d@|j|d?d@|j|d ?d@dSdS) Nryr#rr?rrr rz)rrvrr&)rwrr{rrrr^$s$       z_OutputBitStream.writecCs t|jSrT)rLrr~rrrr_6s z_OutputBitStream.toBytes) rrrrr[rxr^rLr_rrrrr\s r\)rr)rrU collectionsrtypingrrrrrrr[__annotations__ ExceptionrrLr"rrJrArrBrCr,r%r\rrrrsJ   9  J (