bucket.cpp

Go to the documentation of this file.

#include "bucket.hpp"
 
using std::cerr;
using std::endl;
 
void Bucket::generate(const int& particleNum) {
    next.resize(particleNum);
}
 
Bucket::Bucket(const double& reMax, const Domain& domain, const size_t& particleSize) {
    this->length = reMax;
    this->domain = domain;
 
    this->numX = (int) (domain.xLength / length) + 3;
    this->numY = (int) (domain.yLength / length) + 3;
    this->numZ = (int) (domain.zLength / length) + 3;
    this->num  = numX * numY * numZ;
 
    this->first.resize(num);
    this->last.resize(num);
    this->next.resize(particleSize);
}
 
void Bucket::storeParticles(Particles& particles) {
 
#pragma omp parallel for
    for (int i = 0; i < num; i++) {
        first[i] = -1;
        last[i]  = -1;
    }
#pragma omp parallel for
    for (int i = 0; i < particles.size(); i++) {
        next[i] = -1;
    }
 
    for (auto& p : particles) {
        if (p.type == ParticleType::Ghost)
            continue;
 
        bool isInDomain = true;
        if (p.position.x() < domain.xMin || domain.xMax < p.position.x())
            isInDomain = false;
        if (p.position.y() < domain.yMin || domain.yMax < p.position.y())
            isInDomain = false;
        if (p.position.z() < domain.zMin || domain.zMax < p.position.z())
            isInDomain = false;
        if (!isInDomain) {
            cerr << "WARNING: particle " << p.id << " is out of domain." << endl;
            cerr << "x = " << p.position.x() << " ";
            cerr << "y = " << p.position.y() << " ";
            cerr << "z = " << p.position.z() << endl;
            p.type = ParticleType::Ghost;
            continue;
            // std::exit(-1);
        }
 
        int ix      = (int) ((p.position.x() - domain.xMin) / length) + 1;
        int iy      = (int) ((p.position.y() - domain.yMin) / length) + 1;
        int iz      = (int) ((p.position.z() - domain.zMin) / length) + 1;
        int iBucket = ix + iy * numX + iz * numX * numY;
 
        if (last[iBucket] == -1)
            first[iBucket] = p.id;
        else
            next[last[iBucket]] = p.id;
        last[iBucket] = p.id;
    }
}