script to plot abs(R) versus k for a different M at one specific lambda_s/lambda_f ; marked positions in stability plots by a cross

Author: Daniel Ruprecht

examples/fwsw/plot_stab_vs_k.py

@@ -16,9 +16,13 @@
 
 if __name__ == "__main__":
   mvals = [2, 3, 4]
-  kvals = np.arange(2,10)
-  lambda_fast = 15j
-  lambda_slow = 3j
+  kvals = np.arange(1,10)
+  lambda_fast = 10j
+  slow_resolved = False
+  if slow_resolved:
+    lambda_slow = 1j
+  else:
+    lambda_slow = 4j
   stabval = np.zeros((np.size(mvals), np.size(kvals)))
 
   for i in range(0,np.size(mvals)):
@@ -68,18 +72,25 @@
   rcParams['figure.figsize'] = 2.5, 2.5
   fig = plt.figure()
   fs = 8
-  plt.plot(kvals, stabval[0,:], 'o-', color='b', label=("M=%2i" % mvals[0]))
-  plt.plot(kvals, stabval[1,:], 's-', color='r', label=("M=%2i" % mvals[1]))
-  plt.plot(kvals, stabval[2,:], 'd-', color='g', label=("M=%2i" % mvals[2]))
+  plt.plot(kvals, stabval[0,:], 'o-', color='b', label=("M=%2i" % mvals[0]), markersize=fs-2)
+  plt.plot(kvals, stabval[1,:], 's-', color='r', label=("M=%2i" % mvals[1]), markersize=fs-2)
+  plt.plot(kvals, stabval[2,:], 'd-', color='g', label=("M=%2i" % mvals[2]), markersize=fs-2)
   plt.plot(kvals, 1.0+0.0*kvals, '--', color='k')
   plt.xlabel('Number of iterations K', fontsize=fs)
   plt.ylabel(r'Modulus of stability function $\left| R \right|$', fontsize=fs)
   plt.ylim([0.0, 1.2])
-  plt.legend(loc='lower left', fontsize=fs, prop={'size':fs})
+  if slow_resolved:
+    plt.legend(loc='upper right', fontsize=fs, prop={'size':fs})
+  else:
+    plt.legend(loc='lower left', fontsize=fs, prop={'size':fs})
+
   plt.gca().get_xaxis().get_major_formatter().labelOnlyBase = False
   plt.gca().get_xaxis().set_major_formatter(ScalarFormatter())
-  plt.show()
-
-#  filename = 'stablimit-M'+str(mvals[0])+'.pdf'
-#  fig.savefig(filename, bbox_inches='tight')
-#  call(["pdfcrop", filename, filename])
+  #plt.show()
+  if slow_resolved:
+    filename = 'stab_vs_k_resolved.pdf'
+  else:
+    filename = 'stab_vs_k_unresolved.pdf'
+  
+  fig.savefig(filename, bbox_inches='tight')
+  call(["pdfcrop", filename, filename])

examples/fwsw/plot_stab_vs_m.py

@@ -17,8 +17,8 @@
 if __name__ == "__main__":
   mvals = np.arange(2,10)
   kvals = [3, 5, 7]
-  lambda_fast = 8j
-  lambda_slow = 4j
+  lambda_fast = 10j
+  lambda_slow = 3j
   stabval = np.zeros((np.size(kvals), np.size(mvals)))
 
   for i in range(0,np.size(mvals)):

examples/fwsw/plot_stability.py

@@ -35,10 +35,10 @@
     swparams['collocation_class'] = collclass.CollGaussRadau_Right
 
     ### SET NUMBER OF QUADRATURE NODES ###
-    swparams['num_nodes'] = 4
+    swparams['num_nodes'] = 3
 
     ### SET NUMBER OF ITERATIONS - SET K=0 FOR COLLOCATION SOLUTION ###
-    K = 3
+    K = 5
 
     do_coll_update = True
 
@@ -89,6 +89,9 @@
 #    levels = np.array([1.0])
     CS1 = plt.contour(lambda_s.imag, lambda_f.imag, np.absolute(stab), levels, colors='k', linestyles='dashed')
     CS2 = plt.contour(lambda_s.imag, lambda_f.imag, np.absolute(stab), [1.0],  colors='k')
+    # Set markers at points used in plot_stab_vs_k
+    plt.plot(4, 10, 'x', color='k', markersize=fs-4)
+    plt.plot(1, 10, 'x', color='k', markersize=fs-4)
     plt.clabel(CS1, inline=True, fmt='%3.2f', fontsize=fs-2)
     manual_locations = [(1.5, 2.5)]
     if K>0: # for K=0 and no 1.0 isoline, this crashes Matplotlib for somer reason