IntegratorFactory.hpp

Go to the documentation of this file.

#pragma once

 
#include <icarus/sim/integrators/Integrator.hpp>
#include <icarus/sim/integrators/IntegratorTypes.hpp>
#include <icarus/sim/integrators/RK45Integrator.hpp>
#include <icarus/sim/integrators/RK4Integrator.hpp>
#include <memory>
#include <stdexcept>
 
namespace icarus {

template <typename Scalar> class IntegratorFactory {
  public:
    static std::unique_ptr<Integrator<Scalar>> Create(const IntegratorConfig<Scalar> &config) {
        switch (config.type) {
        case IntegratorType::Euler:
            return std::make_unique<EulerIntegrator<Scalar>>();
 
        case IntegratorType::RK2:
            return std::make_unique<RK2Integrator<Scalar>>();
 
        case IntegratorType::RK4:
            return std::make_unique<RK4Integrator<Scalar>>();
 
        case IntegratorType::RK45: {
            auto rk45 = std::make_unique<RK45Integrator<Scalar>>(config.abs_tol, config.rel_tol);
            rk45->SetMinDt(config.min_dt);
            rk45->SetMaxDt(config.max_dt);
            rk45->SetSafetyFactor(config.safety_factor);
            return rk45;
        }
        }
        throw std::invalid_argument("Unknown integrator type");
    }

    static std::unique_ptr<Integrator<Scalar>> Create(const std::string &type_name) {
        return Create(IntegratorConfig<Scalar>::ForMethod(parse_integrator_type(type_name)));
    }

    static std::unique_ptr<Integrator<Scalar>> CreateDefault() {
        return Create(IntegratorConfig<Scalar>::RK4Default());
    }
};
 
} // namespace icarus