diff --git a/signalgen.py b/signalgen.py index 782fee7..235c663 100755 --- a/signalgen.py +++ b/signalgen.py @@ -125,13 +125,14 @@ def addBurst(values): #and save the new plot show_dynamic(injectedSpectra, f"{basename} Dynamic Spectra and {values.dm} DM Pulse", save=True) #now generate new filterbank file + newName = f"{basename}_injected.fil" #build header from original file headerDic = filterbankObj.your_header.__dict__ #remove extraneous keys for key in ['filelist', 'filename']: headerDic.pop(key, None) - sigprocObj = make_sigproc_object(**headerDic) - newName = f"{basename}_injected.fil" + headerDic['rawdatafile'] = newName + sigprocObj = make_sigproc_object(**headerDic) sigprocObj.write_header(newName) sigprocObj.append_spectra(injectedSpectra, newName) logger.info(f"{newName} successfully written.")