Skip to content
Snippets Groups Projects
Forked from lscsoft / bilby
2782 commits behind the upstream repository.
Code owners
Assign users and groups as approvers for specific file changes. Learn more.
plot_time_domain_data.py 1.31 KiB
#!/bin/python
"""
"""
from __future__ import division, print_function
import numpy as np
import tupak

np.random.seed(1)

duration = 4
sampling_frequency = 2048.

outdir = 'outdir'
label = 'example'

injection_parameters = dict(
    mass_1=36., mass_2=29., a_1=0.4, a_2=0.3, tilt_1=0.5, tilt_2=1.0,
    phi_12=1.7, phi_jl=0.3, luminosity_distance=1000., iota=0.4, psi=2.659,
    phase=1.3, geocent_time=1126259642.413, ra=1.375, dec=-1.2108)

waveform_arguments = dict(waveform_approximant='IMRPhenomPv2',
                          reference_frequency=50.)

waveform_generator = tupak.gw.WaveformGenerator(
    duration=duration, sampling_frequency=sampling_frequency,
    frequency_domain_source_model=tupak.gw.source.lal_binary_black_hole,
    parameters=injection_parameters, waveform_arguments=waveform_arguments)
hf_signal = waveform_generator.frequency_domain_strain()

H1 = tupak.gw.detector.get_interferometer_with_fake_noise_and_injection(
    'H1', injection_polarizations=hf_signal,
    injection_parameters=injection_parameters, duration=duration,
    sampling_frequency=sampling_frequency, outdir=outdir)

t0 = injection_parameters['geocent_time']
H1.plot_time_domain_data(outdir=outdir, label=label, notches=[50],
                         bandpass_frequencies=(50, 200), start_end=(-0.5, 0.5),
                         t0=t0)