g.\dZddlmZddlmZddlmZddlmZm Z m Z m Z gdZ edZ edd d gief Ze r!erejZejjZejjZGd d ZdZifdZddZGdde ZGdde ZGddeZddZdS)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! ) annotations)Mul) import_module)Gate OneQubitGateCGateCGateS) CircuitPlot circuit_plotlabellerMzMxCreateOneQubitGate CreateCGatenumpy matplotlibfromlistpyplot) import_kwargscatchceZdZUdZdZdZdZdZdZdZ gZ de d<iZ de d <d Z d Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdS)r z$A class for managing a circuit plot.g?g4@g?g333333?z list[str]labelszdict[str, str]inits?c trtstd||_t |jj|_||_||| | | | | dS)Nz"numpy or matplotlib not available.)npr ImportErrorcircuitlenargsngatesnqubitsupdate _create_grid_create_figure _plot_wires _plot_gates_finish)selfcr"kwargss ]/var/www/html/ai-engine/env/lib/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.updateIs 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)r3 wire_grid gate_grids r,r$zCircuitPlot._create_gridMs[ 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!r3r"_figure add_subplot set_axis_offset_xlimr7set_ylimr6 set_aspect_axes)r)axoffsets r,r%zCircuitPlot._create_figureUs}}[+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.rrAkcolorlwg?center)sizerPhavaN)r7r3ranger"r6Line2D linewidthrJadd_linerrgettext label_buffer render_labelfontsize_plot_measured_wires)r)xstartxstopxdataiydatalineinit_label_buffers r,r&zCircuitPlot._plot_wiresgs?##$*$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)NrAg{Gz?rNrO) _measurementsr7r3r6rWrXrJrY enumerate_gates isinstancerr controlstargetsminmax) r) ismeasuredradyimrbrdrercgwireswires r,r_z CircuitPlot._plot_measured_wires~s'')) #  & &B_Z^4U4:5EFE_R(+DOB,?,BCEu>D J   % % % %T[[]]++ 2 2CAa!eV_-- 2 QY.! 2 2Dz))?1- 4@P0QQQ #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.)rkrrreversedr rappend)r)gatesrss r,rjzCircuitPlot._gatess dlC ( ( 'dl/00 $ $a&&$LLOOO $ d + + ' LL & & & r.ct|D]\}}|||dS)z0Iterate through the gates and plot each of them.N)rirj plot_gate)r)rcgates 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)rirjgetattrrm)r)rprcrstargets r,rhzCircuitPlot._measurementss T[[]]++ / /CAaqu-- /i//F++%f-1112Jv.-. 6**r.ch|jD]}|ddS)NF)rDfindobj set_clip_on)r)os r,r(zCircuitPlot._finishs>%%'' ! !A MM%  ! !r.c |j|}|j|}|j|||dddddd|jd|jdS)z#Draw a box for a single qubit gate.rNrRr;TecfcfillrQ)rPrTrUbboxrSN)r7r6rJr[rXr^)r)tgate_idxwire_idxxys r, one_qubit_boxzCircuitPlot.one_qubit_boxsg OH % OH %  q!3DNKK      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$)rZ)labelrinits r,r]r].sB 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).0rcnsymbols r, zlabeller..Ms* 7 7 7Gvac!en $ 7 7 7r.)rV)rrs``r,r r <s)" 8 7 7 7 7eAhh 7 7 77r.ceZdZdZdZdZdZdS)r zMock-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,r r O) KIOOOr.r ceZdZdZdZdZdZdS)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,rrYrr.rceZdZddZdS)rNcD|s|}t|dztf||dS)Nr)rr)typer)mclname latexnames r,__new__zCreateOneQubitGate.__new__ds6 ID6ML?9 = =?? ?r.N)rrrrrr.r,rrcs(??????r.rNc:|s|}t||fd}|S)z5Use a lexical closure to make a controlled gate. cPtt||Sr)rtuple)ctrlsr onequbitgates r,ControlledGatez#CreateCGate..ControlledGateps#U5\\,,v"6"6777r.)r)rrrrs @r,rrjs@  %dI66L88888 r.)rr)r __future__rsympy.core.mulrsympy.externalrsympy.physics.quantum.gaterrrr __all__r RuntimeErrorrrlinesrWpatchesrr r r]r r rrrrrr.r,rs #"""""((((((HHHHHHHHHHHH   ]7 ]hZ 8 / '*'  F   $F   &F o o o o o o o o b - - -! . . . .8888&????????r.