sedfitting/read_chary_seds.py at master · bjweiner/sedfitting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

# read some SEDs from Chary and Elbaz 2001 from a file
# this uses a text file that I made from the Chary & Elbaz
# IDL save data file, outputting only the CE SEDs that
# have similar luminosity to the Rieke 2009, and converting
# them to be in units of flux at 10 Mpc like the Rieke SEDs

# this is derived from read_rieke_seds and since I formatted
# the file the same way, only difference is file location
# (and number of wavelengths but don't need to set that in Python)

# return a structure,
# which is a list of dictionaries

# if flag rujo2013 is set, then renormalize the luminosities
# according to Wiphu's stretch factor table.  I left this in
# for read_chary_seds since the SEDs are the same luminosity
# as the Rieke set, although the stretch factors were derived
# for Rieke seds so it's not self-consistent

# bjw 5/18/2016

# Hard code locations for the moment

import numpy as np
import os
from astropy.io import ascii
import matplotlib.pyplot as plt

def read_chary_seds(fname = '', outfile = '', rujo2013 = 0, makeplot=0):

    if fname == '':
        fname = 'chary_elbaz_14templ.wlmic_fluxjy_10mpc.dat'
        direc = 'fidel/chary_elbaz/'
        homedir = os.getenv('HOME')
        fname = homedir + '/' + direc + fname

    nseds = 14
    sedlabels = 9.75 + 0.25 * np.arange(nseds)

    f1 = open(fname, 'r')
    fdata = ascii.read(f1, data_start=0, delimiter='\s')
    f1.close()

    # wavelengths in first column
    wave1 = fdata.columns[0]
    sedstruct=[]
    for i in range(nseds):
        flux1 = fdata.columns[i+1]
        name1 = str(sedlabels[i])
        label1 = sedlabels[i]
        sed1 = {'label': label1,
                'name': name1,
                'wave': wave1,
                'flux': flux1}
        sedstruct.append(sed1)

    if rujo2013 == 1:
        # If labels are strings, would need to iterate through loop
        # changing labels to floats and changing back
        logstretchfac = [-0.118, 0.013, 0.173, 0.408, 0.717, 1.101,
                         1.560, 2.095, 2.704, 3.388]
        newlabels = sedlabels + logstretchfac
        # This may not make a copy
        tmpstruct = sedstruct
        for i in range(nseds):
            sedstruct[i]['name'] = str(newlabels[i])
            sedstruct[i]['label'] = newlabels[i]
            sedstruct[i]['flux'] = tmpstruct[i]['flux'] * 10**logstretchfac[i]

    # write SED structure to a file like a FITS table?
    if outfile != '':
        print 'Writing SED structure not implemented yet'

    # make a plot
    if makeplot != 0 :
        plt.clf()
        for i in range(nseds):
            style = 'k-'
            plt.plot(np.log10(sedstruct[i]['wave']), np.log10(sedstruct[i]['flux']), style)
        plt.show()

    return sedstruct