Ed.xdZddlmZmZddlmZddlmZddl m Z m Z m Z m Z ddlmZddlmZgdZed Zed d d gief Zer!erejZejjZejjZGddZdZifdZddZGdde ZGdde Z GddeZ!ddZ"dS)aMatplotlib based plotting of quantum circuits. Todo: * Optimize printing of large circuits. * Get this to work with single gates. * Do a better job checking the form of circuits to make sure it is a Mul of Gates. * Get multi-target gates plotting. * Get initial and final states to plot. * Get measurements to plot. Might need to rethink measurement as a gate issue. * Get scale and figsize to be handled in a better way. * Write some tests/examples! )ListDict)Mul) import_module)Gate OneQubitGateCGateCGateS) BasicMeta)ManagedProperties) CircuitPlot circuit_plotlabellerMzMxCreateOneQubitGate CreateCGatenumpy matplotlibfromlistpyplot) import_kwargscatchceZdZdZdZdZdZdZdZdZ gZ iZ dZ dZ dZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdS)r z$A class for managing a circuit plot.g?g4@g?g333333??c trtstd||_t |jj|_||_||| | | | | dS)Nz"numpy or matplotlib not available.)npr ImportErrorcircuitlenargsngatesnqubitsupdate _create_grid_create_figure _plot_wires _plot_gates_finish)selfcr#kwargss Alib/python3.11/site-packages/sympy/physics/quantum/circuitplot.py__init__zCircuitPlot.__init__>s D DBCC C $,+,,   F     c:|j|dS)z'Load the kwargs into the instance dict.N)__dict__r$)r*r,s r-r$zCircuitPlot.updateKs V$$$$$r/c|j}td|j|z|t}td|j|z|t}||_||_dS)zCreate the grid of wires.g)dtypeN)scaleraranger#floatr" _wire_grid _gate_grid)r*r4 wire_grid gate_grids r-r%zCircuitPlot._create_gridOs[ IIc4<#5uEIJJ IIc4;u#4e5III ##r/ct|j|jz|j|jzfdd|_|jdddd}|d|jz}||j d|z |j d|z| |j d|z |j d|z| d ||_ d S) z"Create the main matplotlib figure.w)figsize facecolor edgecolorT)frameonrrequalN)rfigurer"r4r#_figure add_subplot set_axis_offset_xlimr8set_ylimr7 set_aspect_axes)r*axoffsets r-r&zCircuitPlot._create_figureWs}}[+T\$*-DE%   \ % % q!&   TZ DOA&/1Dv1MNNN DOA&/1Dv1MNNN g r/c f|jd}|jd}||jz ||jzf}t|jD]}|j||j|f}t ||d|j}|j||j rd}|j |j |rd}|j |d|j z |z |dt|j ||j |jddd|dS) z&Plot the wires of the circuit diagram.rrBkcolorlwg?center)sizerQhavaN)r8r4ranger#r7Line2D linewidthrKadd_linelabelsinitsgettext label_buffer render_labelfontsize_plot_measured_wires)r*xstartxstopxdataiydatalineinit_label_buffers r-r'zCircuitPlot._plot_wiresis?##$*$edj&89t|$$ 7 7A_Q');D J   % % %{ 7$%!:>>$+a.11Kt3D !HT../@@q Q ;;X  777 !!#####r/c|}|jd}d}|D]q}|j||||jzf}|j||z|j||zf}t ||d|j}|j|rt| D]\}} t| ttfr| j | jz} | D]} | |vr|j||j|| krpt| t!| f}|j||z |j||z f}t ||d|j}|j|dS)NrBg{Gz?rOrP) _measurementsr8r4r7rXrYrKrZ enumerate_gates isinstancer r controlstargetsminmax) r* ismeasuredrddyimrergrhrfgwireswires r-rbz CircuitPlot._plot_measured_wiress'')) #  & &B_Z^4U4:5EFE_R(+DOB,?,BCEu>D J   % % % %T[[]]++ 2 2CAa!eV_-- 2 QY.! 2 2Dz) 2?1- 4@P0QQ 2 #E CJJ 6 $ 22 5tq7I"7L L%!5"%#~     ++D111 2 2r/c4g}t|jtrGt|jjD],}t|t r||-n4t|jt r||j|S)z/Create a list of all gates in the circuit plot.)rnrrreversedr!rappend)r*gatesrvs r-rmzCircuitPlot._gatess dlC ( ( 'dl/00 $ $a&&$LLOOO $ d + + ' LL & & & r/ct|D]\}}|||dS)z0Iterate through the gates and plot each of them.N)rlrm plot_gate)r*rfgates r-r(zCircuitPlot._plot_gatessF // $ $GAt NN4 # # # # $ $r/ci}t|D];\}}t|ddr%|jD]}||vr|||kr|||<|||<<|S)zReturn a dict ``{i:j}`` where i is the index of the wire that has been measured, and j is the gate where the wire is measured. measurementF)rlrmgetattrrp)r*rsrfrvtargets r-rkzCircuitPlot._measurementss T[[]]++ / /CAaqu-- /i//F+/%f-1312Jv.-. 6**r/ch|jD]}|ddS)NF)rEfindobj set_clip_on)r*os r-r)zCircuitPlot._finishs>%%'' ! !A MM%  ! !r/c|j|}|j|}|j|||dddt ddd|j|jdS)z#Draw a box for a single qubit gate.rOrSr<TecfcfillrR)rQrUrVbboxrTN)r8r7rKr^dictrYra)r*tgate_idxwire_idxxys r- one_qubit_boxzCircuitPlot.one_qubit_boxsl OH % OH %  q!4DNCCC      r/cVt|jt|jdS)z>> from sympy.physics.quantum.circuitplot import render_label >>> render_label('q0') '$\\left|q0\\right\\rangle$' >>> render_label('q0', {'q0':'0'}) '$\\left|q0\\right\\rangle=\\left|0\\right\\rangle$' z$\left|z\right\rangle=\left|z\right\rangle$z$\left|%s\right\rangle$)r])labelr\inits r-r`r`0sB 99U  D PPCH55$$$OO % --r/qc>fdtDS)aAutogenerate labels for wires of quantum circuits. Parameters ========== n : int number of qubits in the circuit. symbol : string A character string to precede all gate labels. E.g. 'q_0', 'q_1', etc. >>> from sympy.physics.quantum.circuitplot import labeller >>> labeller(2) ['q_1', 'q_0'] >>> labeller(3,'j') ['j_2', 'j_1', 'j_0'] c*g|]}d|z dz fzS)z%s_%dr@r).0rfnsymbols r- zlabeller..Os* 7 7 7Gvac!en $ 7 7 7r/)rW)rrs``r-rr>s)" 8 7 7 7 7eAhh 7 7 77r/ceZdZdZdZdZdZdS)rzMock-up of a z measurement gate. This is in circuitplot rather than gate.py because it's not a real gate, it just draws one. TM_zNrrrrr gate_namegate_name_latexrr/r-rrQ) KIOOOr/rceZdZdZdZdZdZdS)rzMock-up of an x measurement gate. This is in circuitplot rather than gate.py because it's not a real gate, it just draws one. TM_xNrrr/r-rr[rr/rceZdZddZdS)rNcP|s|}tj||dztf||dS)Nr)rr)r __new__r)mclname latexnames r-rzCreateOneQubitGate.__new__fs> I dVml_/3 !R!RTT Tr/N)rrrrrr/r-rres.TTTTTTr/rNc:|s|}t||fd}|S)z5Use a lexical closure to make a controlled gate. cPtt||Sr)r tuple)ctrlsr onequbitgates r-ControlledGatez#CreateCGate..ControlledGaters#U5\\,,v"6"6777r/)r)rrrrs @r-rrls@  %dI66L88888 r/)rr)#rtypingrrtDictsympy.core.mulrsympy.externalrsympy.physics.quantum.gaterrr r sympy.core.corer sympy.core.assumptionsr __all__r RuntimeErrorrrlinesrXpatchesrr rr`rrrrrrr/r-rs '&&&&&&&((((((HHHHHHHHHHHH%%%%%%444444   ]7 ]hZ 8 / '*'  F   $F   &F o o o o o o o o b - - -! . . . .8888&TTTTT*TTTr/