feat: update with delta (#1922)

Co-authored-by: Roberto Pastor Muela <37798125+RobPasMue@users.noreply.github.com>
Co-authored-by: pyansys-ci-bot <92810346+pyansys-ci-bot@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: 4 people

doc/changelog.d/1922.added.md

@@ -0,0 +1 @@
+Update with delta

src/ansys/geometry/core/__init__.py

@@ -70,3 +70,6 @@
 
 DOCUMENTATION_BUILD: bool = os.environ.get("PYANSYS_GEOMETRY_DOC_BUILD", "false").lower() == "true"
 """Global flag for the documentation to use the proper PyVista Jupyter backend."""
+
+USE_TRACKER_TO_UPDATE_DESIGN: bool = False
+"""Global constant for checking whether to use the tracker to update designs."""

src/ansys/geometry/core/_grpc/_services/v0/repair_tools.py

@@ -344,13 +344,18 @@ def find_and_fix_simplify(self, **kwargs) -> dict:  # noqa: D102
         # Call the gRPC service
         response = self.stub.FindAndSimplify(request)
 
+        serialized_tracker_response = self._serialize_tracker_command_response(
+            response.complete_command_response
+        )
+
         # Return the response - formatted as a dictionary
         return {
             "success": response.success,
             "found": response.found,
             "repaired": response.repaired,
             "created_bodies_monikers": [],
             "modified_bodies_monikers": [],
+            "complete_command_response": serialized_tracker_response,
         }
 
     @protect_grpc
@@ -378,13 +383,18 @@ def find_and_fix_stitch_faces(self, **kwargs) -> dict:  # noqa: D102
         # Call the gRPC service
         response = self.stub.FindAndFixStitchFaces(request)
 
+        serialized_tracker_response = self._serialize_tracker_command_response(
+            response.complete_command_response
+        )
+
         # Return the response - formatted as a dictionary
         return {
             "success": response.success,
             "created_bodies_monikers": response.created_bodies_monikers,
             "modified_bodies_monikers": response.modified_bodies_monikers,
             "found": response.found,
             "repaired": response.repaired,
+            "complete_command_response": serialized_tracker_response,
         }
 
     @protect_grpc
@@ -457,13 +467,18 @@ def find_and_fix_short_edges(self, **kwargs):  # noqa: D102
         # Call the gRPC service
         response = self.stub.FindAndFixShortEdges(request)
 
+        serialized_tracker_response = self._serialize_tracker_command_response(
+            response.complete_command_response
+        )
+
         # Return the response - formatted as a dictionary
         return {
             "success": response.success,
             "found": response.found,
             "repaired": response.repaired,
             "created_bodies_monikers": [],
             "modified_bodies_monikers": [],
+            "complete_command_response": serialized_tracker_response,
         }
 
     @protect_grpc
@@ -479,13 +494,18 @@ def find_and_fix_extra_edges(self, **kwargs) -> dict:  # noqa: D102
         # Call the gRPC service
         response = self.stub.FindAndFixExtraEdges(request)
 
+        serialized_tracker_response = self._serialize_tracker_command_response(
+            response.complete_command_response
+        )
+
         # Return the response - formatted as a dictionary
         return {
             "success": response.success,
             "found": response.found,
             "repaired": response.repaired,
             "created_bodies_monikers": response.created_bodies_monikers,
             "modified_bodies_monikers": response.modified_bodies_monikers,
+            "complete_command_response": serialized_tracker_response,
         }
 
     @protect_grpc
@@ -505,13 +525,18 @@ def find_and_fix_split_edges(self, **kwargs) -> dict:  # noqa: D102
         # Call the gRPC service
         response = self.stub.FindAndFixSplitEdges(request)
 
+        serialized_tracker_response = self._serialize_tracker_command_response(
+            response.complete_command_response
+        )
+
         # Return the response - formatted as a dictionary
         return {
             "success": response.success,
             "found": response.found,
             "repaired": response.repaired,
             "created_bodies_monikers": [],
             "modified_bodies_monikers": [],
+            "complete_command_response": serialized_tracker_response,
         }
 
     def __serialize_inspect_result_response(self, response) -> dict:  # noqa: D102
@@ -567,3 +592,52 @@ def serialize_issue(issue):
                 for body_issues in response.issues_by_body
             ]
         }
+
+    def _serialize_tracker_command_response(self, response) -> dict:
+        """Serialize a TrackerCommandResponse object into a dictionary.
+
+        Parameters
+        ----------
+        response : TrackerCommandResponse
+            The gRPC TrackerCommandResponse object to serialize.
+
+        Returns
+        -------
+        dict
+            A dictionary representation of the TrackerCommandResponse object.
+        """
+
+        def serialize_body(body):
+            return {
+                "id": body.id,
+                "name": body.name,
+                "can_suppress": body.can_suppress,
+                "transform_to_master": {
+                    "m00": body.transform_to_master.m00,
+                    "m11": body.transform_to_master.m11,
+                    "m22": body.transform_to_master.m22,
+                    "m33": body.transform_to_master.m33,
+                },
+                "master_id": body.master_id,
+                "parent_id": body.parent_id,
+            }
+
+        def serialize_entity_identifier(entity):
+            """Serialize an EntityIdentifier object into a dictionary."""
+            return {
+                "id": entity.id,
+            }
+
+        return {
+            "success": response.success,
+            "created_bodies": [
+                serialize_body(body) for body in getattr(response, "created_bodies", [])
+            ],
+            "modified_bodies": [
+                serialize_body(body) for body in getattr(response, "modified_bodies", [])
+            ],
+            "deleted_bodies": [
+                serialize_entity_identifier(entity)
+                for entity in getattr(response, "deleted_bodies", [])
+            ],
+        }

src/ansys/geometry/core/designer/body.py

@@ -1940,7 +1940,16 @@ def __generic_boolean_command(
             for b in other_bodies:
                 b.parent_component.delete_body(b)
 
-        parent_design._update_design_inplace()
+        from ansys.geometry.core import USE_TRACKER_TO_UPDATE_DESIGN
+
+        if not USE_TRACKER_TO_UPDATE_DESIGN:
+            parent_design._update_design_inplace()
+        else:
+            # If USE_TRACKER_TO_UPDATE_DESIGN is True, we serialize the response
+            # and update the parent design with the serialized response.
+            tracker_response = response.result.complete_command_response
+            serialized_response = self._serialize_tracker_command_response(tracker_response)
+            parent_design._update_from_tracker(serialized_response)
 
     @reset_tessellation_cache
     @ensure_design_is_active

src/ansys/geometry/core/designer/design.py

@@ -1254,3 +1254,141 @@ def _update_design_inplace(self) -> None:
 
         # Read the existing design
         self.__read_existing_design()
+
+    def _update_from_tracker(self, tracker_response: list[dict]):
+        """Update the design with the changed bodies while preserving unchanged ones."""
+        self._grpc_client.log.debug(
+            f"Starting _update_from_tracker with response: {tracker_response}"
+        )
+        self._handle_modified_bodies(tracker_response.get("modified_bodies", []))
+        self._handle_deleted_bodies(tracker_response.get("deleted_bodies", []))
+        self._handle_created_bodies(tracker_response.get("created_bodies", []))
+
+    def _handle_modified_bodies(self, modified_bodies):
+        for body_info in modified_bodies:
+            body_id = body_info["id"]
+            body_name = body_info["name"]
+            self._grpc_client.log.debug(
+                f"Processing modified body: ID={body_id}, Name='{body_name}'"
+            )
+            updated = False
+
+            for body in self.bodies:
+                if body.id == body_id:
+                    self._update_body(body, body_info)
+                    updated = True
+                    self._grpc_client.log.debug(
+                        f"Modified body '{body_name}' (ID: {body_id}) updated at root level."
+                    )
+                    break
+
+            if not updated:
+                for component in self.components:
+                    if self._find_and_update_body(body_info, component):
+                        break
+
+    def _handle_deleted_bodies(self, deleted_bodies):
+        for body_info in deleted_bodies:
+            body_id = body_info["id"]
+            self._grpc_client.log.debug(f"Processing deleted body: ID={body_id}")
+            removed = False
+
+            for body in self.bodies:
+                if body.id == body_id:
+                    body._is_alive = False
+                    self.bodies.remove(body)
+                    removed = True
+                    self._grpc_client.log.info(
+                        f"Deleted body (ID: {body_id}) removed from root level."
+                    )
+                    break
+
+            if not removed:
+                for component in self.components:
+                    if self._find_and_remove_body(body_info, component):
+                        break
+
+    def _handle_created_bodies(self, created_bodies):
+        for body_info in created_bodies:
+            body_id = body_info["id"]
+            body_name = body_info["name"]
+            is_surface = body_info.get("is_surface", False)
+            self._grpc_client.log.debug(
+                f"Processing created body: ID={body_id}, Name='{body_name}'"
+            )
+
+            if any(body.id == body_id for body in self.bodies):
+                self._grpc_client.log.debug(
+                    f"Created body '{body_name}' (ID: {body_id}) already exists at root level."
+                )
+                continue
+
+            added = any(self._find_and_add_body(body_info, self.components))
+            if not added:
+                new_body = MasterBody(body_id, body_name, self._grpc_client, is_surface=is_surface)
+                self.bodies.append(new_body)
+                self._grpc_client.log.debug(
+                    f"Added new body '{body_name}' (ID: {body_id}) to root level."
+                )
+
+    def _update_body(self, existing_body, body_info):
+        self._grpc_client.log.debug(
+            f"Updating body '{existing_body.name}' "
+            f"(ID: {existing_body.id}) with new info: {body_info}"
+        )
+        existing_body.name = body_info["name"]
+        existing_body._template._is_surface = body_info.get("is_surface", False)
+
+    def _find_and_add_body(self, body_info, components):
+        for component in components:
+            if component.id == body_info["parent_id"]:
+                new_body = MasterBody(
+                    body_info["id"],
+                    body_info["name"],
+                    self._grpc_client,
+                    is_surface=body_info.get("is_surface", False),
+                )
+                component.bodies.append(new_body)
+                self._grpc_client.log.debug(
+                    f"Added new body '{new_body.name}' (ID: {new_body.id}) "
+                    f"to component '{component.name}' (ID: {component.id})"
+                )
+                return True
+
+            if self._find_and_add_body(body_info, component.components):
+                return True
+
+        return False
+
+    def _find_and_update_body(self, body_info, component):
+        for body in component.bodies:
+            if body.id == body_info["id"]:
+                self._update_body(body, body_info)
+                self._grpc_client.log.debug(
+                    f"Updated body '{body.name}' (ID: {body.id}) in component "
+                    f"'{component.name}' (ID: {component.id})"
+                )
+                return True
+
+        for subcomponent in component.components:
+            if self._find_and_update_body(body_info, subcomponent):
+                return True
+
+        return False
+
+    def _find_and_remove_body(self, body_info, component):
+        for body in component.bodies:
+            if body.id == body_info["id"]:
+                body._is_alive = False
+                component.bodies.remove(body)
+                self._grpc_client.log.debug(
+                    f"Removed body '{body_info['name']}' (ID: {body_info['id']}) from component "
+                    f"'{component.name}' (ID: {component.id})"
+                )
+                return True
+
+        for subcomponent in component.components:
+            if self._find_and_remove_body(body_info, subcomponent):
+                return True
+
+        return False

src/ansys/geometry/core/tools/prepare_tools.py

@@ -270,7 +270,9 @@ def enhanced_share_topology(
         from ansys.geometry.core.designer.body import Body
 
         if not bodies:
-            return RepairToolMessage(False, [], [], 0, 0)
+            return RepairToolMessage(
+                success=False, created_bodies=[], modified_bodies=[], found=0, repaired=0
+            )
 
         # Verify inputs
         check_type_all_elements_in_iterable(bodies, Body)
@@ -282,11 +284,11 @@ def enhanced_share_topology(
         )
 
         message = RepairToolMessage(
-            response.get("success"),
-            response.get("created_bodies_monikers"),
-            response.get("modified_bodies_monikers"),
-            response.get("found"),
-            response.get("repaired"),
+            success=response.get("success"),
+            created_bodies=response.get("created_bodies_monikers"),
+            modified_bodies=response.get("modified_bodies_monikers"),
+            found=response.get("found"),
+            repaired=response.get("repaired"),
         )
         return message