In [1]:
import os
import numpy as np
import h5py
from prospect.sources.star_basis import StarBasis

In [2]:
libname = os.path.join(os.getenv('DESI_ROOT'), 'data', 'mwsinterp', 'miles_prugniel.h5')

In [3]:
f = h5py.File(libname, "r", driver=None)
stellar_params1 = f['parameters'][:]
f.close()

In [6]:
# Toss our the NaNs
if False:
    param_names = stellar_params1.dtype.names
    good = np.ndarray( (len(param_names), len(stellar_params1)) )
    for ii, pp in enumerate(param_names):
        good[ii, :] = ~np.isnan(stellar_params1[pp])
    these = np.where(np.sum(good, axis=0) == 3)[0]

    stellar_params = np.ndarray( (len(these), ), dtype=[(p, 'f4') for p in param_names])
    for pp in param_names:
        stellar_params[pp] = stellar_params1[pp][these]

In [7]:
ss = StarBasis(libname, verbose=True, logify_Z=False)


(979, 3)

In [8]:
ss.weights(teff=6435, feh=0.0, logg=1.0)


Out[8]:
(array([ 43, 741, 750, 891], dtype=int32),
 array([ 0.20431003,  0.54597613,  0.11161181,  0.13810203]))

In [9]:
wave, flux, ferr = ss.get_star_spectrum(teff=6435, feh=0.0, logg=1.0)

In [ ]: