1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package p3j.simulation.calculation.deterministic;
17
18 import p3j.experiment.results.BasicResults;
19 import p3j.misc.math.Matrix2D;
20 import p3j.simulation.calculation.deterministic.parameters.BasicParameters;
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39 public abstract class AbstractPopulation<P extends BasicParameters, R extends BasicResults> {
40
41
42
43
44
45
46 public static final double FORCE_MORT_INTERVAL = 0.5;
47
48
49
50
51
52
53
54
55 public abstract R calculatePopulation(P parameters);
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71 protected static void getFirstHalfyearSurvProb(Matrix2D target,
72 Matrix2D deathProbInfant1half, Matrix2D mortality, int numOfYears,
73 int maxAge) {
74
75 for (int year = 0; year < numOfYears; year++) {
76
77 double mortCurrentYear = mortality.getQuick(year, 0);
78 double mortNextYear = mortality.getQuick(year, 1);
79
80
81 target
82 .setQuick(
83 0,
84 year,
85 1.0 - ((deathProbInfant1half.getQuick(year, 0) * (mortCurrentYear - mortNextYear)) / mortCurrentYear));
86
87
88 for (int age = 1; age < maxAge; age++) {
89 mortCurrentYear = mortNextYear;
90 mortNextYear = mortality.getQuick(year, age + 1);
91 target
92 .setQuick(
93 age,
94 year,
95 1.0 - ((FORCE_MORT_INTERVAL * (mortCurrentYear - mortNextYear)) / mortCurrentYear));
96 }
97 }
98 }
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114 protected static void getSecondHalfyearSurvProb(Matrix2D target,
115 Matrix2D deathProbInfant1half, Matrix2D mortality, int numOfYears,
116 int maxAge) {
117
118 for (int year = 0; year < numOfYears; year++) {
119
120 double deathProbInfant = deathProbInfant1half.getQuick(year, 0);
121 double mortCurrentYear = mortality.getQuick(year, 0);
122 double mortNextYear = mortality.getQuick(year, 1);
123 double mortDiff = mortCurrentYear - mortNextYear;
124
125
126 target
127 .setQuick(
128 0,
129 year,
130 1.0 - (((1.0 - deathProbInfant) * mortDiff) / (mortCurrentYear - deathProbInfant
131 * mortDiff)));
132
133
134 for (int age = 1; age < maxAge; age++) {
135 mortCurrentYear = mortNextYear;
136 mortNextYear = mortality.getQuick(year, age + 1);
137 mortDiff = mortCurrentYear - mortNextYear;
138 target
139 .setQuick(
140 age,
141 year,
142 1.0 - ((FORCE_MORT_INTERVAL * mortDiff) / (mortCurrentYear - FORCE_MORT_INTERVAL
143 * mortDiff)));
144 }
145 }
146
147 }
148
149
150
151
152
153
154
155
156
157 protected static void calculateSurvivalProbabilities(
158 BasicParameters parameters, BasicResults results) {
159
160
161 getFirstHalfyearSurvProb(results.getP1f(), parameters
162 .getDeathProbInfant1halfFemale(), parameters.getMortXf(), parameters
163 .getNumOfYears(), parameters.getMaxAge());
164 getFirstHalfyearSurvProb(results.getP1m(), parameters
165 .getDeathProbInfant1halfMale(), parameters.getMortXm(), parameters
166 .getNumOfYears(), parameters.getMaxAge());
167
168
169 getSecondHalfyearSurvProb(results.getP2f(), parameters
170 .getDeathProbInfant1halfFemale(), parameters.getMortXf(), parameters
171 .getNumOfYears(), parameters.getMaxAge());
172 getSecondHalfyearSurvProb(results.getP2m(), parameters
173 .getDeathProbInfant1halfMale(), parameters.getMortXm(), parameters
174 .getNumOfYears(), parameters.getMaxAge());
175 }
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195 protected static void calculateByMultMin1(int startAge, int endAge, int year,
196 Matrix2D target, Matrix2D source, Matrix2D mult) {
197 for (int age = startAge; age < endAge; age++) {
198 target.setQuick(age, year, source.getQuick(age - 1, year - 1)
199 * mult.getQuick(age - 1, year - 1));
200 }
201 }
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220 protected static void calculateByMult(int startAge, int endAge, int year,
221 Matrix2D target, Matrix2D source, Matrix2D mult) {
222 for (int age = startAge; age < endAge; age++) {
223 target.setQuick(age, year, source.getQuick(age, year)
224 * mult.getQuick(age, year));
225 }
226 }
227
228
229
230
231
232
233
234
235
236
237
238
239 protected static double getNumOfChilds(Matrix2D meanFemalePop,
240 Matrix2D fertility, int year) {
241
242 double numOfChilds = 0;
243
244 for (int age = Constants.FERT_AGE_BEGIN; age < Constants.FERT_AGE_END; age++) {
245 numOfChilds += meanFemalePop.getQuick(age + 1, year)
246 * fertility.getQuick(year, age + 1);
247 }
248
249 return numOfChilds;
250 }
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273 protected static void calculateRestOfMeanPopulation(Matrix2D target,
274 Matrix2D endPopulation, Matrix2D p2, Matrix2D survO100,
275 Matrix2D liveBirthProp, int year, double numberOfChilds, int maxAge) {
276
277 target.setQuick(0, year, numberOfChilds * liveBirthProp.getQuick(year, 0));
278
279 target
280 .setQuick(maxAge, year, endPopulation.getQuick(maxAge - 1, year - 1)
281 * p2.getQuick(maxAge - 1, year)
282 + endPopulation.getQuick(maxAge, year - 1)
283 * survO100.getQuick(year, 0));
284 }
285
286 }