Changeset 4172 for TOOLS/MOSAIX/CalvingWeights.py

Timestamp:

12/05/18 11:17:10 (6 years ago)

Author:

omamce

Message:

O.M. :

• More documentation
• Update run-off weights computation
• Update calving weights computation

File:

• TOOLS/MOSAIX/CalvingWeights.py (modified) (8 diffs)

TOOLS/MOSAIX/CalvingWeights.py

@@ -42 +42 @@
 parser.add_argument ('--repartition_file', help='Data files with iceberg melting' , type=str, default='./runoff-icb_DaiTrenberth_Depoorter_eORCA1_JD.nc' )
 parser.add_argument ('--repartition_var' , help='Variable name for iceshelf'      , type=str, default=None)
+parser.add_argument ('--output'  , help='output rmp file name', default='rmp_tlmd_to_torc_calving_64bit.nc' )
+parser.add_argument ('--fmt'     , help='NetCDF file format, using nco syntax', default='64bits', choices=['classic', 'netcdf3', '64bit', '64bit_data', '64bit_data', 'netcdf4', 'netcdf4_classsic'] )
 
 # Parse command line
@@ -63 +65 @@
 
 nb_zone = len(limit_lat)
+
+if myargs.fmt == 'classic'         : FmtNetcdf = 'CLASSIC'
+if myargs.fmt == 'netcdf3'         : FmtNetcdf = 'CLASSIC'
+if myargs.fmt == '64bit'           : FmtNetcdf = 'NETCDF3_64BIT_OFFSET'
+if myargs.fmt == '64bit_data'      : FmtNetcdf = 'NETCDF3_64BIT_DATA'
+if myargs.fmt == '64bit_offset'    : FmtNetcdf = 'NETCDF3_64BIT_OFFSET'
+if myargs.fmt == 'netcdf4'         : FmtNetcdf = 'NETCDF4'
+if myargs.fmt == 'netcdf4_classic' : FmtNetcdf = 'NETCDF4_CLASSIC'
 ### ==========================================================================================
 
@@ -258 +268 @@
         print ('  ')
 
+
+
 ## End of loop
 print (' ')
@@ -265 +277 @@
 print ( 'Total num_links : {:10d}'.format(num_links) )
 
+# Diag : interpolate uniform field
+src_field = np.zeros(shape=(src_grid_size))
+for n_bas in np.arange ( nb_zone ) :
+    index_src = ((src_grid_size - 1)*n_bas) // (nb_zone-1) + 1
+    src_field[index_src-1] = n_bas
+
+dst_field = np.zeros(shape=(dst_grid_size,))
+for link in np.arange (num_links) :
+    dst_field [dst_address [link]-1] +=  remap_matrix[link] * src_field[src_address[link]-1]
+
 ### ===== Writing the weights file, for OASIS MCT ==========================================
 
 # Output file
-calving = 'rmp_t' + src_Name + '_to_' + 't' + dst_Name + '_calving_' + myargs.type + '.nc'
-f_calving = netCDF4.Dataset ( calving, 'w', format='NETCDF3_64BIT' )
-
+calving = myargs.output
+f_calving = netCDF4.Dataset ( calving, 'w', format=FmtNetcdf )
+    
 print ('Output file: ' + calving )
 
@@ -285 +307 @@
 f_calving.title           = calving
 f_calving.Program         = "Generated by " + sys.argv[0] + " with flags " + str(sys.argv[1:])
+f_calving.repartitionType = myargs.type
+if myargs.type in ['iceberg', 'iceshelf' ] :
+    f_calving.repartitionFile = myargs.repartition_file
+    f_calving.repartitionVar  = repartitionVar
+f_calving.gridsFile       = grids
+f_calving.areasFile       = areas
+f_calving.masksFile       = masks
 f_calving.timeStamp       = time.asctime()
 f_calving.uuid            = f_areas.uuid
@@ -293 +322 @@
 f_calving.release         = platform.release()
 f_calving.hardware        = platform.machine()
-f_calving.Comment         = "Preliminary attempt - Do not trust !"
 f_calving.conventions     = "SCRIP"
 if src_name == 'lmd' : f_calving.source_grid = "curvilinear"
@@ -381 +409 @@
 v_dst_repartition[:] = key_repartition.reshape( (nb_zone,dst_grid_size) )
 
-# Additoonnal fields for debugging
+# Additionnal fields for debugging
 # ================================
 v_dst_grid_imaskutil      = f_calving.createVariable ( 'dst_grid_imaskutil'     , 'f8', ('dst_grid_size',)  )
@@ -407 +435 @@
 v_dst_imaskutil_addressed[:] = dst_mskutil.ravel()[dst_address-1].ravel()
 
+v_src_field = f_calving.createVariable ( 'src_field', 'f8', ('src_grid_size',) )
+v_dst_field = f_calving.createVariable ( 'dst_field', 'f8', ('dst_grid_size',) )
+v_src_field[:] = src_field
+v_dst_field[:] = dst_field
+
 f_calving.close ()