Model.h

Go to the documentation of this file.

//
// Created by daniel on 22.05.24.
//
 
#pragma once
#include "fileHandling/FileHandler.h"
#include "moleculeSimulator/forceCalculation/Force.h"
#include "particleRepresentation/container/ParticleContainer.h"
 
class Model {
    //Thermostats need direct access to the model to meassure and regulate the temperature of the particles.
    friend class DefaultThermostat;
    friend class Thermostat;
    friend class FlowThermostat;
 
private:
    FileHandler fileHandler;
    FileHandler::outputFormat outputFormat;
 
protected:
    ParticleContainer &particles;
    Force &force;
    double deltaT;
    bool gravityOn;
    std::array<double, 3> g;
 
    Model(ParticleContainer &particles, Force &force, double deltaT, FileHandler::outputFormat outputFormat,
          bool gravityOn, std::array<double, 3> g = {});
 
    void updatePositions() const;
 
    void updateVelocities() const;
 
    void applyGravity();
 
public:
    virtual ~Model() = default;
 
    void plot(int iteration, std::string &baseName);
 
    void addCuboid(const std::array<double, 3> &position, unsigned N1, unsigned N2, unsigned N3, double h, double mass,
                   const std::array<double, 3> &initVelocity, int dimensions, double brownianMotionAverageVelocity,
                   double epsilon = 5, double sigma = 1, bool fixed = false);
 
    void addDisc(const std::array<double, 3> &center,
                 const std::array<double, 3> &initVelocity, int N, double h, double mass, int dimensions,
                 double brownianMotionAverageVelocity, double epsilon = 5, double sigma = 1, bool fixed = false);
 
    void addSphere(const std::array<double, 3> &center, const std::array<double, 3> &initVelocity, int N, double h,
                   double mass, int dimensions, double brownianMotionAverageVelocity, double epsilon = 5, double sigma = 1,
                   bool fixed = false);
 
    void addParticle(Particle &p);
 
    void addViaFile(std::string &filepath, FileHandler::inputFormat inputFormat);
 
    void saveState();
 
    void updateForces() const;
 
    virtual void step(int iteration) = 0;
 
    [[nodiscard]] ParticleContainer &getParticles() const {
        return particles;
    }
 
    virtual void initializeForces() = 0;
};