signalgen.py

@@ -37,7 +37,7 @@ rng = np.random.default_rng()
 def show_dynamic(
     dynamic_spectra: np.ndarray,
     title: Union[str, None] = None,
-    save: Union[bool, None] = False
+    save: Union[bool, None] = False,
 ) -> None:
     """
     Show a dynamic spectra by first flattening it
@@ -264,9 +264,6 @@ if __name__ == "__main__":
 	parser.add_argument(
 		"-o", "--output", dest="output", type=str, help="Set output directory."
 	)
-	parser.add_argument(
-		"-s", "--start", type=int, help="Zoomed plot start time bin."
-	)
 	parser.set_defaults(dm=250.0)
 	parser.set_defaults(pWidth=0.001)
 	parser.set_defaults(nsamp=int(3e5))
@@ -276,7 +273,6 @@ if __name__ == "__main__":
 	parser.set_defaults(plot=False)
 	parser.set_defaults(rsamp=False)
 	parser.set_defaults(output=None)
-	parser.set_defaults(start=-1)
 	values = parser.parse_args()
 
 	#set working directory to ignored directory or set output
@@ -294,10 +290,7 @@ if __name__ == "__main__":
 		logging.info(f"Running with file {values.file}")
 		if values.plot:
 			filterbankObj = Your(values.file)
-			if values.start == -1:
-				spectra = filterbankObj.get_data(0, 524288)
-			else:
-				spectra = filterbankObj.get_data(values.start, 8900)
+			spectra = filterbankObj.get_data(0, 524288)
 			show_dynamic(spectra, f"{values.file} Dynamic Spectra", save=True)
 		else:
 			addBurst(values)

testanalysis.py

@@ -11,9 +11,8 @@ import pandas as pd
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO) #change for debug prints
 
-INJECTION_FILE = os.path.join(".","out","2025-08-08T11-28-06.txt")
+INJECTION_FILE = os.path.join(".","out","2025-07-31T17-25-54.txt")
 DETECTION_FILE = os.path.join(".","out","plotOut.txt")
-EXCLUDE_FILES = True
 
 def linesplit(line):
     """
@@ -92,22 +91,20 @@ detections = detections.reset_index(drop=True)
 summary(2)
 
 #five files have SO MANY FALSE POSITIVES so get rid of them here?
-if EXCLUDE_FILES:
-    remFiles = ["data_2025-04-30_07-53-07", "data_2025-05-01_07-47-34", "data_2025-04-24_07-36-04", "data_2025-04-29_07-50-16", "data_2025-04-30_08-18-17"]
-    logger.info("Filtering out the five problem files...")
-    remMask = [True] * len(detections['dm'])
-    for detection in detections.itertuples():
-        if detection.file in remFiles:
-            remMask[detection.Index] = False
-    detections = detections[remMask]
-    detections = detections.reset_index(drop=True)
-    summary(3)
+remFiles = ["data_2025-04-30_07-53-07", "data_2025-05-01_07-47-34", "data_2025-04-24_07-36-04", "data_2025-04-29_07-50-16", "data_2025-04-30_08-18-17"]
+logger.info("Filtering out the five problem files...")
+remMask = [True] * len(detections['dm'])
+for detection in detections.itertuples():
+    if detection.file in remFiles:
+        remMask[detection.Index] = False
+detections = detections[remMask]
+detections = detections.reset_index(drop=True)
+summary(3)
 
-    postFilterCount = len(detections['dm'])
-
-    logger.info(f"Removed {preFilterCount-postFilterCount} detections by filtering the following:")
-    for file in remFiles:
-        logger.info(file+"_injected.fil")
+postFilterCount = len(detections['dm'])
+logger.info(f"Removed {preFilterCount-postFilterCount} detections by filtering the following:")
+for file in remFiles:
+    logger.info(file+"_injected.fil")
 
 #Let's do detection matching! Yaaaay!
 #What detections line up to which injections? This will determine which ones got missed entirely.