legend-dataflow-scripts 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

{legend_dataflow_scripts-0.2.3.dist-info → legend_dataflow_scripts-0.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: legend-dataflow-scripts
-Version: 0.2.3
+Version: 0.2.4
 Summary: Python package for the processing scripts for LEGEND-200 data
 Author-email: George Marshall <ggmarsh@uw.edu>, Luigi Pertoldi <gipert@pm.me>
 Maintainer: The LEGEND Collaboration

{legend_dataflow_scripts-0.2.3.dist-info → legend_dataflow_scripts-0.2.4.dist-info}/RECORD

@@ -1,5 +1,5 @@
 legenddataflowscripts/__init__.py,sha256=hlpvTxSBjOyXlZUyOyYx3VwT5LS6zNzhAZnTmfT3NjU,303
-legenddataflowscripts/_version.py,sha256=kBRz0P2plw1eVdIpt70W6m1LMbEIhLY3RyOfVGdubaI,704
+legenddataflowscripts/_version.py,sha256=NRw4Jle4n9v_DD2wtplRqflGCvX8OU5eAjycYY0vY3Y,704
 legenddataflowscripts/par/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 legenddataflowscripts/par/geds/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 legenddataflowscripts/par/geds/dsp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -11,10 +11,10 @@ legenddataflowscripts/par/geds/dsp/pz.py,sha256=l9nc3MmSjghmYOR3LBbbGZMnHc52MFy5
 legenddataflowscripts/par/geds/dsp/svm.py,sha256=eDneRB_PQZp8Q4n2VheTX3kbu4ufZQ-jnuCCjvtwFpk,826
 legenddataflowscripts/par/geds/dsp/svm_build.py,sha256=w5-vT6rXmpl7V9rdkfc7_g6GTzn86i41tHkIT-3f5YI,1931
 legenddataflowscripts/par/geds/hit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-legenddataflowscripts/par/geds/hit/aoe.py,sha256=jAH0Rh3JCnV67vhv9xUZtPdGB8ADJXgG9Lo8t8YQjqs,10841
-legenddataflowscripts/par/geds/hit/ecal.py,sha256=j8Z90r9UP5Hn-cawf2lvDL8tF2Uvfp4rDjlM4sGbM-M,27009
-legenddataflowscripts/par/geds/hit/lq.py,sha256=vXgK83RlJJ4UUjQQJWfmFGIbT0AEP3EaLALM5LhvZ6s,11133
-legenddataflowscripts/par/geds/hit/qc.py,sha256=vr6j5sRTvjmDmErW2uUteCPsj5qfBrpn7ssvLKzov4M,12408
+legenddataflowscripts/par/geds/hit/aoe.py,sha256=DEruYu-mFcgUMdR6jTdgoBY2Q3ijvktqcctu_Cexa6s,10973
+legenddataflowscripts/par/geds/hit/ecal.py,sha256=Ki1YGzFYMcQhh2S2VgNs1ApkMjJjGVq0V-AdKpcCV4A,27125
+legenddataflowscripts/par/geds/hit/lq.py,sha256=aY2tRwbTI9U7b5bxwDohQitY-Tuhe1tXdAA5dqeaznc,11264
+legenddataflowscripts/par/geds/hit/qc.py,sha256=j3oWN28zfGrR_WMMZMGWJ46dpegKf38Fq8J3zH815sU,12606
 legenddataflowscripts/tier/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 legenddataflowscripts/tier/dsp.py,sha256=J_ABEqbQurZHJOg8LV2aporNjZQpPg2bDZG6DkmuAL4,8633
 legenddataflowscripts/tier/hit.py,sha256=-LCfsqYAYPAuN2OKbPuh4G2P5B4uA4YmQGcqUtbZxis,4989
@@ -30,8 +30,8 @@ legenddataflowscripts/workflow/execenv.py,sha256=qTG4N9ovEPxA0QtqG0wWUIuK50BZIcY
 legenddataflowscripts/workflow/filedb.py,sha256=6scz8DjdvbXs8OXOy2-6BJWVSALQzmy5cLLn-OsdXsU,3656
 legenddataflowscripts/workflow/pre_compile_catalog.py,sha256=cEK0KXh-ClSE2Bo9MK471o79XG22bMY5r-2tIihtCfk,790
 legenddataflowscripts/workflow/utils.py,sha256=VVCsj7wNaqV6sw2Xnk_xykhVv3BKTX4hqQtKE4UUayg,3170
-legend_dataflow_scripts-0.2.3.dist-info/METADATA,sha256=BhkwgP_I60cx2ZfV0ExKx0uye1LGTWlxfNTUsApT_gw,3085
-legend_dataflow_scripts-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-legend_dataflow_scripts-0.2.3.dist-info/entry_points.txt,sha256=B197waSm-orA_ZS-9rkxNDsmOHdCn8CbWodnlqXQKRg,1313
-legend_dataflow_scripts-0.2.3.dist-info/top_level.txt,sha256=s8E2chjJNYUbrN6whFG_VCsJKySFp1IOXLcUefA7DB0,22
-legend_dataflow_scripts-0.2.3.dist-info/RECORD,,
+legend_dataflow_scripts-0.2.4.dist-info/METADATA,sha256=vQ9uxbFCQUTNbkVqZlpOVWKOnKxnKj8Ew-b8e6Zr4O0,3085
+legend_dataflow_scripts-0.2.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+legend_dataflow_scripts-0.2.4.dist-info/entry_points.txt,sha256=B197waSm-orA_ZS-9rkxNDsmOHdCn8CbWodnlqXQKRg,1313
+legend_dataflow_scripts-0.2.4.dist-info/top_level.txt,sha256=s8E2chjJNYUbrN6whFG_VCsJKySFp1IOXLcUefA7DB0,22
+legend_dataflow_scripts-0.2.4.dist-info/RECORD,,

legenddataflowscripts/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.2.3'
-__version_tuple__ = version_tuple = (0, 2, 3)
+__version__ = version = '0.2.4'
+__version_tuple__ = version_tuple = (0, 2, 4)
 
 __commit_id__ = commit_id = None

legenddataflowscripts/par/geds/hit/aoe.py

@@ -251,14 +251,13 @@ def par_geds_hit_aoe() -> None:
         ]
 
         if "dt_param" in kwarg_dict:
-            params += kwarg_dict["dt_param"]
+            params.append(kwarg_dict["dt_param"])
         else:
             params.append("dt_eff")
 
         if "dt_cut" in kwarg_dict and kwarg_dict["dt_cut"] is not None:
             cal_dict.update(kwarg_dict["dt_cut"]["cut"])
             params.append(kwarg_dict["dt_cut"]["out_param"])
-
         # load data in
         data, threshold_mask = load_data(
             files,
@@ -271,9 +270,12 @@ def par_geds_hit_aoe() -> None:
         msg = f"Loaded {len(data)} events"
         log.info(msg)
 
-        mask = get_pulser_mask(
-            pulser_file=args.pulser_file,
-        )
+        if args.pulser_file is not None:
+            mask = get_pulser_mask(
+                pulser_file=args.pulser_file,
+            )
+        else:
+            mask = np.zeros(len(threshold_mask), dtype=bool)
 
         data["is_pulser"] = mask[threshold_mask]
 

legenddataflowscripts/par/geds/hit/ecal.py

@@ -532,9 +532,12 @@ def par_geds_hit_ecal() -> None:
         cal_energy_param="trapTmax",
     )
 
-    mask = get_pulser_mask(
-        pulser_file=args.pulser_file,
-    )
+    if args.pulser_file is not None:
+        mask = get_pulser_mask(
+            pulser_file=args.pulser_file,
+        )
+    else:
+        mask = np.zeros(len(threshold_mask), dtype=bool)
 
     data["is_pulser"] = mask[threshold_mask]
 

legenddataflowscripts/par/geds/hit/lq.py

@@ -288,9 +288,12 @@ def par_geds_hit_lq() -> None:
         msg = f"Loaded {len(data)} events"
         log.info(msg)
 
-        mask = get_pulser_mask(
-            pulser_file=args.pulser_file,
-        )
+        if args.pulser_file is not None:
+            mask = get_pulser_mask(
+                pulser_file=args.pulser_file,
+            )
+        else:
+            mask = np.zeros(len(threshold_mask), dtype=bool)
 
         data["is_pulser"] = mask[threshold_mask]
 

legenddataflowscripts/par/geds/hit/qc.py

@@ -39,17 +39,18 @@ def build_qc(
 ):
     search_name = table_name if table_name[-1] == "/" else table_name + "/"
 
-    kwarg_dict_fft = config["fft_fields"]
+    kwarg_dict_fft = config.get("fft_fields", None)
     kwarg_dict_cal = config["cal_fields"]
 
     cut_fields = get_keys(
         [key.replace(search_name, "") for key in ls(cal_files[0], search_name)],
         kwarg_dict_cal["cut_parameters"],
     )
-    cut_fields += get_keys(
-        [key.replace(search_name, "") for key in ls(cal_files[0], search_name)],
-        kwarg_dict_fft["cut_parameters"],
-    )
+    if kwarg_dict_fft is not None:
+        cut_fields += get_keys(
+            [key.replace(search_name, "") for key in ls(cal_files[0], search_name)],
+            kwarg_dict_fft["cut_parameters"],
+        )
 
     if "initial_cal_cuts" in config:
         init_cal = config["initial_cal_cuts"]
@@ -147,9 +148,12 @@ def build_qc(
         cal_energy_param="trapTmax",
     )
 
-    mask = get_pulser_mask(
-        pulser_file=pulser_file,
-    )
+    if pulser_file is not None:
+        mask = get_pulser_mask(
+            pulser_file=pulser_file,
+        )
+    else:
+        mask = np.zeros(len(threshold_mask), dtype=bool)
 
     data["is_pulser"] = mask[threshold_mask]
 
@@ -289,7 +293,9 @@ def build_qc(
 def par_geds_hit_qc() -> None:
     argparser = argparse.ArgumentParser()
     argparser.add_argument("--cal-files", help="cal_files", nargs="*", type=str)
-    argparser.add_argument("--fft-files", help="fft_files", nargs="*", type=str)
+    argparser.add_argument(
+        "--fft-files", help="fft_files", nargs="*", type=str, default=[]
+    )
 
     argparser.add_argument(
         "--tcm-filelist", help="tcm_filelist", type=str, required=False