Skip to content

Commit 0d98ad0

Browse files
fix: have new elements pick-up the indexes of the already-existing elements, with 'None' for nonexisting steps
1 parent 56a80d5 commit 0d98ad0

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/gh/diffCheck/diffCheck/df_poses.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,9 @@ def add_step(self, new_poses: list[DFPose]):
5353
for i, pose in enumerate(new_poses):
5454
if f"element_{i}" not in self.poses_per_element_dictionary:
5555
self.poses_per_element_dictionary[f"element_{i}"] = DFPosesBeam({}, 4)
56-
self.poses_per_element_dictionary[f"element_{i}"].add_pose(pose, self.n_step + 1)
56+
for j in range(self.n_step):
57+
self.poses_per_element_dictionary[f"element_{i}"].add_pose(None, j)
58+
self.poses_per_element_dictionary[f"element_{i}"].add_pose(pose, self.n_step)
5759
self.n_step += 1
5860

5961
def get_last_poses(self):
@@ -64,7 +66,7 @@ def get_last_poses(self):
6466
return None
6567
last_poses = []
6668
for i in range(len(self.poses_per_element_dictionary)):
67-
last_poses.append(self.poses_per_element_dictionary[f"element_{i}"].poses_dictionnary[f"pose_{self.n_step}"])
69+
last_poses.append(self.poses_per_element_dictionary[f"element_{i}"].poses_dictionnary[f"pose_{self.n_step-1}"])
6870
return last_poses
6971

7072
def reset(self):

0 commit comments

Comments
 (0)