o Á¿iQã@sZUdZddlZdedefdd„ZeedƒrejneZejegefe d<Gd d „d ƒZ dS)zM NOTE: BETA - this API is currently in development and is subject to change. éNÚ_byteÚreturncCst|ƒ d¡S)NÚ1)ÚbinÚcount)r©rúc/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/test_visibility/coverage_lines.pyÚ_bit_count_compatsr Ú bit_countÚ _bit_countc@sÄeZdZd"defdd„Zdedefdd„Zd d „Zdd„Z d d„Z dd„Zdefdd„Zdefdd„Z deefdd„Zd#dd„Zdefdd„Zedeeddfdd„ƒZededdfdd „ƒZd!S)$Ú CoverageLinesé Úinitial_sizecCst|ƒ|_dS©N)Ú bytearrayÚ_lines)ÚselfrrrrÚ__init__szCoverageLines.__init__ÚotherrcCst|tƒstS|j|jkSr)Ú isinstancerÚNotImplementedr)rrrrrÚ__eq__s zCoverageLines.__eq__cCs| ¡Sr©Ú _num_lines©rrrrÚ__len__szCoverageLines.__len__cCs| ¡dkS)NrrrrrrÚ__bool__szCoverageLines.__bool__cCstƒ}|j ¡|_|Sr)rrÚcopy)rÚnew_instancerrrÚ__copy__szCoverageLines.__copy__cCsd| ¡›dS)NzCoverageLines(num_lines=ú)rrrrrÚ__repr__$szCoverageLines.__repr__cCstdd„|jDƒƒS)Ncss|]}t|ƒVqdSr)r)Ú.0ÚbyterrrÚ (s€z+CoverageLines._num_lines..)Úsumrrrrrr'szCoverageLines._num_linesÚline_numbercCsV|d}||j ¡kr|j t||j ¡dƒ¡d|d?}|j||O<dS)Nééé€)rrÚextendr)rr&Ú lines_byteÚ lines_bitrrrÚadd*s zCoverageLines.addcCsHg}t|jƒD]\}}tdƒD]}|d|?@r | |d|¡qq|S)z-Returns a sorted list of covered line numbersr'r))Ú enumeraterÚrangeÚappend)rÚlinesÚidxrÚ_bitrrrÚto_sorted_list5s€ýzCoverageLines.to_sorted_listcCs^|j ¡|j ¡kr|j t|j ¡|j ¡ƒ¡t|jƒD] \}}|j||O<qdSr)rrr*rr.)rrÚ _byte_idxrrrrÚupdate@s ÿzCoverageLines.updatecCs|jS)zdThis exists as a simple interface in case we ever decide to change the internal lines representation©rrrrrÚto_bytesHszCoverageLines.to_bytesr1cCs|ƒ}|D]}| |¡q|Sr)r-)Úclsr1ÚcoverageÚlinerrrÚ from_listLszCoverageLines.from_listcCs|ƒ}||_|Srr7)r9r1r:rrrÚfrom_bytearraySszCoverageLines.from_bytearrayN)r )rr)Ú__name__Ú __module__Ú__qualname__ÚintrÚobjectÚboolrrrrr!rr-Úlistr4r6Úbytesr8Úclassmethodr<rr=rrrrrs r)Ú__doc__ÚtypingÚtrAr Úhasattrr rÚCallableÚ__annotations__rrrrrÚs (