Cell.h

Go to the documentation of this file.

 
#pragma once
 
#include "Particle.h"
#include "utils/CellUtils.h"
#include <algorithm>
#include <array>
#include <bitset>
#include <forward_list>
#include <sstream>
#include <string>
#include <vector>
#define VEC_CONTAINS(vec, el) ((std::find((vec).begin(), (vec).end(), (el)) != (vec).end()))
 
class Cell {
    using ContainerType = std::forward_list<std::reference_wrapper<Particle>>;
 
  private:
    ContainerType m_particles{};
    std::array<double, 3> m_size;
    std::array<double, 3> m_position;
    std::vector<HaloLocation> m_haloLocation;
    std::vector<BorderLocation> m_borderLocation;
    std::vector<int> m_neighbors;
    CellType m_type;
    int m_index;
 
  public:
    /* constructor */
    Cell(const std::array<double, 3> &size, const std::array<double, 3> &position, CellType type, int index,
         const std::vector<HaloLocation> &haloLocation, const std::vector<BorderLocation> &borderLocation);
 
    /* iterator */
    ContainerType::iterator begin();
 
    ContainerType::iterator end();
 
    /* corner cells */
    std::array<double, 3> getRelativePosition(const Particle &p) const;
 
    void validatePosition(const std::array<double, 3> &relPos) const;
 
    bool isCorner(HaloLocation loc1, HaloLocation loc2) const;
 
    HaloLocation handle2DCorner(const std::array<double, 3> &relPos) const;
 
    HaloLocation handle3DCorner(const std::array<double, 3> &relPos) const;
 
    HaloLocation handle3DDiagonal(const std::array<double, 3> &relPos) const;
 
    HaloLocation handle3DTripleCorner(const std::array<double, 3> &relPos) const;
 
    /* main functionality */
    void addParticle(Particle &particle);
 
    void removeParticle(Particle &particle);
 
    HaloLocation getCornerRegion(const Particle &p) const;
 
    const std::array<double, 3> &getX() const;
 
    const std::array<double, 3> &getSize() const;
 
    const std::vector<HaloLocation> &getHaloLocation() const;
 
    const std::vector<BorderLocation> &getBorderLocation() const;
 
    std::vector<int> &getNeighbors();
 
    const std::vector<int> &getNeighbors() const;
 
    std::forward_list<std::reference_wrapper<Particle>> &getParticles();
 
    const std::forward_list<std::reference_wrapper<Particle>> &getParticles() const;
 
    CellType getType() const;
 
    int getIndex() const;
 
    std::string toString() const;
};