Add basic auto-flagging (#134)

Co-authored-by: Daniel Morcuende <[email protected]>
juanep97 · Sep 18, 2024 · c060037 · c060037
1 parent 48a2669
commit c060037
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/iop4lib/iop4.py b/iop4lib/iop4.py
@@ -79,6 +79,13 @@ def process_epochs(epochname_list: Iterable[str], args):
         epoch.compute_relative_photometry()
         epoch.compute_relative_polarimetry()
 
+    logger.info("Auto-flagging points.")
+
+    for result in PhotoPolResult.objects.filter(epoch__in=epoch_L).all():
+        if result.p is not None and not (0 <= result.p <= 1):
+            result.flags.add(PhotoPolResult.FLAGS.ERROR_POLARIMETRY)
+            result.save()
+
     logger.info("Applying corrections.")
 
     for epoch in epoch_L: