FeatureVector.h

Go to the documentation of this file.

 
#pragma once
 
#include <Eigen/Core>
#include <vector>
 
#include "autopas/tuning/Configuration.h"
#include "autopas/utils/NumberSet.h"
#include "autopas/utils/Random.h"
 
namespace autopas {
 
class FeatureVector : public Configuration {
 public:
  static constexpr size_t featureSpaceDims = 4;
 
  using ContainerTraversalEstimatorOption = std::tuple<ContainerOption, TraversalOption, LoadEstimatorOption>;
 
  FeatureVector() : Configuration() {}
 
  FeatureVector(ContainerOption container, double cellSizeFactor, TraversalOption traversal,
                LoadEstimatorOption loadEstimator, DataLayoutOption dataLayout, Newton3Option newton3,
                InteractionTypeOption interactionType)
      : Configuration(container, cellSizeFactor, traversal, loadEstimator, dataLayout, newton3, interactionType) {}
 
  FeatureVector(Configuration conf) : Configuration(conf) {}
 
  Eigen::VectorXd operator-(const FeatureVector &other) const {
    Eigen::VectorXd result(featureSpaceDims);
    result << cellSizeFactor - other.cellSizeFactor,
        (container == other.container and traversal == other.traversal and loadEstimator == other.loadEstimator) ? 0.
                                                                                                                 : 1.,
        dataLayout == other.dataLayout ? 0. : 1., newton3 == other.newton3 ? 0. : 1.;
    return result;
  }
};
 
inline std::ostream &operator<<(std::ostream &os, const FeatureVector &featureVector) {
  return os << featureVector.toString();
}
 
}  // namespace autopas