def main(config, outpath, magflag, clobber=False):
files = config['Runtime']['outpath']
files = glob(files)
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
size = comm.Get_size()
files = files[rank::size]
cc = config['Cosmology']
nb_config = config['NBody']
cosmo = Cosmology(**cc)
domain = Domain(cosmo, **nb_config.pop('Domain'))
domain.decomp(comm, comm.rank, comm.size)
d = domain.dummyDomain()
nbody = NBody(cosmo, d, **nb_config)
model = ADDGALSModel(nbody, **config['GalaxyModel']['ADDGALSModel'])
filters = config['GalaxyModel']['ADDGALSModel']['colorModelConfig'][
'filters']
train = fitsio.read(config['GalaxyModel']['ADDGALSModel']
['colorModelConfig']['trainingSetFile'])
nk = len(filters)
for f in files:
pixnum = f.split('.')[-2]
fname = '{}-{}.{}.fits'.format(outpath, magflag, pixnum)
if os.path.exists(fname) & (not clobber):
continue
g = fitsio.read(f, columns=['SEDID', 'Z', 'MAG_R_EVOL', 'MU'])
mags = np.zeros(len(g),
dtype=np.dtype([('TMAG', (np.float, nk)),
('AMAG', (np.float, nk)),
('LMAG', (np.float, nk)),
('OMAG', (np.float, nk)),
('OMAGERR', (np.float, nk)),
('FLUX', (np.float, nk)),
('IVAR', (np.float, nk)),
('Z', np.float)]))
mags['Z'] = g['Z']
z_a = copy(mags['Z'])
z_a[z_a < 1e-6] = 1e-6
mags['TMAG'], mags['AMAG'] = model.colorModel.computeMagnitudes(
g['MAG_R_EVOL'], g['Z'], train['COEFFS'][g['SEDID']], filters)
for i in range(len(filters)):
mags['LMAG'][:, i] = mags['TMAG'][:, i] - 2.5 * np.log10(g['MU'])
fitsio.write(fname, mags, clobber=clobber)
def main(config, outpath, magflag, clobber=False):
files = config['Runtime']['outpath']
files = glob(files)
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
size = comm.Get_size()
files = files[rank::size]
cc = config['Cosmology']
nb_config = config['NBody']
cosmo = Cosmology(**cc)
domain = Domain(cosmo, **nb_config.pop('Domain'))
domain.decomp(comm, comm.rank, comm.size)
d = domain.dummyDomain()
nbody = NBody(cosmo, d, **nb_config)
model = ADDGALSModel(nbody, **config['GalaxyModel']['ADDGALSModel'])
filters = config['GalaxyModel']['ADDGALSModel']['colorModelConfig'][
'filters']
train = fitsio.read(config['GalaxyModel']['ADDGALSModel']
['colorModelConfig']['trainingSetFile'])
nk = len(filters)
for f in files:
pixnum = f.split('.')[-2]
fname = '{}-{}.{}.fits'.format(outpath, magflag, pixnum)
if os.path.exists(fname) & (not clobber):
continue
g = fitsio.read(f,
columns=[
'SEDID', 'Z', 'MAG_R_EVOL', 'MU', 'Z_COS', 'VX',
'VY', 'VZ', 'PX', 'PY', 'PZ'
])
ngal = len(g)
pos = np.zeros((ngal, 3))
vel = np.zeros((ngal, 3))
pos[:, 0] = g['PX']
pos[:, 1] = g['PY']
pos[:, 2] = g['PZ']
vel[:, 0] = g['VX']
vel[:, 1] = g['VY']
vel[:, 2] = g['VZ']
v_r = np.sum(pos * vel, axis=1) / np.sqrt(np.sum(pos**2, axis=1))
z_rsd = g['Z_COS'] + v_r * (1 + g['Z_COS']) / 299792.458
del pos, vel, v_r
m_r = np.copy(g['MAG_R_EVOL'])
mu = np.copy(g['MU'])
coeffs = train['COEFFS'][g['SEDID']]
del g
mags = np.zeros(ngal,
dtype=np.dtype([('TMAG', (np.float, nk)),
('AMAG', (np.float, nk)),
('LMAG', (np.float, nk)),
('OMAG', (np.float, nk)),
('OMAGERR', (np.float, nk)),
('FLUX', (np.float, nk)),
('IVAR', (np.float, nk)),
('Z', np.float)]))
mags['Z'] = z_rsd
z_a = copy(mags['Z'])
z_a[z_a < 1e-6] = 1e-6
mags['TMAG'], mags['AMAG'] = model.colorModel.computeMagnitudes(
m_r, z_rsd, coeffs, filters)
for i in range(len(filters)):
mags['LMAG'][:, i] = mags['TMAG'][:, i] - 2.5 * np.log10(mu)
fitsio.write(fname, mags, clobber=clobber)