Changeset 805 for codes/icosagcm/devel/Python/dynamico/meshes.py

Timestamp:

02/08/19 08:10:04 (5 years ago)

Author:

dubos

Message:

devel/Python : moving Fortran-based features to dynamico.dev module (TBC)

File:

• codes/icosagcm/devel/Python/dynamico/meshes.py (modified) (11 diffs)

codes/icosagcm/devel/Python/dynamico/meshes.py

@@ -90 +90 @@
     return z.transpose()
 
-class Cartesian_mesh(Base_class):
+class Cartesian_Mesh(Base_class):
     def __init__(self,nx,ny,llm,nqdyn,Lx,Ly,f):
         dx,dy = Lx/nx, Ly/ny
@@ -115 +115 @@
         def zeros(shape,n=1): return [np.zeros(shape) for i in range(n)] if n>1 else np.zeros(shape)
         def indices(shape,n=1): return [np.zeros(shape,np.int32) for i in range(n)] if n>1 else np.zeros(shape,np.int32)
+
+        primal_num, dual_num, edge_num = nx*ny, nx*ny, 2*nx*ny
 
         Aiv, lon_i,lon_v,lat_i,lat_v   = zeros( nx*ny,   5)
@@ -178 +180 @@
 
         Aiv[:]=dx*dy # Ai=Av=dx*dy
-        le_de=le/de
-        unst.init_mesh(llm,nqdyn,2*nx*ny,nx*ny,nx*ny,4,4,4,
-                  primal_deg,primal_edge,primal_ne,
-                  dual_deg,dual_edge,dual_ne,dual_vertex,
-                  left,right,down,up,trisk_deg,trisk,
-                  Aiv,Aiv,f+0.*Aiv,le_de,Riv2,wee)
-        self.set_members(locals(), 'primal_deg', 'primal_edge', 'primal_ne', 'primal_vertex',
+        le_de, fv = le/de, f+0.*Aiv
+        
+        self.set_members(locals(), 'primal_num',  'dual_num', 'edge_num',
+                         'primal_deg', 'primal_edge', 'primal_ne', 'primal_vertex',
                          'dual_deg', 'dual_edge', 'dual_ne', 'dual_vertex',
                          'left','right','down','up',
                          'trisk_deg','trisk','Riv2','wee',
-                         'lon_i','lat_i','lon_v','lat_v',
+                         'lon_i','lat_i','lon_v','lat_v', 'fv',
                          'le_de','le','de','lon_e','lat_e','angle_e',
                          'primal_i','primal_j','edge_i','edge_j')
         self.Ai, self.Av = Aiv, Aiv
+        self.to_dynamico()
+    def to_dynamico(self): pass
     def ncwrite(self, name):
         """The method writes Cartesian mesh on the disc.
@@ -399 +400 @@
 
 class Abstract_Mesh(Base_class):
+    def to_dynamico(self): pass
     def field_theta(self,n=1): return zeros((n,self.nqdyn,self.primal_num,self.llm))
     def field_mass(self,n=1):  return zeros((n,self.primal_num,self.llm))
@@ -436 +438 @@
         self.lon_v, self.lat_v = mapping.map(self.lon_v, self.lat_v)
         self.lon_e, self.lat_e = mapping.map(self.lon_e, self.lat_e)
-    
+
 class Unstructured_Mesh(Abstract_Mesh):
     def __init__(self, gridfile, llm, nqdyn, radius, f):
@@ -466 +468 @@
                          'primal_num', 'dual_num', 'edge_num', 'primal_deg', 'primal_vertex', 'primal_ne',
                          'lat_i', 'lon_i', 'lat_v', 'lon_v', 'lat_e', 'lon_e', 'angle_e',
-                         'Ai', 'Av', 'fv', 'le', 'de', 'le_de',
+                         'Ai', 'Av', 'fv', 'le', 'de', 'down', 'up', 'le_de',
                          'trisk_deg', 'trisk', 'wee', 'Riv2', 'dual_ne',
                          'left', 'right',
                          'primal_edge')
         gridfile.normalize(self)
-        max_primal_deg, max_dual_deg, max_trisk_deg = [ x.shape[1] for x in primal_edge, dual_edge, trisk]
-        unst.init_mesh(llm, nqdyn, edge_num, primal_num,dual_num,
-                  max_trisk_deg, max_primal_deg, max_dual_deg,
-                  primal_deg,primal_edge,primal_ne,
-                  dual_deg,dual_edge,dual_ne,dual_vertex,
-                  left,right,down,up,trisk_deg,trisk,
-                  self.Ai,self.Av,self.fv,le_de,Riv2,wee)
+        self.to_dynamico()
         self.primal  = tri.Triangulation(lon_i*radian, lat_i*radian)
         self.dual    = tri.Triangulation(lon_v*radian, lat_v*radian)
@@ -604 +600 @@
         for edge in range(edge_num): 
             if edge_deg[edge]<2: up[edge]=down[edge]
-        max_primal_deg, max_dual_deg, max_trisk_deg = [x.shape[1] for x in primal_edge, dual_edge, trisk]
 
         # construct own primal mesh for XIOS output
@@ -638 +633 @@
                          'trisk_deg', 'trisk', 'wee',
                          'dual_deg', 'dual_edge', 'dual_ne', 'dual_vertex', 'Riv2',
-                         'left','right','primal_deg','primal_edge','primal_ne',
+                         'left', 'right', 'down', 'up', 
+                         'primal_deg', 'primal_edge', 'primal_ne',
                          'vertices_V1', 'edges_E2',
                          'lon_i', 'lat_i', 'Ai',
@@ -651 +647 @@
         self.fv = mapping.coriolis(self.lon_v,self.lat_v) # Coriolis parameter
         # propagate info to Fortran side
-        unst.init_mesh(llm, nqdyn, self.edge_num, self.primal_num, self.dual_num,
-                  max_trisk_deg, max_primal_deg, max_dual_deg,
-                  primal_deg,primal_edge,primal_ne,
-                  dual_deg,dual_edge,dual_ne,dual_vertex,
-                  left,right,down,up,trisk_deg,trisk,
-                  self.Ai,self.Av,self.fv,self.le_de,self.Riv2,self.wee)
-
+        self.to_dynamico()
         # setup halo transfers - NB : llm must be set before we call set_dynamico_transfer
         self.com_primal = parallel.Halo_Xchange(
@@ -767 +757 @@
     return bounds
 
-class Stencil_glob:
+class Stencil_glob(object):
     def __init__(self, degree, neigh, weight=None):
         self.degree, self.neigh, self.weight = degree, neigh, weight
@@ -773 +763 @@
         return Stencil(cells, self.degree, self.neigh, neigh_dict, self.weight)
             
-class Stencil:
+class Stencil(object):
     def __init__(self, cells, degree, neigh, neigh_dict, weight=None):
         get = degree.dim.getter(cells)