Differential Algebraic Equations
This tutorial will introduce you to the functionality for solving DAEs. Other introductions can be found by checking out DiffEqTutorials.jl. This tutorial assumes you have read the Ordinary Differential Equations tutorial.
In this example we will solve the implicit ODE equation
f is the a variant of the Roberts equation. This equation is of the form
or is also known as a constrained differential equation where
g is the constraint equation. The Robertson model can be written in the form:
with initial conditions $y_1(0) = 1$, $y_2(0) = 0$, $y_3(0) = 0$, $dy_1 = - 0.04$, $dy_2 = 0.04$, and $dy_3 = 0.0$.
The workflow for DAEs is the same as for the other types of equations, where all you need to know is how to define the problem. A DAEProblem is specified by defining an in-place update
f(out,du,u,p,t) which uses the values to mutate
out as the output. To makes this into a DAE, we move all of the variables to one side. Thus we can define the function:
function f(out,du,u,p,t) out = - 0.04u + 1e4*u*u - du out = + 0.04u - 3e7*u^2 - 1e4*u*u - du out = u + u + u - 1.0 end
with initial conditions
u₀ = [1.0, 0, 0] du₀ = [-0.04, 0.04, 0.0] tspan = (0.0,100000.0)
and make the DAEProblem:
using DifferentialEquations differential_vars = [true,true,false] prob = DAEProblem(f,du₀,u₀,tspan,differential_vars=differential_vars)
differential_vars is an option which states which of the variables are differential, i.e. not purely algebraic (which means that their derivative shows up in the residual equations). This is required for the algorithm to be able to find consistant initial conditions. Notice that the first two variables are determined by their changes, but the last is simply determined by the conservation equation. Thus we use
differential_vars = [true,true,false].
As with the other DifferentialEquations problems, the commands are then to solve and plot. Here we will use the IDA solver from Sundials:
using Sundials sol = solve(prob,IDA())
In order to clearly see all the features of this solution, it should be plotted on a logarithmic scale. We'll also plot each on a different subplot to allow scaling the y-axis appropriately.
using Plots; plotly() # Using the Plotly backend plot(sol, xscale=:log10, tspan=(1e-6, 1e5), layout=(3,1))
This gives the following plot: