bahamas.psd_strain

Submodules

bahamas.psd_strain.psd_function

Module for generating power spectral density (PSD) models for gravitational wave signals. This module includes various signal models, noise models, and functions to compute the total PSD from multiple sources. It also provides functions for generating Gaussian processes and averaging data. The module uses JAX for efficient computation and supports both time and frequency domain analysis.

bahamas.psd_strain.psd_function.Omega_extra_foreground(freqs, par)

Extra galactic foreground of white dwarf binaries. https://arxiv.org/abs/2407.10642

Parameters:

  • freqs (array) – Frequency array.

  • par (dict) – Parameters for the model.

Returns:

Power spectral density.

Return type:

array

bahamas.psd_strain.psd_function.Omega_gaussian_bump(freqs, par)

Power spectral density for gravitational waves modeled as a Gaussian bump.

Parameters:

  • freqs (array) – Frequency array.

  • par (dict) – Parameters for the model.

Returns:

Power spectral density.

Return type:

array

bahamas.psd_strain.psd_function.Omega_phase_transition(freqs, par)

Power spectral density for gravitational waves from phase transitions.

Parameters:

  • freqs (array) – Frequency array.

  • par (dict) – Parameters for the model.

Returns:

Power spectral density.

Return type:

array

bahamas.psd_strain.psd_function.Omega_pl(freqs, par)

” Power-law model for gravitational wave signal.

Parameters:

  • freqs (array) – Frequency array.

  • par (dict) – Parameters for the model.

Returns:

Power spectral density.

Return type:

array

bahamas.psd_strain.psd_function.galactic_foreground(freqs, par, injected=False, gen2=False)

Galactic foreground of white dwarf binaries. https://arxiv.org/abs/2103.14598

Parameters:

  • freqs (array) – Frequency array.

  • par (dict) – Parameters for the model.

  • injected (bool) – If True, uses injected parameters.

Returns:

Power spectral density.

Return type:

array

bahamas.psd_strain.psd_function.galactic_foreground_time(freqs, par, injected=False, t1=0, t2=0, tdi=0, gen2=False)

Galactic foreground model for gravitational wave signal with time dependence. https://arxiv.org/abs/2410.08274, https://arxiv.org/abs/2410.08263

Parameters:

  • freqs (array) – Frequency array.

  • injected (bool) – If True, uses injected parameters.

  • par (dict) – Parameters for the model.

  • t1 (float) – Start time for the envelope.

  • t2 (float) – End time for the envelope.

  • tdi (int) – TDI channel (0 = A, 1 = E).

Returns:

Power spectral density.

Return type:

array

bahamas.psd_strain.psd_function.model_psd(freqs, sources, response, injected=False, tdi=0, **kwargs)

Computes the total power spectral density (PSD) by combining multiple sources.

Parameters:

  • freqs (array) – Frequency array.

  • sources (dict) – Dictionary of sources with their parameters.

  • response (float) – Response factor to scale the PSD.

  • injected (bool, optional) – If True, uses injected parameters for sources. Defaults to False.

  • tdi (int, optional) – Time-delay interferometry channel (0 = A, 1 = E). Defaults to 0.

  • **kwargs – Additional arguments for specific sources (e.g., t1, t2).

Returns:

Total PSD if injected is False. tuple: Total PSD and list of true PSDs for each source if injected is True.

Return type:

array

bahamas.psd_strain.psd_function.noise(freq, par, gen2=False)

Computes the noise power spectral density (PSD).

This function calculates the noise PSD for LISA, including contributions from test mass noise and the optical metrology system. It supports both standard and TDI2 (Time-Delay Interferometry 2) configurations.

Parameters:

  • freq (array) – Frequency array in Hz.

  • par (dict) – Dictionary containing the parameters: - ‘A’ (float): Amplitude of the test mass noise. - ‘P’ (float): Amplitude of the optical metrology system noise.

  • tdi2 (bool, optional) – If True, applies the TDI2 factor. Defaults to False.

Returns:

Noise power spectral density (PSD) as a function of frequency.

Return type:

array