NEDAS.models.vort2d package

class NEDAS.models.vort2d.Vort2DModel(**kwargs)[source]

Bases: Model[RegularGrid]

nx: int
ny: int
dx: float
dt: float
restart_dt: float
Vmax: float
Rmw: float
Vbg: float
Vslope: float
loc_sprd: int
gen: float
diss: float
memory: dict = {}
filename(**kwargs)[source]
read_grid(**kwargs)[source]

Read the grid information from the model output.

Parameters:

**kwargs – Keyword arguments for reading the grid.

read_mask(**kwargs)[source]
read_var_from_file(**kwargs)[source]
write_var_to_file(var, **kwargs)[source]
z_coords(**kwargs)[source]

Get the vertical coordinates of the model.

Parameters:

**kwargs – Keyword arguments for getting the vertical coordinates.

Returns:

The vertical coordinates.

Return type:

np.ndarray

generate_initial_condition(loc_sprd=None)[source]
preprocess(**kwargs)[source]

Preprocess the model data.

Parameters:

**kwargs – Keyword arguments for preprocessing.

postprocess(*args, **kwargs)[source]

Postprocess the model data.

Parameters:

**kwargs – Keyword arguments for postprocessing.

run(*args, **kwargs)[source]

Run the model forward in time.

Parameters:

  • *args – Arguments

  • **kwargs – Keyword arguments

Keyword Arguments:

  • time (datetime) – current time when forecast starts

  • restart_dir (str) – directory where restart files are located

  • forecast_period (int) – forecast period in hours

If self.ens_run_strategy == ‘batch’, the method will run all ensemble members in one go, expect additional kwargs[‘nens’] to be the ensemble size. If self.ens_run_strategy == ‘scheduler’, the method runs a single member indexed by kwargs[‘member’], and kwargs[‘worker_id’] is the pid assigned by the scheduler to run this method.

generate_truth(*args, **kwargs) → None[source]

Generate truth (nature run) model states. Use for running synthetic observation experiments.

generate_init_ensemble(*args, **kwargs) → None[source]

Generate initial perturbed model states for ensemble forecasts.

Parameters:

  • nens (int) – ensemble size

  • **kwargs

Submodules