gdZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z mZdd lmZerdd lmZGd d eZGd dZdS)z0Circuit operation representing a ``while`` loop.) annotations) TYPE_CHECKING)ClbitClassicalRegister)expr) CircuitError)validate_conditioncondition_resources) ControlFlowOpQuantumCircuitceZdZdZ ddfd Zed Zed Zejd Zejd ZedZ dZ dZ xZ S) WhileLoopOpzA circuit operation which repeatedly executes a subcircuit (``body``) until a condition (``condition``) evaluates as False. The classical bits used in ``condition`` must be a subset of those attached to ``body``. N condition=tuple[ClassicalRegister, int] | tuple[Clbit, int] | expr.Exprbodyrlabel str | Nonec|j}|j}td|||g|t ||_dS)a Args: condition: A condition to be checked prior to executing ``body``. Can be specified as either a tuple of a ``ClassicalRegister`` to be tested for equality with a given ``int``, or as a tuple of a ``Clbit`` to be compared to either a ``bool`` or an ``int``. body: The loop body to be repeatedly executed. label: An optional label for identifying the instruction. while_looprN) num_qubits num_clbitssuper__init__r _condition)selfrrrrr __class__s [/granary/edev/quantum/lib/python3.11/site-packages/qiskit/circuit/controlflow/while_loop.pyrzWhileLoopOp.__init__%sK_ _  z:vUSSS,Y77c|jSN_paramsrs r paramszWhileLoopOp.params:s |r!c|jSr#rr&s r rzWhileLoopOp.condition>s r!c||_dSr#r))rvalues r rzWhileLoopOp.conditionBs r!c *ddlm}|\}t||s tdt |d|j|jks|j|jkr0td|jd|jd|jd|jd |g|_dS)Nrr zJWhileLoopOp expects a body parameter of type QuantumCircuit, but received .zAttempted to assign a body parameter with a num_qubits or num_clbits different than that of the WhileLoopOp. WhileLoopOp num_qubits/clbits: /z" Supplied body num_qubits/clbits: )qiskit.circuitr isinstancertyperrr%)r parametersrrs r r'zWhileLoopOp.paramsFs 211111$// >04T >>>  ?do - -DO1S1SY26/YYDHOYY59OYYGKoYYY v r!c|jdfS)Nrr$r&s r blockszWhileLoopOp.blocks]s Q!!r!cB|\}t|j||jS)Nr)rrr)rr4rs r replace_blockszWhileLoopOp.replace_blocksas"4?D CCCCr!c td)NzfWhileLoopOp cannot be classically controlled through Instruction.c_if. Please use an IfElseOp instead.)NotImplementedError)r classicalvals r c_ifzWhileLoopOp.c_ifes! .   r!r#)rrrrrr) __name__ __module__ __qualname____doc__rpropertyr'rsetterr4r6r; __classcell__)rs@r rrs! 8888888*XX    ]],""X"DDD       r!rc0eZdZdZdZdddd Zd Zd ZdS)WhileLoopContextaIA context manager for building up while 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. 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.while_loop` in its context-manager form, i.e. by not supplying a ``body`` or sets of qubits and clbits. Example usage:: from qiskit.circuit import QuantumCircuit, Clbit, Qubit bits = [Qubit(), Qubit(), Clbit()] qc = QuantumCircuit(bits) with qc.while_loop((bits[2], 0)): qc.h(0) qc.cx(0, 1) qc.measure(0, 0) .. warning:: This is an internal interface and no part of it should be relied upon outside of Qiskit Terra. )_circuitr_labelNrcircuitrrrrrcJ||_t||_||_dSr#)rEr rrF)rrGrrs r rzWhileLoopContext.__init__s%  ,Y77 r!czt|j}|j|j|jdS)N)clbits registers)r rrE _push_scoperJcregs)r resourcess r __enter__zWhileLoopContext.__enter__s7'88  !!)9Y_!UUUUUr!cf||jdS|j}|||}|jt |j||j|j|jdS)NFr) rE _pop_scopebuildqubitsrJappendrrrF)rexc_typeexc_valexc_tbscopers r __exit__zWhileLoopContext.__exit__s   M $ $ & & &5 ((**{{5<<>>5<<>>::  T[ A A A K K   ur!)rGrrrrr)r<r=r>r? __slots__rrOrYr!r rDrDlsi:5I!      VVVr!rDN)r? __future__rtypingr qiskit.circuit.classicalregisterrrqiskit.circuit.classicalrqiskit.circuit.exceptionsr_builder_utilsr r control_flowr r/rrrDr[r!r rcs76"""""" EEEEEEEE))))))222222CCCCCCCC''''''.------L L L L L -L L L ^??????????r!