1 files changed, 77 insertions, 0 deletions
diff --git a/Source/Initialization/InjectorDensity.cpp b/Source/Initialization/InjectorDensity.cpp
new file mode 100644
index 000000000..7fed85b75
--- /dev/null
+++ b/Source/Initialization/InjectorDensity.cpp
@@ -0,0 +1,77 @@
+#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;
+    }
+    }
+}
+
+// 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
+        // is stored in shared memory.
+        return amrex::Gpu::numThreadsPerBlockParallelFor() * sizeof(double) * 3;
+    }
+    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() > 6,
+            "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);
+}