From d5639168232d1c111d6924b10c6abd74204abec1 Mon Sep 17 00:00:00 2001 From: Tavian Barnes Date: Wed, 6 Oct 2010 20:44:32 -0400 Subject: Add Heun's method. --- src/Makefile.am | 1 + src/vZ.hpp | 1 + src/vZ/Heun.hpp | 95 +++++++++++++++++++++++++++++++++++++++++++++++++++ src/vZ/Integrator.hpp | 1 - tests/Heun.cpp | 37 ++++++++++++++++++++ tests/Makefile.am | 2 ++ 6 files changed, 136 insertions(+), 1 deletion(-) create mode 100644 src/vZ/Heun.hpp create mode 100644 tests/Heun.cpp diff --git a/src/Makefile.am b/src/Makefile.am index ff03023..9ad815b 100644 --- a/src/Makefile.am +++ b/src/Makefile.am @@ -19,6 +19,7 @@ nobase_include_HEADERS = vZ.hpp \ vZ/Euler.hpp \ + vZ/Heun.hpp \ vZ/Integrator.hpp \ vZ/Midpoint.hpp \ vZ/RK.hpp \ diff --git a/src/vZ.hpp b/src/vZ.hpp index 9d47b33..7a311b1 100644 --- a/src/vZ.hpp +++ b/src/vZ.hpp @@ -29,6 +29,7 @@ #include #include #include +#include #include #endif // VZ_HPP diff --git a/src/vZ/Heun.hpp b/src/vZ/Heun.hpp new file mode 100644 index 0000000..b70df9e --- /dev/null +++ b/src/vZ/Heun.hpp @@ -0,0 +1,95 @@ +/************************************************************************* + * Copyright (C) 2009-2010 Tavian Barnes * + * * + * This file is part of The vZ Library. * + * * + * The vZ Library is free software; you can redistribute it and/or * + * modify it under the terms of the GNU Lesser General Public License as * + * published by the Free Software Foundation; either version 3 of the * + * License, or (at your option) any later version. * + * * + * The vZ Library is distributed in the hope that it will be useful, but * + * WITHOUT ANY WARRANTY; without even the implied warranty of * + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * + * Lesser General Public License for more details. * + * * + * You should have received a copy of the GNU Lesser General Public * + * License along with this program. If not, see * + * . * + *************************************************************************/ + +#ifndef VZ_HEUN_HPP +#define VZ_HEUN_HPP + +namespace vZ +{ + // Heun's method + // + // Second order Runge-Kutta method + // Two function evaluations per step + // Its tableau is: + // + // 0 | + // 1 | 1 + // --+--------- + // | 1/2 1/2 + // + // k1 = dt*f(y[n]) + // k2 = dt*f(y[n] + dt*k1) + // y[n + 1] = y[n] + 1/2*(k1 + k2) + template + class GenericHeunIntegrator : public GenericSimpleIntegrator + { + public: + typedef typename GenericSimpleIntegrator::Scalar Scalar; + typedef typename GenericSimpleIntegrator::Function Function; + + GenericHeunIntegrator(Function f) + : GenericSimpleIntegrator(f, s_a, s_b) { } + ~GenericHeunIntegrator() { } + + private: + typedef typename GenericSimpleIntegrator::ACoefficients ACoefficients; + typedef typename GenericSimpleIntegrator::BCoefficients BCoefficients; + + static ACoefficients s_a; + static BCoefficients s_b; + + static Scalar s_a2Arr[1]; + static std::vector s_aArr[1]; + static Scalar s_bArr[2]; + }; + + // Type alias + typedef GenericHeunIntegrator HeunIntegrator; + + // Implementation + + template + typename GenericHeunIntegrator::Scalar + GenericHeunIntegrator::s_a2Arr[1] = { + Scalar(1) + }; + + template + std::vector::Scalar> + GenericHeunIntegrator::s_aArr[1] = { + std::vector(s_a2Arr, s_a2Arr + 1) + }; + + template + typename GenericHeunIntegrator::ACoefficients + GenericHeunIntegrator::s_a(s_aArr, s_aArr + 1); + + template + typename GenericHeunIntegrator::Scalar + GenericHeunIntegrator::s_bArr[2] = { + Scalar(1)/Scalar(2), Scalar(1)/Scalar(2) + }; + + template + typename GenericHeunIntegrator::BCoefficients + GenericHeunIntegrator::s_b(s_bArr, s_bArr + 2); +} + +#endif // VZ_HEUN_HPP diff --git a/src/vZ/Integrator.hpp b/src/vZ/Integrator.hpp index 34bc870..9346eb8 100644 --- a/src/vZ/Integrator.hpp +++ b/src/vZ/Integrator.hpp @@ -82,5 +82,4 @@ namespace vZ } } - #endif // VZ_INTEGRATOR_HPP diff --git a/tests/Heun.cpp b/tests/Heun.cpp new file mode 100644 index 0000000..758524b --- /dev/null +++ b/tests/Heun.cpp @@ -0,0 +1,37 @@ +#include "vZ.hpp" +#include +#include +#include +#include + +// y' = y (y == C*exp(t)) +double +f(double t, double y) +{ + return y; +} + +int +main() +{ + vZ::HeunIntegrator integrator(f); + integrator.y(1.0).x(0.0).h(0.02); + + integrator.integrate(2.0); + + double actual = integrator.y(); + double expected = std::exp(2.0); + + std::cout << std::setprecision(10) + << "Numerical: " << actual << std::endl + << "Expected: " << expected << std::endl; + + double error = std::fabs(expected - actual)/expected; + if (error > 1.4e-4) { + std::cerr << "Error: " << 100.0*error << "%" << std::endl; + return EXIT_FAILURE; + } else { + std::cout << "Error: " << 100.0*error << "%" << std::endl; + return EXIT_SUCCESS; + } +} diff --git a/tests/Makefile.am b/tests/Makefile.am index edef1ae..0baa463 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -21,9 +21,11 @@ INCLUDES = -I$(top_srcdir)/src check_PROGRAMS = Euler-test \ Midpoint-test \ + Heun-test \ RK4-test TESTS = $(check_PROGRAMS) Euler_test_SOURCES = Euler.cpp Midpoint_test_SOURCES = Midpoint.cpp +Heun_test_SOURCES = Heun.cpp RK4_test_SOURCES = RK4.cpp -- cgit v1.2.3