From 768093cffd18a2c37d90be1033ae94c3ff38776a Mon Sep 17 00:00:00 2001
From: MaxThevenet <mthevenet@lbl.gov>
Date: Thu, 9 May 2019 15:53:08 -0700
Subject: move particle plot options to PPC

---
 Source/Particles/PhysicalParticleContainer.cpp | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

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

diff --git a/Source/Particles/PhysicalParticleContainer.cpp b/Source/Particles/PhysicalParticleContainer.cpp
index 2fa39d87d..c6bc92ff6 100644
--- a/Source/Particles/PhysicalParticleContainer.cpp
+++ b/Source/Particles/PhysicalParticleContainer.cpp
@@ -82,6 +82,26 @@ PhysicalParticleContainer::PhysicalParticleContainer (AmrCore* amr_core, int isp
     pp.query("do_splitting", do_splitting);
     pp.query("split_type", split_type);
     pp.query("do_continuous_injection", do_continuous_injection);
+    {
+        pp.queryarr("plot_vars", plot_vars);
+        if (plot_vars.size() == 0){
+            if (WarpX::do_boosted_frame_diagnostic && WarpX::do_boosted_frame_particles){
+                plot_flags.resize(PIdx::nattribs + 6, 1);
+            } else {
+                plot_flags.resize(PIdx::nattribs, 1);
+            }
+        } else {
+            if (WarpX::do_boosted_frame_diagnostic && WarpX::do_boosted_frame_particles){
+                plot_flags.resize(PIdx::nattribs + 6, 0);
+            } else {
+                plot_flags.resize(PIdx::nattribs, 0);
+            }
+
+            for (const auto& var : plot_vars){
+                plot_flags[ParticleStringNames::to_index.at(var)] = 1;
+            }
+        }
+    }
 }
 
 PhysicalParticleContainer::PhysicalParticleContainer (AmrCore* amr_core)
-- 
cgit v1.2.3