blob: b70df9ef7c8b0c49075d96efe668ee8d695d71ac (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/*************************************************************************
* Copyright (C) 2009-2010 Tavian Barnes <tavianator@gmail.com> *
* *
* 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 *
* <http://www.gnu.org/licenses/>. *
*************************************************************************/
#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 <typename Y>
class GenericHeunIntegrator : public GenericSimpleIntegrator<Y>
{
public:
typedef typename GenericSimpleIntegrator<Y>::Scalar Scalar;
typedef typename GenericSimpleIntegrator<Y>::Function Function;
GenericHeunIntegrator(Function f)
: GenericSimpleIntegrator<Y>(f, s_a, s_b) { }
~GenericHeunIntegrator() { }
private:
typedef typename GenericSimpleIntegrator<Y>::ACoefficients ACoefficients;
typedef typename GenericSimpleIntegrator<Y>::BCoefficients BCoefficients;
static ACoefficients s_a;
static BCoefficients s_b;
static Scalar s_a2Arr[1];
static std::vector<Scalar> s_aArr[1];
static Scalar s_bArr[2];
};
// Type alias
typedef GenericHeunIntegrator<double> HeunIntegrator;
// Implementation
template <typename Y>
typename GenericHeunIntegrator<Y>::Scalar
GenericHeunIntegrator<Y>::s_a2Arr[1] = {
Scalar(1)
};
template <typename Y>
std::vector<typename GenericHeunIntegrator<Y>::Scalar>
GenericHeunIntegrator<Y>::s_aArr[1] = {
std::vector<Scalar>(s_a2Arr, s_a2Arr + 1)
};
template <typename Y>
typename GenericHeunIntegrator<Y>::ACoefficients
GenericHeunIntegrator<Y>::s_a(s_aArr, s_aArr + 1);
template <typename Y>
typename GenericHeunIntegrator<Y>::Scalar
GenericHeunIntegrator<Y>::s_bArr[2] = {
Scalar(1)/Scalar(2), Scalar(1)/Scalar(2)
};
template <typename Y>
typename GenericHeunIntegrator<Y>::BCoefficients
GenericHeunIntegrator<Y>::s_b(s_bArr, s_bArr + 2);
}
#endif // VZ_HEUN_HPP
|
