Img_processing: Add optional travel status

Author: TziporaGoldshmidt

img_processing/data.json

@@ -0,0 +1,12 @@
+[
+    {
+        "path": "C:/Users/user1/Desktop/day/sun/sun_in_street.mov",
+        "focal_length": 1500,
+        "is_travel": true
+    },
+    {
+        "path": "C:/Users/user1/Desktop/day/close_cars/road.mov",
+        "focal_length": 2000,
+        "is_travel": true
+    }
+]

img_processing/include/manager.h

@@ -10,6 +10,7 @@
 #include "log_manager.h"
 #include "velocity.h"
 #include "communication.h"
+
 class Manager {
    public:
     static logger imgLogger;
@@ -45,7 +46,7 @@ class Manager {
     bool isResetTracker(bool isTravel);
     bool isTrack(bool isTravel);
     void sendAlerts(std::vector<std::vector<uint8_t>> &alerts);
-    void runOnVideo(std::string videoPath);
+    void runOnVideo(std::string videoPath, bool isTravel);
     int readIdFromJson(const char *target);
 };
 #endif  //__MANAGER_H__

img_processing/src/manager.cpp

@@ -1,9 +1,13 @@
+#include <iostream>
+#include <fstream>
+#include "nlohmann/json.hpp"
 #include "manager.h"
 #include "alert.h"
 
 #define ESC 27
 #define NUM_OF_TRACKING 10
 
+using json=nlohmann::json;
 using namespace std;
 using namespace cv;
 
@@ -44,40 +48,50 @@ void Manager::init()
 
 void Manager::mainDemo()
 {
-    string filePath = "../data.txt";
-    // open the file
-    ifstream file(filePath);
+    string path;
+    int focalLength;
+    bool isTravel;
+    // Open the JSON file
+    std::ifstream file("../data.json");
+
     if (!file.is_open()) {
-        LogManager::logErrorMessage(ErrorType::FILE_ERROR);
+        LogManager::logErrorMessage(ErrorType::FILE_ERROR,
+                                    "Failed to open the file");
+        throw runtime_error("Failed to open the file");
         return;
     }
-    string line;
-    // run over the file and read the lines
-    while (getline(file, line)) {
-        // intialize the iteration cnt
-        iterationCnt = 1;
-        istringstream iss(line);
-        string videoPath;
-        double focalLength;
-        // read the parameters
-        if (getline(iss, videoPath, '|') && iss >> focalLength) {
-            // Trim leading and trailing whitespaces from videoPath
-            videoPath.erase(0, videoPath.find_first_not_of(" \t\n\r\f\v"));
-            videoPath.erase(videoPath.find_last_not_of(" \t\n\r\f\v") + 1);
-        }
-        else {
-            LogManager::logErrorMessage(ErrorType::VIDEO_ERROR);
-            return;
+
+    // Read the content of the file into a JSON object
+    json jsonData;
+    file >> jsonData;
+
+    // Check if the JSON data is an array
+    if (jsonData.is_array()) {
+        // Iterate over each object in the array
+        for (const auto &obj : jsonData) {
+            iterationCnt = 1;
+            if (obj.find("path") != obj.end() && obj["path"].is_string()) {
+                path = obj["path"];
+            }
+
+            if (obj.find("focal_length") != obj.end() &&
+                obj["focal_length"].is_number_integer()) {
+                focalLength = obj["focal_length"];
+            }
+
+            if (obj.find("is_travel") != obj.end() &&
+                obj["is_travel"].is_boolean()) {
+                isTravel = obj["is_travel"];
+            }
+            // Get the distance instance and set the focal length
+            Distance &distance = Distance::getInstance();
+            distance.setFocalLength(focalLength);
+            runOnVideo(path, isTravel);
         }
-        // intialize focal length
-        Distance &distance = Distance::getInstance();
-        distance.setFocalLength(focalLength);
-        runOnVideo(videoPath);
     }
-    cout << "finish reading data";
 }
 
-void Manager::runOnVideo(string videoPath)
+void Manager::runOnVideo(string videoPath, bool isTravel)
 {
     // Convert Windows file path to WSL file path format
     if (videoPath.length() >= 3 && videoPath[1] == ':') {
@@ -95,13 +109,14 @@ void Manager::runOnVideo(string videoPath)
         throw runtime_error("video not found");
         return;
     }
+    //run on video
     while (1) {
         capture >> frame;
         if (frame.empty()) {
             LogManager::logInfoMessage(InfoType::MEDIA_FINISH);
             break;
         }
-        int result = processing(frame, true);
+        int result = processing(frame, isTravel);
         if (result == -1)
             return;
     }
@@ -161,19 +176,19 @@ int Manager::processing(const Mat &newFrame, bool isTravel)
         iterationCnt = iterationCnt % NUM_OF_TRACKING + 1;
     }
 
-    #ifdef LANE_DETECT
+#ifdef LANE_DETECT
     laneDetector.manageLaneDetector(this->currentFrame);
-    #endif
-    // visual
-    #ifdef SHOW_FRAMES
-        drawOutput();
-        imshow("currentFrame", *currentFrame);
-        int key = cv::waitKey(1);
-        if (key == ESC) {
-            return -1;
-        }
-        return 1;
-    #endif
+#endif
+// visual
+#ifdef SHOW_FRAMES
+    drawOutput();
+    imshow("currentFrame", *currentFrame);
+    int key = cv::waitKey(1);
+    if (key == ESC) {
+        return -1;
+    }
+    return 1;
+#endif
 }
 
 void Manager::drawOutput()
@@ -244,9 +259,9 @@ void Manager::drawOutput()
     putText(*currentFrame, "velocity", Point(legendX + 15, legendY + 50),
             FONT_HERSHEY_SIMPLEX, 0.6, Scalar(255, 255, 255), 1);
 
-    #ifdef LANE_DETECT
+#ifdef LANE_DETECT
     laneDetector.drawLanesOnImage(currentFrame);
-    #endif  
+#endif
 }
 
 void Manager::sendAlerts(vector<vector<uint8_t>> &alerts)