Skip to content

Commit

Permalink
fix bug
Browse files Browse the repository at this point in the history
  • Loading branch information
@lucafedeli88
lucafedeli88 committed May 6, 2024
1 parent 31bfd4d commit 58ee62e
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 49 deletions.
8 changes: 0 additions & 8 deletions Source/Particles/Radiation/RadiationHandler.H
View file
Original file line number Diff line number Diff line change
Expand Up @@ -82,14 +82,6 @@ private:
amrex::Gpu::DeviceVector<amrex::Real> m_det_n_y;
amrex::Gpu::DeviceVector<amrex::Real> m_det_n_z;

amrex::Gpu::DeviceVector<amrex::Real> m_b_x;
amrex::Gpu::DeviceVector<amrex::Real> m_b_y;
amrex::Gpu::DeviceVector<amrex::Real> m_b_z;

amrex::Gpu::DeviceVector<amrex::Real> m_bp_x;
amrex::Gpu::DeviceVector<amrex::Real> m_bp_y;
amrex::Gpu::DeviceVector<amrex::Real> m_bp_z;

amrex::Gpu::DeviceVector<ablastr::math::Complex> m_radiation_data;
amrex::Gpu::DeviceVector<amrex::Real> m_radiation_calculation;

Expand Down
56 changes: 15 additions & 41 deletions Source/Particles/Radiation/RadiationHandler.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -440,41 +440,6 @@ void RadiationHandler::add_radiation_contribution(

WARPX_ALWAYS_ASSERT_WITH_MESSAGE((np-1) == static_cast<int>(np-1), "too many particles!");


m_b_x.resize(np);
m_b_y.resize(np);
m_b_z.resize(np);
m_bp_x.resize(np);
m_bp_y.resize(np);
m_bp_z.resize(np);

auto* p_b_x = m_b_x.dataPtr();
auto* p_b_y = m_b_y.dataPtr();
auto* p_b_z = m_b_z.dataPtr();
auto* p_bp_x = m_bp_x.dataPtr();
auto* p_bp_y = m_bp_y.dataPtr();
auto* p_bp_z = m_bp_z.dataPtr();


amrex::ParallelFor(np, [=] AMREX_GPU_DEVICE(int ip){
const auto ux = 0.5_prt*(p_ux[ip] + p_ux_old[ip]);
const auto uy = 0.5_prt*(p_uy[ip] + p_uy_old[ip]);
const auto uz = 0.5_prt*(p_uz[ip] + p_uz_old[ip]);

auto const u2 = ux*ux + uy*uy + uz*uz;

auto const one_over_gamma = 1._prt/std::sqrt(1.0_rt + u2*inv_c2);
auto const one_over_gamma_c = one_over_gamma*inv_c;
const auto one_over_dt_gamma_c = one_over_gamma_c/dt;

p_b_x[ip] = ux*one_over_gamma_c;
p_b_y[ip] = uy*one_over_gamma_c;
p_b_z[ip] = uz*one_over_gamma_c;
p_bp_x[ip] = (p_ux[ip] - p_ux_old[ip])*one_over_dt_gamma_c;
p_bp_y[ip] = (p_uy[ip] - p_uy_old[ip])*one_over_dt_gamma_c;
p_bp_z[ip] = (p_uz[ip] - p_uz_old[ip])*one_over_dt_gamma_c;
});

const auto np_times_det_pos = np*how_many_det_pos;

amrex::ParallelFor(
Expand All @@ -485,13 +450,22 @@ void RadiationHandler::add_radiation_contribution(
amrex::ParticleReal xp, yp, zp;
GetPosition.AsStored(ip, xp, yp, zp);

const auto bx = p_b_x[ip];
const auto by = p_b_y[ip];
const auto bz = p_b_z[ip];
const auto ux = 0.5_prt*(p_ux[ip] + p_ux_old[ip]);
const auto uy = 0.5_prt*(p_uy[ip] + p_uy_old[ip]);
const auto uz = 0.5_prt*(p_uz[ip] + p_uz_old[ip]);

auto const u2 = ux*ux + uy*uy + uz*uz;

auto const one_over_gamma = 1._prt/std::sqrt(1.0_rt + u2*inv_c2);
auto const one_over_gamma_c = one_over_gamma*inv_c;
const auto one_over_dt_gamma_c = one_over_gamma_c/dt;

const auto bpx = p_bp_x[ip];
const auto bpy = p_bp_y[ip];
const auto bpz = p_bp_z[ip];
const auto bx = ux*one_over_gamma_c;
const auto by = uy*one_over_gamma_c;
const auto bz = uz*one_over_gamma_c;
const auto bpx = (p_ux[ip] - p_ux_old[ip])*one_over_dt_gamma_c;
const auto bpy = (p_uy[ip] - p_uy_old[ip])*one_over_dt_gamma_c;
const auto bpz = (p_uz[ip] - p_uz_old[ip])*one_over_dt_gamma_c;

const auto w = p_w[ip];

Expand Down

0 comments on commit 58ee62e

Please sign in to comment.