Merge branch 'classical_radiation_reaction' of github.com:lucafedeli88/WarpX into classical_radiation_reaction

1 files changed, 26 insertions, 4 deletions

diff --git a/Source/Particles/MultiParticleContainer.cpp b/Source/Particles/MultiParticleContainer.cpp
index 715c97b99..c860d21f5 100644
--- a/Source/Particles/MultiParticleContainer.cpp
+++ b/Source/Particles/MultiParticleContainer.cpp
@@ -1,11 +1,12 @@
-#include <limits>
-#include <algorithm>
-#include <string>
-
 #include <MultiParticleContainer.H>
 #include <WarpX_f.H>
 #include <WarpX.H>
 
+#include <limits>
+#include <algorithm>
+#include <string>
+#include <memory>
+
 using namespace amrex;
 
 MultiParticleContainer::MultiParticleContainer (AmrCore* amr_core)
@@ -149,6 +150,11 @@ MultiParticleContainer::InitData ()
     // For each species, get the ID of its product species.
     // This is used for ionization and pair creation processes.
     mapSpeciesProduct();
+
+#ifdef WARPX_QED
+    InitQED();
+#endif
+
 }
 
 
@@ -726,3 +732,19 @@ MultiParticleContainer::doFieldIonization ()
         } // lev
     } // pc_source
 }
+
+#ifdef WARPX_QED
+void MultiParticleContainer::InitQED ()
+{
+    for (auto& pc : allcontainers) {
+        if(pc->has_quantum_sync()){
+            pc->set_quantum_sync_engine_ptr
+                (std::make_shared<QuantumSynchrotronEngine>(qs_engine));
+        }
+        if(pc->has_breit_wheeler()){
+            pc->set_breit_wheeler_engine_ptr
+                (std::make_shared<BreitWheelerEngine>(bw_engine));
+        }
+    }
+}
+#endif