GaussianPointCloud.h

Go to the documentation of this file.

 
 
#pragma once
 
// base code includes
#include <Debug.h>
#include <random>
 
namespace ttk {
 
  class GaussianPointCloud : virtual public Debug {
  public:
    GaussianPointCloud();
 
    template <class dataType>
    int castSample(const int &dimension,
                   std::mt19937 &gen,
                   dataType &x,
                   dataType &y,
                   dataType &z) const;
 
    // Execute the sampling
    template <class dataType>
    int generate(const int &dimension,
                 const int &numberOfSamples,
                 const int &seed,
                 dataType *const outputData) const;
  };
} // namespace ttk
 
template <class dataType>
int ttk::GaussianPointCloud::castSample(const int &dimension,
                                        std::mt19937 &gen,
                                        dataType &x,
                                        dataType &y,
                                        dataType &z) const {
 
  dataType u, v, w;
 
  std::uniform_real_distribution<dataType> uniformDistribution(-1, 1);
  u = uniformDistribution(gen);
  v = (dimension >= 2 ? uniformDistribution(gen) : 0);
  w = (dimension == 3 ? uniformDistribution(gen) : 0);
 
  dataType norm = std::sqrt(u * u + v * v + w * w);
 
  std::normal_distribution<dataType> normalDistribution{0, 1};
 
  dataType gaussianScale = normalDistribution(gen);
 
  x = (u / norm) * gaussianScale;
  y = (v / norm) * gaussianScale;
  z = (w / norm) * gaussianScale;
 
  return 0;
}
 
template <class dataType>
int ttk::GaussianPointCloud::generate(const int &dimension,
                                      const int &numberOfSamples,
                                      const int &seed,
                                      dataType *const outputData) const {
 
  Timer t;
 
  std::mt19937 gen{};
  gen.seed(seed);
 
  for(int i = 0; i < numberOfSamples; i++) {
    this->castSample(dimension, gen, outputData[3 * i], outputData[3 * i + 1],
                     outputData[3 * i + 2]);
  }
 
  this->printMsg(std::vector<std::vector<std::string>>{
    {"#Samples", std::to_string(numberOfSamples)}});
  this->printMsg("Samples generated in " + std::to_string(dimension) + "D", 1.0,
                 t.getElapsedTime(), 1);
 
  return 0;
}