|
2 | 2 |
|
3 | 3 | import warnings |
4 | 4 | from dataclasses import InitVar, dataclass, field |
5 | | -from enum import auto |
| 5 | +from enum import Enum |
6 | 6 | from pathlib import Path |
7 | 7 | from typing import TYPE_CHECKING, Any, TypeVar |
8 | 8 |
|
9 | 9 | import numpy as np |
10 | | -from strenum import LowercaseStrEnum # TODO: EOL 3.10: replace with native StrEnum |
11 | 10 |
|
12 | 11 | from eitprocessing.datahandling import DataContainer |
13 | 12 | from eitprocessing.datahandling.continuousdata import ContinuousData |
@@ -173,12 +172,28 @@ def calculate_global_impedance(self) -> np.ndarray: |
173 | 172 | return np.nansum(self.pixel_impedance, axis=(1, 2)) |
174 | 173 |
|
175 | 174 |
|
176 | | -class Vendor(LowercaseStrEnum): |
177 | | - """Enum indicating the vendor (manufacturer) of the source EIT device.""" |
| 175 | +class Vendor(Enum): |
| 176 | + """Enum indicating the vendor (manufacturer) of the source EIT device. |
| 177 | +
|
| 178 | + The enum values are all lowercase strings. For some manufacturers, multiple ways of wrinting are provided mapping to |
| 179 | + the same value, to prevent confusion over conversion of special characters. The "simulated" vendor is provided to |
| 180 | + indicate simulated data. |
| 181 | + """ |
| 182 | + |
| 183 | + DRAEGER = "draeger" |
| 184 | + """Dräger (PulmoVista V500)""" |
| 185 | + |
| 186 | + TIMPEL = "timpel" |
| 187 | + """Timpel (Enlight 2100)""" |
| 188 | + |
| 189 | + SENTEC = "sentec" |
| 190 | + """Sentec (Lumon)""" |
178 | 191 |
|
179 | | - DRAEGER = auto() |
180 | | - TIMPEL = auto() |
181 | | - SENTEC = auto() |
182 | 192 | DRAGER = DRAEGER |
183 | | - DRÄGER = DRAEGER # noqa: PLC2401 |
184 | | - SIMULATED = auto() |
| 193 | + """Synonym of DREAGER""" |
| 194 | + |
| 195 | + DRÄGER = DRAEGER # noqa: PIE796, PLC2401 |
| 196 | + """Synonym of DREAGER""" |
| 197 | + |
| 198 | + SIMULATED = "simulated" |
| 199 | + """Simulated data""" |
0 commit comments