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

Timestamp:

12/19/17 15:26:51 (7 years ago)

Author:

dubos

Message:

devel/unstructured : bubble test case with Fortran time stepping

File:

• codes/icosagcm/devel/Python/dynamico/meshes.py (modified) (1 diff)

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

@@ -148 +148 @@
                   Aiv,Aiv,f+0.*Aiv,le_de,Riv2,-wee)
 
-    def field_theta(self): return squeeze((self.nqdyn,self.ny,self.nx,self.llm))
-    def field_mass(self): return squeeze((self.ny,self.nx,self.llm))
-    def field_z(self): return squeeze((self.ny,self.nx,self.llm))
-    def field_w(self): return squeeze((self.ny,self.nx,self.llm+1))
-    def field_u(self): return np.zeros((self.ny,2*self.nx,self.llm))
-    def field_ps(self): return squeeze((self.ny,self.nx))
+    def field_theta(self,n=1): return squeeze((n,self.nqdyn,self.ny,self.nx,self.llm))
+    def field_mass(self,n=1): return squeeze((n,self.ny,self.nx,self.llm))
+    def field_z(self,n=1): return squeeze((n,self.ny,self.nx,self.llm))
+    def field_w(self,n=1): return squeeze((n,self.ny,self.nx,self.llm+1))
+    def field_u(self,n=1): 
+        if n==1:
+            return np.zeros((self.ny,2*self.nx,self.llm))
+        else:
+            return np.zeros((n,self.ny,2*self.nx,self.llm))
+    def field_ps(self,n=1): return squeeze((n,self.ny,self.nx))
     def ucomp(self,u): return u[:,range(0,2*self.nx,2),:]
     def set_ucomp(self,uv,u): uv[:,range(0,2*self.nx,2),:]=u