use macro to initialize species masses and gammas

AMReX-Astro · Aug 29, 2024 · bba236b · bba236b
1 parent 0401c93
commit bba236b
Showing 1 changed file with 25 additions and 84 deletions.
diff --git a/EOS/primordial_chem/actual_eos.H b/EOS/primordial_chem/actual_eos.H
@@ -27,90 +27,31 @@ void actual_eos_init ()
 
     int idx;
 
-    idx = network_spec_index(eos_rp::species_1_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_1_gamma;
-        spmasses[idx] = eos_rp::species_1_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_2_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_2_gamma;
-        spmasses[idx] = eos_rp::species_2_mass;
-   }
-
-    idx = network_spec_index(eos_rp::species_3_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_3_gamma;
-        spmasses[idx] = eos_rp::species_3_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_4_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_4_gamma;
-        spmasses[idx] = eos_rp::species_4_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_5_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_5_gamma;
-        spmasses[idx] = eos_rp::species_5_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_6_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_6_gamma;
-        spmasses[idx] = eos_rp::species_6_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_7_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_7_gamma;
-        spmasses[idx] = eos_rp::species_7_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_8_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_8_gamma;
-        spmasses[idx] = eos_rp::species_8_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_9_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_9_gamma;
-        spmasses[idx] = eos_rp::species_9_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_10_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_10_gamma;
-        spmasses[idx] = eos_rp::species_10_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_11_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_11_gamma;
-        spmasses[idx] = eos_rp::species_11_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_12_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_12_gamma;
-        spmasses[idx] = eos_rp::species_12_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_13_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_13_gamma;
-        spmasses[idx] = eos_rp::species_13_mass;
-    }
-
-    idx = network_spec_index(eos_rp::species_14_name);
-    if (idx >= 0) {
-        gammas[idx] = eos_rp::species_14_gamma;
-        spmasses[idx] = eos_rp::species_14_mass;
-    }
-
+    // Set the gammas & masses for the species
+    #define GET_SPECIES_PARAMS(num) do { \
+        idx = network_spec_index(eos_rp::species_##num##_name); \
+        if (idx >= 0) { \
+            gammas[idx] = eos_rp::species_##num##_gamma; \
+            spmasses[idx] = eos_rp::species_##num##_mass; \
+        } \
+    } while (0)
+
+    GET_SPECIES_PARAMS(1);
+    GET_SPECIES_PARAMS(2);
+    GET_SPECIES_PARAMS(3);
+    GET_SPECIES_PARAMS(4);
+    GET_SPECIES_PARAMS(5);
+    GET_SPECIES_PARAMS(6);
+    GET_SPECIES_PARAMS(7);
+    GET_SPECIES_PARAMS(8);
+    GET_SPECIES_PARAMS(9);
+    GET_SPECIES_PARAMS(10);
+    GET_SPECIES_PARAMS(11);
+    GET_SPECIES_PARAMS(12);
+    GET_SPECIES_PARAMS(13);
+    GET_SPECIES_PARAMS(14);
+
+    #undef GET_SPECIES_PARAMS
 }