gT6dZddlmZddlmZddlZddlZddlm Z m Z m Z ddl m Z GddZdd ZddZddZddZddZddZddZdS)z1Algebra utilities and the ``GateSequence`` class.) annotations)SequenceN)GateQuantumCircuitQubit) UnitaryGateceZdZdZd"d#dZd$d Zd%dZdZdZd&dZ d'dZ d'dZ d(dZ d)dZ d*dZd*dZed+dZd,d Zd!S)- GateSequencezA class implementing a sequence of gates. This class stores the sequence of gates along with the unitary they implement. gatesSequence[Gate]returnNonect||_d|D|_d|D|_t jd}|jD]}||}t|\}}t|}d|_ d |j|_ ||_ ||_ ||_dS)zxCreate a new sequence of gates. Args: gates: The gates in the sequence. The default is []. cNg|]"}tj|tj#S)dtype)npasarray complex128.0gates c/granary/edev/quantum/lib/python3.11/site-packages/qiskit/synthesis/discrete_basis/gate_sequence.py z)GateSequence.__init__..&s)QQQ4D >>>QQQcg|] }|j Sr namers rrz)GateSequence.__init__..'s333Tty333rN )listr matriceslabelsridentitydot_convert_u2_to_su2_convert_su2_to_so3_eulersjoinr global_phaseproduct product_su2)selfr u2_matrixmatrix su2_matrixr+ so3_matrixs r__init__zGateSequence.__init__s %[[ QQ5QQQ 33U333 KNN m . .F 9--II$6i#@#@ L)44  HHT[)) (! %rindices Sequence[int]ct|dddD]6}|j||j|7d|j|_dS)zQRemove a pair of indices that cancel each other and *do not* change the matrices.Nr!)r"r popr$r*r)r.r4indexs rremove_cancelling_pairz#GateSequence.remove_cancelling_pair<si'$$B$-(( # #E JNN5 ! ! ! KOOE " " " "HHT[)) rother'GateSequence'boolct|jt|jksdSt|j|jD]\}}||krdS|j|jkrdSdS)zCheck if this GateSequence is the same as the other GateSequence. Args: other: The GateSequence that will be compared to ``self``. Returns: True if ``other`` is equivalent to ``self``, false otherwise. FT)lenr zipr+)r.r;gate1gate2s r__eq__zGateSequence.__eq__Esy4:#ek"2"2225 EK88  LE5~~uu   2 2 25trct|jdkrotj|jtjdsCt d|j}t|j}| |dg|St d|j}|jD]}| |dg|S)zConvert to a circuit. If no gates set but the product is not the identity, returns a circuit with a unitary operation to implement the matrix. r)r+) r?r rallcloser,r%rr+_convert_so3_to_su2unitaryappend)r.circuitsu2rs r to_circuitzGateSequence.to_circuit[s tz??a   DL"+a..(Q(Q $QT5FGGGG%dl33C OOC! % % %N 1BCCCJ & &D NN4! % % % %rcddlm}tf}|}||t |jdkrgt j|jt j ds;t|j}| t||d|S|j |_ |jD]}| ||d|S)zConvert to a :class:`.DAGCircuit`. If no gates set but the product is not the identity, returns a circuit with a unitary operation to implement the matrix. r) DAGCircuitrEF)check)qiskit.dagcircuitrOr add_qubitsr?r rrGr,r%rHapply_operation_backrr+)r.rOqregdagrLrs rto_dagzGateSequence.to_dagms 100000zjll t tz??a   DL"+a..(Q(Q %dl33C  $ $[%5%5t5 $ I I IJ,J > >D  $ $T4u $ = = = = rrrcPd|_tj|tj}t |\}}t |}||j|_||j|_|j |z|_ |j |t|j dkr|xjd|jz c_n |j|_|j |j|j ||S)zAppend gate to the sequence of gates. Args: gate: The gate to be appended. Returns: GateSequence with ``gate`` appended. Nrrr!)r)rarrayrr'r(r&r,r-r+r rJr?r$rr#)r.rr0rLphaseso3s rrJzGateSequence.appends $bm444'// U!#&&wwt|,, 774#344 -5 $ t{  a   IITY (III DI 49%%% V$$$ rcxt}dt|jD|_d|jD|_d|j|_t j|jj |_t j|j j |_ |j |_ |S)zGet the complex conjugate.c6g|]}|Sr )inversers rrz(GateSequence.adjoint..s IIIDIIIrcg|] }|j Sr r)rinvs rrz(GateSequence.adjoint..s<<>tz1,411 6$, >> $ 1E4F FrN)r )r r rr)r4r5rr)r;r<rr=)rrrr<)rr<)rri)r9rirr)rrn)r0ryrr<)r;r<rr<)__name__ __module__ __qualname____doc__r3r:rCrMrVrJrcrgrkrmrvrx classmethodrr&r rrr r s> &&&&&:****,$.    D    $ ! ! ! !        [.rr r/ryrtuple[np.ndarray, float]cdtjtj|z }||z}tjtj|tj|}||fS)z8Convert a U(2) matrix to SU(2) by adding a global phase.rF)rsqrtlinalgdetarctan2imagreal)r/zr1rYs rr'r'$sY BGBIMM),, - --AYJ Jrwqzz271:: . .E u rr0tuple[float, float, float]cVtj|d}|dddkr|dddkrtj|dd }tj|ddtj|z |ddtj|z }tj|ddtj|z |ddtj|z }|||fSd}|dddkr@tjdz }|tj|dd|ddz}nCtj dz }| tj|dd |dd z}|||fS)aComputes the Euler angles from the SO(3)-matrix u. Uses the algorithm from Gregory Slabaugh, see `here `_. Args: matrix: The SO(3)-matrix for which the Euler angles need to be computed. Returns: Tuple (phi, theta, psi), where phi is rotation about z-axis, theta rotation about y-axis and psi rotation about x-axis. )decimalsr rrFr7)rroundmathasinatan2cospi)r0thetapsiphis r_compute_euler_angles_from_so3r-skXfr * * *F ay|qVAYq\R//6!9Q<(((j171QV9WXXj171QV9WXXE3 !9Q<1  GaKE 6!9Q<1>>>CCWHqLE$VAYq\MF1IaL=AAACE3ranglesc|\}}}tjtjd|zdgdtjd|zggt}tjt j|dz t j|dz gt j|dz  t j|dz ggt}tjt j|dz t j|dz dzgt j|dz dzt j|dz ggt}tj|tj||S)a Computes SU(2)-matrix from Euler angles. Args: angles: The tuple containing the Euler angles for which the corresponding SU(2)-matrix needs to be computed. Returns: The SU(2)-matrix corresponding to the Euler angles in angles. yry?rr y?)rrXexpcomplexrrsinr&)rrrruz_phiuy_thetaux_psis r_compute_su2_from_euler_anglesrKsJOC Xus{++Q/!RVD3J5G5G1HIQX Y Y YFx (519  tx 22 3tx 7J7J6JDHUZ]^U^L_L_5`aHX (37  TXcAg..3 4txa7H7H27MtxX[^_X_O`O`6abF 6&"&622 3 33rc t||t}tj|dd}tj|dd}tj|dd }tj|dd }tj|dz|dzz |dzz |dzzd|z|zd|z|zzd|z|zd|z|zzgd|z|zd|z|zz|dz|dzz |dzz|dzz d|z|zd|z|zzgd|z|zd|z|zzd|z|zd|z|zz |dz|dzz|dzz |dzz ggt}|S)a'Computes SO(3)-matrix from input SU(2)-matrix. Args: matrix: The SU(2)-matrix for which a corresponding SO(3)-matrix needs to be computed. Returns: The SO(3)-matrix corresponding to ``matrix``. Raises: ValueError: if ``matrix`` is not an SU(2)-matrix. rrFr r) _check_is_su2astyperrrrrXfloat)r0abcdrotations rr(r(bs& ]]7 # #F q ! A q ! A 1  A 1  Ax TAqD[1a4 !Q$ &A AEAI(=rAvzAPQETUI?U V !VaZ!a%!) #QTAqD[1a4%7!Q$%>A APQETUI@U V UQYQ "AEAIA $91a4!Q$;A;MPQSTPT;T U  H OrcXt|tt|S)aConverts an SO(3)-matrix to a corresponding SU(2)-matrix. Args: matrix: SO(3)-matrix to convert. Returns: SU(2)-matrix corresponding to SO(3)-matrix ``matrix``. Raises: ValueError: if ``matrix`` is not an SO(3)-matrix. ) _check_is_so3rrr0s rrHrHs)& )*H*P*P Q QQrrc|jdkrtd|jdttj|dz dkr0tdtj|ddS)zrsK87""""""$$$$$$ 6666666666HHHHHHHHHHHHHHV<4444.< R R R R ^^^^^^^^^^r