o `j/ @sddlZgdZgdZgdZeeeeZddZddZd d Z d d Z d dZ ddZ ddZ ddZd,ddZddZddZddZddZd,dd Zd,d!d"Zd,d#d$Zed%krQee ee eD]\ZZeZe eeeeks|JqjeeeD]\ZZed&ZeeeZedeeksJqeD]\ZZeZeeeeeksJqe ee eeeeeeeeZe e efD]ZeeD]\Z Z!ed'"eje d(ee!qqe eefD]#ZeeD]\Z Z!ejZ#ej$d'"e#e d(d)"e#e!d*"e#d+qqeefD]ZeeD]\Z Z!ed'"eje d(ee!qqeeefD]ZeeD]\Z Z!ed'"eje d(ee!q;q5dSdS)-N)')r)))~?)i)s@)si)si))si )si))si)si)si)s)i)i)si)sl)s)sl)sl))sl)sl)sl)sl)sl)sl)sl)sl)sl)s l)s l)s l)s l)s l)s l)<r rrrr)rr r rrrcCs&tD]\}}|||ksJ|qdSN) test_data)enc_funcencodeddecodedr[/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/kafka/benchmarks/varint_speed.py_assert_valid_encDs rcCsNtD]"\}}|t|\}}||ksJ||f|t|ks$J||fqdSr)r bytearraylen)dec_funcrrresposrrr_assert_valid_decIs r#cCs*tD]\}}||t|ksJ|qdSr)rr) size_funcrrrrr_assert_valid_sizePs r%cCsn|d>|d?A}td}tdD]}|d@|dkrdndB||<|d?}|dkr*nqtd|d |dS) aZ Encode an integer to a varint presentation. See https://developers.google.com/protocol-buffers/docs/encoding?csw=1#varints on how those can be produced. Arguments: num (int): Value to encode Returns: bytearray: Encoded presentation of integer with length from 1 to 10 bytes rr rOut of double rangeN)rrange ValueError)numbufirrrencode_varint_1Us  r0cCs t|fSr)bytes)r/rrrint2bytess r2cCsX|d>|d?A}|d@}|dL}d}|r&|td|B7}|d@}|dL}|s|t|S)Nrrr'r)r(r2valuebitsr!rrrencode_varint_2vs r8cCsV|j}|d>|d?A}|d@}|dL}|r%|d|B|d@}|dL}|s|||SNrrr'r)r(append)r6r.r;r7rrrencode_varint_3s r<cCsp|d>|d?A}|dkrt|S|dkr"td|d@Bt|d?S|dkr>td|d@Btd|d?d@Bt|d?S|d krdtd|d@Btd|d?d@Btd|d?d@Bt|d ?S|d krtd|d@Btd|d?d@Btd|d?d@Btd|d ?d@Bt|d ?S|d@}|dL}d }|r|td|B7}|d@}|dL}|s|t|S)Nrrr'?r(r)r3r4r5rrrencode_varint_4sL    rDcCs\|d>|d?A}|d@}|dL}|r&d|B||<|d@}|dL}|d7}|s|||<|dSr9r)r6r.r"r7rrrencode_varint_5s rEcCs|j}|d>|d?A}|dkr||dS|dkr)|d|d@B||d?dS|dkrG|d|d@B|d|d?d@B||d ?d S|d kro|d|d@B|d|d?d@B|d|d ?d@B||d ?d S|dkr|d|d@B|d|d?d@B|d|d ?d@B|d|d ?d@B||d?dS|d@}|dL}d}|r|d|B|d@}|dL}|d7}|s|||S)Nrrr'r=r(r)r>r?r@rArBrCrr:)r6r.r;r7r/rrrencode_varint_6sL     rJcCs6|d>|d?A}d} |d7}|d?}|dkr |Sq )L Number of bytes needed to encode an integer in variable-length format. rrrTr)r)r6r!rrrsize_of_varint_1srLcCs|d>|d?A}|dkrdS|dkrdS|dkrdS|dkr d S|d kr&d S|d kr,d S|dkr2dS|dkr8dS|dkr>dSdS)rKrrr'r=rFr>rGr@rHrBrIllr)ll r&r)r6rrrsize_of_varint_2s(rPcCs||S)zu Read a byte from memoryview as an integer Raises: IndexError: if position is out of bounds r)memviewr"rrr _read_bytesrRc Csd}d}t|}t||dD]/}zt||}Wn ty#tdw|d@dkr7||d@|>O}|d7}q|||>O}ntd|d?|d@ A|dfS) a Decode an integer from a varint presentation. See https://developers.google.com/protocol-buffers/docs/encoding?csw=1#varints on how those can be produced. Arguments: buffer (bytes-like): any object acceptable by ``memoryview`` pos (int): optional position to read from Returns: (int, int): Decoded int value and next read position rr&zEnd of byte streamr(r'r)r*r) memoryviewr+rR IndexErrorr,)bufferr"r6shiftrQr/byterrrdecode_varint_1s      rXcCsbd}d} ||}||d@|>O}|d7}|d@s$|d?|d@ A|fS|d7}|dkr0tdq)Nrrr'r(r)r Out of int64 ranger,rUr"resultrVbrrrdecode_varint_2;sr^cCs||}|d@s|d?|dfS|d@s|d?dA|dfS|dM}|d7}d} ||}||d@|>O}|d7}|d@sH|d?|d@ A|fS|d7}|dkrTtdq)) Nrr(rr'r)r rYrZr[rrrdecode_varint_3Js$r`__main__r&z {}_{}byterz{}({}, buffer)z/from __main__ import {}; buffer = bytearray(10))stmtsetup)r)%pyperfrBENCH_VALUES_ENCBENCH_VALUES_DEClistmaprrr#r%r0r2r8r<rDrErJrLrPrRrXr^r`__name__rrr!writtenRunnerrunner bench_func enumerater/r6formatfnametimeitrrrrs+   " +  !          D