heat.cpp

Go to the documentation of this file.

/// @file backends/omp/heat.cpp
/// @brief OpenMP heat diffusion backend
 
#include "impl.hpp"
#include "backends/seq/impl.hpp"
#include "kernel.hpp"
#include <cmath>
#include <algorithm>
 
#ifdef NUMERICS_HAS_OMP
#  include <omp.h>
#endif
 
namespace physics::backends::omp {
 
void heat_compute(std::vector<Particle>& particles,
                  const std::vector<RigidBody>& bodies,
                  const SpatialHash& grid,
                  const HeatParams& params) {
#ifdef NUMERICS_HAS_OMP
    const float h       = params.h;
    const float alpha   = params.alpha_T;
    const float m       = params.mass;
    const float h_conv  = params.h_conv;
    const float supp_sq = 4.0f * h * h;
    const float eps2    = 0.01f * h * h;
    const int   n       = static_cast<int>(particles.size());
 
#   pragma omp parallel for schedule(static)
    for (int i = 0; i < n; ++i) {
        Particle& pi = particles[i];
        float dT = 0.0f;
 
        grid.query(pi.x, pi.y, [&](int j) {
            if (j == i) return;
            const Particle& pj = particles[j];
            const float rx = pi.x - pj.x, ry = pi.y - pj.y;
            const float r2 = rx * rx + ry * ry;
            if (r2 >= supp_sq || r2 < 1e-10f) return;
            const float r             = std::sqrt(r2);
            const float rij_dot_gradW = Kernel::dW_dr(r, h) * r;
            dT += 2.0f * alpha * (m / pj.density)
                  * (pi.temperature - pj.temperature)
                  * rij_dot_gradW / (r2 + eps2);
        });
 
        for (const RigidBody& body : bodies) {
            const float dx = pi.x - body.x, dy = pi.y - body.y;
            const float d  = std::sqrt(dx * dx + dy * dy);
            if (d < body.radius + 2.0f * h) {
                const float excess = std::max(0.0f, d - body.radius);
                const float phi    = 1.0f - excess / (2.0f * h);
                dT += h_conv * (body.temperature - pi.temperature) * phi;
            }
        }
 
        pi.dT_dt = dT;
    }
#else
    seq::heat_compute(particles, bodies, grid, params);
#endif
}
 
} // namespace physics::backends::omp