ó ¡¼™\c@ s´dZddlmZmZddlmZddlmZddlm Z ddl m Z m Z m Z mZddlmZddlmZd d d d d ddgZe dƒZe ddidgd6defƒZe sèe r d efd„ƒYZd„Zn@ejZejjZejjZd efd„ƒYZd„Zid„Zdd„Z d e fd„ƒYZ!d e fd„ƒYZ"defd„ƒYZ#d!d „Z%d!S("sÉMatplotlib 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! iÿÿÿÿ(tprint_functiontdivision(tMul(trange(t import_module(tGatet OneQubitGatetCGatetCGateS(t BasicMeta(tManagedPropertiest CircuitPlott circuit_plottlabellertMztMxtCreateOneQubitGatet CreateCGatetnumpyt matplotlibt__import__kwargstpyplottfromlisttcatchcB seZd„ZRS(cO stdƒ‚dS(Ns"numpy or matplotlib not available.(t ImportError(targstkwargs((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyt__init__,s(t__name__t __module__R(((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR +scO stdƒ‚dS(Ns"numpy or matplotlib not available.(R(RR((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR /scB sÔeZdZdZdZdZdZdZdZgZ iZ dZ d„Z d„Z d„Zd „Zd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„ZRS(s$A class for managing a circuit plot.gð?g4@gš™™™™™©?g333333Ã?gà?cK sj||_t|jjƒ|_||_|j|ƒ|jƒ|jƒ|jƒ|j ƒ|j ƒdS(N( tcircuittlenRtngatestnqubitstupdatet _create_gridt_create_figuret _plot_wirest _plot_gatest_finish(tselftcR!R((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyRFs       cC s|jj|ƒdS(s'Load the kwargs into the instance dict.N(t__dict__R"(R(R((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR"QscC sc|j}tjd|j||dtƒ}tjd|j||dtƒ}||_||_dS(sCreate the grid of wires.gtdtypeN(tscaletnptarangeR!tfloatR t _wire_gridt _gate_grid(R(R,t wire_gridt gate_grid((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR#Us  "" cC sÖtjd|j|j|j|jfddddƒ|_|jjddddtƒ}|jƒd|j}|j |j d||j d |ƒ|j |j d||j d |ƒ|j d ƒ||_d S( s"Create the main matplotlib figure.tfigsizet facecolortwt edgecoloritframeongà?iiÿÿÿÿtequalN(RtfigureR R,R!t_figuret add_subplottTruet set_axis_offtset_xlimR1tset_ylimR0t set_aspectt_axes(R(taxtoffset((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR$]s        && c C s/|jd}|jd}||j||jf}xêt|jƒD]Ù}|j||j|f}t||ddd|jƒ}|jj|ƒ|j rDd}|j j |j |ƒrÃd}n|jj |d|j ||dt|j ||j ƒd|jdddd d d ƒqDqDW|jƒd S( s&Plot the wires of the circuit diagram.iiÿÿÿÿtcolortktlwgÐ?tsizethatcentertvaN(R1R,RR!R0tLine2Dt linewidthRBtadd_linetlabelstinitstgetttextt label_buffert render_labeltfontsizet_plot_measured_wires(R(txstarttxstoptxdatatitydatatlinetinit_label_buffer((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR%os(        c C sœ|jƒ}|jd}d}x||D]t}|j||||jf}|j|||j||f}t||ddd|jƒ}|jj|ƒq&Wx÷t|j ƒƒD]ã\}} t | t ƒsÛt | t ƒr±| j | j} x¦| D]›} | |krò|j||j|| kròt| ƒt| ƒf}|j|||j||f}t||ddd|jƒ}|jj|ƒqòqòWq±q±WdS(Niÿÿÿÿg{®Gáz¤?RERFRG(t _measurementsR1R,R0RLRMRBRNt enumeratet_gatest isinstanceRRtcontrolsttargetstmintmax( R(t ismeasuredRXtdytimRYR[R\RZtgtwirestwire((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyRV†s0   "    "  cC s€g}t|jtƒrWxat|jjƒD]%}t|tƒr+|j|ƒq+q+Wn%t|jtƒr||j|jƒn|S(s/Create a list of all gates in the circuit plot.(RaRRtreversedRRtappend(R(tgatesRi((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR`£scC s7x0t|jƒƒD]\}}|j||ƒqWdS(s0Iterate through the gates and plot each of them.N(R_R`t plot_gate(R(RZtgate((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR&®scC sŒi}xt|jƒƒD]k\}}t|dtƒrxJ|jD]<}||krs|||kr}|||>> 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$' s-$\left|%s\right\rangle=\left|%s\right\rangle$s$\left|%s\right\rangle$(RQ(tlabelRPtinit((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyRT5s tqcC s/gt|ƒD]}d|||df^q S(s‚Autogenerate 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'] s%s_%di(R(tntsymbolRZ((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR CscB s eZdZeZdZdZRS(sŠ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. RuM_z(RRR–R=Rqt gate_nametgate_name_latex(((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyRUscB s eZdZeZdZdZRS(s‹Mock-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. RuM_x(RRR–R=RqRœR(((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyR_scB seZdd„ZRS(cC s:|s|}ntj||dtfi|d6|d6ƒS(NRRœR(R t__new__R(tmcltnamet latexname((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyRžjs N(RRtNoneRž(((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyRisc s1|s|}nt||ƒ‰‡fd†}|S(s5Use a lexical closure to make a controlled gate. c stt|ƒˆ|ƒƒS(N(Rttuple(tctrlsRt(t onequbitgate(s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pytControlledGatevs(R(R R¡R¦((R¥s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyRps  N(&R–t __future__RRtsympyRtsympy.core.compatibilityRtsympy.externalRtsympy.physics.quantum.gateRRRRtsympy.core.coreR tsympy.core.assumptionsR t__all__R-t RuntimeErrorRtobjectR R RtlinesRLtpatchesRŠRTR RRRR¢R(((s@lib/python2.7/site-packages/sympy/physics/quantum/circuitplot.pyts@"       î