Merge pull request #539 from Yin-YinjianZhao/collision

Adding Collision Module into WarpX

1 files changed, 9 insertions, 0 deletions

diff --git a/Source/Particles/MultiParticleContainer.H b/Source/Particles/MultiParticleContainer.H
index 72064ce8c..d7ddc7a72 100644
--- a/Source/Particles/MultiParticleContainer.H
+++ b/Source/Particles/MultiParticleContainer.H
@@ -16,6 +16,8 @@
     #include <QuantumSyncEngineWrapper.H>
 #endif
 
+#include "CollisionType.H"
+
 #include <memory>
 #include <map>
 #include <string>
@@ -142,6 +144,8 @@ public:
 
     void doFieldIonization ();
 
+    void doCoulombCollisions ();
+
     void Checkpoint (const std::string& dir) const;
 
     void WritePlotFile (const std::string& dir) const;
@@ -214,6 +218,10 @@ protected:
 
     std::vector<std::string> lasers_names;
 
+    std::vector<std::string> collision_names;
+
+    amrex::Vector<std::unique_ptr<CollisionType> > allcollisions;
+
     //! instead of depositing (current, charge) on the finest patch level, deposit to the coarsest grid
     std::vector<bool> m_deposit_on_main_grid;
 
@@ -295,5 +303,6 @@ private:
     // runtime parameters
     int nlasers = 0;
     int nspecies = 1;   // physical particles only. nspecies+nlasers == allcontainers.size().
+    int ncollisions = 0;
 };
 #endif /*WARPX_ParticleContainer_H_*/