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