wip: update example files (some examples have bugs. will create an is…

…sue to be fixed)

.gitignore

@@ -7,6 +7,7 @@ __pycache__/
 *.so
 
 *.blend
+*.blend1
 *.swp
 *.npz
 

examples/br2/br2_pushed.py

@@ -75,7 +75,6 @@ def setup(
 
 class BR2Environment(BaseEnvironment):
     def __init__(self, *args, **kwargs) -> None:
-        bsr.clear_mesh_objects()
         super().__init__(*args, **kwargs)
 
     def setup(
@@ -209,6 +208,8 @@ def main(
     time_step: float = 1.0e-5,
     recording_fps: int = 30,
 ):
+    bsr.clear_mesh_objects()
+
     # Initialize the environment
     env = BR2Environment(
         final_time=final_time, time_step=time_step, recording_fps=recording_fps
@@ -221,6 +222,21 @@ def main(
         time = env.step(time=time)
     print("Simulation finished!")
 
+    # Set the final keyframe number
+    bsr.frame_manager.set_frame_end()
+
+    # Set the frame rate
+    bsr.frame_manager.set_frame_rate(fps=recording_fps)
+
+    # Set the view distance
+    bsr.set_view_distance(distance=1)
+
+    # Deslect all objects
+    bsr.deselect_all()
+
+    # Select the camera object
+    bsr.camera_manager.select()
+
     # Save the simulation
     env.save("BR2_simulation")
 

examples/br2/callbacks.py

@@ -160,12 +160,11 @@ def __init__(
             default_centerline_position=system.position_collection,
             default_centerline_director=system.director_collection,
         )
-        self.keyframe = 0
 
     def save_params(self, system: ea.CosseratRod, time: float) -> None:
         self.bpy_objs.update_states(
             centerline_position=system.position_collection,
             centerline_director=system.director_collection,
         )
-        self.bpy_objs.set_keyframe(self.keyframe)
-        self.keyframe += 1
+        self.bpy_objs.set_keyframe(bsr.frame_manager.current_frame)
+        bsr.frame_manager.update()

examples/camera_movement.py

@@ -15,17 +15,15 @@ def main(filename: str = "camera_movement"):
 
     bsr.camera_manager.look_at = np.array([0.0, 0.0, 0.0])
 
-    for k, angle in enumerate(
-        np.linspace(0.0, 360.0, frame_rate * total_time + 1)
-    ):
+    for angle in np.linspace(0.0, 360.0, frame_rate * total_time + 1):
         bsr.camera_manager.location = np.array(
             [
                 camera_radius * np.cos(np.radians(angle)),
                 camera_radius * np.sin(np.radians(angle)),
                 camera_heigh,
             ]
         )
-        bsr.camera_manager.set_keyframe(k)
+        bsr.camera_manager.set_keyframe(bsr.frame_manager.current_frame)
         bsr.frame_manager.update()
 
     # Set the final keyframe number