SortingThresholdInfo2B.h

Go to the documentation of this file.

 
#pragma once
 
#include <cstddef>
 
#include "autopas/options/Newton3Option.h"
#include "autopas/options/SortingDirectionOption.h"
#include "autopas/utils/SortingThresholdInfoInterface.h"
 
namespace autopas {
 
struct SortingThresholdInfo2B : SortingThresholdInfoInterface {
  size_t noN3FaceThresholdBidirectional;
  size_t noN3FaceThresholdUnidirectional;
  size_t noN3EdgeThresholdBidirectional;
  size_t noN3EdgeThresholdUnidirectional;
  size_t noN3CornerThresholdBidirectional;
  size_t noN3CornerThresholdUnidirectional;
  size_t n3FaceThreshold;
  size_t n3EdgeThreshold;
  size_t n3CornerThreshold;
 
  explicit SortingThresholdInfo2B(size_t uniformThreshold)
      : noN3FaceThresholdBidirectional(uniformThreshold),
        noN3FaceThresholdUnidirectional(uniformThreshold),
        noN3EdgeThresholdBidirectional(uniformThreshold),
        noN3EdgeThresholdUnidirectional(uniformThreshold),
        noN3CornerThresholdBidirectional(uniformThreshold),
        noN3CornerThresholdUnidirectional(uniformThreshold),
        n3FaceThreshold(uniformThreshold),
        n3EdgeThreshold(uniformThreshold),
        n3CornerThreshold(uniformThreshold) {}
 
  SortingThresholdInfo2B(size_t noN3FaceThresholdBidirectional, size_t noN3FaceThresholdUnidirectional,
                         size_t noN3EdgeThresholdBidirectional, size_t noN3EdgeThresholdUnidirectional,
                         size_t noN3CornerThresholdBidirectional, size_t noN3CornerThresholdUnidirectional,
                         size_t n3FaceThreshold, size_t n3EdgeThreshold, size_t n3CornerThreshold)
      : noN3FaceThresholdBidirectional(noN3FaceThresholdBidirectional),
        noN3FaceThresholdUnidirectional(noN3FaceThresholdUnidirectional),
        noN3EdgeThresholdBidirectional(noN3EdgeThresholdBidirectional),
        noN3EdgeThresholdUnidirectional(noN3EdgeThresholdUnidirectional),
        noN3CornerThresholdBidirectional(noN3CornerThresholdBidirectional),
        noN3CornerThresholdUnidirectional(noN3CornerThresholdUnidirectional),
        n3FaceThreshold(n3FaceThreshold),
        n3EdgeThreshold(n3EdgeThreshold),
        n3CornerThreshold(n3CornerThreshold) {}
 
  size_t getThresholdByConfig(bool newton3, std::array<double, 3> sortingDirection, bool bidirectional) const {
    const SortingDirectionOption cellDirection = SortingDirectionOption::fromRawArray(sortingDirection);
    if (newton3) {
      if (cellDirection == SortingDirectionOption::corner) {
        return n3CornerThreshold;
      }
      if (cellDirection == SortingDirectionOption::edge) {
        return n3EdgeThreshold;
      }
      return n3FaceThreshold;
    } else if (bidirectional) {
      if (cellDirection == SortingDirectionOption::corner) {
        return noN3CornerThresholdBidirectional;
      }
      if (cellDirection == SortingDirectionOption::edge) {
        return noN3EdgeThresholdBidirectional;
      }
      return noN3FaceThresholdBidirectional;
    } else {
      if (cellDirection == SortingDirectionOption::corner) {
        return noN3CornerThresholdUnidirectional;
      }
      if (cellDirection == SortingDirectionOption::edge) {
        return noN3EdgeThresholdUnidirectional;
      }
      return noN3FaceThresholdUnidirectional;
    }
  }
  void setThresholdByOption(Newton3Option n3, SortingDirectionOption cellDirection, bool bidirectional, size_t value) {
    if (n3 == Newton3Option::enabled) {
      switch (cellDirection) {
        case SortingDirectionOption::corner:
          n3CornerThreshold = value;
          break;
        case SortingDirectionOption::edge:
          n3EdgeThreshold = value;
          break;
        case SortingDirectionOption::face:
          n3FaceThreshold = value;
          break;
      }
    } else if (bidirectional) {
      switch (cellDirection) {
        case SortingDirectionOption::corner:
          noN3CornerThresholdBidirectional = value;
          break;
        case SortingDirectionOption::edge:
          noN3EdgeThresholdBidirectional = value;
          break;
        case SortingDirectionOption::face:
          noN3FaceThresholdBidirectional = value;
          break;
      }
    } else {
      switch (cellDirection) {
        case SortingDirectionOption::corner:
          noN3CornerThresholdUnidirectional = value;
          break;
        case SortingDirectionOption::edge:
          noN3EdgeThresholdUnidirectional = value;
          break;
        case SortingDirectionOption::face:
          noN3FaceThresholdUnidirectional = value;
          break;
      }
    }
  }
};
 
}  // namespace autopas