From d3268dc3460383e060587b8c78dbb3c9976ba2c1 Mon Sep 17 00:00:00 2001
From: tlatorre <tlatorre@uchicago.edu>
Date: Mon, 4 Mar 2019 16:12:11 -0600
Subject: add a --min-nhit command line argument

---
 src/fit.c | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/fit.c b/src/fit.c
index 1338ddd..177204c 100644
--- a/src/fit.c
+++ b/src/fit.c
@@ -5766,6 +5766,7 @@ void usage(void)
     fprintf(stderr,"  --max-time           maximum time in seconds per fit (default: 3600)\n");
     fprintf(stderr,"  --skip-second-event  only fit the first event after a MAST bank\n");
     fprintf(stderr,"  --max-particles      maximum number of particles to fit for (default: 1)\n");
+    fprintf(stderr,"  --min-nhit           minimum nhit to fit an event (default: 100)\n");
     fprintf(stderr,"  -h      display this help message\n");
     exit(1);
 }
@@ -5915,6 +5916,7 @@ int main(int argc, char **argv)
     size_t combos[100];
     size_t len;
     char tmp[256];
+    size_t nhit, min_nhit = 100;
 
     for (i = 1; i < argc; i++) {
         if (strlen(argv[i]) >= 2 && !strncmp(argv[i], "--", 2)) {
@@ -5927,6 +5929,9 @@ int main(int argc, char **argv)
             } else if (!strcmp(argv[i]+2,"max-particles")) {
                 max_particles = atoi(argv[++i]);
                 continue;
+            } else if (!strcmp(argv[i]+2,"min-nhit")) {
+                min_nhit = atoi(argv[++i]);
+                continue;
             }
         } else if (argv[i][0] == '-') {
             switch (argv[i][1]) {
@@ -6166,9 +6171,13 @@ skip_mc:
 
             rv = get_event(f,&ev,&b);
 
+            nhit = get_nhit(&ev);
+
+            if (nhit < min_nhit) break;
+
             if (fout) {
                 fprintf(fout, "      - gtid: %i\n", ev.gtid);
-                fprintf(fout, "        nhit: %zu\n", get_nhit(&ev));
+                fprintf(fout, "        nhit: %zu\n", nhit);
                 fprintf(fout, "        fit:\n");
             }
 
-- 
cgit