gZETdZddlmZdZddlZddlmZmZmZm Z m Z m Z m Z m Z ddlmZmZddlmZmZddlmZd d lmZmZmZd d lmZd d lmZmZmZe rdd l m!Z!GddZ"e"Z#GddeZ$GddeZ%GddZ&GddZ'dS)zselfs \/granary/edev/quantum/lib/python3.11/site-packages/qiskit/circuit/controlflow/switch_case.py__repr__z_DefaultCaseType.__repr__+sN)__name__ __module__ __qualname____doc__r$r r%r#rr"s/CC      r%rcfeZdZdZdddfd Zfd ZddZdZedZ ddZ dZ xZ S)rzA circuit operation that executes one particular circuit block based on matching a given ``target`` against an ordered list of ``values``. The special value :data:`.CASE_DEFAULT` can be used to represent a default condition. Nlabeltarget%Clbit | ClassicalRegister | expr.Exprcases$Iterable[Tuple[Any, QuantumCircuit]]r, Optional[str]c 8ddlm}t|tjr=|jjtjtj fvrtd|jdn+t|ttfstdt|tjr'|jjtj urdn |jj }n&t|trdnt|}d|zdz }t}d\}} ||_i|_ g|_ g|_ t)|D]\} \} } t| t*t,frt+| n| f} | D]}||jvrtd|t.|jvrtd |t.urLt|t0r|dkrtd ||krtd |d |d |d|d | |j|<|j| t| |stdt5| |vrtd|t5| || j| j} }| j|ks | j| krtd|j| |jstdt=d|| |j|dS)aD Args: target: the real-time value to switch on. cases: an ordered iterable of the corresponding value of the ``target`` and the circuit block that should be executed if this is matched. There is no fall-through between blocks, and the order matters. rrzNthe switch target must be an expression with type 'Uint(n)' or 'Bool()', not ''z5the switch target must be a classical bit or registerr)NNzduplicate case value z'cases after the default are unreachablez5case values must be Booleans or non-negative integerszswitch target 'z' has z bit(s) of precision, but case z is larger than the maximum of .z,case blocks must be QuantumCircuit instancesz.ungrouped cases cannot point to the same blockNzincompatible bits between casesz"must have at least one case to run switch_caser+) qiskit.circuitr isinstancerExprtypekindrUintBoolrrrwidthlensetr- _case_map _label_spec_params enumeratetuplelistrintappendidadd num_qubits num_clbitssuper__init__)r"r-r/r,r target_bits target_maxcase_idsrJrKi value_speccase_valuesvalue __class__s r#rMzSwitchCaseOp.__init__8s 211111 fdi ( ( X{ EJ'???",#[,,,@ FU,=$>?? XVWW W fdi ( ( J%{/5:==!!6;CTKK)&%88I!!c&kkK;&!+ 55!+ J  &QS   S'0&6&6 ' ' "A" E*4Z%*O*ObU:&&&V`UbF * *DN**&'Fu'F'FGGG4>11&'PQQQ ,,%eS11dUQYY*+bcccz))*]f]]K]]).]]OY]]])*u%%   # #F + + +e^44 S"#QRRR%yyH$$"#STTT LLE # # #!).)95;KJ :--1AZ1O1O"#DEEE L   & & & &| ECDD D  J TYZZZZZr%ct|o@|j|jko0tdt |j|jDS)Nc3\K|]'\}}t|t|kV(dSN)r?).0 labels_self labels_others r# z&SwitchCaseOp.__eq__..sO-KK  C $5$55r%)rL__eq__r-allziprA)r"otherrVs r#r^zSwitchCaseOp.__eq__sl GGNN5 ! !  u|+ 14T5EuGX1Y1Y r%return&Iterable[Tuple[Tuple, QuantumCircuit]]c6t|j|jS)aReturn an iterable where each element is a 2-tuple whose first element is a tuple of jump values, and whose second is the single circuit block that is associated with those values. This is an abstract specification of the jump table suitable for creating new :class:`.SwitchCaseOp` instances. .. seealso:: :meth:`.SwitchCaseOp.cases` Create a lookup table that you can use for your own purposes to jump from values to the circuit that would be executed.)r`rArBr!s r#cases_specifierzSwitchCaseOp.cases_specifiers4#T\222r%cNfdjDS)aReturn a lookup table from case labels to the circuit that would be executed in that case. This object is not generally suitable for creating a new :class:`.SwitchCaseOp` because any keys that point to the same object will not be grouped. .. seealso:: :meth:`.SwitchCaseOp.cases_specifier` An alternate method that produces its output in a suitable format for creating new :class:`.SwitchCaseOp` instances. c2i|]\}}|j|Sr )rB)rZkeyindexr"s r# z&SwitchCaseOp.cases..s&RRRZS%T\%(RRRr%)r@itemsr!s`r#r/zSwitchCaseOp.casess.SRRR4>;O;O;Q;QRRRRr%c*t|jSrY)rDrBr!s r#blockszSwitchCaseOp.blockssT\"""r%rmIterable[QuantumCircuit]'SwitchCaseOp'c"t|}t|t|jkr4tdt|jdt|t |jt|j|S)Nzneeded z blocks but received ) rDr>rBrr@rr-r`rA)r"rms r#replace_blockszSwitchCaseOp.replace_blocksszv v;;#dl++ + +`T^)<)<``SVW]S^S^``aa aDKT-=v)F)FGGGr%c td)NznSwitchCaseOp cannot be classically controlled through Instruction.c_if. Please nest it in an IfElseOp instead.)NotImplementedError)r" classicalvals r#c_ifzSwitchCaseOp.c_ifs! 5   r%)r-r.r/r0r,r1)rbrc)rmrnrbro) r&r'r(r)rMr^rer/propertyrmrqrv __classcell__rVs@r#rr2s $ O[O[O[O[O[O[O[O[b       3 3 3 3 S S S##X#HHHH        r%rc<eZdZdZdddfd Zd Zd Zd ZxZS)SwitchCasePlaceholderaA placeholder instruction to use in control-flow context managers, when calculating the number of resources this instruction should block is deferred until the construction of the outer loop. This generally should not be instantiated manually; only :obj:`.SwitchContext` should do it when it needs to defer creation of the concrete instruction. .. warning:: This is an internal interface and no part of it should be relied upon outside of Qiskit Terra. Nr+r-r.r/)List[Tuple[Any, ControlFlowBuilderBlock]]r,r1c||_||_||_t dt |jjt |jjg|dS)Nr5r+) _SwitchCasePlaceholder__target_SwitchCasePlaceholder__cases _calculate_placeholder_resources!_SwitchCasePlaceholder__resourcesrLrMr>qubitsclbits)r"r-r/r,rVs r#rMzSwitchCasePlaceholder.__init__s}  @@BB    ' ( (  ' ( (       r%cPt}t}t}t}t|jtr||jnt|jt r5||j||jnHt|j}||j||j |j D]T\}}|| z}||z}t|j \}} ||z}|| z}Utt|t|t|t|S)Nrrqregscregs)r?r7r~rrIrupdaterrrrrr registersrrD) r"rrrr resources_body body_qregs body_cregss r#rz6SwitchCasePlaceholder._calculate_placeholder_resourcessg dmU + + * JJt} % % % %  '8 9 9 * MM$- ( ( ( IIdm $ $ $ $&t}55I MM)* + + + LL ) ) )|  GAt dkkmm #F dkkmm #F%8%H%H "J Z E Z EE#====,,,,     r%c|jSrY)rr!s r#placeholder_resourcesz+SwitchCasePlaceholder.placeholder_resourcess r%c dt|jtfd|jDD}|rtt |ddjt |ddjt |ddjt |ddj}n|j }| t|j ||j |fS)Nc"g|] \\}}}||f Sr r )rZlabelsr unified_bodys r# z>SwitchCasePlaceholder.concrete_instruction..s5   ) \\ "   r%c3JK|]\}}|VdSrY)build)rZrrrrs r#r]z=SwitchCasePlaceholder.concrete_instruction..s5'_'_wq$ 66(B(B'_'_'_'_'_'_r%rrrr+)r`rrrrDrrrrr_copy_mutable_propertiesrr~r,)r"rrr/rs `` r#concrete_instructionz*SwitchCasePlaceholder.concrete_instructions   -0 ''_'_'_'_'_RVR^'_'_'___..     ),U1Xa[/00U1Xa[/00E!HQK-..E!HQK-.. II(I  ) ),t}eSWS]*^*^*^ _ _   r%)r-r.r/r|r,r1) r&r'r(r)rMrrrrxrys@r#r{r{s  $ $         $   6          r%r{c:eZdZdZdddd Zd ZddZdZdZdS) SwitchContextaA context manager for building up ``switch`` statements onto circuits in a natural order, without having to construct the case bodies first. The return value of this context manager can be used within the created context to build up the individual ``case`` statements. No other instructions should be appended to the circuit during the `switch` context. This context should almost invariably be created by a :meth:`.QuantumCircuit.switch_case` 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:`.SwitchCaseOp`). .. warning:: This is an internal interface and no part of it should be relied upon outside of Qiskit Terra. Nr+circuitrr-r.in_loopboolr,r1c|||_||_t|tr|f|_d|_nYt|t rt||_|f|_n't|}|j |_|j |_||_ d|_ ||_ g|_t|_dS)Nr F)r_targetr7r target_clbits target_cregsrrDrrrrcomplete _op_label_casesr? _label_set)r"rr-rr,rs r#rMzSwitchContext.__init__(s  fe $ $ 05;ID ?AD    1 2 2 0!&vD !' D  &v..I!*!1D  )D   MO %%r%c||jvS)zMReturn whether a case label is already accounted for in the switch statement.)r)r"r,s r# label_in_usezSwitchContext.label_in_useBs''r%r-Tuple[Union[int, Literal[CASE_DEFAULT]], ...]blockrcr|j||j||fdS)zAdd a sequence of conditions and the single block that should be run if they are triggered to the context. The labels are assumed to have already been validated using :meth:`label_in_use`.N)rrrrG)r"rrs r#add_casezSwitchContext.add_caseFs: v&&& FE?+++++r%cV|jdt|S)Nz'Cannot have instructions outside a case)forbidden_message)r _push_scope CaseBuilderr!s r# __enter__zSwitchContext.__enter__Qs*   3\ ]]]4   r%cd|_|j|dSt|j|j|j}|}|jr'|j ||j |j nc| t|j t|j \}}|j ||j |j dS)NTFr+)rr _pop_scoper{rrrrrrGrrrr?)r"exc_typeexc_valexc_tb placeholderinitial_resources operationrs r#__exit__zSwitchContext.__exit__Us  !!!  5,DL$+T^\\\ '==?? < O L   ->-EGXG_ ` ` ` `#.#C#C%,--s3D3K/L/L$$ Iy L   9+;Y=M N N Nur%)rrr-r.rrr,r1)rrrr) r&r'r(r)rMrrrrr r%r#rrs0 $       4((( , , , ,!!!r%rc@eZdZdZeZ ddZejdZ dS)ra%A child context manager for building up the ``case`` blocks of ``switch`` statements onto circuits in a natural order, without having to construct the case bodies first. This context should never need to be created manually by a user; it is the return value of the :class:`.SwitchContext` context manager, which in turn should only be created by suitable :meth:`.QuantumCircuit.switch_case` calls. .. warning:: This is an internal interface and no part of it should be relied upon outside of Qiskit Terra. parentrc"||_d|_dS)NF)switchentered)r"rs r#rMzCaseBuilder.__init__{s  r%c'K|jrtd|jjrtdt d|Dstdt }|D]H}|j|s||vrtd|d||I|jj |jj |jj |jj  d|_dVd |_|jj }n*#d |_|jj }wxYw|j||dS) NzCannot enter more than one case at once. If you want multiple labels to point to the same block, pass them all to a single case context, such as `with case(1, 2, 3):`.z6Cannot add a new case to a completed switch statement.c3RK|]"}|tupt|tV#dSrY)rr7rF)rZrUs r#r]z'CaseBuilder.__call__..s6WWu5L(BJuc,B,BWWWWWWr%z.Case values must be integers or `CASE_DEFAULT`zduplicate case label: 'r3)rr allow_jumpsTF)rrrrr_r?rrIrrrrrrr)r"rTseenrUrs r#__call__zCaseBuilder.__call__s < 2  ;  YWXX XWWPVWWWWW QOPP Puu  E{''.. G%4--"#EU#E#E#EFFF HHUOOOO '';,k. + (    5DL EEE DLK'2244EE!DLK'2244E 4 4 4 4 VU+++++s 6 D'''EN)rr) r&r'r(r)rDEFAULTrM contextlibcontextmanagerrr r%r#rrjs[  GG!,!,!,!,!,r%r)(r) __future__r__all__rtypingrrrr r r r r qiskit.circuit.classicalregisterrrqiskit.circuit.classicalrrqiskit.circuit.exceptionsrbuilderrrr control_flowr_builder_utilsrrrr6rrrrr{rrr r%r#rsCB"""""" *VVVVVVVVVVVVVVVVVVVVEEEEEEEE00000000222222ZZZZZZZZZZ''''''XXXXXXXXXX.------          !! K K K K K =K K K \R R R R R 2R R R jRRRRRRRRj7,7,7,7,7,7,7,7,7,7,r%