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
56
57
58
59 public abstract R calculatePopulation(String subPopName, int generation,
60 P parameters);
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76 protected static void getFirstHalfyearSurvProb(Matrix2D target,
77 Matrix2D deathProbInfant1half, Matrix2D mortality, int numOfYears,
78 int maxAge) {
79
80 for (int year = 0; year < numOfYears; year++) {
81
82 double mortCurrentYear = mortality.getQuick(year, 0);
83 double mortNextYear = mortality.getQuick(year, 1);
84
85
86 target
87 .setQuick(
88 0,
89 year,
90 1.0 - ((deathProbInfant1half.getQuick(year, 0) * (mortCurrentYear - mortNextYear)) / mortCurrentYear));
91
92
93 for (int age = 1; age < maxAge; age++) {
94 mortCurrentYear = mortNextYear;
95 mortNextYear = mortality.getQuick(year, age + 1);
96 target
97 .setQuick(
98 age,
99 year,
100 1.0 - ((FORCE_MORT_INTERVAL * (mortCurrentYear - mortNextYear)) / mortCurrentYear));
101 }
102 }
103 }
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119 protected static void getSecondHalfyearSurvProb(Matrix2D target,
120 Matrix2D deathProbInfant1half, Matrix2D mortality, int numOfYears,
121 int maxAge) {
122
123 for (int year = 0; year < numOfYears; year++) {
124
125 double deathProbInfant = deathProbInfant1half.getQuick(year, 0);
126 double mortCurrentYear = mortality.getQuick(year, 0);
127 double mortNextYear = mortality.getQuick(year, 1);
128 double mortDiff = mortCurrentYear - mortNextYear;
129
130
131 target
132 .setQuick(
133 0,
134 year,
135 1.0 - (((1.0 - deathProbInfant) * mortDiff) / (mortCurrentYear - deathProbInfant
136 * mortDiff)));
137
138
139 for (int age = 1; age < maxAge; age++) {
140 mortCurrentYear = mortNextYear;
141 mortNextYear = mortality.getQuick(year, age + 1);
142 mortDiff = mortCurrentYear - mortNextYear;
143 target
144 .setQuick(
145 age,
146 year,
147 1.0 - ((FORCE_MORT_INTERVAL * mortDiff) / (mortCurrentYear - FORCE_MORT_INTERVAL
148 * mortDiff)));
149 }
150 }
151
152 }
153
154
155
156
157
158
159
160
161
162 protected static void calculateSurvivalProbabilities(
163 BasicParameters parameters, BasicResults results) {
164
165
166 getFirstHalfyearSurvProb(results.getP1f(),
167 parameters.getDeathProbInfant1halfFemale(), parameters.getMortXf(),
168 parameters.getNumOfYears(), parameters.getMaxAge());
169 getFirstHalfyearSurvProb(results.getP1m(),
170 parameters.getDeathProbInfant1halfMale(), parameters.getMortXm(),
171 parameters.getNumOfYears(), parameters.getMaxAge());
172
173
174 getSecondHalfyearSurvProb(results.getP2f(),
175 parameters.getDeathProbInfant1halfFemale(), parameters.getMortXf(),
176 parameters.getNumOfYears(), parameters.getMaxAge());
177 getSecondHalfyearSurvProb(results.getP2m(),
178 parameters.getDeathProbInfant1halfMale(), parameters.getMortXm(),
179 parameters.getNumOfYears(), parameters.getMaxAge());
180 }
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200 protected static void calculateByMultMin1(int startAge, int endAge, int year,
201 Matrix2D target, Matrix2D source, Matrix2D mult) {
202 for (int age = startAge; age < endAge; age++) {
203 target
204 .setQuick(
205 age,
206 year,
207 source.getQuick(age - 1, year - 1)
208 * mult.getQuick(age - 1, year - 1));
209 }
210 }
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229 protected static void calculateByMult(int startAge, int endAge, int year,
230 Matrix2D target, Matrix2D source, Matrix2D mult) {
231 for (int age = startAge; age < endAge; age++) {
232 target.setQuick(age, year,
233 source.getQuick(age, year) * mult.getQuick(age, year));
234 }
235 }
236
237
238
239
240
241
242
243
244
245
246
247
248 protected static double getNumOfChilds(Matrix2D meanFemalePop,
249 Matrix2D fertility, int year) {
250
251 double numOfChilds = 0;
252
253 for (int age = Constants.FERT_AGE_BEGIN; age < Constants.FERT_AGE_END; age++) {
254 numOfChilds += meanFemalePop.getQuick(age + 1, year)
255 * fertility.getQuick(year, age + 1);
256 }
257
258 return numOfChilds;
259 }
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282 protected static void calculateRestOfMeanPopulation(Matrix2D target,
283 Matrix2D endPopulation, Matrix2D p2, Matrix2D survO100,
284 Matrix2D liveBirthProp, int year, double numberOfChilds, int maxAge) {
285
286 target.setQuick(0, year, numberOfChilds * liveBirthProp.getQuick(year, 0));
287
288 target.setQuick(
289 maxAge,
290 year,
291 endPopulation.getQuick(maxAge - 1, year - 1)
292 * p2.getQuick(maxAge - 1, year)
293 + endPopulation.getQuick(maxAge, year - 1)
294 * survO100.getQuick(year, 0));
295 }
296
297 }