Changeset 440 for trunk/SRC/ToBeReviewed/CALCULS/moyenne.pro

Timestamp:

12/21/10 14:26:05 (14 years ago)

Author:

smasson

Message:

take into account the ssh in moyenne and grossemoyenne

File:

• trunk/SRC/ToBeReviewed/CALCULS/moyenne.pro (modified) (8 diffs)

trunk/SRC/ToBeReviewed/CALCULS/moyenne.pro

@@ -45 +45 @@
 ; To make an integral rather than an average
 ;
+; @keyword SSH
+; array giving the sea level elenation (in meter) that must be added
+; to the thickness of the first level when computing vertical mean
+;
 ; @keyword WDEPTH
 ; to specify that the field is at W depth instead of T
@@ -73 +77 @@
 ;-
 FUNCTION moyenne, tab, direc, BOXZOOM=boxzoom, INTEGRATION=integration, KEEPBOTTOM = keepbottom $
-                , NAN=nan, NODOMDEF=nodomdef, WDEPTH=wdepth, _EXTRA=ex
+                , NAN=nan, NODOMDEF=nodomdef, SSH = ssh, WDEPTH=wdepth, _EXTRA=ex
 ;
   compile_opt idl2, strictarrsubs
@@ -143 +147 @@
   IF vargrid EQ 'W' THEN wdepth = 1
   grille, mask, glam, gphi, gdep, nx, ny, nz, firstx, firsty, firstz, lastx, lasty, lastz, e1, e2, e3, WDEPTH = wdepth
+;---------------------------------------------------------------
+; check the use of ssh keyword...
+;---------------------------------------------------------------
+  IF keyword_set(ssh) AND dirz NE 0 AND firstz EQ 0 AND taille[0] GE 2 THEN BEGIN
+    szssh = size(ssh)
+    IF szssh[1] NE taille[1] OR szssh[2] NE taille[2] THEN $
+       return, report('ssh and input arrays must have the same size')
+    sshh = ssh
+  ENDIF
 ;------------------------------------------------------------
 ;------------------------------------------------------------
@@ -317 +330 @@
 ;---------------------------------------------------------------
     case 1 of
-      taille[1] eq jpi and taille[2] eq jpj and taille[3] eq jpk: $
+      taille[1] eq jpi and taille[2] eq jpj and taille[3] eq jpk:BEGIN
         res = tab[firstx:lastx, firsty:lasty, firstz:lastz]
-      taille[1] eq jpi and taille[2] eq jpj and taille[3] eq nz: $
+        IF keyword_set(sshh) THEN sshh = sshh[firstx:lastx, firsty:lasty]
+      END
+      taille[1] eq jpi and taille[2] eq jpj and taille[3] eq nz:BEGIN 
         res = tab[firstx:lastx, firsty:lasty, *]
+        IF keyword_set(sshh) THEN sshh = sshh[firstx:lastx, firsty:lasty]
+      END
       taille[1] EQ  nx and taille[2] eq  ny and taille[3] eq nz :res = tab
       taille[1] EQ  nx and taille[2] eq  ny and taille[3] eq jpk : $
@@ -424 +441 @@
         e33 = replicate(1, 1.*nx*ny)#e3
         e33 = reform(e33, nx, ny, nz, /over)
+        IF keyword_set(sshh) THEN e33[*, *, 0] = e33[*, *, 0] + sshh
         IF keyword_set(key_partialstep) AND bottom[0] NE -1 THEN BEGIN
           IF keyword_set(wdepth) THEN $
@@ -476 +494 @@
         e133 = e1[*]#e3
         e133 = reform(e133, nx, ny, nz, /over)
+        IF keyword_set(sshh) THEN e133[*, *, 0] = e133[*, *, 0] + sshh*e1
         IF keyword_set(key_partialstep) AND bottom[0] NE -1 THEN BEGIN
           IF keyword_set(wdepth) THEN $
@@ -500 +519 @@
         e233 = e2[*]#e3
         e233 = reform(e233, nx, ny, nz, /over)
+        IF keyword_set(sshh) THEN e233[*, *, 0] = e233[*, *, 0] + sshh*e2
         IF keyword_set(key_partialstep) AND bottom[0] NE -1 THEN BEGIN
           IF keyword_set(wdepth) THEN $
@@ -524 +544 @@
         e1233 = (e1*e2)[*]#e3
         e1233 = reform(e1233, nx, ny, nz, /over)
+        IF keyword_set(sshh) THEN e1233[*, *, 0] = e1233[*, *, 0] + sshh*e1*e2
         IF keyword_set(key_partialstep) AND bottom[0] NE -1 THEN BEGIN
           IF keyword_set(wdepth) THEN $