Skip to content
Snippets Groups Projects
Commit ea2dde97 authored by Aditya Vijaykumar's avatar Aditya Vijaykumar
Browse files

changes

parent 8c7b8d34
No related branches found
No related tags found
1 merge request!1105Relative Binning in bilby
......@@ -1402,13 +1402,22 @@ def compute_snrs(sample, likelihood, npool=1):
"""
if likelihood is not None:
if likelihood.__class__.__name__ == "RelativeBinningGravitationalWaveTransient":
logger.info("Relative Binning Likelihood; Calculating SNRs from Summary Data")
if isinstance(sample, dict):
if likelihood.__class__.__name__ == "RelativeBinningGravitationalWaveTransient":
waveform_ratio = likelihood.compute_waveform_ratio(sample)
else:
signal_polarizations = likelihood.waveform_generator.frequency_domain_strain(sample)
likelihood.parameters.update(sample)
signal_polarizations =\
likelihood.waveform_generator.frequency_domain_strain(sample)
for ifo in likelihood.interferometers:
per_detector_snr = likelihood.calculate_snrs(
signal_polarizations, ifo)
if likelihood.__class__.__name__ == "RelativeBinningGravitationalWaveTransient":
per_detector_snr = likelihood.calculate_snrs_relative_binning(waveform_ratio[ifo.name], ifo)
else:
per_detector_snr = likelihood.calculate_snrs(
signal_polarizations, ifo)
sample['{}_matched_filter_snr'.format(ifo.name)] =\
per_detector_snr.complex_matched_filter_snr
sample['{}_optimal_snr'.format(ifo.name)] = \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment