atmospy/ contains useful calculation tools. pot_vort/ contains tools for calculating potential vorticity.
xcontour is adapted from https://github.com/miniufo/xcontour, with thanks to the author for making their code available.
mars_analysis/ contains scripts used to plot figures.
In the following files, savedata=False will allow the user to plot figures from the published data. To calculate various diagnostics from model output, savedata=True can be turned on. All files can be run using the plot_env environment (plot_env.yml), unless otherwise specified.
Plotted in mars_analysis/plot_HC_anomaly.py using the function plot_psi_cross_section_one
Plotted in mars_analysis/plot_HC_anomaly.py using the functions plot_HC_edge_evolution and plot_HC_strength_evolution
Plotted in mars_analysis/plot_jet_anomaly.py using the functions plot_jet_lat_evolution and plot_jet_strength_evolution
Plotted in mars_analysis/plot_PV_anomaly.py using the functions plot_PV_lat_evolution and plot_PV_max_evolution
Plotted in mars_analysis/polar_tracer_ratios.py
Plotted in mars_analysis/plot_keff_cross_sections.py
Plotted in mars_analysis/plot_keff_lat.py
Plotted in mars_analysis/plot_summary_of_lats.py
Plotted in mars_analysis/plot_deposition_timeseries.py
Plotted in mars_analysis/plot_PV_maps_anomaly.py These should be run in the environment plotting using plotting.yml
Plotted in mars_analysis/plot_HC_anomaly.py using the function plot_psi_cross_section
Plotted in mars_analysis/plot_deposition_timeseries.py
The gradient of tracer is calculated in mars_analysis/add_gradStracer.py (either on pressure or theta levels)
Effective diffusivity is then calculated in mars_analysis/calculate_keff.py (either on pressure or theta levels; this should be run in keff environment using keff.yml)
This is then all saved in one place using the script mars_analysis/concat_eps_gamma.py