g[%TdZddlmZmZmZmZmZmZmZddl m Z ddl m Z m Z mZddlmZmZmZmZmZmZmZddgZeeeefeeeefeeeefee eeeeefeeeeiZdeDZd Zd Zd Zd Zd Z dZ!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_operatorsci|]\}}|| Sr).0kvs ]/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/physics/quantum/operatorset.py r,s 5 5 5tq!a 5 5 5c t|ttfs$t|tst dt|tr|D];}t|ts$t|tst d>> 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!c"g|] }| Srr)rops r z&operators_to_state..ts333333rc,g|]}t|Sr)type)ros rr"z&operators_to_state..{s(((q477(((rN) isinstancer set issubclassNotImplementedError frozenset op_mapping _get_stater$) operatorsoptionssops op_instancesrettmpclasses op_instances rrr/s l y8S/ 2 2KjH6U6UK!"IJJJ)S!!( G GAq(++ G H-- G)*EFFF "" *   &33s333  C#l2C2COOwOO& & & & o &J((C(((CnnG*$$ G!4cEEWEEJ  " " ,'ikk  I!6 OOwOO& , , , + ,J )__ * *ji99PPPP P4s$-2C C:9C: #E//F F c 0t|ts$t|tstd|tvr]t |} t |tt|fi|}n#ttf$rt|}YnmwxYwt|tvr5t |ttt|fi|}nt|trV| tvr;t |tt| }nt|tr| tvrt |} t |tt| }n8#ttf$r"t| }YnwxYwd}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 TypeErrorr$r dual_class)stater. state_instr2s rrrsv ui ( (>Jui,H,H>!"<=== "5))  ':$]5%9::GG>EGGCC#Y/ ' ' '&CCC ' e % %u tE{{!;<<II@GII E7 # # (8(8(:(:m(K(Ku u/?/?/A/A!BCCEE E7 # #(8(8(:(:m(K(K"5))  4:$]53C3C3E3E%FGGIICC#Y/ 4 4 4 0 0 2 23CCC 4 S>>s$&A;;BB:G0HHcB |}n#t$r|}YnwxYw|SN)r;)exprr2s rr8r8s?dff  Js c d |j|fi|}n#t$rt|}YnwxYw|Sr@)_operators_to_stater)r8) state_classr0r.r2s rr,r,sV)-k-c==W== )))K(() Js --c < |j|fi|}n[#t$rNt|ttt frt d|D}nt |}YnwxYwt|trt|dkr|dS|S)Nc34K|]}t|VdSr@)r8)rxs r z_get_ops.. s*==Q a((======rr)_state_to_operatorsr)r&r'tupler*r8len)r> op_classesr.r2s rr9r9s,,j,ZCC7CC ,,, j3y"9 : : ,==*=====CC ++C , #sCA 1v JsAA)(A)cht|tttfrt |S|Sr@)r&rKlistr*r')r0s rr:r:s+#tY/003xx rN)"__doc__sympy.physics.quantum.cartesianrrrrrrr sympy.physics.quantum.operatorr sympy.physics.quantum.stater r r sympy.physics.quantum.spinrrrrrrr__all__r*r7itemsr+rrr8r,r9r:rrrrWs <<<<<<<<<<<<<<<<<<333333??????????////////////////// D$<00D$<00D$<00xCc?!;!;  6 5}2244 5 5 5 aaaHUUUp   "r