fixed radiopropa import error

nu-radio · Jan 25, 2024 · 0d9caa2 · 0d9caa2
1 parent f954996
commit 0d9caa2
Showing 1 changed file with 39 additions and 35 deletions.
diff --git a/NuRadioMC/SignalProp/examples/birefringence_examples/01_simple_propagation.py b/NuRadioMC/SignalProp/examples/birefringence_examples/01_simple_propagation.py
@@ -5,7 +5,11 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-from NuRadioMC.SignalProp import radioproparaytracing
+try:
+    from NuRadioMC.SignalProp import radioproparaytracing
+    propa = True
+except:
+    propa = False
 from NuRadioMC.SignalProp import analyticraytracing
 
 """
@@ -87,40 +91,40 @@ def birefringence_propagation(config, raytracing, isolution, plot, lim, tracing_
 
 birefringence_propagation(config, 'analytical', ray_tracing_solution, 1, x_limits, 'analyticraytracing.py', 'analyticraytracing.py', 'southpole_B')
 
-
-# ---------------   raytracing:            analyticraytracing.py ---------------------
-# ---------------   e-field propagation:   radioproparaytracing.py ---------------------
-# ---------------   biefringence model:    'southpole_A' ---------------------
-
-config['propagation']['birefringence'] = True
-config['propagation']['birefringence_model'] = 'southpole_A'
-config['propagation']['birefringence_propagation'] = 'numerical'
-
-birefringence_propagation(config, 'analytical', ray_tracing_solution, 2, x_limits, 'analyticraytracing.py', 'radioproparaytracing.py', 'southpole_A')
-
-
-# ---------------   raytracing:            radioproparaytracing.py ---------------------
-# ---------------   e-field propagation:   radioproparaytracing.py ---------------------
-# ---------------   biefringence model:    'southpole_A' ---------------------
-
-propa_config = dict()
-propa_config['propagation'] = dict(
-    attenuate_ice = True,
-    focusing_limit = 2,
-    focusing = False,
-    birefringence = True,
-    radiopropa = dict(
-        mode = 'iterative',
-        iter_steps_channel = [25., 2., .5], #unit is meter
-        iter_steps_zenith = [.5, .05, .005], #unit is degree
-        auto_step_size = False,
-        max_traj_length = 10000) #unit is meter
-)
-propa_config['speedup'] = dict(
-    delta_C_cut = 40 * units.degree
-)
-
-birefringence_propagation(propa_config, 'numerical', ray_tracing_solution, 3, x_limits, 'radioproparaytracing.py', 'radioproparaytracing.py', 'southpole_A')
+if propa:
+    # ---------------   raytracing:            analyticraytracing.py ---------------------
+    # ---------------   e-field propagation:   radioproparaytracing.py ---------------------
+    # ---------------   biefringence model:    'southpole_A' ---------------------
+
+    config['propagation']['birefringence'] = True
+    config['propagation']['birefringence_model'] = 'southpole_A'
+    config['propagation']['birefringence_propagation'] = 'numerical'
+
+    birefringence_propagation(config, 'analytical', ray_tracing_solution, 2, x_limits, 'analyticraytracing.py', 'radioproparaytracing.py', 'southpole_A')
+
+
+    # ---------------   raytracing:            radioproparaytracing.py ---------------------
+    # ---------------   e-field propagation:   radioproparaytracing.py ---------------------
+    # ---------------   biefringence model:    'southpole_A' ---------------------
+
+    propa_config = dict()
+    propa_config['propagation'] = dict(
+        attenuate_ice = True,
+        focusing_limit = 2,
+        focusing = False,
+        birefringence = True,
+        radiopropa = dict(
+            mode = 'iterative',
+            iter_steps_channel = [25., 2., .5], #unit is meter
+            iter_steps_zenith = [.5, .05, .005], #unit is degree
+            auto_step_size = False,
+            max_traj_length = 10000) #unit is meter
+    )
+    propa_config['speedup'] = dict(
+        delta_C_cut = 40 * units.degree
+    )
+
+    birefringence_propagation(propa_config, 'numerical', ray_tracing_solution, 3, x_limits, 'radioproparaytracing.py', 'radioproparaytracing.py', 'southpole_A')
 
 # ---------------   plotting the results ---------------------
 axs[0].legend(loc = 1)