signalgen.py

@@ -132,8 +132,8 @@ def addBurst(values):
 	pulseNum = rng.integers(1, high=MAX_BURSTS, endpoint=True)
 	logger.info(f"{basename} loaded. Sampling {pulseNum} pulse(s) {values.nsamp} times.")
 	starts = np.linspace(samples//4, 3*(samples//4), num=pulseNum, dtype=int)
-	for run in range(pulseNum):
-		start = starts[run]
+	while pulseNum > 0:
+		start = starts[pulseNum-1]
 		#create pulse
 
 		#check if this is part of a rng run
@@ -164,21 +164,16 @@ def addBurst(values):
 		
 		logger.info("Injecting pulse and saving file.")
 		#inject pulse
-		if run != pulseNum-1:
-			savename = f"{basename}_injected_{run}.fil"
-		else:
-			savename = f"{basename}_injected.fil"
 		inject.inject_constant_into_file(
 			yr_input = filterbankObj,
 			pulse = pulse,
 			start = start,
-			out_fil = path.join("./", savename)
+			out_fil = path.join("./", f"{basename}_injected.fil")
 		)
+		sleep(1) #was a write then read too fast causing issues?
 		#reload into object
-		if run != pulseNum-1: 
-			filterbankObj = Your(path.join("./", f"{basename}_injected_{run}.fil"))
-		if run > 0:
-			os.remove(path.join("./", f"{basename}_injected_{run-1}.fil"))
+		filterbankObj = Your(path.join("./", f"{basename}_injected.fil"))
+		pulseNum -= 1
 	os.remove(path.join("./", f"{basename}_trunc.fil")) #delete truncated file to save 2GB of disk space
 	logger.info(f"Truncated file removed.")