pol = args.do_pol if pol: raise NotImplementedError shape_sim, wcs_sim, shape_dat, wcs_dat = aio.enmaps_from_config( Config, sim_section, analysis_section, pol=pol) if rank == 0: print(("Dat shape : ", shape_dat)) print(("Sim shape : ", shape_sim)) 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(Config, expf_name, shape_dat, wcs_dat, dimensionless=False) parray_sim = aio.patch_array_from_config(Config, expf_name, shape_sim, wcs_sim, dimensionless=False) # 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]
min_ell = fmaps.minimum_ell(shape_dat, wcs_dat) if rank == 0: print("Ell bounds...") lb = aio.ellbounds_from_config(Config, "reconstruction_sigurd", min_ell) tellmin = lb['tellminY'] tellmax = lb['tellmaxY'] pellmin = lb['pellminY'] pellmax = lb['pellmaxY'] kellmin = lb['kellmin'] kellmax = lb['kellmax'] if rank == 0: print("Patches data...") parray_dat = aio.patch_array_from_config(Config, expf_name, shape_dat, wcs_dat, dimensionless=True) if rank == 0: print("Attributes...") lxmap_dat, lymap_dat, modlmap_dat, angmap_dat, lx_dat, ly_dat = fmaps.get_ft_attributes_enmap( shape_dat, wcs_dat) if rank == 0: print("Binners...") lbin_edges = np.arange(kellmin, kellmax, 200) lbinner_dat = stats.bin2D(modlmap_dat, lbin_edges) if rank == 0: print("Cosmology...")