Printing injected SNRs

kazewong · Feb 16, 2024 · 7bb44ea · 7bb44ea
1 parent c477887
commit 7bb44ea
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/jimgw/single_event/detector.py b/src/jimgw/single_event/detector.py
@@ -406,6 +406,14 @@ def inject_signal(
         signal = self.fd_response(freqs, h_sky, params) * align_time
         self.data = signal + noise_real + 1j * noise_imag
 
+        # also calculate the optimal SNR and match filter SNR
+        optimal_SNR = jnp.sqrt(jnp.sum(signal * signal.conj() / var).real)
+        match_filter_SNR = jnp.sum(self.data * signal.conj() / var) / optimal_SNR
+
+        print(f"For detector {self.name}:")
+        print(f"The injected optimal SNR is {optimal_SNR}")
+        print(f"The injected match filter SNR is {match_filter_SNR}")
+
     @jaxtyped
     def load_psd(
         self, freqs: Float[Array, " n_sample"], psd_file: str = ""