add append_table_data func [skip ci]

Author: juliasloan25

experiments/ClimaEarth/user_io/benchmarks.jl

@@ -138,19 +138,35 @@ headers = [build_id_str, "Horiz. res.: 30 elems", "CPU Run [64 processes]", "GPU
 data = [
     ["" "Vert. res.: 63 levels" "" ""]
     ["" "dt: 120secs" "" ""]
-    ["" "job ID:" cpu_job_id_coupled gpu_job_id_coupled]
-    ["Coupled" "SYPD:" cpu_sypd_coupled gpu_sypd_coupled]
-    ["" "CPU max RSS allocs:" cpu_allocs_coupled gpu_cpu_allocs_coupled]
-    ["" "job ID:" cpu_job_id_atmos gpu_job_id_atmos]
-    ["Atmos-only" "SYPD:" cpu_sypd_atmos gpu_sypd_atmos]
-    ["" "CPU max RSS allocs:" cpu_allocs_atmos gpu_cpu_allocs_atmos]
 ]
 
+"""
+    append_table_data!(table_data, setup_id, cpu_job_id, gpu_job_id, cpu_artifacts_dir, gpu_artifacts_dir)
+
+Append data for a given setup to the table data.
+"""
+function append_table_data!(table_data, setup_id, cpu_job_id, gpu_job_id, cpu_artifacts_dir, gpu_artifacts_dir)
+    # Get SYPD and allocation info for both input runs
+    cpu_sypd, cpu_allocs = get_sypd_allocs(cpu_artifacts_dir)
+    gpu_sypd, gpu_cpu_allocs = get_sypd_allocs(gpu_artifacts_dir)
+
+    # Create rows containing data for these runs
+    new_table_data = [
+        ["" "job ID:" cpu_job_id gpu_job_id]
+        [setup_id "SYPD:" cpu_sypd gpu_sypd]
+        ["" "CPU max RSS allocs:" cpu_allocs gpu_cpu_allocs]
+    ]
+    table_data = vcat(table_data, new_table_data)
+end
+
+append_table_data!(data, "Coupled", cpu_job_id_coupled, gpu_job_id_coupled, cpu_artifacts_dir_coupled, gpu_artifacts_dir_coupled)
+append_table_data!(data, "Atmos with diag. EDMF", cpu_job_id_atmos, gpu_job_id_atmos, cpu_artifacts_dir_atmos, gpu_artifacts_dir_atmos)
+
 # Use the coupled CPU job ID for the output dir
 table_output_dir = joinpath(output_dir, "compare_$(mode_name)_$(mode_name_atmos)_$(cpu_job_id_coupled)")
 !isdir(table_output_dir) && mkdir(table_output_dir)
 table_path = joinpath(table_output_dir, "table.txt")
 open(table_path, "w") do f
     # Output the table, including lines before and after the header
-    PrettyTables.pretty_table(f, data, header = headers, hlines = [0, 3, 6, 9])
+    PrettyTables.pretty_table(f, data, header = headers, hlines = [0, 3, 6, 9]) # TODO don't hardcode hlines
 end