gX% :dZddlZddlZddlZgdZgdZejeZ de de fdZ dde d e d e d e dej f d Zde de defdZde ded e dej fdZde d e dej fdZde dej fdZde d e dej fdZde d e dej fdZdS)z\ These are the CNOT structure methods: anything that you need for creating CNOT structures. N)sequspincart cyclic_spin cyclic_line)fulllinestar num_qubitsreturncLtjd|zd|zz dz dz }|S)z Returns lower limit on the number of CNOT units that guarantees exact representation of a unitary operator by quantum gates. Args: num_qubits: number of qubits. Returns: lower limit on the number of CNOT units. g@)mathceil)r num_cnotss b/granary/edev/quantum/lib/python3.11/site-packages/qiskit/synthesis/unitary/aqc/cnot_structures.py _lower_limitrs0 1j=1z>9A=DEEI rrnetwork_layoutconnectivity_typedepthc|dkrtd|dkr-t|}td|||}|dkr#t ||}t |||S|dkrt || S|d krFt| }td t||jd |S|dkr)|dkrtd|dt||S|dkr)|dkrtd|dt||Stdtd|)a Generates a network consisting of building blocks each containing a CNOT gate and possibly some single-qubit ones. This network models a quantum operator in question. Note, each building block has 2 input and outputs corresponding to a pair of qubits. What we actually return here is a chain of indices of qubit pairs shared by every building block in a row. Args: num_qubits: number of qubits. network_layout: type of network geometry, ``{"sequ", "spin", "cart", "cyclic_spin", "cyclic_line"}``. connectivity_type: type of inter-qubit connectivity, ``{"full", "line", "star"}``. depth: depth of the CNOT-network, i.e. the number of layers, where each layer consists of a single CNOT-block; default value will be selected, if ``L <= 0``. Returns: A matrix of size ``(2, N)`` matrix that defines layers in cnot-network, where ``N`` is either equal ``L``, or defined by a concrete type of the network. Raises: ValueError: if unsupported type of CNOT-network layout or number of qubits or combination of parameters are passed. z.Number of qubits must be greater or equal to 2rzPNumber of CNOT units chosen as the lower limit: %d, got a non-positive value: %dr)r connectivity)r linksrr)r rr)r z)Optimal lower bound: %d; Cartan CNOTs: %drrr'z$' layout expects 'full' connectivityrr z$' layout expects 'line' connectivityz4Unknown type of CNOT-network layout, expects one of , got ) ValueErrorrloggerdebug_get_connectivity_sequential_network _spin_network_cartan_networkshape_cyclic_spin_network_cyclic_line_network_NETWORK_LAYOUTS)r rrr new_depthrcnotss rmake_cnot_networkr-*s8A~~IJJJ zz ,,  ^     !ZFWXXX"jUSSSS 6 ! ! %@@@@ 6 ! !:666 7j9Q9QSXS^_`Sa    = ( (  & &UUUUVV V#J666 = ( (  & &UUUUVV V#J666 $CS $ $! $ $   rrcdkrddgi}n|dkrfdtD}n|dkr5dtddz D}ddg|d<dz dz g|dz <nZ|dkr:d tdD}tt|d<ntd td ||S) aY Generates connectivity structure between qubits. Args: num_qubits: number of qubits. connectivity: type of connectivity structure, ``{"full", "line", "star"}``. Returns: dictionary of allowed links between qubits. Raises: ValueError: if unsupported type of CNOT-network layout is passed. rrrcJi|]}|tt S)listrange).0ir s r z%_get_connectivity..s+GGGDz**++GGGrr c&i|]}||dz ||dzgS)rr0r3r4s rr5z%_get_connectivity..s)HHH!QUAq1u%HHHrrr ci|]}|d|gS)rr0r7s rr5z%_get_connectivity..s999qQF999rz*Unknown connectivity type, expects one of r)r2r1r _CONNECTIVITY_TYPES)r rrs` rr#r#rsQQC   GGGGU:5F5FGGG   HHuQ Q/G/GHHHq6a",aa @j1n   99E!Z$8$8999j))**a b9L b bT` b b    Lrrcd}tjd|ft} td|dz D]A}t|dz|D]+}|||vr||d|f<||d|f<|dz }||kr|ccS,BV)a) Generates a sequential network. Args: num_qubits: number of qubits. links: dictionary of connectivity links. depth: depth of the network (number of layers of building blocks). Returns: A matrix of ``(2, N)`` that defines layers in qubit network. rrdtypeTrnpzerosintr2)r rrlayerr,r4js rr$r$s E HaZs + + +E%q*q.)) % %A1q5*-- % %a==&'E!U(O&'E!U(OQJE~~$  %%rc d}tjd|ft} td|dz dD]"}||d|f<|dz|d|f<|dz }||kr|cS#td|dz dD]"}||d|f<|dz|d|f<|dz }||kr|cS#o)z Generates a spin-like network. Args: num_qubits: number of qubits. depth: depth of the network (number of layers of building blocks). Returns: A matrix of size ``2 x L`` that defines layers in qubit network. rrr;Trr=)r rrAr,r4s rr%r%s E HaZs + + +E q*q.!,,  AE!U(O!eE!U(O QJE~~ q*q.!,,  AE!U(O!eE!U(O QJE~~  rc |}|dkrtjgdgdg}tj|dz |dz |dz |dz g|dz |dz |dz |dz gg}t|dz D]d}tjtjtj||fd|f}|dxxdzcc<tj|d}en4|dkrtjgdgdg}nt d |d |S) a? Cartan decomposition in a recursive way, starting from n = 3. Args: num_qubits: number of qubits. Returns: 2xN matrix that defines layers in qubit network, where N is the depth of Cartan decomposition. Raises: ValueError: if number of qubits is less than 3. r)rrr)rrrrr)r)rrrrrrrrrrrrrrrrrrrrrr)rrrrrrrrrrrrrrrrrrrrrrz'The number of qubits must be >= 3, got .)r>asarrayr2hstacktiler )r nr,mult_s rr&r&s@ A1uu IIIyyy122zAE1q5!a%Q7!a%QAqSTu9UVWWq1u $ $AIrwry%'?'?CCUKLLE KKK1 KKK74##DD $ a RRRRRR   G1GGGHHH Lrc~tjd|ft}d} td|dD].}|dz|dz kr||d|f<|dz|d|f<|dz }||kr|cS/td|dD]K}|dz|dz kr||d|f<|dz|d|f<|dz }n||dz kr||d|f<d|d|f<|dz }||kr|cSL)a4 Same as in the spin-like network, but the first and the last qubits are also connected. Args: num_qubits: number of qubits. depth: depth of the network (number of layers of building blocks). Returns: A matrix of size ``2 x L`` that defines layers in qubit network. rr;rTrr=)r rr,zr4s rr(r(s9 HaZs + + +E Aq*a((  A1u Q&&ad !ead QEzz q*a((  A1u Q&&ad !ead Qj1n$$ad ad QEzz %rctjd|ft}t|D]}|dz|z|d|f<|dz|z|d|f<|S)a Generates a line based CNOT structure. Args: num_qubits: number of qubits. depth: depth of the network (number of layers of building blocks). Returns: A matrix of size ``2 x L`` that defines layers in qubit network. rr;rrr=)r rr,r4s rr)r)sd HaZs + + +E 5\\++1u *ad 1u *ad Lr)rrr)__doc__loggingrnumpyr>r*r9 getLogger__name__r!r@rstrndarrayr-dictr#r$r%r&r(r)r0rrrXs III...  8 $ $ S S    "!# E E E E E   E Z E E E E P*#*S*T****Z%C%%S%RZ%%%%2c#"*:      F!S!!!!!!HSr