From 9e0b3fd9a9a080e6b98213070fd99038583c1720 Mon Sep 17 00:00:00 2001
From: dan-du-car <dan.du@leidos.com>
Date: Wed, 21 Aug 2024 20:54:50 +0000
Subject: [PATCH] additional class values

---
 .../MUSTSensorDriverPlugin/src/MUSTSensorDetection.h      | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/v2i-hub/MUSTSensorDriverPlugin/src/MUSTSensorDetection.h b/src/v2i-hub/MUSTSensorDriverPlugin/src/MUSTSensorDetection.h
index 313204811..cd2dfcd6f 100644
--- a/src/v2i-hub/MUSTSensorDriverPlugin/src/MUSTSensorDetection.h
+++ b/src/v2i-hub/MUSTSensorDriverPlugin/src/MUSTSensorDetection.h
@@ -16,6 +16,9 @@ namespace MUSTSensorDriverPlugin {
     enum class DetectionClassification {
         SEDAN,
         TRUCK,
+        BUS,
+        PICKUP_TRUCK,
+        PEDESTRIAN,
         VAN,
         NA
     };
@@ -46,7 +49,10 @@ namespace MUSTSensorDriverPlugin {
     const static std::unordered_map<std::string, DetectionClassification> stringToDetectionClassificationMap = {
         {"sedan", DetectionClassification::SEDAN},
         {"truck", DetectionClassification::TRUCK},
-        {"van", DetectionClassification::VAN}
+        {"van", DetectionClassification::VAN},
+        {"bus", DetectionClassification::BUS},
+        {"pickup truck", DetectionClassification::PICKUP_TRUCK},
+        {"pedestrian", DetectionClassification::PEDESTRIAN}
     };
 
     /**