"""
Services package for RTSP stream processing with GPU acceleration.
"""

from .base_model_controller import BaseModelController, BatchFrame, BufferState
from .inference_engine import (
    BackendType,
    EngineMetadata,
    IInferenceEngine,
    NativeTensorRTEngine,
    UltralyticsEngine,
    create_engine,
)
from .jpeg_encoder import JPEGEncoderFactory, encode_frame_to_jpeg
from .model_repository import (
    ExecutionContext,
    ModelMetadata,
    SharedEngine,
    TensorRTModelRepository,
)
from .modelstorage import FileModelStorage, IModelStorage
from .pt_converter import PTConverter
from .stream_connection_manager import (
    StreamConnection,
    StreamConnectionManager,
    TrackingResult,
)
from .stream_decoder import ConnectionStatus, StreamDecoder, StreamDecoderFactory
from .tensorrt_model_controller import TensorRTModelController
from .tracking_controller import Detection, ObjectTracker, TrackedObject
from .ultralytics_exporter import UltralyticsExporter
from .ultralytics_model_controller import UltralyticsModelController
from .yolo import COCO_CLASSES, YOLOv8Utils

__all__ = [
    "StreamDecoderFactory",
    "StreamDecoder",
    "ConnectionStatus",
    "JPEGEncoderFactory",
    "encode_frame_to_jpeg",
    "TensorRTModelRepository",
    "ModelMetadata",
    "ExecutionContext",
    "SharedEngine",
    "ObjectTracker",
    "TrackedObject",
    "Detection",
    "YOLOv8Utils",
    "COCO_CLASSES",
    "BaseModelController",
    "TensorRTModelController",
    "UltralyticsModelController",
    "BatchFrame",
    "BufferState",
    "StreamConnectionManager",
    "StreamConnection",
    "TrackingResult",
    "PTConverter",
    "IModelStorage",
    "FileModelStorage",
    "IInferenceEngine",
    "NativeTensorRTEngine",
    "UltralyticsEngine",
    "EngineMetadata",
    "BackendType",
    "create_engine",
    "UltralyticsExporter",
]