def test_UnresolvedCluster():
from popstar import synthetic as syn
from popstar import atmospheres as atm
from popstar import evolution
from popstar.imf import imf
from popstar.imf import multiplicity
log_age = 6.7
AKs = 0.0
distance = 4000
metallicity=0
cluster_mass = 10**4.
startTime = time.time()
multi = multiplicity.MultiplicityUnresolved()
imf_in = imf.Kroupa_2001(multiplicity=multi)
evo = evolution.MergedBaraffePisaEkstromParsec()
atm_func = atm.get_merged_atmosphere
iso = syn.Isochrone(log_age, AKs, distance, metallicity=metallicity,
evo_model=evo, atm_func=atm_func, mass_sampling=10)
print('Made Isochrone: %d seconds' % (time.time() - startTime))
cluster = syn.UnresolvedCluster(iso, imf_in, cluster_mass)
print('Constructed unresolved cluster: %d seconds' % (time.time() - startTime))
# Plot an integrated spectrum of the whole cluster.
wave = cluster.wave_trim
flux = cluster.spec_trim
plt.clf()
plt.plot(wave, flux, 'k.')
return
def test_isochrone(plot=False):
from popstar import synthetic as syn
logAge = 6.7
AKs = 2.7
distance = 4000
startTime = time.time()
iso = syn.Isochrone(logAge, AKs, distance)
print('Test completed in: %d seconds' % (time.time() - startTime))
# Typically takes 104 - 120 seconds.
# Limited by pysynphot.Icat call in atmospheres.py
assert iso.points.meta['LOGAGE'] == logAge
assert iso.points.meta['AKS'] == AKs
assert iso.points.meta['DISTANCE'] == distance
assert len(iso.points) > 100
if plot:
plt.figure(1)
iso.plot_HR_diagram()
plt.figure(2)
iso.plot_mass_luminosity()
return iso
def test_UnresolvedCluster():
from popstar import synthetic as syn
from popstar import atmospheres as atm
from popstar import evolution
from popstar.imf import imf
from popstar.imf import multiplicity
log_age = 6.7
AKs = 0.0
distance = 4000
cluster_mass = 30.
startTime = time.time()
multi = multiplicity.MultiplicityUnresolved()
imf_in = imf.Kroupa_2001(multiplicity=multi)
evo = evolution.MergedBaraffePisaEkstromParsec()
iso = syn.Isochrone(log_age, AKs, distance, evo, mass_sampling=10)
print 'Made cluster: %d seconds' % (time.time() - startTime)
cluster = syn.UnresolvedCluster(iso, imf_in, cluster_mass)
print 'Constructed unresolved cluster: %d seconds' % (time.time() -
startTime)
# Plot an integrated spectrum of the whole cluster.
wave = cluster.spec_trim.wave
flux = cluster.spec_trim.flux
plt.clf()
plt.plot(wave, flux, 'k.')
pdb.set_trace()
return
def model_young_cluster_object(resolved=False):
from popstar import synthetic as syn
from popstar import atmospheres as atm
from popstar import evolution
from popstar.imf import imf
from popstar.imf import multiplicity
log_age = 6.5
AKs = 0.1
distance = 8000.0
cluster_mass = 10000.
multi = multiplicity.MultiplicityUnresolved()
imf_in = imf.Kroupa_2001(multiplicity=multi)
evo = evolution.MergedPisaEkstromParsec()
atm_func = atm.get_merged_atmosphere
iso = syn.Isochrone(log_age, AKs, distance, evo, mass_sampling=10)
if resolved:
cluster = syn.ResolvedCluster(iso, imf_in, cluster_mass)
else:
cluster = syn.UnresolvedCluster(iso,
imf_in,
cluster_mass,
wave_range=[19000, 24000])
# Plot the spectrum of the most massive star
idx = cluster.mass_all.argmax()
print('Most massive star is {0:f} M_sun.'.format(cluster.mass_all[idx]))
#bigstar = cluster.spec_list_trim[idx]
plt.figure(1)
plt.clf()
plt.plot(cluster.spec_list_trim[idx]._wavetable,
cluster.spec_list_trim[idx]._fluxtable, 'k.')
# Plot an integrated spectrum of the whole cluster.
wave, flux = cluster.spec_list_trim[idx]._wavetable, cluster.spec_trim
plt.figure(2)
plt.clf()
plt.plot(wave, flux, 'k.')
return