diff --git a/signalgen.py b/signalgen.py index e6e7ce4..cab48ad 100755 --- a/signalgen.py +++ b/signalgen.py @@ -164,15 +164,19 @@ 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("./", f"{basename}_injected.fil") + out_fil = path.join("./", savename) ) - sleep(1) #was a write then read too fast causing issues? #reload into object - filterbankObj = Your(path.join("./", f"{basename}_injected_{run}.fil")) + 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")) os.remove(path.join("./", f"{basename}_trunc.fil")) #delete truncated file to save 2GB of disk space