Fix the output directory code.

It wasn't taking into account any directories in the input file name.
IDInteraction · Dec 4, 2015 · 107d466 · 107d466
1 parent 057fc6d
commit 107d466
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/main.cpp b/src/main.cpp
@@ -73,13 +73,19 @@ void draw_bbox_centre(Mat frame, vector<float> bbox) {
 }
 
 string output_filename(string input_filename, string output_dir="") {
-  int dot = input_filename.find_last_of('.');
   int out_len = output_dir.length();
 
-  if(out_len > 0 && !(0 == output_dir.compare((out_len - 1), 1, "/"))) {
-    output_dir += '/';
+  if(out_len > 0) {
+    int slash = input_filename.find_last_of('/');
+    input_filename = input_filename.substr(slash + 1);
+
+    if(!(0 == output_dir.compare((out_len - 1), 1, "/"))) {
+      output_dir += '/';
+    }
   }
 
+  int dot = input_filename.find_last_of('.');
+
   return output_dir + input_filename.substr(0, dot) + "_out.avi";
 }