[enums] add mapping for trackname, id and color

This commit is contained in:
Jan Grewe 2025-02-21 16:18:24 +01:00
parent 2e918866e1
commit f09c78adb5

View File

@ -1,7 +1,29 @@
from enum import Enum
from PySide6.QtGui import QColor
class DetectionData(Enum):
ID = 0
FRAME = 1
COORDINATES = 2
TRACK_ID = 3
TRACK_ID = 3
class Tracks(Enum):
TRACKONE = 1
TRACKTWO = 2
UNASSIGNED = -1
def toColor(self):
track_colors = {
Tracks.TRACKONE: QColor.fromString("orange"),
Tracks.TRACKTWO: QColor.fromString("green"),
Tracks.UNASSIGNED: QColor.fromString("red")
}
return track_colors.get(self, QColor(128, 128, 128)) # Default to black if not found
@staticmethod
def fromValue(value):
for track in Tracks:
if track.value == value:
return track
return Tracks.UNASSIGNED