raphael dussin raphaeldussin

crashing ocean general circulation models since 2008...

raphaeldussin / func2_roms2zarr.py

Last active December 3, 2020 02:20

function for the roms2zarr tutorial

	def select_interior(ds):
	"""
	discard "exterior" u,v,rho-points to build a symetric grid

	Parameters:
	ds (xarray.Dataset): ROMS dataset
	"""
	ds = ds.isel(xi_rho=slice(1,-1), eta_rho=slice(1,-1))
	if 'xi_v' in ds.dims:
	ds = ds.isel(xi_v=slice(1,-1))

raphaeldussin / func3_roms2zarr.py

Created December 3, 2020 02:26

function for roms2zarr tutorial

	def rename_dims(ds):
	""" rename dimensions

	Parameters:
	ds (xarray.Dataset): ROMS dataset

	"""
	ds = ds.rename({'xi_rho': 'xh', 'xi_v': 'xh', 'xi_u': 'xq', 'xi_psi': 'xq',
	'eta_rho': 'yh', 'eta_v': 'yq', 'eta_u': 'yh', 'eta_psi': 'yq',
	'ocean_time': 'time'

raphaeldussin / func4_roms2zarr.py

Created December 3, 2020 02:33

function for roms2zarr tutorial

	def add_coords(ds):
	""" set coordinate variables as xarray coordinates

	Parameters:
	ds (xarray.Dataset): ROMS dataset
	"""
	ds = ds.set_coords(['Cs_r', 'Cs_w', 'hc', 'h', 'Vtransform', 'time',
	'lon_rho', 'lon_v', 'lon_u', 'lon_psi',
	'lat_rho', 'lat_v', 'lat_u', 'lat_psi'])
	return ds

raphaeldussin / main_roms2zarr.py

Created December 3, 2020 02:45

main code for roms2zarr conversion

	import xarray as xr
	import glob

	# use glob to create a list of files to convert
	filelist = glob.glob('data/CCS2_avg_*.nc')
	# path to grid file
	gridfile = 'grid/CCS2_grid.nc'
	# output store name
	store_name = 'CCS2_store'

raphaeldussin / open_zarr_store_ROMS.py

Last active December 8, 2020 22:06

opening a ROMS zarr store

	import xarray as xr
	from xgcm import Grid

	CCS2 = xr.open_zarr(CCS2_store, consolidated=True)

	# Create xgcm grid object
	grid_ccs2 = Grid(CCS2, coords={'X': {'center': 'xh', 'outer': 'xq'},
	'Y': {'center': 'yh', 'outer': 'yq'},
	'Z': {'center': 's_rho', 'outer': 's_w'}},
	periodic=False)

raphaeldussin / compute_depth_ROMS.py

Last active December 8, 2020 00:01

compute depth of ROMS layers

	def compute_depth_layers(ds, grid, hmin=0.1):
	""" compute depths of ROMS vertical levels (Vtransform = 2) """

	# compute vertical transformation functional
	S_rho = (ds.hc * ds.s_rho + ds.Cs_r * ds.h) / (ds.hc + ds.h)
	S_w = (ds.hc * ds.s_w + ds.Cs_w * ds.h) / (ds.hc + ds.h)

	# compute depth of rho (layers) and w (interfaces) points
	z_rho = ds.zeta + (ds.zeta + ds.h) * S_rho
	z_w = ds.zeta + (ds.zeta + ds.h) * S_w

raphaeldussin / plot_section_xarray_ROMS.py

Last active December 8, 2020 01:18

plotting a ROMS velocity section with xarray

	import matplotlib.pyplot as plt
	import cmocean

	plt.figure(figsize=[10,8])

	CCS2['v'].isel(time=12, yq=500).plot(x='lon_v', y='z_v',
	vmin=-0.25, vmax=0.25,
	cmap=cmocean.cm.balance,
	subplot_kws={'facecolor': 'grey'})

raphaeldussin / plot_temp100m_xarray_ROMS.py

Last active February 8, 2022 18:07

interpolate and plot temperature in ROMS

	import cmocean
	import cartopy.crs as ccrs
	import matplotlib.pyplot as plt

	# extract temperature and depth for a given time
	snapshot = CCS2.isel(time=50)
	temp_snapshot = snapshot['temp']
	depth_snapshot = snapshot['z_rho']

	# compute temperature at 100 meters depth using xgcm.transform

raphaeldussin / movie.py

Created March 23, 2023 22:30

rotating globe movie for surface speed

	import xarray as xr
	import numpy as np
	from matplotlib import pyplot as plt
	from matplotlib import animation
	import matplotlib.image as image
	import cartopy.crs as ccrs
	from owslib.wms import WebMapService
	from matplotlib import cm
	import os
	from xgcm import Grid

raphaeldussin / meatpotatoes.yaml

Last active February 28, 2024 22:39