LCC04HCPTraversal.h

Go to the documentation of this file.

 
#pragma once
 
#include "autopas/containers/cellTraversals/C08BasedTraversal.h"
#include "autopas/containers/linkedCells/traversals/LCC08CellHandler.h"
#include "autopas/containers/linkedCells/traversals/LCTraversalInterface.h"
#include "autopas/utils/ArrayUtils.h"
#include "autopas/utils/ThreeDimensionalMapping.h"
#include "autopas/utils/WrapOpenMP.h"
 
namespace autopas {
 
template <class ParticleCell, class PairwiseFunctor>
class LCC04HCPTraversal : public C08BasedTraversal<ParticleCell, PairwiseFunctor>, public LCTraversalInterface {
 public:
  LCC04HCPTraversal(const std::array<unsigned long, 3> &dims, PairwiseFunctor &pairwiseFunctor,
                    const double interactionLength, const std::array<double, 3> &cellLength,
                    DataLayoutOption dataLayout, bool useNewton3)
      : C08BasedTraversal<ParticleCell, PairwiseFunctor>(dims, pairwiseFunctor, interactionLength, cellLength,
                                                         dataLayout, useNewton3),
        _cellHandler(pairwiseFunctor, this->_cellsPerDimension, interactionLength, cellLength, this->_overlap,
                     dataLayout, useNewton3),
        _end(utils::ArrayMath::subScalar(utils::ArrayUtils::static_cast_copy_array<long>(this->_cellsPerDimension),
                                         1l)) {}
 
  void traverseParticles() override;
 
  [[nodiscard]] TraversalOption getTraversalType() const override { return TraversalOption::lc_c04_HCP; }
 
  [[nodiscard]] bool isApplicableToDomain() const override {
    // The cellsize cannot be smaller than the cutoff, if OpenMP is used.
    // Also see: https://github.com/AutoPas/AutoPas/issues/464
    const double minLength = *std::min_element(this->_cellLength.cbegin(), this->_cellLength.cend());
 
    return minLength >= this->_interactionLength;
  }
 
  void setAoSSortingThresholds(const SortingThresholdInfoInterface &aosSortingThreshold) override {
    _cellHandler.setAoSSortingThresholds(aosSortingThreshold);
  }
 
  void setSoASortingThresholds(const SortingThresholdInfoInterface &soaSortingThreshold) override {
    _cellHandler.setSoASortingThresholds(soaSortingThreshold);
  }
 
 private:
  void traverseSingleColor(std::vector<ParticleCell> &cells, int color);
 
  void processBasePack6(std::vector<ParticleCell> &cells, const std::array<long, 3> &base3DIndex);
 
  LCC08CellHandler<ParticleCell, PairwiseFunctor> _cellHandler;
 
  const std::array<long, 3> _end;
};
 
template <class ParticleCell, class PairwiseFunctor>
void LCC04HCPTraversal<ParticleCell, PairwiseFunctor>::processBasePack6(std::vector<ParticleCell> &cells,
                                                                        const std::array<long, 3> &base3DIndex) {
  using utils::ThreeDimensionalMapping::threeToOneD;
  std::array<long, 3> index{};
  const std::array<long, 3> signedDims = utils::ArrayUtils::static_cast_copy_array<long>(this->_cellsPerDimension);
 
  // go through the six cells
  for (long z = 0; z < 3; ++z) {
    for (long x = 0; x < 2; ++x) {
      index[0] = base3DIndex[0] + x;
      index[1] = base3DIndex[1];
      index[2] = base3DIndex[2] + z;
 
      bool isIn = true;
      for (int d = 0; d < 3; ++d) {
        // prevent using overlapping cells and cells outside the boundaries
        isIn &= (index[d] >= 0l) and (index[d] <= (_end[d] - this->_overlap[d]));
      }
 
      // skip cells outside radius
      if (isIn) {
        const unsigned long ulIndex = threeToOneD(index, signedDims);
        _cellHandler.processBaseCell(cells, ulIndex);
      }
    }
  }
}
 
template <class ParticleCell, class PairwiseFunctor>
void LCC04HCPTraversal<ParticleCell, PairwiseFunctor>::traverseParticles() {
  auto &cells = *(this->_cells);
  AUTOPAS_OPENMP(parallel) {
    for (int color = 0; color < 4; ++color) {
      traverseSingleColor(cells, color);
 
      if (color < 3) {
        AUTOPAS_OPENMP(barrier)
      }
    }
  }  // close parallel region
}
 
template <class ParticleCell, class PairwiseFunctor>
void LCC04HCPTraversal<ParticleCell, PairwiseFunctor>::traverseSingleColor(std::vector<ParticleCell> &cells,
                                                                           int color) {
  // determine a starting point of one of the grids
  std::array<long, 3> startOfThisColor{};  // coordinates: {x,y,z}
 
  // different starting points for different colors
  // some colors are starting outside the grid because only part of their cuboids are part of the grid
  // this way the starting points of sticking out cuboids can be determined as well
  switch (color) {
    case 0:
      startOfThisColor = {0l, 0l, 0l};
      break;
    case 1:
      startOfThisColor = {-4l, 0l, 1l};
      break;
    case 2:
      startOfThisColor = {-4l, 0l, -2l};
      break;
    case 3:
      startOfThisColor = {-2l, 0l, -1l};
      break;
    default:
      autopas::utils::ExceptionHandler::exception("LCC04HCPTraversal::traverseSingleColor: invalid color ({})", color);
  }
 
  // to fix intel64 icpc compiler complaints about perfectly nested loop.
  const long startX = startOfThisColor[0], endX = _end[0];
  const long startY = startOfThisColor[1], endY = _end[1];
  const long startZ = startOfThisColor[2], endZ = _end[2];
 
  // iterate over cartesian grid
  AUTOPAS_OPENMP(for schedule(dynamic, 1) collapse(3) nowait)
  for (long z = startZ; z < endZ; z += 4) {
    for (long y = startY; y < endY; y++) {
      /* color starts every 6th column again, the +4 is needed to prevent ending too early, since it
      will be shifted back inside the loop */
      for (long x = startX; x < (endX + 4); x += 6) {
        long x_index = x;
        /*  shift on x-axis according to z-value: shift two times and then go back to original x-value
            first: no shift
            second: -4 shift
            third: -2 shift
            fourth: go back to first
            every 12th z, the shifting pattern repeats again at the origin of x without shift,
            because z is shifted by 4 in every loop run and every third z-shift the pattern repeats
        */
        switch ((z - startZ) % 12 / 4) {
          case 0:
            break;
          case 1:
            x_index -= 4;
            break;
          case 2:
            x_index -= 2;
            break;
          default:
            break;
        }
        // shift x-axis every second y-row
        if ((y - startY) % 2 != 0) {
          x_index += 3;
        }
        processBasePack6(cells, {x_index, y, z});
      }
    }
  }
}
 
}  // namespace autopas