Commit 3dc54155 authored by Anika Treffehn's avatar Anika Treffehn
Browse files

Merge branch 'fix_preamble_noise' into 'main'

fixed noise preamble

See merge request !11
parents 275445ab 3f4f54a1
Loading
Loading
Loading
Loading
Loading
+6 −6
Original line number Diff line number Diff line
@@ -85,30 +85,30 @@ def trim(
        pre_trim = limits[0]
        post_trim = limits[1]

    if pre_trim <= 0:
    if pre_trim < 0:
        if pad_noise:
            # pad with uniformly distributed noise between -4 and 4
            np.random.seed(SEED_PADDING)
            noise = np.random.randint(
                low=-4, high=5, size=(np.abs(pre_trim), np.shape(x)[1])
            ).astype("float")
            x = np.concatenate((x, noise), axis=0)
            x = np.concatenate((noise, x), axis=0)
        else:
            x = np.pad(x, [[np.abs(pre_trim), 0], [0, 0]])
    else:
    elif pre_trim > 0:
        x = x[pre_trim:, :]

    if post_trim <= 0:
    if post_trim < 0:
        if pad_noise:
            # pad with uniformly distributed noise between -4 and 4
            np.random.seed(SEED_PADDING)
            noise = np.random.randint(
                low=-4, high=5, size=(np.abs(post_trim), np.shape(x)[1])
            ).astype("float")
            x = np.concatenate((noise, x), axis=0)
            x = np.concatenate((x, noise), axis=0)
        else:
            x = np.pad(x, [[0, np.abs(post_trim)], [0, 0]])
    else:
    elif post_trim > 0:
        x = x[:-post_trim, :]

    return x