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

readability

Browse Source
This commit is contained in:
augustin_s
2024-07-31 17:50:40 +02:00
parent 1c129bc5be
commit 851a186538
1 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
dap/algos/radprof.py
View File

@ -6,13 +6,18 @@ def calc_radial_integration(results, data, keep_pixels, pixel_mask_pf, center_ra
if keep_pixels is None and pixel_mask_pf is not None: if keep_pixels is None and pixel_mask_pf is not None:
keep_pixels = (pixel_mask_pf != 0) keep_pixels = (pixel_mask_pf != 0)
if center_radial_integration is None: if center_radial_integration is None:
center_radial_integration = [results["beam_center_x"], results["beam_center_y"]] center_radial_integration = [
results["beam_center_x"],
results["beam_center_y"]
]
r_radial_integration = None r_radial_integration = None
if r_radial_integration is None: if r_radial_integration is None:
r_radial_integration, nr_radial_integration = prepare_radial_profile(data, center_radial_integration, keep_pixels) r_radial_integration, nr_radial_integration = prepare_radial_profile(data, center_radial_integration, keep_pixels)
r_min_max = [int(np.min(r_radial_integration)), int(np.max(r_radial_integration)) + 1] r_min = int(np.min(r_radial_integration))
r_max = int(np.max(r_radial_integration)) + 1
apply_threshold = results.get("apply_threshold", False) apply_threshold = results.get("apply_threshold", False)
@ -32,16 +37,16 @@ def calc_radial_integration(results, data, keep_pixels, pixel_mask_pf, center_ra
silent_region_min is not None and silent_region_min is not None and
silent_region_max is not None and silent_region_max is not None and
silent_region_max > silent_region_min and silent_region_max > silent_region_min and
silent_region_min > r_min_max[0] and silent_region_min > r_min and
silent_region_max < r_min_max[1] silent_region_max < r_max
): ):
silent_region = rp[silent_region_min:silent_region_max]
integral_silent_region = np.sum(rp[silent_region_min:silent_region_max]) integral_silent_region = np.sum(silent_region)
rp = rp / integral_silent_region rp = rp / integral_silent_region
results["radint_normalised"] = [silent_region_min, silent_region_max] results["radint_normalised"] = [silent_region_min, silent_region_max]
results["radint_I"] = list(rp[r_min_max[0]:]) results["radint_I"] = rp[r_min:].tolist()
results["radint_q"] = r_min_max results["radint_q"] = [r_min, r_max]
return keep_pixels, center_radial_integration, r_radial_integration return keep_pixels, center_radial_integration, r_radial_integration