diff --git a/signalgen.py b/signalgen.py index 916be76..e94768d 100755 --- a/signalgen.py +++ b/signalgen.py @@ -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()