TEM diagnostics

[justin-richling]

This will calculate TEM files and plots for:

• uzm - Zonal-Mean zonal wind
• epfy - northward component of the Eliassen–Palm flux
• epfz - upward component of the Eliassen–Palm flux
• vtem - Transformed Eulerian mean northward wind
• wtem - Transformed Eulerian mean upward wind
• psitem - Transformed Eulerian mean mass stream function
• utendepfd - tendency of eastward wind due to Eliassen-Palm flux divergence
• utendvtem - tendency of eastward wind due to TEM northward wind advection and the coriolis term
• utendwtem - tendency of eastward wind due to TEM upward wind advection

There are two scripts being added, scripts/averaging/create_TEM_files.py for reading in h4 history files, calculating TEM variables and creating new netcdf files. And scripts/plotting/tem.py for reading in either the newly created netcdf files, or existing TEM netcdf files and plotting.

Update adf_variable_defaults.yaml with TEM variables, including Obs files and variable names.

Update config_cam_baseline_example.yaml for TEM scripts

NOTE: If QBO is present, all variables will be calculated, if not, only uzm epfy epfz vtem wtem psitem and utendepfd will be calculated.

NOTE: If there are different vertical levels between cases, currently the difference plots don't work. I am working towards a vertical and horizontal interpolation for TEM as noted in issue #225

[nusbaume]

Great work! However, I do have some concerns, particularly with making sure we are actually grabbing all of the model history files. If you have any questions or concerns with my change requests just let me know. Thanks!

[justin-richling]

@nusbaume I think I've tackled all your suggestions and have run a couple tests. Let me know if anything else needs addressing, thanks!

[justin-richling]

@nusbaume Last changes are in place if you want to review this again, thanks!

[nusbaume]

Looking a lot better! I just have some (hopefully) final config variable checking and cleanup requests.

[justin-richling]

@nusbaume Can you take another look and see if this is all good now. I ran it with your suggested changes and it appears to work as intended for when tem_opts is included or missing in the config file.

[nusbaume]

Everything looks great to me now, so feel free to merge whenever you are ready. Thanks!