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 *.sigmf-meta
images/ images/
recordings/ 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: class RecordingGenerator:
def __init__(self, sample_rate): def __init__(self, sample_rate: int | float = 10e6):
self.sample_rate = int(sample_rate) self.sample_rate = int(sample_rate)
def generate_collision( def generate_collision(
@ -78,13 +78,14 @@ class RecordingGenerator:
def convert_to_dat( def convert_to_dat(
self, self,
source_directory: str = "/recordings", source_directory: str = "recordings",
save_directory: str = "/dat_recordings", save_directory: str = "dat_recordings",
): ):
os.makedirs(save_directory, exist_ok=True)
for root, _, files in os.walk(source_directory): for root, _, files in os.walk(source_directory):
for name in files: for name in files:
filename = os.path.join(root, name) filename = os.path.join(root, name)
savename = save_directory + name[:-4] + ".dat" savename = save_directory + "/" + name[:-4] + ".dat"
recording = from_npy(file=filename) recording = from_npy(file=filename)
data = recording.data[0] data = recording.data[0]