blob: fa54b342c49e64ba016dc9303b883fbf227cecf2 (
plain) (
blame)
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
|
#include <InjectorDensity.H>
#include <PlasmaInjector.H>
using namespace amrex;
InjectorDensity::~InjectorDensity ()
{
switch (type)
{
case Type::parser:
{
object.parser.m_parser.clear();
break;
}
case Type::custom:
{
object.custom.clear();
break;
}
case Type::predefined:
{
object.predefined.clear();
break;
}
default:
return;
}
}
// Compute the amount of memory needed in GPU Shared Memory.
std::size_t
InjectorDensity::sharedMemoryNeeded () const noexcept
{
switch (type)
{
case Type::parser:
{
// For parser injector, the 3D position of each particle
// and time, t, is stored in shared memory.
return amrex::Gpu::numThreadsPerBlockParallelFor() * sizeof(double) * 4;
}
default:
return 0;
}
}
InjectorDensityPredefined::InjectorDensityPredefined (
std::string const& a_species_name) noexcept
: profile(Profile::null)
{
ParmParse pp(a_species_name);
std::vector<amrex::Real> v;
// Read parameters for the predefined plasma profile,
// and store them in managed memory
pp.getarr("predefined_profile_params", v);
p = static_cast<amrex::Real*>
(amrex::The_Managed_Arena()->alloc(sizeof(amrex::Real)*v.size()));
for (int i = 0; i < static_cast<int>(v.size()); ++i) {
p[i] = v[i];
}
// Parse predefined profile name, and update member variable profile.
std::string which_profile_s;
pp.query("predefined_profile_name", which_profile_s);
std::transform(which_profile_s.begin(), which_profile_s.end(),
which_profile_s.begin(), ::tolower);
if (which_profile_s == "parabolic_channel"){
profile = Profile::parabolic_channel;
AMREX_ALWAYS_ASSERT_WITH_MESSAGE(v.size() > 5,
"InjectorDensityPredefined::parabolic_channel: not enough parameters");
}
}
// Note that we are not allowed to have non-trivial destructor.
// So we rely on clear() to free memory.
void InjectorDensityPredefined::clear ()
{
amrex::The_Managed_Arena()->free(p);
}
|
