Homec4science
Diffusion ARRAKIHS (master) Browse

arrakihs/Workspace/InjectedDwarfs_conflimit/ark_idealized_dwarfs_get_properties
d192e801c42bmaster

Last Change
Raw File

ark_idealized_dwarfs_get_properties
View Options

#!/usr/bin/env python3
import argparse
import numpy as np
import os
from pNbody import *
from pNbody.Mockimgs import filters
from pNbody import libgrid
####################################################################
# option parser
####################################################################
description="""generate a set of idealized dwarf galaxies in the hdf5 mockimg format
"""
epilog ="""
Examples:
--------
ark_idealized_dwarfs_get_properties snapshots*.hdf5
"""
parser = argparse.ArgumentParser(description=description,epilog=epilog,formatter_class=argparse.RawDescriptionHelpFormatter)
parser.add_argument(action="store",
dest="files",
metavar='FILE',
type=str,
default=None,
nargs='*',
help='a list of files')
parser.add_argument("-o",
action="store",
type=str,
dest="outputfilename",
default=None,
help="Name of the output file")
####################################################################
# main
####################################################################
filters_list = [
'BastI_GAIA_G',
'BastI_GAIA_G_RP',
'BastI_GAIA_G_BP',
'BastI_HST_F475X',
'BPASS230_SDSS_u',
'BPASS230_SDSS_g',
'BPASS230_SDSS_r',
'BPASS230_SDSS_i',
'BPASS230_SDSS_z',
'BPASS230_Euclid_VIS',
'BPASS230_Euclid_J',
'BPASS230_Euclid_Y',
'BPASS230_Euclid_H',
'BPASS230_ARK_VIS1',
'BPASS230_ARK_VIS2',
'BPASS230_ARK_NIR1',
'BPASS230_ARK_NIR2']
if __name__ == '__main__':
opt = parser.parse_args()
database = {}
# loop over files
for filename in opt.files:
print(filename)
nb = Nbody(filename)
data = {}
#####################################
# luminosity
#####################################
Lum = nb.Luminosity().sum()
data["Lv"] = Lum
#####################################
# initial IMF mass
#####################################
Mass = nb.Mass(units="Msol").sum()
data["Mass"] = Mass
#####################################
# magnitudes in different filters
#####################################
Masses = nb.mass * 1e10
Ages = nb.age
Fes = nb.mh
for filter_name in filters_list:
# set the filter
mags = filters.Filter(filter_name).Magnitudes(Masses,Ages,Fes)
# convert to flux
F = np.sum(10**(-mags/2.5))
# convert to magnitude
mag = -2.5 * np.log10(F)
data["Mag_%s"%filter_name] = mag
#####################################
# Half light radius
#####################################
R12 = nb.CylindricalHalfMassFromIntegratedMass(nlos=1)
data["R12"] = R12
#####################################
# Mode metallicity
#####################################
# do the 1d histogram
G = libgrid.Generic_1d_Grid(-4,0.5,40)
y = G.get_MassMap(nb.mh,np.ones(nb.nbody))/nb.nbody
x = G.get_r()
Fe_mode = x[np.argmax(y)]
data["Fe_mode"] = Fe_mode
#####################################
# Done for this file : store
#####################################
name = os.path.splitext(os.path.basename(filename))[0]
database[name]=data
if opt.outputfilename is not None:
import pickle
with open(opt.outputfilename, 'wb') as f:
pickle.dump(database, f)
c4science · Help