Support for different alt/az MC selection radii.

Author: Ievgen Vovk

sim_in_box.py

@@ -61,7 +61,10 @@ def info_message(text):
 
     for emission_type in search_radius:
         if search_radius[emission_type] is not None:
-            search_radius[emission_type] = u.Quantity(search_radius[emission_type])
+            if isinstance(search_radius[emission_type], str):
+                search_radius[emission_type] = u.Quantity(search_radius[emission_type])
+            else:
+                search_radius[emission_type] = [u.Quantity(s) for s in search_radius[emission_type]]
 
     info_message('Preparing sources')
     srcs = srcgen(cfg['sources'])

sim_in_box.yaml

@@ -7,8 +7,8 @@ mc:
         max_samples: 10
     proton:
         files: 'mc/dl2_proton_20deg_180deg_20201125_v0.6.3_prod5_v00_testing.h5'
-        search_radius: '180 deg'
-        max_samples: 10
+        search_radius: ['180 deg', '180 deg']
+        max_samples: 100
 
 rungen:
     pointing:

srcsim/run.py

@@ -70,8 +70,16 @@ def predict(self, mccollections, source, tel_pos_tolerance=None, time_step=1*u.m
 
             if tel_pos_tolerance is None:
                 mc = mccollections[source.emission_type].get_closest(tel_pos.altaz)
-            else:
+            elif isinstance(tel_pos_tolerance, u.Quantity):
                 mc = mccollections[source.emission_type].get_nearby(tel_pos, tel_pos_tolerance)
+            elif isinstance(tel_pos_tolerance, list) or isinstance(tel_pos_tolerance, tuple):
+                mc = mccollections[source.emission_type].get_in_box(
+                    tel_pos,
+                    max_lon_offset=tel_pos_tolerance[0],
+                    max_lat_offset=tel_pos_tolerance[1],
+                )
+            else:
+                raise ValueError(f"Data type '{type(tel_pos_tolerance)}' for argument 'tel_pos_tolerance' is not supported")
 
             nsamples = len(mc.samples)