fixed error in including friction

with_c_lsa.py

@@ -19,17 +19,18 @@ ax.set_xlabel(r'$q$')
 ax.set_ylabel(r'$Re[\, \lambda(q) \, ]_{\rm max}$')
 ax.set_ylabel(r'$Re[\, \lambda(q) \, ]_{\rm max}$')
 ax.axhline(y=0, color='black')
 ax.axhline(y=0, color='black')
 
 
-lamda0 = 3.0
-K0 = 1.0
-krho0 = 3.0
-eta0 = 3.0
+lamda0 = 6.0
+K0 = 4.0
+krho0 = 1.0
+eta0 = 4.0
 rho00 = 1.0
 rho00 = 1.0
 Dc0 = 1.0
 Dc0 = 1.0
 Drho0 = 0.0
 Drho0 = 0.0
 kc0 = 1.0
 kc0 = 1.0
 c00 = 1.0
 c00 = 1.0
+gamma0 = 1.0
 
 
-lamda = np.array([largest_real_eigval(q, K = K0, lamda = lamda0, eta = eta0, krho = krho0, rho0 = rho00, c0 = c00, kc = kc0, Drho = Drho0, Dc = Dc0) for q in kyu])
+lamda = np.array([largest_real_eigval(q, K = K0, lamda = lamda0, eta = eta0, krho = krho0, rho0 = rho00, c0 = c00, kc = kc0, Drho = Drho0, Dc = Dc0, gamma = gamma0) for q in kyu])
 lambda_plot, = ax.plot(kyu, lamda)
 lambda_plot, = ax.plot(kyu, lamda)
 lamda_min, lamda_max = min(0, lamda.min()), max(0, lamda.max())
 lamda_min, lamda_max = min(0, lamda.min()), max(0, lamda.max())
 ax.set_ylim(lamda_min, lamda_max)
 ax.set_ylim(lamda_min, lamda_max)
@@ -44,6 +45,7 @@ ax_Dc  = plt.axes([0.7, 0.20, 0.2, 0.02])
 ax_kc = plt.axes([0.1, 0.10, 0.2, 0.02])
 ax_kc = plt.axes([0.1, 0.10, 0.2, 0.02])
 ax_c0 = plt.axes([0.4, 0.10, 0.2, 0.02])
 ax_c0 = plt.axes([0.4, 0.10, 0.2, 0.02])
 ax_Drho = plt.axes([0.7, 0.10, 0.2, 0.02])
 ax_Drho = plt.axes([0.7, 0.10, 0.2, 0.02])
+ax_gamma = plt.axes([0.7, 0.05, 0.2, 0.02])
 
 
 # sliders for controlling parameters
 # sliders for controlling parameters
 s_K  = Slider(ax_K, r'$K$', valmin = 0.0, valmax = 10.0, valinit = K0, valstep = 0.001)
 s_K  = Slider(ax_K, r'$K$', valmin = 0.0, valmax = 10.0, valinit = K0, valstep = 0.001)
@@ -55,10 +57,11 @@ s_Dc  = Slider(ax_Dc, r'$D_c$', valmin = 0.0, valmax = 10.0, valinit = Dc0, vals
 s_kc  = Slider(ax_kc, r'$k_c$', valmin = 0.0, valmax = 10.0, valinit = kc0, valstep = 0.001)
 s_kc  = Slider(ax_kc, r'$k_c$', valmin = 0.0, valmax = 10.0, valinit = kc0, valstep = 0.001)
 s_c0  = Slider(ax_c0, r'$c_0$', valmin = 0.0, valmax = 10.0, valinit = c00, valstep = 0.001)
 s_c0  = Slider(ax_c0, r'$c_0$', valmin = 0.0, valmax = 10.0, valinit = c00, valstep = 0.001)
 s_Drho  = Slider(ax_Drho, r'$D_\rho{}$', valmin = 0.0, valmax = 10.0, valinit = Drho0, valstep = 0.001)
 s_Drho  = Slider(ax_Drho, r'$D_\rho{}$', valmin = 0.0, valmax = 10.0, valinit = Drho0, valstep = 0.001)
+s_gamma  = Slider(ax_gamma, r'$\gamma$', valmin = 0.0, valmax = 10.0, valinit = gamma0, valstep = 0.001)
 
 
 # slider update function
 # slider update function
 def update(val):
 def update(val):
-    lamda = np.array([largest_real_eigval(q, K=s_K.val, lamda = s_lamda.val, eta = s_eta.val, krho = s_krho.val, rho0 = s_rho0.val, c0 = s_c0.val, kc = s_kc.val, Drho = s_Drho.val, Dc = s_Dc.val)
+    lamda = np.array([largest_real_eigval(q, K=s_K.val, lamda = s_lamda.val, eta = s_eta.val, krho = s_krho.val, rho0 = s_rho0.val, c0 = s_c0.val, kc = s_kc.val, Drho = s_Drho.val, Dc = s_Dc.val, gamma = s_gamma.val)
                             for q in kyu])
                             for q in kyu])
     lambda_plot.set_ydata(lamda)
     lambda_plot.set_ydata(lamda)
     lamda_min, lamda_max = min(0, lamda.min()), max(0, lamda.max())
     lamda_min, lamda_max = min(0, lamda.min()), max(0, lamda.max())
@@ -74,5 +77,6 @@ s_Dc.on_changed(update)
 s_Drho.on_changed(update)
 s_Drho.on_changed(update)
 s_kc.on_changed(update)
 s_kc.on_changed(update)
 s_c0.on_changed(update)
 s_c0.on_changed(update)
+s_gamma.on_changed(update)
 
 
 plt.show()
 plt.show()