1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
/* Copyright 2019 Yinjian Zhao
*
* This file is part of WarpX.
*
* License: BSD-3-Clause-LBNL
*/
#ifndef WARPX_PARTICLES_COLLISION_ELASTIC_COLLISION_PEREZ_H_
#define WARPX_PARTICLES_COLLISION_ELASTIC_COLLISION_PEREZ_H_
#include "UpdateMomentumPerezElastic.H"
#include "ComputeTemperature.H"
#include "Utils/WarpXConst.H"
#include <AMReX_Random.H>
/** \brief Prepare information for and call
* UpdateMomentumPerezElastic().
* @param[in] I1s,I2s is the start index for I1,I2 (inclusive).
* @param[in] I1e,I2e is the start index for I1,I2 (exclusive).
* @param[in] I1 and I2 are the index arrays.
* @param[in,out] u1 and u2 are the velocity arrays (u=v*gamma),
* they could be either different or the same,
* their lengths are not needed,
* @param[in] I1 and I2 determine all elements that will be used.
* @param[in] w1 and w2 are arrays of weights.
* @param[in] q1 and q2 are charges. m1 and m2 are masses.
* @param[in] T1 and T2 are temperatures (Joule)
* and will be used if greater than zero,
* otherwise will be computed.
* @param[in] dt is the time step length between two collision calls.
* @param[in] L is the Coulomb log and will be used if greater than zero,
* otherwise will be computed.
* @param[in] dV is the volume of the corresponding cell.
*/
template <typename T_index, typename T_R>
AMREX_GPU_HOST_DEVICE AMREX_INLINE
void ElasticCollisionPerez (
T_index const I1s, T_index const I1e,
T_index const I2s, T_index const I2e,
T_index *I1, T_index *I2,
T_R *u1x, T_R *u1y, T_R *u1z,
T_R *u2x, T_R *u2y, T_R *u2z,
T_R const *w1, T_R const *w2,
T_R const q1, T_R const q2,
T_R const m1, T_R const m2,
T_R const T1, T_R const T2,
T_R const dt, T_R const L, T_R const dV,
amrex::RandomEngine const& engine)
{
int NI1 = I1e - I1s;
int NI2 = I2e - I2s;
// get local T1t and T2t
T_R T1t; T_R T2t;
if ( T1 <= T_R(0.0) && L <= T_R(0.0) )
{
T1t = ComputeTemperature(I1s,I1e,I1,u1x,u1y,u1z,m1);
}
else { T1t = T1; }
if ( T2 <= T_R(0.0) && L <= T_R(0.0) )
{
T2t = ComputeTemperature(I2s,I2e,I2,u2x,u2y,u2z,m2);
}
else { T2t = T2; }
// local density
T_R n1 = T_R(0.0);
T_R n2 = T_R(0.0);
T_R n12 = T_R(0.0);
for (int i1=I1s; i1<static_cast<int>(I1e); ++i1) { n1 += w1[ I1[i1] ]; }
for (int i2=I2s; i2<static_cast<int>(I2e); ++i2) { n2 += w2[ I2[i2] ]; }
n1 = n1 / dV; n2 = n2 / dV;
{
int i1 = I1s; int i2 = I2s;
for (int k = 0; k < amrex::max(NI1,NI2); ++k)
{
n12 += amrex::min( w1[ I1[i1] ], w2[ I2[i2] ] );
++i1; if ( i1 == static_cast<int>(I1e) ) { i1 = I1s; }
++i2; if ( i2 == static_cast<int>(I2e) ) { i2 = I2s; }
}
n12 = n12 / dV;
}
// compute Debye length lmdD
T_R lmdD;
lmdD = T_R(1.0)/std::sqrt( n1*q1*q1/(T1t*PhysConst::ep0) +
n2*q2*q2/(T2t*PhysConst::ep0) );
T_R rmin = std::pow( T_R(4.0) * MathConst::pi / T_R(3.0) *
amrex::max(n1,n2), T_R(-1.0/3.0) );
lmdD = amrex::max(lmdD, rmin);
// call UpdateMomentumPerezElastic()
{
int i1 = I1s; int i2 = I2s;
for (int k = 0; k < amrex::max(NI1,NI2); ++k)
{
UpdateMomentumPerezElastic(
u1x[ I1[i1] ], u1y[ I1[i1] ], u1z[ I1[i1] ],
u2x[ I2[i2] ], u2y[ I2[i2] ], u2z[ I2[i2] ],
n1, n2, n12,
q1, m1, w1[ I1[i1] ], q2, m2, w2[ I2[i2] ],
dt, L, lmdD,
engine);
++i1; if ( i1 == static_cast<int>(I1e) ) { i1 = I1s; }
++i2; if ( i2 == static_cast<int>(I2e) ) { i2 = I2s; }
}
}
}
#endif // WARPX_PARTICLES_COLLISION_ELASTIC_COLLISION_PEREZ_H_
|