diff --git a/simulationmodelfactory.cpp b/simulationmodelfactory.cpp index b0a2ec9..eb344ca 100644 --- a/simulationmodelfactory.cpp +++ b/simulationmodelfactory.cpp @@ -2,17 +2,25 @@ SimulationModelFactory::SimulationModelFactory() {} -std::unique_ptr SimulationModelFactory::create( - const std::string &simulationModelLabel) +std::unique_ptr +SimulationModelFactory::create(const std::string& simulationModelLabel) { - if (simulationModelLabel == DRMSimulationModel::label) { - return std::make_unique(); - } else if (simulationModelLabel == ChronosEulerSimulationModel::label) { - return std::make_unique(); - } - std::cerr << "Simulation model used for computing the optimization results was " - "not recognized" - << std::endl; - assert(false); - return nullptr; + if (simulationModelLabel == DRMSimulationModel::label) { + return std::make_unique(); + } else if (simulationModelLabel == ChronosEulerSimulationModel::label) { + return std::make_unique(); + } else if (simulationModelLabel == ChronosEulerLinearSimulationModel::label) { + return std::make_unique(); + } else if (simulationModelLabel == + ChronosEulerNonLinearSimulationModel::label) { + return std::make_unique(); + } else if (simulationModelLabel == LinearSimulationModel::label) { + return std::make_unique(); + } + std::cerr + << "Simulation model used for computing the optimization results was " + "not recognized" + << std::endl; + assert(false); + return nullptr; } diff --git a/simulationmodelfactory.hpp b/simulationmodelfactory.hpp index 543a011..f26576d 100644 --- a/simulationmodelfactory.hpp +++ b/simulationmodelfactory.hpp @@ -1,17 +1,19 @@ #ifndef SIMULATIONMODELFACTORY_HPP #define SIMULATIONMODELFACTORY_HPP +#include "chronoseulerlinearsimulationmodel.hpp" +#include "chronoseulernonlinearsimulationmodel.hpp" #include "chronoseulersimulationmodel.hpp" #include "der_leimer.hpp" #include "drmsimulationmodel.hpp" #include "linearsimulationmodel.hpp" #include -class SimulationModelFactory -{ +class SimulationModelFactory { public: - SimulationModelFactory(); - static std::unique_ptr create(const std::string &simulationModelLabel); + SimulationModelFactory(); + static std::unique_ptr + create(const std::string &simulationModelLabel); }; #endif // SIMULATIONMODELFACTORY_HPP