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
96
97
98
99
100
101
102
103
|
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.commons.math3.ode;
import org.apache.commons.math3.exception.DimensionMismatchException;
import org.apache.commons.math3.exception.MaxCountExceededException;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
/**
* Wrapper class to compute Jacobian matrices by finite differences for ODE which do not compute
* them by themselves.
*
* @since 3.0
*/
class ParameterJacobianWrapper implements ParameterJacobianProvider {
/** Main ODE set. */
private final FirstOrderDifferentialEquations fode;
/**
* Raw ODE without Jacobian computation skill to be wrapped into a ParameterJacobianProvider.
*/
private final ParameterizedODE pode;
/** Steps for finite difference computation of the Jacobian df/dp w.r.t. parameters. */
private final Map<String, Double> hParam;
/**
* Wrap a {@link ParameterizedODE} into a {@link ParameterJacobianProvider}.
*
* @param fode main first order differential equations set
* @param pode secondary problem, without parameter Jacobian computation skill
* @param paramsAndSteps parameters and steps to compute the Jacobians df/dp
* @see JacobianMatrices#setParameterStep(String, double)
*/
ParameterJacobianWrapper(
final FirstOrderDifferentialEquations fode,
final ParameterizedODE pode,
final ParameterConfiguration[] paramsAndSteps) {
this.fode = fode;
this.pode = pode;
this.hParam = new HashMap<String, Double>();
// set up parameters for jacobian computation
for (final ParameterConfiguration param : paramsAndSteps) {
final String name = param.getParameterName();
if (pode.isSupported(name)) {
hParam.put(name, param.getHP());
}
}
}
/** {@inheritDoc} */
public Collection<String> getParametersNames() {
return pode.getParametersNames();
}
/** {@inheritDoc} */
public boolean isSupported(String name) {
return pode.isSupported(name);
}
/** {@inheritDoc} */
public void computeParameterJacobian(
double t, double[] y, double[] yDot, String paramName, double[] dFdP)
throws DimensionMismatchException, MaxCountExceededException {
final int n = fode.getDimension();
if (pode.isSupported(paramName)) {
final double[] tmpDot = new double[n];
// compute the jacobian df/dp w.r.t. parameter
final double p = pode.getParameter(paramName);
final double hP = hParam.get(paramName);
pode.setParameter(paramName, p + hP);
fode.computeDerivatives(t, y, tmpDot);
for (int i = 0; i < n; ++i) {
dFdP[i] = (tmpDot[i] - yDot[i]) / hP;
}
pode.setParameter(paramName, p);
} else {
Arrays.fill(dFdP, 0, n, 0.0);
}
}
}
|
