o jg6@sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZmZmZmZd d gZGd d d eZd dZddZGdd d eZddZddZddZdS))Expr)Symbol)sympify)Matrix) prettyForm)Tuple) is_sequence)Dagger) numpy_ndarrayscipy_sparse_matrixto_sympyto_numpyto_scipy_sparse QuantumErrorQExprc@s eZdZdS)rN)__name__ __module__ __qualname__rrS/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/physics/quantum/qexpr.pyrscCs tt|S)aConvert elements of a sequence to standard form. This is like sympify, but it performs special logic for arguments passed to QExpr. The following conversions are done: * (list, tuple, Tuple) => _qsympify_sequence each element and convert sequence to a Tuple. * basestring => Symbol * Matrix => Matrix * other => sympify Strings are passed to Symbol, not sympify to make sure that variables like 'pi' are kept as Symbols, not the SymPy built-in number subclasses. Examples ======== >>> from sympy.physics.quantum.qexpr import _qsympify_sequence >>> _qsympify_sequence((1,2,[3,4,[1,]])) (1, 2, (3, 4, (1,))) )tuple__qsympify_sequence_helper)seqrrr_qsympify_sequences rcCsTt|st|tr |St|trt|St|St|tr|Sdd|D}t|S)z^ Helper function for _qsympify_sequence This function does the actual work. cSsg|]}t|qSr)r).0itemrrr Lsz.__qsympify_sequence_helper..)r isinstancerstrrrrr)rresultrrrr8s   rc@s eZdZdZdZdZdZeddZddZ e d d Z ed d Z ed dZ e ddZddZe ddZe ddZddZddZddZddZdAd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Z d7d8Z!d9d:Z"d;d<d=d>Z#d?d@Z$d;S)Brz>A base class for all quantum object like operators and states.) hilbert_spaceFcCs|hSNrselfrrr free_symbolsdszQExpr.free_symbolscOsZ|j|fi|}t|dkr|jt|fi|}tj|g|R}|||_|S)a:Construct a new quantum object. Parameters ========== args : tuple The list of numbers or parameters that uniquely specify the quantum object. For a state, this will be its symbol or its set of quantum numbers. Examples ======== >>> from sympy.physics.quantum.qexpr import QExpr >>> q = QExpr(0) >>> q 0 >>> q.label (0,) >>> q.hilbert_space H >>> q.args (0,) >>> q.is_commutative False r) _eval_argslenr default_argsr__new___eval_hilbert_spacer )clsargskwargsinstrrrr)hs   z QExpr.__new__cOs"tj|g|Ri|}||_|S)aCreate new instance of this class with hilbert_space and args. This is used to bypass the more complex logic in the ``__new__`` method in cases where you already have the exact ``hilbert_space`` and ``args``. This should be used when you are positive these arguments are valid, in their final, proper form and want to optimize the creation of the object. )rr)r )r+r r,old_assumptionsobjrrr _new_rawargss zQExpr._new_rawargscCs&t|jdkr|t|S|jS)aThe label is the unique set of identifiers for the object. Usually, this will include all of the information about the state *except* the time (in the case of time-dependent objects). This must be a tuple, rather than a Tuple. r)r'r,r&listr(r#rrrlabels z QExpr.labelcCsdS)NTrr#rrr is_symbolicszQExpr.is_symboliccCtd)a6If no arguments are specified, then this will return a default set of arguments to be run through the constructor. NOTE: Any classes that override this MUST return a tuple of arguments. Should be overridden by subclasses to specify the default arguments for kets and operators z$No default arguments for this class!NotImplementedErrorr#rrrr(szQExpr.default_argscCs4t|}|durtt|}t|tr|j|_|Sr")r _eval_adjointr)r rrr )r$r0rrrr8s   zQExpr._eval_adjointcCst|S)zrProcess the args passed to the __new__ method. This simply runs args through _qsympify_sequence. )r)r+r,rrrr&szQExpr._eval_argscCsddlm}|S)z:Compute the Hilbert space instance from the args. r) HilbertSpace)sympy.physics.quantum.hilbertr9)r+r,r9rrrr*s zQExpr._eval_hilbert_spacecGs0g}|D]}||j|g|Rq||Sr")append_printjoin)r$rsepprinterr,rrrrr_print_sequences zQExpr._print_sequencecGsV|j|dg|R}|ddD]}t||}t||j|g|R}q|S)Nr)r<rright)r$rr>r?r,pformrrrr_print_sequence_prettys zQExpr._print_sequence_prettycCsBt|d|}t|d|}t||dtjiS)N binding)rleftwidthrBbelowPOW)r$abtopbotrrr_print_subscript_prettyszQExpr._print_subscript_prettycCs||Sr"r)r$rKrLrrr_print_superscript_prettyzQExpr._print_superscript_pretty()cCst|j||dS)N)rGrB)rparens)r$rCrGrBrrr_print_parens_prettyzQExpr._print_parens_prettycG|j|j|j|g|RS)zPrints the label of the QExpr This method prints self.label, using self._label_separator to separate the elements. This method should not be overridden, instead, override _print_contents to change printing behavior. r@r3_label_separatorr$r?r,rrr _print_labels  zQExpr._print_labelcGs|j|jd|g|RS)N,)r@r3rZrrr_print_label_reprs zQExpr._print_label_reprcGrWr")rDr3rYrZrrr_print_label_pretty   zQExpr._print_label_prettycGrWr"rXrZrrr_print_label_latexr_zQExpr._print_label_latexcG|j|g|RS)aSPrinter for contents of QExpr Handles the printing of any unique identifying contents of a QExpr to print as its contents, such as any variables or quantum numbers. The default is to print the label, which is almost always the args. This should not include printing of any brackets or parentheses. )r[rZrrr_print_contentsszQExpr._print_contentscGrar")r^rZrrr_print_contents_prettyrVzQExpr._print_contents_prettycGrar")r`rZrrr_print_contents_latex"rVzQExpr._print_contents_latexcGra)aDefault printing behavior of QExpr objects Handles the default printing of a QExpr. To add other things to the printing of the object, such as an operator name to operators or brackets to states, the class should override the _print/_pretty/_latex functions directly and make calls to _print_contents where appropriate. This allows things like InnerProduct to easily control its printing the printing of contents. )rbrZrrr _sympystr's zQExpr._sympystrcGs&|jj}|j|g|R}d||fS)Nz%s(%s)) __class__rr])r$r?r, classnamer3rrr _sympyrepr3s zQExpr._sympyreprcGs|j|g|R}|Sr")rc)r$r?r,rCrrr_pretty8sz QExpr._prettycGrar")rdrZrrr_latex<rVz QExpr._latexcKr5)Nz)This object does not have a default basisr6)r$optionsrrr_represent_default_basisCrQzQExpr._represent_default_basisN)basiscKsJ|dur |jdi|}n t|d|fi|}|dd}|||}|S)a`Represent this object in a given basis. This method dispatches to the actual methods that perform the representation. Subclases of QExpr should define various methods to determine how the object will be represented in various bases. The format of these methods is:: def _represent_BasisName(self, basis, **options): Thus to define how a quantum object is represented in the basis of the operator Position, you would define:: def _represent_Position(self, basis, **options): Usually, basis object will be instances of Operator subclasses, but there is a chance we will relax this in the future to accommodate other types of basis sets that are not associated with an operator. If the ``format`` option is given it can be ("sympy", "numpy", "scipy.sparse"). This will ensure that any matrices that result from representing the object are returned in the appropriate matrix format. Parameters ========== basis : Operator The Operator whose basis functions will be used as the basis for representation. options : dict A dictionary of key/value pairs that give options and hints for the representation, such as the number of basis functions to be used. N _representformatsympyr)rldispatch_methodget_format_represent)r$rmrkrrorrrrnFs "  zQExpr._representcCsR|dkr t|ts t|S|dkrt|tst|S|dkr't|ts't|S|S)Nrpnumpyz scipy.sparse)rrr r r r r)r$rrorrrrsrszQExpr._format_represent)rRrS)%rrr__doc__ __slots__is_commutativerYpropertyr%r) classmethodr1r3r4r(r8r&r*r@rDrOrPrUr[r]r^r`rbrcrdrerhrirjrlrnrsrrrrrUsP %              ,cCs|\}}t|}||fS)z1Split into commutative and non-commutative parts.)args_cncr2)ec_partnc_partrrrsplit_commutative_parts~s r~cCs<g}g}|jD]}t|ts||q||q||fS)z=Split an expression into Expr and noncommutative QExpr parts.)r,rrr;)r{ expr_part qexpr_partargrrrsplit_qexpr_partss    rcKsVd||jjf}t||r t||}||fi|}|dur |Std|jj||f)z)Dispatch a method to the proper handlers.z%s_%sNz%s.%s cannot handle: %r)rfrhasattrgetattrr7)r$basenamerrk method_namefrrrrrqs   rqN)sympy.core.exprrsympy.core.symbolrsympy.core.sympifyrsympy.matrices.denser sympy.printing.pretty.stringpictrsympy.core.containersrsympy.utilities.iterablesrsympy.physics.quantum.daggerr !sympy.physics.quantum.matrixutilsr r r r r__all__ Exceptionrrrrr~rrqrrrrs*        +