PyNOT-redux 2.1.3__tar.gz → 2.2__tar.gz

{pynot_redux-2.1.3 → pynot_redux-2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyNOT-redux
-Version: 2.1.3
+Version: 2.2
 Summary: Data Reduction Pipeline for NOT/ALFOSC
 Home-page: https://github.com/jkrogager/PyNOT
 Author: Jens-Kristian Krogager

{pynot_redux-2.1.3 → pynot_redux-2.2}/PyNOT_redux.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyNOT-redux
-Version: 2.1.3
+Version: 2.2
 Summary: Data Reduction Pipeline for NOT/ALFOSC
 Home-page: https://github.com/jkrogager/PyNOT
 Author: Jens-Kristian Krogager

{pynot_redux-2.1.3 → pynot_redux-2.2}/PyNOT_redux.egg-info/SOURCES.txt

@@ -28,7 +28,6 @@ pynot/fitsio.py
 pynot/functions.py
 pynot/identify_gui.py
 pynot/images.py
-pynot/india.py
 pynot/insconfig.py
 pynot/logging.py
 pynot/main.py
@@ -92,4 +91,24 @@ pynot/data/obs.py
 pynot/data/organizer.py
 pynot/data/help/welcome_msg_extract.html
 pynot/data/help/welcome_msg_identify.html
-pynot/data/help/welcome_msg_response.html
+pynot/data/help/welcome_msg_response.html
+pynot/viewer/__init__.py
+pynot/viewer/containers.py
+pynot/viewer/default_linelists.json
+pynot/viewer/dust.py
+pynot/viewer/linelists.py
+pynot/viewer/messages.py
+pynot/viewer/models.py
+pynot/viewer/notes.py
+pynot/viewer/spectrum.py
+pynot/viewer/tablemodels.py
+pynot/viewer/targets.py
+pynot/viewer/viewer.py
+pynot/viewer/templates/Assef_2010.dat
+pynot/viewer/templates/FeII.Vestergaard.dat
+pynot/viewer/templates/FeIII.Vestergaard.dat
+pynot/viewer/templates/Hernan_Caballero_quasar_NIR_template.txt
+pynot/viewer/templates/QSO_selsing.dat
+pynot/viewer/templates/QSO_template.Krawczyk_all.dat
+pynot/viewer/templates/QSO_template.Krawczyk_high.dat
+pynot/viewer/templates/QSO_template.Krawczyk_low.dat

pynot_redux-2.2/pynot/VERSION

@@ -0,0 +1 @@
+2.2

{pynot_redux-2.1.3 → pynot_redux-2.2}/pynot/extract_gui.py

@@ -51,27 +51,6 @@ def run_gui(input_fname, output_fname, app=None, **ext_kwargs):
     del gui
 
 
-# def save_ascii_spectrum(fname, wl, flux, err, hdr, bg=None):
-#     """Write spectrum to an ascii text file with header saved to separate text file."""
-#     if bg is not None:
-#         data_table = np.column_stack([wl, flux, err, bg])
-#         fmt = "%12.4f  % .3e  %.3e  %.3e"
-#         col_names = "# Wavelength       Flux        Error      Sky"
-#     else:
-#         data_table = np.column_stack([wl, flux, err])
-#         fmt = "%12.4f  % .3e  %.3e"
-#         col_names = "# Wavelength       Flux        Error"
-#
-#     basename, ext = os.path.splitext(fname)
-#     header_fname = basename + '_hdr.txt'
-#
-#     with open(fname, 'w') as output:
-#         output.write(col_names + "\n")
-#         np.savetxt(output, data_table, fmt=fmt)
-#     hdr.tofile(header_fname, sep='\n', endcard=False, padding=False, overwrite=True)
-#     return True, "File saved successfully"
-
-
 def get_FWHM(y, x=None):
     """
     Measure the FWHM of the profile given as `y`.
@@ -1669,7 +1648,10 @@ class ExtractGUI(QtWidgets.QMainWindow):
     def update_xmask_in_points(self):
         # Clear old shapes:
         for item in self.xmask:
-            item.remove()
+            try:
+                item.remove()
+            except NotImplementedError:
+                pass
         self.xmask = list()
         xmin, xmax, ymin, ymax = self.get_limits()
         if xmin > 0 or xmax < self.image2d.data.shape[1]+1:

{pynot_redux-2.1.3 → pynot_redux-2.2}/pynot/fitsio.py

@@ -3,6 +3,8 @@ __author__ = "Jens-Kristian Krogager"
 
 import warnings
 from astropy.io import fits
+from astropy import units as u
+from astropy.table import QTable
 import numpy as np
 import os
 
@@ -129,8 +131,8 @@ def get_wavelength_from_header(hdr):
 
 # -- These names are used to define proper column names for Wavelength, Flux and Error:
 wavelength_column_names = ['wl', 'lam', 'lambda', 'loglam', 'wave', 'wavelength', 'awav']
-flux_column_names = ['data', 'spec', 'flux', 'flam', 'fnu', 'flux_density']
-error_column_names = ['err', 'sig', 'error', 'ivar', 'sigma', 'var', 'err_flux']
+flux_column_names = ['data', 'spec', 'flux', 'flam', 'fnu', 'flux_density', 'reduced_flux']
+error_column_names = ['err', 'sig', 'error', 'ivar', 'sigma', 'var', 'err_flux', 'flux_ivar', 'flux_var', 'flux_err']
 mask_column_names = ['mask', 'qual', 'dq', 'qc']
 
 # -- These names are used to define proper ImageHDU names for Flux and Error:
@@ -139,7 +141,7 @@ error_HDU_names = ['ERR', 'ERRS', 'SIG', 'SIGMA', 'ERROR', 'ERRORS', 'IVAR', 'VA
 mask_HDU_names = ['MASK', 'QUAL', 'QC', 'DQ']
 
 
-def get_spectrum_fits_table(tbdata):
+def get_spectrum_fits_table(table_hdu):
     """
     Scan the TableData for columns containing wavelength, flux, error and mask.
     All arrays of {wavelength, flux and error} must be present.
@@ -163,7 +165,12 @@ def get_spectrum_fits_table(tbdata):
         Numpy boolean array of pixel mask. `True` if the pixel is 'good',
         `False` if the pixel is bad and should not be used.
     """
-    table_names = [name.lower() for name in tbdata.names]
+    tbdata = QTable.read(table_hdu)
+    # table_names = [name.lower() for name in tbdata.names]
+    for colname in tbdata.colnames:
+        tbdata.rename_column(colname, colname.lower())
+    table_names = tbdata.colnames
+
     wl_in_table = False
     for colname in wavelength_column_names:
         if colname in table_names:
@@ -185,9 +192,10 @@ def get_spectrum_fits_table(tbdata):
     for colname in error_column_names:
         if colname in table_names:
             error_in_table = True
-            if colname == 'ivar':
-                error = 1./np.sqrt(tbdata[colname])
-            elif colname == 'var':
+            if 'ivar' in colname.lower():
+                with np.errstate(divide='ignore', invalid='ignore'):
+                    error = 1 / np.sqrt(tbdata[colname])
+            elif 'var' in colname.lower():
                 error = np.sqrt(tbdata[colname])
             else:
                 error = tbdata[colname]
@@ -365,17 +373,19 @@ def load_fits_spectrum(fname, ext=None, iraf_obj=None):
         else:
             is_fits_table = isinstance(HDUlist[1], fits.BinTableHDU) or isinstance(HDUlist[1], fits.TableHDU)
             if is_fits_table:
-                if ext:
-                    tbdata = HDUlist[ext].data
-                    data_hdr = HDUlist[ext].header
-                else:
-                    tbdata = HDUlist[1].data
-                    data_hdr = HDUlist[1].header
+                if not ext:
+                    ext = 1
+
+                table_hdu = HDUlist[ext]
+                tbdata = table_hdu.data
+                data_hdr = table_hdu.header
 
                 has_multi_extensions = len(HDUlist) > 2
                 if has_multi_extensions and (ext is None):
                     msg = "[WARNING] - More than one data extension detected in the file"
-                wavelength, data, error, mask = get_spectrum_fits_table(tbdata)
+                wavelength, data, error, mask = get_spectrum_fits_table(table_hdu)
+                if 'CUNIT1' in data_hdr:
+                    wavelength *= u.Unit(data_hdr['CUNIT1'])
                 return wavelength, data, error, mask, data_hdr, msg
 
             elif len(HDUlist) == 2:
@@ -538,3 +548,11 @@ def create_error_image(base_fname, overwrite=False):
             hdu.append(ext)
     output_msg = "          - Successfully created an error image"
     return output_msg
+
+
+def detect_4most_MEC(fname):
+    prim = fits.getheader(fname)
+    is_qmost = prim.get('INSTRUME', 'NONE').strip() == 'QMOST'
+    hdr = fits.getheader(fname, 1)
+    has_mec_extname = hdr.get('EXTNAME') in ['OBMETATAB', 'SPECTAB', 'FIBMETATAB']
+    return is_qmost & has_mec_extname

{pynot_redux-2.1.3 → pynot_redux-2.2}/pynot/main.py

@@ -372,6 +372,10 @@ def main(inspect=False):
                               help="Set this option for 2D spectra of extended sources to turn off automatic localization")
     parser_scomb.add_argument("--mef", action="store_false",
                               help="Set this option to save output as a multiextension FITS file instead of a FITS table.")
+    parser_scomb.add_argument("--imin", type=int, default=0,
+                              help="Slice each input spectrum along dispersion axis [imin:imax], only for 1D spectra")
+    parser_scomb.add_argument("--imax", type=int, default=None,
+                              help="Slice each input spectrum along dispersion axis [imin:imax], only for 1D spectra")
 
 
     # -- extract :: Extraction of 1D spectrum from 2D
@@ -389,6 +393,31 @@ def main(inspect=False):
     set_default_pars(parser_ext, section='extract', default_type=int,
                      ignore_pars=['interactive'])
 
+    # -- view :: View 1D spectra in an interactive window
+    parser_view = tasks.add_parser('view', formatter_class=set_help_width(31),
+                                   help="Display and interact with 1D spectra")
+    parser_view.add_argument("files", type=str, nargs='*',
+                             help="Filenames of spectral data to load. Each file is loaded as one target")
+    parser_view.add_argument("-t", "--table", type=str,
+                             help="Filename of association table. Each line in the file gives a comma-separated list of filenames. "
+                                  "All files in one line are loaded as a single target.")
+    parser_view.add_argument("-c", "--container", action="store_true",
+                             help="Load the file(s) as a FITS container "
+                                  "(such as 4MOST MEC, SDSS bricks, or a large collection of files)")
+    parser_view.add_argument("-l", "--list", type=str, default='',
+                             help="File list. Name of a text file where each line is the path of one spectrum."
+                                  "(Useful for loading many files in `container` mode with `-c`)")
+    # redshift_table=None, z_col=None, name_col=None, cls_col=None
+    parser_view.add_argument("-z", type=str, default='',
+                             help="Filename of redshift catalog with a name, redshift and spectral type column. "
+                                  "Set column names using `--ncol`, `--zcol`, `--tcol`")
+    parser_view.add_argument("--ncol", type=str, default='',
+                             help="Name of the target name column in `-z REDSHIFT_TABLE`")
+    parser_view.add_argument("--zcol", type=str, default='',
+                             help="Name of the redshift column in `-z REDSHIFT_TABLE`")
+    parser_view.add_argument("--tcol", type=str, default='',
+                             help="Name of the spectral type column in `-z REDSHIFT_TABLE`")
+
 
     # Spectral Redux:
     parser_redux = tasks.add_parser('spex', formatter_class=set_help_width(30),
@@ -770,7 +799,7 @@ def main(inspect=False):
 
         if data_is_1d:
             out_args = combine_1d(filelist, output=args.output, method=args.method,
-                                  scale=args.scale, table_output=args.mef)
+                                  scale=args.scale, table_output=args.mef, imin=args.imin, imax=args.imax)
         else:
             out_args = combine_2d(filelist, output=args.output, method=args.method, trim=args.trim,
                                   scale=args.scale, extended=args.extended, dispaxis=args.axis)
@@ -796,6 +825,28 @@ def main(inspect=False):
             gui.show()
             app.exit(app.exec_())
 
+    elif task == 'view':
+        from PyQt5 import QtWidgets
+        from pynot.viewer.viewer import MainWindow
+
+        app = QtWidgets.QApplication(sys.argv)
+        screenSize = app.primaryScreen().size()
+        ratio = 0.85
+        if args.list:
+            with open(args.list) as l:
+                input_files = [line.strip() for line in l.readlines()]
+        else:
+            input_files = args.files
+        main = MainWindow(input_files,
+                          assn_table=args.table,
+                          container_mode=args.container,
+                          width=ratio*screenSize.width(),
+                          height=ratio*screenSize.height(),
+                          redshift_table=args.z,
+                          z_col=args.zcol, name_col=args.ncol, cls_col=args.tcol,
+                          )
+        main.show()
+        app.exit(app.exec_())
 
     # -- Imaging tasks:
     elif task == 'phot':

{pynot_redux-2.1.3 → pynot_redux-2.2}/pynot/scombine.py

@@ -353,7 +353,7 @@ def combine_2d(files, output=None, method='mean', scale=False, extended=False, d
 
 
 
-def combine_1d(files, output=None, method='mean', scale=False, table_output=True):
+def combine_1d(files, output=None, method='mean', scale=False, table_output=True, imin=0, imax=None):
     """Combine a list of 1d-spectra using either median or mean combination.
     For median combination, only the overlapping parts of the spectra will be
     combined. The mean combination uses a weighted average over the entire
@@ -376,6 +376,12 @@ def combine_1d(files, output=None, method='mean', scale=False, table_output=True
 
     table_output : bool   [default=True]
         Use FITS table for the output format? Otherwise use a MultiExtension Fits File
+
+    imin : int
+        Apply slicing to each spectrum from imin to imax.
+
+    imax : int | None
+        Apply slicing to each spectrum from imin to imax.
     """
 
     wl_all = list()
@@ -385,6 +391,7 @@ def combine_1d(files, output=None, method='mean', scale=False, table_output=True
     size_all = list()
     scales = list()
 
+    cut = slice(imin, imax)
     msg = list()
     for fname in files:
         if fname.endswith('.fits') or fname.endswith('.fit'):
@@ -401,11 +408,11 @@ def combine_1d(files, output=None, method='mean', scale=False, table_output=True
                 msg.append(load_msg)
             msg.append("          - Loaded ASCII spectrum: %s" % fname)
 
-        wl_all.append(wl)
-        flux_all.append(flux)
-        err_all.append(err)
-        mask_all.append(mask)
-        size_all.append(len(wl))
+        wl_all.append(wl[cut])
+        flux_all.append(flux[cut])
+        err_all.append(err[cut])
+        mask_all.append(mask[cut])
+        size_all.append(len(wl[cut]))
         if scale:
             nonzero = flux.nonzero()[0]
             idx_0 = min(nonzero) + len(nonzero)/2

pynot_redux-2.2/pynot/viewer/containers.py

@@ -0,0 +1,82 @@
+import os
+import numpy as np
+from astropy.io import fits
+import astropy.units as u
+
+from pynot.viewer.spectrum import Spectrum
+from pynot.viewer.targets import Target
+
+    
+class GenericFileContainer:
+    def __init__(self, filelist):
+        self.filelist = filelist
+        self.view = []
+        for f in self.filelist:
+            for key in ['OBJ_NME', 'OBJ_UID', 'OBJECT']:
+                try:
+                    name = fits.getval(f, key)
+                except Exception:
+                    pass
+            else:
+                name = f
+            self.view.append(os.path.basename(name))
+
+    def __len__(self):
+        return len(self.filelist)
+
+    def __getitem__(self, index):
+        target = Target(name=self.view[index])
+        spec = Spectrum.read(self.filelist[index])
+        target.add_spectrum(spec)
+        return target
+
+
+class QMEC:
+    def __init__(self, filename, view):
+        self.filename = filename
+        self.view = view
+
+    def __len__(self):
+        return len(self.view)
+
+    def __getitem__(self, index):
+        return self.get_target(index)
+
+    def get_target(self, index):
+        target = Target(name=self.view[index])
+        with fits.open(self.filename) as hdu:
+            data = hdu['SPECTAB'].data[index]
+            hdr = hdu['SPECTAB'].header
+
+            flux = data['FLUX'].flatten()
+            try:
+                ivar = data['FLUX_IVAR'].flatten()
+                with np.errstate(divide='ignore', invalid='ignore'):
+                    error = 1 / np.sqrt(ivar)
+            except Exception:
+                error = data['ERR_FLUX'].flatten()
+            npix = len(flux)
+            wavelength = np.arange(npix) * hdr['1CDLT1'] + hdr['1CRVL1']
+            wavelength *= u.Unit(hdr['1CUNI1'])
+            flux_unit = u.Unit(hdr['TUNIT1'])
+            spectrum = Spectrum(wavelength=wavelength,
+                                flux=flux*flux_unit,
+                                error=error*flux_unit,
+                                name=f"[{index}]",
+                                filename=self.filename+f"[{index}]",
+                                meta=dict(hdr))
+            target.add_spectrum(spectrum)
+            return target
+
+
+    @staticmethod
+    def read(filename):
+        with fits.open(filename) as hdu:
+            names_columns = ['OBJECT', 'OBJ_NME', 'OBJ_UID']
+            for colname in names_columns:
+                if colname in hdu['FIBMETATAB'].data.names:
+                    view = hdu['FIBMETATAB'].data[colname]
+                    break
+            else:
+                view = [f"Object #{num}" for num in range(len(hdu['FIBMETATAB'].data))]
+        return QMEC(filename, view)

pynot_redux-2.2/pynot/viewer/default_linelists.json

@@ -0,0 +1,103 @@
+{
+    "": [],
+
+    "Common Lines": [
+        ["Ly-α", 1215.67, true],
+        ["N V 1240", 1240.14, true],
+        ["Si IV / O IV]", 1399, true],
+        ["C IV", 1549.06, true],
+        ["C III]", 1908.73, true],
+        ["Mg II", 2798.75, true],
+        ["H-α", 6564.61, true],
+        ["H-β", 4862.68, true],
+        ["H-γ", 4341.68, true],
+        ["H-δ", 4102.89, true],
+        ["H-ε", 3971.20, true],
+        ["[O II] 3727", 3727.09, true],
+        ["[O II] 3729", 3729.88, true],
+        ["[Ne III] 3870", 3869.86, true],
+        ["[O III] 4364", 4364.44, true],
+        ["[O III] 4960", 4960.30, true],
+        ["[O III] 5008", 5008.24, true],
+        ["[N II] 6550", 6549.86, true],
+        ["[N II] 6585", 6585.27, true],
+        ["[S II] 6718", 6718.29, true],
+        ["[S II] 6733", 6732.67, true],
+        ["Ca II K", 3934.78, true],
+        ["Ca II H", 3969.59, true],
+        ["Mg I b", 5176.70, true],
+        ["Na I D", 5891.58, true]
+    ],
+
+    "Quasar Lines": [
+        ["O VI", 1033.83, true],
+        ["Ly-α", 1215.67, true],
+        ["Ly-β", 1025.72, true],
+        ["N V 1240", 1240.14, true],
+        ["O I", 1304.35, true],
+        ["Si IV / O IV]", 1399, true],
+        ["C IV", 1549.06, true],
+        ["He II 1640", 1640.42, true],
+        ["C III]", 1908.73, true],
+        ["Mg II", 2798.75, true],
+        ["H-α", 6564.61, true],
+        ["H-β", 4862.68, true],
+        ["H-γ", 4341.68, true],
+        ["H-δ", 4102.89, true],
+        ["H-ε", 3971.20, true],
+        ["[O II] 3727", 3727.09, true],
+        ["[O II] 3729", 3729.88, true],
+        ["[Ne III] 3870", 3869.86, true],
+        ["[O III] 4960", 4960.30, true],
+        ["[O III] 5008", 5008.24, true],
+        ["[N II] 6550", 6549.86, true],
+        ["[N II] 6585", 6585.27, true],
+        ["[S II] 6718", 6718.29, true],
+        ["[S II] 6733", 6732.67, true]
+    ],
+
+    "ISM Absorption": [
+        ["CI_1656", 1656.93, true],
+        ["CI_1560", 1560.31, true],
+        ["CI_1328", 1328.83, true],
+        ["SII_1250", 1250.578, true],
+        ["SII_1253", 1253.805, true],
+        ["SII_1259", 1259.518, true],
+        ["OI_1302", 1302.17, true],
+        ["FeII_1608", 1608.45, true],
+        ["SiII_1260", 1260.42, true],
+        ["SiII_1304", 1304.37, true],
+        ["CII_1334", 1334.53, true],
+        ["SiIV_1393", 1393.76, true],
+        ["SiIV_1402", 1402.77, true],
+        ["SiII_1526", 1526.71, true],
+        ["CIV_1548", 1548.19, true],
+        ["CIV_1550", 1550.77, true],
+        ["FeII_1608", 1608.45, true],
+        ["AlII_1670", 1670.79, true],
+        ["SiII_1808", 1808.01, true],
+        ["AlIII_1854", 1854.72, true],
+        ["AlIII_1862", 1862.79, true],
+        ["TiII_1910", 1910.61, true],
+        ["TiII_3073", 3073.86, true],
+        ["TiII_3242", 3242.91, true],
+        ["TiII_3384", 3384.73, true],
+        ["FeII_2344", 2344.21, true],
+        ["FeII_2374", 2374.46, true],
+        ["FeII_2382", 2382.76, true],
+        ["MnII_2576", 2576.87, true],
+        ["FeII_2586", 2586.65, true],
+        ["MnII_2594", 2594.50, true],
+        ["FeII_2600", 2600.17, true],
+        ["MnII_2606", 2606.46, true],
+        ["MgII_2796", 2796.35, true],
+        ["MgII_2803", 2803.53, true],
+        ["MgI_2852", 2852.96, true],
+        ["NaI_5891", 5891.58, true],
+        ["NaI_5897", 5897.56, true],
+        ["ZnII_2026", 2026.14, true],
+        ["ZnII_2062", 2062.66, true],
+        ["CaII_3934", 3934.77, true],
+        ["CaII_3969", 3969.59, true]
+    ]
+}

pynot_redux-2.2/pynot/viewer/dust.py

@@ -0,0 +1,149 @@
+
+import numpy as np
+from scipy.interpolate import UnivariateSpline as spline
+
+parameters = {
+    'SMC': {'c1': -4.959, 'c2': 2.264, 'c3': 0.389, 'c4': 0.0, 'x0': 4.579, 'gamma': 0.934, 'Rv': 2.74},
+    'LMC': {'c1': -0.89, 'c2': 0.998, 'c3': 2.719, 'c4': 0.0, 'x0': 4.579, 'gamma': 0.934, 'Rv': 3.41},
+    }
+
+
+def MW_reddening(wl, Rv=3.1):
+    """
+    Average Galactic reddening law parametrized by Fitzpatrick & Massa (2007).
+
+    INPUT
+    wl: wavelength in Angstrom
+
+    Rv: if not given, the average value of 3.1 is assumed.
+
+    Returns A(l)/A(V) evaluated at the input wavelengths.
+    """
+
+    if isinstance(wl, float):
+        wl = np.array([wl])
+        convert2float = True
+    else:
+        convert2float = False
+
+    k_in = 1./(wl*1.e-4)
+
+    c4 = 0.319
+    c3 = 2.991
+    c2 = 5.0/Rv - 0.85
+    c1 = 2.09 - 2.84*c2
+    x0 = 4.592
+    gamma = 0.922
+    pars_avg = {'c1': c1, 'c2': c2, 'c3': c3, 'c4': c4,
+                'x0': x0, 'gamma': gamma, 'Rv': Rv}
+
+    ksi = FM2007(k_in, **pars_avg)
+
+    if convert2float:
+        ksi = float(ksi)
+    return ksi
+
+
+def FM2007(wl, c1, c2, c3, c4, Rv, x0=4.579, gamma=0.934):
+    """
+    Fitzpatrick & Mazza extinction curve
+    Return ksi = A(lambda)/A(V) instead of E(B-lambda)/E(B-V).
+    Using O2 and O3 as free parameters for optical spline.
+      O2 = 1.33 +/- 0.01
+      O3 = 2.0 +/- 0.1
+    O1 is kept fixed to ensure correct normalization.
+
+    Parameters
+    ----------
+    wl : np.ndarray
+        Wavelength array in angstrom
+
+    pars : dict
+        Parameters for the extinction curve:
+        (c1, c2, c3, c4, x0, gamma, Rv)
+
+    Returns
+    -------
+    ksi : np.ndarray
+        Array of A(lambda) / A(v) for the given input wavelength array.
+    """
+    k = 1 / (wl*1.e-4)
+
+    D = k**2/((k**2 - x0**2)**2 + k**2*gamma**2)
+
+    F = 0.5392*(k-5.9)**2 + 0.05644*(k-5.9)**3
+    F[k < 5.9] = 0.
+
+    # Use Fitzpatrick & Massa 1990 original formulation
+    # with fixed UV polynomial:
+    Ebv = c1 + c2*k + c3*D + c4*F
+
+    # Use IR power-law from Fitzpatrick & Massa 2007
+    # assuming the correlation ketween k_IR and Rv.
+    # Their eq. 7:
+    Ebv_IR = (-0.83 + 0.63*Rv)*k**1.84 - Rv
+
+    # Use spline points from 1 < x < 3.7
+    # Anchor to UV and IR parts to make smooth transition: (FM2007)
+    U1 = 3.85                                    # anchor at 2600A
+    U2 = 3.7                                    # anchor at 2700A
+    O_UV1 = c1 + c2*U1 + c3*U1**2/((U1**2 - x0**2)**2 + U1**2*gamma**2)
+    O_UV2 = c1 + c2*U2 + c3*U2**2/((U2**2 - x0**2)**2 + U2**2*gamma**2)
+    O_IRopt = (-0.83 + 0.63*Rv)*1.0**1.84 - Rv    # anchor at 1.0
+    O_IR = (-0.83 + 0.63*Rv)*0.75**1.84 - Rv    # anchor at 0.75
+
+    # Array of anchor points for spline
+    O = np.array([O_IR, O_IRopt, 0., O_UV2, O_UV1])
+
+    # Array of inverse wavelength for spline anchors:
+    k_anchor = np.array([0.75, 1., 1.808, U2, U1])
+
+    Ebv_spline_func = spline(k_anchor, O)
+    Ebv_spline = Ebv_spline_func(k)
+
+    # stitch together the pieces
+    Ebv = Ebv*(k >= 3.7) + Ebv_spline*(k < 3.7)*(k > 1.) + Ebv_IR*(k <= 1.)
+
+    ksi = Ebv/Rv + 1.
+
+    return ksi
+
+
+class DustModel:
+    name = ''
+
+    def __init__(self, name: str, pars: dict[str, float], functor=FM2007):
+        self.name = name
+        self.pars = pars
+        self.functor = functor
+
+    def __call__(self, x: np.ndarray, Av=1.0):
+        model = 10**(-0.4 * self.functor(x, **self.pars) * Av)
+        return 10**(-0.4 * self.functor(x, **self.pars) * Av)
+
+    def __repr__(self):
+        return self.__str__()
+
+    def __str__(self):
+        return self.name
+
+
+class SMCDustModel(DustModel):
+    name = 'SMC'
+
+    def __init__(self):
+        super().__init__(name='SMC', pars=parameters['SMC'])
+
+
+class LMCDustModel(DustModel):
+    name = 'LMC'
+
+    def __init__(self):
+        super().__init__(name='LMC', pars=parameters['LMC'])
+
+
+class MWDustModel(DustModel):
+    name = 'MW'
+
+    def __init__(self):
+        super().__init__(name='MW', pars={'Rv': 3.1}, functor=MW_reddening)