Ticket #1761: make_INITICE.py

#!/opt/local/bin/python

import os,sys
from netCDF4 import Dataset as netcdf
import numpy as np
import matplotlib.pyplot as plt
from math import exp

resname=''

# input file
###fcoord='coordinates_'+str(resname)+'.nc'
#fcoord='mesh_mask_'+str(resname)+'.nc'
fcoord='mesh_mask.nc'

# output file
#fflx='initice_'+str(resname)+'.nc'
fflx='initice.nc'

print '   creating init ice file  ' +fflx

# Reading coordinates file
nccoord=netcdf(fcoord,'r')
nav_lon=nccoord.variables['nav_lon']
nav_lat=nccoord.variables['nav_lat']
LON1= nav_lon.shape[1]
LAT1= nav_lon.shape[0]
print 'nav_lon.shape[1]' ,nav_lon.shape[1]
print 'LON1 ', LON1
print 'LAT1 ', LAT1

# Creating INITICE netcdf file
nc=netcdf(fflx,'w')
nc.createDimension('y',LAT1)
nc.createDimension('x',LON1)
nc.createDimension('time_counter',0)    # Setting dimension size to 0 or None makes it unlimited. 

cdflon=nc.createVariable('nav_lon','f',('y','x'))
cdflat=nc.createVariable('nav_lat','f',('y','x'))
cdftimecounter=nc.createVariable('time_counter','f',('time_counter'))
# ati : Fraction of open waters in sea ice - units %
# hti : Sea ice thickness - units m
# hts : Snow thickness - units m
# smi :
# tmi : Sea ice internal temperature - units K
# tsu : Sea ice surface temperature - units K
#
# Take constant values from namelist &namiceini of NEMO
rn_hts_ini=0.3            #  initial real snow thickness (m)
rn_hti_ini=3.0            #  initial real ice thickness  (m)
rn_ati_ini=0.9            #  initial ice concentration   (-)
rn_smi_ini=6.3            #  initial ice salinity     (g/kg)
rn_tmi_ini=270.           #  initial ice/snw temperature (K)
rn_tsu_ini=270.           #  initial sea ice temperature (K)
#
cdfati=nc.createVariable('ati','f',('time_counter','y','x'))
cdfati.units='Percentage'
cdfati.long_name='Fraction of open waters in sea ice'
cdfhti=nc.createVariable('hti','f',('time_counter','y','x'))
cdfhti.long_name='Sea ice thickness'
cdfhti.units='m'
cdfhts=nc.createVariable('hts','f',('time_counter','y','x'))
cdfhts.long_name='Snow thickness'
cdfhts.units='m'
cdfsmi=nc.createVariable('smi','f',('time_counter','y','x'))
cdfsmi.long_name=''
cdfsmi.units=''
cdftmi=nc.createVariable('tmi','f',('time_counter','y','x'))
cdftmi.long_name='Sea ice internal temperature'
cdftmi.units='Kelvin'
cdftsu=nc.createVariable('tsu','f',('time_counter','y','x'))
cdftsu.long_name='Sea ice surface temperature'
cdftsu.units='Kelvin'

cdflon[:,:]=nav_lon[:,:]
cdflat[:,:]=nav_lat[:,:]

# Fill fields
#print 'cdfati[:,1]', cdfati[:,1]  -> 32 values
## Add a gaussian for sea ice thickness here
cdfhti[:,:,:]=0.
A=7.     # max of gaussian for sea ice thichness in meters
a=-0.1
xshift=5.

x=np.arange(0,LON1,1)
y = [A*exp(a*t**2)for t in x-xshift]

plt.plot(x,y,'ro')

for t in np.arange(1,LON1,1) :
    cdfhti[:,:,t] = A*exp(a*(t-xshift)**2)
    
#
# Intialise all other arrays to constant values
    
cdfhts[:,:,:]=rn_hts_ini
cdfati[:,:,:]=rn_ati_ini
cdfsmi[:,:,:]=rn_smi_ini
cdftmi[:,:,:]=rn_tmi_ini
cdftsu[:,:,:]=rn_tsu_ini


nc.close()
nccoord.close()

#sys.exit()