76 lines
2.0 KiB
C++
76 lines
2.0 KiB
C++
|
/*+
|
||
|
|
ARES/HADES/BORG Package -- ./libLSS/tests/test_window3d.cpp
|
||
|
|
Copyright (C) 2014-2020 Guilhem Lavaux <guilhem.lavaux@iap.fr>
|
||
|
|
Copyright (C) 2009-2020 Jens Jasche <jens.jasche@fysik.su.se>
|
||
|
|
|
||
|
|
Additional contributions from:
|
||
|
|
Guilhem Lavaux <guilhem.lavaux@iap.fr> (2023)
|
||
|
|
|
||
|
|
+*/
|
||
|
|
#include <string>
|
||
|
|
#include "libLSS/mpi/generic_mpi.hpp"
|
||
|
|
#include <cmath>
|
||
|
|
#include <healpix_cxx/healpix_map.h>
|
||
|
|
#include <boost/multi_array.hpp>
|
||
|
|
#include <boost/array.hpp>
|
||
|
|
#include "libLSS/tools/console.hpp"
|
||
|
|
#include "libLSS/tools/static_init.hpp"
|
||
|
|
#include "libLSS/data/window3d.hpp"
|
||
|
|
#include "libLSS/samplers/core/random_number.hpp"
|
||
|
|
#include "libLSS/samplers/rgen/gsl_random_number.hpp"
|
||
|
|
#include <CosmoTool/hdf5_array.hpp>
|
||
|
|
|
||
|
|
static const int N0 = 128;
|
||
|
|
|
||
|
|
struct BasicSelFunction
|
||
|
|
{
|
||
|
|
Healpix_Map<double> C;
|
||
|
|
|
||
|
|
double getRadialSelection(double r, int i) const {
|
||
|
|
return std::exp(-0.5*(r*r/400.));
|
||
|
|
}
|
||
|
|
|
||
|
|
int getNumRadial() const { return 1; }
|
||
|
|
|
||
|
|
double get_sky_completeness(double x, double y, double z) const {
|
||
|
|
return C[C.vec2pix(vec3(x,y,z))];
|
||
|
|
}
|
||
|
|
|
||
|
|
double get_sky_completeness(double ra, double dec) const {
|
||
|
|
return C[C.ang2pix(pointing(0.5*M_PI-dec, ra))];
|
||
|
|
}
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
using namespace LibLSS;
|
||
|
|
|
||
|
|
int main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
Console& console = Console::instance();
|
||
|
|
MPI_Communication *comm = LibLSS::setupMPI(argc, argv);
|
||
|
|
boost::multi_array<double, 3> selFuncData(boost::extents[N0][N0][N0]);
|
||
|
|
double L[3] = {200.,200.,200.};
|
||
|
|
double xmin[3] = {-100,-100,-100};
|
||
|
|
double delta[3] = {200./N0, 200./N0, 200./N0 };
|
||
|
|
StaticInit::execute();
|
||
|
|
|
||
|
|
RandomNumberThreaded<GSL_RandomNumber> rng(-1);
|
||
|
|
|
||
|
|
console.setVerboseLevel<LOG_INFO>();
|
||
|
|
|
||
|
|
LibLSS::smp_set_nested(true);
|
||
|
|
|
||
|
|
BasicSelFunction sel;
|
||
|
|
|
||
|
|
sel.C.SetNside(1, RING);
|
||
|
|
sel.C.fill(0);
|
||
|
|
sel.C[0] = 1.0;
|
||
|
|
|
||
|
|
compute_window_value_elem(comm, rng, sel, selFuncData, L, delta, xmin, 0.001);
|
||
|
|
|
||
|
|
H5::H5File f("test_window.h5", H5F_ACC_TRUNC);
|
||
|
|
CosmoTool::hdf5_write_array(f, "selData", selFuncData);
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|