CompatibleLoadEstimators.h

Go to the documentation of this file.

 
#pragma once
 
#include <array>
#include <set>
#include <vector>
 
#include "autopas/options/ContainerOption.h"
#include "autopas/options/LoadEstimatorOption.h"
#include "autopas/options/TraversalOption.h"
 
namespace autopas::loadEstimators {
 
static std::set<autopas::LoadEstimatorOption> allCompatibleLoadEstimators(autopas::ContainerOption container) {
  switch (container) {
    case ContainerOption::linkedCells: {
      return std::set<autopas::LoadEstimatorOption>{LoadEstimatorOption::none,
                                                    LoadEstimatorOption::squaredParticlesPerCell};
    }
    case ContainerOption::verletListsCells: {
      return std::set<autopas::LoadEstimatorOption>{LoadEstimatorOption::none,
                                                    LoadEstimatorOption::squaredParticlesPerCell,
                                                    LoadEstimatorOption::neighborListLength};
    }
    case ContainerOption::verletClusterLists: {
      return std::set<autopas::LoadEstimatorOption>{LoadEstimatorOption::none, LoadEstimatorOption::neighborListLength};
    }
    default: {
      return std::set<autopas::LoadEstimatorOption>{};
    }
  }
}
 
static bool usesLoadEstimator(autopas::TraversalOption traversal) {
  switch (traversal) {
    case TraversalOption::lc_sliced_balanced:
      [[fallthrough]];
    case TraversalOption::vcl_sliced_balanced:
      [[fallthrough]];
    case TraversalOption::vlc_sliced_balanced: {
      return true;
    }
    default: {
      return false;
    }
  }
}
 
[[maybe_unused]] static std::set<autopas::LoadEstimatorOption> getApplicableLoadEstimators(
    autopas::ContainerOption container, autopas::TraversalOption traversal,
    const std::set<autopas::LoadEstimatorOption> &allowedOptions) {
  if (usesLoadEstimator(traversal)) {
    auto compatible = allCompatibleLoadEstimators(container);
    std::set<autopas::LoadEstimatorOption> intersection;
    std::set_intersection(allowedOptions.begin(), allowedOptions.end(), compatible.begin(), compatible.end(),
                          std::inserter(intersection, intersection.begin()));
    if (not intersection.empty()) {
      return intersection;
    }
  }
  return std::set<autopas::LoadEstimatorOption>{LoadEstimatorOption::none};
}
 
}  // namespace autopas::loadEstimators