diff --git a/.gitignore b/.gitignore index adb409d..0565243 100644 --- a/.gitignore +++ b/.gitignore @@ -27,3 +27,4 @@ venv.bak/ *.sigmf-meta images/ recordings/ +dat_recordings/ diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/recording_generation.py b/recording_generation.py index a01f559..f6fdf82 100644 --- a/recording_generation.py +++ b/recording_generation.py @@ -10,7 +10,7 @@ from signal_generation import (create_birdie_recording, create_ctnb_recording, class RecordingGenerator: - def __init__(self, sample_rate): + def __init__(self, sample_rate: int | float = 10e6): self.sample_rate = int(sample_rate) def generate_collision( @@ -78,13 +78,14 @@ class RecordingGenerator: def convert_to_dat( self, - source_directory: str = "/recordings", - save_directory: str = "/dat_recordings", + source_directory: str = "recordings", + save_directory: str = "dat_recordings", ): + os.makedirs(save_directory, exist_ok=True) for root, _, files in os.walk(source_directory): for name in files: filename = os.path.join(root, name) - savename = save_directory + name[:-4] + ".dat" + savename = save_directory + "/" + name[:-4] + ".dat" recording = from_npy(file=filename) data = recording.data[0]