guilhem_lavaux/cosmotool
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Python CIC is now more memory and CPU efficient using numexpr

Browse source
This commit is contained in:
Guilhem Lavaux 2014-09-25 09:55:32 +02:00
parent 0d688fbf7d
commit 4e79e2eec6
2 changed files with 31 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

11
python_sample/build_dipole_ksz_from_galaxies.py
View file

@ -99,6 +99,7 @@ def get_args():
parser.add_argument('--ksz_map', type=str, required=True)
parser.add_argument('--base_fig', type=str, default="kszfig.png")
parser.add_argument('--build_dipole', type=bool, default=False)
parser.add_argument('--degrade', type=int, default=-1)
return parser.parse_args()
def main():
@ -113,11 +114,19 @@ def main():
proj,mask = generate_from_catalog(args.depth_min,args.depth_max,args.Nside)
if args.degrade > 0:
proj *= mask
proj = hp.ud_grade(proj, nside_out=args.degrade)
mask = hp.ud_grade(mask, nside_out=args.degrade)
Nside = args.degrade
else:
Nside = args.Nside
hp.write_map(args.ksz_map + ".fits", proj)
hp.write_map(args.ksz_map + "_mask.fits", mask)
if args.build_dipole:
x,y,z=hp.pix2vec(args.Nside, np.arange(hp.nside2npix(args.Nside)))
x,y,z=hp.pix2vec(Nside, np.arange(hp.nside2npix(Nside)))
hp.write_map(args.ksz_map + "_x.fits", proj*x)
hp.write_map(args.ksz_map + "_y.fits", proj*y)
hp.write_map(args.ksz_map + "_z.fits", proj*z)