def load_model(cfg):
config_file = cfg
config = parseConfig(config_file)
cc = config['Cosmology']
nb_config = config['NBody']
cosmo = Cosmology(**cc)
d_config = nb_config.pop('Domain')
domain = Domain(cosmo, **d_config)
domain.decomp(None, 1, 1)
for d in domain.yieldDomains():
nbody = NBody(cosmo, d, **nb_config)
break
model = ADDGALSModel(nbody, **config['GalaxyModel']['ADDGALSModel'])
return model, config, d_config
def load_model(cfg):
config = parseConfig(cfg)
cc = config['Cosmology']
nb_config = config['NBody']
cosmo = Cosmology(**cc)
domain = Domain(cosmo, **nb_config.pop('Domain'))
domain.decomp(None, 0, 1)
for d in domain.yieldDomains():
nbody = NBody(cosmo, d, **nb_config)
break
model = ADDGALSModel(nbody, **config['GalaxyModel']['ADDGALSModel'])
filters = [
'desy3/desy3std_g.par', 'desy3/desy3std_r.par', 'desy3/desy3std_i.par',
'desy3/desy3std_z.par', 'desy3/desy3_Y.par'
]
return model, filters
('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)
if __name__ == '__main__':
config_file = sys.argv[1]
outpath = sys.argv[2]
magflag = sys.argv[3]
if len(sys.argv) > 4:
clobber = True
else:
clobber = False
config = parseConfig(config_file)
main(config, outpath, magflag, clobber=clobber)
if i == 0:
h = hi
else:
h = np.hstack([h, hi])
h = h[h['HOST_HALOID'] == -1]
del hi
hr = np.sqrt(h['PX']**2 + h['PY']**2 + h['PZ']**2)
for i in range(len(files)):
print(files[i])
g = fitsio.read(files[i])
r = np.sqrt(g['PX']**2 + g['PY']**2 + g['PZ']**2)
config = parseConfig(cfg)
cc = config['Cosmology']
nb_config = config['NBody']
nb_config['Domain']['pixlist'] = [0]
cosmo = Cosmology(**cc)
domain = Domain(cosmo, **nb_config.pop('Domain'))
domain.decomp(None, 0, 1)
for d in domain.yieldDomains():
nbody = NBody(cosmo, d, **nb_config)
print('[{}]: working on rbin {}'.format(
rank, domain.rbins[d.boxnum][d.rbin]))
sys.stdout.flush()
idx = ((domain.rbins[d.boxnum][d.rbin] <= r) &
footprint_cfg_file = sys.argv[2]
partpath = sys.argv[3]
outbase = sys.argv[4]
if len(sys.argv) > 5:
mastercat_file = sys.argv[5]
write_to_mastercat = True
else:
write_to_mastercat = False
if len(sys.argv) > 6:
only_mastercat = bool(sys.argv[6])
else:
only_mastercat = False
addgals_config = parseConfig(addgals_cfg_file)
with open(footprint_cfg_file, 'r') as fp:
footprint_cfg = yaml.load(fp)
cc = addgals_config['Cosmology']
cosmo = Cosmology(**cc)
mask_file = footprint_cfg['DepthFile']
mask = fitsio.read(mask_file)
mask = mask['HPIX']
rotation_file = footprint_cfg['MatPath']
with open(rotation_file, 'rb') as fp:
rot_matrix = pickle.load(fp, encoding='latin1')