Method hostPartitionTile

Src/Particle/AMReX_ParticleContainerI.H:1351–1434

template <typename ParticleType, int NArrayReal, int NArrayInt,
          template<class> class Allocator, class CellAssignor>
AMREX_FORCE_INLINE
int
ParticleContainer_impl<ParticleType, NArrayReal, NArrayInt, Allocator, CellAssignor>
::hostPartitionTile (ParticleTileType& src_tile,
                     int lev, int gid, int tid,
                     int lev_min, int lev_max, int nGrow, int local,
                     bool remove_negative, int myproc,
                     Gpu::DeviceVector<int>& boxes,
                     Gpu::DeviceVector<int>& levels,
                     Gpu::DeviceVector<int>& tiles,
                     Gpu::DeviceVector<int>& src_indices,
                     Gpu::DeviceVector<IntVect>& periodic_shift)
{
    ParticleLocData pld;
    auto ptd = src_tile.getParticleTileData();
    int num_move = 0;

    Long last = src_tile.numParticles() - 1;
    Long pindex = 0;
    int who = -1;
    while (pindex <= last) {
        decltype(auto) p = ptd[pindex];

        // remove invalid if needed
        if (!ptd.id(pindex).is_valid()) {
            if (remove_negative) {
                copyParticle(ptd, ptd, last, pindex);
                correctCellVectors(last, pindex, gid, p);
                --last;
            } else {
                ++pindex;
            }
            continue;
        }

        // Fast path: if on the finest level, try the last good location first
        // If not on finest level, we need to do the full locate because even if
        // the last assignment works, the particle might belong on a finer level.
        if (lev == lev_max && pld.m_tile == tid && pld.m_grid == gid && pld.m_lev == lev && who == myproc)  {
            const auto iv = Index(p, lev);
            if (pld.m_tilebox.contains(iv)) {
                ++pindex;
                continue; // don't need correctCellVectors or particlePostLocate if it stays where it is
            }
        }

        // full locate
        locateParticle(p, pld, lev_min, lev_max, nGrow, local ? gid : -1);
        particlePostLocate(p, pld, lev);

        // particle may have been invalidated, so check if we need to remove again.
        if (!ptd.id(pindex).is_valid()) {
            copyParticle(ptd, ptd, last, pindex);
            correctCellVectors(last, pindex, gid, p);
            --last;
            continue;
        }