def make_maps(args, comm, mode):
hp.disable_warnings()
nprocs = comm.Get_size()
rank = comm.Get_rank()
name = MPI.Get_processor_name()
if rank == 0:
reals = np.arange(args.nreal)
chunks = np.array_split(reals, nprocs)
else:
chunks = None
chunk = comm.scatter(chunks, root=0)
if mode is not False:
if mode == 'E':
cl = make_cl(args, comm, mode)
if mode == 'B':
cl = make_cl(args, comm, mode)
else:
cl = load_cl(args, comm)
for i in range(chunk[0], chunk[-1]+1):
if mode is not False:
print(f'Rank {rank} is processing {mode} realization {i} on processor {name}')
outdir = f"{args.outpath}/{mode}"
else:
print(f'Rank {rank} is processing realization {i} on processor {name}')
outdir = f"{args.outpath}"
m = hp.synfast(cl, args.nside, lmax=3*args.nside-1, pol=True, new=True)
#m_smooth = hp.smoothing(m, args.beamfwhm *np.pi/10800)
if not os.path.isdir(outdir):
os.mkdir(outdir)
hp.write_map(f"{outdir}/map_{i}.fits", hp.reorder(m, r2n=True), overwrite=True, nest= True, dtype=np.float64)
def small_to_zero(args, v):
hp.disable_warnings()
#mask = hp.ud_grade(hp.read_map(args.mask, verbose=False, dtype=np.float64), nside_out=args.nside)
#mask = hp.read_map(args.mask, verbose=False, dtype=np.float64)
mask = np.ones(12 * args.nside**2)
mask = np.concatenate((mask, mask))
zeros = np.where(mask == 0)[0]
print(f'{len(zeros)} zeros in each column')
for i in range(v.shape[1]):
v[:, i][zeros] = 0
return v
def main():
comm = MPI.COMM_WORLD
parser = argparse.ArgumentParser()
add_data_args(parser)
args = parser.parse_args()
print(f'INFO: NSIDE {args.nside}')
hp.disable_warnings()
write_distributed_data(args, comm)
gather_data(args, comm)
from re import A
import time
totaltime = time.time()
import click
import sys
import os
import healpy as hp
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as col
from src.tools import arcmin2rad
import os
hp.disable_warnings()
# Fix for macos openMP duplicate bug
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
#print("Importtime:", (time.time() - totaltime))
def trygveplot(input,
dataset=None,
nside=None,
auto=False,
min=False,
max=False,
mid=[],
rng="auto",
colorbar=False,
graticule=False,
lmax=None,
fwhm=0.0,