From cc12437725741908eea5195bbb491f1a78996598 Mon Sep 17 00:00:00 2001
From: "E. Almqvist" <elalmqvist@gmail.com>
Date: Thu, 31 Mar 2022 09:38:23 +0200
Subject: [PATCH] Shrodprob stuff

---
 .../__pycache__/phys_const.cpython-310.pyc    | Bin 0 -> 762 bytes
 .../__pycache__/phys_func.cpython-310.pyc     | Bin 0 -> 777 bytes
 mas/shrodequ_superpos/main.py                 |  21 ++++++++++++++
 mas/shrodequ_superpos/phys/__init__.py        |   4 +++
 mas/shrodequ_superpos/phys/__main__.py        |   0
 .../phys/__pycache__/__init__.cpython-310.pyc | Bin 0 -> 294 bytes
 .../phys/__pycache__/box.cpython-310.pyc      | Bin 0 -> 645 bytes
 .../phys/__pycache__/const.cpython-310.pyc    | Bin 0 -> 771 bytes
 .../phys/__pycache__/func.cpython-310.pyc     | Bin 0 -> 777 bytes
 .../phys/__pycache__/particle.cpython-310.pyc | Bin 0 -> 559 bytes
 mas/shrodequ_superpos/phys/box.py             |   9 ++++++
 mas/shrodequ_superpos/phys/const.py           |  26 ++++++++++++++++++
 mas/shrodequ_superpos/phys/func.py            |  17 ++++++++++++
 mas/shrodequ_superpos/phys/particle.py        |  11 ++++++++
 14 files changed, 88 insertions(+)
 create mode 100644 mas/shrodequ_superpos/__pycache__/phys_const.cpython-310.pyc
 create mode 100644 mas/shrodequ_superpos/__pycache__/phys_func.cpython-310.pyc
 create mode 100755 mas/shrodequ_superpos/main.py
 create mode 100644 mas/shrodequ_superpos/phys/__init__.py
 create mode 100644 mas/shrodequ_superpos/phys/__main__.py
 create mode 100644 mas/shrodequ_superpos/phys/__pycache__/__init__.cpython-310.pyc
 create mode 100644 mas/shrodequ_superpos/phys/__pycache__/box.cpython-310.pyc
 create mode 100644 mas/shrodequ_superpos/phys/__pycache__/const.cpython-310.pyc
 create mode 100644 mas/shrodequ_superpos/phys/__pycache__/func.cpython-310.pyc
 create mode 100644 mas/shrodequ_superpos/phys/__pycache__/particle.cpython-310.pyc
 create mode 100644 mas/shrodequ_superpos/phys/box.py
 create mode 100644 mas/shrodequ_superpos/phys/const.py
 create mode 100644 mas/shrodequ_superpos/phys/func.py
 create mode 100644 mas/shrodequ_superpos/phys/particle.py

diff --git a/mas/shrodequ_superpos/__pycache__/phys_const.cpython-310.pyc b/mas/shrodequ_superpos/__pycache__/phys_const.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..c80850bcea25f803ea2c6c8f1ec439263c9d7d81
GIT binary patch
literal 762
zcmZWnJ&zMH5cPPyo6W~2fuzfcI%#etIwS-qT~0J8IteOEC|cH5lFNQL_NEA2m#g>{
zx$+u*kUK<`h95u!<2{5TurzNxv*Vff>@B0HMR5IifAw)d3Hk1V)8%7ufo8rS2qGXO
z9MIE@<{tNQ#u*UdabJ2okZ@=SCjAk)2Odhso5B~t9pNp}5Fz^Zg7A)v@O*Pmr{qJE
zcV*|$lP%fZ!#x1mjIpBsPemUq+J7rv2zW;(tt0XwvkA4qb(R$A+D0;yX{F0zo+rjE
zJ`caX`S$7V^$}H(6kPS6<@OBi63v_;ROFTnNd+U6H$Z8^tLTu7Fo%QKOuOQvJ9P{Z
zCZ2_*g|`;+2e42)5Z%G_&uP3Ya~aDl$>M8W-XIwhn{-*0S*&!qcpf9K*erD^<YqfJ
zTP3wBO{|tXGf&IHROf1E;mS2Lj`SJmF1r7?yC<o+k8x9x0|=<d4Hb0YRCs9Ymh1t&
z?{e-jP=`0u0oK=ICV{1evdnCBWvN=}ru*WC>4pRMOVpwsA;QqYx{t0VHU|9^_s2HE
z<cal*ZLW5@hY|0YbftFZwVb4=jCMk_=fK_FV|&-g!t`XSYsVv5NWIw2O_fyASYIc_
zLTWd}28mL#5Y|^_HD8}lbC;!Se&Nf!6x&R`at~#mA%vt4%wy1{3_>qtjCBDsw8me*
E0pzBfQ~&?~

literal 0
HcmV?d00001

diff --git a/mas/shrodequ_superpos/__pycache__/phys_func.cpython-310.pyc b/mas/shrodequ_superpos/__pycache__/phys_func.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..178b31c2b4fa5d8fd46a94dbb1f3c393d23547cc
GIT binary patch
literal 777
zcma)4L2uJA6!x=|HZ5y`ZcO4QgaeT{BZM?|t&lb@IZ&l>O2(4d*p5V{-O3MP$NnW>
zIql4i6Yn{dn6w>m<!67-eee7HIcheWG8~^jov(Mm*e?S2hd}Tir@h1=nS_Sr!wNY^
z%o}mZq^P0|mywKrM0tEbBRNJiI-o>O5RDILDxV-qBwVoc=ngxxWeOf%`H`tyZ@Q9G
z?l5Q{<2=J@Pcb;=;3s$D26k}6PS`D+unV?_Wg2<-;CZLLuveXj>pl1I-9xu$<_HV_
zoMp9YtE_5@Ci`gA*Q#_jvt_MRlWC*Ayvf=E6SYyY`qr;(uPdXK&2+uBt4-gPul3do
zZLdby?KcTK1vAIxe+9o}{{))|3)9HZ-3&D}V+?tW?N`@kLJ1N*gc<Jqn2wZo;p{1$
zJ;U&To8By#Mamp9qETGJF`2}dcO*L<WIxIhp%ZK3v-Hn7m}6WXkdXL5vcL_!$Uw7$
zfyg|^bjrOb);6eolqw-Ax)I$Y&@I5cpvp-emu=x{A9sDLx51x$8s4E)opnCmG^%jH
m{GmUo47K-M3;qS`lDDmreN(-|kDOUDCSeIAFc<iX7UD1AJ&#ua

literal 0
HcmV?d00001

diff --git a/mas/shrodequ_superpos/main.py b/mas/shrodequ_superpos/main.py
new file mode 100755
index 0000000..d05a823
--- /dev/null
+++ b/mas/shrodequ_superpos/main.py
@@ -0,0 +1,21 @@
+#!/usr/bin/python3
+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)
+
+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
new file mode 100644
index 0000000..24de9b4
--- /dev/null
+++ b/mas/shrodequ_superpos/phys/__init__.py
@@ -0,0 +1,4 @@
+from phys.func import *
+from phys.const import *
+from phys.box import box
+from phys.particle import particle
diff --git a/mas/shrodequ_superpos/phys/__main__.py b/mas/shrodequ_superpos/phys/__main__.py
new file mode 100644
index 0000000..e69de29
diff --git a/mas/shrodequ_superpos/phys/__pycache__/__init__.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/__init__.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..95dbd8dfc5a61aab28f202c8d9fd0e19ea90bb3d
GIT binary patch
literal 294
zcmYjJ%}T^D5Y9BMWETs*MnyKTXAuvIHxcw8xrA;SZ6)c%qy_hpe5Jj5@)bNejVw4Y
z-ybs{*7cgv+&tgzJn`3J{!c;Xnl@b!bgTo%p8gmz7q0MJaac0$N?)mpbot1d>YF}T
z0|z)K;5+;v5}}RLV4S^gPSh&&`?PtPN2AtR8XVK4MAmd4ITffn+v8s>Df?uxBFgY{
zCwuf(TGu-H5b@2LDamAdL?=VU*RAv|1%1TMzRx<%!N!0|F11qYVH_r{H(`Gemma&>
Q*(;##BnSi(a0+Me3+cm3wEzGB

literal 0
HcmV?d00001

diff --git a/mas/shrodequ_superpos/phys/__pycache__/box.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/box.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..d287653feec182e6970fb25d1a89dc198c03405a
GIT binary patch
literal 645
zcmYjOu};G<5VaGh4W&h>AHYnOMq)+?fe>4T)CFa+(%5ZUIdy8M60O+aW9Tn=W#R)^
zn7DH(%1L**duP9U=jg$pN5H<HAEqDh{lubeAr!aZ_5y++f+|wcj8gJQL?FU<BEq*I
z3j~{xB>I9gB%v;@bJHveB?|=F_D^>Ss3aK`1TIq<&H_)zGA5Xepo&E2on;--1>)F+
zQ~lyPO0MQjSp+Rc%+c?`Ew1uPUkL}L6SAk{gc=4|5;$h1n)zZ+Oc$`9yR_7mOr^?I
z`egJ{7L849QEIKy+UWUJTIB$gMhm&#aJ#9cskKe(a%)qNGpe^P=6tbQG@RoUi#mFA
z*T?3u;aIJw%Ys7dXqcmW$#@aPng|jv3^d~&K0kyQO4rYY;Kz5x1^hThcjq{+-5e)f
z6t>DM$+_!sUTLvWnD;qfZ*uk5iDpXYjXA+#DBYZ5K<ZY&*#`d_y6r%S$r&An9vdg2
f8{m74j=~N^c(%piDg6@h^+C%G?r(ATJ{bnTB|nAZ

literal 0
HcmV?d00001

diff --git a/mas/shrodequ_superpos/phys/__pycache__/const.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/const.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..ff4b28629a52bdbfcc3535dd222f21962ac4a884
GIT binary patch
literal 771
zcmZWnJ&zMH5cPPy$>w8|z@^KHI%%>J9TEZriH-t=lRyGlLea9elHBcwV{eMUb-9XP
zkt?s^|KJW$<uA~{cn_fnEX^B_{X8?zb`ge6g5%q}>Bli8<cAB6hmXNCH1h>P5CJLS
zfS#r_^SGBW&VUGy`_khz3H!QW(jSl;;DKbkA$(E0CcG)?B0%3-5Z;y{o^R~vgnVf5
zj%@F{vMD<|xB(#RAu@XZFnY*n{bf87@P>?<2keK;M%32M(>PBqZ75Tjl)A|0S!~SW
zbN}nv_fOYP52%VH;MjkLn=`Z*Xyz25Bv+(QN*JKL6O<;rlJ>~}*RT+qX-7V|Q-=Uy
z<XKqS=JVI@=dY%-v$roTWVc`;za_ej`#&eqvdCm4(>RUJb#Z}RnaCu|qDUj9i^anT
z8;s0S7ecN#bF)!WtHMNTxiwKz<fa^}t%a#eW?c8DpgZXP6XD*b5;4R{N%kP1Bo|cB
z6DP+*V^?Gc=zX{5UIhhtGhIV|rD{~OG*_0HwJt4HE8TEAu9&VnaDSr`bq67a7A||}
zs>X((@8kT?LKxk%e!j`nR(CPtU6ZWTc3jDc^U`Q1M0*b0CElfXi7ZTxs=9PMl)2Q4
z?c9`cDUJ1YoG+wyMQklrO6J1)%B<#>N3FTb(p7)K<E#*yR6cPpW*#5}qzBAn(4h<h
OFJO#y09R;*zkUP1E}rxN

literal 0
HcmV?d00001

diff --git a/mas/shrodequ_superpos/phys/__pycache__/func.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/func.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..fc4dc392bf6de4a0dcf9838ef0ed4474f570184f
GIT binary patch
literal 777
zcma)4v2NQi5ap4QExWPXI%$A@f;vP%=b|W(b~TD7Kmjd*EXq|HNn}Y1h=FYGha{ta
ziEF3K-8%J-GSC3YP#Qe)9ewv6AEjoqDZ}yQ>!;Ng82d%w;SdNu;I!u$B$Lpvd{`mp
zh<PK<nG{vD<}#A;&nS=gXe7soMthXV38L{HP31F0iG(wj9^7MRwoJjp8$U9Y>rGd3
z%Iyd3W1Ocr?J)+&9NcmzZeR;H?1<gL5j$f$Sf-JOkDhnh3wzmlxY}_K*B-hZGY44s
z=QOKTTV+*KG}&jPzE`EQnJsIjnoJw@?QPZ;n5d1C)sOzd_PR1!*-Y0Po2~n<Jkc93
zw7nc*x8Ee}7|a}({}ue2{S#~=EKDOqcQe$`j4|Xfw!gSC6H1WiA<S^+$8@B$3un*i
z>;;BL-1KI_EK=r>5sl&!4#_0GyeHY|Ap1#{2%T6FpQV4!!5rf9h=jxkk_B$)MFyHJ
z3`FK7rc>@kv9dwslT-;&(T(UHfo=ij6;)31xNHko`?%{{y$SyKDZRr<sXFU?ylzzC
ng84&#QW<I=xEA~i)+O&+CHtm&j~_X+WK6;mNMJ7T7cInJhZ~P<

literal 0
HcmV?d00001

diff --git a/mas/shrodequ_superpos/phys/__pycache__/particle.cpython-310.pyc b/mas/shrodequ_superpos/phys/__pycache__/particle.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..28d76dce2d4e1d29291b3eee5f1ad6a22593da6c
GIT binary patch
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#$

literal 0
HcmV?d00001

diff --git a/mas/shrodequ_superpos/phys/box.py b/mas/shrodequ_superpos/phys/box.py
new file mode 100644
index 0000000..4b0f2fb
--- /dev/null
+++ b/mas/shrodequ_superpos/phys/box.py
@@ -0,0 +1,9 @@
+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
new file mode 100644
index 0000000..f1b529b
--- /dev/null
+++ b/mas/shrodequ_superpos/phys/const.py
@@ -0,0 +1,26 @@
+import numpy as np
+from scipy.constants import Planck, electron_mass 
+
+# constants
+BOX_LENGTH = 0.01 # box length (meters)
+h = Planck # plancks constant
+
+D = np.sqrt(2/BOX_LENGTH) # norm factor
+m = electron_mass # mass
+
+# total energy function
+def energy(n):
+    return (((h*n)/BOX_LENGTH)**2)*(1/(8*m))
+
+# Wave func inner coef
+def k(n):
+    return np.sqrt( (8 * ((np.pi)**2) * m * energy(n))/(h**2) )
+
+# pre calculate energy states
+energy_states = []
+for n in range(0, 11):
+    energy_states.append( energy(n) )
+
+psi_k = []
+for n in range(0, 11):
+    psi_k.append( k(n) )
diff --git a/mas/shrodequ_superpos/phys/func.py b/mas/shrodequ_superpos/phys/func.py
new file mode 100644
index 0000000..a63247e
--- /dev/null
+++ b/mas/shrodequ_superpos/phys/func.py
@@ -0,0 +1,17 @@
+import cmath
+import numpy as np
+from phys.const import *
+
+# wave function
+def psi(x, n) -> float:
+    return D * np.sin( k(n) * x ) 
+
+def psi_opt(x, n) -> float: # optimized version
+    return D * np.sin( psi_k[n] * x )
+
+# probability density function
+def prob(x, n) -> float:
+    return (psi(x, n))**2
+
+def prob_opt(x, n) -> float: # optimized version
+    return abs(psi_opt(x,n)) ** 2
diff --git a/mas/shrodequ_superpos/phys/particle.py b/mas/shrodequ_superpos/phys/particle.py
new file mode 100644
index 0000000..72db7a5
--- /dev/null
+++ b/mas/shrodequ_superpos/phys/particle.py
@@ -0,0 +1,11 @@
+from phys.func import psi, prob, E
+
+class particle:
+    def __init__(self, mass: float, n_state: int):
+        self.mass = mass
+        self.n_state = n_state
+        
+        self.energy = 
+        
+
+