From 02e3548c6b74906e08049eb3b2bdc6006ee5b48a Mon Sep 17 00:00:00 2001
From: "E. Almqvist" <elalmqvist@gmail.com>
Date: Thu, 31 Mar 2022 10:18:14 +0200
Subject: [PATCH] Quantum systems and classes

---
 mas/shrodequ_superpos/main.py                 |  28 +++++++-------
 mas/shrodequ_superpos/phys/__init__.py        |   2 +-
 .../phys/__pycache__/__init__.cpython-310.pyc | Bin 294 -> 290 bytes
 .../phys/__pycache__/const.cpython-310.pyc    | Bin 771 -> 781 bytes
 .../phys/__pycache__/func.cpython-310.pyc     | Bin 777 -> 760 bytes
 .../phys/__pycache__/particle.cpython-310.pyc | Bin 559 -> 1459 bytes
 .../quantum_systems.cpython-310.pyc           | Bin 0 -> 1222 bytes
 mas/shrodequ_superpos/phys/box.py             |   9 -----
 mas/shrodequ_superpos/phys/const.py           |  10 ++---
 mas/shrodequ_superpos/phys/func.py            |   1 -
 mas/shrodequ_superpos/phys/particle.py        |  35 +++++++++++++++---
 mas/shrodequ_superpos/phys/quantum_systems.py |  21 +++++++++++
 12 files changed, 72 insertions(+), 34 deletions(-)
 create mode 100644 mas/shrodequ_superpos/phys/__pycache__/quantum_systems.cpython-310.pyc
 delete mode 100644 mas/shrodequ_superpos/phys/box.py
 create mode 100644 mas/shrodequ_superpos/phys/quantum_systems.py

diff --git a/mas/shrodequ_superpos/main.py b/mas/shrodequ_superpos/main.py
index d05a823..fac4444 100755
--- a/mas/shrodequ_superpos/main.py
+++ b/mas/shrodequ_superpos/main.py
@@ -3,19 +3,21 @@ import matplotlib.pyplot as plt
 import numpy as np
 from phys import *
 
-x = np.arange(0, BOX_LENGTH, 0.00001)
-n = np.arange(1, 10, 1)
+quantum_system = box(0.001) # box with inf walls
 
-X, N = np.meshgrid(x, n)
-P = prob(X, N)
-
-fig = plt.figure()
-ax = plt.axes(projection='3d')
-ax.plot_surface(X, N, P, rstride=1, cstride=1, cmap='viridis', edgecolor='none')
-ax.set_xlabel('x [m]')
-ax.set_ylabel('n [int]')
-ax.set_zlabel('probability [frac]');
-
-plt.show()
+# x = np.arange(0, BOX_LENGTH, 0.00001)
+# n = np.arange(1, 10, 1)
+# 
+# X, N = np.meshgrid(x, n)
+# P = prob(X, N)
+# 
+# fig = plt.figure()
+# ax = plt.axes(projection='3d')
+# ax.plot_surface(X, N, P, rstride=1, cstride=1, cmap='viridis', edgecolor='none')
+# ax.set_xlabel('x [m]')
+# ax.set_ylabel('n [int]')
+# ax.set_zlabel('probability [frac]');
+# 
+# plt.show()
 
 
diff --git a/mas/shrodequ_superpos/phys/__init__.py b/mas/shrodequ_superpos/phys/__init__.py
index 24de9b4..0453745 100644
--- a/mas/shrodequ_superpos/phys/__init__.py
+++ b/mas/shrodequ_superpos/phys/__init__.py
@@ -1,4 +1,4 @@
 from phys.func import *
 from phys.const import *
-from phys.box import box
+from phys.quantum_systems import *
 from phys.particle import particle
diff --git a/mas/shrodequ_superpos/phys/__pycache__/__init__.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/__init__.cpython-310.pyc
index 95dbd8dfc5a61aab28f202c8d9fd0e19ea90bb3d..4dc77025e5f704d90e4ab152f954015ab71da9e0 100644
GIT binary patch
delta 155
zcmZ3+w1|l}pO=@50SJ<kT$9Qt^2)kv068fPDU3M`ArP7oN;83JrX1#6mM9iRh7{&t
z22GZiAibK5w-~j61V=$)QAuWUPO6_K>%<xfBawoP%3{63(!{)y(%kss%HopL+~O!+
cuqaGv5i`($l?+8JK&psk;{Ie74n_e+0L$Jc;s5{u

delta 159
zcmZ3)w2X;2pO=@50SH_oT$6Gq^2!G3068fPDU3M`ArP7oOf%&$<uXSxGlImJb69d&
zqga7#mS6@=)|VjtnvAy?wSWY3Qho)H;wVTgD#=XFN%hlYn^-8J%~6n1S*!<?D`EyJ
ej^YJ#VQPw4fMP2didcbE5$nX=$t*mK0*nBr93-*;

diff --git a/mas/shrodequ_superpos/phys/__pycache__/const.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/const.cpython-310.pyc
index ff4b28629a52bdbfcc3535dd222f21962ac4a884..d292ebc59b938e4e1b4db34df9fdb84c3c18393b 100644
GIT binary patch
delta 91
zcmZo>>t*B3=jG*M0D?17u1W42d3zbz*;4aTi_$A6&taU-%34sI89&*CDN&0HA{w8P
kns$q|C^0WRwTK(2?G{^NK|yL>N)#tZWj089@;)Y40L8l<1^@s6

delta 81
zcmeBWYi8rk=jG*M0D_z0u1R_ud3za|7+ohXWSq{(IN6pdQIj_{FSRJWGQPMZu_U$l
j7Hd&rUV3T~H&DYZw#0&h)V!2itOdoH@!6A4Fu4K%mWml-

diff --git a/mas/shrodequ_superpos/phys/__pycache__/func.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/func.cpython-310.pyc
index fc4dc392bf6de4a0dcf9838ef0ed4474f570184f..475579c6a890283bc3febae161bd4467cb6f6c39 100644
GIT binary patch
delta 159
zcmeBV`@zba&&$ij00gP=u1RGRd1ZZkfSeSD6viBeC`KU7l*1IlkiwkGoW_*GlET`;
z62$`Lv8AvBd8|+#M+zs9#|GtbrEmjz>=WyDF|tm!WUOW6n7o}aiIHcr6jPoMKhTC^
hCLkfiSR@LhG&v?OWC|1D0WvulI2ieufsjRr830dQ9j^cY

delta 176
zcmeyt+R4V7&&$ij00cqdu1U=kd1XUFfSeSD6viBeC<x6M#RTLt<uHdZq%fzlq%ozi
zq_DQIM6p77Y$@zO9vhU$k-`b&u|s)WDcnFF$HcN-jBJw)8EY9iC$DErV&t7H#FQr~
u0JN!?2}lSr7Ks8WP0lFR<lMxPjLDOj!i4yMA{-1Hj2w)7%s|K@#0&tOydj|g

diff --git a/mas/shrodequ_superpos/phys/__pycache__/particle.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/particle.cpython-310.pyc
index 28d76dce2d4e1d29291b3eee5f1ad6a22593da6c..6e78f4b28e9054cf82c841e2498c7e37101dafbb 100644
GIT binary patch
literal 1459
zcmZuxOOM+&5avTKKi5g!c99gQdsGf<w3lAmCO}Z6Cu4yGJ?J9HA|1=CS4cXs4ee?7
z2lO5U*c|(pc<pKbLQkEMA6*os!A~<3hn)H5ldHj?OJF>o|2qG}A>=P~9u5x93z+(M
z06_%JNkJo8u!vFgS<VY5atb$cTc79N8(epCKMLAsC+fh{&AUY}>QVBF2v7L8MEEx>
z>I>&J34?n$n}p2rT4&a&RW%34lciK|x7LNl_$s89KAQo0mw8gAS1E=(96mTNVCvrh
zB#EdX5ff*ENiMi_gd^RBCtTs(x)CHG`GSc+bZ(u<M^AKN_GAEi`eFd99WjK|MmDGu
zWwJDv_Y@AcPLXJBJ-yXN7S=CgZ4x6>pzbu#M%rG&)NcS9F6b3e0|Rt5{3H9sZYYxV
znl!W_F>N>?YZxFmPD57k9&E?BCmSt(=QZT>Au{AM)q%5jKVBa4o9y|Tv<XeJQpnAu
zSZgzx%gOhXN==GN$?qnIjDHS&U{+<NQ78b_N9ZA7`pN~c{X~fPSbpeON6UP%-`?)T
zm_Lac!Q<~&(`8l2smzmnda0_FOpTuEbXisT6dL;Whba&@)k{?g`EDKSbuCq0>1n;(
z0vt-XsJAwV<E+d~9AjGgF+f0v)S({ps7L#(Pe*L`=oE3>ZRsXuu=zO9#v<A#oPz-b
zk`*=V4#-|P4IKlOV`xE0XQA6xZhK{=ig=Nvrc#!bwdFc1TPi}P&fpC71p?N)hev4P
zO?ZKB4Z7=}a%<IWvO-!<NwZdEin2Hk6znO(890I($Z(9AGC-JyKP5eK4WdBd+^}nq
z1c<=-Os`Q8{wgr!t^-#0SkJZBc#YfE#b@`JkwV#qJ@q9#EUUA9fAMnbr&Uqsa%0`J
zNX*i5xvBRW7-Qg92$*5$KdAm|Sctq`oK|vS=`QMFfd|rBF8;%hbv`8T<x_~#*wH7n
z3%_$3u-)K96z+g;yA^cUMG2G%92(nh92b>X=V<ri_}x0m4{zK>UM0qQS-F^38_QA1
zD3*N+c%Awd;ooMpR@<=vUf{Q*(Fn*n9S)yxg-$rI9aQASVqK>Dqp)SR+b*V6sm+7$
zW8AJnQl(iJ@%}GCTen;n^|s|(eS^VgI?d|sMH>xUwL<dS>;5l08~n5{`4?=Q9s-QW
Nhz7t|K>PGD`v+@vNGJdR

literal 559
zcmYjOy-ve05Vjqsq!pD410w@lmPTSms1icmDkLVdSZVCiASX`kWGL!JUxNqWmAo?X
z3QXLE7IM;k_WkJdcV{ymj|f=&biaIu^Am%<1rW@@-4%c&l4?@Yl$I=Il+1}_GT0Lt
zys<Qp{E@`r7xa;s`Ji@%=e0G<95a365X`{cC4eF+l_X`7sz5TuS6l+qo~NM<fp6fW
zI<rlYYvp;FIp<F*;hL;bIZ8Vk-Wm8ExSIg%=#B`oV}L*pM061|8$!hn0OG)Nr}XMS
z^FeF`h3|Q?Hl<3G&UEr%&5Oz#m$-awj80&h=j#MiByMeuRGU_~wpO+_E~(dBfP>U&
zz4eh0MO8FH;J6N9Ku7fT?62ooU*M2JR9UHn@FO8gBU_F4Qz15OrVo{HrA^lOpr{&)
zpVsN^!4Px!4lbj6Fd`SU%PeB<hp2G6YOCDhrYv@K#U2H2z#@;2ZhlqVQiwXT2{~s!
DuBm#$

diff --git a/mas/shrodequ_superpos/phys/__pycache__/quantum_systems.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/quantum_systems.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..97b024b3e23d0f15ad3cc8411d2e7af3f8131edf
GIT binary patch
literal 1222
zcmZWo&2AGh5VpPE&88$NKNXc)gdz^KxpXBCoDeD?gt$aV-1cH^*G`ja|7tr@6ZO_g
zeF%h9j(sIxIrSBKV#aAo0u#-|Gaiq3zWH2tyBh@LrTBalGeUl&vnm{%Ls0V;j3km4
zq@*dOxMl@QgD*q|l3x<ZFIdXK3uOder1(XUhLU|JgVrsaB{$T?O{VQ6FVrx{$f^Y3
z9D<sCFp8v9l9WlRf)SHU2A4eLlB*Crtk2U(wxDwCdZ*K@veQzSnX#&Lyvj<ILnjY{
zYt&Dm<}sLadQJp6CSc(fs6vMWrbF0q?d1oW0dw4_V&vmDM8~jwH%!KLsS;IWMe<G8
zCn~olG5NTzi=@%@@tdT~ATZXoRHsv6rj61~ZIWg@Gs*ggnuBKMVj(8g#0r6z8-zNd
z8+5+Co_es%(g;#c!7K&W5vcDCk+bB4UbA!Z3npX(37!6cIYE@efABNB@RNg{`Eb7Z
zK}se2%a;4|t-h`M_jmUfc=q@1=-wdIEvT*WCKqK*qbljbMo%j1;n4`MM`&T%nAgdq
z6wBHe@%?q^3p(HZr@z(A;h=@E;1q>$8$y(|oEEt52m$Ac`<=j8t&zGO6y00@0-Tn{
zOErRuYVhM@(xXqgkLBR52_~VKS76o&ctu9kldzNWE2x%+2%w85)##``b5WtHV>|wj
zbXXB5W?qA_1fC2Mk9zts8(eb<gF0cmFf53No{GV6-uhaN`XB4FTSRpbxzN*S0rVDx
z9IG1O=Bc*sQB~_wjI!L;8Yt|CgemaU+Ct<-W{ePjXX5cfqzvN2X#~ILbiTRh<AG*_
zNN>Wv#<!xMfN|lds59%2yY7AATouK#xQBZO(>><TVATcn6OYLb)fj<Ry7&+dmZb2L
mb73_to0;Cm(}<YQT_MxMC|&$+!G-t8df!5a#I(n@+J69mISK;+

literal 0
HcmV?d00001

diff --git a/mas/shrodequ_superpos/phys/box.py b/mas/shrodequ_superpos/phys/box.py
deleted file mode 100644
index 4b0f2fb..0000000
--- a/mas/shrodequ_superpos/phys/box.py
+++ /dev/null
@@ -1,9 +0,0 @@
-from phys.particle import particle
-
-class box:
-    def __init__(self, length: float):
-        self.length = length
-
-    def add_particle(pobj: particle):
-        pass
-
diff --git a/mas/shrodequ_superpos/phys/const.py b/mas/shrodequ_superpos/phys/const.py
index f1b529b..4afbd48 100644
--- a/mas/shrodequ_superpos/phys/const.py
+++ b/mas/shrodequ_superpos/phys/const.py
@@ -13,14 +13,14 @@ def energy(n):
     return (((h*n)/BOX_LENGTH)**2)*(1/(8*m))
 
 # Wave func inner coef
-def k(n):
+def psi_k(n):
     return np.sqrt( (8 * ((np.pi)**2) * m * energy(n))/(h**2) )
 
 # pre calculate energy states
-energy_states = []
+energy_def = []
 for n in range(0, 11):
-    energy_states.append( energy(n) )
+    energy_def.append( energy(n) )
 
-psi_k = []
+psi_k_def = []
 for n in range(0, 11):
-    psi_k.append( k(n) )
+    psi_k_def.append( psi_k(n) )
diff --git a/mas/shrodequ_superpos/phys/func.py b/mas/shrodequ_superpos/phys/func.py
index a63247e..34c139e 100644
--- a/mas/shrodequ_superpos/phys/func.py
+++ b/mas/shrodequ_superpos/phys/func.py
@@ -1,4 +1,3 @@
-import cmath
 import numpy as np
 from phys.const import *
 
diff --git a/mas/shrodequ_superpos/phys/particle.py b/mas/shrodequ_superpos/phys/particle.py
index 72db7a5..868d8d6 100644
--- a/mas/shrodequ_superpos/phys/particle.py
+++ b/mas/shrodequ_superpos/phys/particle.py
@@ -1,11 +1,36 @@
-from phys.func import psi, prob, E
+from phys.func import psi, prob
+from phys.const import energy, psi_k
+from phys.quantum_systems import *
+
+import numpy as np
+import cmath
+from scipy.constants import Planck
 
 class particle:
-    def __init__(self, mass: float, n_state: int):
-        self.mass = mass
-        self.n_state = n_state
+    def __init__(self, mass: float, system: infbox, n_state: int=1):
+        if n_state < 1:
+            print(f"Particle energy state index must be 1 or more! {n_state=}")
+            return
+
+        self.mass = mass # particle mass
+        self.n_state = n_state # energy state index
+        self.system = system # quantum system
         
-        self.energy = 
+        self.energy = energy(self.n_state) # total energy "for" the particle
+        self.psi_k = psi_k(self.n_state)   # wave function coeff-
+
+        self.system.add_particle(self)
+
+    def psi(self, x: float):
+        return self.system.norm_factor * np.sin(self.psi_k * x)
+
+    def wave(self, x: float, time: float) -> complex:
+        z_coef = -( (2*np.pi*self.energy*time)/Planck ) # exponent coef
+        z = complex(0, z_coef) # full coef (complex)
+        time_factor = cmath.exp(z) # TODO: cmath.exp is weird, verify...
+        psi = self.system.norm_factor * np.sin(self.psi_k * x)
+
+        return psi * time_factor
         
 
 
diff --git a/mas/shrodequ_superpos/phys/quantum_systems.py b/mas/shrodequ_superpos/phys/quantum_systems.py
new file mode 100644
index 0000000..bf31a46
--- /dev/null
+++ b/mas/shrodequ_superpos/phys/quantum_systems.py
@@ -0,0 +1,21 @@
+from phys.particle import particle
+import numpy as np
+
+class quantum_system:
+    def __init__(self, name: str):
+        self.name = name
+        self.particles = []
+
+    def add_particle(particle: particle):
+        self.particles.append(particle)
+        print(f"Added particle {particle} to quantum system '{self.name}'")
+
+class infbox(quantum_system):
+    def __init__(self, length: float):
+        super
+        self.name = "Inf Box"
+        self.length = length
+        self.norm_factor = np.sqrt(2/self.length)
+
+
+