Second try for optimisation of initial conditions

graph_opti.py

@@ -9,20 +9,22 @@ def couche_lim(eta, f):
 
 def optim(Pr, f_2_0,th_1_0):
     n = 7
-    sol = solve_ivp(couche_lim, (0,40), [0,0,f_2_0,1,th_1_0])
-    while abs(sol.y[1][-1]) > 0.001:
-        if n > 50 :
-            print("n > 50")
+    sol = solve_ivp(couche_lim, (0,30), [0,0,f_2_0,1,th_1_0])
+    while abs(sol.y[1][-1]) > 0.005 or abs(sol.y[3][-1]) > 0.005:
+        if n > 300 :
+            print("Too much iterations")
             break
-        
-        if sol.y[1][-1] > 0.001:
-            f_2_0 *= 1 - 1/(2**(n+1))
+        if sol.y[3][-1] > 0.005:
+            th_1_0 *= 1 + 1/(2**n)
+        elif sol.y[3][-1] < -0.005:
             th_1_0 *= 1 - 1/(2**n)
-        elif sol.y[1][-1] < -0.001:
+        sol = solve_ivp(couche_lim, (0,30), [0,0,f_2_0,1,th_1_0])
+                
+        if sol.y[3][-1] > 0.005:
+            f_2_0 *= 1 - 1/(2**(n+1))
+        elif sol.y[1][-1] < -0.005:
             f_2_0 *= 1 + 1/(2**(n+1))
-            th_1_0 *= 1 + 1/(2**n)
-            
-        sol = solve_ivp(couche_lim, (0,40), [0,0,f_2_0,1,th_1_0])
+        sol = solve_ivp(couche_lim, (0,30), [0,0,f_2_0,1,th_1_0])
         n += 1
     return sol
     
@@ -37,4 +39,5 @@ if __name__ == "__main__":
     mpl.plot(sol.t,sol.y[1], label = "f'(η)")
     mpl.plot(sol.t,sol.y[3], label = "θ(η)")
     mpl.legend()
-    mpl.show()
\ No newline at end of file
+    mpl.show()
+    print(sol.y[1][-1],sol.y[3][-1])
\ No newline at end of file