Allow no velocities for instrument/sun position

DOI-USGS · Sep 9, 2024 · 69b7604 · 69b7604
1 parent 0ba7b24
commit 69b7604
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/ale/formatters/formatter.py b/ale/formatters/formatter.py
@@ -176,7 +176,11 @@ def to_isd(driver):
     instrument_position['spk_table_original_size'] = len(times)
     instrument_position['ephemeris_times'] = times
     # Rotate positions and velocities into J2000 then scale into kilometers
-    velocities = j2000_rotation.rotate_velocity_at(positions, velocities, times)/1000
+    # If velocities are provided, then rotate, otherwise create array of 0s to match shape of position
+    if velocities is not None:
+        velocities = j2000_rotation.rotate_velocity_at(positions, velocities, times)/1000
+    else:
+        velocities = np.zeros(positions.shape)
     positions = j2000_rotation.apply_at(positions, times)/1000
     instrument_position['positions'] = positions
     instrument_position['velocities'] = velocities
@@ -191,7 +195,11 @@ def to_isd(driver):
     sun_position['spk_table_original_size'] = len(times)
     sun_position['ephemeris_times'] = times
     # Rotate positions and velocities into J2000 then scale into kilometers
-    velocities = j2000_rotation.rotate_velocity_at(positions, velocities, times)/1000
+    # If velocities are provided, then rotate, otherwise create array of 0s to match shape of position
+    if velocities is not None:
+        velocities = j2000_rotation.rotate_velocity_at(positions, velocities, times)/1000
+    else:
+        velocities = np.zeros(positions.shape)
     positions = j2000_rotation.apply_at(positions, times)/1000
     sun_position['positions'] = positions
     sun_position['velocities'] = velocities