gUdZddlmZddlmZddlmZddlZddlZddl m Z m Z m Z m Z ddl mZmZmZddl mZmZmZmZmZdd lmZdd lmZdd lmZmZmZmZmZdd l m!Z!m"Z"dd l#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5dZ6dZ7dZ8Gdde5Z9GddZ:Gdde:Z;Gdde:Z<Gdde:Z=Gdd e<Z>Gd!d"e:Z?Gd#d$e?e<Z@Gd%d&e?ZAGd'd(eAe<ZBGd)d*e?e<ZCGd+d,e:ZDGd-d.e?e<ZEGd/d0e?e<ZFGd1d2e?e<ZGGd3d4e?e;ZHGd5d6eAe;ZIGd7d8e?e;ZJGd9d:e:ZKGd;deKZMGd?d@eKZNGdAdBeKZOGdCdDeKZPGdEdFe:ZQGdGdHeQZRGdIdJeQZSGdKdLe:ZTGdMdNe:ZUGdOdPe:ZVGdQdRZWGdSdTZXdS)UzN A module for drawing circuits in ascii art or some other text representation )StringIO)warn)get_terminal_sizeN)QubitClbitClassicalRegister CircuitError)ControlledGateResetMeasure) ControlFlowOp WhileLoopOpIfElseOp ForLoopOp SwitchCaseOp)expr)node_resources)IGateRZZGateSwapGateSXGateSXdgGate)_canonicalize_modifiersControlModifier)pi_check)ast) BasicPrinter) _ExprBuilder)get_gate_ctrl_text get_param_str get_wire_mapget_bit_registerget_bit_reg_indexget_wire_labelget_condition_label_val_get_layered_instructions)VisualizationErrorceZdZdZdS)TextDrawerEncodingErrorzA problem with encodingN)__name__ __module__ __qualname____doc__W/granary/edev/quantum/lib/python3.11/site-packages/qiskit/visualization/circuit/text.pyr+r+5s!!Dr1r+ceZdZdZd dZedZedZedZedZ edZ e j d Z d d Z dS) DrawElementz2An element is an operation that needs to be drawn.Ncd|_|x|_|_dx|_x|_|_dx|_|_dx|_x|_ |_ dx|_ x|_ |_ i|_i|_dx|_x|_|_d|_dS)N%s r)_widthlabel mid_content top_format mid_format bot_format top_connect bot_connecttop_pad _mid_paddingbot_padmid_bcktop_bckbot_bck bot_connector top_connector right_fill left_fill layer_width wire_label)selfr:s r2__init__zDrawElement.__init__>s (-- T%>BBB$/DO.114+:== =t(4<588 8t|dl>???$.4+;r1c*|jdzdkrit|jdzdkrNt|jdkr6|j|j|jz|j|jz}n-|j|j|j|jz}|jr ||j|j}|jr | |j|j}||j |j }|S)z&Constructs the top line of the elementr(rr) widthlenr<r?rAcenterrIljustrJrjustrKrErMrets r2topzDrawElement.topJs JNq S%9%9A%=%B%Bs4K[G\G\`aGaGa/T\D4D%D$L$L DL%%CC/D$4$;$;DJ $U$UUC ? ;))DOT\::C > :))DNDL99Cjj)4<88 r1c<|j|j|j|jz}|jr ||j|j}|jr ||j|j}||j |j }|S)z)Constructs the middle line of the element) r=r;rRrPrBrIrSrJrTrKrDrUs r2midzDrawElement.midZso 0 7 7 DDU V VV ? @))DOT->??C > ?))DND,=>>Cjj)4<88 r1c|jdzdkrQt|jdzdkr6|j|j|jz|j|jz}n-|j|j|j|jz}|jr ||j|j}|j r | |j |j}||j |j }|S)z)Constructs the bottom line of the elementr(rr) rPrQr<r>rCr@rRrIrSrJrTrKrFrUs r2botzDrawElement.botes JNq S%9%9A%=%B%B/T\D4D%D$L$L DL%%CC/D$4$;$;DJ $U$UUC ? ;))DOT\::C > :))DNDL99Cjj)4<88 r1ctt|jt|jt|jS)z>d0>#1)>d0>#1)rArCrDr?r@r;)rMr:r?r@ __class__s r2rNzBoxOnClWire.__init__sc *&*&++ t| && r1)r8rnrnr,r-r.r/rN __classcell__rts@r2rmrmsG ! ! ! ! ! ! ! ! ! !r1rmc$eZdZdZdfd ZxZS) BoxOnQuWireuDraws a box on the quantum wire. :: top: ┌───┐ ┌───┐ mid: ┤ A ├ ──┤ A ├── bot: └───┘ └───┘ r8rnFct|d|_d|_d|_dx|_x|_|_||_|rdnd|_ ||_ ddi|_ ddi|_ dS) Nrpu ┤ %s ├rqrn╥│┴┬) rsrNr<r=r>rArCrDr?r@r;rHrG)rMr:r? conditionalrts r2rNzBoxOnQuWire.__init__s *&*5:: :t|dl&$/:55U #U^#U^r1r8rnFrurws@r2ryrysG , , , , , , , , , ,r1ryc$eZdZdZdfd ZxZS) MeasureTouThe element on the classic wire to which the measure is performed. :: top: ║ ║ mid: ═╩═ ═══╩═══ bot: r8ctd|_d|_||_d|_dS)Nu ║ u ═╩═rr)rsrNr?r;r@rDrMr:rts r2rNzMeasureTo.__init__s< "&  r1r8rurws@r2rrsGr1rc"eZdZdZfdZxZS) MeasureFromuThe element on the quantum wire in which the measure is performed. :: top: ┌─┐ ┌─┐ mid: ┤M├ ───┤M├─── bot: └╥┘ └╥┘ ctdx|_x|_|_d|_d|_d|_dx|_|_ d|_ dS)Nr6u ┌─┐u┤M├u └╥┘r7rn) rsrNr<r=r>r?r;r@rArCrB)rMrts r2rNzMeasureFrom.__init__sb >BBB$/DO&$&&)) t|!r1rurws@r2rrsB"""""""""r1rc.eZdZdZdZedZdS)MultiBoxz,Elements that are drawn over multiple wires.cR||cxkrdkrnn|j|_dSd|dzdz ddz}|dzdz}|dz}||cxkr|kr=ndS||kr|j|_dS||dzkr|j|_dS|j|_dSdS)zIn multi-bit elements, the label is centered vertically. Args: input_length (int): Rhe amount of wires affected. order (int): Which middle element is this one? rN*r(r)r:r?rRindexr;r@)rM input_lengthorderlocation_in_the_box top_limit bot_limits r2 center_labelzMultiBox.center_labels 5 % % % %A % % % % %#zD  F!jj)9A)=>>DDSIIAMAIM M + 7 7 7 7i 7 7 7 7 7 7"i//#':   $ A 55#':   #':   8 7r1cF|jr|jSt|jSra)r9rQr:r^s r2rPzMultiBox.widths$ ; ; 4:r1N)r,r-r.r/rrjrPr0r1r2rrsD66...*Xr1rc$eZdZdZdfd ZxZS)BoxOnQuWireTopzCDraws the top part of a box that affects more than one quantum wirer8Nct|||_dx|_|_d|_t |j|_dd|jdzdzdz|_ |j dd|_ d |jd |_ d |j|jzd |_ |r|nd|_ dS) Nr7r8┌─srrn─┐r6┤ %s ├r| %s │)rsrNrLr@rCr;rQrJrRr<replacer=r>r?rMr:r?rLrts r2rNzBoxOnQuWireTop.__init__s $*--4<T_--"SZZ0BE%J%JJXU/11#t<<8888F t~ =FFF*5@;;5r1r8Nr8rurws@r2rr sMMM A A A A A A A A A Ar1rc$eZdZdZdfd ZxZS) BoxOnWireMidzA generic middle boxr8cxt|dx|_x|_x|_|_||_t|j|_d|j|jzd|_ d|j|jzd|_ dx|_x|_|_ | ||dS)Nr7r|rr8) rsrNrArCr?r@rLrQrJr<r>r;r)rMr:rrrLrts r2rNzBoxOnWireMid.__init__s LOO Ot|Od&69I$T_--F t~ =FFFF t~ =FFFACCC4+d.> ,.....r1rrurws@r2rrsC//////////r1rc$eZdZdZdfd ZxZS)BoxOnQuWireMidzFDraws the middle part of a box that affects more than one quantum wirer8Nct|||||r||jd|_dSd|jd|_dS)NrLrrrsrNrLr=)rMr:rrrL control_labelrts r2rNzBoxOnQuWireMid.__init__-s_  e KKK  =!.HHHHDOOOrr@r;r?)rMr:rr@rLrrts r2rNzBoxOnQuWireBot.__init__8s $ T_--F t~ =FFF8888"SZZ0BE%J%JJXU/11#t<<%0;kke $/@55[.004+ 1  $D     r1Nr8Frurws@r2rr5sCPP%%%%%%%%%%r1rc$eZdZdZdfd ZxZS) FlowOnQuWirez5Draws a box for a ControlFlowOp using a single qubit.r8rnFc6t||tkrd|_d|_d|_nd|_d|_d|_dx|_x|_|_||_ |rdnd|_ ||_ d d i|_ d d i|_ dS) Nu ─%s─┐u %s ├u ─%s─┘u ┌─%s─ u┤ %s u └─%s─ rnr{r|r}r~)rsrNCF_RIGHTr<r=r>rArCrDr?r@r;rHrG)rMsectionr:r?rrts r2rNzFlowOnQuWire.__init__Ls  h  ,DO(DO,DOO,DO(DO,DO5:: :t|dl&$/:55U #U^#U^r1rrurws@r2rrIsC??,,,,,,,,,,r1rc$eZdZdZdfd ZxZS)FlowOnQuWireTopzIDraws the top of a box for a ControlFlowOp that uses more than one qubit.r8Nct|||_dx|_|_d|_t |j|_|tkrod |jdzddz|_ |j dd|_ d|jd|_ d|j|jzd |_ nqd d |jdzdzdz|_ |j dd|_ d |jd |_ d |j|jzd |_ |r|nd|_dS)Nr7r8rr(rnrr6rrrr %s r|)rsrNrLr@rCr;rQrJrrRr<rr=r>r?)rMrr:r?rLrts r2rNzFlowOnQuWireTop.__init__asF $*--4<T_-- h  !jj!);UCChNDO"o55c4@@DO:$/:::DOH$,"?HHHDOO&DNQ4F)N)NNQTTDO"o55c4@@DO:DO:::DOHDL4>$AHHHDO*5@;;5r1rrurws@r2rr^sMSSAAAAAAAAAAr1rc$eZdZdZdfd ZxZS)FlowOnQuWireMidzLDraws the middle of a box for a ControlFlowOp that uses more than one qubit.r8c0t|dx|_x|_x|_|_||_t|j|_|tkrAd|j|jzd|_ d|j|jzd|_ d|jd|_ n@d|j|jzd|_ d|j|jzd|_ d|jd|_ dx|_x|_|_ |||dS)Nr7rrr|rrr8)rsrNrArCr?r@rLrQrJrr<r>r=r;r)rMrr:rrrLrts r2rNzFlowOnQuWireMid.__init__ws& LOO Ot|Od&69I$T_-- h  H$,"?HHHDOH$,"?HHHDO:$/:::DOOHDL4>$AHHHDOHDL4>$AHHHDO:DO:::DOACCC4+d.> ,.....r1rrurws@r2rrtsCVV//////////r1rc*eZdZdZ dfd ZxZS)FlowOnQuWireBotzLDraws the bottom of a box for a ControlFlowOp that uses more than one qubit.Nr8Fct|||_d|_t |j|_|t krrd|j|jzd|_d|jd|_dd |jdzdzdz|_ |j dd|_ nqd |j|jzd |_d |jd |_d d |jdzdzdz|_ |j dd|_ |r|nd}|rd n||_ dx|_ |_|dkr ||_dSdS)Nr7rrrr(rnrr6r|rrrr{r8)rsrNrLrArQrJrr<r=rRr>rr@r;r?)rMrr:rr@rLrrts r2rNzFlowOnQuWireBot.__init__sv $ T_-- h  H$,"?HHHDO:$/:::DO!CJJt~/A5$I$IIHTDO"o55c4@@DOOHDL4>$AHHHDO:DO:::DO&DNQ4F)N)NNQTTDO"o55c4@@DO%0;kke $/@55[.004+ 1  $D     r1rrurws@r2rrsNVV%%%%%%%%%%r1rc$eZdZdZdfd ZxZS)BoxOnClWireTopzQDraws the top part of a conditional box that affects more than one classical wirer8Nct|||_d|_d|_|r|nd|_dx|_|_dS)Nr8u │ %s │rnr7)rsrNrLr;r>r?r@rCrs r2rNzBoxOnClWireTop.__init__sU $&*5@;;5*--4<<rr@r;r?)rMr:rr@rLrrts r2rNzBoxOnClWireBot.__init__s $T_--  F t~ =FFF8888"SZZ0BE%J%JJXU/11#t<<%0;kke &.004+ 1  $D     r1)rnr8rurws@r2rrsC^^%%%%%%%%%%r1rc$eZdZdZdfd ZxZS)DirectOnQuWirez0 Element to the wire (without the box). r8ct|d|_d|_d|_dx|_|_ddd|_ddd|_dS)N %s u─%s─rnr|║r|r rsrNr<r=r>rBrDrHrGrs r2rNzDirectOnQuWire.__init__e  $ +00DL%*599%*599r1rrurws@r2rrG::::::::::r1rc$eZdZdZdfd ZxZS)BarrieruDraws a barrier with a label at the top if there is one. :: top: ░ label mid: ─░─ ───░─── bot: ░ ░ r8ctd|r|nd|_d|_i|_i|_dS)Nu░)rsrNr?r@rHrGrs r2rNzBarrier.__init__sJ $)455u r1rrurws@r2rrsG          r1rc$eZdZdZdfd ZxZS)ExuDraws an X (usually with a connector). E.g. the top part of a swap gate. :: top: mid: ─X─ ───X─── bot: │ │ r7Fcntd|rdn||_||_dS)NXr)rsrNr@r?)rMr@r?rrts r2rNz Ex.__init__s: $/@55[&r1)r7r7Frurws@r2rrsG''''''''''r1rc$eZdZdZdfd ZxZS) ResetDisplayzDraws a reset gateFc`td|r d|_dSdS)N|0>r)rsrNr@)rMrrts r2rNzResetDisplay.__init__s;   %$D    % %r1Frurws@r2rrsC%%%%%%%%%%r1rc$eZdZdZdfd ZxZS)BulletuDraws a bullet (usually with a connector). E.g. the top part of a CX gate. :: top: mid: ─■─ ───■─── bot: │ │ r8FNctd||_||_|rdn||_|r |r||_n |r||_d|_dS)N■rrnrsrNrr?r@rDrMr?r@rr:bottomrts r2rNzBullet.__init__sq &&$/@55[  %V %$D    %$D  r1r8r8FNFrurws@r2rr G          r1rc$eZdZdZdfd ZxZS) OpenBulletuDraws an open bullet (usually with a connector). E.g. the top part of a CX gate. :: top: mid: ─o─ ───o─── bot: │ │ r8FNctd||_||_|rdn||_|r |r||_n |r||_d|_dS)Norrnrrs r2rNzOpenBullet.__init__-sq &&$/@55[  %V %$D    %$D  r1rrurws@r2rr#rr1rc$eZdZdZdfd ZxZS)DirectOnClWirez: Element to the classical wire (without the box). r8ct|d|_d|_d|_dx|_|_ddd|_ddd|_dS)Nru═%s═rrr|rrrrs r2rNzDirectOnClWire.__init__>rr1rrurws@r2rr9rr1rc$eZdZdZdfd ZxZS)ClBulletuDraws a bullet on classical wire (usually with a connector). E.g. the top part of a CX gate. :: top: mid: ═■═ ═══■═══ bot: │ │ r8FNctd||_|rdn||_|r |r||_n |r||_d|_dS)NrrrrrsrNr?r@rDrs r2rNzClBullet.__init__Rsi &$/@55[  %V %$D    %$D  r1rrurws@r2rrHGr1rc$eZdZdZdfd ZxZS) ClOpenBulletuDraws an open bullet on classical wire (usually with a connector). E.g. the top part of a CX gate. :: top: mid: ═o═ ═══o═══ bot: │ │ r8FNctd||_|rdn||_|r |r||_n |r||_d|_dS)Nrrrrrrs r2rNzClOpenBullet.__init__gsi &$/@55[  %V %$D    %$D  r1rrurws@r2rr]rr1rc8eZdZdZfdZedZxZS) EmptyWirez2This element is just the wire, with no operations.cft||x|_|_dSrc)rsrNrBrD)rMwirerts r2rNzEmptyWire.__init__us. +//DLLLr1cdt|DD])}||krtdntd||<*|S)aGiven a layer, replace the Nones in it with EmptyWire elements. Args: layer (list): The layer that contains Nones. first_clbit (int): The first wire that is classic. Returns: list: The new layer, with no Nones. cg|] \}}|| Srcr0).0ixs r2 z*EmptyWire.fillup_layer..sBBBDAq a r1rrrn) enumerater)layer first_clbitnoness r2 fillup_layerzEmptyWire.fillup_layerys\CBIe$4$4BBB Z ZE/4 /C/C9U+++SXIYIYE%LL r1r,r-r.r/rN staticmethodrrvrws@r2rrrsY<<00000  \     r1rc8eZdZdZfdZedZxZS) BreakWirez;This element is used to break the drawing in several pages.ctdx|_x|_|_||_||_||_dS)Nr6)rsrNr<r=r>r?r;r@)rM arrow_charrts r2rNzBreakWire.__init__sL >BBB$/DO%%%r1crg}t|D]$}|t|%|S)aCreates a layer with BreakWire elements. Args: layer_length (int): The length of the layer to create arrow_char (char): The char used to create the BreakWire element. Returns: list: The new layer. )rangeappendr ) layer_lengthr breakwire_layerrs r2rzBreakWire.fillup_layersE|$$ : :A  " "9Z#8#8 9 9 9 9r1rrws@r2r r sYEE&&&&&  \     r1r c8eZdZdZfdZedZxZS) InputWirez:This element is the label and the initial value of a wire.cJt|dSrc)rsrNrs r2rNzInputWire.__init__s! r1ctd|D}g}|D]7}|t||8|S)zCreates a layer with InputWire elements. Args: names (list): List of names for the wires. Returns: list: The new layer c34K|]}t|VdSrc)rQ)rnames r2 z)InputWire.fillup_layer..s(22Dc$ii222222r1)r]rrrT)nameslongest inputs_wiresrs r2rzInputWire.fillup_layersg22E22222  @ @D    $**W*=*= > > ? ? ? ?r1rrws@r2rrsYDD       \     r1rceZdZdZ ddZdZd Zd Zd Zdd Z dd Z ddZ dZ dZ edZeddZedZedZdZdZdZddZdS) TextDrawingzThe text drawingFTNhighc||_||_||_||_| r+|jjr|jjj|_nd|_nd|_| |_|j|_||_ ||_ ||_ | |_ |dvrtd||_i|_| |_| pt"jjpd|_d|_d|_d|_dS)N)rmediumlowz;Vertical compression can only be 'high', 'medium', or 'low'utf8rr8)qubitsclbitsnodes_circuit_layoutinitial_layoutlayout initial_state global_phase plotbarriers reverse_bits line_lengthexpr_len ValueErrorvertical_compression _wire_map cregbundlesysstdoutencoding _nest_depth _expr_text_single_string)rMr$r%r&circuitr.r-r/r2r+r4r7 with_layoutr0s r2rNzTextDrawing.__init__s      }$ #"m3B " DK*#0((&  '@ @ @Z[[ [$8!$ ACJ$7A6  !r1c*|Src single_stringr^s r2__str__zTextDrawing.__str__!!###r1c2d|dS)Nz
z
r>r^s r2 _repr_html_zTextDrawing._repr_html_s, , !!##  , , , r1c*|Srcr>r^s r2__repr__zTextDrawing.__repr__rAr1c|jr|jS d||j|j|_n#t tf$rtd|jdtd|_d||j|j|_YnwxYw|jS)z{Creates a long string with the ascii art. Returns: str: The lines joined by a newline (``\n``)  z The encoding zf has a limited charset. Consider a different encoding in your environment. UTF-8 is being used insteadzutf-8) r:joinlinesencoder7decodeUnicodeEncodeErrorUnicodeDecodeErrorrRuntimeWarningr^s r2r?zTextDrawing.single_strings    '& &  $**,,''..t}==DDT]SS   #$67    ; ;;;     $DM $**,,''..t}==DDT]SS     ""sAA--BDDct|d|p|j5}||ddddS#1swxYwYdS)zDumps the ascii art in the file. Args: filename (str): File to dump the ascii art. encoding (str): Optional. Force encoding, instead of self.encoding. w)moder7N)openr7writer?)rMfilenamer7 text_files r2dumpzTextDrawing.dumps(x/H4= I I I 2Y OOD..00 1 1 1 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2s(AAAcb||j}|s0dtjvrdtjvrd}nt\}}t |j}|}gg}|}t|D]\}}t ||||<t ||dkr|d |`||dj } | |kr!|d ||| z}|d t t |d| t t |dg||dddj z }|d t |d ||dddj z}|d |||dddj z}g} |jr,| d t%|jd |D]0} t't)| } | || z } 1| S)aDGenerates a list with lines. These lines form the text drawing. Args: line_length (int): Optional. Breaks the circuit drawing to this length. This is useful when the drawing does not fit in the console. If None (default), it will try to guess the console width using shutil.get_terminal_size(). If you don't want pagination at all, set line_length=-1. Returns: list: A list of lines with the text drawing. N ipykernelspyderPrfr»«Fwith_initial_statezglobal phase: ndigits)r/r5modulesrrQr$ build_layersrrrrnormalize_widthrr_r r wire_namesr,rlistzip draw_wires) rMr/rnoqubitslayers layer_groupsrest_of_the_linelayernorrrI layer_groupwiress r2rIzTextDrawing.lines$s  *K 5s{**1L1L !2!4!4 Qt{##""$$t &'// C CNGU'44UHEEF7O  ' ' . . .b  R ''..."'?1-4L...R ''... L0  R '' (>(>s5zz4(P(PQQQ##Y%;%;CJJ%M%M$NOOO#.b1A"1Ea1H1O#O R ''**4??e?+T+TUU!L$4R$8$;$BB R ''... L$4R$8$;$BB     T LLR(43Da*P*P*PRR S S S' , ,Kk*++E T__U++ +EE r1cV|rd}d}nd}d}t|j|j|jz|j|_g}|jD]\}}t|tr|}nt|j|\}}} ||n| }td|||j |j} | |j t|trdndz } d} t|tr|} n"|} |jr|t|jd z} || | z| z|S) zReturns a list of names for each wire. Args: with_initial_state (bool): Optional (Default: False). If true, adds the initial value to the name. Returns: List: The list of wire names. rz0 r8Ntext)r*r4r7z: /)r"r'r$r%r4r3items isinstancerr$r%r*rstrsizer) rMr^initial_qubit_valueinitial_clbit_value wire_labelsrrregister bit_index reg_indexrL cregb_addinitial_bit_values r2rezTextDrawing.wire_namesisn  %"' "&  "$ "$ %dmdkDK6ORVRabb >//11 K KKD%$ 122 E1B4=RV1W1W.)Y%-%5 9'% J !8Ze=T=T!8##Z^ ^JI$&& 9$7!!$7!?9x'; #HM 2 2S 8I   z,== I J J J Jr1c|jdkrdS|jdkrdSt||D]G\}}|dvr|dvrdS|r|dks|r |dkrdSHdS)zdDecides if the top_line and bot_line should be merged, based on `self.vertical_compression`.rTr"F)r}╨)r~r{r7)r2rgisalnum)rMtop_linebot_linerWr[s r2should_compresszTextDrawing.should_compresss  $ . .4  $ - -5Hh//  HCn$$)>)>uu  #**#++--*C3JJuutr1cg}d}|D]G}d}|D] }||jz } |||n|||rA|t||n6|t|d|dd}|D] }||jz } |t|d|dd}|D] }||jz } |t|d|dI|S)zGiven a list of wires, creates a list of lines with the text drawing. Args: wires (list): A list of wires with nodes. Returns: list: A list of lines with the text drawing. Nr8rfr[)icod)rWrrr merge_linespoprYr[)rMrorIrrrnodemid_lines r2rhzTextDrawing.draw_wiresst S SDH % %DH$ X&&&&''(;;[LL!8!8h!O!OPPPPLL!8!8rHSX!8!Y!YZZZH % %DH$ LL00rH50QQ R R RH % %DH$ LL00rH50QQ R R R R r1c~tdtdtdi}t|dd}||dS)z%Some instructions have special labelsIu√Xu√Xdg base_classN)rrrgetattrget)rlabels node_types r2 special_labelzTextDrawing.special_labels;ffhAD,55 zz)T***r1rWc d}t||D]y\}}||kr||z }|dvr |dkr|dz }"|dkr||z }.|dvr|dvr |dkr||z }B|dvr|dkr |d kr|dz }X|d vr|dkr |d kr|d z }n|d vr |d kr|dz }~|d vr |dkr|dz }|dvr|dkr |dkr||z }|dvr|dkr |dkr||z }|dvr|dkr |d kr||z }|dvr |d vr|dz }|dvr |dvr|dz }|dvr |dvr|d z }|dvr |dvr|dz } |dkr|dkr |dkr|dz }"|dkr|dkr |dkr|dz };|dvr |dkr|dz }L|dvr|dvr |dkr|d z }a|dkr |dkr||z }t||z }{|S)!aIMerges two lines (top and bot) in a way that the overlapping makes sense. Args: top (str): the top line bot (str): the bottom line icod (top or bot): in case of doubt, which line should have priority? Default: "top". Returns: str: The merge of both lines. r8u┼╪r7r|u┬╥u ║│rWr~r[r{ru┬│rru╪rnu┼u└┘║│░u─═u║╥u╬u╫u ║╫╬u │┼╪u└u┌u├u┘u┐ru┐┌u┘└r})rg)rWr[rrVtopcbotcs r2rzTextDrawing.merge_linessc3--* * JD$t||t !!dckku t !!di&7&7DEMMt 43;;45==u 43;;45==u !!demmu !!demmu ***ts{{tu}}t !!dckkdemmt !!dckkdemmt !!demmu !!demmu $$u $$u 45==TU]]u 45==TU]]u !!demmu !!demm u t t  r1cttd|}td|D}|D] }||_ dS)z When the elements of the layer have different widths, sets the width to the max elements. Args: layer (list): A list of elements. c |duSrcr0)rs r2z-TextDrawing.normalize_width..s atmr1c3$K|] }|jV dSrc)r_)rrs r2rz.TextDrawing.normalize_width..s$44ddk444444r1N)rffilterr]rK)rr&rrs r2rdzTextDrawing.normalize_widths_V33U;;<<44e44444 ' 'D&D   ' 'r1c |j}|r|jn|j}|jd|}|j|d|r|jn|j}|d|dddd}g} g} g} t fdD} t ||D]} t| | dkr| | 7t| | dkr| | l| | g}tt|D]}t|ddd}||d kr3|t||t|  T|t||t|  || | | fS) a; Analyzes the node in the layer and checks if the controlled arguments are in the box or out of the box. Args: node (DAGNode): node to analyse wire_map (dict): map of qubits/clbits to position ctrl_text (str): text for a control label conditional (bool): is this a node with a condition mod_control (ControlModifier): an instance of a modifier for an AnnotatedOperation Returns: Tuple(list, list, list): - tuple: controlled arguments on top of the "node box", and its status - tuple: controlled arguments on bottom of the "node box", and its status - tuple: controlled arguments in the "node box", and its status - the rest of the arguments Nb0rfc30K|]}|v|VdSrcr0)rr args_qubitswire_maps r2rz/TextDrawing.controlled_wires..4s1QQq[@P@Px{@P@P@P@PQQr1r conditionT1)rr:r)opnum_ctrl_qubitsqargs ctrl_staterTsortedrgminrr]rrQrrboolr)rr ctrl_textr mod_controlrr ctrl_qubitsrin_boxtop_boxbot_box qubit_indices ctrl_qubitgatesrrs ` @r2controlled_wireszTextDrawing.controlled_wiress*W9D\+55"J\j!1/!12 j!1!12 /:M[++ "&&,,_cBB44R4H QQQQQHQQQQQ k:66 * *J=!!HZ]$;;;z****]##hz!}&===z**** j))))s;''((  Ar;--9" !}## V 9UYZaUbUbcccdddd ;iPTU\P]P]^^^w==r1c<!|j}g}g}d}d!t|dd}t|dd} t|tt t fsFt|dds5t|d\} } } t |p| } | | z} t|dd"|| |j |d z }d !d } d}t|d dr2t|j }|D]}t|tr|}nt|trt}||jd|t%j|jd\}} }jr?|=||jdt/t1|n ||jdt/nt|ddrmjs||||fSt5|jfd}|jD]=}|jvr2||kr|jnd}||t;|>n_t|t r>!fdt=t?|jD}| |||||n t|t r1||jdtA!nt|tBr6d| }tE!tE!g}| |||||n{t?|jdkr9|js2||jdtG| !n*t|tHs|rt%||| !|}|\}}}}}|rt?|dkr&|&tG| !n|rdnd}|rdnd}|'|| !|||}t=t5|tQ|dzD]&} |&| tSdf'n|j*dkr%|&tE!np|j*dvrB|j*+| }|&tE!n%|j*dkr3|tY!tY!gz }| |||||n|j*dkr)d| }|tE!tE!gz }nt?|dkr||rdnd}|rdnd}|'|| !|||}t=t5|tQ|dzD]&} |&| tSdf'n$|&tG| !| |||||nt?|jdkr%|js|'|j| !nG|jr+|jr$|-| |j|j!nt]d|j*|/d d!|D}|/d"d#|D}||||fS)$zConvert a dag op node into its corresponding Gate object, and establish any connections it introduces between qubits. gate_wire_map is the flow_wire_map if gate is inside a ControlFlowOp, else it's self._wire_mapNF base_gaterqr_r` _directiverrr?Tct|D]^\}}||j|}|d|Dvr8||j|||||f_dS)Ncg|]\}}|Sr0r0rrjs r2rzITextDrawing._node_to_gate..add_connected_gate..fs'C'C'Cda'C'C'Cr1)rr set_qubitr)rrr current_cons gate_wire_maprgate actual_indexs r2add_connected_gatez5TextDrawing._node_to_gate..add_connected_gatecs$U++ > >4,TZ]; 'C'Cl'C'C'CCCOODJqM4888 ''t(<===  > >r1 modifiersrcTj|tdS)Ninf)r3rfloat)qrMs r2rz+TextDrawing._node_to_gate..s dn6H6HERWLL6Y6Yr1keyr8c0g|]}tS)r)r)rrrs r2rz-TextDrawing._node_to_gate..s$QQQQRK000QQQr1rZZrr}r~)rcontrolled_edger?r@z)u1pswaprzzr()rcargsrz7Text visualizer does not know how to handle this node: c|dSNrr0tups r2rz+TextDrawing._node_to_gate..s #a&r1cg|]\}}|Sr0r0)rrgs r2rz-TextDrawing._node_to_gate..s333da333r1c|dSrr0rs r2rz+TextDrawing._node_to_gate..s s1vr1cg|]\}}|Sr0r0)rcrs r2rz-TextDrawing._node_to_gate..s===41aQ===r1)0rrr!rtr rr r rrset_cl_multiboxrrrrrrrr$r'rr4 set_clbitrrur-rr$r:rrrQrrrryr rrset_qu_multiboxr]r4rupperr _set_multiboxr)sort)"rMrrrrrcurrent_cons_condconnection_labelrparams gate_textrrrrcanonical_modifiersmodifierrrzr| top_qubitqubitr:rcontrols_arraycontrolled_topcontrolled_botrrestr?r@indexesrrs"` @r2 _node_to_gatezTextDrawing._node_to_gateLs W  B T22 r61555"w%899 +'"l\aBbBb +&8V&D&D #Iy!#11"55BI!F*I 2{D ) ) 5 !6!6r|]`e!6!f!f f K > > > 2{D ) ) "9","G"G /  h88"*KE b' " "o ==D OODJqM4 0 0 0%6t}djQRm%T%T "Ha <8#7JqMc)nn--  1 y{{;;;; Ru - -c $ Pl,=?OOODJ,Y,Y,Y,YZZZI ; ;DK''(-(:(:BHHEOOE75>>::: ; H % %X QQQQ%DJ:P:PQQQE  tUE< O O O O E " "S  OODJqM>?D/ Nt99>>LLYK!P!P!PQQQQ+9"C%%tK+9"C%%tK#33!$/(7$/$/ 4G"'s7||S\\A5E!F!FFF$++UKOO,DEEEEF3&& V <<<====;..&/n&:&:&<&<#Ff#F#F  V <<<====6))"555rk7R7R7RSS""4|]SSSS5((#0== &[9996k;Z;Z;Z[[TQ'5?ee4 '5?ee4 // +$3 + + 0#3w<<W1ABBBBE '' B(@AAAAB [ LLLMMM  tUE< O O O O __ ! !$* !  ! !$*i[ ! Q Q Q Q Z DJ    jj'     %I27  0011133l333 #5#5666==+<===l$57GGGr1c|||j}|sgSt|g}|jD]}t |j|j|j|j |j }|D]}t|j tr$d|_||||j @||||j \}}}}|j||f|jd|f|d||j|S)z Constructs layers. Returns: list: List of DrawElements. Raises: VisualizationError: When the drawing is, for some reason, impossible to be drawn. r]rNr|)rer+rrr&Layerr$r%r4r'r3rtrr r8add_control_flowr connectionsr connect_with full_layer) rMrerj node_layerrrrrrs r2rczTextDrawing.build_layerss^__8J_KK  I((445* , ,J    E# H Hdg}55H'(D$))$GGGGOSOaOaeT^PPLE<): -MNNN%,,d4E-FGGGG   u % % % MM%* + + + + r1c  t|jtr$t|jjtjs;t |jddr"t|jjtjrfd}t|jtr |jjn |jj}tt|j }t}t|d |t||_t%jjkrjdjdz_nt|jt( }|t,|}||jt3|jj} t|jtrX| dt3|jdd t=| D]@\} } } | fd tC|j"| j#DtC|j$| j D]s\} }j%rJtMj'|x}3tMj'| }| ||iV| || it| dkr<|| tP| d z d }||jtS| | \}}}|D]}tUj#j j%j'| }|D]}t|jtVr8xj,d z c_,-||| xj,d zc_,T.||| \}}}}|j/||f|j/d|f|0d ||jB|| tbd }||jdS)z,Add control flow ops to the circuit drawing.rNct|tjrtj|jSt|t rtj|jS tj|\}}}n-#t$r tjd|cYSwxYw|tjd|Stj |jtj |S)zLook up a classical-expression variable or register/bit in our internal symbol table, and return an OQ3-like identifier._bit) rtrVarr Identifierrrr$r'r SubscriptedIdentifierIntegerLiteral)varrzr{r|rMrs r2 lookup_varz0TextDrawing.add_control_flow..lookup_vars c48,,4>#(333c#4554>#(333 B5Ft}VY5Z5Z2Hi#BBB>*@#*@*@AAAAAB#>*<*<*<===0@RS\@]@]^^^s$A>>'B('B(z )indentz...rrrc(i|]\}}||Sr0r0)routerinnerrs r2 z0TextDrawing.add_control_flow..Ks#\\\LE5\\\r1F)rr|)2rtrrtargetrExprrrrrr%rrvisitacceptrgetvaluer9rQr0r draw_flow_boxCF_LEFTrrrfblocksinsertcases_specifiercopy_empty_likercopyupdatergrr$rr4r#r'CF_MIDr'rr r8rrrrr)rMrrjrrrdraw_conditionalstream flow_layer circuit_listcirc_numr; flow_wire_maprrin_regout_regrr& layer_nodes flow_layer2 layer_noderrrs` ` r2rzTextDrawing.add_control_flows tw - -% B*TW^TY2W2W% B DG[$ / /% B4>tw?PRVR[4\4\% B _ _ _ _ _ _0+5TWl*K*KbQUQXQbI#N9$=$=$DEE ZZF  - - - 3 3I4D4D\R\E]E]4^4^ _ _ _$oo//DO4?##dm33"&//DM/"BU"J#-dgy#A#AA ''hM]'^^  j+,,,DGN++ dg| , , \   4(?(?(A(A#B#B1#Ea#H#X#X#Z#Z [ [ [!*O7PQQQQ((/// k45555- 62''mXSX'YY  j+,,,,,r1rc |j}t|j}|tkrd}|jr d|jz}t |t r d|z|z} nlt |tr d|z|z} nMt |trq|j d} dt| vr8t| dkr%t| dd} | dd d z} nt| } d |zdz| z} nd |z|z} n|tkrt |t rd |z} ng} t| D]\} }| | dt| |dvrd}n)t| |dd}d|zdz|z} nd|z} t!|j|j|j|j}t|jdkrC||j|jdt/|| |nfd|jD}t1|}t3|}||z dz}||j|t5|| dt7t9|dz|D]8\}}||j|t;|| ||d9||j|t=|| ||d|rt |jt>rt |jj tBj"r |jj }n^t |jj tFr|jj df}n0|jj d|jj j$zdz f}n |jj%}|&|d}|S)z5Draw the left, middle, or right of a control flow boxr8r7zIf-zWhile-rrNrfz, ...)zFor-zSwitch-zElse-defaultz,))zCase-zEnd-r)r:rc g|] }| Sr0r0)rqargrs r2rz-TextDrawing.draw_flow_box..sCCC d+CCCr1)r:rL)r:rrrL)r:rrrLr(rr)'rrur8r r9rtrrrrrQrrfrrrrr$r%r4r'rrrrr]rrrrrrrrrrrvrr)rMrrrrrrdepthetextr:indexset index_str jump_list jump_valuesrjump_strridx_listmin_idxmax_idx box_heightrrrs ` r2r zTextDrawing.draw_flow_boxzsSWD$%% g  E .do-"h'' 2 -B ,, 2 5(50B ** 29Q<#h--//CMMA4E4E #HRaRL 1 1I )#2# 9II #H I,y8!E)E1   "h'' 9% &*2+=+=+?+?&@&@22NK$$[1111Ih$7$: ; ;;;(HH"9X#677??cJJH%#-8UNE K K O M     tz??a     M$*Q-89WE{KKK     DCCC CCCH(mmG(mmG 7*Q.J   G$ogUWY&Z&Z&Z   &eGaK&A&ABB  q$$KN##%/##%        G$!+ +!      X$'<00 .dgndi88Q $II66Q!% 3II!%tw~7J1Ka1O PII G- **9mQV*WWAr1) FTNrTNNFrrcr)rW)rF)r,r-r.r/rNr@rCrEr?rVrIrerrhrrrrdrrrcrr r0r1r2rrs#0!0!0!0!d$$$   $$$###42222CCCCJ((((T   $$$L++\+ 666\6p ' '\ '5>5>\5>n_H_H_HB###Jf-f-f-Paaaaaar1rcreZdZdZdZedZdZdZ ddZ dd Z d Z dd Z d Z dS)rz'A layer is the "column" of the circuit.c||_||_|rfg|_d}|D]Y}t|j|}|r||kr ||j|=|}|j|Zn||_dgt |z|_g|_dgt |z|_||_ ||_ dSrc) r$r'r%r#rrQ qubit_layerr clbit_layerr4r3) rMr$r%r4r;r previous_cregbitrzs r2rNzLayer.__init__s   !DK M 6 6+DM3?? ]h%>%>#K&&s++++$,MK&&}5555 6!DK 6CKK/ 6CKK/$!r1c |j|jzS)z Returns the composition of qubits and classic wires. Returns: String: self.qubit_layer + self.clbit_layer )r1r2r^s r2rzLayer.full_layers$"222r1c0||j|j|<dS)zSets the qubit to the element. Args: qubit (qbit): Element of self.qubits. element (DrawElement): Element to set in the qubit N)r1r3)rMrelements r2rzLayer.set_qubits3:.///r1ct|j|}|jr.|,||j|j|t |jz <dS||j|j|t |jz <dS)zSets the clbit to the element. Args: clbit (cbit): Element of self.clbits. element (DrawElement): Element to set in the clbit N)r#r'r4r2r3rQr$)rMclbitr7rzs r2rzLayer.set_clbit sw$DM599 ? Qx3LSD T^H5DK8H8HH I I IIPD T^E2S5E5EE F F Fr1NFc 8 96tfdt|jD}tfdt|jD} t t t t dz t t t dz fd|jD} fd|jD} t|jjt|jjt |jt|z t | zdz} |  dt|| dd} ttt|dzt |jD]m\} }||vr+ d}| d}n|j|}d z}| |t|| | |nttt | | dzD]m\} }|| vr+ d}| d}n|j|}d z}||t|| | |n| dt!|| | d| St#tfd t|jD}t t t dz |jjd gt z} |j}t&}t(}t}t }nӉ‰t#tfd t|jD}t t t dz fd |jD} |jj|j }t*}t}t}t,}nt/di}|r5t|jD] \}}|D]\}}||kr |dkrdnd||<!t |dkr|d|||nXt |t|z dz} | d|||| dttt|dzt |D]}\} }||vr+ d}| d}n|j|jz|}d z}||}||||| | ||~| d||| || d||S)Nc3*K|] \}}|v |VdSrcr0)rrrrs r2rz&Layer._set_multibox.."+!U!U1!u**!****!U!Ur1c3*K|] \}}|v |VdSrcr0)rrrrs r2rz&Layer._set_multibox..#r<r1rcg|]<}|vt|d=Sr7rurrS)rqbitrwire_label_lens r2rz'Layer._set_multibox..(N5==EKK%%&&,,^SAA ==r1cg|]<}|vt|d=Sr?r@)rcbitrrBs r2rz'Layer._set_multibox..-rCr1rrrr7c3*K|] \}}|v |VdSrcr0rrrbitss r2rz&Layer._set_multibox..U+ S Stq!d S Sr1r8c3*K|] \}}|v |VdSrcr0rGs r2rz&Layer._set_multibox..arIr1cg|]<}|vt|d=Sr?r@)rrArHrBs r2rz'Layer._set_multibox..csN4<<DJJt$$%%++NC@@<?R?R;S;STTN KI  KI 5dk&7888E5dk&7888ET[))C ,=,==L@Q@QQTUUJ NN599Q<<)--XYJZJZ)[)[)[ \ \ \E )%L0A0AA0Es4;GWGW*X*X Y Y   uL(( % ! I!*q!1!1JJ $ E 2I!$~!5J~eZS]^^^!*%L0A0A*B*BEAI N N   uL(( % ! I!*q!1!1JJ $ E 2I!$~!5J~eZS]^^^ NN ! nUJ9==YZK[K[\\\     =U.;;D S S S Sy/E/E S S SSSK SYY]!3!344N II$++I , , ,s4yy(InG F&I&I&II ]u0;;D S S S Sy/E/E S S SSSK SYY]!3!344N KI II$++I , , ,nG F&I&I&II$%<== =  N )$+ 6 6 N N u,;NN(M5 --8= uu# e,N {  q GDGVVE{CCC D D D D[))C ,<,<.sHHHC(3-HHHr1)rtrrrr% collections defaultdictrfr#r'rrextendset_cond_bulletsrQr4rrmrsr&rrrrv)rMrrr?r:outcondition_bits registersr4 registerlessrzrHval_bitsr%r]s @r2rzLayer.set_cl_multiboxs i + + EC+I66=N#/55I% L L*4=#>>?FFsKKKK(}}T2666|  ))%#\9J9J1JLZbcc `!*ccHNN8A; %U`0a0a0abbbbc'0oo&7&7``NHdJJt44USECII8TTTTNN9Q<5Va1b1b1bccccy|A %U`0a0a0abbbI)A,.. I#A,HHHH53G3GHHH((&(KK Kr1c 4g}tfd|D}t|D]\}}d} ||kr|} ||dkr5td| |j|t |jz <n@||dkr4t d| |j|t |jz <|} | d|Dvr=|| |j|t |jz f|S)aSets bullets for classical conditioning when cregbundle=False. Args: label (str): String to display below the condition val_bits (list(int)): A list of bit values clbits (list[Clbit]): The list of classical bits on which the instruction is conditioned. wire_map (dict): Map of bits to indices Returns: List: list of tuples of open or closed bullets for condition bits c3(K|] }|V dSrcr0)rr4rs r2rz)Layer.set_cond_bullets..s'77x}777777r1r7rr)r?r@rcg|]\}}|Sr0r0rs r2rz*Layer.set_cond_bullets..s#?#?#?$!QA#?#?#?r1)r]rrr2rQr$rr) rMr:rfr%rrwire_maxrr4r@rs ` r2razLayer.set_cond_bulletssZ 777777777''  FAsK}((# {c!!EM %;FFF #T[1A1A!ABB!##EQ %;FFF #T[1A1A!AB$C=L#?#?,#?#?#???##!4#3HSMC DTDT4T#UVr1c8|||||||S)aSets the multi qubit box. Args: bits (list[int]): A list of affected bits. label (string): The label for the multi qubit box. top_connect (char): None or a char connector on the top bot_connect (char): None or a char connector on the bottom conditional (bool): If the box has a conditional controlled_edge (list): A list of bit that are controlled (to draw them at the edge) Return: List: A list of indexes of the box. )rr?r@rr)r)rMrHr:r?r@rrs r2rzLayer.set_qu_multiboxs3*!! ###+ "   r1cF|jD]\}}|s t|D]\}}t|ttfrzd}|dkr+t |dkr||dgU|t |dz kr||dg||ddg|dkr||dg|t |dz kr||dg|||ddg|rz|D]w}t |t |jz|_t|ttfr0|j r)t |t |jz|_ xdS)uConnects the elements in the layer using wire_char. Args: wire_char (char): For example '║' or '│'. rrrr[rWN) rrrtrrrQrirYrIrrrrJ)rMrgr: affected_bitsr affected_bits r2rzLayer.connect_withs%)$4 T T E=  '0'?'? H H#|lX|,DEEH %Izzc-&8&81&<&<$,,Y@@@@#m"4"4q"888$,,Y@@@@$,,YGGGGzz$,,Y@@@@#m"4"4q"888$,,YGGGG$,,YGGGG T$1TTL.1%jj3|?O;P;P.PL+!,0DEET,JbT14Uc,BR>S>S1S .5 T Tr1)NNNNFN)r})NNFN)r,r-r.r/rNrjrrrrrrarrr0r1r2rrs11""".33X3::: Q Q Q @@@@D9L9L9L9Lv   L    r}s$$$$$$ HHHHHHHHHHHH9999999999XXXXXXXXXXXXXX))))))555555\\\\\\\\\\\\\\WWWWWWWW222222------......                    ,+++++        0   ZNZNZNZNZNZNZNZNz!!!!!+!!!,,,,,,+,,,. $"""""+"""*{@ A A A A AX{ A A A / / / / /8 / / /=====\;===%%%%%X{%%%(,,,,,;,,,*AAAAAh AAA,/////h ///(%%%%%h %%%D . . . . .X{ . . .99999\;999%%%%%X{%%%* : : : : :[ : : :     n   $ ' ' ' ' ' ' ' ' %%%%%>%%%^,, : : : : :[ : : :~*>* . 6 .` ` ` ` ` ` ` ` FWTWTWTWTWTWTWTWTWTWTr1