madrigal/recording-generation
1
0
Fork 0
Code Issues Pull Requests Workflows Packages Projects Releases Wiki Activity

Added __init__, added dat_recordings to gitignore, minor fixes to convert_to_dat

Browse Source
This commit is contained in:
madrigal 2025-07-24 10:12:50 -04:00
parent 93738c28c5
commit f6173b4037
3 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -27,3 +27,4 @@ venv.bak/
*.sigmf-meta
images/
recordings/
dat_recordings/

0
__init__.py Normal file
View File

9
recording_generation.py
View File

@ -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]