vide_public/external/healpix/Healpix_cxx/anafast_cxx_module.cc
2012-10-30 13:56:48 -04:00

129 lines
4 KiB
C++

/*
* This file is part of Healpix_cxx.
*
* Healpix_cxx is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* Healpix_cxx is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with Healpix_cxx; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
* For more information about HEALPix, see http://healpix.jpl.nasa.gov
*/
/*
* Healpix_cxx is being developed at the Max-Planck-Institut fuer Astrophysik
* and financially supported by the Deutsches Zentrum fuer Luft- und Raumfahrt
* (DLR).
*/
/*
* Copyright (C) 2003-2010 Max-Planck-Society
* Author: Martin Reinecke
*/
#include "xcomplex.h"
#include "cxxutils.h"
#include "paramfile.h"
#include "healpix_data_io.h"
#include "alm.h"
#include "alm_fitsio.h"
#include "healpix_map.h"
#include "healpix_map_fitsio.h"
#include "powspec.h"
#include "powspec_fitsio.h"
#include "alm_healpix_tools.h"
#include "alm_powspec_tools.h"
#include "fitshandle.h"
#include "levels_facilities.h"
#include "lsconstants.h"
using namespace std;
namespace {
template<typename T> void anafast_cxx (paramfile &params)
{
int nlmax = params.template find<int>("nlmax");
int nmmax = params.template find<int>("nmmax",nlmax);
string infile = params.template find<string>("infile");
string outfile = params.template find<string>("outfile","");
string outfile_alms = params.template find<string>("outfile_alms","");
planck_assert ((outfile!="") || (outfile_alms!=""),
"no output specified, nothing done");
bool polarisation = params.template find<bool>("polarisation");
int num_iter = params.template find<int>("iter_order",0);
if (!polarisation)
{
Healpix_Map<T> map;
read_Healpix_map_from_fits(infile,map,1,2);
arr<double> weight;
get_ring_weights (params,map.Nside(),weight);
Alm<xcomplex<T> > alm(nlmax,nmmax);
double avg=map.average();
map.Add(T(-avg));
if (map.Scheme()==NEST) map.swap_scheme();
map2alm_iter(map,alm,num_iter,weight);
alm(0,0) += T(avg*sqrt(fourpi));
if (outfile!="")
{
PowSpec powspec;
extract_powspec (alm,powspec);
write_powspec_to_fits (outfile,powspec,1);
}
if (outfile_alms!="")
write_Alm_to_fits(outfile_alms,alm,nlmax,nmmax,planckType<T>());
}
else
{
Healpix_Map<T> mapT, mapQ, mapU;
read_Healpix_map_from_fits(infile,mapT,mapQ,mapU,2);
arr<double> weight;
get_ring_weights (params,mapT.Nside(),weight);
Alm<xcomplex<T> > almT(nlmax,nmmax), almG(nlmax,nmmax), almC(nlmax,nmmax);
double avg=mapT.average();
mapT.Add(T(-avg));
if (mapT.Scheme()==NEST) mapT.swap_scheme();
if (mapQ.Scheme()==NEST) mapQ.swap_scheme();
if (mapU.Scheme()==NEST) mapU.swap_scheme();
map2alm_pol_iter
(mapT,mapQ,mapU,almT,almG,almC,num_iter,weight);
almT(0,0) += T(avg*sqrt(fourpi));
if (outfile!="")
{
PowSpec powspec;
extract_powspec (almT,almG,almC,powspec);
params.template find<bool>("full_powerspectrum",false) ?
write_powspec_to_fits (outfile,powspec,6) :
write_powspec_to_fits (outfile,powspec,4);
}
if (outfile_alms!="")
write_Alm_to_fits (outfile_alms,almT,almG,almC,nlmax,nmmax,planckType<T>());
}
}
} // unamed namespace
int anafast_cxx_module (int argc, const char **argv)
{
module_startup ("anafast_cxx", argc, argv, 2, "<parameter file>");
paramfile params (argv[1]);
bool dp = params.find<bool> ("double_precision",false);
dp ? anafast_cxx<double>(params) : anafast_cxx<float>(params);
return 0;
}