Changeset 792 for codes/icosagcm/devel/Python

Timestamp:

12/10/18 12:48:58 (6 years ago)

Author:

jisesh

Message:

devel/unstructured : added kernel for curl curl ; used by Baroclinic_3D_ullrich

Location:

codes/icosagcm/devel/Python

Files:

• src/kernels_transport.jin (modified) (2 diffs)
• src/unstructured.pyx (modified) (1 diff)
• test/py/Baroclinic_3D_ullrich.py (modified) (2 diffs)

codes/icosagcm/devel/Python/src/kernels_transport.jin

@@ -312 +312 @@
 
 KERNEL(scalar_laplacian)
-
   FORALL_CELLS_EXT()
     ON_EDGES
@@ -330 +329 @@
 
 END_BLOCK
+
+KERNEL(curl_laplacian)
+  FORALL_CELLS_EXT()
+    ON_DUAL
+      etav = 0.d0
+      FORALL_EDGES
+         etav = etav + SIGN*u(EDGE)
+      END_BLOCK
+      qv(DUAL_CELL) = etav/AV
+    END_BLOCK
+  END_BLOCK
+
+  FORALL_CELLS_EXT()
+    ON_EDGES
+       curlcurl(EDGE) = SIGN*(qv(VERTEX1)-qv(VERTEX2))*(1./LE_DE)
+    END_BLOCK
+  END_BLOCK
+
+END_BLOCK

codes/icosagcm/devel/Python/src/unstructured.pyx

@@ -78 +78 @@
                      ['dynamico_caldyn_unstructured', c_num, c_void_p,20],
                      ['dynamico_scalar_laplacian', c_void_p,2],
+                     ['dynamico_curl_laplacian', c_void_p,2],
                      ['dynamico_partition_graph', c_int,2, c_void_p,3, c_int, c_void_p],
                      ['dynamico_init_transfer', c_int, c_int,2,c_void_p,3, c_int,2,c_void_p,3],

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

@@ -201 +201 @@
                 caldyn_step.next()
                 davies.relax(llm, caldyn_step, flow0)
-                m,S = caldyn_step.mass, caldyn_step.theta_rhodz
+                m,S,u = caldyn_step.mass, caldyn_step.theta_rhodz, caldyn_step.u
                 s = S/m
-                laps = mesh.field_mass()
-                bilaps = mesh.field_mass()
+                laps, bilaps = mesh.field_mass(), mesh.field_mass()
+                lapu, bilapu = mesh.field_u(), mesh.field_u()
                 unst.ker.dynamico_scalar_laplacian(s,laps)
                 unst.ker.dynamico_scalar_laplacian(laps,bilaps)
+                unst.ker.dynamico_curl_laplacian(u,lapu)
+                unst.ker.dynamico_curl_laplacian(lapu,bilapu)
                 caldyn_step.theta_rhodz[:] = S - dt*kappa*bilaps*m # Euler step
+                caldyn_step.u[:] = u - dt*kappa*bilapu # Euler step
+
             return (caldyn_step.mass.copy(), caldyn_step.theta_rhodz.copy(), caldyn_step.u.copy(),
                     caldyn_step.geopot.copy(), caldyn_step.W.copy())
@@ -230 +234 @@
         v = mesh.field_mass() # specific volume (diagnosed)
         
-        for i in range(Nslice):
+        for i in range(Nslice+1):
             context.update_calendar(i+1)