From 08e4d7cebca16077353106cda365d28db7377db8 Mon Sep 17 00:00:00 2001
From: sniedbalski <77023264+sniedbalski@users.noreply.github.com>
Date: Tue, 23 Apr 2024 12:10:28 -0400
Subject: [PATCH] Update socket_grex.py

Adding capability to set min_dm, max_ibox, min_snr, and min_snr_t2out for filter_candidates rather than having them hardcoded in.
---
 grex_t2/socket_grex.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/grex_t2/socket_grex.py b/grex_t2/socket_grex.py
index dc4c618..5dc3041 100644
--- a/grex_t2/socket_grex.py
+++ b/grex_t2/socket_grex.py
@@ -12,7 +12,7 @@
 
 nbeams_queue = deque(maxlen=10)
 
-def filter_candidates(candsfile, output=True, trigger=True, last_trigger_time=0.0):
+def filter_candidates(candsfile, output=True, trigger=True, last_trigger_time=0.0, min_dm=50, max_ibox=64, min_snr=12.5, min_snr_t2out=15.0):
     """Take a single gulp of candidates,
     parse, cluster, and then filter to
     produce highest S/N candidate and save
@@ -21,10 +21,6 @@ def filter_candidates(candsfile, output=True, trigger=True, last_trigger_time=0.
     outroot = "/hdd/data/candidates/T2/"
 
     col_heimdall = ["snr", "if", "itime", "mjds", "ibox", "idm", "dm", "ibeam"]
-    min_dm = 50
-    max_ibox = 64
-    min_snr = 12.5
-    min_snr_t2out = 15.0
     max_ncl = np.inf
     max_cntb = np.inf
     max_cntb0 = np.inf