Merge branch 'fix-bam-new-bp-table' into 'master'

get_bam(): allows for varying bunch pattern table length

Closes #80

See merge request !279

doc/changelog.rst

@@ -7,6 +7,7 @@ unreleased
 - **Bug fixes**
 
     - fix :issue:`75` regarding pulse Id assignment when reading BAM data :mr:`272`
+    - fix :issue:`80` regarding new bunch pattern table size when reading BAM data :mr:`279`
 
 - **Improvements**
 

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':