Feature/diagnostics level select - Githubissues

ai2cm / pace

Implementation of the FV3GFS / SHiELD atmospheric model in Python

https://ai2cm.github.io/pace/

Apache License 2.0

37 stars 12 forks source link

Feature/diagnostics level select #370

Closed elynnwu closed 2 years ago

elynnwu commented 2 years ago

Purpose

This PR adds the capability of saving a vertical slice using z_select in the diagnostics config.

The expected syntax is as follows:

diagnostics_config:
  path: output
  output_format: netcdf
  z_select:
    - level: 1
       names:
         - ua
         - va
    - level: 10
       names:
         - pt

Code changes:

diagnostics.py: added ZSelect class for storing selected vertical slice and the variables to be saved
- fix column integral mistake, need to multiple by 1/gravity

Checklist

Before submitting this PR, please make sure:

[x] You have followed the coding standards guidelines established at Code Review Checklist.
[ ] Docstrings and type hints are added to new and updated routines, as appropriate
[ ] All relevant documentation has been updated or added (e.g. README, CONTRIBUTING docs)
[ ] For each public change and fix in pace-util, HISTORY has been updated
[ ] Unit tests are added or updated for non-stencil code changes