Particle.h

Go to the documentation of this file.

/*
 * Particle.h
 *
 *  Created on: 23.02.2010
 *      Author: eckhardw
 */
 
#pragma once
 
#include <array>
#include <string>
#include <algorithm>
#include "spdlog/spdlog.h"
#include "utils/ArrayUtils.h"
 
class Particle {
 
private:
 static int nextId;
 
    std::array<double, 3> x{};
 
    std::array<double, 3> v{};
 
    std::array<double, 3> f{};
 
    std::array<double, 3> old_f{};
 
    double m{};
 
    int type;
 
    double epsilon;
 
    double sigma;
 
    bool marked;
 
    int id;
 
    std::vector<int> directNeighbors;
 
    std::vector<int> diagonalNeighbors;
 
    bool fixed;
 
    std::vector<bool> forceMarker;
 
    std::vector<std::array<double,3>> forceAccumulator;
 
 
public:
 
    explicit Particle(int type = 0);
 
    Particle(const Particle &other);
 
    Particle(
        // for visualization, we need always 3 coordinates
        // -> in case of 2d, we use only the first and the second
        std::array<double, 3> x_arg, std::array<double, 3> v_arg, double m_arg, int type = 0, double epsilon_arg = 5, double sigma_arg = 1);
 
    void resetForce();
 
    double calculateEKin() const;
 
    double calculateEKinFlow(std::array<double, 3>& avgVelocity) const;
 
    void addDirectNeighbor(int idToAdd);
 
    void addDiagonalNeighbor(int idToAdd);
 
    bool isDirectNeighbor(Particle& neighbor);
 
    bool isDiagonalNeighbor(Particle& neighbor);
 
    virtual ~Particle();
 
 //Only needed for parallelization strategy "Reduction"
 
#ifdef _OPENMP
    void initializeForceAccumulator(int numberThreads);
 
    void addForceToAccumulator(std::array<double, 3>& force, int threadId);
 
    void subForceFromAccumulator(std::array<double, 3>& force, int threadId);
 
    void reduceForce();
#endif
 
    //Getter and setter
 
    [[nodiscard]] const std::array<double, 3> &getX() const;
 
    [[nodiscard]] const std::array<double, 3> &getV() const;
 
    [[nodiscard]] const std::array<double, 3> &getF() const;
 
    [[nodiscard]] const std::array<double, 3> &getOldF() const;
 
    [[nodiscard]] double getM() const;
 
    [[nodiscard]] int getType() const;
 
    [[nodiscard]] double getEpsilon() const;
 
    [[nodiscard]] double getSigma() const;
 
    [[nodiscard]] int getId() const;
 
    [[nodiscard]] bool isMarked()const;
 
    [[nodiscard]] std::vector<int> & getDirectNeighbors();
 
    [[nodiscard]] std::vector<int> & getDiagonalNeighbors();
 
    [[nodiscard]] bool isFixed() const;
 
    void setOldF(const std::array<double, 3> &oldF);
 
    void setF(const std::array<double, 3> &f);
 
    void setX(const std::array<double, 3> &x);
 
    void setV(const std::array<double, 3> &v);
 
    void setType(int type);
 
    void setMarked(bool status);
 
    void setFixed(bool status);
 
    bool operator==(Particle &other) const;
 
    [[nodiscard]] std::string toString() const;
 
    static void resetID();
};
 
std::ostream &operator<<(std::ostream &stream, Particle &p);