1 files changed, 10 insertions, 0 deletions
diff --git a/Source/Particles/WarpXParticleContainer.H b/Source/Particles/WarpXParticleContainer.H
index 1561fcfcf..40f352df9 100644
--- a/Source/Particles/WarpXParticleContainer.H
+++ b/Source/Particles/WarpXParticleContainer.H
@@ -317,6 +317,13 @@ public:
     }
 
     int doBackTransformedDiagnostics () const { return do_back_transformed_diagnostics; }
+    /** Whether back-transformed diagnostics need to be performed for a particular species.
+     *
+     * \param[in] do_back_transformed_particles The parameter to set if back-transformed particles are set to true/false
+     */
+    void SetDoBackTransformedParticles(const bool do_back_transformed_particles) {
+        m_do_back_transformed_particles = do_back_transformed_particles;
+    }
 
     std::map<std::string, int> getParticleComps () const noexcept { return particle_comps;}
     std::map<std::string, int> getParticleiComps () const noexcept { return particle_icomps;}
@@ -417,6 +424,8 @@ protected:
     int do_resampling = 0;
 
     int do_back_transformed_diagnostics = 1;
+    /** Whether back-transformed diagnostics is turned on for the corresponding species.*/
+    bool m_do_back_transformed_particles = false;
 
 #ifdef WARPX_QED
     //Species can receive a shared pointer to a QED engine (species for
@@ -445,6 +454,7 @@ public:
                                        TmpIdx::nattribs>;
     using TmpParticles = amrex::Vector<std::map<PairIndex, TmpParticleTile> >;
 
+    TmpParticles getTmpParticleData () const noexcept {return tmp_particle_data;}
 protected:
     TmpParticles tmp_particle_data;