Changeset 504

Timestamp:

01/05/17 00:34:26 (7 years ago)

Author:

smasson

Message:

update smallmeshmask.pro

File:

• trunk/SRC/Grid/smallmeshmask.pro (modified) (6 diffs)

trunk/SRC/Grid/smallmeshmask.pro

@@ -66 +66 @@
 ;
 ;-
-PRO smallmeshmask, ncfilein, ncfileout, IODIR = iodir, HGRFILE = hgrfile, ZGRFILE = zgrfile, MSKFILE = mskfile
+PRO smallmeshmask, ncfilein, ncfileout, IODIR = iodir, HGRFILE = hgrfile, ZGRFILE = zgrfile, MSKFILE = mskfile, USEBATHY = usebathy
 ;
   compile_opt idl2, strictarrsubs
@@ -73 +73 @@
   test = (file_search(filein))[0]
   IF test EQ '' THEN BEGIN
-    IF keyword_set(hgrfile) THEN filein_hgr = isafile(FILE = hgrfile, IODIR = iodir) ELSE filein_hgr = (file_search(filein+'.mesh_hgr.nc'))[0]
-    IF keyword_set(zgrfile) THEN filein_zgr = isafile(FILE = zgrfile, IODIR = iodir) ELSE filein_zgr = (file_search(filein+'.mesh_zgr.nc'))[0]
-    IF keyword_set(mskfile) THEN filein_msk = isafile(FILE = mskfile, IODIR = iodir) ELSE filein_msk = (file_search(filein+'.mask.nc'))[0]
+    IF keyword_set(hgrfile) THEN filein_hgr = isafile(FILE = hgrfile, IODIR = iodir) ELSE filein_hgr = (file_search(filein+'[._]mesh_hgr.nc'))[0]
+    IF keyword_set(zgrfile) THEN filein_zgr = isafile(FILE = zgrfile, IODIR = iodir) ELSE filein_zgr = (file_search(filein+'[._]mesh_zgr.nc'))[0]
+    IF keyword_set(usebathy) THEN filein_msk = 'not used' ELSE BEGIN
+      IF keyword_set(mskfile) THEN filein_msk = isafile(FILE = mskfile, IODIR = iodir) ELSE filein_msk = (file_search(filein+'[._]mask.nc'))[0]
+    ENDELSE
     IF filein_hgr EQ '' OR filein_zgr EQ '' OR filein_msk EQ ''  THEN BEGIN
       ras = report(['meshmask file(s) not found...', $
@@ -171 +173 @@
     varid = [varid, ncdf_vardef(cdfidout, 'mbathy', [dimidx, dimidy], /short)]
 ; mask variable
-  msklist = ['tmask', 'umask', 'vmask', 'fmask']
+  IF keyword_set(usebathy) THEN msklist = ['tmask'] ELSE msklist = ['tmask', 'umask', 'vmask', 'fmask']
   FOR m = 0, n_elements(msklist)-1 DO $
     varid = [varid, ncdf_vardef(cdfidout, msklist[m], [dimidx, dimidy, dimidz], /byte)]
@@ -208 +210 @@
   IF (where(zgr_varlist EQ 'mbathy'))[0] NE -1 THEN ncdf_transfer, cdfid, cdfidout, 'mbathy'
 ;
+  bat = -1
   IF keyword_set(key_e3_3d)  THEN BEGIN
 ; get the bottom
@@ -233 +236 @@
   IF keyword_set(key_gdep_3d)  THEN BEGIN
 ; get the bottom
-    CASE (ncdf_varinq(cdfid, 'mbathy')).ndims OF
-      2:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj]
-      3:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj, 1]
-      4:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj, 1, 1]
-    ENDCASE
+    IF bat[0] EQ -1 THEN BEGIN
+      CASE (ncdf_varinq(cdfid, 'mbathy')).ndims OF
+        2:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj]
+        3:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj, 1]
+        4:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj, 1, 1]
+      ENDCASE
+    ENDIF
     bottom = 0L > ( long(bat) - 1L )
     bottom = lindgen(jpi, jpj) + jpi*jpj*temporary(bottom)
@@ -257 +262 @@
 ; mask
 ;
-  IF n_elements(filein_msk) NE 0  THEN BEGIN
-    ncdf_close, cdfid
-    cdfid = ncdf_open(filein_msk)
-  ENDIF
+  IF keyword_set(usebathy) THEN BEGIN
+    IF bat[0] EQ -1 THEN BEGIN
+      CASE (ncdf_varinq(cdfid, 'mbathy')).ndims OF
+        2:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj]
+        3:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj, 1]
+        4:ncdf_varget, cdfid, 'mbathy', bat, count = [jpi, jpj, 1, 1]
+      ENDCASE
+    ENDIF
+    FOR k = 0, jpk-1 DO BEGIN
+      zz = bat GT k
+      ncdf_varput, cdfidout, 'tmask', temporary(zz), offset = [0, 0, k], count = [jpi, jpj, 1]
+    ENDFOR  
+  ENDIF ELSE BEGIN
+    IF n_elements(filein_msk) NE 0  THEN BEGIN
+      ncdf_close, cdfid
+      cdfid = ncdf_open(filein_msk)
+    ENDIF
 ; loop on the vertical levels to limit the memory use
-  FOR k = 0, jpk-1 DO BEGIN
-    FOR m = 0, 3 DO BEGIN
-      CASE (ncdf_varinq(cdfid, msklist[m])).ndims OF
-        3:ncdf_varget, cdfid, msklist[m], zzz, offset = [0, 0, k   ], count = [jpi, jpj, 1]
-        4:ncdf_varget, cdfid, msklist[m], zzz, offset = [0, 0, k, 0], count = [jpi, jpj, 1, 1]
-      ENDCASE
-      ncdf_varput, cdfidout, msklist[m], byte(temporary(zzz)), offset = [0, 0, k], count = [jpi, jpj, 1]
+    FOR k = 0, jpk-1 DO BEGIN
+      FOR m = 0, n_elements(msklist)-1 DO BEGIN
+        CASE (ncdf_varinq(cdfid, msklist[m])).ndims OF
+          3:ncdf_varget, cdfid, msklist[m], zzz, offset = [0, 0, k   ], count = [jpi, jpj, 1]
+          4:ncdf_varget, cdfid, msklist[m], zzz, offset = [0, 0, k, 0], count = [jpi, jpj, 1, 1]
+        ENDCASE
+        ncdf_varput, cdfidout, msklist[m], byte(temporary(zzz)), offset = [0, 0, k], count = [jpi, jpj, 1]
+      ENDFOR
     ENDFOR
-  ENDFOR
+  ENDELSE
 ;------------------------------------------------------
 ;------------------------------------------------------