source: branches/2015/dev_r5102_CNRS11_TestCase/NEMOGCM/TOOLS/TEST_CASES/CREATE_FILES/make_TScst_init.py @ 5346

#!/opt/local/bin/python

import os,sys
from netCDF4 import Dataset as netcdf
import numpy as np

resname='r025'

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

# Output files
fTSinit='TScst_init_'+str(resname)+'.nc'

DEPTH1=31

print '   creating TS init file  '+fTSinit+'...'

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

# Creating TS init netcdf file
nc=netcdf(fTSinit,'w')
nc.createDimension('z',DEPTH1)
nc.createDimension('y',LAT1)
nc.createDimension('x',LON1)
nc.createDimension('time_counter',None)

cdflon=nc.createVariable('nav_lon','f',('y','x'))
cdflat=nc.createVariable('nav_lat','f',('y','x'))
cdfdepth=nc.createVariable('deptht','f',('z'))
cdftimecounter=nc.createVariable('time_counter','f',('time_counter'))
cdfthetao=nc.createVariable('thetao','f',('time_counter','z','y','x'))
cdfso=nc.createVariable('so','f',('time_counter','z','y','x'))

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

## test 1 : T & S Constants
cdfso[0,:,:,:]=35
cdfthetao[0,:,:,:]=10    

#sys.exit()

nc.close()
nccoord.close()