Skip to content

Commit

Permalink
SVAR data access (#1560)
Browse files Browse the repository at this point in the history
SVAR access.

Co-authored-by: Prithwish Mukherjee <[email protected]>
  • Loading branch information
ajain-work and prmukherj authored Apr 26, 2023
1 parent 5da5465 commit 64d3e38
Show file tree
Hide file tree
Showing 3 changed files with 624 additions and 1 deletion.
13 changes: 12 additions & 1 deletion src/ansys/fluent/core/fluent_connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
from ansys.fluent.core.services.reduction import Reduction, ReductionService
from ansys.fluent.core.services.scheme_eval import SchemeEval, SchemeEvalService
from ansys.fluent.core.services.settings import SettingsService
from ansys.fluent.core.services.svar import SVARData, SVARInfo, SVARService
from ansys.fluent.core.streaming_services.datamodel_event_streaming import (
DatamodelEvents,
)
Expand Down Expand Up @@ -237,11 +238,13 @@ def __init__(

self._field_data_service = FieldDataService(self._channel, self._metadata)
self.field_info = FieldInfo(self._field_data_service)

self.field_data = FieldData(
self._field_data_service, self.field_info, _IsDataValid(self.scheme_eval)
)

self._svar_service = SVARService(self._channel, self._metadata)
self.svar_info = SVARInfo(self._svar_service)

self.journal = Journal(self.scheme_eval)

self._cleanup_on_exit = cleanup_on_exit
Expand All @@ -266,6 +269,14 @@ def __init__(
)
FluentConnection._monitor_thread.cbs.append(self._finalizer)

@property
def svar_data(self) -> SVARData:
"""Return the SVARData handle."""
try:
return SVARData(self._svar_service, self.svar_info)
except RuntimeError:
return None

@property
def id(self) -> str:
"""Return the session id."""
Expand Down
6 changes: 6 additions & 0 deletions src/ansys/fluent/core/services/field_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -664,6 +664,12 @@ class _FieldDataConstants:
FieldDataProtoModule.FieldType.FLOAT_ARRAY: np.float32,
FieldDataProtoModule.FieldType.DOUBLE_ARRAY: np.float64,
}
np_data_type_to_proto_field_type = {
np.int32: FieldDataProtoModule.FieldType.INT_ARRAY,
np.int64: FieldDataProtoModule.FieldType.LONG_ARRAY,
np.float32: FieldDataProtoModule.FieldType.FLOAT_ARRAY,
np.float64: FieldDataProtoModule.FieldType.DOUBLE_ARRAY,
}
chunk_size = 256 * 1024
bytes_stream = True
payloadTags = {
Expand Down
Loading

0 comments on commit 64d3e38

Please sign in to comment.