Merge pull request #91 from Fonotec/master

Include the SNIa rates inside the aperatures in the library

Author: bwvdnbro

velociraptor/catalogue/registration.py

@@ -1523,6 +1523,31 @@ def registration_element_masses_in_stars(
     return
 
 
+def registration_snia_rates(
+    field_path: str, unit_system: VelociraptorUnits
+) -> (unyt.Unit, str, str):
+    """
+    Registers the SNIa rates within apertures
+    """
+
+    unit = unit_system.velocity / unit_system.length
+    # Capture aperture size
+    match_string = "Aperture_SNIaRates_aperture_total_star_([0-9]*)_kpc"
+    regex = cached_regex(match_string)
+
+    match = regex.match(field_path)
+
+    if match:
+        aperture_size = match.group(1)
+
+        full_name = f"SNIa rate ({aperture_size} kpc)"
+        snake_case = f"snia_rates_{aperture_size}_kpc"
+
+        return unit, full_name, snake_case
+    else:
+        raise RegistrationDoesNotMatchError
+
+
 # TODO
 # lambda_B
 # q
@@ -1545,6 +1570,7 @@ def registration_element_masses_in_stars(
 global_registration_functions = {
     k: globals()[f"registration_{k}"]
     for k in [
+        "snia_rates",
         "metallicity",
         "ids",
         "energies",