Formatting
This commit is contained in:
parent
0270e7fcad
commit
1932b5365a
|
|
@ -60,9 +60,7 @@ class RecordingGenerator:
|
|||
|
||||
def generate_wb(self, num: int = 2, length: int = 8192):
|
||||
for i in range(num):
|
||||
recording = create_noise_recording(
|
||||
length=length, rms_power=0.2, seed=num
|
||||
)
|
||||
recording = create_noise_recording(length=length, rms_power=0.2, seed=num)
|
||||
recording.to_npy(filename=f"wb{i + 1}", overwrite=True)
|
||||
print(f"wb{i + 1} file saved.")
|
||||
|
||||
|
|
|
|||
|
|
@ -97,15 +97,13 @@ def create_lfm_recording(
|
|||
|
||||
|
||||
def create_noise_recording(
|
||||
rms_power: float,
|
||||
length: int,
|
||||
seed: int | None = None
|
||||
rms_power: float, length: int, seed: int | None = None
|
||||
) -> Recording:
|
||||
"""Generate a Recording of Additive White Gaussian Noise (AWGN)."""
|
||||
# 1. Create a repeating pseudo-random envelope
|
||||
if seed is not None:
|
||||
np.random.seed(seed)
|
||||
|
||||
|
||||
# 2. Sigma for complex AWGN:
|
||||
sigma = np.sqrt(rms_power / 2)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user