moldflow-api/src/moldflow/data_transform.py at 8f6b3fd4bb37e46bba464ad991d998ea73dbda86 · Autodesk/moldflow-api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
# SPDX-FileCopyrightText: 2025 Autodesk, Inc.
# SPDX-License-Identifier: Apache-2.0

"""
Usage:
    DataTransform Class API Wrapper
"""

from .logger import process_log, LogMessage
from .helper import check_type, check_and_coerce_optional, get_enum_value
from .com_proxy import safe_com
from .common import TransformFunctions, TransformOperations, TransformScalarOperations
from .integer_array import IntegerArray
from .double_array import DoubleArray


class DataTransform:
    """
    Wrapper for DataTransform class of Moldflow Synergy.
    """

    def __init__(self, _data_transform):
        """
        Initialize the DataTransform with a DataTransform instance from COM.

        Args:
            _data_transform: The DataTransform instance.
        """
        process_log(__name__, LogMessage.CLASS_INIT, locals(), name="DataTransform")
        self.data_transform = safe_com(_data_transform)

    # pylint: disable=R0913, R0917
    def func(
        self,
        func_name: TransformFunctions | str,
        label_in: IntegerArray | None,
        data_in: DoubleArray | None,
        label_out: IntegerArray | None,
        data_out: DoubleArray | None,
    ) -> bool:
        """
        This function calculates data using given function.

        Args:
            func_name (TransformFunctions | str): The name of the function to be applied.
            label_in (IntegerArray): The input label array.
            data_in (DoubleArray): The input data array.
            label_out (IntegerArray): The output label array.
            data_out (DoubleArray): The output data array.

        Returns:
            bool: True if the function was applied successfully, False otherwise.
        """
        process_log(__name__, LogMessage.FUNCTION_CALL, locals(), name="func")
        func_name = get_enum_value(func_name, TransformFunctions)

        return self.data_transform.Func(
            func_name,
            check_and_coerce_optional(label_in, IntegerArray),
            check_and_coerce_optional(data_in, DoubleArray),
            check_and_coerce_optional(label_out, IntegerArray),
            check_and_coerce_optional(data_out, DoubleArray),
        )

    # pylint: disable=R0913, R0917
    def op(
        self,
        label_1: IntegerArray | None,
        data_1: DoubleArray | None,
        op: TransformOperations | str,
        label_2: IntegerArray | None,
        data_2: DoubleArray | None,
        label_out: IntegerArray | None,
        data_out: DoubleArray | None,
    ) -> bool:
        """
        This function calculates data using given operation.

        Args:
            label_1 (IntegerArray): The first input label array.
            data_1 (DoubleArray): The first input data array.
            op (TransformOperations | str): The operation to be applied.
            label_2 (IntegerArray): The second input label array.
            data_2 (DoubleArray): The second input data array.
            label_out (IntegerArray): The output label array.
            data_out (DoubleArray): The output data array.

        Returns:
            bool: True if the operation was applied successfully, False otherwise.
        """
        process_log(__name__, LogMessage.FUNCTION_CALL, locals(), name="op")

        op = get_enum_value(op, TransformOperations)

        return self.data_transform.Op(
            check_and_coerce_optional(label_1, IntegerArray),
            check_and_coerce_optional(data_1, DoubleArray),
            op,
            check_and_coerce_optional(label_2, IntegerArray),
            check_and_coerce_optional(data_2, DoubleArray),
            check_and_coerce_optional(label_out, IntegerArray),
            check_and_coerce_optional(data_out, DoubleArray),
        )

    # pylint: disable=R0913, R0917
    def scalar(
        self,
        label_in: IntegerArray | None,
        data_in: DoubleArray | None,
        op: TransformScalarOperations | str,
        scalar_value: float,
        label_out: IntegerArray | None,
        data_out: DoubleArray | None,
    ) -> bool:
        """
        This function calculates data using given scalar operation.

        Args:
            label_in (IntegerArray): The input label array.
            data_in (DoubleArray): The input data array.
            op (TransformScalarOperations | str): The scalar operation to be applied.
            scalar_value (float): The scalar value to be used in the operation.
            label_out (IntegerArray): The output label array.
            data_out (DoubleArray): The output data array.

        Returns:
            bool: True if the scalar operation was applied successfully, False otherwise.
        """
        process_log(__name__, LogMessage.FUNCTION_CALL, locals(), name="scalar")

        op = get_enum_value(op, TransformScalarOperations)
        check_type(scalar_value, (float, int))

        return self.data_transform.Scalar(
            check_and_coerce_optional(label_in, IntegerArray),
            check_and_coerce_optional(data_in, DoubleArray),
            op,
            scalar_value,
            check_and_coerce_optional(label_out, IntegerArray),
            check_and_coerce_optional(data_out, DoubleArray),
        )