adding updating of vega.hd5 file

beast/tools/make_libfile.py

@@ -1,17 +1,19 @@
 # script to generate BEAST library files
+import os
 import numpy as np
 
 import stsynphot as stsyn
 from pandeia.engine.instrument_factory import InstrumentFactory
 import astropy.units as u
+from astropy.table import QTable
 
 import h5py
 
 from beast.config import __ROOT__
 from beast.observationmodel import phot
 
 
-def make_libfile():
+def make_filters_libfile():
     """
     Extract filters from STSYNPHOT and save to main library file.
     """
@@ -644,5 +646,64 @@ def make_libfile():
     hf.close()
 
 
+def make_vega_libfile():
+
+    # Read an updated filters.hd5 lib
+    __default_filtlist__ = __ROOT__ + "/filters.hd5"
+    filtlist = QTable.read(__default_filtlist__, path="content")
+    filters = [cfilt.decode("UTF-8") for cfilt in filtlist["TABLENAME"].data]
+
+    # filenames for vega info
+    __default_vega__ = __ROOT__ + "/vega.hd5"
+    __default_vega_old__ = __ROOT__ + "/vega_old.hd5"
+
+    # rename the current file so we can write a new version
+    os.rename(__default_vega__, __default_vega_old__)
+
+    # Get a spectrum from the need-to-be updated vega lib
+    vega_old = QTable.read(__default_vega_old__, path="spectrum")
+    vl = vega_old["WAVELENGTH"].data
+    vf = vega_old["FLUX"].data
+
+    # Write out an updated vega.hd5 file
+    vega = h5py.File(__default_vega__, "w")
+
+    vega.create_dataset("spectrum", data=vega_old)
+    flist = phot.load_filters(filters, interp=True, lamb=vl, filterLib=__default_filtlist__)
+
+    fname = []
+    cwave = []
+    lum = []
+    mag = []
+
+    for i in range(len(flist)):
+        fname.append(flist[i].name)
+        cwave.append(flist[i].cl)
+        flux = flist[i].getFlux(vl, vf)
+        lum.append(flux)
+        mag.append(-2.5 * np.log10(flux))
+
+    contents = np.array(
+            list(
+                zip(
+                    fname,
+                    cwave,
+                    lum,
+                    mag,
+                )
+            ),
+            dtype=[
+                ("FNAME", "S30"),
+                ("CWAVE", "<f8"),
+                ("LUM", "<f8"),
+                ("MAG", "<f8"),
+            ],
+        )
+
+    vega.create_dataset("sed", data=contents)
+    vega.close()
+
+
 if __name__ == "__main__":  # pragma: no cover
-    make_libfile()
+    make_filters_libfile()
+    make_vega_libfile()