diff --git a/signalgen.py b/signalgen.py index 235c663..96348f0 100755 --- a/signalgen.py +++ b/signalgen.py @@ -126,13 +126,27 @@ def addBurst(values): 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) - headerDic['rawdatafile'] = newName - sigprocObj = make_sigproc_object(**headerDic) + sigprocObj = make_sigproc_object( + rawdatafile=newName, #d + source_name="TEMP", #d + nchans=filterbankObj.nchans, + foff=filterbankObj.your_header.foff, # MHz + fch1=filterbankObj.your_header.fch1, # MHz + tsamp=filterbankObj.your_header.tsamp, # seconds + tstart=filterbankObj.your_header.tstart, # MJD + src_raj=filterbankObj.your_header.ra_deg, + src_dej=filterbankObj.your_header.dec_deg, + machine_id=0, + nbeams=0, + ibeam=0, + nbits=filterbankObj.your_header.nbits, + nifs=1, + barycentric=0, + pulsarcentric=0, + data_type=0, + az_start=-1, + za_start=-1, + ) sigprocObj.write_header(newName) sigprocObj.append_spectra(injectedSpectra, newName) logger.info(f"{newName} successfully written.")