Add Runge-Kutta numerical integration algorithm

marco-compiler · Jul 30, 2024 · 99c5ab9 · 99c5ab9
1 parent 4b2b0c7
commit 99c5ab9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/clang/include/clang/Driver/Options.td b/clang/include/clang/Driver/Options.td
@@ -8696,7 +8696,7 @@ def model : Joined<["--"], "model=">,
 def solver : Joined<["--"], "solver=">,
     Group<Simulation_Group>,
     Visibility<[MarcoOption, MC1Option]>,
-    HelpText<"Solver (possible options: euler-forward, ida)">;
+    HelpText<"Solver (possible options: euler-forward, ida, rk4, rk-euler-forward, rk-midpoint, rk-heun, rk-ralston, rk-heun-euler, rk-bogacki-shampine, rk-fehlberg, rk-cash-karp, rk-dormand-prince, rk-euler-backward)">;
 
 def variables_filter : Joined<["--"], "variables-filter=">,
     Group<Simulation_Group>,

diff --git a/clang/lib/Driver/ToolChains/CommonArgs.cpp b/clang/lib/Driver/ToolChains/CommonArgs.cpp
@@ -1373,6 +1373,10 @@ void tools::addMarcoLinkerArgs(
   } else if (solverString == "ida") {
     CmdArgs.push_back("-lMARCORuntimeDriverIDA");
     CmdArgs.push_back("-lMARCORuntimeSolverIDA");
+  } else if (solverString == "rk4" ||
+             solverString.starts_with("rk-")) {
+    CmdArgs.push_back("-lMARCORuntimeDriverRungeKutta");
+    CmdArgs.push_back("-lMARCORuntimeSolverRungeKutta");
   }
 
   // Add the remaining runtime libraries.