TimeStepping.h

Go to the documentation of this file.

#pragma once
 
 
#include "common/ForwardDecl.h"
#include "objects/containers/Array.h"
#include "objects/geometry/Vector.h"
#include "objects/wrappers/SharedPtr.h"
 
NAMESPACE_SPH_BEGIN
 
class IScheduler;
 
class ITimeStepping : public Polymorphic {
protected:
    SharedPtr<Storage> storage;
 
    Float timeStep;
 
    Float maxTimeStep;
 
    AutoPtr<ITimeStepCriterion> criterion;
 
public:
    ITimeStepping(const SharedPtr<Storage>& storage, const RunSettings& settings);
 
    ITimeStepping(const SharedPtr<Storage>& storage,
        const RunSettings& settings,
        AutoPtr<ITimeStepCriterion>&& criterion);
 
    ~ITimeStepping() override;
 
    INLINE Float getTimeStep() const {
        return timeStep;
    }
 
    void step(IScheduler& scheduler, ISolver& solver, Statistics& stats);
 
protected:
    virtual void stepImpl(IScheduler& scheduler, ISolver& solver, Statistics& stats) = 0;
};
 
 
class EulerExplicit : public ITimeStepping {
public:
    explicit EulerExplicit(const SharedPtr<Storage>& storage, const RunSettings& settings)
        : ITimeStepping(storage, settings) {}
 
    virtual void stepImpl(IScheduler& scheduler, ISolver& solver, Statistics& stats) override;
};
 
class PredictorCorrector : public ITimeStepping {
private:
    SharedPtr<Storage> predictions;
 
public:
    PredictorCorrector(const SharedPtr<Storage>& storage, const RunSettings& settings);
 
    ~PredictorCorrector() override;
 
protected:
    virtual void stepImpl(IScheduler& scheduler, ISolver& solver, Statistics& stats) override;
 
    void makePredictions(IScheduler& scheduler);
 
    void makeCorrections(IScheduler& scheduler);
};
 
class LeapFrog : public ITimeStepping {
public:
    LeapFrog(const SharedPtr<Storage>& storage, const RunSettings& settings)
        : ITimeStepping(storage, settings) {}
 
protected:
    virtual void stepImpl(IScheduler& scheduler, ISolver& solver, Statistics& stats) override;
};
 
class RungeKutta : public ITimeStepping {
private:
    SharedPtr<Storage> k1, k2, k3, k4;
 
public:
    RungeKutta(const SharedPtr<Storage>& storage, const RunSettings& settings);
 
    ~RungeKutta() override;
 
protected:
    virtual void stepImpl(IScheduler& scheduler, ISolver& solver, Statistics& stats) override;
 
    void integrateAndAdvance(ISolver& solver, Statistics& stats, Storage& k, const Float m, const Float n);
};
 
 
class ModifiedMidpointMethod : public ITimeStepping {
private:
    SharedPtr<Storage> mid;
    Size n;
 
public:
    ModifiedMidpointMethod(const SharedPtr<Storage>& storage, const RunSettings& settings);
 
protected:
    virtual void stepImpl(IScheduler& scheduler, ISolver& solver, Statistics& stats) override;
};
 
class BulirschStoer : public ITimeStepping {
private:
    Float eps;
 
public:
    BulirschStoer(const SharedPtr<Storage>& storage, const RunSettings& settings);
 
protected:
    virtual void stepImpl(IScheduler& scheduler, ISolver& solver, Statistics& stats) override;
};
 
NAMESPACE_SPH_END