dynamic scene

Author: Licini

CHANGELOG.md

@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Added
 
+* Added ability to add and remove objects in real-time after viewer starts running.
+
 ### Changed
 
 * Made `linewidth` working again through `GeometryShader`.

scripts/dynamic_scene.py

@@ -0,0 +1,47 @@
+from compas.geometry import Box
+from compas_viewer import Viewer
+from compas.colors import Color
+
+viewer = Viewer()
+
+boxes = [
+    Box.from_width_height_depth(1, 1, 1),
+    Box.from_width_height_depth(2, 2, 0.5),
+    Box.from_width_height_depth(3, 3, 0.2),
+]
+
+obj1 = viewer.scene.add(boxes[0], facecolor=Color.blue())
+
+objects_to_remove = [obj1]
+
+
+@viewer.on(interval=1000)
+def dynamic_update(frame):
+    """
+    This function is called every 1000ms for 6 frames.
+    """
+    print(f"Frame {frame}")
+
+    if frame == 1:
+        print("Adding second box...")
+        added_object = viewer.scene.add(boxes[1], facecolor=Color.red())
+        objects_to_remove.append(added_object)
+
+    if frame == 2:
+        print("Adding third box...")
+        added_object = viewer.scene.add(boxes[2], facecolor=Color.green())
+        objects_to_remove.append(added_object)
+
+    if frame == 4:
+        print("Removing first box...")
+        viewer.scene.remove(objects_to_remove.pop(-1))
+
+    if frame == 5:
+        print("Removing second box...")
+        viewer.scene.remove(objects_to_remove.pop(-1))
+
+    if frame == 6:
+        print("Removing third box...")
+        viewer.scene.remove(objects_to_remove.pop(-1))
+
+viewer.show()

src/compas_viewer/renderer/renderer.py

@@ -445,6 +445,22 @@ def init(self):
         self.shader_lines.uniformBuffer("settingsBuffer", self.buffer_manager.settings_texture, unit=1)
         self.shader_lines.release()
 
+    def rebuild_buffers(self):
+        """Rebuild the buffers."""
+        GL.glBindVertexArray(self._vao)
+        self.buffer_manager.clear()
+
+        # Ensure all objects are initialized before adding to buffer
+        for obj in self.viewer.scene.objects:
+            if not isinstance(obj, (Group, TagObject)) and not obj._inited:
+                obj.init()
+
+        for obj in self.viewer.scene.objects:
+            if not isinstance(obj, (Group, TagObject)):
+                self.buffer_manager.add_object(obj)
+        self.buffer_manager.create_buffers()
+        GL.glBindVertexArray(0)
+
     def update_projection(self, w=None, h=None):
         """
         Update the projection matrix.

src/compas_viewer/scene/buffermanager.py

@@ -281,6 +281,7 @@ def clear(self) -> None:
             self.elements[buffer_type] = np.array([], dtype=np.int32)
             self.buffer_ids[buffer_type] = {}
 
+        self.objects = {}
         self.transforms = []
         self.settings = []
         self.object_settings_cache = {}

src/compas_viewer/scene/scene.py

@@ -177,6 +177,12 @@ def add(
             u=u,
             **kwargs,
         )
+
+        if self.viewer.running:
+            self.viewer.renderer.rebuild_buffers()
+            self.viewer.renderer.update()
+            self.viewer.ui.sidebar.update()
+
         return sceneobject
 
     def add_group(self, name: str = None, parent: Optional[ViewerSceneObject] = None, **kwargs) -> Group:
@@ -200,3 +206,19 @@ def add_group(self, name: str = None, parent: Optional[ViewerSceneObject] = None
         group = Group(name=name, **kwargs)
         self.add(group, parent=parent)
         return group
+
+    def remove(self, sceneobject: ViewerSceneObject):
+        """
+        Remove an scene object from the scene.
+
+        Parameters
+        ----------
+        sceneobject : :class:`compas_viewer.scene.ViewerSceneObject`
+            The scene object to remove.
+        """
+        super().remove(sceneobject)
+
+        if self.viewer.running:
+            self.viewer.renderer.rebuild_buffers()
+            self.viewer.renderer.update()
+            self.viewer.ui.sidebar.update()