Skip to content

Commit b819e7f

Browse files
committed
feat: use Enum classes for constants
1 parent 6fba567 commit b819e7f

File tree

4 files changed

+11
-5
lines changed

4 files changed

+11
-5
lines changed

src/ansys/additive/machine.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,13 @@
11
# (c) 2023 ANSYS, Inc. Unauthorized use, distribution, or duplication is prohibited.
2+
from enum import Enum
23
import math
34

45
from ansys.api.additive.v0.additive_domain_pb2 import MachineSettings as MachineMessage
56

67
import ansys.additive.conversions as conversions
78

89

9-
class MachineConstants:
10+
class MachineConstants(Enum):
1011
#: Default laser power in Watts.
1112
DEFAULT_LASER_POWER = 195
1213
MIN_LASER_POWER = 50

src/ansys/additive/parametric_study/constants.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,8 @@
11
# (c) 2023 ANSYS, Inc. Unauthorized use, distribution, or duplication is prohibited.
2-
class ColumnNames:
2+
from enum import Enum
3+
4+
5+
class ColumnNames(Enum):
36
"""Column names for the parametric study data frame.
47
58
Values are stored internally as a :class:`Pandas DataFrame

src/ansys/additive/simulation.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# (c) 2023 ANSYS, Inc. Unauthorized use, distribution, or duplication is prohibited.
2+
from enum import Enum
23
from typing import Union
34

45
from ansys.additive.microstructure import MicrostructureInput
@@ -7,7 +8,7 @@
78
from ansys.additive.thermal_history import ThermalHistoryInput
89

910

10-
class SimulationType:
11+
class SimulationType(Enum):
1112
"""Simulation types."""
1213

1314
#: Single bead simulation.
@@ -18,7 +19,7 @@ class SimulationType:
1819
MICROSTRUCTURE = "Microstructure"
1920

2021

21-
class SimulationStatus:
22+
class SimulationStatus(Enum):
2223
"""Simulation status values."""
2324

2425
#: Simulation is awaiting execution.

src/ansys/additive/single_bead.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# (c) 2023 ANSYS, Inc. Unauthorized use, distribution, or duplication is prohibited.
2+
from enum import Enum
23
from ansys.api.additive.v0.additive_domain_pb2 import MeltPool as MeltPoolMessage
34
from ansys.api.additive.v0.additive_domain_pb2 import SingleBeadInput as SingleBeadInputMessage
45
from ansys.api.additive.v0.additive_simulation_pb2 import SimulationRequest
@@ -101,7 +102,7 @@ def _to_simulation_request(self) -> SimulationRequest:
101102
return SimulationRequest(id=self.id, single_bead_input=input)
102103

103104

104-
class MeltPoolColumnNames:
105+
class MeltPoolColumnNames(Enum):
105106
"""Column names for melt pool data frame."""
106107

107108
#: Width of melt pool (m).

0 commit comments

Comments
 (0)