Multi-height states from xarray Dataset.

Attributes¶

data_source: str or xarray.Dataset

Either path to a file or data

state_coord: str

Name of the state coordinate

h_coord: str

Name of the height coordinate

xr_read_pars: dict

Parameters for xarray.open_dataset

Public members¶

MultiHeightNCStates(data_source, *args, state_coord='state', ...)[source]

Constructor.

load_data(algo, verbosity=0)[source]

Load and/or create all model data that is subject to chunking.

RDICT = {'index_col': 0}

__repr__()[source]

Return repr(self).

property data_source

The data source

reset(algo=None, states_sel=None, states_loc=None, verbosity=0)[source]

Reset the states, optionally select states

set_running(algo, data_stash, sel=None, isel=None, verbosity=0)[source]

Sets this model status to running, and moves all large data to stash.

unset_running(algo, data_stash, sel=None, isel=None, verbosity=0)[source]

Sets this model status to not running, recovering large data from stash

size()[source]

The total number of states.

index()[source]

The index list

output_point_vars(algo)[source]

The variables which are being modified by the model.

weights(algo)[source]

The statistical weights of all states.

calculate(algo, mdata, fdata, tdata)[source]

The main model calculation.

finalize(algo, verbosity=0)[source]

Finalizes the model.

classmethod new(states_type, *args, **kwargs)[source]

Run-time states factory.

output_coords()[source]

Gets the coordinates of all output arrays

ensure_variables(algo, mdata, fdata, tdata)[source]

Add variables to tdata, initialized with NaN

run_calculation(algo, *data, out_vars, **calc_pars)[source]

Starts the model calculation in parallel, via xarray’s apply_ufunc.

property model_id

Unique id based on the model type.

var(v)[source]

Creates a model specific variable name.

property initialized

Initialization flag.

sub_models()[source]

List of all sub-models

initialize(algo, verbosity=0, force=False)[source]

Initializes the model.

property running

Flag for currently running models

get_data(variable, target, lookup='smfp', mdata=None, ...)[source]

Getter for a data entry in the model object or provided data sources