From 290aaf809a4a5646dbde05a139ab59899eb79d6a Mon Sep 17 00:00:00 2001
From: Andi <andi.gerken@gmail.com>
Date: Tue, 25 Mar 2025 12:25:56 +0100
Subject: [PATCH] Fixed rendering for files without zone information.
---
src/robofish/io/file.py | 19 ++++++++++---------
1 file changed, 10 insertions(+), 9 deletions(-)
diff --git a/src/robofish/io/file.py b/src/robofish/io/file.py
index 0afc5cd..e734258 100644
--- a/src/robofish/io/file.py
+++ b/src/robofish/io/file.py
@@ -1437,16 +1437,17 @@ class File(h5py.File):
ori_deg = np.rad2deg(self.fov_orientations[i_entity])
- for zone in zones[i_entity]:
- zone.set_center(
- (
- this_pose[i_entity, 0],
- this_pose[i_entity, 1],
+ if len(zones) > i_entity:
+ for zone in zones[i_entity]:
+ zone.set_center(
+ (
+ this_pose[i_entity, 0],
+ this_pose[i_entity, 1],
+ )
)
- )
- if fovs[i_entity] < 360:
- zone.theta1 = ori_deg - fovs[i_entity] / 2
- zone.theta2 = ori_deg + fovs[i_entity] / 2
+ if fovs[i_entity] < 360:
+ zone.theta1 = ori_deg - fovs[i_entity] / 2
+ zone.theta2 = ori_deg + fovs[i_entity] / 2
if render_swarm_center:
swarm_center[0].set_offsets(swarm_center_position[file_frame])
--
GitLab