gdZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z mZddlmZddlmZdd lmZdd lmZGd d eZddZddZddZdS)zATranslate parameterized gates only, and leave others as they are.) annotations) InstructionParameterExpressionQubitClbit)EquivalenceLibrary)circuit_to_dag) DAGCircuit DAGOpNode) QiskitError)TransformationPass)BasisTranslator)Targetc4eZdZdZ ddfd ZddZxZS)TranslateParameterizedGatesu Translate parameterized gates to a supported basis set. Once a parameterized instruction is found that is not in the ``supported_gates`` list, the instruction is decomposed one level and the parameterized sub-blocks are recursively decomposed. The recursion is stopped once all parameterized gates are in ``supported_gates``, or if a gate has no definition and a translation to the basis is attempted (this might happen e.g. for the ``UGate`` if it's not in the specified gate list). Example: The following, multiply nested circuit:: from qiskit.circuit import QuantumCircuit, ParameterVector from qiskit.transpiler.passes import TranslateParameterizedGates x = ParameterVector("x", 4) block1 = QuantumCircuit(1) block1.rx(x[0], 0) sub_block = QuantumCircuit(2) sub_block.cx(0, 1) sub_block.rz(x[2], 0) block2 = QuantumCircuit(2) block2.ry(x[1], 0) block2.append(sub_block.to_gate(), [0, 1]) block3 = QuantumCircuit(3) block3.ccx(0, 1, 2) circuit = QuantumCircuit(3) circuit.append(block1.to_gate(), [1]) circuit.append(block2.to_gate(), [0, 1]) circuit.append(block3.to_gate(), [0, 1, 2]) circuit.cry(x[3], 0, 2) supported_gates = ["rx", "ry", "rz", "cp", "crx", "cry", "crz"] unrolled = TranslateParameterizedGates(supported_gates)(circuit) is decomposed to:: ┌──────────┐ ┌──────────┐┌─────────────┐ q_0: ┤ Ry(x[1]) ├──■──┤ Rz(x[2]) ├┤0 ├─────■────── ├──────────┤┌─┴─┐└──────────┘│ │ │ q_1: ┤ Rx(x[0]) ├┤ X ├────────────┤1 circuit-92 ├─────┼────── └──────────┘└───┘ │ │┌────┴─────┐ q_2: ─────────────────────────────┤2 ├┤ Ry(x[3]) ├ └─────────────┘└──────────┘ Nsupported_gateslist[str] | Noneequivalence_libraryEquivalenceLibrary | Nonetarget Target | NonereturnNonect|ddlm}|}||j}n|t d||_||_t||||_ dS)a Args: supported_gates: A list of suppported basis gates specified as string. If ``None``, a ``target`` must be provided. equivalence_library: The equivalence library to translate the gates. Defaults to the equivalence library of all Qiskit standard gates. target: A :class:`.Target` containing the supported operations. If ``None``, ``supported_gates`` must be set. Note that this argument takes precedence over ``supported_gates``, if both are set. Raises: ValueError: If neither of ``supported_gates`` and ``target`` are passed. Nr)_selz;One of ``supported_gates`` or ``target`` must be specified.)r) super__init__9qiskit.circuit.library.standard_gates.equivalence_libraryroperation_names ValueError_supported_gates_targetr _translator)selfrrrr __class__s l/granary/edev/quantum/lib/python3.11/site-packages/qiskit/transpiler/passes/basis/translate_parameterized.pyrz$TranslateParameterizedGates.__init__Os&   & V V V V V V"&   $4OO  $Z[[ [ / *+>X^___dagr c|D]}t|jrt||j|js|jj}|#|t|}nP |j t|j}n"#t$r}td|d}~wwxYw| |||S)a)Run the transpiler pass. Args: dag: The DAG circuit in which the parameterized gates should be unrolled. Returns: A DAG where the parameterized gates have been unrolled. Raises: QiskitError: If the circuit cannot be unrolled. Nz Failed to translate final block.)op_nodes_is_parameterizedop _is_supportedr!r" definitionrunr r#_instruction_to_dag Exceptionr substitute_node_with_dag)r$r(noder.unrolledexcs r&r/zTranslateParameterizedGates.runusLLNN = =D )) =-d+T\33 ="W/ )#xxz(B(BCCHHW#'#3#7#78KDG8T8T#U#U$WWW)*LMMSVVW,,T8<<< s8,B%% C/B??C)NNN)rrrrrrrr)r(r rr )__name__ __module__ __qualname____doc__rr/ __classcell__)r%s@r&rrst11j-19= $ $`$`$`$`$`$`$`L        r'rr,rrboolc>td|jDS)Nc3pK|]1}t|tot|jdkV2dS)rN) isinstancerlen parameters).0params r& z$_is_parameterized..sSQV 5-..L3u7G3H3H13Lr')anyparams)r,s r&r+r+s3 Z\Zc  r'r3r r list[str]rrc`|||jjS|jj|vS)zCheck whether the node is supported. If the target is provided, check using the target, otherwise the supported_gates are used. )instruction_supportedr,name)r3rrs r&r-r-s1 ++DGL999 7z'_instruction_to_dag..:::EGG:::r'c*g|]}tSrL)rrMs r&rOz'_instruction_to_dag..rPr'F)check)r add_qubitsrange num_qubits num_clbitsapply_operation_backqubitsclbits)r,r(s r&r0r0s ,,CNN::U2=%9%9:::;;;NN::U2=%9%9:::;;;RSZuEEE Jr'N)r,rrr;)r3r rrFrrrr;)r,rrr )r9 __future__rqiskit.circuitrrrr"qiskit.circuit.equivalence_libraryrqiskit.convertersr qiskit.dagcircuitr r qiskit.exceptionsr qiskit.transpiler.basepassesr /qiskit.transpiler.passes.basis.basis_translatorrqiskit.transpiler.targetrrr+r-r0rLr'r&rcsBHG""""""IIIIIIIIIIIIAAAAAA,,,,,,33333333))))));;;;;;KKKKKK++++++zzzzz"4zzzz ++++r'