def __init__(self, *args, wimp_kwargs=None, **kwargs):
# Compute the energy spectrum in a given time range
# Times used by wimprates are J2000 timestamps
assert self.n_time_bins >= 1, "Need >= 1 time bin"
if hasattr(self, 'n_in'):
raise RuntimeError(
"n_in is gone! Use n_time_bins to control accuracy, or set "
"pretend_wimps_dont_modulate to use a time-averaged spectrum.")
times = np.linspace(wr.j2000(self.t_start.value),
wr.j2000(self.t_stop.value), self.n_time_bins + 1)
time_centers = self.bin_centers(times)
if wimp_kwargs is None:
# No arguments given at all;
# use default mass, xsec and energy range
wimp_kwargs = dict(mw=self.mw,
sigma_nucleon=self.sigma_nucleon,
es=self.es)
else:
assert 'mw' in wimp_kwargs and 'sigma_nucleon' in wimp_kwargs, \
"Pass at least 'mw' and 'sigma_nucleon' in wimp_kwargs"
if 'es' not in wimp_kwargs:
# Energies not given, use default energy bin edges
wimp_kwargs['es'] = self.es
es = wimp_kwargs['es']
es_centers = self.bin_centers(es)
del wimp_kwargs['es'] # To avoid confusion centers / edges
# Transform wimp_kwargs to arguments that can be passed to wimprates
# which means transforming es from edges to centers
spectra = np.array([
wr.rate_wimp_std(t=t, es=es_centers, **wimp_kwargs) * np.diff(es)
for t in time_centers
])
assert spectra.shape == (len(time_centers), len(es_centers))
self.energy_hist = Histdd.from_histogram(spectra,
bin_edges=(times, es))
if self.pretend_wimps_dont_modulate:
self.energy_hist.histogram = (
np.ones_like(self.energy_hist.histogram) *
self.energy_hist.sum(axis=0).histogram.reshape(1, -1) /
self.n_time_bins)
# Initialize the rest of the source, needs to be after energy_hist is
# computed because of _populate_tensor_cache
super().__init__(*args, **kwargs)
def __init__(self, *args, wimp_kwargs=None, **kwargs):
# Compute the energy spectrum in a given time range
# Times used by wimprates are J2000 timestamps
assert self.n_in > 1, \
f"Number of time bin edges needs to be at least 2"
times = np.linspace(wr.j2000(date=self.t_start),
wr.j2000(date=self.t_stop), self.n_in)
time_centers = self.bin_centers(times)
if wimp_kwargs is None:
# No arguments given at all;
# use default mass, xsec and energy range
wimp_kwargs = dict(mw=self.mw,
sigma_nucleon=self.sigma_nucleon,
es=self.es)
else:
assert 'mw' in wimp_kwargs and 'sigma_nucleon' in wimp_kwargs, \
"Pass at least 'mw' and 'sigma_nucleon' in wimp_kwargs"
if 'es' not in wimp_kwargs:
# Energies not given, use default energy bin edges
wimp_kwargs['es'] = self.es
es = wimp_kwargs['es']
es_centers = self.bin_centers(es)
del wimp_kwargs['es'] # To avoid confusion centers / edges
# Transform wimp_kwargs to arguments that can be passed to wimprates
# which means transforming es from edges to centers
spectra = np.array([
wr.rate_wimp_std(t=t, es=es_centers, **wimp_kwargs) * np.diff(es)
for t in time_centers
])
assert spectra.shape == (len(time_centers), len(es_centers))
self.energy_hist = Histdd.from_histogram(spectra,
bin_edges=(times, es))
# Initialize the rest of the source, needs to be after energy_hist is
# computed because of _populate_tensor_cache
super().__init__(*args, **kwargs)
