\h.|SrSSKJr SSKJr SSKJr SSKJrJ r J r J r /SQr \"S5r \"SS S /0\4S 9r\ (a?\(a8\R r\R"R$r\R&R(r"S S 5rSr04SjrSSjr"SS\ 5r"SS\ 5r"SS\5rSSjrg)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_kwargscatchc\rSrSr%SrSrSrSrSrSr Sr /r S\ S'0r S \ S 'S rS rS rSrSrSrSrSrSrSrSrSrSrSrSrSrSrSrg)r /z$A class for managing a circuit plot.g?g4@g?g333333?z list[str]labelszdict[str, str]inits?c h[(a [(d [S5eXl[ URR 5UlX lURU5 UR5 UR5 UR5 UR5 UR5 g)Nz"numpy or matplotlib not available.)npr ImportErrorcircuitlenargsngatesnqubitsupdate _create_grid_create_figure _plot_wires _plot_gates_finish)selfcr#kwargss Y/var/www/auris/envauris/lib/python3.13/site-packages/sympy/physics/quantum/circuitplot.py__init__CircuitPlot.__init__<s|rBC C $,,++,   F     c:URRU5 g)z'Load the kwargs into the instance dict.N)__dict__r$)r*r,s r-r$CircuitPlot.updateIs V$r0cURn[RSURU-U[S9n[RSUR U-U[S9nX lX0lg)zCreate the grid of wires.g)dtypeN)scaleraranger#floatr" _wire_grid _gate_grid)r*r6 wire_grid gate_grids r-r%CircuitPlot._create_gridMsS IIc4<<#5uEIJ IIc4;;u#4e5II ##r0c[RURUR-URUR-4SSS9UlUR R SSSSS9nUR5 SUR-nURURSU- URSU-5 URURSU- URSU-5 URS 5 Xl g ) z"Create the main matplotlib figure.w)figsize facecolor edgecolorT)frameonrrequalN)rfigurer"r6r#_figure add_subplot set_axis_offset_xlimr:set_ylimr9 set_aspect_axes)r*axoffsets r-r&CircuitPlot._create_figureUs}}[[+T\\$**-DE%  \\ % % q!&  TZZ DOOA&/1Dv1MN DOOA&/1Dv1MN g r0c URSnURSnXR- X R-4n[UR5HnURUURU4n[ X5SUR S9nURRU5 UR(dMdSnURRURU5(aSnURRUSUR- U- US[URUUR5URSSSS9 M UR!5 g) z&Plot the wires of the circuit diagram.rrEkcolorlwg?center)sizerUhavaN)r:r6ranger#r9Line2D linewidthrNadd_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% !!#r0c *UR5nURSnSnUHvnURXX R-4nURUU-URUU-4n[ XVSUR S9nUR RU5 Mx [UR55Hup[U [[45(dM"U RU R-n U Hn X;dM URUURX:dM.[U 5[!U 54nURUU- URUU- 4n[ XVSUR S9nUR RU5 M M g)NrEg{Gz?rSrT) _measurementsr:r6r9r\r]rNr^ enumerate_gates isinstancerr controlstargetsminmax) r* ismeasuredrfdyimrgrirjrhgwireswires r-rd 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",r0c\/n[UR[5(aP[URR5H+n[U[ 5(dMUR U5 M- U$[UR[ 5(aUR UR5 U$)z/Create a list of all gates in the circuit plot.)rqrrreversedr!rappend)r*gatesrys r-rpCircuitPlot._gatessx dllC ( (dll//0a&&LLO1   d + + LL & r0cj[UR55HupURX5 M g)z0Iterate through the gates and plot each of them.N)rorp plot_gate)r*rhgates r-r(CircuitPlot._plot_gatess% /GA NN4 #0r0c0n[UR55HEup#[USS5(dMURHnXA;aXU:aX!U'MMX!U'M MG U$)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)rorpgetattrrs)r*rvrhrytargets r-rnCircuitPlot._measurementssm T[[]+CAqu--iiF+%-112v.2./6* (,r0chURR5HnURS5 M g)NF)rHfindobj set_clip_on)r*os r-r)CircuitPlot._finishs%%%'A MM% (r0c URUnURUnURRXEUSSSSSSURS.UR S9 g)z#Draw a box for a single qubit gate.rSrWr?TecfcfillrV)rUrYrZbboxrXN)r:r9rNr`r]rc)r*tgate_idxwire_idxxys r- one_qubit_boxCircuitPlot.one_qubit_boxsX OOH % OOH %  !3DNNK  r0cX[UR5 [UR5 g)zL %$$$.2: $ !  " "  * r0r c [X40UD6$)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+r#r,s r-r r s q ,V ,,r0cRURU5nU(a SU<SU<S3$SU-$)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$)r_)labelrinits r-rbrb.s+ 99U D CH$OO % --r0cX[U5Vs/sHnSXU- S- 4-PM sn$s snf)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_%drC)r[)nsymbolrhs r-r r <s/"/4Ah 7hGvc!en $h 77 7s'c$\rSrSrSrSrSrSrSrg)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_latexrrr0r-r r O KIOr0r c$\rSrSrSrSrSrSrSrg)riYzMock-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_xrNrrr0r-rrYrr0rc\rSrSrSSjrSrg)ricNcBU(dUn[US-[4XS.5$)Nr)rr)typer)mclname latexnames r-__new__CreateOneQubitGate.__new__ds'ID6ML? =? ?r0rN)rrrrrrrr0r-rrcs?r0rNc<^U(dUn[X5mU4SjnU$)z5Use a lexical closure to make a controlled gate. c:>[[U5T"U55$r)rtuple)ctrlsr onequbitgates r-ControlledGate#CreateCGate..ControlledGatepsU5\,v"677r0)r)rrrrs @r-rrjs"  %d6L8 r0)qr)r __future__rsympy.core.mulrsympy.externalrsympy.physics.quantum.gaterrrr __all__r RuntimeErrorrrlinesr\patchesrr r rbr r rrrrrr0r-rs #(HH 7 hZ 8 / *   F    $ $F    & &F o o b -! .8&??r0