From f6173b40371e9f7f1a125aad858e6386bb2aba3c Mon Sep 17 00:00:00 2001 From: madrigal Date: Thu, 24 Jul 2025 10:12:50 -0400 Subject: [PATCH] Added __init__, added dat_recordings to gitignore, minor fixes to convert_to_dat --- .gitignore | 1 + __init__.py | 0 recording_generation.py | 9 +++++---- 3 files changed, 6 insertions(+), 4 deletions(-) create mode 100644 __init__.py 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]