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
|
/* Copyright 2019-2020 Andrew Myers, Axel Huebl,
* Maxence Thevenet
*
* This file is part of WarpX.
*
* License: BSD-3-Clause-LBNL
*/
#ifndef DEFAULTINITIALIZATION_H_
#define DEFAULTINITIALIZATION_H_
#include <AMReX_REAL.H>
#include <AMReX_GpuContainers.H>
#include <map>
#include <string>
#include <cmath>
/**
* \brief This set of initialization policies describes what happens
* when we need to create a new particle due to an elementary process.
* For example, when an ionization event creates an electron, these
* policies control the initial values of the electron's components.
* These can always be over-written later.
*
* The specific meanings are as follows:
* Zero - set the component to zero
* One - set the component to one
* RandomExp - a special flag for the optical depth component used by
* certain QED processes, which gets a random initial value
* extracted from an exponential distribution
*
*/
enum struct InitializationPolicy {Zero=0, One, RandomExp};
/**
* \brief This map sets the initialization policy for each particle component
* used in WarpX.
*/
static std::map<std::string, InitializationPolicy> initialization_policies = {
{"w", InitializationPolicy::Zero },
{"ux", InitializationPolicy::Zero },
{"uy", InitializationPolicy::Zero },
{"uz", InitializationPolicy::Zero },
#ifdef WARPX_DIM_RZ
{"theta", InitializationPolicy::Zero},
#endif
#ifdef WARPX_QED
{"optical_depth_BW", InitializationPolicy::RandomExp},
{"optical_depth_QSR", InitializationPolicy::RandomExp}
#endif
};
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
amrex::ParticleReal initializeRealValue (const InitializationPolicy policy, amrex::RandomEngine const& engine) noexcept
{
switch (policy) {
case InitializationPolicy::Zero : return 0.0;
case InitializationPolicy::One : return 1.0;
case InitializationPolicy::RandomExp : {
return -std::log(amrex::Random(engine));
}
default : {
amrex::Abort("Initialization Policy not recognized");
return 1.0;
}
}
}
AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
int initializeIntValue (const InitializationPolicy policy) noexcept
{
switch (policy) {
case InitializationPolicy::Zero : return 0;
case InitializationPolicy::One : return 1;
default : {
amrex::Abort("Initialization Policy not recognized");
return 1;
}
}
}
#endif
|
