From cf6d093de9dd8870e3fd5e3004b3f3916eb5155f Mon Sep 17 00:00:00 2001
From: Yohsuke Fukai <ysk@yfukai.net>
Date: Mon, 25 Mar 2024 19:17:07 +0900
Subject: [PATCH] added option to enable sorting

---
 main.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/main.py b/main.py
index 83009ca..db8dc8a 100755
--- a/main.py
+++ b/main.py
@@ -141,6 +141,15 @@ def get_args():
         default="cpu",
         help="Device to use, cpu or gpu [default = 'cpu'].",
     )
+    optional.add_argument(
+        "-s",
+        "--sort_intensity",
+        action="store_true",
+        dest="sort_intensity",
+        required=False,
+        default=False,
+        help="If True, sort the intensity pixelwise (suitable for non-timelapse images).",
+    )
 
     arg = parser.parse_args()
 
@@ -164,6 +173,7 @@ def main(args):
         max_reweight_iterations=args.max_reweight_iterations,
         fitting_mode=args.fitting_mode,
         get_darkfield=args.darkfield,
+        sort_intensity=args.sort_intensity,
     )
 
     # Initialize flatfields and darkfields