U Zʹj[%@sdZddlmZmZmZmZmZmZmZddl m Z ddl m Z m Z mZddlmZmZmZmZmZmZmZddgZeeeefeeeefeeeefee eeeeefeeeeiZdd eDZd dZd dZd d ZddZddZ ddZ!dS)a A module for mapping operators to their corresponding eigenstates and vice versa It contains a global dictionary with eigenstate-operator pairings. If a new state-operator pair is created, this dictionary should be updated as well. It also contains functions operators_to_state and state_to_operators for mapping between the two. These can handle both classes and instances of operators and states. See the individual function descriptions for details. TODO List: - Update the dictionary with a complete list of state-operator pairs )XOpYOpZOpXKetPxOpPxKet PositionKet3D)Operator) StateBaseBraBaseKet)JxOpJyOpJzOpJ2OpJxKetJyKetJzKetoperators_to_statestate_to_operatorscCsi|]\}}||qSr).0kvrr`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/physics/quantum/operatorset.py ,src Ks\t|ttfs t|ts tdt|tr|D] }t|ts.t|ts.tdq.t|}|tkrz(dd|D}tt|t|f|}Wntk rt|}YnX|Sdd|D}t|}|tkrtt||f|}nd}|Snr|tkr0z|}tt||f|}Wntk r*t|}YnX|St|tkrTttt||f|SdSdS)a. Returns the eigenstate of the given operator or set of operators A global function for mapping operator classes to their associated states. It takes either an Operator or a set of operators and returns the state associated with these. This function can handle both instances of a given operator or just the class itself (i.e. both XOp() and XOp) There are multiple use cases to consider: 1) A class or set of classes is passed: First, we try to instantiate default instances for these operators. If this fails, then the class is simply returned. If we succeed in instantiating default instances, then we try to call state._operators_to_state on the operator instances. If this fails, the class is returned. Otherwise, the instance returned by _operators_to_state is returned. 2) An instance or set of instances is passed: In this case, state._operators_to_state is called on the instances passed. If this fails, a state class is returned. If the method returns an instance, that instance is returned. In both cases, if the operator class or set does not exist in the state_mapping dictionary, None is returned. Parameters ========== arg: Operator or set The class or instance of the operator or set of operators to be mapped to a state Examples ======== >>> from sympy.physics.quantum.cartesian import XOp, PxOp >>> from sympy.physics.quantum.operatorset import operators_to_state >>> from sympy.physics.quantum.operator import Operator >>> operators_to_state(XOp) |x> >>> operators_to_state(XOp()) |x> >>> operators_to_state(PxOp) |px> >>> operators_to_state(PxOp()) |px> >>> operators_to_state(Operator) |psi> >>> operators_to_state(Operator()) |psi> z%Argument is not an Operator or a set!zSet is not all Operators!cSsg|] }|qSrr)roprrr tsz&operators_to_state..cSsg|] }t|qSr)type)rorrrr{sN) isinstancer set issubclassNotImplementedError frozenset op_mapping _get_stater) operatorsoptionssopsZ op_instancesrettmpclassesZ op_instancerrrr/s@6    c Ks2t|tst|tstd|tkrjt|}zt|tt|f|}Wn ttfk rft|}YnXnt |tkrt|ttt |f|}nt|t r| tkrt|tt| }njt|t r&| tkr&t|}zt|tt| }Wn&ttfk r"t| }YnXnd}t|S)a` Returns the operator or set of operators corresponding to the given eigenstate A global function for mapping state classes to their associated operators or sets of operators. It takes either a state class or instance. This function can handle both instances of a given state or just the class itself (i.e. both XKet() and XKet) There are multiple use cases to consider: 1) A state class is passed: In this case, we first try instantiating a default instance of the class. If this succeeds, then we try to call state._state_to_operators on that instance. If the creation of the default instance or if the calling of _state_to_operators fails, then either an operator class or set of operator classes is returned. Otherwise, the appropriate operator instances are returned. 2) A state instance is returned: Here, state._state_to_operators is called for the instance. If this fails, then a class or set of operator classes is returned. Otherwise, the instances are returned. In either case, if the state's class does not exist in state_mapping, None is returned. Parameters ========== arg: StateBase class or instance (or subclasses) The class or instance of the state to be mapped to an operator or set of operators Examples ======== >>> from sympy.physics.quantum.cartesian import XKet, PxKet, XBra, PxBra >>> from sympy.physics.quantum.operatorset import state_to_operators >>> from sympy.physics.quantum.state import Ket, Bra >>> state_to_operators(XKet) X >>> state_to_operators(XKet()) X >>> state_to_operators(PxKet) Px >>> state_to_operators(PxKet()) Px >>> state_to_operators(PxBra) Px >>> state_to_operators(XBra) X >>> state_to_operators(Ket) O >>> state_to_operators(Bra) O zArgument is not a state!N) r r r"r# state_mapping _make_default_get_ops _make_set TypeErrorrr Z dual_class)stater( state_instr+rrrrs@;  cCs(z |}Wntk r"|}YnX|SN)r2)exprr+rrrr/s   r/cKs4z|j|f|}Wntk r.t|}YnX|Sr5)Z_operators_to_stater#r/)Z state_classr*r(r+rrrr&s r&cKsvz|j|f|}Wn@tk rRt|tttfrFtdd|D}nt|}YnXt|trrt|dkrr|dS|S)Ncss|]}t|VqdSr5)r/)rxrrr sz_get_ops..r)Z_state_to_operatorsr#r r!tupler$r/len)r4Z op_classesr(r+rrrr0sr0cCs t|tttfrt|S|SdSr5)r r:listr$r!)r*rrrr1sr1N)"__doc__Zsympy.physics.quantum.cartesianrrrrrrrZsympy.physics.quantum.operatorr Zsympy.physics.quantum.stater r r Zsympy.physics.quantum.spinr rrrrrr__all__r$r.itemsr%rrr/r&r0r1rrrrs8$ $    dX