grdZddlZddlmZddlmZddlmZej e Z GddeZ dS)zTRemove any swap gates in the circuit by pushing it through into a qubit permutation.N)TransformationPass)Layout)elide_permutationsceZdZdZdZdS)ElidePermutationsa[Remove permutation operations from a pre-layout circuit This pass is intended to be run before a layout (mapping virtual qubits to physical qubits) is set during the transpilation pipeline. This pass iterates over the :class:`~.DAGCircuit` and when a :class:`~.SwapGate` or :class:`~.PermutationGate` are encountered it permutes the virtual qubits in the circuit and removes the swap gate. This will effectively remove any :class:`~SwapGate`\s or :class:`~PermutationGate` in the circuit prior to running layout. If this pass is run after a layout has been set it will become a no-op (and log a warning) as this optimization is not sound after physical qubits are selected and there are connectivity constraints to adhere to. For tracking purposes this pass sets 3 values in the property set if there are any :class:`~.SwapGate` or :class:`~.PermutationGate` objects in the circuit and the pass isn't a no-op. * ``original_layout``: The trivial :class:`~.Layout` for the input to this pass being run * ``original_qubit_indices``: The mapping of qubit objects to positional indices for the state of the circuit as input to this pass. * ``virtual_permutation_layout``: A :class:`~.Layout` object mapping input qubits to the output state after eliding permutations. These three properties are needed for the transpiler to track the permutations in the out :attr:`.QuantumCircuit.layout` attribute. The elision of permutations is equivalent to a ``final_layout`` set by routing and all three of these attributes are needed in the case c2|jdtdStj}|S|\}}dt jD}t||jd<|jd ||jd<tfdt |D}|jdx}rB|| jjj|jd<n ||jd<|S) zRun the ElidePermutations pass on ``dag``. Args: dag (DAGCircuit): the DAG to be optimized. Returns: DAGCircuit: the optimized DAG. layoutNzpElidePermutations is not valid after a layout has been set. This indicates an invalid pass manager construction.ci|]\}}|| Sr ).0indexqubits n/granary/edev/quantum/lib/python3.11/site-packages/qiskit/transpiler/passes/optimization/elide_permutations.py z)ElidePermutations.run..OsVVV uueVVVoriginal_layoutoriginal_qubit_indicesc2i|]\}}j||Sr )qubits)r idxoutdags rrz)ElidePermutations.run..Ts%[[[hc3SZ_c[[[rvirtual_permutation_layout) property_setloggerwarningelide_permutations_rsrun enumeraterrcomposeinverse)selfrresultnew_dag qubit_mappinginput_qubit_mapping new_layoutcurrent_layouts ` rrzElidePermutations.run5sB  X & 2 NN8   J&*3// >J$* -VV #*@U@UVVV/56I/J/J+,  5 6 >:MD 6 7[[[[)MBZBZ[[[\\ !./KL L> I>H>P>P&&sz3:>> ??D : ; ;?ID : ;rN)__name__ __module__ __qualname____doc__rr rrrrs-6&&&&&rr) r,loggingqiskit.transpiler.basepassesrqiskit.transpiler.layoutrqiskit._acceleraterr getLoggerr)rrr rrr2s[Z;;;;;;++++++JJJJJJ  8 $ $BBBBB*BBBBBr