+ j.Rt^RIHt^RIHt^RIHt^RIHtH t H t H t .ROt ]!R 4t ]!R R R ./]3R 7t] 'dA]'d9]P t]P"P$t]P&P(t!RR4tRt/3RltRRlt!RR] 4t!RR] 4t!RR]4tRRltR#)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 CircuitPlotMzMxCreateOneQubitGatenumpy matplotlibfromlistpyplot) import_kwargscatchc]tRt^/t$RtRtRtRtRtRt Rt .t R] R&/t R] R &R tR tR tR tRtRtRtRtRtRtRtRtRtRtRtRtRtRtR#)r z$A class for managing a circuit plot.g?g4@g?g333333?z list[str]labelszdict[str, str]inits?c n\'d \'g \R4hWn\ VPP 4VnW nVPV4VP4VP4VP4VP4VP4R#)z"numpy or matplotlib not available.N)npr ImportErrorcircuitlenargsngatesnqubitsupdate _create_grid_create_figure _plot_wires _plot_gates_finish)selfcrkwargss&&&,b/home/luca/screen-checking/.venv/lib/python3.14/site-packages/sympy/physics/quantum/circuitplot.py__init__CircuitPlot.__init__<s|rBC C $,,++,   F     c <VPPV4R#)z'Load the kwargs into the instance dict.N)__dict__r)r%r's&&r(rCircuitPlot.updateIs V$r+c VPp\PRVPV,V\R7p\PRVP V,V\R7pW nW0nR#)zCreate the grid of wires.g)dtypeN)scalerarangerfloatr _wire_grid _gate_grid)r%r1 wire_grid gate_grids& r(r CircuitPlot._create_gridMsS IIc4<<#5uEIJ IIc4;;u#4e5II ##r+c j\PVPVP,VPVP,3RRR7VnVP P ^^^RR7pVP4RVP,pVPVP^,V, VPR,V,4VPVP^,V, VPR,V,4VPR4Wn R#) z"Create the main matplotlib figure.w)figsize facecolor edgecolorT)frameonrequalN)rfigurerr1r_figure add_subplot set_axis_offset_xlimr5set_ylimr4 set_aspect_axes)r%axoffsets& r(r!CircuitPlot._create_figureUs}}[[+T\\$**-DE%  \\ % % q!&  TZZ DOOA&/1Dv1MN DOOA&/1Dv1MN g r+c  *VP^,pVPR,pWP, W P,3p\VP4EFpVPV,VPV,3p\ W5RVP R7pVPPV4VP'gKq^pVPPVPV,4'dRpVPPV^,VP, V, V^,\VPV,VP4VPRRRR7EK! VP!4R#)z&Plot the wires of the circuit diagram.kcolorlwg?center)sizerOhavaNr@)r5r1rangerr4Line2D linewidthrHadd_linerrgettext label_buffer render_labelfontsize_plot_measured_wires)r%xstartxstopxdataiydatalineinit_label_buffers& r(r"CircuitPlot._plot_wiresgs$##**$ejj&89t||$A__Q');>D JJ   %{{{$%!::>>$++a.11t3D !HT.../@@q Q ;X  7% !!#r+c  VP4pVPR,pRpVFpVPW,,W P,3pVPV,V,VPV,V,3p\ WVRVP R7pVP PV4K \VP44EFwr\V \\34'gK$V PV P,p V Fp W9gK VPV,VPW,,8gK<\V 4\!V 43pVPV,V, VPV,V, 3p\ WVRVP R7pVP PV4K EK R#)g{Gz?rMrNNr@) _measurementsr5r1r4rVrWrHrX enumerate_gates isinstancerrcontrolstargetsminmax) r% ismeasuredr`dyimrarcrdrbgwireswires & r(r^ CircuitPlot._plot_measured_wires~sf'') # B__Z^4U::5EFE__R(+DOOB,?,BCE>>D JJ   %T[[]+CA!eV_-- QYY.!D)??1- @P0QQ #E CJ 6 $ 22 5tq7I"7L L%!"%#~~   ++D1",r+c b.p\VP\4'dR\VPP4F,p\V\ 4'gKVP V4K. V#\VP\ 4'dVP VP4V#)z/Create a list of all gates in the circuit plot.)rlrrreversedrrappend)r%gatesrts& r(rkCircuitPlot._gatessx dllC ( (dll//0a&&LLO1   d + + LL & r+c l\VP44FwrVPW4K R#)z0Iterate through the gates and plot each of them.N)rjrk plot_gate)r%rbgates& r(r#CircuitPlot._plot_gatess% /GA NN4 #0r+c /p\VP44FLwr#\VRR4'gKVPF"pWA9dW,V8dW!V&KKW!V&K$ KN V#)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)rjrkgetattrrn)r%rqrbrttargets& r(riCircuitPlot._measurementssm T[[]+CAqu--iiF+%-112v.2./6* (,r+c jVPP4FpVPR4K R#)FN)rBfindobj set_clip_on)r%os& r(r$CircuitPlot._finishs%%%'A MM% (r+c VPV,pVPV,pVPPWEVRRRRRRRRRRVP/VP R 7R #) z#Draw a box for a single qubit gate.rMrQecfcr:fillTrP)rOrSrTbboxrRN)r5r4rHrZrWr])r%tgate_idxwire_idxxys&&&& r( one_qubit_boxCircuitPlot.one_qubit_boxs` OOH % OOH %  !T3dDNNK  r+c Z\VP4\VP4R#)zL %$$$.2: $ !  " "  * r+c \W3/VB#)a%Draw the circuit diagram for the circuit with nqubits. Parameters ========== c : circuit The circuit to plot. Should be a product of Gate instances. nqubits : int The number of qubits to include in the circuit. Must be at least as big as the largest ``min_qubits`` of the gates. )r )r&rr's&&,r( circuit_plotr s q ,V ,,r+c\VPV4pV'd RV: RV: R2#RV,#)zSlightly more flexible way to render labels. >>> 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$)rY)labelrinits&& r(r\r\.s- 99U D tCH$OO % --r+cn\V4Uu.uFpRWV, ^, 3,NK up#uupi)aXAutogenerate 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'] z%s_%d)rU)nsymbolrbs&& r(labellerr<s2"/4Ah 7hGvc!en $ $h 77 7s!2c&]tRtRtRtRtRtRtRtR#)r iOz~Mock-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_zrN rrrrrr gate_namegate_name_latexrrr+r(r r O KIOr+c&]tRtRtRtRtRtRtRtR#)r iYzMock-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_xrNrrr+r(r r Yrr+c ]tRtRtRRltRtR#)r icNc PV'gTp\VR,\3RVRV/4#)rrr)typer)mclname latexnames&&&r(__new__CreateOneQubitGate.__new__ds.ID6ML? $ 19 =? ?r+rN)rrrrrrrr+r(r r cs ??r+Nc>aV'gTp\W4oV3RlpV#)z5Use a lexical closure to make a controlled gate. c:<\\V4S!V44#r)rtuple)ctrlsr onequbitgates&&r(ControlledGate#CreateCGate..ControlledGatepsU5\,v"677r+)r )rrrrs&& @r( CreateCGaterjs"  %d6L8 r+)r rrr r r r)qr)r __future__rsympy.core.mulrsympy.externalrsympy.physics.quantum.gaterrrr__all__r RuntimeErrorrrlinesrVpatchesrr rr\rr r rr rrr+r(rs #(HH 7 hZ 8 / *   F    $ $F    & &F o o b -! .8&??r+