From 7f178a55df4d0973f6788d9c27fd10b128ad4571 Mon Sep 17 00:00:00 2001
From: Luca Fedeli <luca.fedeli@cea.fr>
Date: Tue, 8 Oct 2019 18:25:10 +0200
Subject: Heavy refactoring

---
 Source/Particles/MultiParticleContainer.cpp | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'Source/Particles/MultiParticleContainer.cpp')

diff --git a/Source/Particles/MultiParticleContainer.cpp b/Source/Particles/MultiParticleContainer.cpp
index 715c97b99..a5632dede 100644
--- a/Source/Particles/MultiParticleContainer.cpp
+++ b/Source/Particles/MultiParticleContainer.cpp
@@ -1,6 +1,7 @@
 #include <limits>
 #include <algorithm>
 #include <string>
+#include <memory>
 
 #include <MultiParticleContainer.H>
 #include <WarpX_f.H>
@@ -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
-- 
cgit v1.2.3