Source file symplectic_d.ml

# 1 "src/ode/symplectic/symplectic_d.ml"
(*
 * OWL - OCaml Scientific and Engineering Computing
 * OWL-ODE - Ordinary Differential Equation Solvers
 *
 * Copyright (c) 2019 Ta-Chu Kao <tck29@cam.ac.uk>
 * Copyright (c) 2019 Marcello Seri <m.seri@rug.nl>
 *)

module Types = Owl_ode_base.Types

type mat = Owl_dense_matrix_d.mat

include Owl_ode_base.Symplectic_generic.Make (Owl_algodiff_primal_ops.D)

module Symplectic_Euler = struct
  type state = mat * mat
  type f = mat * mat -> float -> mat
  type step_output = (mat * mat) * float
  type solve_output = mat * mat * mat

  let step = symplectic_euler_s
  let solve = prepare step
end

module PseudoLeapfrog = struct
  type state = mat * mat
  type f = mat * mat -> float -> mat
  type step_output = (mat * mat) * float
  type solve_output = mat * mat * mat

  let step = pseudoleapfrog_s
  let solve = prepare step
end

module Leapfrog = struct
  type state = mat * mat
  type f = mat * mat -> float -> mat
  type step_output = (mat * mat) * float
  type solve_output = mat * mat * mat

  let step = leapfrog_s
  let solve = prepare step
end

module Ruth3 = struct
  type state = mat * mat
  type f = mat * mat -> float -> mat
  type step_output = (mat * mat) * float
  type solve_output = mat * mat * mat

  let step = ruth3_s
  let solve = prepare step
end

module Ruth4 = struct
  type state = mat * mat
  type f = mat * mat -> float -> mat
  type step_output = (mat * mat) * float
  type solve_output = mat * mat * mat

  let step = ruth4_s
  let solve = prepare step
end