diff options
Diffstat (limited to 'eq_gen/constraint_gen.py')
-rw-r--r-- | eq_gen/constraint_gen.py | 19 |
1 files changed, 14 insertions, 5 deletions
diff --git a/eq_gen/constraint_gen.py b/eq_gen/constraint_gen.py index b7033e8..ac403bb 100644 --- a/eq_gen/constraint_gen.py +++ b/eq_gen/constraint_gen.py @@ -67,18 +67,24 @@ for diff0 in range(psi_funcs.shape[0]): r, theta, phi = coords x, a, L, r_0 = sympy.symbols('x a L r_0') -r_0 = 20 + +#r_0 = 20 +r_0 = 0 +L = 1 + kappa_p = sympy.sqrt(143 / sympy.pi) / 12288 -c_n = (((225 / (8 * sympy.pi)) ** 2)) ** (sympy.Rational(1, 4)) Im2 = sympy.Piecewise((a * kappa_p * (L ** 5) * ((1 - ((x / L) ** 2)) ** 6), sympy.Abs(x) < L), (0, True)) + +#c_n = (((225 / (8 * sympy.pi)) ** 2)) ** (sympy.Rational(1, 4)) #Im2 = a * c_n * (L ** 5) * sympy.exp(-(x ** 2) / (L ** 2)) + Im1 = Im2.diff(x) I0 = Im1.diff(x) Ip1 = I0.diff(x) Ip2 = Ip1.diff(x) -v = r# - r_0 +v = r - r_0 u = -r - r_0 Im2v = Im2.subs(x, v) @@ -93,11 +99,14 @@ I0u = I0.subs (x, u) Ip1u = Ip1.subs(x, u) Ip2u = Ip2.subs(x, u) -Ip1v = sympy.Function('Ip1v')(v) -Ip2v = Ip1v.diff(v) +#Ip1v = sympy.Function('Ip1v')(v) +#Ip2v = Ip1v.diff(v) eta = ((Ip2v / r - 2 * Ip1v / (r ** 2)) * (sympy.sin(theta) ** 2)).simplify() K_rtheta = ((Ip2v / (r ** 2) - 3 * Ip1v / (r ** 3) + 6 * I0v / (r ** 4) - 6 * Im1v / (r ** 5)) * r * sympy.sin(2 * theta)).simplify() +sympy.pretty_print(eta.simplify()) +sys.exit(0) + #eta = (((Ip2u - Ip2v) / r + 2 * (Ip1u + Ip1v) / (r ** 2)) * (sympy.sin(theta) ** 2)).simplify() #eta = 0 #K_rtheta = (((Ip2u + Ip2v) / (r ** 2) + 3 * (Ip1u - Ip1v) / (r ** 3) + |