fmm::FastMultipole3d< value > Class Template Reference

#include <multipole3d.h>

Collaboration diagram for fmm::FastMultipole3d< value >:

Public Types
using	complex_value = std::conditional_t< std::is_same_v< value, double >, std::complex< double >, Vector3cd >
using	particle_t = particle< Vector3d, value >
using	MortonTree_t = MortonTree< 3, point3d, value >
using	TreeCell_t = typename MortonTree_t::TreeCell_t

Public Member Functions
	FastMultipole3d (const std::vector< particle_t > &particles, double eps=1.e-8, int N=FMM_AUTO, int tree_depth=FMM_AUTO)
	FastMultipole3d (std::vector< particle_t > &&particles, double eps=1.e-8, int N=FMM_AUTO, int tree_depth=FMM_AUTO)
	FastMultipole3d (const std::vector< particle_t > &source_particles, const std::vector< particle_t > &target_particles, double eps=1.e-8, int N=FMM_AUTO, int tree_depth=FMM_AUTO)

Public Attributes
std::vector< Vector3d >	forces
std::vector< Vector3d >	moments
std::vector< double >	potentials
std::vector< Vector3d >	dipolevelocities

Private Member Functions
void	Solve (std::vector< particle_t > &&particles, double eps=1.e-8, int N=FMM_AUTO, int tree_depth=FMM_AUTO)
void	Multipole (complex_value *a, const std::pair< size_t, size_t > &particle_range, const Vector3d &z0, double *Pnm, std::complex< double > *eim)
void	M2M (const complex_value *a, const Vector3d &p0, complex_value *b, std::vector< complex_value > &work1, std::vector< complex_value > &work2)
void	M2L (const complex_value *a, const Vector3d &p0, complex_value *b, std::vector< complex_value > &work1, std::vector< complex_value > &work2)
void	L2L (const complex_value *a, const Vector3d &p0, complex_value *b, std::vector< complex_value > &work1, std::vector< complex_value > &work2)
void	RotateY (const complex_value *a, const std::vector< double > &dmatrix, complex_value *res)
void	RotateZ (const complex_value *a, const std::vector< std::complex< double > > &rotation_exp, complex_value *b)
void	RotateZ (complex_value *a, const std::vector< std::complex< double > > &rotation_exp)
void	Upward ()
void	Downward ()
void	ComputeLeaves ()
void	Prepare ()
void	ComputeLeaves ()
void	ComputeLeaves ()

Private Attributes
std::shared_ptr< MortonTree_t >	tree
std::vector< std::vector< complex_value > >	outer_expansions
std::vector< std::vector< complex_value > >	inner_expansions
size_t	tree_depth
int	N
int	Nx2
size_t	num_particles
size_t	targets_num
tbb::enumerable_thread_specific< std::vector< complex_value > >	local_work1
tbb::enumerable_thread_specific< std::vector< complex_value > >	local_work2

Detailed Description

template<typename value>
class fmm::FastMultipole3d< value >

Definition at line 11 of file multipole3d.h.

Member Typedef Documentation

complex_value

template<typename value >

using fmm::FastMultipole3d< value >::complex_value = std::conditional_t<std::is_same_v<value, double>, std::complex<double>, Vector3cd>

Definition at line 14 of file multipole3d.h.

MortonTree_t

template<typename value >

using fmm::FastMultipole3d< value >::MortonTree_t = MortonTree<3, point3d, value>

Definition at line 16 of file multipole3d.h.

particle_t

template<typename value >

using fmm::FastMultipole3d< value >::particle_t = particle<Vector3d, value>

Definition at line 15 of file multipole3d.h.

TreeCell_t

template<typename value >

using fmm::FastMultipole3d< value >::TreeCell_t = typename MortonTree_t::TreeCell_t

Definition at line 17 of file multipole3d.h.

Constructor & Destructor Documentation

FastMultipole3d() [1/3]

template<typename value >

fmm::FastMultipole3d< value >::FastMultipole3d	(	const std::vector< particle_t > &	particles,
		double	eps = 1.e-8,
		int	N = FMM_AUTO,
		int	tree_depth = FMM_AUTO
	)

Definition at line 1481 of file multipole3d.cpp.

{
    auto particles_copy = particles;
    targets_num = particles.size();
    Solve(std::move(particles_copy), eps, N_, tree_depth_);
}

FastMultipole3d() [2/3]

template<typename value >

fmm::FastMultipole3d< value >::FastMultipole3d	(	std::vector< particle_t > &&	particles,
		double	eps = 1.e-8,
		int	N = FMM_AUTO,
		int	tree_depth = FMM_AUTO
	)

Definition at line 1489 of file multipole3d.cpp.

{
    targets_num = particles.size();
    Solve(std::move(particles), eps, N_, tree_depth_);
}

FastMultipole3d() [3/3]

template<typename value >

fmm::FastMultipole3d< value >::FastMultipole3d	(	const std::vector< particle_t > &	source_particles,
		const std::vector< particle_t > &	target_particles,
		double	eps = 1.e-8,
		int	N = FMM_AUTO,
		int	tree_depth = FMM_AUTO
	)

Definition at line 1496 of file multipole3d.cpp.

{
    std::vector<particle_t> particles(source_particles.size() + target_particles.size());
    memcpy(particles.data(), target_particles.data(), target_particles.size() * sizeof(particle_t));
    memcpy(particles.data() + target_particles.size(), source_particles.data(), source_particles.size() * sizeof(particle_t));
    targets_num = target_particles.size();
    Solve(std::move(particles), eps, N_, tree_depth_);
}

Member Function Documentation

ComputeLeaves() [1/3]

void fmm::FastMultipole3d< Vector3d >::ComputeLeaves ( )

private

Definition at line 471 of file multipole3d.cpp.

{
    using namespace std::literals;
 
    tbb::enumerable_thread_specific<Vector3d> tmp_force_ets, dr_ets;
    tbb::enumerable_thread_specific<Matrix3d> matrix_ets;
    tbb::enumerable_thread_specific<std::vector<std::complex<double>>> eim_ets(std::vector<std::complex<double>>(N + 1));
    tbb::enumerable_thread_specific<treevector<double>> Pnm_ets(treevector<double>(N + 3));                                 //up to 3rd derivatives
    tbb::enumerable_thread_specific<std::vector<Vector3d>> buf_forces_ets((std::vector<Vector3d>(num_particles)));
    tbb::enumerable_thread_specific<std::vector<Vector3d>> buf_moments_ets((std::vector<Vector3d>(num_particles)));
 
    const auto& leaves = tree->levels.back();
    const auto& leaves_inner = inner_expansions.back();
    const auto& particles = tree->particles;
 
#ifdef FMM_MPI
    auto [Begin, End] = LocalPart(0, tree->level_sizes.back());
#else
    size_t Begin = 0, End = tree->level_sizes.back();
#endif
 
    decltype(&TreeCell_t::source_range) range_ptr;
    tree->targets_num == 0 ? range_ptr = &TreeCell_t::source_range : range_ptr = &TreeCell_t::target_range;
 
    tbb::parallel_for(Begin, End, [&](size_t leaf_idx) {
        auto& tmp_force = tmp_force_ets.local();
        auto& dr = dr_ets.local();
        auto& eim = eim_ets.local();
        auto& Pnm = Pnm_ets.local();
        auto& m_temp = matrix_ets.local();
        auto& buf_forces = buf_forces_ets.local();
        auto& buf_moments = buf_moments_ets.local();
 
 
        const auto& cell = leaves[leaf_idx];
        const auto& [p1_begin, p1_end] = cell.*range_ptr;
 
        for (auto j = p1_begin; j < p1_end; ++j)
        {
            auto& particle = particles[j];
 
            for (auto k = p1_begin; k < p1_end; ++k)
            {
                dr = particle.center - particles[k].center;
 
                // velDipole //
                //*
                auto Ldr = abs(dr);
 
                auto argLE = (Ldr / FORCE_EPS);
                if (argLE < 3.0)
                {
                    auto [lambda, eta] = lambdaEta(argLE);
                    auto [lambdaA, etaA] = lambdaEtaA(argLE);
                    buf_forces[j] += particles[k].q * lambda / FORCE_EPS3 + dot(particles[k].q, dr) * dr * eta / FORCE_EPS5;
                    buf_moments[j] -= (dot(particles[j].q, particles[k].q) * lambdaA / FORCE_EPS4 + dot(particles[k].q, dr) * dot(particles[j].q, dr) * etaA / FORCE_EPS6) / std::max(Ldr, 1e-10) * dr + \
                        (dot(particles[j].q, dr) * particles[k].q + dot(particles[k].q, dr) * particles[j].q) * eta / FORCE_EPS5;
 
                }
                else
                {
                    auto Ldr2 = Ldr * Ldr;
                    auto Ldr3 = Ldr2 * Ldr;
                    auto Ldr4 = Ldr2 * Ldr2;
                    auto Ldr5 = Ldr3 * Ldr2;
                    auto Ldr6 = Ldr3 * Ldr3;
                    buf_forces[j] += particles[k].q / (-Ldr3) + 3 * dot(particles[k].q, dr) * dr / Ldr5;
                    buf_moments[j] -= (3 * dot(particles[j].q, particles[k].q) / Ldr4 - 15 * dot(particles[k].q, dr) * dot(particles[j].q, dr) / Ldr6 ) / std::max(Ldr, 1e-10) * dr + \
                        3 * (dot(particles[j].q, dr) * particles[k].q + dot(particles[k].q, dr) * particles[j].q) / Ldr5;
                }
                //*/
                // end velDipole //
 
                // vel BS //
                //buf_forces[j] += cross(particles[k].q, dr) / MyPow(std::max(abs(dr), FORCE_EPS), 3);
                // end vel BS //
            }
 
            for (const auto& idx : cell.closeneighbours)
            {
                const auto& neighbour_cell = leaves[idx];
                const auto& [p2_begin, p2_end] = neighbour_cell.source_range;
                for (auto k = p2_begin; k < p2_end; ++k)
                {
                    dr = particle.center - particles[k].center;
 
                    // velDipole //
                    //*
                    auto Ldr = abs(dr);
 
                    auto argLE = (Ldr / FORCE_EPS);
                    if (argLE < 3.0)
                    {
                        auto [lambda, eta] = lambdaEta(argLE);
                        auto [lambdaA, etaA] = lambdaEtaA(argLE);
                        buf_forces[j] += particles[k].q * lambda / FORCE_EPS3 + dot(particles[k].q, dr) * dr * eta / FORCE_EPS5;
                        buf_moments[j] -= ((dot(particles[j].q, particles[k].q) * lambdaA / FORCE_EPS4 + dot(particles[k].q, dr) * dot(particles[j].q, dr) * etaA / FORCE_EPS6) / std::max(Ldr, 1e-10) * dr + \
                            (dot(particles[j].q, dr) * particles[k].q + dot(particles[k].q, dr) * particles[j].q) * eta / FORCE_EPS5);
                    }
                    else
                    {
                        auto Ldr2 = Ldr * Ldr;
                        auto Ldr3 = Ldr2 * Ldr;
                        auto Ldr4 = Ldr2 * Ldr2;
                        auto Ldr5 = Ldr3 * Ldr2;
                        auto Ldr6 = Ldr3 * Ldr3;
                        buf_forces[j] += particles[k].q / (-Ldr3) + 3 * dot(particles[k].q, dr) * dr / Ldr5;
                        buf_moments[j] -= (3 * dot(particles[j].q, particles[k].q) / Ldr4 - 15 * dot(particles[k].q, dr) * dot(particles[j].q, dr) / Ldr6) / std::max(Ldr, 1e-10) * dr + \
                            3 * (dot(particles[j].q, dr) * particles[k].q + dot(particles[k].q, dr) * particles[j].q) / Ldr5;
                    }
                    //*/
                    // end velDipole //
 
                    // vel BS //
                    //buf_forces[j] += cross(particles[k].q, dr) / MyPow(std::max(abs(dr), FORCE_EPS), 3);
                    // end vel BS //
                }
            }
        }
 
 
 
        const auto& inner = leaves_inner.data() + leaf_idx * Nx2;
 
        //if (leaf_idx == 0)
        //{
        //  std::ofstream cftFile("coeffts.txt");
        //  cftFile.precision(16);
        //  for (int n = 1; n < N; ++n)
        //  {
        //      for (int m = 0; m <= n; ++m)
        //      {
        //          cftFile << n << " " << m << " " \
        //              << inner[n * (n + 1) / 2 + m][0].real() << " " \
        //              << inner[n * (n + 1) / 2 + m][0].imag() << "\n";
        //      }
        //  }
        //  cftFile.close();
        //  std::cout << "beg,end = " << p1_begin << ", " << p1_end << std::endl;
        //}
 
 
        std::complex<double> Ynm, SphDTheta, SphDPhi,
            SphDThetaTheta, SphDThetaPhi, SphDPhiPhi,
            SphDThetaThetaTheta, SphDThetaThetaPhi, SphDThetaPhiPhi, SphDPhiPhiPhi;
        Vector3cd coef, coefmulr;
        double x, y, rn, s1, s2, c1, c2, sd1, sd2, cd1, cd2;
 
        //if (leaf_idx == 0)
        //{
        //  std::ofstream coordFile("coord.txt");
        //  coordFile.precision(16);
        //  for (int j = p1_begin; j < p1_end; ++j)
        //  {
        //      coordFile << (particles[j].center - cell.center)[0] << " " \
        //          << (particles[j].center - cell.center)[1] << " " \
        //          << (particles[j].center - cell.center)[2] << " " \
        //          << DecToSph(particles[j].center - cell.center)[0] << " " \
        //          << DecToSph(particles[j].center - cell.center)[1] << " " \
        //          << DecToSph(particles[j].center - cell.center)[2] << "\n";
        //  }
        //  coordFile.close();
        //}
 
        for (auto j = p1_begin; j < p1_end; ++j)
        {
            /* //�������� �� ������ ��� --- ������
            auto& particle = particles[j];
            memset(m_temp.data(), 0, sizeof(Matrix3d));
            dr = DecToSph(particle.center - cell.center);
            ComputePnm(N + 1, dr[1], Pnm.data()); // +1 for computing derivative
 
            for (int m = 0; m <= N; ++m)
            {
                x = m * dr[2];
                eim[m] = std::complex<double>(cos(x), sin(x));
            }
            x = cos(dr[1]);
            y = sin(dr[1]);;
            rn = 1.;
 
 
            for (int n = 1; n < N; ++n)
            {
                for (int m = 0; m <= n; ++m)
                {
                    Ynm = Pnm(n, m) * eim[m] * Knm(n, m);
                    coef = inner[n * (n + 1) / 2 + m] * rn;
                    SphDTheta = Knm(n, m) * eim[m] * ((1 - m + n) * Pnm(n + 1, m) - (n + 1) * x * Pnm(n, m)) / y;
                    SphDPhi = double(m) * 1.0i * Ynm;
 
                    for (int s = 0; s < 3; ++s)
                    {
                        m_temp[s][0] -= (double(n) * coef[s] * Ynm).real();
                        m_temp[s][1] -= (coef[s] * SphDTheta).real() * dr[0];
                        m_temp[s][2] -= (coef[s] * SphDPhi).real() * dr[0];
                        if (m != 0)
                        {
                            m_temp[s][0] -= (double(n) * coef[s] * Ynm).real();
                            m_temp[s][1] -= (coef[s] * SphDTheta).real() * dr[0];
                            m_temp[s][2] -= (coef[s] * SphDPhi).real() * dr[0];
                        }
                    }
                }
                rn *= dr[0];
            }
 
            s1 = sin(dr[1]); c2 = cos(dr[2]); c1 = cos(dr[1]); s2 = sin(dr[2]);
            auto& f = buf_forces[j];
 
            Matrix3d m_copy;
            Matrix3d JacT;
            JacT[0] = { s1 * c2, c1 * c2 / dr[0], -s2 / (s1 * dr[0]) };
            JacT[1] = { s1 * s2, c1 * s2 / dr[0], c2 / (s1 * dr[0]) };
            JacT[2] = { c1, -s1 / dr[0], 0.0 };
 
            Matrix3d Jac;
            Jac[0] = { JacT[0][0], JacT[1][0], JacT[2][0] };
            Jac[1] = { JacT[0][1], JacT[1][1], JacT[2][1] };
            Jac[2] = { JacT[0][2], JacT[1][2], JacT[2][2] };
 
            //for (int s = 0; s < 3; ++s)
            //  mv(JacT, m_temp[s], m_copy[s]);
 
            mm(m_temp, Jac, m_copy); //�������������� ����������� ��� -> ���
 
            f[0] += m_copy[1][2] - m_copy[2][1]; //�-� ��� --- ������
            f[1] += m_copy[2][0] - m_copy[0][2];
            f[2] += m_copy[0][1] - m_copy[1][0];
 
            if (leaf_idx == 0)
                std::cout << "gr: " << m_copy[0][0] << ", " << m_copy[0][1] << ", " << m_copy[0][2] << std::endl;
            //*/ //�������� �� ������ ��� --- ������
 
 
 
            //* //�������� �������������� ��������
            auto& particle = particles[j];
            memset(m_temp.data(), 0, sizeof(Matrix3d));
            dr = DecToSph(particle.center - cell.center);
 
            /* //������������� �������� � DecToSph
            if (dr[0] < 1e-10)
                dr[0] = 1e-10;
            if (sin(dr[1]) < 1e-10)
            {
                if (cos(dr[1]) > 0)
                    dr[1] = 1e-10;
                else
                    dr[1] = 3.141592653589793 - 1e-10;
            }
            */
 
            ComputePnm(N + 3, dr[1], Pnm.data()); // +3 for computing 3rd derivative
 
            for (int m = 0; m <= N; ++m)
            {
                x = m * dr[2];
                eim[m] = std::complex<double>(cos(x), sin(x));
            }
            x = cos(dr[1]);
            y = sin(dr[1]);
            rn = 1.0;
 
            Matrix3d t_temp[3];
            memset(t_temp[0].data(), 0, 3*sizeof(Matrix3d));
 
            Matrix3d u_temp[3][3];
            memset(u_temp[0][0].data(), 0, 9 * sizeof(Matrix3d));
 
            for (int n = 1; n < N; ++n)
            {
                for (int m = 0; m <= n; ++m)
                {
                    Ynm = Pnm(n, m) * eim[m] * Knm(n, m);
                    coef = inner[n * (n + 1) / 2 + m] * rn;
                    coefmulr = coef * dr[0];
 
                    SphDTheta = Knm(n, m) * eim[m] * ((1 + n - m) * Pnm(n + 1, m) - (n + 1) * x * Pnm(n, m)) / y;
 
                    SphDPhi = double(m) * 1.0i * Ynm;
 
 
                    SphDThetaTheta = Knm(n, m) * eim[m] *
                        (
                            (m * m + (1 + n) * (2 + n) - (3 + 2 * n) * m) * Pnm(n + 2, m)
                            - 2.0 * (n + 2) * (1 + n - m) * x * Pnm(n + 1, m)
                            + (n + 1) * (1 + (n + 1) * x * x) * Pnm(n, m)
                        ) / (y * y);
 
                    SphDThetaPhi = SphDTheta * 1.0i * (double)m;
 
                    SphDPhiPhi = double(-m * m) * Ynm;
 
 
                    SphDThetaThetaTheta = Knm(n, m) * eim[m] *
                        (
                            (2 + n - m) * (1 + n - m) * ((3 + n - m) * Pnm(n + 3, m) - 3 * (3 + n) * x * Pnm(n + 2, m))
                            + 0.5 * ((3 * n * n + 18 * n + 23) + (3 * n * n + 12 * n + 13) * (x * x - y * y)) * (1 + n - m) * Pnm(n + 1, m)
                            - 0.5 * x * (n + 1) * ((n * n + 8 * n + 11) + (n + 1) * (n + 1) * (x * x - y * y)) * Pnm(n, m)
                        ) / (y * y * y);
 
                    SphDThetaThetaPhi = SphDThetaTheta * 1.0i * (double)m;
 
                    SphDThetaPhiPhi = double(-m * m) * SphDTheta;
 
                    SphDPhiPhiPhi = double(-m * m * m) * 1.0i * Ynm;
 
 
 
                    double times = (m ? 2.0 : 1.0);
 
                    for (int s = 0; s < 3; ++s)
                    {
                        m_temp[s][0] -= times * (double(n) * coef[s] * Ynm).real();
                        m_temp[s][1] -= times * (coefmulr[s] * SphDTheta).real();
                        m_temp[s][2] -= times * (coefmulr[s] * SphDPhi).real();
 
                        t_temp[s][0][0] -= times * (double(n * (n - 1)) * coef[s] * Ynm).real();
                        t_temp[s][0][1] -= times * (double(n) * coef[s] * SphDTheta).real();
                        t_temp[s][0][2] -= times * (double(n) * coef[s] * SphDPhi).real();
                        t_temp[s][1][1] -= times * (coefmulr[s] * SphDThetaTheta).real();
                        t_temp[s][1][2] -= times * (coefmulr[s] * SphDThetaPhi).real();
                        t_temp[s][2][2] -= times * (coefmulr[s] * SphDPhiPhi).real();
 
                        u_temp[s][0][0][0] -= times * (double(n * (n - 1) * (n - 2)) * coef[s] * Ynm).real();
                        u_temp[s][0][0][1] -= times * (double(n * (n - 1)) * coef[s] * SphDTheta).real();
                        u_temp[s][0][0][2] -= times * (double(n * (n - 1)) * coef[s] * SphDPhi).real();
                        u_temp[s][0][1][1] -= times * (double(n) * coef[s] * SphDThetaTheta).real();
                        u_temp[s][0][1][2] -= times * (double(n) * coef[s] * SphDThetaPhi).real();
                        u_temp[s][0][2][2] -= times * (double(n) * coef[s] * SphDPhiPhi).real();
                        u_temp[s][1][1][1] -= times * (coefmulr[s] * SphDThetaThetaTheta).real();
                        u_temp[s][1][1][2] -= times * (coefmulr[s] * SphDThetaThetaPhi).real();
                        u_temp[s][1][2][2] -= times * (coefmulr[s] * SphDThetaPhiPhi).real();
                        u_temp[s][2][2][2] -= times * (coefmulr[s] * SphDPhiPhiPhi).real();
                    }
                }
                rn *= dr[0];
            }
 
            for (int s = 0; s < 3; ++s)
            {
                if (dr[0] > 1e-10)
                {
                    t_temp[s][0][0] /= dr[0];
 
                    u_temp[s][0][0][0] /= (dr[0] * dr[0]);
                    u_temp[s][0][0][1] /= dr[0];
                    u_temp[s][0][0][2] /= dr[0];
                }
 
                t_temp[s][1][0] = t_temp[s][0][1];
                t_temp[s][2][0] = t_temp[s][0][2];
                t_temp[s][2][1] = t_temp[s][1][2];
 
                u_temp[s][0][1][0] = u_temp[s][1][0][0] = u_temp[s][0][0][1];
                u_temp[s][0][2][0] = u_temp[s][2][0][0] = u_temp[s][0][0][2];
                u_temp[s][1][1][0] = u_temp[s][1][0][1] = u_temp[s][0][1][1];
                u_temp[s][1][2][0] = u_temp[s][2][0][1] = u_temp[s][2][1][0] = u_temp[s][0][2][1] = u_temp[s][1][0][2] = u_temp[s][0][1][2];
                u_temp[s][2][2][0] = u_temp[s][2][0][2] = u_temp[s][0][2][2];
                u_temp[s][1][2][1] = u_temp[s][2][1][1] = u_temp[s][1][1][2];
                u_temp[s][2][1][2] = u_temp[s][2][2][1] = u_temp[s][1][2][2];
            }
 
            //if (leaf_idx == 0)
            //{
            //  std::cout << dr[0] << " " << dr[1] << " " << dr[2] << "\n";
            //}
 
            c1  = cos(dr[1]);       s1  = sin(dr[1]);       c2  = cos(dr[2]);       s2  = sin(dr[2]);
            cd1 = cos(2.0 * dr[1]); sd1 = sin(2.0 * dr[1]); cd2 = cos(2.0 * dr[2]); sd2 = sin(2.0 * dr[2]);
 
            double ct1 = cos(3.0 * dr[1]), ct2 = cos(3.0 * dr[2]), cq1 = cos(4.0 * dr[1]);
            double st1 = sin(3.0 * dr[1]), st2 = sin(3.0 * dr[2]);
 
            double s1sq = s1 * s1, c1sq = c1 * c1, c2sq = c2 * c2, s2sq = s2 * s2, drsq = dr[0] * dr[0], drcu = drsq * dr[0];
 
            auto& f = buf_forces[j];
            auto& m = buf_moments[j];
 
            //Matrix3d m_copy;
            Matrix3d JacT;
            JacT[0] = { s1 * c2, c1 * c2 / dr[0], -s2 / (s1 * dr[0]) };
            JacT[1] = { s1 * s2, c1 * s2 / dr[0],  c2 / (s1 * dr[0]) };
            JacT[2] = { c1, -s1 / dr[0], 0.0 };
 
            //�������������� ����������� ��� -> ���
            Matrix3d pm[3], t_copy[3];
            for (int s = 0; s < 3; ++s)
            {
                mm(JacT, t_temp[s], pm[s]);
                mmt(pm[s], JacT, t_copy[s]);
            }
 
            double izn = 1.0 / (s1sq * drsq);
 
            Matrix3d HCoordTransform[3];
            HCoordTransform[0][0] = { (c1sq + s1sq * s2sq) / dr[0], -(c2 * s1sq * s2) / dr[0], -(c1 * c2 * s1) / dr[0] };
            HCoordTransform[0][1] = { HCoordTransform[0][0][1], (c1sq + c2sq * s1sq) / dr[0], -(c1 * s1 * s2) / dr[0]};
            HCoordTransform[0][2] = { HCoordTransform[0][0][2], HCoordTransform[0][1][2], s1sq / dr[0]};
 
            HCoordTransform[1][0] = { c1 * (c2sq * cd1 - cd2) / (s1 * drsq), c1 * (-2.0 + cd1) * sd2 / (2.0 * s1 * drsq), -(c2 * cd1) / drsq };
            HCoordTransform[1][1] = { HCoordTransform[1][0][1], c1 * (cd2 + cd1 * s2sq) / (s1 * drsq), -cd1 * s2 / drsq };
            HCoordTransform[1][2] = { HCoordTransform[1][0][2], HCoordTransform[1][1][2], sd1 / drsq };
 
            HCoordTransform[2][0] = { sd2 * izn, -cd2 * izn, 0.0 };
            HCoordTransform[2][1] = { HCoordTransform[2][0][1], -HCoordTransform[2][0][0], 0.0 };
            HCoordTransform[2][2] = { 0.0, 0.0, 0.0 };
 
            // cos(theta)  -> c1,   sin(theta)  -> s1,   cos(phi)  -> c2,   sin(phi)  -> s2
            // cos(2theta) -> cd1,  sin(2theta) -> sd1,  cos(2phi) -> cd2,  sin(2phi) -> sd2
            Matrix3d TCoordTransform[3][3];
            {
                double A = -3 * c2 * s1 * (c1sq + s1sq * s2sq) / drsq;
                double P = -3 * s2 * s1 * (c1sq + s1sq * c2sq) / drsq;
                double B = s1 * s2 * (-1 + 3 * c2sq * s1sq) / drsq;
                double G = s1 * c2 * (-1 + 3 * s2sq * s1sq) / drsq;
                double F = c1 * (-c1sq + 0.5 * (1 + 3 * cd2) * s1sq) / drsq;
                double Q = c1 * (-c1sq + 0.5 * (1 - 3 * cd2) * s1sq) / drsq;
                double J = 0.5 * (1 + 3 * cd1) * s1 * c2 / drsq;
                double R = 0.5 * (1 + 3 * cd1) * s1 * s2 / drsq;
                double H = 3 * c1 * c2 * s1sq * s2 / drsq;
                double M = -3 * c1 * s1sq / drsq;
 
                TCoordTransform[0][0] = Matrix3d({ {A, B, F}, {B, G, H}, {F, H, J} });
                TCoordTransform[0][1] = Matrix3d({ {B, G, H}, {G, P, Q}, {H, Q, R} });
                TCoordTransform[0][2] = Matrix3d({ {F, H, J}, {H, Q, R}, {J, R, M} });
            }
 
            {
                double A = 0.5 * c2 * (-4 + cq1 + (8 - 6 * cd1 + cq1) * cd2) * c1 / (s1sq * drcu);
                double B = 0.25 * ((-2 * cd1 + cq1) * s2 + (8 - 6 * cd1 + cq1) * st2) * c1 / (s1sq * drcu);
                double G = (-cq1 * c2sq + cd1 * cd2) / (s1 * drcu);
                double D = (2 - 4 * cd2 + cd1 * (-2 + 3 * cd2) + cq1 * s2sq) * c2 * c1 / (s1sq * drcu);
                double E = (2 * cd1 - cq1) * c2 * s2 / (s1 * drcu);
                double H = 2 * ct1 * c2 / drcu;
                double K = 0.25 * (4 * cq1 * s2sq * s2 - 8 * st2 + 6 * cd1 * (-s2 + st2)) * c1 / (s1sq * drcu);
                double L = -(cd1 * cd2 + cq1 * s2sq) / (s1 * drcu);
                double M = 2 * ct1 * s2 / drcu;
                double N = -2 * st1 / drcu;
 
                TCoordTransform[1][0] = Matrix3d({ {A, B, G}, {B, D, E}, {G, E, H} });
                TCoordTransform[1][1] = Matrix3d({ {B, D, E}, {D, K, L}, {E, L, M} });
                TCoordTransform[1][2] = Matrix3d({ {G, E, H}, {E, L, M}, {H, M, N} });
            }
 
            {
                double iz = 2 / (s1sq * s1 * drcu);
                double P = iz * st2;
                double S = iz * ct2;
 
                TCoordTransform[2][0] = Matrix3d({ {-P, S, 0}, {S, P, 0}, {0, 0, 0} });
                TCoordTransform[2][1] = Matrix3d({ {S, P, 0}, {P, -S, 0}, {0, 0, 0} });
                TCoordTransform[2][2] = Matrix3d({ {0, 0, 0}, {0, 0, 0}, {0, 0, 0} });
            }
 
 
            Matrix3d add[3];
            for (int s = 0; s < 3; ++s)
            {
                vt(m_temp[s], HCoordTransform, add[s]);
                for (int ii = 0; ii < 3; ++ii)
                    t_copy[s][ii] += add[s][ii];
            }
 
            auto S1 = [&](int s, int ii, int jj, int kk)
                {
                    double res = 0;
                    for (int ll = 0; ll < 3; ++ll)
                        for (int mm = 0; mm < 3; ++mm)
                            for (int nn = 0; nn < 3; ++nn)
                                res += JacT[ii][ll] * JacT[jj][mm] * JacT[kk][nn] * u_temp[s][ll][mm][nn];
                    return res;
                };
 
            auto S2 = [&](int s, int ii, int jj, int kk)
                {
                    double res = 0;
                    for (int ll = 0; ll < 3; ++ll)
                        for (int mm = 0; mm < 3; ++mm)
                            res += (HCoordTransform[ll][ii][jj] * JacT[kk][mm] + \
                                HCoordTransform[ll][ii][kk] * JacT[jj][mm] + \
                                HCoordTransform[ll][jj][kk] * JacT[ii][mm]) * t_temp[s][ll][mm];
                    return res;
                };
 
            auto S3 = [&](int s, int ii, int jj, int kk)
                {
                    double res = 0;
                    for (int ll = 0; ll < 3; ++ll)
                        res += TCoordTransform[ll][ii][jj][kk] * m_temp[s][ll];
                    return res;
                };
 
            Matrix3d u_copy[3][3];
            memset(u_copy[0][0].data(), 0, 9 * sizeof(Matrix3d));
            for (int s = 0; s < 3; ++s)
            {
                for (int ii = 0; ii < 3; ++ii)
                    for (int jj = 0; jj < 3; ++jj)
                        for (int kk = 0; kk < 3; ++kk)
                        {
                            u_copy[s][ii][jj][kk] += S1(s, ii, jj, kk) + S2(s, ii, jj, kk) + S3(s, ii, jj, kk);
                        }
            }
 
 
            //if (fabs(t_copy[0][0][0] + t_copy[1][0][1] + t_copy[2][0][2]) > 1e-10)
            //  std::cout << "f: " << t_copy[0][0][0] << " " << t_copy[1][0][1] << " " << t_copy[2][0][2] << "\n";
 
 
            f[0] -= t_copy[0][0][0] + t_copy[1][0][1] + t_copy[2][0][2];
            f[1] -= t_copy[0][1][0] + t_copy[1][1][1] + t_copy[2][1][2];
            f[2] -= t_copy[0][2][0] + t_copy[1][2][1] + t_copy[2][2][2];
            //*/ //�������� �������������� ��������
 
            Matrix3d G3;
            for (int line = 0; line < 3; ++line)
                G3[line] = (u_copy[0][0][line] + u_copy[1][1][line] + u_copy[2][2][line]);
            Vector3d add_mom;
            mv(G3, particle.q, add_mom);
 
            m += add_mom;
        }
    });
 
    auto& buf_forces = *buf_forces_ets.begin();
    for (auto it = buf_forces_ets.begin() + 1; it < buf_forces_ets.end(); it++)
    {
        const auto& x = *it;
        tbb::parallel_for(tbb::blocked_range<size_t>(0, num_particles),
            [&](tbb::blocked_range<size_t> r) {
                for (auto i = r.begin(); i < r.end(); ++i)
                {
                    buf_forces[i] += x[i];
                }
            });
    }
 
    auto& buf_moments = *buf_moments_ets.begin();
    for (auto it = buf_moments_ets.begin() + 1; it < buf_moments_ets.end(); it++)
    {
        const auto& x = *it;
        tbb::parallel_for(tbb::blocked_range<size_t>(0, num_particles),
            [&](tbb::blocked_range<size_t> r) {
                for (auto i = r.begin(); i < r.end(); ++i)
                {
                    buf_moments[i] += x[i];
                }
            });
    }
 
 
#ifdef FMM_MPI
    AllReduce(buf_forces.data(), buf_forces.size());
    AllReduce(buf_moments.data(), buf_moments.size());
#endif
 
    const auto& index_mapping = tree->positions_map;
 
    tbb::parallel_for(size_t(0), targets_num, [&](size_t i) {
        forces[i] = buf_forces[index_mapping[i]];
        moments[i] = buf_moments[index_mapping[i]];
    });
}

Here is the call graph for this function:

ComputeLeaves() [2/3]

void fmm::FastMultipole3d< double >::ComputeLeaves ( )

private

Definition at line 1040 of file multipole3d.cpp.

{
    using namespace std::literals;
 
    tbb::enumerable_thread_specific<Vector3d> tmp_force_ets, dr_ets;
    tbb::enumerable_thread_specific<Vector3d> matrix_ets;
    tbb::enumerable_thread_specific<std::vector<std::complex<double>>> eim_ets(std::vector<std::complex<double>>(N + 1));
    tbb::enumerable_thread_specific<treevector<double>> Pnm_ets(treevector<double>(N + 1));
    tbb::enumerable_thread_specific<std::vector<Vector3d>> buf_forces_ets((std::vector<Vector3d>(num_particles)));
    tbb::enumerable_thread_specific<std::vector<double>> buf_potentials_ets((std::vector<double>(num_particles)));
 
    const auto& leaves = tree->levels.back();
    const auto& leaves_inner = inner_expansions.back();
    const auto& particles = tree->particles;
 
#ifdef FMM_MPI
    auto [Begin, End] = LocalPart(0, tree->level_sizes.back());
#else
    size_t Begin = 0, End = tree->level_sizes.back();
#endif
 
    decltype(&TreeCell_t::source_range) range_ptr;
    tree->targets_num == 0 ? range_ptr = &TreeCell_t::source_range : range_ptr = &TreeCell_t::target_range;
 
    using simd_vec_t = std::conditional_t<detail::avx_vec_length == 2, __m128d, std::conditional_t<detail::avx_vec_length == 4, __m256d, __m512d>>;
    simd_vec_t oneVec, epsVec;
    avx_set_constant(epsVec, FORCE_EPS2);
    avx_set_constant(oneVec, 1.0);
 
    std::unique_ptr<size_t[]> shifts(new size_t[tree->level_sizes.back() + 1]);
    shifts[0] = 0;
    for (size_t i = 0; i < tree->level_sizes.back(); ++i)
    {
        auto [a, b] = leaves[i].source_range;
        shifts[i + 1] = shifts[i] + (b - a + detail::avx_vec_length - 1) / detail::avx_vec_length;
    }
    size_t num_vecs = shifts[tree->level_sizes.back()];
    std::unique_ptr<simd_vec_t[]> x_simd(new simd_vec_t[num_vecs]),
                                  y_simd(new simd_vec_t[num_vecs]),
                                  z_simd(new simd_vec_t[num_vecs]),
                                  q_simd(new simd_vec_t[num_vecs]);
 
    double t1 = omp_get_wtime();
    tbb::parallel_for(size_t(0), tree->level_sizes.back(), [&](size_t i)
        {
            auto [a, b] = leaves[i].source_range;
            auto xptr = x_simd.get() + shifts[i];
            auto yptr = y_simd.get() + shifts[i];
            auto zptr = z_simd.get() + shifts[i];
            auto qptr = q_simd.get() + shifts[i];
            for (auto idx = a, k = (decltype(a))0; idx < b; idx += detail::avx_vec_length, ++k)
            {
#if defined(AVX128)
                xptr[k] = _mm_set_pd(idx + 2 > b ? 0 : particles[idx + 1].center[0], particles[idx].center[0]);
                yptr[k] = _mm_set_pd(idx + 2 > b ? 0 : particles[idx + 1].center[1], particles[idx].center[1]);
                zptr[k] = _mm_set_pd(idx + 2 > b ? 0 : particles[idx + 1].center[2], particles[idx].center[2]);
                qptr[k] = _mm_set_pd(idx + 2 > b ? 0 : particles[idx + 1].q, particles[idx].q);
#elif defined(AVX256)
                xptr[k] = _mm256_set_pd(idx + 4 > b ? 0 : particles[idx + 3].center[0],
                    idx + 3 > b ? 0 : particles[idx + 2].center[0],
                    idx + 2 > b ? 0 : particles[idx + 1].center[0], particles[idx].center[0]);
                yptr[k] = _mm256_set_pd(idx + 4 > b ? 0 : particles[idx + 3].center[1],
                    idx + 3 > b ? 0 : particles[idx + 2].center[1],
                    idx + 2 > b ? 0 : particles[idx + 1].center[1], particles[idx].center[1]);
                zptr[k] = _mm256_set_pd(idx + 4 > b ? 0 : particles[idx + 3].center[2],
                    idx + 3 > b ? 0 : particles[idx + 2].center[2],
                    idx + 2 > b ? 0 : particles[idx + 1].center[2], particles[idx].center[2]);
                qptr[k] = _mm256_set_pd(idx + 4 > b ? 0 : particles[idx + 3].q,
                    idx + 3 > b ? 0 : particles[idx + 2].q,
                    idx + 2 > b ? 0 : particles[idx + 1].q, particles[idx].q);
#elif defined(AVX512)
                xptr[k] = _mm512_set_pd(idx + 8 > b ? 0 : particles[idx + 7].center[0],
                    idx + 7 > b ? 0 : particles[idx + 6].center[0],
                    idx + 6 > b ? 0 : particles[idx + 5].center[0],
                    idx + 5 > b ? 0 : particles[idx + 4].center[0],
                    idx + 4 > b ? 0 : particles[idx + 3].center[0],
                    idx + 3 > b ? 0 : particles[idx + 2].center[0],
                    idx + 2 > b ? 0 : particles[idx + 1].center[0], particles[idx].center[0]);
                yptr[k] = _mm512_set_pd(idx + 8 > b ? 0 : particles[idx + 7].center[1],
                    idx + 7 > b ? 0 : particles[idx + 6].center[1],
                    idx + 6 > b ? 0 : particles[idx + 5].center[1],
                    idx + 5 > b ? 0 : particles[idx + 4].center[1],
                    idx + 4 > b ? 0 : particles[idx + 3].center[1],
                    idx + 3 > b ? 0 : particles[idx + 2].center[1],
                    idx + 2 > b ? 0 : particles[idx + 1].center[1], particles[idx].center[1]);
                zptr[k] = _mm512_set_pd(idx + 8 > b ? 0 : particles[idx + 7].center[2],
                    idx + 7 > b ? 0 : particles[idx + 6].center[2],
                    idx + 6 > b ? 0 : particles[idx + 5].center[2],
                    idx + 5 > b ? 0 : particles[idx + 4].center[2],
                    idx + 4 > b ? 0 : particles[idx + 3].center[2],
                    idx + 3 > b ? 0 : particles[idx + 2].center[2],
                    idx + 2 > b ? 0 : particles[idx + 1].center[2], particles[idx].center[2]);
                qptr[k] = _mm512_set_pd(idx + 8 > b ? 0 : particles[idx + 7].q,
                    idx + 7 > b ? 0 : particles[idx + 6].q,
                    idx + 6 > b ? 0 : particles[idx + 5].q,
                    idx + 5 > b ? 0 : particles[idx + 4].q,
                    idx + 4 > b ? 0 : particles[idx + 3].q,
                    idx + 3 > b ? 0 : particles[idx + 2].q,
                    idx + 2 > b ? 0 : particles[idx + 1].q, particles[idx].q);
#endif
        }
    });
 
    tbb::enumerable_thread_specific<std::vector<simd_vec_t>> fx_simd_ets((std::vector<simd_vec_t>(num_vecs))),
                                                             fy_simd_ets((std::vector<simd_vec_t>(num_vecs))),
                                                             fz_simd_ets((std::vector<simd_vec_t>(num_vecs))),
                                                             phi_simd_ets((std::vector<simd_vec_t>(num_vecs)));
 
    tbb::parallel_for(Begin, End, [&](size_t leaf_idx) {
        auto& tmp_force = tmp_force_ets.local();
        auto& dr = dr_ets.local();
        auto& eim = eim_ets.local();
        auto& Pnm = Pnm_ets.local();
        auto& m_temp = matrix_ets.local();
        auto& buf_forces = buf_forces_ets.local();
        auto& buf_potentials = buf_potentials_ets.local();
        auto& fx_simd = fx_simd_ets.local();
        auto& fy_simd = fy_simd_ets.local();
        auto& fz_simd = fz_simd_ets.local();
        auto& phi_simd = phi_simd_ets.local();
        double phi = 0.0;
 
        const auto& cell = leaves[leaf_idx];
        const auto& [p1_begin, p1_end] = cell.*range_ptr;
 
        simd_vec_t x_target, y_target, z_target, q_target,
                   x_source, y_source, z_source, q_source, dx, dy, dz, fx, fy, fz, pot, invdr, invdr2, q_invdr, q_invdr3;
 
        auto x_leaf = x_simd.get() + shifts[leaf_idx];
        auto y_leaf = y_simd.get() + shifts[leaf_idx];
        auto z_leaf = z_simd.get() + shifts[leaf_idx];
        auto q_leaf = q_simd.get() + shifts[leaf_idx];
 
        auto fx_leaf = fx_simd.data() + shifts[leaf_idx];
        auto fy_leaf = fy_simd.data() + shifts[leaf_idx];
        auto fz_leaf = fz_simd.data() + shifts[leaf_idx];
        auto phi_leaf = phi_simd.data() + shifts[leaf_idx];
 
        for (auto j = p1_begin; j < p1_end; ++j)
        {
            auto& particle = particles[j];
 
            if (range_ptr == &TreeCell_t::target_range)
            {
                avx_set_constant(x_target, particle.center[0]);
                avx_set_constant(y_target, particle.center[1]);
                avx_set_constant(z_target, particle.center[2]);
                avx_set_constant(q_target, particle.q);
                fx = avx_zero_vec<simd_vec_t>();
                fy = avx_zero_vec<simd_vec_t>();
                fz = avx_zero_vec<simd_vec_t>();
                pot = avx_zero_vec<simd_vec_t>();
 
                {
                    const auto& [p2_begin, p2_end] = cell.source_range;
                    for (auto k = p2_begin, i = (decltype(p2_begin))0; k < p2_end; k += detail::avx_vec_length, ++i)
                    {
                        x_source = x_leaf[i];
                        y_source = y_leaf[i];
                        z_source = z_leaf[i];
                        q_source = q_leaf[i];
                        avx_inv_dr(oneVec, epsVec, x_target, y_target, z_target, x_source, y_source, z_source, dx, dy, dz, invdr, invdr2);
                        q_invdr = avx_mul(q_source, invdr);
                        q_invdr3 = avx_mul(invdr2, q_invdr);
 
                        fx = avx_add(avx_mul(dx, q_invdr3), fx);
                        fy = avx_add(avx_mul(dy, q_invdr3), fy);
                        fz = avx_add(avx_mul(dz, q_invdr3), fz);
                        pot = avx_add(q_invdr, pot);
                    }
                }
 
                for (const auto& idx : cell.closeneighbours)
                {
                    const auto& neighbour_cell = leaves[idx];
                    const auto& [p2_begin, p2_end] = neighbour_cell.source_range;
 
                    auto x_nbr = x_simd.get() + shifts[idx];
                    auto y_nbr = y_simd.get() + shifts[idx];
                    auto z_nbr = z_simd.get() + shifts[idx];
                    auto q_nbr = q_simd.get() + shifts[idx];
 
                    for (auto k = p2_begin, i = (decltype(p2_begin))0; k < p2_end; k += detail::avx_vec_length, ++i)
                    {
                        x_source = x_nbr[i];
                        y_source = y_nbr[i];
                        z_source = z_nbr[i];
                        q_source = q_nbr[i];
                        avx_inv_dr(oneVec, epsVec, x_target, y_target, z_target, x_source, y_source, z_source, dx, dy, dz, invdr, invdr2);
                        q_invdr = avx_mul(q_source, invdr);
                        q_invdr3 = avx_mul(invdr2, q_invdr);
 
                        fx = avx_add(avx_mul(dx, q_invdr3), fx);
                        fy = avx_add(avx_mul(dy, q_invdr3), fy);
                        fz = avx_add(avx_mul(dz, q_invdr3), fz);
                        pot = avx_add(q_invdr, pot);
                    }
                }
            }
            else
            {
                avx_set_constant(x_target, particle.center[0]);
                avx_set_constant(y_target, particle.center[1]);
                avx_set_constant(z_target, particle.center[2]);
                avx_set_constant(q_target, particle.q);
                fx = avx_zero_vec<simd_vec_t>();
                fy = avx_zero_vec<simd_vec_t>();
                fz = avx_zero_vec<simd_vec_t>();
                pot = avx_zero_vec<simd_vec_t>();
 
                auto pletnum = (j - p1_begin) / detail::avx_vec_length;
                {
                    x_source = x_leaf[pletnum];
                    y_source = y_leaf[pletnum];
                    z_source = z_leaf[pletnum];
                    q_source = q_leaf[pletnum];
                    avx_inv_dr(oneVec, epsVec, x_target, y_target, z_target, x_source, y_source, z_source, dx, dy, dz, invdr, invdr2);
                    q_invdr = avx_mul(q_source, invdr);
                    q_invdr3 = avx_mul(invdr2, q_invdr);
 
                    fx = avx_add(avx_mul(dx, q_invdr3), fx);
                    fy = avx_add(avx_mul(dy, q_invdr3), fy);
                    fz = avx_add(avx_mul(dz, q_invdr3), fz);
                    pot = avx_add(q_invdr, pot);
                }
                pletnum++;
                for (auto k = p1_begin + pletnum * detail::avx_vec_length, i = pletnum; k < p1_end; k += detail::avx_vec_length, ++i)
                {
                    x_source = x_leaf[i];
                    y_source = y_leaf[i];
                    z_source = z_leaf[i];
                    q_source = q_leaf[i];
                    avx_inv_dr(oneVec, epsVec, x_target, y_target, z_target, x_source, y_source, z_source, dx, dy, dz, invdr, invdr2);
                    q_invdr = avx_mul(q_source, invdr);
                    q_invdr3 = avx_mul(invdr2, q_invdr);
 
                    fx = avx_add(avx_mul(dx, q_invdr3), fx);
                    fy = avx_add(avx_mul(dy, q_invdr3), fy);
                    fz = avx_add(avx_mul(dz, q_invdr3), fz);
                    pot = avx_add(q_invdr, pot);
 
                    q_invdr = avx_mul(q_target, invdr);
                    q_invdr3 = avx_mul(invdr2, q_invdr);
 
                    fx_leaf[i] = avx_sub(fx_leaf[i], avx_mul(dx, q_invdr3));
                    fy_leaf[i] = avx_sub(fy_leaf[i], avx_mul(dy, q_invdr3));
                    fz_leaf[i] = avx_sub(fz_leaf[i], avx_mul(dz, q_invdr3));
                    phi_leaf[i] = avx_add(q_invdr, phi_leaf[i]);
                }
 
                for (const auto& idx : cell.closeneighbours)
                {
                    const auto& neighbour_cell = leaves[idx];
                    const auto& [p2_begin, p2_end] = neighbour_cell.source_range;
 
                    auto x_nbr = x_simd.get() + shifts[idx];
                    auto y_nbr = y_simd.get() + shifts[idx];
                    auto z_nbr = z_simd.get() + shifts[idx];
                    auto q_nbr = q_simd.get() + shifts[idx];
 
                    auto fx_nbr = fx_simd.data() + shifts[idx];
                    auto fy_nbr = fy_simd.data() + shifts[idx];
                    auto fz_nbr = fz_simd.data() + shifts[idx];
                    auto phi_nbr = phi_simd.data() + shifts[idx];
 
                    for (auto k = p2_begin, i = (decltype(p2_begin))0; k < p2_end; k += detail::avx_vec_length, ++i)
                    {
                        x_source = x_nbr[i];
                        y_source = y_nbr[i];
                        z_source = z_nbr[i];
                        q_source = q_nbr[i];
                        avx_inv_dr(oneVec, epsVec, x_target, y_target, z_target, x_source, y_source, z_source, dx, dy, dz, invdr, invdr2);
                        q_invdr = avx_mul(q_source, invdr);
                        q_invdr3 = avx_mul(invdr2, q_invdr);
 
                        fx = avx_add(avx_mul(dx, q_invdr3), fx);
                        fy = avx_add(avx_mul(dy, q_invdr3), fy);
                        fz = avx_add(avx_mul(dz, q_invdr3), fz);
                        pot = avx_add(q_invdr, pot);
 
                        q_invdr = avx_mul(q_target, invdr);
                        q_invdr3 = avx_mul(invdr2, q_invdr);
 
                        fx_nbr[i] = avx_sub(fx_nbr[i], avx_mul(dx, q_invdr3));
                        fy_nbr[i] = avx_sub(fy_nbr[i], avx_mul(dy, q_invdr3));
                        fz_nbr[i] = avx_sub(fz_nbr[i], avx_mul(dz, q_invdr3));
                        phi_nbr[i] = avx_add(q_invdr, phi_nbr[i]);
                    }
                }
            }
            tmp_force[0] = avx_hsum(fx);
            tmp_force[1] = avx_hsum(fy);
            tmp_force[2] = avx_hsum(fz);
            buf_forces[j] += tmp_force;
            buf_potentials[j] += avx_hsum(pot);
        }
        const auto& inner = leaves_inner.data() + leaf_idx * Nx2;
        std::complex<double> Ynm, coef, SphDTheta, SphDPhi;
        double x, y, rn, s1, s2, c1, c2;
 
        for (auto j = p1_begin; j < p1_end; ++j)
        {
            auto& particle = particles[j];
            memset(m_temp.data(), 0, sizeof(Vector3d));
            dr = DecToSph(particle.center - cell.center);
 
            ComputePnm(N + 1, dr[1], Pnm.data()); // +1 for computing derivative
 
            for (int m = 0; m <= N; ++m)
            {
                x = m * dr[2];
                eim[m] = std::complex<double>(cos(x), sin(x));
            }
            x = cos(dr[1]);
            y = sin(dr[1]);;
            rn = 1. / dr[0];
 
            phi = 0.0;
 
            for (int n = 0; n < N; ++n)
            {
                for (int m = 0; m <= n; ++m)
                {
                    Ynm = Pnm(n, m) * eim[m] * Knm(n, m);
                    coef = inner[n * (n + 1) / 2 + m] * rn;
 
                    phi += (dr[0] * Ynm * coef).real();
                    if (m != 0)
                        phi += (dr[0] * Ynm * coef).real();
 
                    SphDTheta = Knm(n, m) * eim[m] * ((1 - m + n) * Pnm(n + 1, m) - (n + 1) * x * Pnm(n, m)) / y;
                    SphDPhi = double(m) * 1.0i * Ynm;
 
                    m_temp[0] -= (double(n) * coef * Ynm).real();
                    m_temp[1] -= (coef * SphDTheta).real();
                    m_temp[2] -= (coef * SphDPhi).real();
                    if (m != 0)
                    {
                        m_temp[0] -= (double(n) * coef * Ynm).real();
                        m_temp[1] -= (coef * SphDTheta).real();
                        m_temp[2] -= (coef * SphDPhi).real();
                    }
                }
                rn *= dr[0];
            }
            buf_potentials[j] += phi;
 
            s1 = sin(dr[1]); c2 = cos(dr[2]); c1 = cos(dr[1]); s2 = sin(dr[2]);
            auto& f = buf_forces[j];
            f[0] += m_temp[0] * s1 * c2 + m_temp[1] * c1 * c2 - m_temp[2] * s2 / s1;
            f[1] += m_temp[0] * s1 * s2 + m_temp[1] * c1 * s2 + m_temp[2] * c2 / s1;
            f[2] += m_temp[0] * c1 - m_temp[1] * s1;
        }
    });
    std::cout << omp_get_wtime() - t1 << std::endl;
 
    auto& buf_forces = *buf_forces_ets.begin();
    for (auto it = buf_forces_ets.begin() + 1; it < buf_forces_ets.end(); it++)
    {
        const auto& x = *it;
        tbb::parallel_for(tbb::blocked_range<size_t>(0, num_particles),
            [&](tbb::blocked_range<size_t> r) {
                for (auto i = r.begin(); i < r.end(); ++i)
                    buf_forces[i] += x[i];
            });
    }
 
    auto& buf_potentials = *buf_potentials_ets.begin();
    for (auto it = buf_potentials_ets.begin() + 1; it < buf_potentials_ets.end(); it++)
    {
        const auto& x = *it;
        tbb::parallel_for(tbb::blocked_range<size_t>(0, num_particles),
            [&](tbb::blocked_range<size_t> r) {
                for (auto i = r.begin(); i < r.end(); ++i)
                    buf_potentials[i] += x[i];
            });
    }
 
    auto& fx_simd = *fx_simd_ets.begin();
    auto& fy_simd = *fy_simd_ets.begin();
    auto& fz_simd = *fz_simd_ets.begin();
    auto& phi_simd = *phi_simd_ets.begin();
 
    tbb::parallel_for(size_t(0), tree->level_sizes.back(), [&](size_t i) {
        auto [a, b] = leaves[i].source_range;
        auto sz = (b - a + detail::avx_vec_length - 1) / detail::avx_vec_length;
 
        auto fx = fx_simd.data() + shifts[i];
        auto fy = fy_simd.data() + shifts[i];
        auto fz = fz_simd.data() + shifts[i];
        auto phi = phi_simd.data() + shifts[i];
        double bufVec[detail::avx_vec_length];
        for (int j = 0; j < sz; ++j)
        {
            for (auto it = fx_simd_ets.begin() + 1; it < fx_simd_ets.end(); it++) {
                const auto& x = it->data() + shifts[i];
                fx[j] = avx_add(fx[j], x[j]);
            }
            for (auto it = fy_simd_ets.begin() + 1; it < fy_simd_ets.end(); it++) {
                const auto& x = it->data() + shifts[i];
                fy[j] = avx_add(fy[j], x[j]);
            }
            for (auto it = fz_simd_ets.begin() + 1; it < fz_simd_ets.end(); it++) {
                const auto& x = it->data() + shifts[i];
                fz[j] = avx_add(fz[j], x[j]);
            }
            for (auto it = phi_simd_ets.begin() + 1; it < phi_simd_ets.end(); it++) {
                const auto& x = it->data() + shifts[i];
                phi[j] = avx_add(phi[j], x[j]);
            }
 
            auto k = a + detail::avx_vec_length * j;
            avx_store(bufVec, fx[j]);
            for (int s = 0; s < detail::avx_vec_length; ++s)
                if (k + s < b) buf_forces[k + s][0] += bufVec[s];
            avx_store(bufVec, fy[j]);
            for (int s = 0; s < detail::avx_vec_length; ++s)
                if (k + s < b) buf_forces[k + s][1] += bufVec[s];
            avx_store(bufVec, fz[j]);
            for (int s = 0; s < detail::avx_vec_length; ++s)
                if (k + s < b) buf_forces[k + s][2] += bufVec[s];
            avx_store(bufVec, phi[j]);
            for (int s = 0; s < detail::avx_vec_length; ++s)
                if (k + s < b) buf_potentials[k + s] += bufVec[s];
        }
    });
 
#ifdef FMM_MPI
    AllReduce(buf_forces.data(), buf_forces.size());
    AllReduce(buf_potentials.data(), buf_potentials.size());
#endif
 
    const auto& index_mapping = tree->positions_map;
 
    tbb::parallel_for(size_t(0), targets_num, [&](size_t i) {
        forces[i] = buf_forces[index_mapping[i]];
        potentials[i] = buf_potentials[index_mapping[i]];
    });
}

Here is the call graph for this function:

ComputeLeaves() [3/3]

template<typename value >

void fmm::FastMultipole3d< value >::ComputeLeaves ( )

private

Downward()

template<typename value >

void fmm::FastMultipole3d< value >::Downward ( )

private

Definition at line 323 of file multipole3d.cpp.

{
#ifdef FMM_MPI
    std::vector<int> sizes(NProc());
    std::vector<int> displs(NProc());
#endif
 
    for (int i = 2; i < tree->tree_depth; ++i)
    {
        auto& bottom_level = tree->levels[i];
        auto& bottom_inner = inner_expansions[i];
        const auto& bottom_outer = outer_expansions[i];
#ifdef FMM_MPI
        auto [Begin, End] = LocalPart(0, tree->level_sizes[i]);
#else
        size_t Begin = 0, End = tree->level_sizes[i];
#endif
        tbb::parallel_for(Begin, End, [&](size_t j) {
            TreeCell_t& cell = bottom_level[j];
            auto& work1 = local_work1.local();
            auto& work2 = local_work2.local();
 
            for (const auto& idx : cell.farneighbours)
            {
                const auto& far_neighbour = bottom_level[idx];
                M2L(bottom_outer.data() + idx * Nx2, far_neighbour.center - cell.center, bottom_inner.data() + Nx2 * j, work1, work2);
            }
        });
 
#ifdef FMM_MPI
        {
            MPI_Datatype MPI_COMPLEX_VALUE;
            MPI_Type_contiguous(sizeof(complex_value) / sizeof(std::complex<double>), MPI_COMPLEX16, &MPI_COMPLEX_VALUE);
            MPI_Type_commit(&MPI_COMPLEX_VALUE);
 
            std::vector<complex_value> buf(bottom_inner.size());
            for (int j = 0; j < NProc(); ++j)
            {
                sizes[j] = Nx2 * LocalPart(0, tree->level_sizes[i], j);
            }
            for (int j = 1; j < NProc(); ++j)
            {
                displs[j] = displs[j - 1] + sizes[j - 1];
            }
            MPI_Allgatherv(bottom_inner.data() + displs[MyID()], sizes[MyID()], MPI_COMPLEX_VALUE,
                buf.data(), sizes.data(), displs.data(), MPI_COMPLEX_VALUE, MPI_COMM_WORLD);
            std::swap(buf, bottom_inner);
        }
#endif
    }
 
    for (int i = 3; i < tree->tree_depth; ++i)
    {
        const auto& top_level = tree->levels[i - 1];
        const auto& top_inner = inner_expansions[i - 1];
        auto& bottom_level = tree->levels[i];
        auto& bottom_inner = inner_expansions[i];
#ifdef FMM_MPI
        auto [Begin, End] = LocalPart(0, tree->level_sizes[i]);
#else
        size_t Begin = 0, End = tree->level_sizes[i];
#endif
        tbb::parallel_for(Begin, End, [&](size_t j) {
            TreeCell_t& cell = bottom_level[j];
            auto& work1 = local_work1.local();
            auto& work2 = local_work2.local();
            auto& parent_cell = top_level[cell.parent];
            L2L(top_inner.data() + Nx2 * cell.parent, parent_cell.center - cell.center, bottom_inner.data() + Nx2 * j, work1, work2);
        });
 
#ifdef FMM_MPI
        {
            MPI_Datatype MPI_COMPLEX_VALUE;
            MPI_Type_contiguous(sizeof(complex_value) / sizeof(std::complex<double>), MPI_COMPLEX16, &MPI_COMPLEX_VALUE);
            MPI_Type_commit(&MPI_COMPLEX_VALUE);
 
            std::vector<complex_value> buf(bottom_inner.size());
            for (int j = 0; j < NProc(); ++j)
            {
                sizes[j] = Nx2 * LocalPart(0, tree->level_sizes[i], j);
            }
            for (int j = 1; j < NProc(); ++j)
            {
                displs[j] = displs[j - 1] + sizes[j - 1];
            }
            MPI_Allgatherv(bottom_inner.data() + displs[MyID()], sizes[MyID()], MPI_COMPLEX_VALUE,
                buf.data(), sizes.data(), displs.data(), MPI_COMPLEX_VALUE, MPI_COMM_WORLD);
            std::swap(buf, bottom_inner);
        }
#endif
    }
}

L2L()

template<typename value >

void fmm::FastMultipole3d< value >::L2L ( const complex_value *  a, const Vector3d &  p0, complex_value *  b, std::vector< complex_value > &  work1, std::vector< complex_value > &  work2  )

private

Definition at line 195 of file multipole3d.cpp.

{
    auto rho = DecToSph(p0);
    const auto& theta = rho[1];
    const auto& phi = rho[2];
 
    const auto& dm1 = dmatrix.at(doublehash(theta));
    const auto& dm2 = dmatrix.at(doublehash(-theta));
    const auto& exp1 = rotation_exponents.at(doublehash(phi));
    const auto& exp2 = rotation_exponents.at(doublehash(-phi));
 
    memset(work2.data(), 0, work2.size() * sizeof(complex_value));
    RotateZ(a, exp1, work2.data());
    RotateY(work2.data(), dm1, work1.data());
    memset(work2.data(), 0, work2.size() * sizeof(complex_value));
 
    std::vector<double> rn(N, 1);
    for (int i = 1; i < N; ++i)
        rn[i] = rn[i - 1] * rho[0];
 
    for (int j = 0; j < N; ++j)
    {
        for (int k = 0; k <= j; ++k)
        {
            for (int n = j; n < N; ++n)
            {
                work2[j * (j + 1) / 2 + k] += Anm(n - j, 0) * Anm(j, k) * ni(n + j) * rn[n - j] / Anm(n, k) * work1[n * (n + 1) / 2 + k];
            }
        }
    }
 
    RotateY(work2.data(), dm2, work1.data());
    RotateZ(work1.data(), exp2, b);
}

Here is the call graph for this function:

M2L()

template<typename value >

void fmm::FastMultipole3d< value >::M2L ( const complex_value *  a, const Vector3d &  p0, complex_value *  b, std::vector< complex_value > &  work1, std::vector< complex_value > &  work2  )

private

Definition at line 155 of file multipole3d.cpp.

{
    auto rho = DecToSph(p0);
    const auto& theta = rho[1];
    const auto& phi = rho[2];
 
    const auto& dm1 = dmatrix.at(doublehash(theta));
    const auto& dm2 = dmatrix.at(doublehash(-theta));
    const auto& exp1 = rotation_exponents.at(doublehash(phi));
    const auto& exp2 = rotation_exponents.at(doublehash(-phi));
 
    memset(work2.data(), 0, work2.size() * sizeof(complex_value));
    RotateZ(a, exp1, work2.data());
    RotateY(work2.data(), dm1, work1.data());
    memset(work2.data(), 0, work2.size() * sizeof(complex_value));
 
    std::vector<double> rn(2 * N + 1, 1);
    for (int i = 1; i < 2 * N + 1; ++i)
        rn[i] = rn[i - 1] / rho[0];
 
    int idx1, idx2, idx3;
    for (int j = 0; j < N; ++j)
    {
        idx1 = (j + 2) * (j + 1) / 2;
        for (int k = 0; k <= j; ++k)
        {
            idx2 = detail::_3d_MAX_MULTIPOLE_NUM * (idx1 + k + 1) + 1;
            for (int n = k; n < N; ++n)
            {
                idx3 = idx2 + n;
                work2[j * (j + 1) / 2 + k] += (m2lcoef.m2lcoef[idx3] * rn[j + n + 1]) * work1[n * (n + 1) / 2 + k];
            }
        }
    }
 
    RotateY(work2.data(), dm2, work1.data());
    RotateZ(work1.data(), exp2, b);
}

Here is the call graph for this function:

M2M()

template<typename value >

void fmm::FastMultipole3d< value >::M2M ( const complex_value *  a, const Vector3d &  p0, complex_value *  b, std::vector< complex_value > &  work1, std::vector< complex_value > &  work2  )

private

Definition at line 118 of file multipole3d.cpp.

{
    auto rho = DecToSph(p0);
    const auto& theta = rho[1];
    const auto& phi = rho[2];
 
    const auto& dm1 = dmatrix.at(doublehash(theta));
    const auto& dm2 = dmatrix.at(doublehash(-theta));
    const auto& exp1 = rotation_exponents.at(doublehash(phi));
    const auto& exp2 = rotation_exponents.at(doublehash(-phi));
 
    memset(work2.data(), 0, work2.size() * sizeof(complex_value));
    RotateZ(a, exp1, work2.data());
    RotateY(work2.data(), dm1, work1.data());
    memset(work2.data(), 0, work2.size() * sizeof(complex_value));
 
    std::vector<double> rn(N, 1);
    for (int i = 1; i < N; ++i)
        rn[i] = rn[i - 1] * rho[0];
 
    for (int j = 0; j < N; ++j)
    {
        for (int k = 0; k <= j; ++k)
        {
            for (int n = 0; j - n >= k; ++n)
            {
                int j_minus_n = j - n;
                work2[j * (j + 1) / 2 + k] += Anm(n, 0) * Anm(j_minus_n, k) * rn[n] / Anm(j, k) * work1[j_minus_n * (j_minus_n + 1) / 2 + k];
            }
        }
    }
 
    RotateY(work2.data(), dm2, work1.data());
    RotateZ(work1.data(), exp2, b);
}

Here is the call graph for this function:

Multipole()

template<typename value >

void fmm::FastMultipole3d< value >::Multipole ( complex_value *  a, const std::pair< size_t, size_t > &  particle_range, const Vector3d &  z0, double *  Pnm, std::complex< double > *  eim  )

private

Definition at line 23 of file multipole3d.cpp.

{
    auto [p_begin, p_end] = particle_range;
    Vector3d rho;
    double r, rn, mphi;
 
    for (auto i = p_begin; i < p_end; ++i)
    {
        auto& particle = tree->particles[i];
        rho = DecToSph(particle.center - p0);
        rn = r = rho[0];
        ComputePnm(N, rho[1], Pnm);
        for (int m = 0; m < N; ++m)
        {
            mphi = m * rho[2];
            eim[m] = std::complex<double>(cos(mphi), -sin(mphi));
        }
 
        for (int n = 1; n < N; ++n)
        {
            for (int m = 0; m <= n; ++m)
            {
                a[n * (n + 1) / 2 + m] += particle.q * rn * Knm(n, m) * Pnm[n * (n + 1)/ 2 + m]* eim[m];
            }
            rn *= r;
        }
        a[0] += particle.q;
    }
}

Here is the call graph for this function:

Prepare()

template<typename value >

void fmm::FastMultipole3d< value >::Prepare ( )

private

Definition at line 1568 of file multipole3d.cpp.

{
    local_work1 = tbb::enumerable_thread_specific<std::vector<complex_value>>{ std::vector<complex_value>(Nx2) };
    local_work2 = tbb::enumerable_thread_specific<std::vector<complex_value>>{ std::vector<complex_value>(Nx2) };
 
    /*m2lcoef.resize(2 * N * N * N);
    for (int j = 0; j < N; ++j)
    {
        for (int k = 0; k <= j; ++k)
        {
            for (int n = k; n < N; ++n)
            {
                int idx = N * (j + 2) * (j + 1) / 2 + N * (k + 1) + n + 1;
                m2lcoef[idx] = Anm(n, k) * Anm(j, k) * ni(j + k) / Anm(j + n, 0);
            }
        }
    }*/
}

RotateY()

template<typename value >

void fmm::FastMultipole3d< value >::RotateY ( const complex_value *  a, const std::vector< double > &  dmatrix, complex_value *  res  )

private

Definition at line 82 of file multipole3d.cpp.

{
    int idx0, idx1, idx2;
    complex_value val;
    for (int n = 0; n < N; ++n)
    {
        idx0 = (n * (5 + n * (3 + 4 * n))) / 6;
        idx1 = 2 * n + 1;
        for (int m = 0; m <= n; ++m)
        {
            idx2 = idx0 + idx1 * m;
            if constexpr (std::is_same_v<value, double>)
                val = 0.0;
            else
                val = { 0,0,0 };
            for (int k = 1; k <= n; ++k)
            {
                const auto z1 = dmatrix[idx2 + k];
                const auto z2 = dmatrix[idx2 - k];
                const auto w = a[n * (n + 1) / 2 + k];
                if constexpr (std::is_same_v<value, double>)
                {
                    val.real(w.real() * (z1 + z2) + val.real());
                    val.imag(w.imag() * (z1 - z2) + val.imag());
                }
                else
                    for (int s = 0; s < 3; ++s)
                        val[s] += std::complex<double>(w[s].real() * (z1 + z2), w[s].imag() * (z1 - z2));
            }
            val += dmatrix[idx2] * a[n * (n + 1) / 2];
            res[n * (n + 1) / 2 + m] = val;
        }
    }
}

Here is the call graph for this function:

RotateZ() [1/2]

template<typename value >

void fmm::FastMultipole3d< value >::RotateZ ( complex_value *  a, const std::vector< std::complex< double > > &  rotation_exp  )

private

Definition at line 54 of file multipole3d.cpp.

{
    int idx;
    for (int n = 0; n < N; ++n)
    {
        idx = n * (n + 1) / 2;
        for (int m = 0; m <= n; ++m)
        {
            a[idx + m] *= rotation_exp[m];
        }
    }
}

RotateZ() [2/2]

template<typename value >

void fmm::FastMultipole3d< value >::RotateZ ( const complex_value *  a, const std::vector< std::complex< double > > &  rotation_exp, complex_value *  b  )

private

Definition at line 68 of file multipole3d.cpp.

{
    int idx;
    for (int n = 0; n < N; ++n)
    {
        idx = n * (n + 1) / 2;
        for (int m = 0; m <= n; ++m)
        {
            b[idx + m] += a[idx + m] * rotation_exp[m];
        }
    }
}

Solve()

template<typename value >

void fmm::FastMultipole3d< value >::Solve ( std::vector< particle_t > &&  particles, double  eps = 1.e-8, int  N = FMM_AUTO, int  tree_depth = FMM_AUTO  )

private

Definition at line 1506 of file multipole3d.cpp.

{
    double T = omp_get_wtime();
    // if (IAmRoot()) std::cout << "\n***************** Start FMM **************" << std::endl;
    num_particles = particles.size();
    eps = std::max(eps, 1.e-11);
    if (N_ == FMM_AUTO)
        N = std::min(fmm::detail::_3d_MAX_MULTIPOLE_NUM, int(54.2 - 10.73 * sqrt(log(2.4 * eps) + 25.5)));
    else
        N = std::min(detail::_3d_MAX_MULTIPOLE_NUM, N_);
    Nx2 = N * (N + 1) / 2;
    // if (IAmRoot()) std::cout << "multipole num = " << N << std::endl;
 
    size_t initial_depth;
    if (tree_depth_ == FMM_AUTO)
        initial_depth = (size_t)std::max(2.0, 1 + log(double(num_particles) / 100.0) / log(8.0));
    else
        initial_depth = tree_depth_;
    // if (IAmRoot()) std::cout << "tree depth = " << initial_depth << std::endl;
 
    tree = std::make_shared<MortonTree_t>(std::move(particles), initial_depth);
    tree_depth = tree->tree_depth;
 
    if (tree_depth < initial_depth)
        std::cout << "Warning: tree_depth was reduced from " << initial_depth << " to " << tree_depth << " to ensure H_CELL < EPS_FORCE" << std::endl;
 
    double t = omp_get_wtime();
    outer_expansions.resize(tree_depth);
    inner_expansions.resize(tree_depth);
    for (int i = 0; i < tree->tree_depth; ++i)
    {
        auto& outer = outer_expansions[i];
        auto& inner = inner_expansions[i];
        outer.resize(tree->level_sizes[i] * Nx2);
        inner.resize(tree->level_sizes[i] * Nx2);
    }
    forces.resize(targets_num);
    moments.resize(targets_num);
    if constexpr (std::is_same_v<value, double>)
        potentials.resize(targets_num);
 
    //if (IAmRoot()) std::cout << "allocation time: " << omp_get_wtime() - t << std::endl;
 
    t = omp_get_wtime();
    Prepare();
    //if (IAmRoot()) std::cout << "prepare time: " << omp_get_wtime() - t << std::endl;
 
    Upward();
 
    t = omp_get_wtime();
    Downward();
    //if (IAmRoot()) std::cout << "m2l+l2l time: " << omp_get_wtime() - t << std::endl;
 
    t = omp_get_wtime();
    ComputeLeaves();
    //if (IAmRoot()) std::cout << "leaf time: " << omp_get_wtime() - t << std::endl;
 
    // if (IAmRoot()) std::cout << "total fmm time: " << omp_get_wtime() - T << std::endl;
    // if (IAmRoot()) std::cout << "***************** End FMM **************\n" << std::endl;
}

Upward()

template<typename value >

void fmm::FastMultipole3d< value >::Upward ( )

private

Definition at line 231 of file multipole3d.cpp.

{
    double t1 = omp_get_wtime();
 
    auto& leaves = tree->levels.back();
    auto& leaves_outer = outer_expansions.back();
#ifdef FMM_MPI
    auto [Begin, End] = LocalPart(0, tree->level_sizes[tree->tree_depth - 1]);
#else
    size_t Begin = 0, End = tree->level_sizes[tree->tree_depth - 1];
#endif
    tbb::enumerable_thread_specific<std::vector<std::complex<double>>> eim_ets((std::vector<std::complex<double>>(N)));
    tbb::enumerable_thread_specific<treevector<double>> Pnm_ets(treevector<double>(N + 1));
    tbb::parallel_for(Begin, End, [&](size_t i) {
        TreeCell_t& cell = leaves[i];
        auto& eim = eim_ets.local();
        auto& Pnm = Pnm_ets.local();
        Multipole(leaves_outer.data() + Nx2 * i, cell.source_range, cell.center, Pnm.data(), eim.data());
        });
 
#ifdef FMM_MPI
    std::vector<complex_value> buf(leaves_outer.size());
    std::vector<int> sizes(NProc());
    std::vector<int> displs(NProc());
    for (int i = 0; i < NProc(); ++i)
    {
        sizes[i] = Nx2 * LocalPart(0, tree->level_sizes[tree->tree_depth - 1], i);
    }
    for (int i = 1; i < NProc(); ++i)
    {
        displs[i] = displs[i - 1] + sizes[i - 1];
    }
 
    MPI_Datatype MPI_COMPLEX_VALUE;
    MPI_Type_contiguous(sizeof(complex_value) / sizeof(std::complex<double>), MPI_COMPLEX16, &MPI_COMPLEX_VALUE);
    MPI_Type_commit(&MPI_COMPLEX_VALUE);
 
    MPI_Allgatherv(leaves_outer.data() + displs[MyID()], sizes[MyID()], MPI_COMPLEX_VALUE,
        buf.data(), sizes.data(), displs.data(), MPI_COMPLEX_VALUE, MPI_COMM_WORLD);
    std::swap(buf, leaves_outer);
#endif
 
    //if (IAmRoot()) std::cout << "multipole time: " << omp_get_wtime() - t1 << std::endl;
 
    t1 = omp_get_wtime();
 
    for (int i = (int)tree->tree_depth - 2; i >= 0; --i)
    {
        auto& top_level = tree->levels[i];
        auto& top_outer = outer_expansions[i];
        const auto& bottom_level = tree->levels[i + 1];
        const auto& bottom_outer = outer_expansions[i + 1];
#ifdef FMM_MPI
        auto [Begin, End] = LocalPart(0, tree->level_sizes[i]);
#else
        size_t Begin = 0, End = tree->level_sizes[i];
#endif
        tbb::parallel_for(Begin, End, [&](size_t j) {
            TreeCell_t& cell = top_level[j];
            auto& work1 = local_work1.local();
            auto& work2 = local_work2.local();
            const auto& [begin, end] = cell.source_range;
 
            for (auto k = begin; k < end; ++k)
            {
                const auto& children_cell = bottom_level[k];
                M2M(bottom_outer.data() + k * Nx2, children_cell.center - cell.center, top_outer.data() + j * Nx2, work1, work2);
            }
        });
 
#ifdef FMM_MPI
        {
            std::vector<complex_value> buf(top_outer.size());
            for (int j = 0; j < NProc(); ++j)
            {
                sizes[j] = Nx2 * LocalPart(0, tree->level_sizes[i], j);
            }
            for (int j = 1; j < NProc(); ++j)
            {
                displs[j] = displs[j - 1] + sizes[j - 1];
            }
            MPI_Allgatherv(top_outer.data() + displs[MyID()], sizes[MyID()], MPI_COMPLEX_VALUE,
                buf.data(), sizes.data(), displs.data(), MPI_COMPLEX_VALUE, MPI_COMM_WORLD);
            std::swap(buf, top_outer);
        }
#endif
    }
 
    //if (IAmRoot()) std::cout << "m2m time: " << omp_get_wtime() - t1 << std::endl;
}

Member Data Documentation

dipolevelocities

template<typename value >

std::vector<Vector3d> fmm::FastMultipole3d< value >::dipolevelocities

Definition at line 26 of file multipole3d.h.

forces

template<typename value >

std::vector<Vector3d> fmm::FastMultipole3d< value >::forces

Definition at line 23 of file multipole3d.h.

inner_expansions

template<typename value >

std::vector<std::vector<complex_value> > fmm::FastMultipole3d< value >::inner_expansions

private

Definition at line 47 of file multipole3d.h.

local_work1

template<typename value >

tbb::enumerable_thread_specific<std::vector<complex_value> > fmm::FastMultipole3d< value >::local_work1

private

Definition at line 56 of file multipole3d.h.

local_work2

template<typename value >

tbb::enumerable_thread_specific<std::vector<complex_value> > fmm::FastMultipole3d< value >::local_work2

private

Definition at line 57 of file multipole3d.h.

moments

template<typename value >

std::vector<Vector3d> fmm::FastMultipole3d< value >::moments

Definition at line 24 of file multipole3d.h.

N

template<typename value >

int fmm::FastMultipole3d< value >::N

private

Definition at line 50 of file multipole3d.h.

num_particles

template<typename value >

size_t fmm::FastMultipole3d< value >::num_particles

private

Definition at line 52 of file multipole3d.h.

Nx2

template<typename value >

int fmm::FastMultipole3d< value >::Nx2

private

Definition at line 51 of file multipole3d.h.

outer_expansions

template<typename value >

std::vector<std::vector<complex_value> > fmm::FastMultipole3d< value >::outer_expansions

private

Definition at line 46 of file multipole3d.h.

potentials

template<typename value >

std::vector<double> fmm::FastMultipole3d< value >::potentials

Definition at line 25 of file multipole3d.h.

targets_num

template<typename value >

size_t fmm::FastMultipole3d< value >::targets_num

private

Definition at line 53 of file multipole3d.h.

tree

template<typename value >

std::shared_ptr<MortonTree_t> fmm::FastMultipole3d< value >::tree

private

Definition at line 45 of file multipole3d.h.

tree_depth

template<typename value >

size_t fmm::FastMultipole3d< value >::tree_depth

private

Definition at line 49 of file multipole3d.h.

---

The documentation for this class was generated from the following files:

• VMfmm/cpu/multipole3d.h
• VMfmm/cpu/multipole3d.cpp