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

added extra log message; pulse samples adjustable

Browse source
This commit is contained in:
Sakimori 2025-07-18 16:55:56 -04:00
parent 88d125ccd9
commit 160a37b8cb
No known key found for this signature in database
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
signalgen.py
View file

@ -83,6 +83,8 @@ def addBurst(values):
#get bandpass and store in bpWeights
bpWeights = create.filter_weights(spectra)
logger.info(f"{path.basename(values.file)} loaded. Sampling pulse {values.nsamp} times.")
#create pulse
#first version is very simple, plan on adding more complex injections in future
pulseObj = create.SimpleGaussPulse(
@ -98,8 +100,8 @@ def addBurst(values):
nscint=0,
bandpass = bpWeights
)
pulse = pulseObj.sample_pulse(nsamp=int(3e5)) #sample it 300000 times
pulse = pulseObj.sample_pulse(nsamp=values.nsamp) #30000 by default
#inject pulse
injectedSpectra = inject.inject_constant_into_file(
yr_input = filterbankObj,
@ -131,7 +133,11 @@ if __name__ == "__main__":
parser.add_argument(
"-d", "--dm", dest="dm", type=float, help="DM of injected pulse."
)
parser.add_argument(
"-n", "--nsamp", type=int, help="Number of samples to take of the generated pulse."
)
parser.set_defaults(dm=250.0)
parser.set_defaults(nsamp=int(3e4))
parser.set_defaults(listfile=None)
parser.set_defaults(file=None)
values = parser.parse_args()