Changeset 4083

Timestamp:

10/22/18 16:10:24 (6 years ago)

Author:

omamce

Message:

O.M. : add lbc_mask to nemo.py

File:

• TOOLS/MOSAIX/nemo.py (modified) (2 diffs)

TOOLS/MOSAIX/nemo.py

@@ -22 +22 @@
 
 def lbc (ptab, nperio=6, cd_type='T', psgn=1.0) :
+    # Set periodicity on input fields
     """
     ptab      : Input array
@@ -91 +92 @@
 
     return ptab
+
+
+def lbc_mask (ptab, nperio=6, cd_type='T', psgn=1.0) :
+    # Mask fields on duplicate points
+    """
+    ptab      : Input array
+      rank 2 at least : patb[...., lat, lon]
+    nperio    : Type of periodicity
+      1, 4, 6 : Cyclic on i dimension (generaly longitudes)
+      2       : Obsolete (was symmetric condition at southern boundary ?)
+      3, 4    : North fold T-point pivot (legacy ORCA2)
+      5, 6    : North fold F-point pivot (ORCA1, ORCA025, ORCA2 with new grid for paleo)
+    cd_type   : Grid specification : T, U, V or F
+    psgn      : For change of sign for vector components
+    
+    See NEMO documentation for further details
+    """
         
+    jpi = ptab.shape[-1]
+    #
+    zero = 0
+    
+    #
+    #> East-West boundary conditions
+    # --------------------------------
+
+    if nperio in [1, 4, 6] :
+        # ... cyclic
+        ptab [...,:,  0] = zero
+        ptab [...,:, -1] = zero
+   
+    #
+    #> North-South boundary conditions
+    # ----------------------------------
+    if nperio in [3, 4] :  # North fold T-point pivot     
+        if cd_type in [ 'T', 'W' ] : # T-, W-point
+            ptab[..., -1, 1:       ] = zero
+            ptab[..., -1, 0        ] = zero 
+            ptab[..., -2, jpi//2:  ] = zero
+                      
+        if cd_type == 'U' :
+            ptab[..., -1, 0:-1     ] = zero  
+            ptab[..., -1,  0       ] = zero
+            ptab[..., -1, -1       ] = zero 
+            ptab[..., -2, jpi//2-1:] = zero
+            
+        if cd_type == 'V' : 
+            ptab[..., -2, 1:       ] = zero
+            ptab[..., -1, 1:       ] = zero   
+            ptab[..., -1, 0        ] = zero
+            
+        if cd_type == 'F' :
+            ptab[..., -2, 0:-1     ] = zero
+            ptab[..., -1, 0:-1     ] = zero
+            ptab[..., -1,  0       ] =zero
+            ptab[..., -1, -1       ] = zero
+      
+    if nperio in [5, 6] :            #  North fold F-point pivot  
+        if cd_type in ['T', 'W']  :
+            ptab[..., -1, 0:       ] = zero
+                
+        if cd_type == 'U' :
+            ptab[..., -1, 0:-1     ] = zero       
+            ptab[..., -1, -1       ] = zero
+             
+        if cd_type == 'V' :
+            ptab[..., -1, 0:       ] = zero
+            ptab[..., -2, jpi/2:   ] = zero
+                             
+        if cd_type == 'F' :
+            ptab[..., -1, 0:-1     ] = zero
+            ptab[..., -1, -1       ] = zero
+            ptab[..., -2, jpi//2:-1] = zero
+
+    return ptab
 ## ===========================================================================
 ##