g\!dZddlmZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZerdd lmZGd d eZGd d ZdS)z.Circuit operation representing a ``for`` loop.) annotationsN)IterableOptionalUnion TYPE_CHECKING) Parameter) CircuitError) ControlFlowOpQuantumCircuitczeZdZdZ ddfd Zed Zejd ZedZdZ xZ S) ForLoopOpzA circuit operation which repeatedly executes a subcircuit (``body``) parameterized by a parameter ``loop_parameter`` through the set of integer values provided in ``indexset``. Nindexset Iterable[int]loop_parameterUnion[Parameter, None]bodyr label Optional[str]cv|j}|j}td|||||g|dS)aa Args: indexset: A collection of integers to loop over. loop_parameter: The placeholder parameterizing ``body`` to which the values from ``indexset`` will be assigned. body: The loop body to be repeatedly executed. label: An optional label for identifying the instruction. for_looprN) num_qubits num_clbitssuper__init__)selfrrrrrr __class__s Y/granary/edev/quantum/lib/python3.11/site-packages/qiskit/circuit/controlflow/for_loop.pyrzForLoopOp.__init__"sQ_ _    J>40PX]      c|jSN_paramsrs r paramszForLoopOp.params7s |r!c ddlm}|\}}}t|tt dfs t dt |dt||s t dt |d|j|jks|j|jkr0t d|jd|jd|jd|jd |@||jvr7|j d |jDvrtj d |j d d t|tr|nt|}|||g|_dS)Nrr zbForLoopOp expects a loop_parameter parameter to be either of type Parameter or None, but received .zNForLoopOp expects a body parameter to be of type QuantumCircuit, but received zAttempted to assign a body parameter with a num_qubits or num_clbits different than that of the ForLoopOp. ForLoopOp num_qubits/clbits: /z" Supplied body num_qubits/clbits: c3$K|] }|jV dSr#)name).0ps r z#ForLoopOp.params..Zs$'H'H1'H'H'H'H'H'Hr!zThe Parameter provided as a loop_parameter was not found on the loop body and so no binding of the indexset to loop parameter will occur. A different Parameter of the same name (zs) was found. If you intended to loop over that Parameter, please use that Parameter instance as the loop_parameter.) stacklevel)qiskit.circuitr isinstancertyper rr parametersr,warningswarnrangetupler%)rr5r rrrs r r'zForLoopOp.params;s 211111)3&.$.9d4jj*ABB +''+++  $// >04T >>>  ?do - -DO1S1SY04YYBF/YY59OYYGKoYYY   &do55#'H'H'H'H'HHH M&#'&&&      *(E::O88h .$7 r!c|jdfS)Nr0r$r&s r blockszForLoopOp.blocksls Q!!r!cf|\}t|jd|jd||jS)Nrr r)rr'r)rr;rs r replace_blockszForLoopOp.replace_blocksps.QQTZPPPPr!r#)rrrrrr rr) __name__ __module__ __qualname____doc__rpropertyr'setterr;r= __classcell__)rs@r rrs $        *X ].8.8].8`""X"QQQQQQQr!rc8eZdZdZdZdZ dddddZdZdZdS)ForLoopContexta/A context manager for building up ``for`` loops onto circuits in a natural order, without having to construct the loop body first. Within the block, a lot of the bookkeeping is done for you; you do not need to keep track of which qubits and clbits you are using, for example, and a loop parameter will be allocated for you, if you do not supply one yourself. All normal methods of accessing the qubits on the underlying :obj:`~QuantumCircuit` will work correctly, and resolve into correct accesses within the interior block. You generally should never need to instantiate this object directly. Instead, use :obj:`.QuantumCircuit.for_loop` in its context-manager form, i.e. by not supplying a ``body`` or sets of qubits and clbits. Example usage:: import math from qiskit import QuantumCircuit qc = QuantumCircuit(2, 1) with qc.for_loop(range(5)) as i: qc.rx(i * math.pi/4, 0) qc.cx(0, 1) qc.measure(0, 0) qc.break_loop().c_if(0, True) This context should almost invariably be created by a :meth:`.QuantumCircuit.for_loop` call, and the resulting instance is a "friend" of the calling circuit. The context will manipulate the circuit's defined scopes when it is entered (by pushing a new scope onto the stack) and exited (by popping its scope, building it, and appending the resulting :obj:`.ForLoopOp`). .. warning:: This is an internal interface and no part of it should be relied upon outside of Qiskit Terra. r)_circuit_generate_loop_parameter_loop_parameter _indexset_label_usedNrcircuitr rrrOptional[Parameter]rrc||_|du|_||_t|tr|nt ||_||_d|_dS)NF) rGrHrIr3r8r9rJrKrL)rrMrrrs r rzForLoopContext.__init__sT  (6$(>%-&0%%@%@UeHoo  r!c|jrtdd|_|j|jr9t d|j|_t|xjdz c_|jS)Nz0A for-loop context manager cannot be re-entered.T_loop_i_r ) rLr rG _push_scoperHr_generated_loop_parametersrIr4r&s r __enter__zForLoopContext.__enter__s : SQRR R  !!###  ( 7#,-Y8W-Y-Y#Z#ZD JJ 1 1Q 6 1 1##r!c||jdS|j}|||}|jr|j|jvrd}n|j}|jt|j |||j t|jt|jdS)NFr) rG _pop_scopebuildqubitsclbitsrHrIr5appendrrJrKr9)rexc_typeexc_valexc_tbscoperrs r __exit__zForLoopContext.__exit__s   M $ $ & & &5 ((**{{5<<>>5<<>>::  ( 2T-A-X-X!NN!1N  dnnd$+ N N N $+   $+     ur!r#)rMr rrrrNrr) r>r?r@rArS __slots__rrTr_r!r rFrFusy""L"#I/3  $ "$$$r!rF)rA __future__rr6typingrrrrqiskit.circuit.parameterrqiskit.circuit.exceptionsr control_flowr r2r rrFrar!r rgs54"""""";;;;;;;;;;;;......222222''''''.------VQVQVQVQVQ VQVQVQrddddddddddr!