feat: add any_operator support

Author: Matteo-Baussart-ANSYS

src/ansys/dpf/core/any.py

@@ -122,6 +122,7 @@ def _type_to_new_from_get_as_method(self, obj):
             data_tree,
             custom_type_field,
             collection,
+            dpf_operator,
         )
 
         if issubclass(obj, int):
@@ -190,6 +191,11 @@ def _type_to_new_from_get_as_method(self, obj):
                 self._api.any_new_from_int_collection,
                 self._api.any_get_as_int_collection,
             )
+        elif issubclass(obj, dpf_operator.Operator):
+            return (
+                self._api.any_new_from_operator,
+                self._api.any_get_as_operator,
+            )
 
     @staticmethod
     def new_from(obj, server=None):

src/ansys/dpf/core/dpf_operator.py

@@ -121,7 +121,7 @@ class Operator:
 
     """
 
-    def __init__(self, name, config=None, server=None):
+    def __init__(self, name=None, operator=None, config=None, server=None):
         """Initialize the operator with its name by connecting to a stub."""
         self.name = name
         self._internal_obj = None
@@ -136,14 +136,32 @@ def __init__(self, name, config=None, server=None):
         # step 2: get api
         self._api_instance = None  # see _api property
 
-        # step3: init environment
-        self._api.init_operator_environment(self)  # creates stub when gRPC
 
-        # step4: if object exists: take instance, else create it (server)
-        if self._server.has_client():
-            self._internal_obj = self._api.operator_new_on_client(self.name, self._server.client)
+        # step2: if object exists, take the instance, else create it
+        if operator is not None:
+            if isinstance(operator, Operator):
+                self._server = operator._server
+                self._api = self._server.get_api_for_type(
+                    capi=operator_capi.OperatorCAPI,
+                    grpcapi=operator_grpcapi.OperatorGRPCAPI,
+                )
+                # step3: init environment
+                self._api.init_operator_environment(self)  # creates stub when gRPC
+                core_api = self._server.get_api_for_type(
+                    capi=data_processing_capi.DataProcessingCAPI,
+                    grpcapi=data_processing_grpcapi.DataProcessingGRPCAPI,
+                )
+                core_api.init_data_processing_environment(self)
+                self._internal_obj = core_api.data_processing_duplicate_object_reference(operator)
+                self.name = operator.name
+            else:
+                self._internal_obj = operator
+                self.name = self._api.operator_name(self)
         else:
-            self._internal_obj = self._api.operator_new(self.name)
+            if self._server.has_client():
+                self._internal_obj = self._api.operator_new_on_client(self.name, self._server.client)
+            else:
+                self._internal_obj = self._api.operator_new(self.name)
 
         if self._internal_obj is None:
             raise KeyError(

src/ansys/dpf/gate/any_grpcapi.py

@@ -41,6 +41,7 @@ def _type_to_message_type():
             data_tree,
             custom_type_field,
             collection_base,
+            dpf_operator,
         )
 
         return [(int, base_pb2.Type.INT),
@@ -56,6 +57,7 @@ def _type_to_message_type():
                 (data_tree.DataTree, base_pb2.Type.DATA_TREE),
                 (collection_base.CollectionBase, base_pb2.Type.COLLECTION, base_pb2.Type.ANY),
                 (dpf_vector.DPFVectorInt, base_pb2.Type.COLLECTION, base_pb2.Type.INT),
+                (dpf_operator.Operator, base_pb2.Type.OPERATOR),
                 ]
 
     @staticmethod
@@ -139,6 +141,10 @@ def any_get_as_any_collection(any):
     def any_get_as_int_collection(any):
         return AnyGRPCAPI._get_as(any).collection
 
+    @staticmethod
+    def any_get_as_operator(any):
+        return AnyGRPCAPI._get_as(any).operator
+
     @staticmethod
     def _new_from(any, client=None):
         from ansys.grpc.dpf import dpf_any_pb2
@@ -220,3 +226,7 @@ def any_new_from_scoping(any):
     @staticmethod
     def any_new_from_data_tree(any):
         return AnyGRPCAPI._new_from(any, any._server)
+
+    @staticmethod
+    def any_new_from_operator(any):
+        return AnyGRPCAPI._new_from(any, any._server)

src/ansys/dpf/gate/operator_grpcapi.py

@@ -51,6 +51,10 @@ def get_list(op):
     def operator_get_config(op):
         return OperatorGRPCAPI.get_list(op).config
 
+    @staticmethod
+    def operator_name(op):
+        return OperatorGRPCAPI.get_list(op).op_name
+
     @staticmethod
     def update_init(op, pin):
         from ansys.grpc.dpf import operator_pb2

tests/test_any.py

@@ -120,3 +120,12 @@ def test_cast_scoping_any(server_type):
     new_entity = any_dpf.cast()
 
     assert entity.location == new_entity.location
+
+
+@conftest.raises_for_servers_version_under("7.0")
+def test_cast_operator_any(server_type):
+    entity = dpf.Operator(server=server_type, name="U")
+    any_dpf = dpf.Any.new_from(entity)
+    new_entity = any_dpf.cast()
+
+    assert entity.name == new_entity.name