fpa = focalplane.FocalPlaneArray(latitude=0)
fpa.beam_spacing_u = 0.0
fpa.beam_spacing_v = 0.0
fpa.beam_num_u = 1
fpa.beam_num_v = 1
fpa.beam_size = 5.0
fpa.positive_m_only = True
fpa.num_freq = 5
fpa.freq_lower = 460.0
fpa.freq_upper = 540.0
bt = beamtransfer.BeamTransfer("cylinder/fpatest/", telescope=fpa)
# bt.generate()
klt = kltransform.KLTransform(bt)
# klt.generate()
pse = psestimation.PSEstimation(klt)
pse.bands = np.linspace(0.0, 1.0, 5)
# pse.genbands()
# pse.fisher_mpi()
from drift.core import kltransform
from drift.util import mpiutil
from mpi4py import MPI
## Read arguments in.
parser = argparse.ArgumentParser(description="Filter a map using S/N eigenmodes.")
parser.add_argument("teldir", help="The telescope directory to use.")
parser.add_argument("mapfile", help="Input map.")
parser.add_argument("outfile", help="Output map.")
parser.add_argument("threshold", help="Threshold S/N value to cut at.", type=float)
parser.add_argument("-e", "--evsubdir", help="The subdirectory containing the eigensystem files.")
args = parser.parse_args()
## Read in cylinder system
bt = beamtransfer.BeamTransfer(args.teldir)
klt = kltransform.KLTransform(bt, subdir=args.evsubdir)
cyl = bt.telescope
ntel = bt.ntel * bt.nfreq
mmax = cyl.mmax
cut = args.threshold
nside = 0
alm = np.zeros((cyl.nfreq, cyl.lmax+1, cyl.lmax+1), dtype=np.complex128)
if mpiutil.rank0:
## Useful output