62 lines
1.7 KiB
C++
62 lines
1.7 KiB
C++
|
#ifndef __COSMOTOOL_FOURIER_HEALPIX_DETAILS_SPECTRUM_HP
|
||
|
#define __COSMOTOOL_FOURIER_HEALPIX_DETAILS_SPECTRUM_HP
|
||
|
|
||
|
namespace CosmoTool
|
||
|
{
|
||
|
|
||
|
template<typename T>
|
||
|
class HealpixUtilityFunction: public MapUtilityFunction<T>
|
||
|
{
|
||
|
public:
|
||
|
typedef typename MapUtilityFunction<T>::Spectrum Spectrum;
|
||
|
typedef typename MapUtilityFunction<T>::Spectrum_ptr Spectrum_ptr;
|
||
|
typedef typename MapUtilityFunction<T>::FMap FMap;
|
||
|
|
||
|
Spectrum_ptr estimateSpectrumFromMap(const FMap& m) const
|
||
|
{
|
||
|
const HealpixFourierALM<T>& alms = dynamic_cast<const HealpixFourierALM<T>&>(m);
|
||
|
long Lmax = alms.Lmax(), Mmax = alms.Mmax();
|
||
|
HealpixSpectrum<T> *spectrum = new HealpixSpectrum<T>(Lmax);
|
||
|
T *data_cls = spectrum->data();
|
||
|
std::complex<T> *data_alms = alms.data();
|
||
|
|
||
|
// make an estimate of the cls
|
||
|
std::fill(data_cls, data_cls+Lmax+1, 0);
|
||
|
for (long m = 0, q = 0; m <= Mmax; ++m )
|
||
|
{
|
||
|
for (long l = 0; l <= Lmax; ++l)
|
||
|
{
|
||
|
// Triangular storage
|
||
|
data_cls[l] += std::norm(alms[l+q]);
|
||
|
}
|
||
|
q += Lmax+1;
|
||
|
}
|
||
|
|
||
|
for (long l = 0; l <= Lmax; ++l)
|
||
|
{
|
||
|
int dof = 1 + std::min(l, Mmax);
|
||
|
spectrum->set_dof(l, dof);
|
||
|
data_cls[l] /= dof;
|
||
|
}
|
||
|
|
||
|
return Spectrum_ptr(spectrum);
|
||
|
}
|
||
|
|
||
|
Spectrum_ptr newSpectrumFromRaw(T *data, long size,
|
||
|
Spectrum_ptr like_spec) const
|
||
|
{
|
||
|
Spectrum *s = like_spec.get();
|
||
|
HealpixSpectrum<T>& in_spec = dynamic_cast<HealpixSpectrum<T>&>(*s);
|
||
|
HealpixSpectrum<T> *new_spectrum = new HealpixSpectrum<T>(in_spec.Lmax());
|
||
|
T *out_d = new_spectrum->data();
|
||
|
|
||
|
std::copy(data, data + min(size,new_spectrum->size()), out_d);
|
||
|
|
||
|
return Spectrum_ptr(new_spectrum);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
};
|
||
|
|
||
|
#endif
|