#!/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 file
fflx='flx_'+str(resname)+'.nc'

print '   creating FLX file  '+fflx+'...'

# 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 FLX netcdf file
nc=netcdf(fflx,'w')
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'))
cdftimecounter=nc.createVariable('time_counter','f',('time_counter'))
cdfutau=nc.createVariable('utau','f',('time_counter','y','x'))
cdfvtau=nc.createVariable('vtau','f',('time_counter','y','x'))
cdfqtot=nc.createVariable('qtot','f',('time_counter','y','x'))
cdfqsr=nc.createVariable('qsr','f',('time_counter','y','x'))
cdfemp=nc.createVariable('emp','f',('time_counter','y','x'))

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

# Fill fields

cdfutau[:,:,:]=-0.1
cdfvtau[:,:,:]=0
cdfqtot[:,:,:]=0
cdfqsr[:,:,:]=0
cdfemp[:,:,:]=0

nc.close()
nccoord.close()

#sys.exit()