valentinaschueller / ece-scm-coupling

Helpful scripts to change coupling of the EC-Earth coupled SCM

EC-Earth SCM Tools

This repository contains helpful scripts to run the EC-Earth AOSCM and analyze its output. The assumption is that the AOSCM is compiled on your system, as described in the documentation.

Particularly:

• the Python wrapper SchwarzCoupling allows to do Schwarz Waveform Relaxation with the AOSCM
• experiments can be configured and executed directly from Python (using utils/templates.py and the AOSCM class in utils/helpers.py, respectively)
• user_context.py contains your local environment, in order to find executables, input data and output files

Requirements

The scripts use Python 3.9 as a basis. The following external packages are required:

For configuring/running the AOSCM:

• Jinja
• NumPy
• pandas
• ruamel.yaml
• xarray

For plotting:

• ProPlot

Usage

Tutorials:

1. Configure and run a single experiment
2. Change the coupling algorithm (parallel/atmosphere-first/ocean-first)
3. Run a Schwarz Waveform Relaxation simulation
4. Generate a valid rstas.nc file for OASIS initialization
5. Run an ensemble simulation
6. Visualize output

ece-scm-coupling's Issues

Use a more robust structure than dictionaries for experiment setup

At the moment, I use a dictionary to configure experiments. Handling default cases and throwing exceptions if certain parameters are not set is dealt with when rendering config-run.xml.j2.
However, it would be nicer to have this logic outside of the template.
Ideally, the data structure for experiment configuration knows:

• which information is required to set up an experiment
• which information is optional (and what the default case is)

Thus, a user could check whether the configuration for an experiment is complete (and typos in dict keys could be avoided) directly when configuring the experiment and before calling render_template()

Make sure that starting from existing iteration still works for SWR simulation

I am pretty sure that it does not work anymore to restart an SWR experiment as follows:

experiment = {
    "exp_id": "EXP0",
    # ...
}

schwarz = SchwarzCoupling(experiment)
schwarz.run(5)
schwarz.run(10, current_iter=5)

• test whether restarting still works or not
• fix if this is broken

Support for Tetralith

• check what to change in terms of platform settings
• start model run using sbatch

Update scripts to support the ECE4-AOSCM