g/ZdZddlZddlZddlmZddlmZddlmZd dZ dZ dZ d d Z dS) z6 This module contains utility functions for circuits. N) _numpy_compat) QiskitError) CircuitErrorc|ttj|jd}d|z}t jdgdggd|dz g}||dz }n%t |trt|d}t |trd|cxkr|ksntdntdt j t j ||}t j t j d|zt j ||z t j ||z}|S)aP Compute the controlled version of the input matrix with qiskit ordering. This function computes the controlled unitary with :math:`n` control qubits and :math:`m` target qubits, .. math:: V_n^j(U_{2^m}) = (U_{2^m} \otimes |j\rangle\!\langle j|) + (I_{2^m} \otimes (I_{2^n} - |j\rangle\!\langle j|)). where :math:`|j\rangle \in \mathcal{H}^{2^n}` is the control state. Args: base_mat (ndarray): unitary to be controlled num_ctrl_qubits (int): number of controls for new unitary ctrl_state (int or str or None): The control state in decimal or as a bitstring (e.g. '111'). If None, use 2**num_ctrl_qubits-1. Returns: ndarray: controlled version of base matrix. Raises: QiskitError: unrecognized mode or invalid ctrl_state rNz&Invalid control state value specified.z%Invalid control state type specified.) intmathlog2shapenumpyrepeat isinstancestrrdiagrollkroneye)base_matnum_ctrl_qubits ctrl_state num_targetctrl_dim ctrl_grnd ctrl_projfull_mats K/granary/edev/quantum/lib/python3.11/site-packages/qiskit/circuit/_utils.py_compute_control_matrixrsC2TYx~a01122J/!H qcA3Z!X\):;;I\ J $ $(Q'' *c""CJ))))))))FGG G*ABBB 5:i<<==Iz%)AzM22EIh4G4G)4STTW\Wa)XXH Ocd}t|tri t||ksJt|d}nB#t$r}t d|z|d}~wt $r}t d|d}~wwxYwt|tr%d|cxkr d|zkrnn|}n9t d| d|zdz }nt dt||S) aKConvert ctrl_state to int. Args: ctrl_state (None, str, int): ctrl_state. If None, set to 2**num_ctrl_qubits-1. If str, convert to int. If int, pass. num_ctrl_qubits (int): The number of control qubits. Return: int: ctrl_state Raises: CircuitError: invalid ctrl_state Nrzinvalid control bit string: z5invalid control bit string: length != num_ctrl_qubitsrz#invalid control state specificationrz%invalid control state specification: )rrlenr ValueErrorrAssertionErrorrepr)rrctrl_state_stdexs r_ctrl_state_to_intr'Ds;N*c""` `z??o5555Z++JJ T T T= JKKQS S ` ` `VWW]_ _ `*c""W / / / /Q/ / / / / /'NNDEE E  O+a/U4 CSCSUUVVV s!%? A> A A>)A99A>ctj|tjddtjffd fd}|S)zClass decorator that adds an ``__array__`` method to a :class:`.Gate` instance that returns a singleton nonwritable view onto the complex matrix described by ``base_array``.dtypeFwriteNcH|jn|}tj||SNr*copy)r*r array)_selfr*r0 nonwritables r __array__z"with_gate_array..__array__ms+%*] !!{;e$????rcTt|drtd|_|SNr4zBRefusing to decorate a class that already has '__array__' defined.hasattr RuntimeErrorr4clsr4s r decoratorz"with_gate_array..decoratorq1 3 $ $ ecdd d!  r)r r1 complex128setflagsrCOPY_ONLY_IF_NEEDED) base_arrayr<r4r3s @@rwith_gate_arrayrBgs+j0@AAAKu%%%#-*K@@@@@@ rctj|tjfd|1fdtdzDdtjffd n"fd|Ddtjffd fd }|S) aClass decorator that adds an ``__array__`` method to a :class:`.ControlledGate` instance that returns singleton nonwritable views onto a relevant precomputed complex matrix for the given control state. If ``cached_states`` is not given, then all possible control states are precomputed. If it is given, it should be an iterable of integers, and only these control states will be cached.r)ctjt|tj}|d|S)Nr)Fr+)r asarrayrr>r?)stateoutbasers rmatrix_for_control_statez.matrix_for_control_statesIm #D/5 A A"    5 !!! rNc&g|] }|SrK.0rFrIs r z.with_controlled_gate_array..s%___E0077___rrcb|j}||jn|}tj|||Sr.)rr*r r1)selfr*r0arr nonwritabless rr4z-with_controlled_gate_array..__array__s5t/C!&CIIEE;s%d;;; ;rc(i|]}||SrKrKrLs r z.with_controlled_gate_array..s'ZZZ577>>ZZZrc|jx}"||jn|}tj|||S|dur|t jurtdtjt|j|S)Nr/Fz,could not produce matrix without calculationr)) getrr*r r1rr@r"rEr)rPr*r0rQrHrRrs rr4z-with_controlled_gate_array..__array__s#''888E%*] {3e$????u}}]-N!N!N !OPPP='otOOW\ rcTt|drtd|_|Sr6r7r:s rr<z-with_controlled_gate_array..decoratorr=r)r rEr>rangerr@)rAr cached_statesr<r4rHrIrRs ` @@@@rwith_controlled_gate_arrayrZzs =5+; < < r_s ))))))222222))))X   F&------r