o *ήc.@s dZddlmZmZddlmZddlmZddl m Z m Z m Z m Z ddlmZddlmZgdZed Zed d d giefd ZerPerPejZejjZejjZGdddZddZifddZd ddZGddde ZGddde Z GdddeZ!d!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_kwargscatchc@seZdZdZdZdZdZdZdZdZ gZ iZ dZ ddZ d d Zd d Zd dZddZddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'S)(r z$A class for managing a circuit plot.g?g4@g?g333333??cKs`trtstd||_t|jj|_||_||| | | | | dS)Nz"numpy or matplotlib not available.)npr ImportErrorcircuitlenargsngatesnqubitsupdate _create_grid_create_figure _plot_wires _plot_gates_finish)selfcr kwargsr*H/tmp/pip-target-vg8gfxp4/lib/python/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+r!KszCircuitPlot.updatecCsF|j}tjd|j||td}tjd|j||td}||_||_dS)zCreate the grid of wires.g)dtypeN)scaleraranger floatr _wire_grid _gate_grid)r'r/ wire_grid gate_gridr*r*r+r"Os  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)figsize facecolor edgecolorT)frameonrrequalN)rfigurerr/r _figure add_subplot set_axis_offset_xlimr3set_ylimr2 set_aspect_axes)r'axoffsetr*r*r+r#Ws     zCircuitPlot._create_figurec Cs|jd}|jd}||j||jf}t|jD]I}|j||j|f}t||d|jd}|j||j rbd}|j |j |rDd}|jj |d|j ||dt|j ||j |jddddq|dS) z&Plot the wires of the circuit diagram.rr<kcolorlwg?center)sizerJhavaN)r3r/ranger r2Line2D linewidthrEadd_linelabelsinitsgettext label_buffer render_labelfontsize_plot_measured_wires)r'xstartxstopxdataiydatalineinit_label_bufferr*r*r+r$is,    zCircuitPlot._plot_wiresc Cs|}|jd}d}|D]+}|j||||jf}|j|||j||f}t||d|jd}|j|q t| D]K\}} t | t t fr| j | j} | D]7} | |vr|j||j|| krt| t| f}|j|||j||f}t||d|jd}|j|qRq?dS)Nr<g{Gz?rHrI) _measurementsr3r/r2rQrRrErS enumerate_gates isinstancerr controlstargetsminmax) r' ismeasuredr]dyimr^r`rar_gwireswirer*r*r+r[s:   z CircuitPlot._plot_measured_wirescCsVg}t|jtrt|jjD] }t|tr||q|St|jtr)||j|S)z/Create a list of all gates in the circuit plot.)rfrrreversedrrappend)r'gatesrnr*r*r+res    zCircuitPlot._gatescCs&t|D] \}}|||qdS)z0Iterate through the gates and plot each of them.N)rdre plot_gate)r'r_gater*r*r+r%szCircuitPlot._plot_gatescCs\i}t|D]#\}}t|ddr+|jD]}||vr&|||kr%|||<q|||<qq|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)rdregetattrrh)r'rkr_rntargetr*r*r+rcs    zCircuitPlot._measurementscCs|jD]}|dqdS)NF)r?findobj set_clip_on)r'or*r*r+r&s zCircuitPlot._finishc CsD|j|}|j|}|jj|||dddtddd|jd|jddS)z#Draw a box for a single qubit gate.rHrLr6TecfcfillrK)rJrNrObboxrMN)r3r2rErWdictrRrZ)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$)rV)labelrUinitr*r*r+rY0s  rYqcsfddtDS)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_%dr:r*).0r_nsymbolr*r+ Os zlabeller..)rPrr*rr+r>src@eZdZdZdZdZdZdS)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_zNrrrrrv gate_namegate_name_latexr*r*r*r+rQ rc@r)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_xNrr*r*r*r+r[rrc@seZdZdddZdS)rNcCs$|s|}t||dtf||dS)Nr)rr)r __new__r)mclname latexnamer*r*r+rfs zCreateOneQubitGate.__new__N)rrrrr*r*r*r+resrNcs"|s|}t||fdd}|S)z5Use a lexical closure to make a controlled gate. cstt||Sr)rtuple)ctrlsrx onequbitgater*r+ControlledGatersz#CreateCGate..ControlledGate)r)rrrr*rr+rls   r)rr)#rtypingrrtDictsympy.core.mulrsympy.externalrsympy.physics.quantum.gaterrrr sympy.core.corer sympy.core.assumptionsr __all__r RuntimeErrorrrlinesrQpatchesrr r rYrrrrrr*r*r*r+s4      r