formatter

CliMA · Apr 10, 2024 · bc0de28 · bc0de28
1 parent e91c0ac
commit bc0de28
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 9 deletions.
diff --git a/docs/src/plots/P3TerminalVelocityPlots.jl b/docs/src/plots/P3TerminalVelocityPlots.jl
@@ -252,4 +252,4 @@ function figure_2()
 end
 
 # Terminal Velocity figure
-figure_2()
+figure_2()
diff --git a/src/P3Scheme.jl b/src/P3Scheme.jl
@@ -458,7 +458,7 @@ function terminal_velocity_mass(
     for i in 1:2
         if F_r == 0
             # Velocity coefficients for small particles
-            (ai, bi, ci) = CO.Chen2022_vel_coeffs_small(Chen2022, ρ_a) 
+            (ai, bi, ci) = CO.Chen2022_vel_coeffs_small(Chen2022, ρ_a)
             v += integrate(
                 FT(0),
                 D_th,
@@ -547,7 +547,11 @@ function terminal_velocity_mass(
                 v += integrate(
                     th.D_gr,
                     cutoff,
-                    FT(π) / 6 * th.ρ_g * ai[i] * N_0 * (p3.ρ_i / th.ρ_g)^(2 * κ),
+                    FT(π) / 6 *
+                    th.ρ_g *
+                    ai[i] *
+                    N_0 *
+                    (p3.ρ_i / th.ρ_g)^(2 * κ),
                     bi[i] + μ + 3,
                     ci[i] + λ,
                 )
@@ -559,15 +563,23 @@ function terminal_velocity_mass(
                 v += integrate(
                     cutoff,
                     th.D_cr,
-                    FT(π) / 6 * th.ρ_g * ai[i] * N_0 * (p3.ρ_i / th.ρ_g)^(2 * κ),
+                    FT(π) / 6 *
+                    th.ρ_g *
+                    ai[i] *
+                    N_0 *
+                    (p3.ρ_i / th.ρ_g)^(2 * κ),
                     bi[i] + μ + 3,
                     ci[i] + λ,
                 )
             else
                 v += integrate(
                     th.D_gr,
                     th.D_cr,
-                    FT(π) / 6 * th.ρ_g * ai[i] * N_0 * (p3.ρ_i / th.ρ_g)^(2 * κ),
+                    FT(π) / 6 *
+                    th.ρ_g *
+                    ai[i] *
+                    N_0 *
+                    (p3.ρ_i / th.ρ_g)^(2 * κ),
                     bi[i] + μ + 3,
                     ci[i] + λ,
                 )
@@ -594,7 +606,7 @@ function terminal_velocity_mass(
                     cutoff,
                 )
                 v += I
-                
+
                 # Switch to large particles
                 (ai, bi, ci) = CO.Chen2022_vel_coeffs_large(Chen2022, ρ_a)
                 large = true

diff --git a/test/performance_tests.jl b/test/performance_tests.jl
@@ -122,9 +122,19 @@ function benchmark_test(FT)
 
     # P3 scheme
     bench_press(P3.thresholds, (p3, ρ_r, F_r), 12e6, 2048, 80)
-    if FT == Float64 
-        bench_press(P3.distribution_parameter_solver, (p3, q_ice, N, ρ_r, F_r), 1e5)
-        bench_press(P3.terminal_velocity_mass, (p3, ch2022.snow_ice, q_ice, N, ρ_r, F_r, ρ_air), 2e6, 4e4, 2e3)
+    if FT == Float64
+        bench_press(
+            P3.distribution_parameter_solver,
+            (p3, q_ice, N, ρ_r, F_r),
+            1e5,
+        )
+        bench_press(
+            P3.terminal_velocity_mass,
+            (p3, ch2022.snow_ice, q_ice, N, ρ_r, F_r, ρ_air),
+            2e6,
+            4e4,
+            2e3,
+        )
         bench_press(P3.D_m, (p3, q_ice, N, ρ_r, F_r), 1e5)
     end