g. ddlmZddlZddlmZdedefdZdedefdZ dedefd Z dedefd Z d edefd Z d)d ede defdZd)d ede defdZdedefdZd edeefdZdede fdZd edeefdZdedefdZd edefdZdedefdZd edeefdZdedeefdZdedeefdZdedefdZd edefdZd ede fd Zd!ed"ed#ed$ede f d%Zd&Zd'Z d(Z!dS)*)OptionalN)wraphexstrreturnct|dz}tt|dddt|}|S)zBConvert a hexadecimal string to binary string, with zero fillings.N)lenbinintzfill)r num_of_bitsbinstrs C/home/andy/.local/lib/python3.11/site-packages/pyModeS/py_common.pyhex2binrsJf++/K VR ! !!"" % + +C ,<,< = =F Mc"t|dS)z(Convert a hexadecimal string to integer.r r )rs rhex2intrs vr??rrc"t|dS)z#Convert a binary string to integer.r rrs rbin2intrs vq>>rcHdt|dS)z.Convert a binary string to hexadecimal string.z{0:X}r )formatr rs rbin2hexrs >>#fa.. ) ))rmsgczt|dd}tt|dddS)z*Decode Downlink Format value, bits 1 to 5.Nr r)rminr)rdfbins rdfr#s7 CG  E wuQqSz""B ' ''rFencodec>tddtddtddtddg}|r |dddz}t|}t|d }tt t |}t t|d z D]}t d D]}d |z }|||z} | d kr|||d |z z ||<||d zd|d d |z z|d |z zzz ||d z<||dzd|d d |z z|d|z zzz ||dz<||d zd|dd |z z|d |z zzz ||d z<|ddz|dd zz|dz} | S)a\Mode-S Cyclic Redundancy Check. Detect if bit error occurs in the Mode-S message. When encode option is on, the checksum is generated. Args: msg: 28 bytes hexadecimal message string encode: True to encode the date only and return the checksum Returns: int: message checksum, or partity bits (encoder) 11111111r 111110100000010010000000N000000rr )r rrlistmaprranger ) rr$Gmsgbin msgbin_splitmbytesibyteibitmaskbitsresults rcrcr@#s Z  SQ//Z1C1CSUVEWEWXA "#2#h! S\\F??L #g|,, - -Fs6{{Q''!HH  D4u $*519$5QqTQX-!A$$,?@%uqy!%+519$5QqTQX-!A$$,?@%uqy!%+519$5QqTQX-!A$$,?@%uqy!  RjB6":? 3fRj @F Mrctjgd}t|}tjdt|D}|r dgdz|dd<t t|dz D]7}||dkrtj||||z|||||z<8tj|dddd d }t|}|S) z7Mode-S Cyclic Redundancy Check. (Legacy code, 2x slow).)r/r/r/r/r/r/r/r/r/r/r/r/r/rr/rrrrrrr/rrr/c,g|]}t|Sr).0is r zcrc_legacy..Xs666AQ666rrr iN) separatorr/r3)nparrayr rr6 bitwise_xor array2stringr)rr$ generatorngmsgnpbinrEr8reminders r crc_legacyrQPs SSSI YBx6666677H "r3x==2% & &OO A;!    "~hq1r6z.BINNQV_Xcdd^r : : :1R4 @FvH OrxcDttj|S)zMode-S floor function. Defined as the greatest integer value k, such that k <= x For example: floor(3.6) = 3 and floor(-3.6) = -4 )r rIfloor)rRs rrTrTls rx{{  rct|}|dvr |dd}n8|dvr2t|d}t|ddd }d ||z z}nd}|S) zCalculate the ICAO address from an Mode-S message. Applicable only with DF4, DF5, DF20, DF21 messages. Args: msg (String): 28 bytes hexadecimal message string Returns: String: ICAO address in 6 bytes hexadecimal string ) r r,)rrrr T)r$r*Nr z%06X)r#r@r )rDFaddrc0c1s ricaor_vsx CB \1Q3x $ $ $ T " " " RSS2  b! Krr_c|(t|trt|dkrdSt|d}d|cxkrdkrnndSd|cxkrdkrnndSd |cxkrd krnndSd |cxkrd krnndSd |cxkrdkrnndSd|cxkrdkrnndSd|cxkrdkrnndSd|cxkrdkrnndSd|cxkrdkrnndSdS)z=Check whether the ICAO address is assigned (Annex 10, Vol 3).NFr i i'i(i(iPi_i`igihioiiiiiiiiT) isinstancestrr r )r_icaoints ris_icao_assignedres js33 Tau$mmG'$$$$H$$$$$u'$$$$H$$$$$u'$$$$H$$$$$u'$$$$H$$$$$u'$$$$H$$$$$u'$$$$H$$$$$u'$$$$H$$$$$u'$$$$H$$$$$u'$$$$H$$$$$u 4rct|dvrdSt|dd}t|ddS)zType code of ADS-B message Args: msg (string): 28 bytes hexadecimal message string Returns: int: type code number )rWrXNr, rr)r#rr)rtcbins rtypecoderisE #wwht C"I  E 51:  rlatctj|drdStjt|drdS|dks|dkrdSd}dtjtjd|zz z }tjtjdz t|zdz}dtjztjd||z z z }t |}|S) zNL() function in CPR decoding.r;Wr ir/)rIiscloseabscospiarccosrT)rjnzabnlNLs rcprNLrzs z#qr CHHb ! !q rS3YYq B BF25AF# $ $$A rus{SXX%&&!+A RUbiAE ** +B rB Irct|dvrtdt|}|dd}t|S)zCompute identity code (squawk) encoded in DF5 or DF21 message. Args: msg (String): 28 bytes hexadecimal message string Returns: string: squawk code )rrZz(Message must be Downlink Format 5 or 21. )r# RuntimeErrorrsquawk)rmbin idcodebins ridcodersJ #wwgEFFF 3< qA t8Orct|}t|dd}d}|dkrd}n9|dkrd}n0|dkrd }n'|d krd }n|d krd |dz }||fS)zDecode downlink request for DF 4, 5, 20, and 21. Args: msg (str): 14 hexdigits string Returns: int, str: downlink request, description r,rNrzno downlink requestr/zrequest to send Comm-B messagerzComm-B broadcast 1 availablerzComm-B broadcast 2 availabler z#ELM downlink segments available: {}rn)rrr)rr8drrs rrrsS\\F "  B D Qww$ q/ q- q- r4;;BGDD t8Orct|}t|dd}t|dd}|dkrd}|dkrd}|dkrd }|d krd }|||fS) a>Decode utility message for DF 4, 5, 20, and 21. Utility message contains interrogator identifier and reservation type. Args: msg (str): 14 hexdigits string Returns: int, str: interrogator identifier code that triggered the reply, and reservation type made by the interrogator rrWr|rNr/z#Comm-B interrogator identifier coder z#Comm-C interrogator identifier coder-z#Comm-D interrogator identifier coder)rr8iisidsids_texts rumrs~S\\F &B- C &B- C axx axx8 axx8 axx8 X r)F)"typingrnumpyrItextwraprrcrr rrrr#boolr@rQfloatrTr_rerirzrrrrrrrrrrrrrCrrrszCCCC CC *C*C**** (C(C(((( **S*$*3****ZC#8Uschsm634: # (3-     u$& =3 =3 = = = =F#20S0Xc]0000fSXc]*SScc #$$cs34":8r