import ctypes
from ctypes_enum import CtypesEnum
from .CDetectionStructs import *
from .EveControlOption import *
from .EveErrors import *

EVE_MAX_OBJECT_DETECTION_CLASSES = 128

class EveObjectDetectionOptions(ctypes.Structure):
	_fields_ = [
		("enabled", ctypes.c_int),
		("modelPath", ctypes.c_byte * 512),
		("nmsThreshold", ctypes.c_float),
		("iouThreshold", ctypes.c_float),
		("classNames", ctypes.c_byte * EVE_CLASS_ID_NAME_SIZE * EVE_MAX_OBJECT_DETECTION_CLASSES),
		("classCount", ctypes.c_int),
		("error", ctypes.c_int),
	]

class EvePersonDetectionOptions(ctypes.Structure):
	_fields_ = [
		("enabled", ctypes.c_int),
		("error", ctypes.c_int),
	]

class EveDetectionData(ctypes.Structure):
	_fields_ = [
		("data", ctypes.POINTER(CDetectionData)),
		("error", ctypes.c_int),
	]