sf-daq/dap
0
1
Fork 2
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Bug fixes in new streak finder

Browse Source
This commit is contained in:
Dorofeeva Elizaveta (EXT)
2025-07-04 08:43:28 +02:00
parent 6e695a6734
commit 2852a5320a
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
dap/algos/streakfind.py
View File

@@ -14,6 +14,10 @@ from streak_finder.label import Structure2D
DEFAULT_NUM_THREADS = 16
def calc_cbd_analysis(results, data, pf_pixel_mask):
do_snr = results.get("do_snr", False)
if not do_snr:
return data
try:
cryst_data = _generate_cryst_data(results, data, pf_pixel_mask)
except Exception as error: # Broad exception - we don't want to break anything here
@@ -26,13 +30,10 @@ def calc_cbd_analysis(results, data, pf_pixel_mask):
except Exception as error: # Broad exception - we don't want to break anything here
print(f"Error processing CBD data:\n{error}")
results["cbd_error"] = f"Error processing CBD data:\n{error}"
return cryst_data.snr
print(f"Returning data shape {cryst_data.snr[0].shape}")
return cryst_data.snr[0]
def _generate_cryst_data(results, data, pf_pixel_mask) -> CrystData:
do_snr = results.get("do_snr", False)
if not do_snr:
return
params_required = [
"whitefield_data_file",
"mask_data_file",
@@ -73,7 +74,8 @@ def _generate_cryst_data(results, data, pf_pixel_mask) -> CrystData:
)
if scale_whitefield:
data = data.scale_whitefield(method='median', num_threads=num_threads)
data = data.update_snr()
return data
def _calc_streakfinder_analysis(results, cryst_data: CrystData):