ConfigurationAndRankIteratorHandler.h

Go to the documentation of this file.

 
#pragma once
 
#include "autopas/options/ContainerOption.h"
#include "autopas/options/DataLayoutOption.h"
#include "autopas/options/InteractionTypeOption.h"
#include "autopas/options/LoadEstimatorOption.h"
#include "autopas/options/Newton3Option.h"
#include "autopas/options/TraversalOption.h"
 
namespace autopas::utils {
class ConfigurationAndRankIteratorHandler {
 public:
  ConfigurationAndRankIteratorHandler(
      const std::set<ContainerOption> &containerOptions, const std::set<double> &cellSizeFactors,
      const std::set<TraversalOption> &traversalOptions, const std::set<LoadEstimatorOption> &loadEstimatorOptions,
      const std::set<DataLayoutOption> &dataLayoutOptions, const std::set<Newton3Option> &newton3Options,
      const InteractionTypeOption &interactionType, const int numConfigs, const int commSize)
      : _containers(containerOptions),
        _cellSizeFactors(cellSizeFactors),
        _allowedTraversalOptions(traversalOptions),
        _allowedLoadEstimatorOptions(loadEstimatorOptions),
        _dataLayoutOptions(dataLayoutOptions),
        _newton3Options(newton3Options),
        _interactionType(interactionType) {
    reset(numConfigs, commSize);
  }
 
  void advanceIterators(int numConfigs, int commSize);
 
  void reset(const int numConfigs, const int commSize);
 
  inline void getConfigIterators(std::set<ContainerOption>::iterator &containerIt,
                                 std::set<double>::iterator &cellSizeFactorIt,
                                 std::set<TraversalOption>::iterator &traversalIt,
                                 std::set<LoadEstimatorOption>::iterator &loadEstimatorIt,
                                 std::set<DataLayoutOption>::iterator &dataLayoutIt,
                                 std::set<Newton3Option>::iterator &newton3It) {
    containerIt = _containerIt;
    cellSizeFactorIt = _cellSizeFactorIt;
    traversalIt = _traversalIt;
    loadEstimatorIt = _loadEstimatorIt;
    dataLayoutIt = _dataLayoutIt;
    newton3It = _newton3It;
  }
 
  [[nodiscard]] inline int getRankIterator() const { return _rankIterator; }
 
  [[nodiscard]] inline int getInfiniteCellSizeFactorsOffset() const { return _infiniteCellSizeFactorsOffset; }
 
  [[nodiscard]] inline int getInfiniteCellSizeFactorsBlockSize() const { return _infiniteCellSizeFactorsBlockSize; }
 
  [[nodiscard]] inline std::set<ContainerOption>::iterator getContainerIterator() const { return _containerIt; }
 
  [[nodiscard]] inline std::set<double>::iterator getCellSizeFactorIterator() const { return _cellSizeFactorIt; }
 
  [[nodiscard]] inline std::set<TraversalOption>::iterator getTraversalIterator() const { return _traversalIt; }
 
  [[nodiscard]] inline std::set<LoadEstimatorOption>::iterator getLoadEstimatorIterator() const {
    return _loadEstimatorIt;
  }
 
  [[nodiscard]] inline std::set<DataLayoutOption>::iterator getDataLayoutIterator() const { return _dataLayoutIt; }
 
  [[nodiscard]] inline std::set<Newton3Option>::iterator getNewton3Iterator() const { return _newton3It; }
 
 private:
  void selectTraversalsForCurrentContainer();
 
  void selectLoadEstimatorsForCurrentContainerAndTraversal();
 
  inline void advanceConfigIterators();
 
  const std::set<ContainerOption> &_containers;
  const std::set<double> &_cellSizeFactors;
  const std::set<TraversalOption> &_allowedTraversalOptions;
  const std::set<LoadEstimatorOption> &_allowedLoadEstimatorOptions;
  const std::set<DataLayoutOption> &_dataLayoutOptions;
  const std::set<Newton3Option> &_newton3Options;
  const InteractionTypeOption &_interactionType;
  std::set<TraversalOption> _allowedAndApplicableTraversalOptions;
  std::set<LoadEstimatorOption> _allowedAndApplicableLoadEstimatorOptions;
  std::set<ContainerOption>::iterator _containerIt;
  std::set<double>::iterator _cellSizeFactorIt;
  std::set<TraversalOption>::iterator _traversalIt;
  std::set<LoadEstimatorOption>::iterator _loadEstimatorIt;
  std::set<DataLayoutOption>::iterator _dataLayoutIt;
  std::set<Newton3Option>::iterator _newton3It;
  int _rankIterator;
  int _remainingBlockSize;
  int _remainder;
  int _infiniteCellSizeFactorsOffset;
  int _infiniteCellSizeFactorsBlockSize;
};
 
}  // namespace autopas::utils