UniformGenerator.h

Go to the documentation of this file.

 
#pragma once
 
#include <array>
#include <random>
 
#include "autopas/particles/OwnershipState.h"
#include "autopas/utils/inBox.h"
 
namespace autopasTools::generators {
namespace UniformGenerator {
std::array<double, 3> randomPosition(std::mt19937 &generator, const std::array<double, 3> &boxMin,
                                     const std::array<double, 3> &boxMax);
 
template <class Container, class Particle>
void fillWithParticles(Container &container, const Particle &defaultParticle, const std::array<double, 3> &boxMin,
                       const std::array<double, 3> &boxMax, unsigned long numParticles = 100ul, unsigned int seed = 42);
 
template <class Container, class Particle, class HaloAddFunction>
void fillWithHaloParticles(Container &container, const Particle &defaultParticle, double haloWidth,
                           unsigned long numParticles, const HaloAddFunction &haloAddFunction, unsigned int seed = 42);
 
template <class Container, class Particle>
void fillWithHaloParticles(Container &container, const Particle &defaultParticle, double haloWidth,
                           unsigned long numParticles, unsigned int seed = 42) {
  fillWithHaloParticles(
      container, defaultParticle, haloWidth, numParticles,
      [](Container &c, Particle &p) {
        p.setOwnershipState(autopas::OwnershipState::halo);
        c.addHaloParticle(p);
      },
      seed);
}
};  // namespace UniformGenerator
 
template <class Container, class Particle>
void UniformGenerator::fillWithParticles(Container &container, const Particle &defaultParticle,
                                         const std::array<double, 3> &boxMin, const std::array<double, 3> &boxMax,
                                         unsigned long numParticles, unsigned int seed) {
  std::mt19937 generator(seed);
 
  for (unsigned long i = defaultParticle.getID(); i < defaultParticle.getID() + numParticles; ++i) {
    Particle particle(defaultParticle);
    particle.setR(randomPosition(generator, boxMin, boxMax));
    particle.setID(i);
    particle.setOwnershipState(autopas::OwnershipState::owned);
    container.addParticle(particle);
  }
}
 
template <class Container, class Particle, class HaloAddFunction>
void UniformGenerator::fillWithHaloParticles(Container &container, const Particle &defaultParticle, double haloWidth,
                                             unsigned long numParticles, const HaloAddFunction &haloAddFunction,
                                             unsigned int seed) {
  std::mt19937 generator(seed);
 
  auto haloBoxMin = container.getBoxMin();
  auto haloBoxMax = container.getBoxMax();
 
  // increase the box size not exactly by the width to make it exclusive
  for (unsigned int i = 0; i < 3; ++i) {
    haloBoxMin[i] -= haloWidth * .99;
    haloBoxMax[i] += haloWidth * .99;
  }
 
  for (unsigned long i = defaultParticle.getID(); i < defaultParticle.getID() + numParticles; ++i) {
    auto pos = randomPosition(generator, haloBoxMin, haloBoxMax);
    // we only want to add particles not in the actual box
    while (autopas::utils::inBox(pos, container.getBoxMin(), container.getBoxMax())) {
      pos = randomPosition(generator, haloBoxMin, haloBoxMax);
    }
    Particle particle(defaultParticle);
    particle.setR(pos);
    particle.setID(i);
    haloAddFunction(container, particle);
  }
}
}  // namespace autopasTools::generators