1 files changed, 92 insertions, 0 deletions
diff --git a/Source/Particles/PhotonParticleContainer.cpp b/Source/Particles/PhotonParticleContainer.cpp
new file mode 100644
index 000000000..31a62b739
--- /dev/null
+++ b/Source/Particles/PhotonParticleContainer.cpp
@@ -0,0 +1,92 @@
+#include <limits>
+#include <sstream>
+#include <algorithm>
+
+#ifdef _OPENMP
+#include <omp.h>
+#endif
+
+#include <PhotonParticleContainer.H>
+#include <WarpX_f.H>
+#include <WarpX.H>
+#include <WarpXConst.H>
+
+using namespace amrex;
+
+PhotonParticleContainer::PhotonParticleContainer (AmrCore* amr_core, int ispecies,
+                                                  const std::string& name)
+    : PhysicalParticleContainer(amr_core, ispecies, name)
+{
+
+    // This will read <species>.[...] from the inputs file
+    // where <species> is the name of your species
+    ParmParse pp(species_name);
+
+    // read <species>.size_in_inches in the input file, and
+    // store it into member data.
+    pp.query("size_in_inches", size_in_inches);
+
+#ifdef WARPX_QED
+    AddRealComp("tau");
+    plot_flags.resize(PIdx::nattribs + 1, 1);
+#endif
+
+}
+
+void PhotonParticleContainer::InitData()
+{
+    AddParticles(0); // Note - add on level 0
+
+    if (maxLevel() > 0) {
+        Redistribute();  // We then redistribute
+    }
+}
+
+void
+PhotonParticleContainer::PushPX(WarpXParIter& pti,
+                                Cuda::ManagedDeviceVector<Real>& xp,
+                                Cuda::ManagedDeviceVector<Real>& yp,
+                                Cuda::ManagedDeviceVector<Real>& zp,
+                                Cuda::ManagedDeviceVector<Real>& giv,
+                                Real dt)
+{
+}
+
+void
+PhotonParticleContainer::Evolve (int lev,
+                                        const MultiFab& Ex, const MultiFab& Ey, const MultiFab& Ez,
+                                        const MultiFab& Bx, const MultiFab& By, const MultiFab& Bz,
+                                        MultiFab& jx, MultiFab& jy, MultiFab& jz,
+                                        MultiFab* cjx, MultiFab* cjy, MultiFab* cjz,
+                                        MultiFab* rho, MultiFab* crho,
+                                        const MultiFab* cEx, const MultiFab* cEy, const MultiFab* cEz,
+                                        const MultiFab* cBx, const MultiFab* cBy, const MultiFab* cBz,
+                                        Real t, Real dt)
+{
+    // This does gather, push and depose.
+    // Push and depose have been re-written for photon,
+    // so they do not do anything.
+    // Currently, I guess photons do gather fields from the mesh.
+    PhysicalParticleContainer::Evolve (lev,
+                                       Ex, Ey, Ez,
+                                       Bx, By, Bz,
+                                       jx, jy, jz,
+                                       cjx, cjy, cjz,
+                                       rho, crho,
+                                       cEx, cEy, cEz,
+                                       cBx, cBy, cBz,
+                                       t, dt);
+
+}
+
+
+#ifdef WARPX_QED
+void PhotonParticleContainer::InitOpticalDepth(
+    WarpXParIter& pti,
+    warpx_breit_wheeler_engine& engine)
+{
+    auto& taus = pti.GetAttribs(particle_comps["tau"]);
+    for(auto& tau: taus)
+        tau = engine.get_optical_depth();
+}
+#endif