ó ~9­\c @ s%dZddlmZmZddlmZmZmZmZddl m Z ddl m Z m Z ddlmZddd d d d d ddddddg ZedƒZesÌdefd„ƒYZn ejZeddidgd6ƒZesdefd„ƒYZd2ZnEejZeedƒr=ejjZneedƒr[ejjZnd„Zd„Zd„Zd„Zd „Z d!„Z!d"„Z"d#„Z#d$„Z$d%„Z%d&„Z&d'„Z'd(„Z(d)„Z)d*„Z*d+„Z+d,„Z,d-„Z-d.„Z.d/„Z/d0„Z0d1„Z1d2S(3s<Utilities to deal with sympy.Matrix, numpy and scipy.sparse.iÿÿÿÿ(tprint_functiontdivision(t MatrixBasetItExprtInteger(trange(teyetzeros(t import_modulet numpy_ndarraytscipy_sparse_matrixtsympy_to_numpytsympy_to_scipy_sparsetnumpy_to_sympytscipy_sparse_to_sympytflatten_scalart matrix_daggertto_sympytto_numpytto_scipy_sparsetmatrix_tensor_productt matrix_zerostnumpycB seZRS((t__name__t __module__(((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyR stscipyt__import__kwargstsparsetfromlistcB seZRS((RR(((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyR &stbasecK s—tst‚n|jddƒ}t|tƒrItj|jƒd|ƒSt|tƒrƒ|jsv|j sv|t krƒt |ƒSnt d|ƒ‚dS(sBConvert a sympy Matrix/complex number to a numpy matrix or scalar.tdtypetcomplexs.Expected MatrixBase or complex scalar, got: %rN( tnpt ImportErrortgett isinstanceRtmatrixttolistRt is_Numbertis_NumberSymbolRR t TypeError(tmtoptionsR((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyR 4s  cK s¨t st rt‚n|jddƒ}t|tƒrZtjtj|jƒd|ƒƒSt|t ƒr”|j s‡|j s‡|t kr”t |ƒSntd|ƒ‚dS(sBConvert a sympy Matrix/complex number to a numpy matrix or scalar.RR s.Expected MatrixBase or complex scalar, got: %rN(R!RR"R#R$Rt csr_matrixR%R&RR'R(RR R)(R*R+R((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyR As " cK st|jƒƒS(s0Convert a scipy.sparse matrix to a sympy matrix.(Rttodense(R*R+((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyRNscK s t|ƒS(s)Convert a numpy matrix to a sympy matrix.(R(R*R+((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyRSscK slt|tƒr|St|tƒr,t|ƒSt|tƒrEt|ƒSt|tƒrX|Std|ƒ‚dS(s6Convert a numpy/scipy.sparse matrix to a sympy matrix.s1Expected sympy/numpy/scipy.sparse matrix, got: %rN(R$RR RR RRR)(R*R+((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyRXs  cK sw|jddƒ}t|ttfƒr7t|d|ƒSt|tƒrJ|St|tƒrc|jƒStd|ƒ‚dS(s6Convert a sympy/scipy.sparse matrix to a numpy matrix.RR s1Expected sympy/numpy/scipy.sparse matrix, got: %rN( R#R$RRR R R R-R)(R*R+R((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyRes cK s‰|jddƒ}t|ttfƒr7t|d|ƒSt|tƒrbtsUt‚ntj|ƒSt|t ƒru|St d|ƒ‚dS(s6Convert a sympy/numpy matrix to a scipy.sparse matrix.RR s1Expected sympy/numpy/scipy.sparse matrix, got: %rN( R#R$RRR R RR"R,R R)(R*R+R((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyRqs  cC slt|tƒr.|jdkr.|d}q.nt|ttfƒrh|jdkrht|dƒ}qhn|S(sCFlatten a 1x1 matrix to a scalar, return larger matrices unchanged.ii(ii(ii(ii(R$RtshapeR R R (te((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyRscC sOt|tƒr|jSt|ttfƒr;|jƒjƒStd|ƒ‚dS(s7Return the dagger of a sympy/numpy/scipy.sparse matrix.s1Expected sympy/numpy/scipy.sparse matrix, got: %rN(R$RtHR R t conjugatet transposeR)(R/((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyRŠs cG sddlm}||ŒS(sCCompute the kronecker product of a sequence of sympy Matrices. iÿÿÿÿ(tmatrix_kronecker_product(t$sympy.matrices.expressions.kroneckerR3(tmatricesR3((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt_sympy_tensor_product”scG sDtst‚n|d}x$|dD]}tj||ƒ}q$W|S(s6numpy version of tensor product of multiple arguments.ii(R!R"tkron(tproducttanswertitem((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt_numpy_tensor_productœs   cG sMtst‚n|d}x$|dD]}tj||ƒ}q$Wtj|ƒS(s=scipy.sparse version of tensor product of multiple arguments.ii(RR"R7R,(R8R9R:((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt_scipy_sparse_tensor_product¦s   cG s[t|dtƒrt|ŒSt|dtƒr:t|ŒSt|dtƒrWt|ŒSdS(sGCompute the matrix tensor product of sympy/numpy/scipy.sparse matrices.iN(R$RR6R R;R R<(R8((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyR²s   cC s+tst‚ntjtj|ddƒƒS(snumpy version of complex eye.RR (R!R"R%R(tn((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt _numpy_eye¼s cC s%tst‚ntj||ddƒS(s$scipy.sparse version of complex eye.RR (RR"R(R=((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt_scipy_sparse_eyeÃs cK sh|jddƒ}|dkr(t|ƒS|dkr>t|ƒS|dkrTt|ƒStd|ƒ‚dS(s=Get the version of eye and tensor_product for a given format.tformattsympyRs scipy.sparsesInvalid format: %rN(R#RR>R?tNotImplementedError(R=R+R@((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt matrix_eyeÊs      cK s:|jddƒ}ts!t‚ntj||fd|ƒS(snumpy version of zeros.Rtfloat64(R#R!R"R(R*R=R+R((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt _numpy_zerosÖs cK s|jddƒ}|jddƒ}ts3t‚n|dkrXtj||fd|ƒS|dkr}tj||fd|ƒSdS(sscipy.sparse version of zeros.tspmatrixtcsrRRDtlilN(R#RR"t lil_matrixR,(R*R=R+RFR((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt_scipy_sparse_zerosÞs   cK s›|jddƒ}|jddƒ}|jddƒ}|dkrOt||ƒS|dkrkt|||S|dkr‡t|||Std |ƒ‚d S( s'"Get a zeros matrix for a given format.R@RARRDRFRGRs scipy.sparsesInvaild format: %rN(R#RRERJRB(R*R=R+R@RRF((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyRês    cC s<tst‚ntj|ƒ}tj||ƒr4dS|SdS(s/Convert a numpy zero matrix to the zero scalar.gN(R!R"t zeros_liketallclose(R/ttest((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt_numpy_matrix_to_zeroøs  cC sHtst‚n|jƒ}tj|ƒ}tj||ƒr@dS|SdS(s6Convert a scipy.sparse zero matrix to the zero scalar.gN(R!R"R-RKRL(R/tedenseRM((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pyt_scipy_sparse_matrix_to_zeros  cC svt|tƒr6t|jŒ|krrtdƒ}qrn<t|tƒrTt|ƒ}nt|tƒrrt|ƒ}n|S(s)Convert a zero matrix to the scalar zero.i( R$RRR.RR RNR RP(R/((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pytmatrix_to_zerosN(2t__doc__t __future__RRRARRRRtsympy.core.compatibilityRtsympy.matricesRRtsympy.externalR t__all__R!tobjectR tndarrayRR tNoneRthasattrRRFR R RRRRRRRR6R;R<RR>R?RCRERJRRNRPRQ(((s@lib/python2.7/site-packages/sympy/physics/quantum/matrixutils.pytsj"