ansys-fluent-core 0.28.dev0__py3-none-any.whl → 0.28.2__py3-none-any.whl

ansys/fluent/core/services/datamodel_se.py

@@ -6,7 +6,7 @@ import itertools
 import logging
 import os
 from threading import RLock
-from typing import Any, Callable, Iterator, NoReturn, Sequence
+from typing import Any, Callable, Iterator, NoReturn, Sequence, TypeVar
 
 from google.protobuf.json_format import MessageToDict, ParseDict
 import grpc
@@ -28,7 +28,8 @@ from ansys.fluent.core.solver.error_message import allowed_name_error_message
 from ansys.fluent.core.utils.fluent_version import FluentVersion
 
 Path = list[tuple[str, str]]
-_TValue = None | bool | int | float | str | Sequence["_TValue"] | dict[str, "_TValue"]
+PyMenuT = TypeVar("PyMenuT", bound="PyMenu")
+ValueT = None | bool | int | float | str | Sequence["ValueT"] | dict[str, "ValueT"]
 logger: logging.Logger = logging.getLogger("pyfluent.datamodel")
 
 member_specs_oneof_fields = [
@@ -301,7 +302,7 @@ class DatamodelServiceImpl:
         return self._stub.unsubscribeEvents(request, metadata=self._metadata)
 
 
-def _convert_value_to_variant(val: _TValue, var: Variant) -> None:
+def _convert_value_to_variant(val: ValueT, var: Variant) -> None:
     """Convert a Python data type to Fluent's variant type."""
     if isinstance(val, bool):
         var.bool_state = val
@@ -322,7 +323,7 @@ def _convert_value_to_variant(val: _TValue, var: Variant) -> None:
             _convert_value_to_variant(v, var.variant_map_state.item[k])
 
 
-def _convert_variant_to_value(var: Variant) -> _TValue:
+def _convert_variant_to_value(var: Variant) -> ValueT:
     """Convert Fluent's variant type to a Python data type."""
     if var.HasField("bool_state"):
         return var.bool_state
@@ -495,7 +496,7 @@ class DatamodelService(StreamingService):
         self.cache = DataModelCache() if pyfluent.DATAMODEL_USE_STATE_CACHE else None
         self.version = version
 
-    def get_attribute_value(self, rules: str, path: str, attribute: str) -> _TValue:
+    def get_attribute_value(self, rules: str, path: str, attribute: str) -> ValueT:
         """Get attribute value."""
         request = DataModelProtoModule.GetAttributeValueRequest(
             rules=rules, path=path, attribute=attribute
@@ -503,7 +504,7 @@ class DatamodelService(StreamingService):
         response = self._impl.get_attribute_value(request)
         return _convert_variant_to_value(response.result)
 
-    def get_state(self, rules: str, path: str) -> _TValue:
+    def get_state(self, rules: str, path: str) -> ValueT:
         """Get state."""
         request = DataModelProtoModule.GetStateRequest(rules=rules, path=path)
         response = self._impl.get_state(request)
@@ -568,7 +569,7 @@ class DatamodelService(StreamingService):
                 version=self.version,
             )
 
-    def set_state(self, rules: str, path: str, state: _TValue) -> None:
+    def set_state(self, rules: str, path: str, state: ValueT) -> None:
         """Set state."""
         request = DataModelProtoModule.SetStateRequest(
             rules=rules, path=path, wait=True
@@ -601,7 +602,7 @@ class DatamodelService(StreamingService):
         self,
         rules: str,
         path: str,
-        dict_state: dict[str, _TValue],
+        dict_state: dict[str, ValueT],
         recursive=False,
     ) -> None:
         """Update the dict."""
@@ -633,8 +634,8 @@ class DatamodelService(StreamingService):
             )
 
     def execute_command(
-        self, rules: str, path: str, command: str, args: dict[str, _TValue]
-    ) -> _TValue:
+        self, rules: str, path: str, command: str, args: dict[str, ValueT]
+    ) -> ValueT:
         """Execute the command."""
         request = DataModelProtoModule.ExecuteCommandRequest(
             rules=rules, path=path, command=command, wait=True
@@ -651,8 +652,8 @@ class DatamodelService(StreamingService):
         return _convert_variant_to_value(response.result)
 
     def execute_query(
-        self, rules: str, path: str, query: str, args: dict[str, _TValue]
-    ) -> _TValue:
+        self, rules: str, path: str, query: str, args: dict[str, ValueT]
+    ) -> ValueT:
         """Execute the query."""
         request = DataModelProtoModule.ExecuteQueryRequest(
             rules=rules, path=path, query=query
@@ -722,7 +723,7 @@ class DatamodelService(StreamingService):
         self.subscriptions.unsubscribe_all()
 
     def add_on_child_created(
-        self, rules: str, path: str, child_type: str, obj, cb: Callable
+        self, rules: str, path: str, child_type: str, cb: Callable[[str], None]
     ) -> EventSubscription:
         """Add on child created."""
         request_dict = {
@@ -737,11 +738,18 @@ class DatamodelService(StreamingService):
             ]
         }
         subscription = EventSubscription(self, path, request_dict)
-        self.event_streaming.register_callback(subscription.tag, obj, cb)
+
+        def cb_grpc(child_type: str, child_name: str):
+            ppath = convert_se_path_to_path(path)
+            ppath.append((child_type, child_name))
+            child_path = convert_path_to_se_path(ppath)
+            cb(child_path)
+
+        self.event_streaming.register_callback(subscription.tag, cb_grpc)
         return subscription
 
     def add_on_deleted(
-        self, rules: str, path: str, obj, cb: Callable
+        self, rules: str, path: str, cb: Callable[[], None]
     ) -> EventSubscription:
         """Add on deleted."""
         request_dict = {
@@ -753,11 +761,11 @@ class DatamodelService(StreamingService):
             ]
         }
         subscription = EventSubscription(self, path, request_dict)
-        self.event_streaming.register_callback(subscription.tag, obj, cb)
+        self.event_streaming.register_callback(subscription.tag, cb)
         return subscription
 
     def add_on_changed(
-        self, rules: str, path: str, obj, cb: Callable
+        self, rules: str, path: str, cb: Callable[[ValueT], None]
     ) -> EventSubscription:
         """Add on changed."""
         request_dict = {
@@ -769,11 +777,11 @@ class DatamodelService(StreamingService):
             ]
         }
         subscription = EventSubscription(self, path, request_dict)
-        self.event_streaming.register_callback(subscription.tag, obj, cb)
+        self.event_streaming.register_callback(subscription.tag, cb)
         return subscription
 
     def add_on_affected(
-        self, rules: str, path: str, obj, cb: Callable
+        self, rules: str, path: str, cb: Callable[[], None]
     ) -> EventSubscription:
         """Add on affected."""
         request_dict = {
@@ -785,11 +793,11 @@ class DatamodelService(StreamingService):
             ]
         }
         subscription = EventSubscription(self, path, request_dict)
-        self.event_streaming.register_callback(subscription.tag, obj, cb)
+        self.event_streaming.register_callback(subscription.tag, cb)
         return subscription
 
     def add_on_affected_at_type_path(
-        self, rules: str, path: str, child_type: str, obj, cb: Callable
+        self, rules: str, path: str, child_type: str, cb: Callable[[], None]
     ) -> EventSubscription:
         """Add on affected at type path."""
         request_dict = {
@@ -804,11 +812,16 @@ class DatamodelService(StreamingService):
             ]
         }
         subscription = EventSubscription(self, path, request_dict)
-        self.event_streaming.register_callback(subscription.tag, obj, cb)
+        self.event_streaming.register_callback(subscription.tag, cb)
         return subscription
 
-    def add_on_command_executed(
-        self, rules: str, path: str, command: str, obj, cb: Callable
+    def add_on_command_executed_old(
+        self,
+        rules: str,
+        path: str,
+        command: str,
+        obj,
+        cb: Callable[[str, ValueT], None],
     ) -> EventSubscription:
         """Add on command executed."""
         request_dict = {
@@ -823,11 +836,29 @@ class DatamodelService(StreamingService):
             ]
         }
         subscription = EventSubscription(self, path, request_dict)
-        self.event_streaming.register_callback(subscription.tag, obj, cb)
+        self.event_streaming.register_callback(subscription.tag, cb)
+        return subscription
+
+    def add_on_command_executed(
+        self, rules: str, path: str, cb: Callable[[str, ValueT], None]
+    ) -> EventSubscription:
+        """Add on command executed."""
+        request_dict = {
+            "eventrequest": [
+                {
+                    "rules": rules,
+                    "commandExecutedEventRequest": {
+                        "path": path,
+                    },
+                }
+            ]
+        }
+        subscription = EventSubscription(self, path, request_dict)
+        self.event_streaming.register_callback(subscription.tag, cb)
         return subscription
 
     def add_on_attribute_changed(
-        self, rules: str, path: str, attribute: str, obj, cb: Callable
+        self, rules: str, path: str, attribute: str, cb: Callable[[ValueT], None]
     ) -> EventSubscription:
         """Add on attribute changed."""
         request_dict = {
@@ -842,11 +873,16 @@ class DatamodelService(StreamingService):
             ]
         }
         subscription = EventSubscription(self, path, request_dict)
-        self.event_streaming.register_callback(subscription.tag, obj, cb)
+        self.event_streaming.register_callback(subscription.tag, cb)
         return subscription
 
     def add_on_command_attribute_changed(
-        self, rules: str, path: str, command: str, attribute: str, obj, cb: Callable
+        self,
+        rules: str,
+        path: str,
+        command: str,
+        attribute: str,
+        cb: Callable[[ValueT], None],
     ) -> EventSubscription:
         """Add on command attribute changed."""
         request_dict = {
@@ -862,7 +898,7 @@ class DatamodelService(StreamingService):
             ]
         }
         subscription = EventSubscription(self, path, request_dict)
-        self.event_streaming.register_callback(subscription.tag, obj, cb)
+        self.event_streaming.register_callback(subscription.tag, cb)
         return subscription
 
 
@@ -887,6 +923,30 @@ def convert_path_to_se_path(path: Path) -> str:
     return se_path
 
 
+def convert_se_path_to_path(se_path: str) -> Path:
+    """Convert a StateEngine path to a path structure.
+
+    Parameters
+    ----------
+    se_path : str
+        StateEngine path.
+
+    Returns
+    -------
+    Path
+        path structure
+    """
+    path = []
+    for comp in se_path.split("/"):
+        if comp:
+            if ":" in comp:
+                name, value = comp.split(":")
+            else:
+                name, value = comp, ""
+            path.append((name, value))
+    return path
+
+
 class PyCallableStateObject:
     """Any object which can be called to get its state.
 
@@ -1048,7 +1108,7 @@ class PyStateContainer(PyCallableStateObject):
             return self.get_state()
 
     def add_on_attribute_changed(
-        self, attribute: str, cb: Callable
+        self, attribute: str, cb: Callable[[ValueT], None]
     ) -> EventSubscription:
         """Register a callback for when an attribute is changed.
 
@@ -1056,7 +1116,7 @@ class PyStateContainer(PyCallableStateObject):
         ----------
         attribute : str
             attribute name
-        cb : Callable
+        cb : Callable[[ValueT], None]
             Callback function
 
         Returns
@@ -1065,11 +1125,11 @@ class PyStateContainer(PyCallableStateObject):
             EventSubscription instance which can be used to unregister the callback
         """
         return self.service.add_on_attribute_changed(
-            self.rules, convert_path_to_se_path(self.path), attribute, self, cb
+            self.rules, convert_path_to_se_path(self.path), attribute, cb
         )
 
     def add_on_command_attribute_changed(
-        self, command: str, attribute: str, cb: Callable
+        self, command: str, attribute: str, cb: Callable[[ValueT], None]
     ) -> EventSubscription:
         """Register a callback for when an attribute is changed.
 
@@ -1079,7 +1139,7 @@ class PyStateContainer(PyCallableStateObject):
             command name
         attribute : str
             attribute name
-        cb : Callable
+        cb : Callable[[ValueT], None]
             Callback function
 
         Returns
@@ -1088,7 +1148,7 @@ class PyStateContainer(PyCallableStateObject):
             EventSubscription instance which can be used to unregister the callback
         """
         return self.service.add_on_command_attribute_changed(
-            self.rules, convert_path_to_se_path(self.path), command, attribute, self, cb
+            self.rules, convert_path_to_se_path(self.path), command, attribute, cb
         )
 
     def __dir__(self):
@@ -1235,14 +1295,16 @@ class PyMenu(PyStateContainer):
             self.rules, convert_path_to_se_path(self.path), command
         )
 
-    def add_on_child_created(self, child_type: str, cb: Callable) -> EventSubscription:
+    def add_on_child_created(
+        self, child_type: str, cb: Callable[[PyMenuT], None]
+    ) -> EventSubscription:
         """Register a callback for when a child object is created.
 
         Parameters
         ----------
         child_type : str
             Type of the child object
-        cb : Callable
+        cb : Callable[[PyMenuT], None]
             Callback function
 
         Returns
@@ -1250,16 +1312,23 @@ class PyMenu(PyStateContainer):
         EventSubscription
             EventSubscription instance which can be used to unregister the callback
         """
+
+        def cb_service(child_path: str):
+            child_path = convert_se_path_to_path(child_path)
+            child_type, child_name = child_path[-1]
+            child = getattr(self, child_type)[child_name]
+            cb(child)
+
         return self.service.add_on_child_created(
-            self.rules, convert_path_to_se_path(self.path), child_type, self, cb
+            self.rules, convert_path_to_se_path(self.path), child_type, cb_service
         )
 
-    def add_on_deleted(self, cb: Callable) -> EventSubscription:
+    def add_on_deleted(self, cb: Callable[[], None]) -> EventSubscription:
         """Register a callback for when the object is deleted.
 
         Parameters
         ----------
-        cb : Callable
+        cb : Callable[[], None]
             Callback function
 
         Returns
@@ -1268,15 +1337,15 @@ class PyMenu(PyStateContainer):
             EventSubscription instance which can be used to unregister the callback
         """
         return self.service.add_on_deleted(
-            self.rules, convert_path_to_se_path(self.path), self, cb
+            self.rules, convert_path_to_se_path(self.path), cb
         )
 
-    def add_on_changed(self, cb: Callable) -> EventSubscription:
+    def add_on_changed(self, cb: Callable[[PyMenuT], None]) -> EventSubscription:
         """Register a callback for when the object is modified.
 
         Parameters
         ----------
-        cb : Callable
+        cb : Callable[[PyMenuT], None]
             Callback function
 
         Returns
@@ -1284,16 +1353,20 @@ class PyMenu(PyStateContainer):
         EventSubscription
             EventSubscription instance which can be used to unregister the callback
         """
+
+        def cb_service(value: ValueT):
+            cb(self)
+
         return self.service.add_on_changed(
-            self.rules, convert_path_to_se_path(self.path), self, cb
+            self.rules, convert_path_to_se_path(self.path), cb_service
         )
 
-    def add_on_affected(self, cb: Callable) -> EventSubscription:
+    def add_on_affected(self, cb: Callable[[PyMenuT], None]) -> EventSubscription:
         """Register a callback for when the object is affected.
 
         Parameters
         ----------
-        cb : Callable
+        cb : Callable[[PyMenuT], None]
             Callback function
 
         Returns
@@ -1301,12 +1374,16 @@ class PyMenu(PyStateContainer):
         EventSubscription
             EventSubscription instance which can be used to unregister the callback
         """
+
+        def cb_service():
+            cb(self)
+
         return self.service.add_on_affected(
-            self.rules, convert_path_to_se_path(self.path), self, cb
+            self.rules, convert_path_to_se_path(self.path), cb_service
         )
 
     def add_on_affected_at_type_path(
-        self, child_type: str, cb: Callable
+        self, child_type: str, cb: Callable[[PyMenuT], None]
     ) -> EventSubscription:
         """Register a callback for when the object is affected at child type.
 
@@ -1314,7 +1391,7 @@ class PyMenu(PyStateContainer):
         ----------
         child_type : str
             child type
-        cb : Callable
+        cb : Callable[[PyMenuT], None]
             Callback function
 
         Returns
@@ -1322,18 +1399,24 @@ class PyMenu(PyStateContainer):
         EventSubscription
             EventSubscription instance which can be used to unregister the callback
         """
+
+        def cb_service():
+            cb(self)
+
         return self.service.add_on_affected_at_type_path(
-            self.rules, convert_path_to_se_path(self.path), child_type, self, cb
+            self.rules, convert_path_to_se_path(self.path), child_type, cb_service
         )
 
-    def add_on_command_executed(self, command: str, cb: Callable) -> EventSubscription:
+    def add_on_command_executed_old(
+        self, command: str, cb: Callable[[PyMenuT, str, ValueT], None]
+    ) -> EventSubscription:
         """Register a callback for when a command is executed.
 
         Parameters
         ----------
         command : str
-            command name
-        cb : Callable
+            Command name
+        cb : Callable[[PyMenuT, str, ValueT], None]
             Callback function
 
         Returns
@@ -1341,8 +1424,35 @@ class PyMenu(PyStateContainer):
         EventSubscription
             EventSubscription instance which can be used to unregister the callback
         """
+
+        def cb_service(command: str, args: ValueT):
+            cb(self, command, args)
+
+        return self.service.add_on_command_executed_old(
+            self.rules, convert_path_to_se_path(self.path), command, self, cb_service
+        )
+
+    def add_on_command_executed(
+        self, cb: Callable[[PyMenuT, str, ValueT], None]
+    ) -> EventSubscription:
+        """Register a callback for when a command is executed.
+
+        Parameters
+        ----------
+        cb : Callable[[PyMenuT, str, ValueT], None]
+            Callback function
+
+        Returns
+        -------
+        EventSubscription
+            EventSubscription instance which can be used to unregister the callback
+        """
+
+        def cb_service(command: str, args: ValueT):
+            cb(self, command, args)
+
         return self.service.add_on_command_executed(
-            self.rules, convert_path_to_se_path(self.path), command, self, cb
+            self.rules, convert_path_to_se_path(self.path), cb_service
         )
 
 
@@ -1356,12 +1466,12 @@ class PyParameter(PyStateContainer):
         """Get default value of the parameter."""
         return self.get_attr(Attribute.DEFAULT.value)
 
-    def add_on_changed(self, cb: Callable) -> EventSubscription:
+    def add_on_changed(self, cb: Callable[[PyMenuT], None]) -> EventSubscription:
         """Register a callback for when the object is modified.
 
         Parameters
         ----------
-        cb : Callable
+        cb : Callable[[PyMenuT], None]
             Callback function
 
         Returns
@@ -1369,8 +1479,12 @@ class PyParameter(PyStateContainer):
         EventSubscription
             EventSubscription instance which can be used to unregister the callback
         """
+
+        def cb_service(value: ValueT):
+            cb(self)
+
         return self.service.add_on_changed(
-            self.rules, convert_path_to_se_path(self.path), self, cb
+            self.rules, convert_path_to_se_path(self.path), cb_service
         )
 
 
@@ -1577,7 +1691,7 @@ class PyNamedObjectContainer:
             # On-deleted subscription objects are unsubscribed after the datamodel
             # object is deleted.
             self[key].add_on_deleted(
-                lambda _: self.service.subscriptions.unsubscribe_while_deleting(
+                lambda: self.service.subscriptions.unsubscribe_while_deleting(
                     self.rules, se_path, "after"
                 )
             )

ansys/fluent/core/services/datamodel_tui.py

@@ -125,10 +125,12 @@ class DatamodelService:
         channel: grpc.Channel,
         metadata: list[tuple[str, str]],
         fluent_error_state,
+        app_utilities,
         scheme_eval,
     ) -> None:
         """__init__ method of DatamodelService class."""
         self._impl = DatamodelServiceImpl(channel, metadata, fluent_error_state)
+        self._app_utilities = app_utilities
         self._scheme_eval = scheme_eval
 
     def get_attribute_value(
@@ -170,7 +172,8 @@ class DatamodelService:
         request = DataModelProtoModule.GetStaticInfoRequest()
         request.path = path
         response = self._impl.get_static_info(request)
-        return MessageToDict(response.info, including_default_value_fields=True)
+        # Note: MessageToDict's parameter names are different in different protobuf versions
+        return MessageToDict(response.info, True)
 
 
 class PyMenu:
@@ -233,7 +236,7 @@ class PyMenu:
             Query result (any Python datatype)
         """
         with ApiUpgradeAdvisor(
-            self._service._scheme_eval,
+            self._service._app_utilities,
             self._version,
             self._mode,
         ):

ansys/fluent/core/services/field_data.py

@@ -153,6 +153,7 @@ class FieldInfo:
                 "display_name": field_info.displayName,
                 "section": field_info.section,
                 "domain": field_info.domain,
+                "quantity_name": field_info.quantity_name,
             }
             for field_info in response.fieldInfo
         }

ansys/fluent/core/services/reduction.py

@@ -263,6 +263,8 @@ class Reduction:
             raise ValueError(f"Invalid location input: '{loc}'")
 
     def _get_location_string(self, locations, ctxt) -> List[str]:
+        if locations == []:
+            return []
         for loc in locations:
             if isinstance(loc, str):
                 self._validate_str_location(loc)

ansys/fluent/core/services/settings.py

@@ -136,9 +136,12 @@ def _get_request_instance_for_path(request_class, path: str) -> Any:
 class SettingsService:
     """Service for accessing and modifying Fluent settings."""
 
-    def __init__(self, channel, metadata, scheme_eval, fluent_error_state) -> None:
+    def __init__(
+        self, channel, metadata, app_utilities, scheme_eval, fluent_error_state
+    ) -> None:
         """__init__ method of SettingsService class."""
         self._service_impl = _SettingsServiceImpl(channel, metadata, fluent_error_state)
+        self._app_utilities = app_utilities
         self._scheme_eval = scheme_eval
 
     @_trace
@@ -369,7 +372,7 @@ class SettingsService:
         """Checks whether a name has a wildcard pattern."""
         return self._scheme_eval.is_defined(
             "has-fnmatch-wild-card?"
-        ) and self._scheme_eval.scheme_eval(f'(has-fnmatch-wild-card? "{name}")')
+        ) and self._app_utilities.is_wildcard(name)
 
     @_trace
     def is_interactive_mode(self) -> bool: