.gitignore

@@ -19,7 +19,6 @@ venv.bak/
 .vscode/
 
 # Generated files
-*.dat
 *.dot
 *.hdf5
 *.npy

recording_generation.py

@@ -2,16 +2,11 @@ import os
 import random
 
 import numpy as np
-from utils.data.recording import Recording
 from utils.io.recording import from_npy
 
-from signal_generation import (
-    create_birdie_recording,
-    create_ctnb_recording,
-    create_lfm_recording,
-    create_modulated_signal,
-    create_noise_recording,
-)
+from signal_generation import (create_birdie_recording, create_ctnb_recording,
+                               create_lfm_recording, create_modulated_signal,
+                               create_noise_recording)
 
 
 class RecordingGenerator:
@@ -33,9 +28,7 @@ class RecordingGenerator:
                 recording = create_modulated_signal(
                     modulation=modulation, sps=sps, beta=roll_off, length=length
                 )
-                recording.to_npy(
-                    filename=f"{modulation}_{roll_off_str}", overwrite=True
-                )
+                recording.to_npy(filename=f"{modulation}_{roll_off_str}")
                 print(f"{modulation}_{roll_off_str} file saved.")
 
     def generate_lfm(
@@ -55,21 +48,22 @@ class RecordingGenerator:
             )
 
             print(f"LFM chirp length: {int(self.sample_rate * chirp_period)}")
-            recording.to_npy(filename=f"{chirp_type}_chirp", overwrite=True)
+            recording.to_npy(filename=f"{chirp_type}_chirp")
             print(f"{chirp_type}_chirp file saved.")
 
     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, counter=random.choice([2, 4, 8, 16, 32])
+                length=length,
+                rms_power=0.2,
             )
-            recording.to_npy(filename=f"wb{i + 1}", overwrite=True)
+            recording.to_npy(filename=f"wb{i + 1}")
             print(f"wb{i + 1} file saved.")
 
     def generate_ctnb(self, num: int = 2, length: int = 8192):
         for i in range(num):
             recording = create_ctnb_recording(length=length)
-            recording.to_npy(filename=f"ctnb{i + 1}", overwrite=True)
+            recording.to_npy(filename=f"ctnb{i + 1}")
             print(f"ctnb{i + 1} file saved.")
 
     def generate_birdie(self, num: int = 2, length: int = 8192, wave_num: int = 5):
@@ -79,14 +73,9 @@ class RecordingGenerator:
                 length=length,
                 wave_number=int(wave_num + i),
             )
-            recording.to_npy(filename=f"birdie{i + 1}", overwrite=True)
+            recording.to_npy(filename=f"birdie{i + 1}")
             print(f"birdie{i + 1} file saved.")
 
-    def generate_zeros(self, length: int = 8192):
-        data = np.zeros(shape=length, dtype=np.complex64)
-        recording = Recording(data=data)
-        recording.to_npy(filename="zero", overwrite=True)
-
     def convert_to_dat(
         self,
         source_directory: str = "recordings",

signal_generation.py

@@ -34,9 +34,13 @@ class ModulationRegistry:
         return cls._registry[mod_type]
 
 
-def create_modulated_signal(
-    modulation: str, sps: int, beta: float, length: int
-) -> Recording:
+def periodic_random(length, divisor=4, seed=256):
+    np.random.seed(seed)
+    chunk = np.random.rand(int(length / divisor))
+    return np.tile(chunk, divisor)
+
+
+def create_modulated_signal(modulation: str, sps: int, beta, length: int) -> Recording:
     """Produces a modulated signal Recording."""
     mod_info = ModulationRegistry.get(modulation)
     if mod_info is None:
@@ -52,7 +56,7 @@ def create_modulated_signal(
         num_bits_per_symbol=mod_info.bps,
     )
     upsampler = block_generator.Upsampling(factor=sps)
-    filter = block_generator.RootRaisedCosineFilter(
+    filter = block_generator.RaisedCosineFilter(
         span_in_symbols=10, upsampling_factor=sps, beta=beta
     )
 
@@ -62,20 +66,37 @@ def create_modulated_signal(
 
     upsampled_symbols = upsampler([symbols])
     filtered_samples = filter([upsampled_symbols])
-    start = (len(filtered_samples) - length) // 2
-    end = start + length
-    output_recording = filtered_samples[start:end]
+    output_recording = filtered_samples[length : length * 2]
 
-    metadata = {
-        "modulation": modulation,
-        "bits_per_symbol": mod_info.bps,
-        "constellation": mod_info.constellation,
-        "sps": sps,
-        "beta": beta,
-        "source": "signal.block_generator",
-    }
+    return Recording(data=output_recording)
 
-    return Recording(data=output_recording, metadata=metadata)
+
+# Old create_modulated_signal code
+# def create_modulated_signal(modulation: str, sps: int, beta, length: int) -> Recording:
+#     """Produces a modulated signal Recording."""
+#     mod_info = ModulationRegistry.get(modulation)
+#     if mod_info is None:
+#         raise ValueError(f"Modulation {modulation} not in registry.")
+
+#     source_block = block_generator.RandomBinarySource()
+#     mapper_block = block_generator.Mapper(
+#         constellation_type=mod_info.constellation,
+#         num_bits_per_symbol=mod_info.bps,
+#     )
+#     upsampler_block = block_generator.Upsampling(factor=sps)
+#     filter_block = block_generator.RaisedCosineFilter(upsampling_factor=sps, beta=beta)
+
+#     mapper_block.connect_input([source_block])
+#     upsampler_block.connect_input([mapper_block])
+#     filter_block.connect_input([upsampler_block])
+
+#     dividing_factor = sps * mod_info.bps
+#     while length % dividing_factor != 0:
+#         length = length + 1
+#     double_length = length * 2
+
+#     recording = filter_block.record(num_samples=double_length)
+#     return Recording(data=recording.data[:, :length], metadata=recording.metadata)
 
 
 def create_lfm_recording(
@@ -99,11 +120,10 @@ def create_lfm_recording(
 def create_noise_recording(
     rms_power: float,
     length: int,
-    counter: int,
 ) -> Recording:
     """Generate a Recording of Additive White Gaussian Noise (AWGN)."""
     # 1. Create a repeating pseudo-random envelope
-    np.random.seed(256 + counter)
+    np.random.seed(256)
     chunk = np.random.rand(length // 4)
     tiled = np.tile(chunk, 4)
     amplitude_envelope = np.sqrt(tiled)
@@ -125,13 +145,11 @@ def create_ctnb_recording(length: int) -> Recording:
 
 
 def create_birdie_recording(
-    sample_rate: int, length: int, wave_number: int, sps: int = 1
+    sample_rate: int, length: int, wave_number: int
 ) -> Recording:
     recording_data = np.zeros(int(length))
     for _ in range(wave_number):
-        frequency = np.random.choice(
-            np.arange(-sample_rate / (2 * sps), sample_rate / (2 * sps))
-        )
+        frequency = np.random.choice(np.arange(-sample_rate / 2, sample_rate / 2))
         recording = complex_sine(
             sample_rate=int(sample_rate), length=int(length), frequency=int(frequency)
         )