GaussianClusterLogger.h

Go to the documentation of this file.

 
#pragma once
 
#include <unordered_set>
 
#include "autopas/tuning/tuningStrategy/GaussianModel/GaussianModelTypes.h"
#include "autopas/tuning/tuningStrategy/GaussianModel/GaussianProcess.h"
 
namespace autopas {
 
class GaussianClusterLogger {
  const std::string node_start_marker = "GaussianCluster Graph: Nodes";
  const std::string edge_start_marker = "GaussianCluster Graph: Edges";
  const std::string end_marker = "GaussianCluster Graph: End";
 
  constexpr static auto streamMode = std::ios_base::out | std::ios_base::app;
 
 public:
  explicit GaussianClusterLogger(GaussianModelTypes::VectorToStringFun vecToStringFun,
                                 const std::string &outputSuffix = "");
 
  void setVectorToStringFun(const GaussianModelTypes::VectorToStringFun &fun);
 
  void add(const std::vector<GaussianProcess> &clusters,
           const std::vector<GaussianModelTypes::VectorDiscrete> &discreteVectorMap,
           const GaussianModelTypes::VectorContinuous &currentContinuous, const std::vector<double> &means,
           const std::vector<double> &vars, const GaussianModelTypes::NeighboursWeights &neighbourWeights);
 
  void flush();
 
 private:
  void reset();
 
  std::string _outputFileName;
 
  std::stringstream _nodeStream;
  std::stringstream _edgeStream;
 
  std::vector<GaussianModelTypes::VectorContinuous> _currentContinuous{};
 
  GaussianModelTypes::VectorToStringFun _vecToStringFun;
};
}  // namespace autopas