o װiq@sddlZiddgddgddgddgddgddgd gd d dgd dgd d gdgddgddgdddgdgdddgdd ggddgdgdgdggddggdgdd gd!ggd"d#ggd$d%Zd&Zeed'Zd)d(d'ZdS)*N)rrr )r r r )r r r )rr r)rrr)rrr)rr))rrr)rr)r )rrrrrrrrrrrr r a #include #include #include extern "C" __global__ void max_len_seq( long long length, long long n_taps, int n_state, long long* taps, signed char* state, signed char* seq) { int idx = blockDim.x * blockIdx.x + threadIdx.x; for(long long i = 0; i < length; i++) { signed char next_state = state[(idx + 1) % n_state]; if(idx == n_state - 1) { seq[i] = state[0]; for(int n_tap = 0; n_tap < n_taps; n_tap++) { long long tap = taps[n_tap]; next_state ^= state[tap]; } } state[idx] = next_state; } } max_len_seqc Csx|dur'|tvrttt}td||ftt|tj}n+t t|tjddd}t |dksIt ||ksI|j dkrMtdt|}d|d}|dur_|}n t |}|dkrktd|durytj |tjd d }n tj|td d tj}|jdks|j |krtd t|dkrtd tj|tjd d }t|}td |f||||||f||fS)a Maximum length sequence (MLS) generator. Parameters ---------- nbits : int Number of bits to use. Length of the resulting sequence will be ``(2**nbits) - 1``. Note that generating long sequences (e.g., greater than ``nbits == 16``) can take a long time. state : array_like, optional If array, must be of length ``nbits``, and will be cast to binary (bool) representation. If None, a seed of ones will be used, producing a repeatable representation. If ``state`` is all zeros, an error is raised as this is invalid. Default: None. length : int, optional Number of samples to compute. If None, the entire length ``(2**nbits) - 1`` is computed. taps : array_like, optional Polynomial taps to use (e.g., ``[7, 6, 1]`` for an 8-bit sequence). If None, taps will be automatically selected (for up to ``nbits == 32``). Returns ------- seq : array Resulting MLS sequence of 0's and 1's. state : array The final state of the shift register. Notes ----- The algorithm for MLS generation is generically described in: https://en.wikipedia.org/wiki/Maximum_length_sequence The default values for taps are specifically taken from the first option listed for each value of ``nbits`` in: https://web.archive.org/web/20181001062252/http://www.newwaveinstruments.com/resources/articles/m_sequence_linear_feedback_shift_register_lfsr.htm Nz/nbits must be between %s and %s if taps is NonerrzEtaps must be non-empty with values between zero and nbits (inclusive)rz)length must be greater than or equal to 0c)dtypeorderz'state must be a 1-D array of size nbitszstate must not be all zeros)r) _mls_tapscupyarraylistkeys ValueErrorminmaxint64uniqueanysizeintonesint8boolastypendimallemptylen _max_len_seq)nbitsstatelengthtaps known_tapsn_maxseqn_tapsrES/home/ubuntu/.local/lib/python3.10/site-packages/cupyx/scipy/signal/_max_len_seq.pyr"(s8*&  )NNN)r(r'MAX_LEN_SEQ_KERNEL RawKernelr<r"rErErErFs<<