def __init__(self, filename):
self.filename = get_data_file(filename)
self.handle = pyfits.open(self.filename, memmap=True)
if "MATRIX" in self.handle:
self.mat_key = "MATRIX"
elif "SPECRESP MATRIX" in self.handle:
self.mat_key = "SPECRESP MATRIX"
else:
raise RuntimeError(f"Cannot find the response matrix in the RMF "
f"file {filename}! It should be named "
f"\"MATRIX\" or \"SPECRESP MATRIX\".")
self.header = self.handle[self.mat_key].header
self.num_mat_columns = len(self.handle[self.mat_key].columns)
self.ebounds_header = self.handle["EBOUNDS"].header
self.weights = np.array([w.sum() for w in self.data["MATRIX"]])
self.elo = self.data["ENERG_LO"]
self.ehi = self.data["ENERG_HI"]
self.ebins = np.append(self.data["ENERG_LO"], self.data["ENERG_HI"][-1])
self.emid = 0.5*(self.elo+self.ehi)
self.de = self.ehi-self.elo
self.n_e = self.elo.size
self.n_ch = self.header["DETCHANS"]
num = 0
for i in range(1, self.num_mat_columns+1):
if self.header[f"TTYPE{i}"] == "F_CHAN":
num = i
break
self.cmin = self.header.get(f"TLMIN{num}", 1)
self.cmax = self.header.get(f"TLMAX{num}", self.n_ch)
def __init__(self, inst, prng=None):
super().__init__(prng)
img_file = get_data_file(inst['psf'][1])
hdu = inst['psf'][2]
plate_scale_arcmin = inst['fov'] / inst['num_pixels']
plate_scale_deg = plate_scale_arcmin / 60.0
plate_scale_mm = inst['focal_length'] * 1e3 * np.deg2rad(
plate_scale_deg)
self.imhdu = pyfits.open(get_data_file(img_file))[hdu]
self.imctr = np.array(
[self.imhdu.header["CRPIX1"], self.imhdu.header["CRPIX2"]])
unit = self.imhdu.header.get("CUNIT1", "mm")
self.scale = Quantity(
[self.imhdu.header["CDELT1"], self.imhdu.header["CDELT2"]],
unit).to_value('mm')
self.scale /= plate_scale_mm
def __init__(self, filename):
self.filename = get_data_file(filename)
f = pyfits.open(self.filename)
self.elo = f["SPECRESP"].data.field("ENERG_LO")
self.ehi = f["SPECRESP"].data.field("ENERG_HI")
self.emid = 0.5*(self.elo+self.ehi)
self.eff_area = np.nan_to_num(
f["SPECRESP"].data.field("SPECRESP")).astype("float64")
self.max_area = self.eff_area.max()
f.close()
def __init__(self, inst, prng=None):
super().__init__(prng)
self.img_file = get_data_file(inst['psf'][1])
self.det_ctr = np.array(inst['aimpt_coords'])
plate_scale_arcmin = inst['fov'] / inst['num_pixels']
plate_scale_deg = plate_scale_arcmin / 60.0
plate_scale_mm = inst['focal_length'] * 1e3 * np.deg2rad(
plate_scale_deg)
img_e = []
img_r = []
img_i = []
img_c = []
img_s = []
img_u = []
with pyfits.open(self.img_file, lazy_load_hdus=True) as f:
for i, hdu in enumerate(f):
if not hdu.is_image:
continue
img_e.append(hdu.header["ENERGY"])
key = "THETA" if "OFFAXIS" not in hdu.header else "OFFAXIS"
img_r.append(hdu.header[key])
img_c.append([hdu.header["CRPIX1"], hdu.header["CRPIX2"]])
img_s.append([hdu.header["CDELT1"], hdu.header["CDELT2"]])
img_i.append(i)
img_u.append(hdu.header.get("CUNIT1", "mm"))
self.img_e, ie = np.unique(img_e, return_inverse=True)
if np.all(self.img_e > 100.0):
# this is probably in eV
self.img_e *= 1.0e-3
self.img_r2, ir = np.unique(img_r, return_inverse=True)
self.img_i = {j: (i, ie[j], ir[j]) for j, i in enumerate(img_i)}
self.num_images = len(img_e)
self.img_r2 = (self.img_r2 / plate_scale_arcmin)**2
self.img_c = np.array(img_c)
unit = list(set(img_u))
if len(unit) > 1:
raise RuntimeError("More than one delta unit detected!!")
self.img_s = Quantity(img_s, unit[0]).to_value('mm') / plate_scale_mm
def from_filename(cls, filename, ext_area, focal_length):
"""
Read an instrumental background spectrum from
a FITS PHA file.
Parameters
----------
filename : string
The path to the file containing the spectrum.
focal_length : float, (value, unit) tuple, or :class:`~astropy.units.Quantity`
The default focal length of the instrument
in meters.
"""
fn = get_data_file(filename)
with pyfits.open(fn) as f:
hdu = f["SPECTRUM"]
exp_time = hdu.header["EXPOSURE"]
if "COUNTS" in hdu.data.names:
count_rate = hdu.data["COUNTS"] / exp_time
else:
count_rate = hdu.data["COUNT_RATE"]
count_rate /= ext_area
channel = hdu.data["CHANNEL"]
return cls(channel, count_rate, focal_length, exp_time)
def simulate_spectrum(spec,
instrument,
exp_time,
out_file,
instr_bkgnd=False,
foreground=False,
ptsrc_bkgnd=False,
bkgnd_area=None,
absorb_model="wabs",
nH=0.05,
overwrite=False,
prng=None):
"""
Generate a PI or PHA spectrum from a :class:`~soxs.spectra.Spectrum`
by convolving it with responses. To be used if one wants to
create a spectrum without worrying about spatial response. Similar
to XSPEC's "fakeit".
Parameters
----------
spec : :class:`~soxs.spectra.Spectrum`
The spectrum to be convolved. If None is supplied, only backgrounds
will be simulated (if they are turned on).
instrument : string
The name of the instrument to use, which picks an instrument
specification from the instrument registry.
exp_time : float, (value, unit) tuple, or :class:`~astropy.units.Quantity`
The exposure time in seconds.
out_file : string
The file to write the spectrum to.
instr_bkgnd : boolean, optional
Whether or not to include the instrumental/particle background.
Default: False
foreground : boolean, optional
Whether or not to include the local foreground.
Default: False
ptsrc_bkgnd : boolean, optional
Whether or not to include the unresolved point-source background.
Default: False
bkgnd_area : float, (value, unit) tuple, or :class:`~astropy.units.Quantity`
The area on the sky for the background components, in square arcminutes.
Default: None, necessary to specify if any of the background components
are turned on.
absorb_model : string, optional
The absorption model to use, "wabs" or "tbabs". Default: "wabs"
nH : float, optional
The hydrogen column in units of 10**22 atoms/cm**2.
Default: 0.05
overwrite : boolean, optional
Whether or not to overwrite an existing file. Default: False
prng : :class:`~numpy.random.RandomState` object, integer, or None
A pseudo-random number generator. Typically will only
be specified if you have a reason to generate the same
set of random numbers, such as for a test. Default is None,
which sets the seed based on the system time.
Examples
--------
>>> spec = soxs.Spectrum.from_file("my_spectrum.txt")
>>> soxs.simulate_spectrum(spec, "lynx_lxm", 100000.0,
... "my_spec.pi", overwrite=True)
"""
from soxs.events import _write_spectrum
from soxs.response import RedistributionMatrixFile, \
AuxiliaryResponseFile
from soxs.spectra import ConvolvedSpectrum
from soxs.background.foreground import hm_astro_bkgnd
from soxs.background.spectra import BackgroundSpectrum
from soxs.background.instrument import InstrumentalBackground
prng = parse_prng(prng)
exp_time = parse_value(exp_time, "s")
try:
instrument_spec = instrument_registry[instrument]
except KeyError:
raise KeyError(
f"Instrument {instrument} is not in the instrument registry!")
if foreground or instr_bkgnd or ptsrc_bkgnd:
if instrument_spec["grating"]:
raise NotImplementedError(
"Backgrounds cannot be included in simulations "
"of gratings spectra at this time!")
if bkgnd_area is None:
raise RuntimeError(
"The 'bkgnd_area' argument must be set if one wants "
"to simulate backgrounds! Specify a value in square "
"arcminutes.")
bkgnd_area = np.sqrt(parse_value(bkgnd_area, "arcmin**2"))
elif spec is None:
raise RuntimeError(
"You have specified no source spectrum and no backgrounds!")
arf_file = get_data_file(instrument_spec["arf"])
rmf_file = get_data_file(instrument_spec["rmf"])
arf = AuxiliaryResponseFile(arf_file)
rmf = RedistributionMatrixFile(rmf_file)
event_params = {
"RESPFILE": os.path.split(rmf.filename)[-1],
"ANCRFILE": os.path.split(arf.filename)[-1],
"TELESCOP": rmf.header["TELESCOP"],
"INSTRUME": rmf.header["INSTRUME"],
"MISSION": rmf.header.get("MISSION", "")
}
out_spec = np.zeros(rmf.n_ch)
if spec is not None:
cspec = ConvolvedSpectrum.convolve(spec, arf)
out_spec += rmf.convolve_spectrum(cspec, exp_time, prng=prng)
fov = None if bkgnd_area is None else np.sqrt(bkgnd_area)
if foreground:
mylog.info("Adding in astrophysical foreground.")
cspec_frgnd = ConvolvedSpectrum.convolve(
hm_astro_bkgnd.to_spectrum(fov), arf)
out_spec += rmf.convolve_spectrum(cspec_frgnd, exp_time, prng=prng)
if instr_bkgnd and instrument_spec["bkgnd"] is not None:
mylog.info("Adding in instrumental background.")
bkgnd_spec = instrument_spec["bkgnd"]
# Temporary hack for ACIS-S
if "aciss" in instrument_spec["name"]:
bkgnd_spec = bkgnd_spec[1]
bkgnd_spec = InstrumentalBackground.from_filename(
bkgnd_spec[0], bkgnd_spec[1], instrument_spec['focal_length'])
out_spec += bkgnd_spec.generate_channel_spectrum(exp_time,
bkgnd_area,
prng=prng)
if ptsrc_bkgnd:
mylog.info("Adding in background from unresolved point-sources.")
spec_plaw = BackgroundSpectrum.from_powerlaw(1.45,
0.0,
2.0e-7,
emin=0.01,
emax=10.0,
nbins=300000)
spec_plaw.apply_foreground_absorption(nH, model=absorb_model)
cspec_plaw = ConvolvedSpectrum.convolve(spec_plaw.to_spectrum(fov),
arf)
out_spec += rmf.convolve_spectrum(cspec_plaw, exp_time, prng=prng)
bins = (np.arange(rmf.n_ch) + rmf.cmin).astype("int32")
_write_spectrum(bins,
out_spec,
exp_time,
rmf.header["CHANTYPE"],
event_params,
out_file,
overwrite=overwrite)
def make_background(exp_time,
instrument,
sky_center,
foreground=True,
ptsrc_bkgnd=True,
instr_bkgnd=True,
no_dither=False,
dither_params=None,
roll_angle=0.0,
subpixel_res=False,
input_sources=None,
absorb_model="wabs",
nH=0.05,
aimpt_shift=None,
prng=None):
"""
Make background events.
Parameters
----------
exp_time : float, (value, unit) tuple, or :class:`~astropy.units.Quantity`
The exposure time to use, in seconds.
instrument : string
The name of the instrument to use, which picks an instrument
specification from the instrument registry.
sky_center : array, tuple, or list
The center RA, Dec coordinates of the observation, in degrees.
foreground : boolean, optional
Whether or not to include the Galactic foreground. Default: True
instr_bkgnd : boolean, optional
Whether or not to include the instrumental background. Default: True
no_dither : boolean, optional
If True, turn off dithering entirely. Default: False
dither_params : array-like of floats, optional
The parameters to use to control the size and period of the dither
pattern. The first two numbers are the dither amplitude in x and y
detector coordinates in arcseconds, and the second two numbers are
the dither period in x and y detector coordinates in seconds.
Default: [8.0, 8.0, 1000.0, 707.0].
ptsrc_bkgnd : boolean, optional
Whether or not to include the point-source background. Default: True
Default: 0.05
roll_angle : float, (value, unit) tuple, or :class:`~astropy.units.Quantity`, optional
The roll angle of the observation in degrees. Default: 0.0
subpixel_res: boolean, optional
If True, event positions are not randomized within the pixels
within which they are detected. Default: False
input_sources : string, optional
If set to a filename, input the point source positions, fluxes,
and spectral indices from an ASCII table instead of generating
them. Default: None
absorb_model : string, optional
The absorption model to use, "wabs" or "tbabs". Default: "wabs"
nH : float, optional
The hydrogen column in units of 10**22 atoms/cm**2.
Default: 0.05
aimpt_shift : array-like, optional
A two-float array-like object which shifts the aimpoint on the
detector from the nominal position. Units are in arcseconds.
Default: None, which results in no shift from the nominal aimpoint.
prng : :class:`~numpy.random.RandomState` object, integer, or None
A pseudo-random number generator. Typically will only
be specified if you have a reason to generate the same
set of random numbers, such as for a test. Default is None,
which sets the seed based on the system time.
"""
from soxs.background import make_instrument_background, \
make_foreground, make_ptsrc_background
prng = parse_prng(prng)
exp_time = parse_value(exp_time, "s")
roll_angle = parse_value(roll_angle, "deg")
try:
instrument_spec = instrument_registry[instrument]
except KeyError:
raise KeyError(f"Instrument {instrument} is not in the "
f"instrument registry!")
if not instrument_spec["imaging"]:
raise RuntimeError(f"Instrument '{instrument_spec['name']}' is not "
f"designed for imaging observations!")
fov = instrument_spec["fov"]
input_events = defaultdict(list)
arf_file = get_data_file(instrument_spec["arf"])
arf = AuxiliaryResponseFile(arf_file)
rmf_file = get_data_file(instrument_spec["rmf"])
rmf = RedistributionMatrixFile(rmf_file)
if ptsrc_bkgnd:
mylog.info("Adding in point-source background.")
ptsrc_events = make_ptsrc_background(exp_time,
fov,
sky_center,
area=1.2 * arf.max_area,
input_sources=input_sources,
absorb_model=absorb_model,
nH=nH,
prng=prng)
for key in ["ra", "dec", "energy"]:
input_events[key].append(ptsrc_events[key])
input_events["flux"].append(ptsrc_events["flux"])
input_events["emin"].append(ptsrc_events["energy"].min())
input_events["emax"].append(ptsrc_events["energy"].max())
input_events["src_names"].append("ptsrc_bkgnd")
events, event_params = generate_events(input_events,
exp_time,
instrument,
sky_center,
no_dither=no_dither,
dither_params=dither_params,
roll_angle=roll_angle,
subpixel_res=subpixel_res,
aimpt_shift=aimpt_shift,
prng=prng)
mylog.info(f"Generated {events['energy'].size} photons from "
f"the point-source background.")
else:
nx = instrument_spec["num_pixels"]
plate_scale = instrument_spec["fov"] / nx / 60.0
plate_scale_arcsec = plate_scale * 3600.0
if aimpt_shift is None:
aimpt_shift = np.zeros(2)
aimpt_shift = ensure_numpy_array(aimpt_shift).astype('float64')
aimpt_shift /= plate_scale_arcsec
events = defaultdict(list)
if not instrument_spec["dither"]:
dither_on = False
else:
dither_on = not no_dither
if dither_params is None:
dither_params = [8.0, 8.0, 1000.0, 707.0]
dither_dict = {
"x_amp": dither_params[0],
"y_amp": dither_params[1],
"x_period": dither_params[2],
"y_period": dither_params[3],
"dither_on": dither_on,
"plate_scale": instrument_spec["fov"] / nx * 60.0
}
event_params = {
"exposure_time": exp_time,
"fov": instrument_spec["fov"],
"num_pixels": nx,
"pix_center": np.array([0.5 * (2 * nx + 1)] * 2),
"channel_type": rmf.header["CHANTYPE"],
"sky_center": sky_center,
"dither_params": dither_dict,
"plate_scale": plate_scale,
"chan_lim": [rmf.cmin, rmf.cmax],
"rmf": rmf_file,
"arf": arf_file,
"telescope": rmf.header["TELESCOP"],
"instrument": instrument_spec['name'],
"mission": rmf.header.get("MISSION", ""),
"nchan": rmf.n_ch,
"roll_angle": roll_angle,
"aimpt_coords": instrument_spec["aimpt_coords"],
"aimpt_shift": aimpt_shift
}
if "chips" not in event_params:
event_params["chips"] = instrument_spec["chips"]
if foreground:
mylog.info("Adding in astrophysical foreground.")
bkg_events = make_foreground(event_params, arf, rmf, prng=prng)
for key in bkg_events:
events[key] = np.concatenate([events[key], bkg_events[key]])
if instr_bkgnd and instrument_spec["bkgnd"] is not None:
mylog.info("Adding in instrumental background.")
bkg_events = make_instrument_background(instrument_spec,
event_params,
rmf,
prng=prng)
for key in bkg_events:
events[key] = np.concatenate([events[key], bkg_events[key]])
return events, event_params
def generate_events(source,
exp_time,
instrument,
sky_center,
no_dither=False,
dither_params=None,
roll_angle=0.0,
subpixel_res=False,
aimpt_shift=None,
prng=None):
"""
Take unconvolved events and convolve them with instrumental responses. This
function does the following:
1. Determines which events are observed using the ARF
2. Pixelizes the events, applying PSF effects and dithering
3. Determines energy channels using the RMF
This function is not meant to be called by the end-user but is used by
the :func:`~soxs.instrument.instrument_simulator` function.
Parameters
----------
input_events : string, dict, or None
The unconvolved events to be used as input. Can be one of the
following:
1. The name of a SIMPUT catalog file.
2. A Python dictionary containing the following items:
"ra": A NumPy array of right ascension values in degrees.
"dec": A NumPy array of declination values in degrees.
"energy": A NumPy array of energy values in keV.
"flux": The flux of the entire source, in units of erg/cm**2/s.
out_file : string
The name of the event file to be written.
exp_time : float, (value, unit) tuple, or :class:`~astropy.units.Quantity`
The exposure time to use, in seconds.
instrument : string
The name of the instrument to use, which picks an instrument
specification from the instrument registry.
sky_center : array, tuple, or list
The center RA, Dec coordinates of the observation, in degrees.
no_dither : boolean, optional
If True, turn off dithering entirely. Default: False
dither_params : array-like of floats, optional
The parameters to use to control the size and period of the dither
pattern. The first two numbers are the dither amplitude in x and y
detector coordinates in arcseconds, and the second two numbers are
the dither period in x and y detector coordinates in seconds.
Default: [8.0, 8.0, 1000.0, 707.0].
roll_angle : float, (value, unit) tuple, or :class:`~astropy.units.Quantity`, optional
The roll angle of the observation in degrees. Default: 0.0
subpixel_res : boolean, optional
If True, event positions are not randomized within the pixels
within which they are detected. Default: False
aimpt_shift : array-like, optional
A two-float array-like object which shifts the aimpoint on the
detector from the nominal position. Units are in arcseconds.
Default: None, which results in no shift from the nominal aimpoint.
prng : :class:`~numpy.random.RandomState` object, integer, or None
A pseudo-random number generator. Typically will only
be specified if you have a reason to generate the same
set of random numbers, such as for a test. Default is None,
which sets the seed based on the system time.
"""
exp_time = parse_value(exp_time, "s")
roll_angle = parse_value(roll_angle, "deg")
prng = parse_prng(prng)
if source is None:
source_list = []
elif isinstance(source, dict):
parameters = {}
for key in ["flux", "emin", "emax", "src_names"]:
parameters[key] = source[key]
source_list = []
for i in range(len(parameters["flux"])):
phlist = SimputPhotonList(source["ra"][i], source["dec"][i],
source["energy"][i],
parameters['flux'][i],
parameters['src_names'][i])
source_list.append(phlist)
elif isinstance(source, str):
# Assume this is a SIMPUT catalog
source_list, parameters = read_simput_catalog(source)
try:
instrument_spec = instrument_registry[instrument]
except KeyError:
raise KeyError(
f"Instrument {instrument} is not in the instrument registry!")
if not instrument_spec["imaging"]:
raise RuntimeError(f"Instrument '{instrument_spec['name']}' is not "
f"designed for imaging observations!")
arf_file = get_data_file(instrument_spec["arf"])
rmf_file = get_data_file(instrument_spec["rmf"])
arf = AuxiliaryResponseFile(arf_file)
rmf = RedistributionMatrixFile(rmf_file)
nx = instrument_spec["num_pixels"]
plate_scale = instrument_spec["fov"] / nx / 60. # arcmin to deg
plate_scale_arcsec = plate_scale * 3600.0
if aimpt_shift is None:
aimpt_shift = np.zeros(2)
aimpt_shift = ensure_numpy_array(aimpt_shift).astype('float64')
aimpt_shift /= plate_scale_arcsec
if not instrument_spec["dither"]:
dither_on = False
else:
dither_on = not no_dither
if dither_params is None:
dither_params = [8.0, 8.0, 1000.0, 707.0]
dither_dict = {
"x_amp": dither_params[0],
"y_amp": dither_params[1],
"x_period": dither_params[2],
"y_period": dither_params[3],
"dither_on": dither_on,
"plate_scale": plate_scale_arcsec
}
event_params = {
"exposure_time": exp_time,
"arf": arf.filename,
"sky_center": sky_center,
"pix_center": np.array([0.5 * (2 * nx + 1)] * 2),
"num_pixels": nx,
"plate_scale": plate_scale,
"rmf": rmf.filename,
"channel_type": rmf.chan_type,
"telescope": rmf.header["TELESCOP"],
"instrument": instrument_spec['name'],
"mission": rmf.header.get("MISSION", ""),
"nchan": rmf.n_ch,
"roll_angle": roll_angle,
"fov": instrument_spec["fov"],
"chan_lim": [rmf.cmin, rmf.cmax],
"chips": instrument_spec["chips"],
"dither_params": dither_dict,
"aimpt_coords": instrument_spec["aimpt_coords"],
"aimpt_shift": aimpt_shift
}
# Set up WCS
w = pywcs.WCS(naxis=2)
w.wcs.crval = event_params["sky_center"]
w.wcs.crpix = event_params["pix_center"]
w.wcs.cdelt = [-plate_scale, plate_scale]
w.wcs.ctype = ["RA---TAN", "DEC--TAN"]
w.wcs.cunit = ["deg"] * 2
# Determine rotation matrix
rot_mat = get_rot_mat(roll_angle)
# Set up PSF
psf_type = instrument_spec["psf"][0]
psf_class = psf_model_registry[psf_type]
psf = psf_class(instrument_spec, prng=prng)
all_events = defaultdict(list)
for i, src in enumerate(source_list):
mylog.info(
f"Detecting events from source {parameters['src_names'][i]}")
# Step 1: Use ARF to determine which photons are observed
mylog.info(f"Applying energy-dependent effective area from "
f"{os.path.split(arf.filename)[-1]}.")
refband = [parameters["emin"][i], parameters["emax"][i]]
if src.src_type == "phlist":
events = arf.detect_events_phlist(src.events.copy(),
exp_time,
parameters["flux"][i],
refband,
prng=prng)
elif src.src_type.endswith("spectrum"):
events = arf.detect_events_spec(src, exp_time, refband, prng=prng)
n_evt = events["energy"].size
if n_evt == 0:
mylog.warning("No events were observed for this source!!!")
else:
# Step 2: Assign pixel coordinates to events. Apply dithering and
# PSF. Clip events that don't fall within the detection region.
mylog.info("Pixeling events.")
# Convert RA, Dec to pixel coordinates
xpix, ypix = w.wcs_world2pix(events["ra"], events["dec"], 1)
xpix -= event_params["pix_center"][0]
ypix -= event_params["pix_center"][1]
events.pop("ra")
events.pop("dec")
n_evt = xpix.size
# Rotate physical coordinates to detector coordinates
det = np.dot(rot_mat, np.array([xpix, ypix]))
detx = det[0, :] + event_params["aimpt_coords"][0] + aimpt_shift[0]
dety = det[1, :] + event_params["aimpt_coords"][1] + aimpt_shift[1]
# Add times to events
events['time'] = prng.uniform(size=n_evt,
low=0.0,
high=event_params["exposure_time"])
# Apply dithering
x_offset, y_offset = perform_dither(events["time"], dither_dict)
detx -= x_offset
dety -= y_offset
# PSF scattering of detector coordinates
mylog.info(f"Scattering events with a {psf}-based PSF.")
detx, dety = psf.scatter(detx, dety, events["energy"])
# Convert detector coordinates to chip coordinates.
# Throw out events that don't fall on any chip.
cx = np.trunc(detx) + 0.5 * np.sign(detx)
cy = np.trunc(dety) + 0.5 * np.sign(dety)
events["chip_id"] = -np.ones(n_evt, dtype='int')
for i, chip in enumerate(event_params["chips"]):
rtype = chip[0]
args = chip[1:]
r, _ = create_region(rtype, args, 0.0, 0.0)
inside = r.contains(PixCoord(cx, cy))
events["chip_id"][inside] = i
keep = events["chip_id"] > -1
mylog.info(f"{n_evt-keep.sum()} events were rejected because "
f"they do not fall on any CCD.")
n_evt = keep.sum()
if n_evt == 0:
mylog.warning("No events are within the field "
"of view for this source!!!")
else:
# Keep only those events which fall on a chip
for key in events:
events[key] = events[key][keep]
# Convert chip coordinates back to detector coordinates,
# unless the user has specified that they want subpixel
# resolution
if subpixel_res:
events["detx"] = detx[keep]
events["dety"] = dety[keep]
else:
events["detx"] = cx[keep] + \
prng.uniform(low=-0.5, high=0.5, size=n_evt)
events["dety"] = cy[keep] + \
prng.uniform(low=-0.5, high=0.5, size=n_evt)
# Convert detector coordinates back to pixel coordinates by
# adding the dither offsets back in and applying the rotation
# matrix again
det = np.array([
events["detx"] + x_offset[keep] -
event_params["aimpt_coords"][0] - aimpt_shift[0],
events["dety"] + y_offset[keep] -
event_params["aimpt_coords"][1] - aimpt_shift[1]
])
pix = np.dot(rot_mat.T, det)
events["xpix"] = pix[0, :] + event_params['pix_center'][0]
events["ypix"] = pix[1, :] + event_params['pix_center'][1]
if n_evt > 0:
for key in events:
all_events[key] = np.concatenate(
[all_events[key], events[key]])
if len(all_events["energy"]) == 0:
mylog.warning("No events from any of the sources in "
"the catalog were detected!")
for key in [
"xpix", "ypix", "detx", "dety", "time", "chip_id",
event_params["channel_type"]
]:
all_events[key] = np.array([])
else:
# Step 4: Scatter energies with RMF
mylog.info(f"Scattering energies with "
f"RMF {os.path.split(rmf.filename)[-1]}.")
all_events = rmf.scatter_energies(all_events, prng=prng)
return all_events, event_params
def __init__(self, emin, emax, nbins, var_elem=None, apec_root=None,
apec_vers=None, broadening=True, nolines=False,
abund_table=None, nei=False):
if apec_vers is None:
apec_vers = default_apec_vers
mylog.info(f"Using APEC version {apec_vers}.")
if nei and var_elem is None:
raise RuntimeError("For NEI spectra, you must specify which elements "
"you want to vary using the 'var_elem' argument!")
self.nei = nei
emin = parse_value(emin, "keV")
emax = parse_value(emax, 'keV')
self.emin = emin
self.emax = emax
self.nbins = nbins
self.ebins = np.linspace(self.emin, self.emax, nbins+1)
self.de = np.diff(self.ebins)
self.emid = 0.5*(self.ebins[1:]+self.ebins[:-1])
if nei:
neistr = "_nei"
ftype = "comp"
else:
neistr = ""
ftype = "coco"
cocofile = f"apec_v{apec_vers}{neistr}_{ftype}.fits"
linefile = f"apec_v{apec_vers}{neistr}_line.fits"
if apec_root is None:
self.cocofile = get_data_file(cocofile)
self.linefile = get_data_file(linefile)
else:
self.cocofile = os.path.join(apec_root, cocofile)
self.linefile = os.path.join(apec_root, linefile)
if not os.path.exists(self.cocofile) or not os.path.exists(self.linefile):
raise IOError(f"Cannot find the APEC files!\n {self.cocofile}\n, "
f"{self.linefile}")
mylog.info(f"Using {cocofile} for generating spectral lines.")
mylog.info(f"Using {linefile} for generating the continuum.")
self.nolines = nolines
self.wvbins = hc/self.ebins[::-1]
self.broadening = broadening
self.line_handle = pyfits.open(self.linefile)
self.coco_handle = pyfits.open(self.cocofile)
self.nT = self.line_handle[1].data.shape[0]
self.Tvals = self.line_handle[1].data.field("kT")
self.dTvals = np.diff(self.Tvals)
self.minlam = self.wvbins.min()
self.maxlam = self.wvbins.max()
self.var_elem_names = []
self.var_ion_names = []
if var_elem is None:
self.var_elem = np.empty((0, 1), dtype='int')
else:
self.var_elem = []
if len(var_elem) != len(set(var_elem)):
raise RuntimeError("Duplicates were found in the \"var_elem\" list! %s" % var_elem)
for elem in var_elem:
if "^" in elem:
if not self.nei:
raise RuntimeError("Cannot use different ionization states with a "
"CIE plasma!")
el = elem.split("^")
e = el[0]
ion = int(el[1])
else:
if self.nei:
raise RuntimeError("Variable elements must include the ionization "
"state for NEI plasmas!")
e = elem
ion = 0
self.var_elem.append([elem_names.index(e), ion])
self.var_elem.sort(key=lambda x: (x[0], x[1]))
self.var_elem = np.array(self.var_elem, dtype='int')
self.var_elem_names = [elem_names[e[0]] for e in self.var_elem]
self.var_ion_names = ["%s^%d" % (elem_names[e[0]], e[1]) for e in self.var_elem]
self.num_var_elem = len(self.var_elem)
if self.nei:
self.cosmic_elem = [elem for elem in [1, 2]
if elem not in self.var_elem[:, 0]]
self.metal_elem = []
else:
self.cosmic_elem = [elem for elem in cosmic_elem
if elem not in self.var_elem[:,0]]
self.metal_elem = [elem for elem in metal_elem
if elem not in self.var_elem[:,0]]
if abund_table is None:
abund_table = soxs_cfg.get("soxs", "abund_table")
if not isinstance(abund_table, str):
if len(abund_table) != 30:
raise RuntimeError("User-supplied abundance tables "
"must be 30 elements long!")
self.atable = np.concatenate([[0.0], np.array(abund_table)])
else:
self.atable = abund_tables[abund_table].copy()
self._atable = self.atable.copy()
self._atable[1:] /= abund_tables["angr"][1:]