zhetadelta/greenburstAux
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

now with downsampling, try full filterbank object

Browse source
This commit is contained in:
Abyss Halley 2025-07-22 11:42:33 -04:00
parent 120ae8a1c3
commit b8e41bac6b
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
signalgen.py
View file

@ -46,9 +46,12 @@ def show_dynamic(
""" """
#downsample the spectra first #downsample the spectra first
downFac = 24 downFac = np.shape(dynamic_spectra)[0] // 128000
logging.info(f"Downsampling...") if downFac > 1:
decSpectra = signal.decimate(dynamic_spectra, downFac, axis=0) logging.info(f"Downsampling...")
decSpectra = signal.decimate(dynamic_spectra, downFac, axis=0)
else:
decSpectra = dynamic_spectra
spectra_mads = median_fitter(np.median(decSpectra, axis=0)) spectra_mads = median_fitter(np.median(decSpectra, axis=0))
flat = decSpectra - spectra_mads flat = decSpectra - spectra_mads
@ -85,7 +88,7 @@ def addBurst(values):
filterbankObj = Your(values.file) filterbankObj = Your(values.file)
values.file = values.file.strip(".fil") #removing extension for text manipulation later values.file = values.file.strip(".fil") #removing extension for text manipulation later
basename = path.basename(values.file) basename = path.basename(values.file)
samples = 524288 samples = filterbankObj.your_header.nsamp
#the full filterbanks use 64GB in RAM when injecting burst, so we write out a truncated version and load that instead. #the full filterbanks use 64GB in RAM when injecting burst, so we write out a truncated version and load that instead.
filWriter = Writer(filterbankObj, outdir="./", outname = f"{basename}_trunc", nstart = 0, nsamp = samples) filWriter = Writer(filterbankObj, outdir="./", outname = f"{basename}_trunc", nstart = 0, nsamp = samples)
@ -205,7 +208,7 @@ if __name__ == "__main__":
logging.info(f"Running with file {values.file}") logging.info(f"Running with file {values.file}")
if values.plot: if values.plot:
filterbankObj = Your(values.file) filterbankObj = Your(values.file)
spectra = filterbankObj.get_data(0, 524288) spectra = filterbankObj.get_data(0, filterbankObj.your_header.nsamp)
show_dynamic(spectra, f"{values.file} Dynamic Spectra", save=True) show_dynamic(spectra, f"{values.file} Dynamic Spectra", save=True)
else: else:
addBurst(values) addBurst(values)