Changeset 681 for codes/icosagcm/devel/Python/test

Timestamp:

02/14/18 00:06:52 (6 years ago)

Author:

dubos

Message:

devel/unstructured : local mesh setup + halo exchange

File:

• codes/icosagcm/devel/Python/test/py/RSW2_MPAS_W02.py (modified) (7 diffs)

codes/icosagcm/devel/Python/test/py/RSW2_MPAS_W02.py

@@ +1 @@
+print 'Starting'
+
+from mpi4py import MPI
+comm = MPI.COMM_WORLD
+mpi_rank, mpi_size = comm.Get_rank(), comm.Get_size()
+print '%d/%d starting'%(mpi_rank,mpi_size)
+prefix='fig_RSW2_MPAS_W02/%02d'%mpi_rank
+
 print 'Loading DYNAMICO modules ...'
 from dynamico import unstructured as unst
-from dynamico.meshes import MPAS_Format, Unstructured_Mesh as Mesh
+from dynamico.meshes import MPAS_Format, Unstructured_PMesh as PMesh, Local_Mesh as Mesh
 from dynamico import time_step
 print '...Done'
@@ -11 +19 @@
 print '...Done'
 
-grid, llm, nqdyn = 10242, 1,1 # 2562, 10242, 40962
+grid, llm, nqdyn = 2562, 1,1 # 2562, 10242, 40962
 Omega, radius, g, gh0 = 2.*np.pi/86400., 6.4e6, 1., 2.94e4
-N, T, courant = 40, 10800., 1.2 # simulation length = N*T
+N, T, courant = 40, 10400., 1.2 # simulation length = N*T
 
 print 'Omega, planetary PV', Omega, 2*Omega/gh0
@@ -20 +28 @@
 print 'Reading MPAS mesh ...'
 meshfile = MPAS_Format('grids/x1.%d.grid.nc'%grid)
-mesh = Mesh(meshfile, llm, nqdyn, radius, f)
+pmesh = PMesh(comm,meshfile)
+mesh = Mesh(pmesh, llm, nqdyn, radius, f)
 print '...Done'
 lon, lat = mesh.lon_i, mesh.lat_i
@@ -30 +39 @@
 dx = mesh.de.min()
 dt = courant*dx/c0
+print dx, dt
 nt = int(math.ceil(T/dt))
 dt = T/nt
@@ -61 +71 @@
 flow=gh,u
 
+unst.ker.dynamico_update_halo(mesh.com_edges.index, 1, u.shape[0], u) 
+                                  
 for i in range(20):
     if True:
@@ -67 +79 @@
         print i, gh.shape, gh.min(), gh.max(), u.min(), u.max()
         mesh.plot_i(gh-Phi0) ; plt.title('err(gh)');
-        plt.savefig('fig_RSW2_MPAS_W02/err_gh_%02d.png'%i)
+        plt.savefig('%s_err_gh_%02d.png'%(prefix,i))
         plt.close()
     else:
@@ -100 +112 @@
         ax8.scatter(u_nowref,u_now-u_nowref) ; ax8.set_title('u_now');
         f.subplots_adjust(hspace=1.)
-        plt.savefig('fig_RSW2_MPAS_W02/scatter_%02d.png'%i)
+        plt.savefig('%s_scatter_%02d.png'%(prefix,i))
         plt.close()