LoadEstimators.h

Go to the documentation of this file.

 
#pragma once
 
#include <array>
#include <vector>
 
#include "autopas/containers/verletListsCellBased/verletListsCells/VerletListsCellsHelpers.h"
#include "autopas/utils/ThreeDimensionalMapping.h"
 
namespace autopas::loadEstimators {
 
template <class ParticleCell>
unsigned long squaredParticlesPerCell(const std::vector<ParticleCell> &cells,
                                      const std::array<unsigned long, 3> &cellsPerDimension,
                                      const std::array<unsigned long, 3> &lowerCorner,
                                      const std::array<unsigned long, 3> &upperCorner) {
  unsigned long sum = 0;
  for (unsigned long x = lowerCorner[0]; x <= upperCorner[0]; x++) {
    for (unsigned long y = lowerCorner[1]; y <= upperCorner[1]; y++) {
      for (unsigned long z = lowerCorner[2]; z <= upperCorner[2]; z++) {
        const auto load =
            cells[autopas::utils::ThreeDimensionalMapping::threeToOneD(x, y, z, cellsPerDimension)].size();
        sum += load * load;
      }
    }
  }
  return sum;
}
 
template <class Particle_T>
unsigned long neighborListLengthImpl(
    const typename autopas::VerletListsCellsHelpers::AllCellsNeighborListsType<Particle_T> &neighborLists,
    unsigned long cellIndex) {
  unsigned long cellLoad = 0;
  for (auto &list : neighborLists[cellIndex]) {
    cellLoad += list.second.size();
  }
 
  return cellLoad;
}
 
template <class Particle_T>
unsigned long neighborListLengthImpl(
    const typename autopas::VerletListsCellsHelpers::PairwiseNeighborListsType<Particle_T> &neighborLists,
    unsigned long cellIndex) {
  unsigned long cellLoad = 0;
  for (auto &list : neighborLists[cellIndex]) {
    for (size_t index = 0; index < list.size(); index++) {
      cellLoad += list[index].second.size();
    }
  }
 
  return cellLoad;
}
 
template <class Particle_T, class NeighborList>
unsigned long neighborListLength(NeighborList &neighborLists, const std::array<unsigned long, 3> &cellsPerDimension,
                                 const std::array<unsigned long, 3> &lowerCorner,
                                 const std::array<unsigned long, 3> &upperCorner) {
  auto &internalList = neighborLists.getAoSNeighborList();
  unsigned long sum = 0;
  for (unsigned long x = lowerCorner[0]; x <= upperCorner[0]; x++) {
    for (unsigned long y = lowerCorner[1]; y <= upperCorner[1]; y++) {
      for (unsigned long z = lowerCorner[2]; z <= upperCorner[2]; z++) {
        auto cellIndex = autopas::utils::ThreeDimensionalMapping::threeToOneD(x, y, z, cellsPerDimension);
        sum += neighborListLengthImpl<Particle_T>(internalList, cellIndex);
      }
    }
  }
  return sum;
}
 
}  // namespace autopas::loadEstimators