RemoveRealComp/RemoveIntComp

Add new functions to remove runtime components.
AMReX-Codes · Oct 16, 2023 · 08ee44b · 08ee44b
1 parent ecaa144
commit 08ee44b
Show file tree

Hide file tree

Showing 2 changed files with 56 additions and 0 deletions.
diff --git a/Src/Particle/AMReX_NeighborParticles.H b/Src/Particle/AMReX_NeighborParticles.H
@@ -309,6 +309,30 @@ public:
         calcCommSize();
     }
 
+    /** Remove the last n Real components of the species
+     *
+     * @param n number of components to remove
+     */
+    void RemoveRealComp (int n = 1)
+    {
+        ParticleContainer<NStructReal, NStructInt, NArrayReal, NArrayInt>::
+            RemoveRealComp(n);
+        ghost_real_comp.pop_back();
+        calcCommSize();
+    }
+
+    /** Remove the last n Int components of the species
+     *
+     * @param n number of components to remove
+     */
+    void RemoveIntComp (int n = 1)
+    {
+        ParticleContainer<NStructReal, NStructInt, NArrayReal, NArrayInt>::
+            RemoveIntComp(n);
+        ghost_int_comp.pop_back();
+        calcCommSize();
+    }
+
     void Redistribute (int lev_min=0, int lev_max=-1, int nGrow=0, int local=0)
     {
         clearNeighbors();

diff --git a/Src/Particle/AMReX_ParticleContainer.H b/Src/Particle/AMReX_ParticleContainer.H
@@ -1261,6 +1261,38 @@ public:
         SetParticleSize();
     }
 
+    /** Remove the last n Real components of the species
+     *
+     * @param n number of components to remove
+     */
+    void RemoveRealComp (int n = 1)
+    {
+        m_num_runtime_real--;
+        if (m_num_runtime_real == 0 && m_num_runtime_int == 0) {
+            m_runtime_comps_defined = false;
+        }
+        for (int i = 0; i < n; ++i) {
+            h_redistribute_real_comp.pop_back();
+        }
+        SetParticleSize();
+    }
+
+    /** Remove the last n Int components of the species
+     *
+     * @param n number of components to remove
+     */
+    void RemoveIntComp (int n = 1)
+    {
+        m_num_runtime_int--;
+        if (m_num_runtime_real == 0 && m_num_runtime_int == 0) {
+            m_runtime_comps_defined = false;
+        }
+        for (int i = 0; i < n; ++i) {
+            h_redistribute_int_comp.pop_back();
+        }
+        SetParticleSize();
+    }
+
     int NumRuntimeRealComps () const { return m_num_runtime_real; }
     int NumRuntimeIntComps  () const { return m_num_runtime_int;  }