plots for version sent to coauthors

juliasloan25 · juliasloan25 · commit 532677179be8 · 2025-03-10T14:51:03.000-07:00
diff --git a/experiments/long_runs/data_paper_plots.jl b/experiments/long_runs/data_paper_plots.jl
@@ -74,27 +74,27 @@ function get_sim_var_dict(simdir)
             return sim_var_swd
         end
 
-    # Compute LWN = LWD - LWU
-    sim_var_dict["lwn"] =
-        () -> begin
-            lwd = sim_var_dict["lwd"]()
-            lwu = sim_var_dict["lwu"]()
-            lwn = lwd - lwu
-            lwn.attributes["long_name"] = "Net longwave radiation"
-            lwn.attributes["units"] = "W m⁻²"
-            return lwn
-        end
-
-    # Compute SWN = SWD - SWU
-    sim_var_dict["swn"] =
-        () -> begin
-            swd = sim_var_dict["swd"]()
-            swu = sim_var_dict["swu"]()
-            swn = swd - swu
-            swn.attributes["long_name"] = "Net shortwave radiation"
-            swn.attributes["units"] = "W m⁻²"
-            return swn
-        end
+    # # Compute LWN = LWD - LWU
+    # sim_var_dict["lwn"] =
+    #     () -> begin
+    #         lwd = sim_var_dict["lwd"]()
+    #         lwu = sim_var_dict["lwu"]()
+    #         lwn = lwd - lwu
+    #         lwn.attributes["long_name"] = "Net longwave radiation"
+    #         lwn.attributes["units"] = "W m⁻²"
+    #         return lwn
+    #     end
+
+    # # Compute SWN = SWD - SWU
+    # sim_var_dict["swn"] =
+    #     () -> begin
+    #         swd = sim_var_dict["swd"]()
+    #         swu = sim_var_dict["swu"]()
+    #         swn = swd - swu
+    #         swn.attributes["long_name"] = "Net shortwave radiation"
+    #         swn.attributes["units"] = "W m⁻²"
+    #         return swn
+    #     end
 
     return sim_var_dict
 end
@@ -104,12 +104,12 @@ function get_obs_var_dict()
         ClimaLand.Artifacts.era5_monthly_averages_2008_folder_path(),
         "era5_monthly_surface_fluxes_200801-200812.nc",
     )
-    era5_land_forcing_data_path = joinpath(
-        ClimaLand.Artifacts.era5_land_forcing_data2008_folder_path(
-            lowres = false,
-        ),
-        "era5_2008_0.9x1.25.nc",
-    )
+    # era5_land_forcing_data_path = joinpath(
+    #     ClimaLand.Artifacts.era5_land_forcing_data2008_folder_path(
+    #         lowres = false,
+    #     ),
+    #     "era5_2008_0.9x1.25.nc",
+    # )
 
     # Dict for loading in observational data
     obs_var_dict = Dict{String, Any}()
@@ -304,27 +304,27 @@ function get_obs_var_dict()
         end
 
 
-    # Compute LWN = LWD - LWU
-    obs_var_dict["lwn"] =
-        (start_date) -> begin
-            lwd = obs_var_dict["lwd"](start_date)
-            lwu = obs_var_dict["lwu"](start_date)
-            lwn = lwd - lwu
-            lwn.attributes["long_name"] = "Net longwave radiation"
-            lwn.attributes["units"] = "W m⁻²"
-            return lwu
-        end
-
-    # Compute SWN = SWD - SWU
-    obs_var_dict["swn"] =
-        (start_date) -> begin
-            swd = obs_var_dict["swd"](start_date)
-            swu = obs_var_dict["swu"](start_date)
-            swn = swd - swu
-            swn.attributes["long_name"] = "Net shortwave radiation"
-            swn.attributes["units"] = "W m⁻²"
-            return swn
-        end
+    # # Compute LWN = LWD - LWU
+    # obs_var_dict["lwn"] =
+    #     (start_date) -> begin
+    #         lwd = obs_var_dict["lwd"](start_date)
+    #         lwu = obs_var_dict["lwu"](start_date)
+    #         lwn = lwd - lwu
+    #         lwn.attributes["long_name"] = "Net longwave radiation"
+    #         lwn.attributes["units"] = "W m⁻²"
+    #         return lwu
+    #     end
+
+    # # Compute SWN = SWD - SWU
+    # obs_var_dict["swn"] =
+    #     (start_date) -> begin
+    #         swd = obs_var_dict["swd"](start_date)
+    #         swu = obs_var_dict["swu"](start_date)
+    #         swn = swd - swu
+    #         swn.attributes["long_name"] = "Net shortwave radiation"
+    #         swn.attributes["units"] = "W m⁻²"
+    #         return swn
+    #     end
 
     return obs_var_dict
 end
diff --git a/experiments/long_runs/paper_plots.jl b/experiments/long_runs/paper_plots.jl
@@ -24,17 +24,15 @@ root_path = joinpath(pwd(), "snowy_land_longrun_gpu")
 # outdir = "snowy_land_longrun_gpu-3761-alpha1/output_active" # on local
 # outdir = "snowy_land_longrun_gpu-3771-new_default/output_active"
 # outdir = "snowy_land_longrun_gpu-3761-new_default/output_active" # on local
-outdir = "snowy_land_longrun_gpu-3777-alpha2/output_active" # on local
+outdir = "snowy_land_longrun_gpu-3785"
 root_path = outdir
 
-short_names = ["lhf", "shf", "lwu", "swu", "lwn", "swn"]
+short_names = ["lhf", "shf", "lwu", "swu"]#, "lwn", "swn"]
 title_stubs = Dict(
-    "lhf" => "Latent heat flux",
-    "shf" => "Sensible heat flux",
-    "lwu" => "Upward longwave radiation",
-    "swu" => "Upward shortwave radiation",
-    "lwn" => "Net longwave radiation",
-    "swn" => "Net shortwave radiation",
+    "lhf" => "LE",#"Latent heat flux",
+    "shf" => "H",#"Sensible heat flux",
+    "lwu" => "LWᵤ",#"Upward longwave radiation",
+    "swu" => "SWᵤ",#"Upward shortwave radiation",
 )
 # Define levels for contour colorbars
 levels_dict = Dict(
@@ -55,7 +53,7 @@ function compute_global_average(masked_var)
     land_data = ClimaAnalysis.apply_oceanmask(masked_var).data
     lat_data = masked_var.dims[latitude_name]
     mask = .~isnan.(land_data)
-    nlon = length(masked_var.dims[lon_name])µ
+    nlon = length(masked_var.dims[lon_name])
     resized_lat_data = transpose(repeat(lat_data, 1, nlon))
 
     return sum(land_data[mask] .* cosd.(resized_lat_data[mask])) /
@@ -206,13 +204,13 @@ function make_paper_figures(
                         xgridvisible = false, # don't show lat grid
                         ygridvisible = false, # don't show lon grid
                         height = 235,
-                        ylabel = "$(sim_var.attributes["long_name"]) $units_label", # plot variable label on y-axis for leftmost column (sim)
+                        ylabel = "$title_stub $units_label", # plot variable label on y-axis for leftmost column (sim)
                         ylabelvisible = true,
                         ylabelpadding = 0,
                     ),
                     :cb => ClimaAnalysis.Utils.kwargs(
                         vertical = false, # horizontal colorbar
-                        label = "$(sim_var.attributes["long_name"]) $units_label",
+                        label = "$title_stub $units_label",
                         labelsize = 20,
                         flipaxis = false, # label underneath colorbar
                         height = 15,
@@ -234,7 +232,7 @@ function make_paper_figures(
             contour = viz.contour2D_on_globe!(
                 fig,
                 sim_var_annual_average,
-                # ylabel = "$(sim_var.attributes["long_name"]) $units_label", # plot variable label on y-axis for leftmost column (sim)
+                # ylabel = "$title_stub $units_label", # plot variable label on y-axis for leftmost column (sim)
                 p_loc = (fig_row, 1), # plot in the first column
                 plot_colorbar = false,
                 mask = viz.oceanmask(),
@@ -255,7 +253,7 @@ function make_paper_figures(
                         xgridvisible = false, # don't show lat grid
                         ygridvisible = false, # don't show lon grid
                         height = 235,
-                        ylabel = "$(sim_var.attributes["long_name"]) $units_label", # plot variable label on y-axis for leftmost column (sim)
+                        ylabel = "$title_stub $units_label", # plot variable label on y-axis for leftmost column (sim)
                         ylabelpadding = -5,
                         ylabelvisible = true,
                         # limits = clims,
@@ -264,7 +262,7 @@ function make_paper_figures(
             )
             Makie.Colorbar(
                 fig[fig_row + 1, 1],
-                label = "$(sim_var.attributes["long_name"]) $units_label",
+                label = "$title_stub $units_label",
                 labelsize = 20,
                 vertical = false, # horizontal colorbar
                 flipaxis = false, # label underneath colorbar
@@ -293,7 +291,7 @@ function make_paper_figures(
                 levels,
                 p_loc = (fig_row, 2), # plot in the first column
                 plot_colorbar = true,
-                # colorbar_label = "$(sim_var.attributes["long_name"]) $units_label",
+                # colorbar_label = "$title_stub $units_label",
                 mask = viz.oceanmask(),
                 cmap_extrema = clims,
                 more_kwargs = Dict(
@@ -310,7 +308,7 @@ function make_paper_figures(
                     ),
                     :cb => ClimaAnalysis.Utils.kwargs(
                         vertical = false, # horizontal colorbar
-                        label = "$(sim_var.attributes["long_name"]) $units_label",
+                        label = "$title_stub $units_label",
                         labelsize = 20,
                         flipaxis = false, # label underneath colorbar
                         height = 15,
@@ -359,7 +357,7 @@ function make_paper_figures(
             )
             Makie.Colorbar(
                 fig[fig_row + 1, 2],
-                label = "$(sim_var.attributes["long_name"]) $units_label",
+                label = "$title_stub $units_label",
                 labelsize = 20,
                 vertical = false, # horizontal colorbar
                 flipaxis = false, # label underneath colorbar
@@ -386,7 +384,7 @@ function make_paper_figures(
             bias_title =
                 fig_row == 1 ?
                 CairoMakie.rich("ClimaLand vs ERA5 bias", fontsize = 28) : "" # title of the figure
-            bias_colorbar_label = "$(sim_var.attributes["long_name"]) $units_label"
+            bias_colorbar_label = "$title_stub $units_label"
             viz.plot_bias_on_globe!(
                 fig,
                 sim_var_annual_average,
@@ -510,11 +508,11 @@ function make_paper_figures(
 
 
     end
-    save_name = joinpath(root_path, "combined_figures.pdf")
+    save_name = joinpath(root_path, "combined_figures.png")
     save_name =
-        plot_bias ? joinpath(root_path, "combined_figures_bias.pdf") : save_name
+        plot_bias ? joinpath(root_path, "combined_figures_bias.png") : save_name
     save_name =
-        plot_seasonal ? joinpath(root_path, "combined_figures_seasonal.pdf") :
+        plot_seasonal ? joinpath(root_path, "combined_figures_seasonal.png") :
         save_name
     CairoMakie.save(save_name, fig)
     @show save_name
diff --git a/experiments/long_runs/seasonal_paper_plots.jl b/experiments/long_runs/seasonal_paper_plots.jl
@@ -22,7 +22,8 @@ root_path = joinpath(pwd(), "snowy_land_longrun_gpu")
 # outdir = "snowy_land_longrun_gpu-3771-new_default/output_active"
 # outdir = "snowy_land_longrun_gpu-3761-new_default/output_active" # on local
 # outdir = "snowy_land_longrun_gpu-3761-alpha1/output_active" # on local
-outdir = "snowy_land_longrun_gpu-3777-alpha2/output_active" # on local
+# outdir = "snowy_land_longrun_gpu-3777-alpha2/output_active" # on local
+outdir = "snowy_land_longrun_gpu-3785"
 root_path = outdir
 
 short_names = ["lhf", "shf", "lwu", "swu"]
@@ -192,7 +193,7 @@ function make_seasonal_cycle_figure(root_path, outdir, short_names, title_stubs)
         font = "TeX Gyre Heros Bold Makie",
     )
 
-    save_name = joinpath(root_path, "seasonal_cycle.pdf")
+    save_name = joinpath(root_path, "seasonal_cycle.png")
     CairoMakie.save(save_name, fig)
     @show save_name
     return nothing
