diff --git a/src/toolbox_scs/detectors/bam_detectors.py b/src/toolbox_scs/detectors/bam_detectors.py index fde4b8e09024f40e51eaa92d962cd313cfd2cc73..39f29840109d2b94d455a845bca3ac6995e57d72 100644 --- a/src/toolbox_scs/detectors/bam_detectors.py +++ b/src/toolbox_scs/detectors/bam_detectors.py @@ -59,9 +59,8 @@ def get_bam(run, mnemonics=None, merge_with=None, bunchPattern='sase3', Example ------- >>> import toolbox_scs as tb - >>> run, ds = tb.load(2711, 303, 'BAM1932S') - >>> ds['BAM1932S'] - + >>> run = tb.open_run(2711, 303) + >>> bam = tb.get_bam(run, 'BAM1932S') """ bam_mnemos = ['BAM4', 'BAM1'] m2 = [] @@ -102,10 +101,12 @@ def get_bam(run, mnemonics=None, merge_with=None, bunchPattern='sase3', da_bam = merge_with[m] else: da_bam = get_array(run, m) - da_bam = da_bam.sel(BAMbunchId=slice(0, 5400, 2)) + da_bam = da_bam.sel(BAMbunchId=slice(0, None, 2)) # align the pulse Id if bpt is not None: - da_bam = da_bam.assign_coords(BAMbunchId=np.arange(0, 2700)) + n = mask.sizes[dim_names[bunchPattern]] + da_bam = da_bam.isel(BAMbunchId=slice(0, n)) + da_bam = da_bam.assign_coords(BAMbunchId=np.arange(0, n)) da_bam = da_bam.rename(BAMbunchId=dim_names[bunchPattern]) da_bam = da_bam.where(mask, drop=True) if run_mnemonics[m]['key'] != 'data.lowChargeArrivalTime':