From 160a37b8cba530213bb16d5a2ac2f02078b6b539 Mon Sep 17 00:00:00 2001
From: Sakimori <sakimorihalley@gmail.com>
Date: Fri, 18 Jul 2025 16:55:56 -0400
Subject: [PATCH] added extra log message; pulse samples adjustable

---
 signalgen.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

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()