Otj#<dZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZmZdd lmZmZmZgd ZGd deZGddeZGddeZGddeZedZedZedZ edZ!GddeZ"GddeZ#GddeZ$Gdd ee$Z%Gd!d"ee$Z&Gd#d$eZ'Gd%d&eZ(d'Z)d(Z*d)Z+d*S)+zuOperators and states for 1D cartesian position and momentum. TODO: * Add 3D classes to mappings in operatorset.py )Ipi)S)exp)sqrt) DiracDelta)Interval)hbar)L2)DifferentialOperatorHermitianOperator)KetBraState)XOpYOpZOpPxOpXYZPxXKetXBraPxKetPxBraPositionState3D PositionKet3D PositionBra3Dc\eZdZdZedZedZdZdZdZ ddd Z d S) rz1D cartesian position operator.cdS)N)rselfs ]/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sympy/physics/quantum/cartesian.py default_argszXOp.default_args/vcdtttjtjSNr r rNegativeInfinityInfinityr$argss r%_eval_hilbert_spacezXOp._eval_hilbert_space3(1-qz::;;;r(c ttzSr*)rr )r$others r%_eval_commutator_PxOpzXOp._eval_commutator_PxOp7s v r(c |j|zSr*)positionr$ketoptionss r%_apply_operator_XKetzXOp._apply_operator_XKet:|Cr(c |j|zSr*) position_xr7s r%_apply_operator_PositionKet3Dz!XOp._apply_operator_PositionKet3D=~c!!r(indexc |d|}|dj}|dj}t|}t||z }tt z||zzSN) start_indexrr@)_enumerate_statemomentumr rrr r$basisrBr9statescoord1coord2ddeltas r%_represent_PxKetzXOp._represent_PxKet@sc''u'==##  ( (6F?++vqwr(N) __name__ __module__ __qualname____doc__ classmethodr&r0r4r:r>rPr"r(r%rr,s))[<<[<   """01       r(rcDeZdZdZedZedZdZdS)rz8 Y cartesian coordinate operator (for 2D or 3D systems) cdS)N)rr"r#s r%r&zYOp.default_argsMr'r(cdtttjtjSr*r+r.s r%r0zYOp._eval_hilbert_spaceQr1r(c |j|zSr*) position_yr7s r%r>z!YOp._apply_operator_PositionKet3DUr?r(NrQrRrSrTrUr&r0r>r"r(r%rrJs]BB[<<[<"""""r(rcDeZdZdZedZedZdZdS)rz2 Z cartesian coordinate operator (for 3D systems) cdS)N)rr"r#s r%r&zZOp.default_args\r'r(cdtttjtjSr*r+r.s r%r0zZOp._eval_hilbert_space`r1r(c |j|zSr*) position_zr7s r%r>z!ZOp._apply_operator_PositionKet3Ddr?r(Nr[r"r(r%rrYs]<<[<<[<"""""r(rcPeZdZdZedZedZdZdddZdS) rz1D cartesian momentum operator.cdS)N)rr"r#s r%r&zPxOp.default_argsowr(cdtttjtjSr*r+r.s r%r0zPxOp._eval_hilbert_spacesr1r(c |j|zSr*)rHr7s r%_apply_operator_PxKetzPxOp._apply_operator_PxKetwr;r(r@rAc |d|}|dj}|dj}t|}t||z }t t z||zzSrD)rGr6r rrr rIs r%_represent_XKetzPxOp._represent_XKetzse''u'==##  ( (6F?++r$w%  r(N) rQrRrSrTrUr&r0rfrhr"r(r%rrls{))[<<[<   /0!!!!!!!r(rrrrrceZdZdZedZdZedZedZe dZ dZ dZ d Z d S) rz1D cartesian position eigenket.c >|j|gt|Ri|Sr*__new___lowercase_labelsr$opr9s r%_operators_to_statezXKet._operators_to_state-t|DD#4R#8#8DDDGDDDr(c >|j|gt|Ri|Sr*rl_uppercase_labelsr$op_classr9s r%_state_to_operatorszXKet._state_to_operatorsDxE!24!8!8EEE|j|gt|Ri|Sr*rkrns r%rpz#PositionState3D._operators_to_staterqr(c >|j|gt|Ri|Sr*rsrus r%rwz#PositionState3D._state_to_operatorsrxr(cdS)N)r{yzr"r#s r%r&zPositionState3D.default_argssr(c|jdS)z The x coordinate of the state rrr#s r%r=zPositionState3D.position_xrr(c|jdS)z The y coordinate of the state r@rr#s r%rZzPositionState3D.position_yrr(c|jdS)z The z coordinate of the state rErr#s r%r`zPositionState3D.position_zrr(N) rQrRrSrTrUrprwr&rr=rZr`r"r(r%rrs<<EE[EEEE[XXXr(rc.eZdZdZdZedZdS)rz 3D cartesian position eigenket c |j|jz }|j|jz }|j|jz }t|t|zt|zSr*)r=rZr`r)r$rr9x_diffy_diffz_diffs r% _eval_innerproduct_PositionBra3Dz.PositionKet3D._eval_innerproduct_PositionBra3DsT3>13>13>1&!!*V"4"44Z5G5GGGr(ctSr*)rr#s r%r}zPositionKet3D.dual_classr(N)rQrRrSrTrrUr}r"r(r%rrsG**HHH[r(rc(eZdZdZedZdS)rz 3D cartesian position eigenbra ctSr*)rr#s r%r}zPositionBra3D.dual_classrr(N)rQrRrSrTrUr}r"r(r%rrs3**[r(rceZdZdZedZdZedZedZe dZ dZ dZ d Z d S) rz1D cartesian momentum eigenket.c >|j|gt|Ri|Sr*rkrns r%rpzPxKet._operators_to_staterqr(c >|j|gt|Ri|Sr*rsrus r%rwzPxKet._state_to_operatorsrxr(cdSN)pxr"r#s r%r&zPxKet.default_argsrcr(ctSr*)rr#s r%r}zPxKet.dual_class  r(c|jdSzThe momentum of the state.rrr#s r%rHzPxKet.momentumrr(c"t|g|Ri|Sr*r)r$r/r9s r%rGzPxKet._enumerate_states '?t???w???r(c tt|jz|jztz t dt ztzz Sr)rrrHr6r rrrs r%rzPxKet._eval_innerproduct_XBras51T]?3</455d1R49ooEEr(c :t|j|jz Sr*)rrHrs r%rzPxKet._eval_innerproduct_PxBrarr(N)rQrRrSrTrUrprwr&r}rrHrGrrr"r(r%rrs))EE[EEEE[[X@@@FFF88888r(rcTeZdZdZedZedZedZdS)rz1D cartesian momentum eigenbra.cdSrr"r#s r%r&zPxBra.default_args"rcr(ctSr*)rr#s r%r}zPxBra.dual_class&rr(c|jdSrrr#s r%rHzPxBra.momentum*rr(N) rQrRrSrTrUr&r}rrHr"r(r%rrsi))[[Xr(rc|d}|d}|j}|dg}t|dkr6|dd}tt |||z}dt t|D}t |D]>\}} |jd} |t| dzt| zfi|||<?|S)Nrr@ index_listrFcg|]}dSrr").0is r% z,_enumerate_continuous_1D..>s5551555r(_) __class__poplenlistrange enumerater/str) r/r9stater state_classrrF enum_statesrindrs r%rr4s GEaJ/K\2..J :!kk-33 % [:-EFFGG 55eC OO44555KJ''MM3 1 $SZZ#%5C%@LLGLL A r(cJt|ts|g}d|DS)Ncfg|].}t|jd/Sr)rrlowerrargs r%rz%_lowercase_labels..Ks2 5 5 5#C !   # # % % 5 5 5r( isinstanceset)opss r%rmrmGs/ c3  e 5 5 5 5 55r(cNt|ts|g}d|D}|S)Ncg|]W}t|jddt|jdddzXS)rr@N)rrupperrs r%rz%_uppercase_labels..Rsf666*-CIaL!!!$**,,CIaL!!!""%&666r(r)rnew_argss r%rtrtNs@ c3  e6614666H Or(N),rTsympy.core.numbersrrsympy.core.singletonr&sympy.functions.elementary.exponentialr(sympy.functions.elementary.miscellaneousr'sympy.functions.special.delta_functionsrsympy.sets.setsr sympy.physics.quantum.constantsr sympy.physics.quantum.hilbertr sympy.physics.quantum.operatorr r sympy.physics.quantum.staterrr__all__rrrrrrrrrrrrrrrrrmrtr"r(r%rs;'&&&&&&&""""""666666999999>>>>>>$$$$$$000000,,,,,,RRRRRRRR7777777777   .        < " " " " "  " " " " " " " "  " " "&!!!!! !!!.CHHCHHCHH T$ZZGGGGG3GGGD3"e>     C   $C88888C888DC*&666r(