rbinner = stats.bin2D(modrmap*60.*180./np.pi,rbins) true_mass = 2.e14 true2d = get_nfw(true_mass) kellmin = 200 kellmax = 8500 true2d = enmap.ndmap(fmaps.filter_map(true2d,true2d*0.+1.,modlmap,lowPass=kellmax,highPass=kellmin),wcs) Nsims = 1000 cov = np.ones((1,1,8500))*0.00000001 ngen = enmap.MapGen(shape,wcs,cov) out_dir = os.environ['WWW']+"plots/cgauss_" # Efficiently distribute sims over MPI cores num_each,each_tasks = mpi_distribute(Nsims,numcores) # Initialize a container for stats and stacks mpibox = MPIStats(comm,num_each,tag_start=333) if rank==0: print(("At most ", max(num_each) , " tasks...")) # What am I doing? my_tasks = each_tasks[rank] for k,index in enumerate(my_tasks): noise = ngen.get_map() mkappa = true2d+noise if k==0 and rank==0: io.quickPlot2d(noise,out_dir+"nstamp.png") print(k) cents, prof = rbinner.bin(mkappa) mpibox.add_to_stats('prof',prof)
ells, cleb = b(cls[4, :]) ells, cltb = b(cls[5, :]) mpibox.add_to_stats("TT", cltt) mpibox.add_to_stats("EE", clee) mpibox.add_to_stats("BB", clbb) mpibox.add_to_stats("TE", clte) mpibox.add_to_stats("EB", cleb) mpibox.add_to_stats("TB", cltb) return ells Ntot = 8 # Efficiently distribute sims over MPI cores num_each, each_tasks = mpi_distribute(Ntot, numcores) # Initialize a container for stats and stacks mpibox = MPIStats(comm, num_each, tag_start=333) if rank == 0: print(("At most ", max(num_each), " tasks...")) # What am I doing? my_tasks = each_tasks[rank] k = -1 for index in my_tasks: k += 1 cmb_map = load_fullsky(sim_root, "fullsky_curved_lensed_car", index) print((rank, k, cmb_map.shape))
Nsims = Config.getint("general","Nsims") Nsets = Config.getint("general","Nsets") polsims = Config.getboolean("general","pol") pol = False shape_sim, wcs_sim, shape_dat, wcs_dat = aio.enmaps_from_config(Config,sim_section,analysis_section,pol=pol) analysis_resolution = np.min(enmap.extent(shape_dat,wcs_dat)/shape_dat[-2:])*60.*180./np.pi pixratio = analysis_resolution/Config.getfloat(sim_section,"pixel_arcmin") min_ell = fmaps.minimum_ell(shape_dat,wcs_dat) parray_dat = aio.patch_array_from_config(rConfig,expf_name,shape_dat,wcs_dat,dimensionless=False) # Efficiently distribute sims over MPI cores Nuse = Nsims if args.num_sims is None else args.num_sims assert args.num_sims < Nsims num_each,each_tasks = mpi_distribute(Nuse,numcores) # Initialize a container for stats and stacks mpibox = MPIStats(comm,num_each,tag_start=333) if rank==0: print(("At most ", max(num_each) , " tasks...")) # What am I doing? my_tasks = each_tasks[rank] # === COSMOLOGY === theory, cc, lmax = aio.theory_from_config(Config,cosmology_section,dimensionless=False) parray_dat.add_theory(cc,theory,lmax,orphics_is_dimensionless=False) lxmap_dat,lymap_dat,modlmap_dat,angmap_dat,lx_dat,ly_dat = fmaps.get_ft_attributes_enmap(shape_dat,wcs_dat) lb = aio.ellbounds_from_config(rConfig,recon_section,min_ell*8) tellmin = lb['tellminY']