Prevent exit codes >= 255 in convert.py

Author: m90

convert.py

@@ -27,6 +27,7 @@
 colmap_command = '"{}"'.format(args.colmap_executable) if len(args.colmap_executable) > 0 else "colmap"
 magick_command = '"{}"'.format(args.magick_executable) if len(args.magick_executable) > 0 else "magick"
 use_gpu = 1 if not args.no_gpu else 0
+clamp_exit_code = lambda code: code if code <= 255 else 1
 
 if not args.skip_matching:
     os.makedirs(args.source_path + "/distorted/sparse", exist_ok=True)
@@ -41,7 +42,7 @@
     exit_code = os.system(feat_extracton_cmd)
     if exit_code != 0:
         logging.error(f"Feature extraction failed with code {exit_code}. Exiting.")
-        exit(exit_code)
+        exit(clamp_exit_code(exit_code))
 
     ## Feature matching
     feat_matching_cmd = colmap_command + " exhaustive_matcher \
@@ -50,7 +51,7 @@
     exit_code = os.system(feat_matching_cmd)
     if exit_code != 0:
         logging.error(f"Feature matching failed with code {exit_code}. Exiting.")
-        exit(exit_code)
+        exit(clamp_exit_code(exit_code))
 
     ### Bundle adjustment
     # The default Mapper tolerance is unnecessarily large,
@@ -63,7 +64,7 @@
     exit_code = os.system(mapper_cmd)
     if exit_code != 0:
         logging.error(f"Mapper failed with code {exit_code}. Exiting.")
-        exit(exit_code)
+        exit(clamp_exit_code(exit_code))
 
 ### Image undistortion
 ## We need to undistort our images into ideal pinhole intrinsics.
@@ -75,7 +76,7 @@
 exit_code = os.system(img_undist_cmd)
 if exit_code != 0:
     logging.error(f"Mapper failed with code {exit_code}. Exiting.")
-    exit(exit_code)
+    exit(clamp_exit_code(exit_code))
 
 files = os.listdir(args.source_path + "/sparse")
 os.makedirs(args.source_path + "/sparse/0", exist_ok=True)
@@ -105,20 +106,20 @@
         exit_code = os.system(magick_command + " mogrify -resize 50% " + destination_file)
         if exit_code != 0:
             logging.error(f"50% resize failed with code {exit_code}. Exiting.")
-            exit(exit_code)
+            exit(clamp_exit_code(exit_code))
 
         destination_file = os.path.join(args.source_path, "images_4", file)
         shutil.copy2(source_file, destination_file)
         exit_code = os.system(magick_command + " mogrify -resize 25% " + destination_file)
         if exit_code != 0:
             logging.error(f"25% resize failed with code {exit_code}. Exiting.")
-            exit(exit_code)
+            exit(clamp_exit_code(exit_code))
 
         destination_file = os.path.join(args.source_path, "images_8", file)
         shutil.copy2(source_file, destination_file)
         exit_code = os.system(magick_command + " mogrify -resize 12.5% " + destination_file)
         if exit_code != 0:
             logging.error(f"12.5% resize failed with code {exit_code}. Exiting.")
-            exit(exit_code)
+            exit(clamp_exit_code(exit_code))
 
 print("Done.")