imstkTimeIntegrator.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkVectorizedState.h"

namespace imstk
{
class TimeIntegrator
{
public:
    enum class Type
    {
        ForwardEuler,
        BackwardEuler,
        NewmarkBeta,
        CentralDifference,
        NoTimeStepper,
        None
    };

public:
    TimeIntegrator(Type type, double dT) : m_type(type), m_dT(dT), m_defaultDt(dT) {}
    virtual ~TimeIntegrator() = default;

    TimeIntegrator::Type getType() const { return m_type; }

    void setTimestepSize(const double dT) { m_dT = dT; }
    double getTimestepSize() const { return m_dT; }
    void setTimestepSizeToDefault() { m_dT = m_defaultDt; }

    void setDefaultTimestepSize(const double dT) { m_defaultDt = dT; }
    double getDefaultTimestepSize() const { return m_defaultDt; }

    virtual void updateStateGivenDv(std::shared_ptr<FeDeformBodyState> prevState, std::shared_ptr<FeDeformBodyState> currentState, Vectord& dV) = 0;
    virtual void updateStateGivenDu(std::shared_ptr<FeDeformBodyState> prevState, std::shared_ptr<FeDeformBodyState> currentState, Vectord& dU) = 0;
    virtual void updateStateGivenV(std::shared_ptr<FeDeformBodyState> prevState, std::shared_ptr<FeDeformBodyState> currentState, Vectord& v)   = 0;
    virtual void updateStateGivenU(std::shared_ptr<FeDeformBodyState> prevState, std::shared_ptr<FeDeformBodyState> currentState, Vectord& u)   = 0;

protected:
    Type   m_type; 
    double m_dT;   
    double m_defaultDt;
};
} // namespace imstk