U Zʹj.@sdZddlmZddlmZddlmZddlmZm Z m Z m Z dddd d d d gZ ed Z edddgiefdZe rerejZejjZejjZGdddZddZifddZdddZGdd d e ZGdd d e ZGdd d eZddd ZdS)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)Z import_kwargsZcatchc@seZdZUdZdZdZdZdZdZdZ gZ de d<iZ de d <d Z d d Zd dZddZddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+S),r z$A class for managing a circuit plot.g?g4@g?g333333?z list[str]labelszdict[str, str]inits?cKs`trtstd||_t|jj|_||_||| | | | | dS)Nz"numpy or matplotlib not available.)npr ImportErrorcircuitlenargsngatesnqubitsupdate _create_grid_create_figure _plot_wires _plot_gates_finish)selfcrkwargsr'`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/physics/quantum/circuitplot.py__init__<s zCircuitPlot.__init__cCs|j|dS)z'Load the kwargs into the instance dict.N)__dict__r)r$r&r'r'r(rIszCircuitPlot.updatecCsF|j}tjd|j||td}tjd|j||td}||_||_dS)zCreate the grid of wires.g)ZdtypeN)scalerZarangerfloatr _wire_grid _gate_grid)r$r+Z wire_gridZ gate_gridr'r'r(rMs zCircuitPlot._create_gridcCstj|j|j|j|jfddd|_|jjddddd}|d|j}||j d||j d|| |j d||j d|| d ||_ d S) z"Create the main matplotlib figure.w)ZfigsizeZ facecolorZ edgecolorT)ZframeonrrequalN)rZfigurerr+r_figureZ add_subplotZ set_axis_offZset_xlimr.Zset_ylimr-Z set_aspect_axes)r$Zaxoffsetr'r'r(r Us"    zCircuitPlot._create_figurec Cs|jd}|jd}||j||jf}t|jD]}|j||j|f}t||d|jd}|j||j r2d}|j |j |rd}|jj |d|j ||dt|j ||j |jddddq2|dS) z&Plot the wires of the circuit diagram.rr1kcolorlwg?center)sizer8havaN)r.r+rangerr-Line2D linewidthr4add_linerrgettext label_buffer render_labelfontsize_plot_measured_wires)r$ZxstartxstopxdataiydatalineZinit_label_bufferr'r'r(r!gs4   zCircuitPlot._plot_wiresc Cs|}|jd}d}|D]V}|j||||jf}|j|||j||f}t||d|jd}|j|qt| D]\}} t | t t fr~| j | j} | D]n} | |kr|j||j|| krt| t| f}|j|||j||f}t||d|jd}|j|qq~dS)Nr1g{Gz?r6r7) _measurementsr.r+r-r?r@r4rA enumerate_gates isinstancerrZcontrolstargetsminmax) r$ ismeasuredrHZdyZimrIrKrLrJgZwiresZwirer'r'r(rG~s:  z CircuitPlot._plot_measured_wirescCsTg}t|jtr8t|jjD]}t|tr||qnt|jtrP||j|S)z/Create a list of all gates in the circuit plot.)rPrrreversedrrappend)r$ZgatesrUr'r'r(rOs    zCircuitPlot._gatescCs&t|D]\}}|||q dS)z0Iterate through the gates and plot each of them.N)rNrOZ plot_gate)r$rJZgater'r'r(r"szCircuitPlot._plot_gatescCs\i}t|D]F\}}t|ddr|jD]*}||krL|||krT|||<q*|||<q*q|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)rNrOgetattrrQ)r$rTrJrUtargetr'r'r(rMs     zCircuitPlot._measurementscCs|jD]}|dq dS)NF)r3ZfindobjZ set_clip_on)r$or'r'r(r#szCircuitPlot._finishc CsB|j|}|j|}|jj|||dddddd|jd|jddS)z#Draw a box for a single qubit gate.r6r:r/Tecfcfillr9)r8r<r=Zbboxr;N)r.r-r4rCr@rF)r$tgate_idxwire_idxxyr'r'r( one_qubit_boxs  zCircuitPlot.one_qubit_boxcCst|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|%s\right\rangle=\left|%s\right\rangle$z$\left|%s\right\rangle$)rB)labelrinitr'r'r(rE.s  rEqcsfddtDS)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'] cs g|]}d|dfqS)z%s_%dr0r').0rJnsymbolr'r( Mszlabeller..)r>rr'rr(r <sc@seZdZdZdZdZdZdS)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. TZM_zNrvrwrxryrX gate_namegate_name_latexr'r'r'r(r Osc@seZdZdZdZdZdZdS)r zMock-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. TZM_xNrr'r'r'r(r Ysc@seZdZdddZdS)rNcCs |s|}t|dtf||dS)Nr)rr)typer)Zmclname latexnamer'r'r(__new__ds  zCreateOneQubitGate.__new__)N)rvrwrxrr'r'r'r(rcsNcs"|s|}t||fdd}|S)z5Use a lexical closure to make a controlled gate. cstt||S)N)rtuple)ZctrlsrZZ onequbitgater'r(ControlledGatepsz#CreateCGate..ControlledGate)r)rrrr'rr(rjs   )r})N)ry __future__rZsympy.core.mulrZsympy.externalrZsympy.physics.quantum.gaterrrr__all__r RuntimeErrorrrlinesr?Zpatchesrjr r rEr r r rrrr'r'r'r(s>    r