Xa,Ya = np.meshgrid(xa,ya)
ka = np.arange(ijk1[2],ijk2[2]+1)
dims = list(np.shape(ka))
dims.extend(list(np.shape( Xa )))
Cfs = np.zeros( dims, float )
print(' Cfs.shape={}'.format(np.shape(Cfs)))
# = = = = = = = = = = = = = = =
scales = ya # Rename, that's all
kList = ka
k = 0
for kt in kList:
vt = v[:,kt, ijk1[1], ijk1[0]]
C, freqs = continuousWaveletAnalysis( vt, wlDict )
print(' Cfs[kt,:,:].shape = {}, Q.shape = {}'.format(np.shape(Cfs[k,:,:]), np.shape(C)))
Cfs[k,:,:] = C.copy()
k += 1
# = = output file = = = = =
# Create a NETCDF output dataset (dso) for writing out the data.
dso = netcdfOutputDataset( fileout )
xv = createNetcdfVariable( dso, xa , 'x' , len(xa) , 'm', 'f4', ('x',) , parameter )
yv = createNetcdfVariable( dso, ya , 'y' , len(ya) , 'm', 'f4', ('y',) , parameter )
zv = createNetcdfVariable( dso, ka , 'z' , len(ka) , 'm', 'f4', ('z',) , parameter )
Qv = createNetcdfVariable( dso, Cfs, 'Cfs', dims[0], '-', 'f4',('z','y','x',) , variable )
# - - - - Done , finalize the output - - - - - - - - - -
netcdfWriteAndClose( dso )
fx.write(str(nPc[2]) + "\n")
# Loop through the verical canopy columns
for x in range(nPc[0]):
for y in range(nPc[1]):
if (np.all(lad_3d[x, y, :] == 0)):
# There is no need to write empty columns
continue
# Convert everything into strings and write
# datatype x y col(:)
lineStr = str(1) + "," + str(x) + "," + str(y) + "," + ",".join(
map("{:.3g}".format, lad_3d[x, y, :])) + "\n"
fx.write(lineStr)
fx.close()
elif (args.output == "nc"):
#Save into netCDF4 dataset
dso = nct.netcdfOutputDataset(args.fileout)
nPc = np.shape(lad_3d)
xv = nct.createCoordinateAxis(dso,
nPc,
dpx,
0,
'x',
'f4',
'm',
parameter=True)
yv = nct.createCoordinateAxis(dso,
nPc,
dpx,
1,
'y',
'f4',