Changeset 779 for codes/icosagcm/devel/Python/test/py/reorder.py

Timestamp:

11/16/18 14:03:09 (6 years ago)

Author:

dubos

Message:

devel/unstructured : read dual_ne from mesh file when parallel

File:

• codes/icosagcm/devel/Python/test/py/reorder.py (modified) (5 diffs)

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

@@ -1 +1 @@
 print 'Loading DYNAMICO modules ...'
-from dynamico.meshes import morton_index, key_to_perm, reorder_indices, reorder_values
+from dynamico.meshes import morton_index, key_to_perm, reorder_indices, reorder_values, MPAS_Format, Unstructured_Mesh
 print '...Done'
 import numpy as np
@@ -7 +7 @@
 import time
 
-grid = 10242 # 2562, 10242, 40962
+#------------ Read command-line arguments ----------#
+
+import argparse
+parser = argparse.ArgumentParser()
+parser.add_argument("-n", type=int, default=2562,
+                    help="number of primal cells in MPAS mesh")
+args = parser.parse_args()
+grid = args.n
 
 #-------------- read MPAS mesh ----------------#
 
-print 'Reading MPAS mesh ...'
-mesh=cdf.Dataset('grids_MPAS/x1.%d.grid.nc'%grid, "r")
+meshfile='grids_MPAS/x1.%d.grid.nc'%grid
+print 'Reading MPAS mesh %s ...'%meshfile
+
+def f(lon,lat): return 0.*lon
+mesh=MPAS_Format(meshfile)
+mesh=Unstructured_Mesh(mesh, 1, 1, 1., f)
+primal_ne, dual_ne = mesh.primal_ne, mesh.dual_ne
+
+mesh=cdf.Dataset(meshfile, "r")
 
 def getdims(*names): return [len(mesh.dimensions[name]) for name in names]
@@ -53 +67 @@
 for idx in primal_edge, left_right, dual_edge, down_up, primal_vertex, dual_vertex, trisk : idx[:]=idx[:]-1
 
-reorder_indices(perm_i, primal_deg, primal_vertex, primal_edge, lon_i, lat_i, Ai)
-reorder_indices(perm_v, dual_vertex, dual_edge, lon_v, lat_v, Av, Riv2)
+reorder_indices(perm_i, primal_deg, primal_vertex, primal_edge, primal_ne, lon_i, lat_i, Ai)
+reorder_indices(perm_v, dual_vertex, dual_edge, dual_ne, lon_v, lat_v, Av, Riv2)
 reorder_indices(perm_e, trisk_deg, left_right, down_up, trisk, lon_e, lat_e, angle_e, le, de, wee)
 reorder_values(perm_i, dual_vertex, left_right)
@@ -66 +80 @@
 
 mesh=cdf.Dataset('grids/x1.%d.grid.nc'%grid, "r+")
+
+# create new variables dual_ne, primal_ne
+mesh.createVariable('signOnCell','i4',('nCells','maxEdges'))
+mesh.createVariable('signOnVertex','i4',('nVertices','vertexDegree'))
 
 def putvars(datas,names): 
@@ -83 +101 @@
 putvars( (lat_e,lon_e,angle_e), ('latEdge','lonEdge','angleEdge') )
 putvars( (wee,Riv2), ('weightsOnEdge','kiteAreasOnVertex') )
+putvars( (primal_ne,dual_ne), ('signOnCell','signOnVertex') )
 
 mesh.close()