From 7e02060ebed8a58ebb0828eca03a824f809267fc Mon Sep 17 00:00:00 2001
From: "L. Diana Amorim" <LDianaAmorim@lbl.gov>
Date: Wed, 10 Jun 2020 22:18:17 -0700
Subject: [Mini] Z Offset to loaded openPMD particles (#1070)

* Added description of singleparticle particle injection style

* Revert "Added description of singleparticle particle injection style"

This reverts commit f02d842c935311458024da6e661950742de24f63.

* Added use_q_shift and q_shift variables

* Removed use_q_shift

* Added description of q_shift to the Docs

* Replaced q_shift by z_shift as suggested by @ax3l

* Added recommendation to PWFA example

* z_shift should be optional (used query instead of get)

* Fix compile problem

Co-authored-by: Axel Huebl <axel.huebl@plasma.ninja>
---
 Source/Particles/PhysicalParticleContainer.cpp | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

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

diff --git a/Source/Particles/PhysicalParticleContainer.cpp b/Source/Particles/PhysicalParticleContainer.cpp
index 094a8e556..e138f39a4 100644
--- a/Source/Particles/PhysicalParticleContainer.cpp
+++ b/Source/Particles/PhysicalParticleContainer.cpp
@@ -294,7 +294,8 @@ PhysicalParticleContainer::AddGaussianBeam (
 }
 
 void
-PhysicalParticleContainer::AddPlasmaFromFile(ParticleReal q_tot)
+PhysicalParticleContainer::AddPlasmaFromFile(ParticleReal q_tot,
+                                             ParticleReal z_shift)
 {
     // Declare temporary vectors on the CPU
     Gpu::HostVector<ParticleReal> particle_x;
@@ -359,7 +360,7 @@ PhysicalParticleContainer::AddPlasmaFromFile(ParticleReal q_tot)
 
         for (auto i = decltype(npart){0}; i<npart; ++i){
             ParticleReal const x = ptr_x.get()[i]*position_unit_x;
-            ParticleReal const z = ptr_z.get()[i]*position_unit_z;
+            ParticleReal const z = ptr_z.get()[i]*position_unit_z+z_shift;
 #   ifndef WARPX_DIM_3D
             ParticleReal const y = 0.0_prt;
 #   else
@@ -453,7 +454,8 @@ PhysicalParticleContainer::AddParticles (int lev)
     }
 
     if (plasma_injector->external_file) {
-        AddPlasmaFromFile(plasma_injector->q_tot);
+        AddPlasmaFromFile(plasma_injector->q_tot,
+                          plasma_injector->z_shift);
         return;
     }
 
-- 
cgit v1.2.3