1 | // This file is a part of Framsticks SDK. http://www.framsticks.com/ |
---|
2 | // Copyright (C) 1999-2023 Maciej Komosinski and Szymon Ulatowski. |
---|
3 | // See LICENSE.txt for details. |
---|
4 | |
---|
5 | #include <string> |
---|
6 | #include <limits> |
---|
7 | #include <algorithm> |
---|
8 | #include <frams/util/multirange.h> |
---|
9 | #include <utility> |
---|
10 | #include "fH_general.h" |
---|
11 | |
---|
12 | using std::pair, std::to_string, std::numeric_limits; |
---|
13 | |
---|
14 | |
---|
15 | |
---|
16 | // Methods for loading handles |
---|
17 | |
---|
18 | const char *fH_part_names[FH_PART_PROPS_COUNT] = { "fr" }; // , "dn", "ing", "as" }; |
---|
19 | |
---|
20 | const char *fH_joint_names[FH_JOINT_PROPS_COUNT] = { "rotstif" }; //"stif", "stam" }; //actually we would want to disable "rotstif" too (we assume the default value is the best, lower values are good to test technical abilities of phenotypes, but in most cases not useful for evolution - the structures and muscles get weaker and even more elastic), but having no joint properties was never tested, so better leave at least one |
---|
21 | |
---|
22 | void fH_Handle::loadProperties(Param par) |
---|
23 | { |
---|
24 | // loading values for vectors |
---|
25 | for (int i = 0; i < dimensions; i++) |
---|
26 | { |
---|
27 | first[i] = par.getDouble(i); |
---|
28 | second[i] = par.getDouble(dimensions + i); |
---|
29 | } |
---|
30 | obj = par.getSelected(); |
---|
31 | } |
---|
32 | |
---|
33 | void fH_Builder::addHandle(fH_Handle *handle) |
---|
34 | { |
---|
35 | switch (handle->type) |
---|
36 | { |
---|
37 | case fHBodyType::JOINT: |
---|
38 | sticks.push_back((fH_StickHandle*)handle); |
---|
39 | break; |
---|
40 | case fHBodyType::NEURON: |
---|
41 | neurons.push_back((fH_NeuronHandle*)handle); |
---|
42 | break; |
---|
43 | case fHBodyType::CONNECTION: |
---|
44 | connections.push_back((fH_ConnectionHandle*)handle); |
---|
45 | break; |
---|
46 | } |
---|
47 | } |
---|
48 | |
---|
49 | // Methods for saving properties of handles in params |
---|
50 | |
---|
51 | void fH_Handle::saveProperties(Param &par) |
---|
52 | { |
---|
53 | par.select(obj); |
---|
54 | for (int i = 0; i < dimensions; i++) |
---|
55 | { |
---|
56 | par.setDouble(i, first[i]); |
---|
57 | par.setDouble(dimensions + i, second[i]); |
---|
58 | } |
---|
59 | } |
---|
60 | |
---|
61 | // Destructor of Builder |
---|
62 | |
---|
63 | fH_Builder::~fH_Builder() |
---|
64 | { |
---|
65 | for (fH_StickHandle *obj : sticks) |
---|
66 | { |
---|
67 | delete obj; |
---|
68 | } |
---|
69 | sticks.clear(); |
---|
70 | for (fH_NeuronHandle *obj : neurons) |
---|
71 | { |
---|
72 | delete obj; |
---|
73 | } |
---|
74 | neurons.clear(); |
---|
75 | for (fH_ConnectionHandle *obj : connections) |
---|
76 | { |
---|
77 | delete obj; |
---|
78 | } |
---|
79 | connections.clear(); |
---|
80 | |
---|
81 | if (stickparamtab) ParamObject::freeParamTab(stickparamtab); |
---|
82 | if (neuronparamtab) ParamObject::freeParamTab(neuronparamtab); |
---|
83 | if (connectionparamtab) ParamObject::freeParamTab(connectionparamtab); |
---|
84 | |
---|
85 | } |
---|
86 | |
---|
87 | // Methods for parsing genotype |
---|
88 | |
---|
89 | void fH_Builder::prepareParams() |
---|
90 | { |
---|
91 | for (int i = 0; i < dimensions; i++) // preparing first vector fields |
---|
92 | { |
---|
93 | string x = "x"; |
---|
94 | x += to_string(i); |
---|
95 | stickmut.addProperty(NULL, x.c_str(), HANDLE_VECTOR_TYPE, x.c_str(), "", PARAM_CANOMITNAME, 0, -1); |
---|
96 | neuronmut.addProperty(NULL, x.c_str(), HANDLE_VECTOR_TYPE, x.c_str(), "", PARAM_CANOMITNAME, 0, -1); |
---|
97 | connectionmut.addProperty(NULL, x.c_str(), HANDLE_VECTOR_TYPE, x.c_str(), "", PARAM_CANOMITNAME, 0, -1); |
---|
98 | |
---|
99 | } |
---|
100 | for (int i = 0; i < dimensions; i++) // preparing second vector fields |
---|
101 | { |
---|
102 | string y = "y"; |
---|
103 | y += to_string(i); |
---|
104 | stickmut.addProperty(NULL, y.c_str(), HANDLE_VECTOR_TYPE, y.c_str(), "", PARAM_CANOMITNAME, 0, -1); |
---|
105 | neuronmut.addProperty(NULL, y.c_str(), HANDLE_VECTOR_TYPE, y.c_str(), "", PARAM_CANOMITNAME, 0, -1); |
---|
106 | connectionmut.addProperty(NULL, y.c_str(), HANDLE_VECTOR_TYPE, y.c_str(), "", PARAM_CANOMITNAME, 0, -1); |
---|
107 | |
---|
108 | } |
---|
109 | |
---|
110 | Part p; |
---|
111 | for (int i = 0; i < FH_PART_PROPS_COUNT; i++) |
---|
112 | { |
---|
113 | stickmut.addProperty(&p.properties().getParamTab()[p.properties().findId(fH_part_names[i]) + p.properties().getGroupCount()], -1); |
---|
114 | } |
---|
115 | |
---|
116 | Joint j; |
---|
117 | for (int i = 0; i < FH_JOINT_PROPS_COUNT; i++) |
---|
118 | { |
---|
119 | stickmut.addProperty(&j.properties().getParamTab()[j.properties().findId(fH_joint_names[i]) + j.properties().getGroupCount()], -1); |
---|
120 | } |
---|
121 | stickmut.addProperty(NULL, "l", STICKH_LENGTH_TYPE, "length", "", 0, 0, -1); |
---|
122 | |
---|
123 | Neuro n; |
---|
124 | neuronmut.addProperty(&n.properties().getParamTab()[n.properties().findId(FH_PE_NEURO_DET) + n.properties().getGroupCount()], -1); |
---|
125 | |
---|
126 | Param tmp(f0_neuroconn_paramtab, NULL); |
---|
127 | connectionmut.addProperty(&tmp.getParamTab()[tmp.findId(FH_PE_CONN_WEIGHT) + tmp.getGroupCount()], -1); |
---|
128 | |
---|
129 | stickparamtab = ParamObject::makeParamTab((ParamInterface *)&stickmut, 0, 0, stickmut.firstMutableIndex()); |
---|
130 | neuronparamtab = ParamObject::makeParamTab((ParamInterface *)&neuronmut, 0, 0, neuronmut.firstMutableIndex()); |
---|
131 | connectionparamtab = ParamObject::makeParamTab((ParamInterface *)&connectionmut, 0, 0, connectionmut.firstMutableIndex()); |
---|
132 | } |
---|
133 | |
---|
134 | int fH_Builder::processLine(SString line, int linenumber, int begin, int end) |
---|
135 | { |
---|
136 | // Firstly, method determines if line describes joint, neuron or neural connection |
---|
137 | // and prepares corresponding ParamTab |
---|
138 | fH_Handle *handle = NULL; |
---|
139 | ParamEntry *tab = NULL; |
---|
140 | if (line.startsWith("j:")) //joint |
---|
141 | { |
---|
142 | handle = new fH_StickHandle(dimensions, begin, end); |
---|
143 | tab = stickparamtab; |
---|
144 | } |
---|
145 | else if (line.startsWith("n:")) //neuron |
---|
146 | { |
---|
147 | handle = new fH_NeuronHandle(dimensions, begin, end); |
---|
148 | tab = neuronparamtab; |
---|
149 | } |
---|
150 | else if (line.startsWith("c:")) //connection |
---|
151 | { |
---|
152 | handle = new fH_ConnectionHandle(dimensions, begin, end); |
---|
153 | tab = connectionparamtab; |
---|
154 | } |
---|
155 | else // could not determine type of a handle |
---|
156 | { |
---|
157 | string message = "Cannot determine handle type at line: " + to_string(linenumber); |
---|
158 | logMessage("fH_Builder", "processLine", LOG_ERROR, message.c_str()); |
---|
159 | return begin; |
---|
160 | } |
---|
161 | line = line.substr(2); // skip of "j:", "c:" or "n:" |
---|
162 | |
---|
163 | // Secondly, ParamObject for holding handle properties is created |
---|
164 | void *obj = ParamObject::makeObject(tab); |
---|
165 | Param par(tab, obj); |
---|
166 | par.setDefault(); |
---|
167 | ParamInterface::LoadOptions opts; |
---|
168 | |
---|
169 | // After preparing Param objects, vector values and body properties are parsed |
---|
170 | par.load(ParamInterface::FormatSingleLine, line, &opts); |
---|
171 | |
---|
172 | // If parsing failed, method writes error message and ends processing |
---|
173 | if (opts.parse_failed) |
---|
174 | { |
---|
175 | string message = "Error in parsing handle parameters at line: " + to_string(linenumber); |
---|
176 | logMessage("fH_Builder", "processLine", LOG_ERROR, message.c_str()); |
---|
177 | delete handle; |
---|
178 | ParamObject::freeObject(obj); |
---|
179 | return begin; |
---|
180 | } |
---|
181 | |
---|
182 | // If parsing ended successfully, parsed properties are loaded into handle fields |
---|
183 | handle->loadProperties(par); |
---|
184 | |
---|
185 | // In the end, ready handle is stored in an appropriate vector |
---|
186 | addHandle(handle); |
---|
187 | return 0; |
---|
188 | } |
---|
189 | |
---|
190 | int fH_Builder::parseGenotype(const SString &genotype) |
---|
191 | { |
---|
192 | // Firstly, number of dimensions is parsed |
---|
193 | int pos = 0; |
---|
194 | SString numdimensions; |
---|
195 | genotype.getNextToken(pos, numdimensions, '\n'); |
---|
196 | if (!ExtValue::parseInt(numdimensions.c_str(), dimensions, true, false)) |
---|
197 | { |
---|
198 | logMessage("fH_Builder", "parseGenotype", LOG_ERROR, "Could not parse number of dimensions"); |
---|
199 | return 1; |
---|
200 | } |
---|
201 | if (dimensions < 1) |
---|
202 | { |
---|
203 | logMessage("fH_Builder", "parseGenotype", LOG_ERROR, "Number of dimensions cannot be lower than 1"); |
---|
204 | return 1; |
---|
205 | } |
---|
206 | SString line; |
---|
207 | int linenumber = 2; |
---|
208 | |
---|
209 | // With known number of dimensions ParamTabs for handles are prepared |
---|
210 | prepareParams(); |
---|
211 | |
---|
212 | // After preparing Builder for parsing, each line is processed with processLine |
---|
213 | int lastpos = pos; |
---|
214 | while (genotype.getNextToken(pos, line, '\n')) |
---|
215 | { |
---|
216 | if (line.length() > 0) |
---|
217 | { |
---|
218 | int res = processLine(line, linenumber, lastpos, pos - 1); |
---|
219 | if (res != 0) |
---|
220 | { |
---|
221 | return res; |
---|
222 | } |
---|
223 | } |
---|
224 | lastpos = pos; |
---|
225 | linenumber++; |
---|
226 | } |
---|
227 | if (sticks.size() == 0) |
---|
228 | { |
---|
229 | logMessage("fH_Builder", "parseGenotype", LOG_ERROR, "Genotype does not contain any stick"); |
---|
230 | return 1; |
---|
231 | } |
---|
232 | return 0; |
---|
233 | } |
---|
234 | |
---|
235 | // Distance calculations |
---|
236 | |
---|
237 | double fH_Handle::dist(vector<double> left, vector<double> right) |
---|
238 | { |
---|
239 | double sum = 0; |
---|
240 | for (unsigned int i = 0; i < left.size(); i++) |
---|
241 | { |
---|
242 | sum += (left[i] - right[i]) * (left[i] - right[i]); |
---|
243 | } |
---|
244 | return sqrt(sum); |
---|
245 | } |
---|
246 | |
---|
247 | vector<double> fH_Handle::getVectorsAverage() |
---|
248 | { |
---|
249 | vector<double> result(dimensions, 0); |
---|
250 | for (int i = 0; i < dimensions; i++) |
---|
251 | { |
---|
252 | result[i] = (first[i] + second[i]) / 2; |
---|
253 | } |
---|
254 | return result; |
---|
255 | } |
---|
256 | |
---|
257 | double fH_StickHandle::distance(fH_Handle *right) |
---|
258 | { |
---|
259 | double distance = 0; |
---|
260 | switch (right->type) |
---|
261 | { |
---|
262 | case fHBodyType::JOINT: |
---|
263 | // distance is computed between second vector of current handle and first |
---|
264 | // vector of second handle |
---|
265 | distance = dist(second, right->first); |
---|
266 | break; |
---|
267 | case fHBodyType::NEURON: |
---|
268 | { |
---|
269 | // if neuron has to be connected to joint, then distance is calculated |
---|
270 | // between averages of both handles |
---|
271 | vector<double> avgs = getVectorsAverage(); |
---|
272 | vector<double> avgn = right->getVectorsAverage(); |
---|
273 | distance = dist(avgs, avgn); |
---|
274 | break; |
---|
275 | } |
---|
276 | case fHBodyType::CONNECTION: |
---|
277 | // it is impossible to calculate distance between Joint and Connection |
---|
278 | return numeric_limits<double>::quiet_NaN(); |
---|
279 | } |
---|
280 | return distance; |
---|
281 | } |
---|
282 | |
---|
283 | double fH_NeuronHandle::distance(fH_Handle *right) |
---|
284 | { |
---|
285 | double distance = 0; |
---|
286 | switch (right->type) |
---|
287 | { |
---|
288 | case fHBodyType::JOINT: |
---|
289 | { |
---|
290 | // if neuron has to be connected to joint, then distance is calculated |
---|
291 | // between averages of both handles |
---|
292 | vector<double> avgs = right->getVectorsAverage(); |
---|
293 | vector<double> avgn = getVectorsAverage(); |
---|
294 | distance = dist(avgs, avgn); |
---|
295 | break; |
---|
296 | } |
---|
297 | case fHBodyType::CONNECTION: |
---|
298 | // this calculation is meant for input neuron - it compares second vector |
---|
299 | // of neuron and first vector of connection |
---|
300 | distance = dist(second, right->first); |
---|
301 | break; |
---|
302 | case fHBodyType::NEURON: |
---|
303 | // it is impossible to calculate distance between two Neurons |
---|
304 | return numeric_limits<double>::quiet_NaN(); |
---|
305 | } |
---|
306 | return distance; |
---|
307 | } |
---|
308 | |
---|
309 | double fH_NeuronHandle::distance(fH_StickHandle *right, bool first) |
---|
310 | { |
---|
311 | vector<double> avgn = getVectorsAverage(); |
---|
312 | double distance = 0; |
---|
313 | if (first) |
---|
314 | { |
---|
315 | distance = dist(avgn, right->firstparthandle); |
---|
316 | } |
---|
317 | else |
---|
318 | { |
---|
319 | distance = dist(avgn, right->secondparthandle); |
---|
320 | } |
---|
321 | return distance; |
---|
322 | } |
---|
323 | |
---|
324 | double fH_ConnectionHandle::distance(fH_Handle *right) |
---|
325 | { |
---|
326 | double distance = 0; |
---|
327 | switch (right->type) |
---|
328 | { |
---|
329 | case fHBodyType::NEURON: |
---|
330 | // this calculation is meant for output neuron - it compares second vector |
---|
331 | // of connection and first vector of neuron |
---|
332 | distance = dist(second, right->first); |
---|
333 | break; |
---|
334 | case fHBodyType::JOINT: |
---|
335 | case fHBodyType::CONNECTION: |
---|
336 | // it is impossible to calculate distance between Connection and other |
---|
337 | // Connection or Joint |
---|
338 | return numeric_limits<double>::quiet_NaN(); |
---|
339 | } |
---|
340 | return distance; |
---|
341 | } |
---|
342 | |
---|
343 | // Creature build functions |
---|
344 | |
---|
345 | Part * fH_StickHandle::createPart(ParamEntry *tab, std::vector<fH_StickHandle *> *children, Model *model, bool createmapping) |
---|
346 | { |
---|
347 | Param par(tab, obj); |
---|
348 | double partprops[FH_PART_PROPS_COUNT]; |
---|
349 | for (int i = 0; i < FH_PART_PROPS_COUNT; i++) |
---|
350 | { |
---|
351 | partprops[i] = par.getDouble(2 * getDimensions() + i); |
---|
352 | } |
---|
353 | |
---|
354 | size_t stickscount = children->size() + 1; |
---|
355 | |
---|
356 | MultiRange ranges; |
---|
357 | ranges.add(begin, end); |
---|
358 | |
---|
359 | for (fH_StickHandle *child : (*children)) |
---|
360 | { |
---|
361 | par.select(child->obj); |
---|
362 | for (int i = 0; i < FH_PART_PROPS_COUNT; i++) |
---|
363 | { |
---|
364 | partprops[i] += par.getDouble(2 * getDimensions() + i); |
---|
365 | } |
---|
366 | ranges.add(child->begin, child->end); |
---|
367 | } |
---|
368 | |
---|
369 | for (int i = 0; i < FH_PART_PROPS_COUNT; i++) |
---|
370 | { |
---|
371 | partprops[i] /= stickscount; |
---|
372 | } |
---|
373 | |
---|
374 | Part *newpart = new Part(); |
---|
375 | |
---|
376 | model->addPart(newpart); |
---|
377 | |
---|
378 | newpart->friction = partprops[0]; //Macko 2023-06: TODO partprops[0] is different from par.getDoubleById("fr"); (and par.getDoubleById() is used in createJoint() below, investigate why this difference and whether hardcoded references to fields like here could be avoided? Only use fH_part_names if possible.) |
---|
379 | //newpart->density = partprops[1]; |
---|
380 | //newpart->ingest = partprops[2]; |
---|
381 | //newpart->assim = partprops[3]; |
---|
382 | |
---|
383 | if (createmapping) newpart->addMapping(ranges); |
---|
384 | |
---|
385 | return newpart; |
---|
386 | } |
---|
387 | |
---|
388 | Joint* fH_StickHandle::createJoint(ParamEntry *tab, Model *model, bool createmapping) |
---|
389 | { |
---|
390 | Param par(tab, obj); |
---|
391 | if (firstpart == NULL || secondpart == NULL) |
---|
392 | { |
---|
393 | return NULL; |
---|
394 | } |
---|
395 | Joint *newjoint = new Joint(); |
---|
396 | |
---|
397 | model->addJoint(newjoint); |
---|
398 | |
---|
399 | newjoint->rotstif = par.getDoubleById("rotstif"); |
---|
400 | //newjoint->stif = par.getDoubleById("stif"); |
---|
401 | //newjoint->stamina = par.getDoubleById("stam"); |
---|
402 | newjoint->attachToParts(firstpart, secondpart); |
---|
403 | if (createmapping) newjoint->addMapping(IRange(begin, end)); |
---|
404 | return newjoint; |
---|
405 | } |
---|
406 | |
---|
407 | void fH_Builder::buildBody() |
---|
408 | { |
---|
409 | // stickconnections vector holds information about connections between sticks. |
---|
410 | // Left side of pair should hold pointer to stick that is connected with second |
---|
411 | // vector, and right side of pair should hold pointer to stick that is connected |
---|
412 | // with first vector |
---|
413 | stickconnections.clear(); |
---|
414 | |
---|
415 | // if body consists of single stick, just add it to body |
---|
416 | if (sticks.size() == 1) |
---|
417 | { |
---|
418 | stickconnections.push_back(pair<fH_StickHandle *, fH_StickHandle *>(nullptr, sticks[0])); |
---|
419 | sticksorder.push_back(0); |
---|
420 | return; |
---|
421 | } |
---|
422 | |
---|
423 | vector<bool> remainingsticks(sticks.size(), true); |
---|
424 | |
---|
425 | // first we find two handles that have minimal distances between their second |
---|
426 | // and first vector |
---|
427 | fH_StickHandle *left = sticks[0]; |
---|
428 | fH_StickHandle *right = sticks[1]; |
---|
429 | double mindist = left->distance(right); |
---|
430 | int leftid = 0; |
---|
431 | int rightid = 1; |
---|
432 | for (unsigned int i = 0; i < sticks.size(); i++) |
---|
433 | { |
---|
434 | for (unsigned int j = i + 1; j < sticks.size(); j++) |
---|
435 | { |
---|
436 | double distance = sticks[i]->distance(sticks[j]); |
---|
437 | if (distance < mindist) |
---|
438 | { |
---|
439 | mindist = distance; |
---|
440 | left = sticks[i]; |
---|
441 | right = sticks[j]; |
---|
442 | leftid = i; |
---|
443 | rightid = j; |
---|
444 | } |
---|
445 | distance = sticks[j]->distance(sticks[i]); |
---|
446 | if (distance < mindist) |
---|
447 | { |
---|
448 | mindist = distance; |
---|
449 | left = sticks[j]; |
---|
450 | right = sticks[i]; |
---|
451 | leftid = j; |
---|
452 | rightid = i; |
---|
453 | } |
---|
454 | } |
---|
455 | } |
---|
456 | |
---|
457 | // two found handles are the beginning of creature body |
---|
458 | stickconnections.push_back(pair<fH_StickHandle *, fH_StickHandle *>(nullptr, left)); |
---|
459 | stickconnections.push_back(pair<fH_StickHandle *, fH_StickHandle *>(left, right)); |
---|
460 | |
---|
461 | // after selecting two handles as beginning of body, they are marked as used |
---|
462 | // in the list of remaining sticks |
---|
463 | remainingsticks[leftid] = false; |
---|
464 | remainingsticks[rightid] = false; |
---|
465 | |
---|
466 | sticksorder.push_back(leftid); |
---|
467 | sticksorder.push_back(rightid); |
---|
468 | |
---|
469 | // next stick is selected by minimum distance between first vector of its handle |
---|
470 | // and second vector of any existing StickHandle in body |
---|
471 | int remaining = int(sticks.size()) - 2; |
---|
472 | while (remaining > 0) |
---|
473 | { |
---|
474 | leftid = -1; |
---|
475 | rightid = -1; |
---|
476 | mindist = numeric_limits<double>::max(); |
---|
477 | for (unsigned int i = 0; i < sticks.size(); i++) |
---|
478 | { |
---|
479 | // if stick is not already in |
---|
480 | if (remainingsticks[i]) |
---|
481 | { |
---|
482 | for (int stickid : sticksorder) |
---|
483 | { |
---|
484 | double distance = sticks[stickid]->distance(sticks[i]); |
---|
485 | if (distance < mindist) |
---|
486 | { |
---|
487 | mindist = distance; |
---|
488 | leftid = stickid; |
---|
489 | rightid = i; |
---|
490 | } |
---|
491 | } |
---|
492 | } |
---|
493 | } |
---|
494 | stickconnections.push_back(pair<fH_StickHandle *, fH_StickHandle *>(sticks[leftid], sticks[rightid])); |
---|
495 | remainingsticks[rightid] = false; |
---|
496 | sticksorder.push_back(rightid); |
---|
497 | remaining--; |
---|
498 | } |
---|
499 | } |
---|
500 | |
---|
501 | int fH_Builder::developBrain(Model *model, bool createmapping) |
---|
502 | { |
---|
503 | Param par(neuronparamtab, NULL); |
---|
504 | // First of all, neurons are attached to body |
---|
505 | for (fH_NeuronHandle *currneu : neurons) |
---|
506 | { |
---|
507 | par.select(currneu->obj); |
---|
508 | // create Neuro object and set details |
---|
509 | currneu->neuron = new Neuro(); |
---|
510 | SString det = par.getStringById("d"); |
---|
511 | if (det != "") |
---|
512 | { |
---|
513 | currneu->neuron->setDetails(det); |
---|
514 | } |
---|
515 | else |
---|
516 | { |
---|
517 | currneu->neuron->setDetails("N"); |
---|
518 | } |
---|
519 | |
---|
520 | // get class of neuron. If class with given name does not exist - return error |
---|
521 | NeuroClass *nclass = currneu->neuron->getClass(); |
---|
522 | if (!nclass) |
---|
523 | { |
---|
524 | SString msg = "NeuroClass given in details \""; |
---|
525 | msg += det + "\" does not exist"; |
---|
526 | logMessage("fH_Builder", "developBrain", LOG_ERROR, msg.c_str()); |
---|
527 | delete currneu->neuron; |
---|
528 | return -1; |
---|
529 | } |
---|
530 | // add neuron to model -> required before attaching to body part |
---|
531 | model->addNeuro(currneu->neuron); |
---|
532 | if (nclass->getPreferredLocation() == 2) // attach to Joint |
---|
533 | { |
---|
534 | // find stick that has closest average handle to average handle of |
---|
535 | // neuron |
---|
536 | double mindist = currneu->distance(sticks[0]); |
---|
537 | fH_StickHandle *minstick = sticks[0]; |
---|
538 | for (unsigned int i = 1; i < sticks.size(); i++) |
---|
539 | { |
---|
540 | double distance = currneu->distance(sticks[i]); |
---|
541 | if (distance < mindist) |
---|
542 | { |
---|
543 | mindist = distance; |
---|
544 | minstick = sticks[i]; |
---|
545 | } |
---|
546 | } |
---|
547 | currneu->neuron->attachToJoint(minstick->joint); |
---|
548 | } |
---|
549 | else if (nclass->getPreferredLocation() == 1) // attach to Part |
---|
550 | { |
---|
551 | // in the beginning we take first part of first stick to calculate |
---|
552 | // distance between them as initial minimal distance |
---|
553 | double mindist = currneu->distance(sticks[0], true); |
---|
554 | Part *minpart = sticks[0]->firstpart; |
---|
555 | for (unsigned int i = 0; i < sticks.size(); i++) |
---|
556 | { |
---|
557 | // after this we take only second parts of following sticks to |
---|
558 | // avoid repetition (thats why we start from i = 0) |
---|
559 | double distance = currneu->distance(sticks[i], false); |
---|
560 | if (distance < mindist) |
---|
561 | { |
---|
562 | mindist = distance; |
---|
563 | minpart = sticks[i]->secondpart; |
---|
564 | } |
---|
565 | } |
---|
566 | currneu->neuron->attachToPart(minpart); |
---|
567 | } |
---|
568 | if (createmapping) currneu->neuron->addMapping(IRange(currneu->begin, currneu->end)); |
---|
569 | model->checkpoint(); |
---|
570 | } |
---|
571 | |
---|
572 | par.setParamTab(connectionparamtab); |
---|
573 | // Secondly, connections are created |
---|
574 | for (fH_ConnectionHandle *currcon : connections) |
---|
575 | { |
---|
576 | par.select(currcon->obj); |
---|
577 | // Connection is created as follows: |
---|
578 | // beginneu ---> endneu |
---|
579 | // distance between beginneu and connection is calculated as distance |
---|
580 | // between second handle of beginneu and first handle of connection. |
---|
581 | // This is why calculation is written as beginneu->distance(currcon). |
---|
582 | // In case of connection and endneu distance between them is calculated |
---|
583 | // as distance between second handle of connection and first handle of |
---|
584 | // endneu. This is why calculation is written as currcon->distance(endneu). |
---|
585 | |
---|
586 | fH_NeuronHandle *beginneu = NULL; |
---|
587 | double mindist = numeric_limits<double>::max(); |
---|
588 | // find beginning of connection |
---|
589 | for (fH_NeuronHandle *neuron : neurons) |
---|
590 | { |
---|
591 | // These method checked earlier if all neurons have valid classes. |
---|
592 | // If a neuron does not have output, then it's skipped from comparison. |
---|
593 | // Otherwise: |
---|
594 | if (neuron->neuron->getClass()->getPreferredOutput() > 0) |
---|
595 | { |
---|
596 | double distance = neuron->distance(currcon); |
---|
597 | if (distance < mindist) |
---|
598 | { |
---|
599 | mindist = distance; |
---|
600 | beginneu = neuron; |
---|
601 | } |
---|
602 | } |
---|
603 | } |
---|
604 | // if there was no neuron that could begin a connection, then return warning |
---|
605 | if (!beginneu) |
---|
606 | { |
---|
607 | // due to often appearance of connection genes in fB encoding, this |
---|
608 | // log message is commented |
---|
609 | // logMessage("fH_Builder", "developBrain", LOG_DEBUG, "There are no available neurons with outputs, connection could not be established"); |
---|
610 | continue; |
---|
611 | } |
---|
612 | |
---|
613 | fH_NeuronHandle *endneu = NULL; |
---|
614 | mindist = numeric_limits<double>::max(); |
---|
615 | // find ending of connection |
---|
616 | for (fH_NeuronHandle *neuron : neurons) |
---|
617 | { |
---|
618 | // Method checked earlier if all neurons have valid classes. |
---|
619 | // If neuron does not accept input or all inputs are already connected, |
---|
620 | // then it's skipped from comparison. |
---|
621 | // Otherwise: |
---|
622 | if (neuron->neuron->getClass()->getPreferredInputs() == -1 || |
---|
623 | neuron->neuron->getClass()->getPreferredInputs() > neuron->neuron->getInputCount()) |
---|
624 | { |
---|
625 | double distance = currcon->distance(neuron); |
---|
626 | if (distance < mindist) |
---|
627 | { |
---|
628 | mindist = distance; |
---|
629 | endneu = neuron; |
---|
630 | } |
---|
631 | } |
---|
632 | } |
---|
633 | // if there was no neuron that could end connection, then return warning |
---|
634 | if (!endneu) |
---|
635 | { |
---|
636 | // due to often appearance of connection genes in fB encoding, this |
---|
637 | // log message is commented |
---|
638 | // logMessage("fH_Builder", "developBrain", LOG_DEBUG, "There are no available neurons with free inputs, connection could not be established"); |
---|
639 | continue; |
---|
640 | } |
---|
641 | endneu->neuron->addInput(beginneu->neuron, par.getDoubleById("w")); |
---|
642 | if (createmapping) endneu->neuron->addMapping(IRange(currcon->begin, currcon->end)); |
---|
643 | model->checkpoint(); |
---|
644 | } |
---|
645 | return 0; |
---|
646 | } |
---|
647 | |
---|
648 | Pt3D fH_Builder::getNextDirection(int count, int number) |
---|
649 | { |
---|
650 | // In order to get evenly distributed sticks coming from the same Part, the method |
---|
651 | // uses an algorithm for even distribution of points on a sphere. There are several |
---|
652 | // methods to perform this, usually iterative. The method introduced |
---|
653 | // below offers not fully accurate, yet quite satisfying results. This is |
---|
654 | // the RSZ method (Rakhmanov, Saff and Zhou) with the use of the golden angle. |
---|
655 | // This method is based on the distribution of points along a spiral that covers |
---|
656 | // the sphere surface. |
---|
657 | |
---|
658 | // The following method works partially on spherical coordinates (r and theta is used). |
---|
659 | // The Z coordinate is from Cartesian coordinate system. The golden angle is used |
---|
660 | // to "iterate" along the spiral, while the Z coordinate is used to move down the |
---|
661 | // sphere. |
---|
662 | |
---|
663 | double golden_angle = M_PI * (3.0 - sqrt(5)); |
---|
664 | double dz = 2.0 / (double)count; |
---|
665 | double z = 1 - ((double)number + 0.5) * dz; |
---|
666 | double r = sqrt(1 - z * z); |
---|
667 | double theta = golden_angle * number; |
---|
668 | Pt3D vec; |
---|
669 | // In the end X and Y coordinates are calculated with current values of |
---|
670 | // r and theta. Value z is already calculated |
---|
671 | vec.x = r * cos(theta); |
---|
672 | vec.y = r * sin(theta); |
---|
673 | vec.z = z; |
---|
674 | vec.normalize(); |
---|
675 | return vec; |
---|
676 | } |
---|
677 | |
---|
678 | Orient fH_Builder::getRotationMatrixToFitVector(Pt3D currdir, Pt3D expecteddir) |
---|
679 | { |
---|
680 | Orient res; |
---|
681 | // first method normalizes vectors for easy calculations |
---|
682 | currdir.normalize(); |
---|
683 | expecteddir.normalize(); |
---|
684 | double c = currdir.dotProduct(expecteddir); // dot product of both vectors |
---|
685 | // if the dot product of both vectors equals 0 |
---|
686 | if (c == 0) |
---|
687 | { |
---|
688 | res.x.x = -1; |
---|
689 | res.x.y = 0; |
---|
690 | res.x.z = 0; |
---|
691 | |
---|
692 | res.y.x = 0; |
---|
693 | res.y.y = -1; |
---|
694 | res.y.z = 0; |
---|
695 | |
---|
696 | res.z.x = 0; |
---|
697 | res.z.y = 0; |
---|
698 | res.z.z = -1; |
---|
699 | } |
---|
700 | Pt3D v = Pt3D(0); // cross product of both vectors |
---|
701 | v.x = currdir.y * expecteddir.z - currdir.z * expecteddir.y; |
---|
702 | v.y = currdir.z * expecteddir.x - currdir.x * expecteddir.z; |
---|
703 | v.z = currdir.x * expecteddir.y - currdir.y * expecteddir.x; |
---|
704 | |
---|
705 | // Rotation matrix that enables aligning currdir to expecteddir comes from |
---|
706 | // following calculation |
---|
707 | // R = I + [v]_x + ([v]_x)^2 / (1+c) |
---|
708 | // where [v]_x is the skew-symmetric cross-product matrix of v |
---|
709 | res.x.x = 1 - (v.y * v.y + v.z * v.z) / (1 + c); |
---|
710 | res.x.y = v.z + (v.x * v.y) / (1 + c); |
---|
711 | res.x.z = -v.y + (v.x * v.z) / (1 + c); |
---|
712 | res.y.x = -v.z + (v.x * v.y) / (1 + c); |
---|
713 | res.y.y = 1 - (v.x * v.x + v.z * v.z) / (1 + c); |
---|
714 | res.y.z = v.x + (v.y * v.z) / (1 + c); |
---|
715 | res.z.x = v.y + (v.x * v.z) / (1 + c); |
---|
716 | res.z.y = -v.x + (v.y * v.z) / (1 + c); |
---|
717 | res.z.z = 1 - (v.x * v.x + v.y * v.y) / (1 + c); |
---|
718 | |
---|
719 | return res; |
---|
720 | } |
---|
721 | |
---|
722 | Model* fH_Builder::buildModel(bool using_checkpoints) |
---|
723 | { |
---|
724 | Model *model = new Model(); |
---|
725 | |
---|
726 | // At first, floating sticks are connected |
---|
727 | buildBody(); |
---|
728 | |
---|
729 | model->open(using_checkpoints); |
---|
730 | |
---|
731 | // Secondly, parts and joints are created |
---|
732 | // For every stick in body, starting with initial |
---|
733 | Param par(stickparamtab, NULL); |
---|
734 | for (int currid : sticksorder) |
---|
735 | { |
---|
736 | fH_StickHandle *currstick = sticks[currid]; |
---|
737 | fH_StickHandle *parent = NULL; |
---|
738 | // find parent of current stick - it is first element of pair, in which |
---|
739 | // current stick is second |
---|
740 | for (pair<fH_StickHandle *, fH_StickHandle *> conn : stickconnections) |
---|
741 | { |
---|
742 | if (conn.second == currstick) |
---|
743 | { |
---|
744 | parent = conn.first; |
---|
745 | break; |
---|
746 | } |
---|
747 | } |
---|
748 | |
---|
749 | // if parent is NULL, then create Part with current stick properties and |
---|
750 | // location at (0,0,0) |
---|
751 | if (!parent) |
---|
752 | { |
---|
753 | vector<fH_StickHandle *> emptylist; |
---|
754 | Part *firstpart = currstick->createPart(stickparamtab, &emptylist, model, createmapping); |
---|
755 | firstpart->p = Pt3D(0); |
---|
756 | currstick->firstpart = firstpart; |
---|
757 | currstick->firstparthandle = currstick->first; // this is used to calculate later distance between |
---|
758 | model->checkpoint(); |
---|
759 | } |
---|
760 | else //otherwise first part of current stick is the second part of previous stick |
---|
761 | { |
---|
762 | currstick->firstpart = parent->secondpart; |
---|
763 | currstick->firstparthandle = parent->secondparthandle; |
---|
764 | } |
---|
765 | // position of second part depends on two things |
---|
766 | // 1. direction of previous joint |
---|
767 | // 2. how many sticks are connected to the same parent |
---|
768 | // default direction of growth (without parent) is (1,0,0) |
---|
769 | Pt3D direction(1, 0, 0); |
---|
770 | Pt3D secondposition(currstick->firstpart->p); |
---|
771 | // if parent does exist, then determine how many sticks are connected to |
---|
772 | // parent and distribute them evenly on a sphere surrounding second part |
---|
773 | if (parent) |
---|
774 | { |
---|
775 | // improved RSZ method creates vectors that starts in |
---|
776 | // center of sphere (which will act as shared part), so direction |
---|
777 | // calculated below should point from shared part to previous part |
---|
778 | // in order to perform proper aligning |
---|
779 | direction = parent->secondpart->p - parent->firstpart->p; |
---|
780 | direction.normalize(); |
---|
781 | // determine how many sticks are connected to parent and when connection |
---|
782 | // between parent and current stick appear |
---|
783 | int count = 0; |
---|
784 | int id = -1; |
---|
785 | for (unsigned int i = 0; i < stickconnections.size(); i++) |
---|
786 | { |
---|
787 | if (stickconnections[i].first == parent) |
---|
788 | { |
---|
789 | if (stickconnections[i].second == currstick) |
---|
790 | { |
---|
791 | id = count; |
---|
792 | } |
---|
793 | count++; |
---|
794 | } |
---|
795 | } |
---|
796 | if (id == -1) |
---|
797 | { |
---|
798 | logMessage("fH_Builder", "buildModel", LOG_ERROR, "Invalid behaviour"); |
---|
799 | delete model; |
---|
800 | return NULL; |
---|
801 | } |
---|
802 | |
---|
803 | // if there is only one child, then don't change direction - continue |
---|
804 | // along axis of parent. Otherwise calculate direction of id-th stick |
---|
805 | // (that is currstick) with use of RSZ/Vogel method of distributing points |
---|
806 | // evenly on a sphere |
---|
807 | if (count > 1) |
---|
808 | { |
---|
809 | direction = parent->firstpart->p - parent->secondpart->p; |
---|
810 | direction.normalize(); |
---|
811 | // there has to be count+1 directions, so method needs to generate |
---|
812 | // count+1 evenly distributed points on a sphere to make vectors |
---|
813 | // from point (0,0,0) to those points. First generated vector |
---|
814 | // will act as parent joint direction vector |
---|
815 | Pt3D sphere0direction = getNextDirection(count + 1, 0); |
---|
816 | |
---|
817 | // First generated vector needs to be aligned to parent vector |
---|
818 | Orient rotmatrix = getRotationMatrixToFitVector(sphere0direction, direction); |
---|
819 | |
---|
820 | // Calculation of direction from sphere for currstick |
---|
821 | direction = getNextDirection(count + 1, id + 1); |
---|
822 | // Rotation matrix aligning |
---|
823 | direction = rotmatrix.transform(direction); |
---|
824 | direction.normalize(); |
---|
825 | } |
---|
826 | } |
---|
827 | |
---|
828 | // calculate second position |
---|
829 | par.select(currstick->obj); |
---|
830 | secondposition += direction * par.getDoubleById("l"); |
---|
831 | |
---|
832 | // find every stick connected to current stick in order to calculate second |
---|
833 | // part properties |
---|
834 | vector<fH_StickHandle *> children; |
---|
835 | currstick->secondparthandle = currstick->second; |
---|
836 | for (pair<fH_StickHandle *, fH_StickHandle *> conn : stickconnections) |
---|
837 | { |
---|
838 | if (conn.first == currstick) |
---|
839 | { |
---|
840 | children.push_back(conn.second); |
---|
841 | for (int i = 0; i < dimensions; i++) |
---|
842 | { |
---|
843 | currstick->secondparthandle[i] += conn.second->first[i]; |
---|
844 | } |
---|
845 | } |
---|
846 | } |
---|
847 | // create part from current stick and other sticks connected to this part |
---|
848 | Part *secondpart = currstick->createPart(stickparamtab, &children, model, createmapping); |
---|
849 | secondpart->p = secondposition; |
---|
850 | currstick->secondpart = secondpart; |
---|
851 | double count = (double)children.size() + 1; |
---|
852 | for (int i = 0; i < dimensions; i++) |
---|
853 | { |
---|
854 | currstick->secondparthandle[i] /= count; |
---|
855 | } |
---|
856 | |
---|
857 | //after creating second part connect two parts with joint |
---|
858 | Joint * joint = currstick->createJoint(stickparamtab, model, createmapping); |
---|
859 | if (!joint) |
---|
860 | { |
---|
861 | logMessage("fH_Builder", "buildModel", LOG_ERROR, "Joint cannot be created"); |
---|
862 | delete model; |
---|
863 | return NULL; |
---|
864 | |
---|
865 | } |
---|
866 | currstick->joint = joint; |
---|
867 | model->checkpoint(); |
---|
868 | } |
---|
869 | // after creating a body, attach neurons to body and link them according to |
---|
870 | // connections |
---|
871 | if (developBrain(model, createmapping) == -1) |
---|
872 | { |
---|
873 | delete model; |
---|
874 | return NULL; |
---|
875 | } |
---|
876 | model->close(); |
---|
877 | return model; |
---|
878 | } |
---|
879 | |
---|
880 | int fH_Builder::removeNeuronsWithInvalidClasses() |
---|
881 | { |
---|
882 | int count = (int)neurons.size(); |
---|
883 | if (count == 0) |
---|
884 | { |
---|
885 | return 0; |
---|
886 | } |
---|
887 | vector<fH_NeuronHandle *>::iterator it = neurons.begin(); |
---|
888 | Param par(neuronparamtab, NULL); |
---|
889 | while (it != neurons.end()) |
---|
890 | { |
---|
891 | par.select((*it)->obj); |
---|
892 | SString det = par.getStringById("d"); |
---|
893 | if (det == "") |
---|
894 | { |
---|
895 | it++; |
---|
896 | } |
---|
897 | else |
---|
898 | { |
---|
899 | Neuro *neu = new Neuro(); |
---|
900 | neu->setDetails(det); |
---|
901 | if (neu->getClass()) |
---|
902 | { |
---|
903 | it++; |
---|
904 | } |
---|
905 | else |
---|
906 | { |
---|
907 | fH_NeuronHandle *tmp = (*it); |
---|
908 | it = neurons.erase(it); |
---|
909 | delete tmp; |
---|
910 | } |
---|
911 | delete neu; |
---|
912 | } |
---|
913 | |
---|
914 | } |
---|
915 | return count - (int)neurons.size(); |
---|
916 | } |
---|
917 | |
---|
918 | SString fH_Builder::toString() |
---|
919 | { |
---|
920 | SString result = ""; |
---|
921 | result += to_string(dimensions).c_str(); |
---|
922 | result += "\n"; |
---|
923 | // first method stringifies parts |
---|
924 | Param par(stickparamtab, NULL); |
---|
925 | void *def = ParamObject::makeObject(stickparamtab); |
---|
926 | par.select(def); |
---|
927 | par.setDefault(); |
---|
928 | for (fH_StickHandle *currstick : sticks) |
---|
929 | { |
---|
930 | currstick->saveProperties(par); |
---|
931 | SString props; |
---|
932 | par.saveSingleLine(props, def, true, false); |
---|
933 | result += "j:"; |
---|
934 | result += props; |
---|
935 | } |
---|
936 | ParamObject::freeObject(def); |
---|
937 | par.setParamTab(neuronparamtab); |
---|
938 | def = ParamObject::makeObject(neuronparamtab); |
---|
939 | par.select(def); |
---|
940 | par.setDefault(); |
---|
941 | for (fH_NeuronHandle *currneuron : neurons) |
---|
942 | { |
---|
943 | currneuron->saveProperties(par); |
---|
944 | SString props; |
---|
945 | par.saveSingleLine(props, def, true, false); |
---|
946 | result += "n:"; |
---|
947 | result += props; |
---|
948 | } |
---|
949 | ParamObject::freeObject(def); |
---|
950 | par.setParamTab(connectionparamtab); |
---|
951 | def = ParamObject::makeObject(connectionparamtab); |
---|
952 | par.select(def); |
---|
953 | par.setDefault(); |
---|
954 | for (fH_ConnectionHandle *currconnection : connections) |
---|
955 | { |
---|
956 | currconnection->saveProperties(par); |
---|
957 | SString props; |
---|
958 | par.saveSingleLine(props, def, true, false); |
---|
959 | result += "c:"; |
---|
960 | result += props; |
---|
961 | } |
---|
962 | ParamObject::freeObject(def); |
---|
963 | return result; |
---|
964 | } |
---|
965 | |
---|
966 | ParamEntry* fH_Builder::getParamTab(fHBodyType type) |
---|
967 | { |
---|
968 | switch (type) |
---|
969 | { |
---|
970 | case fHBodyType::JOINT: |
---|
971 | return stickparamtab; |
---|
972 | break; |
---|
973 | case fHBodyType::NEURON: |
---|
974 | return neuronparamtab; |
---|
975 | break; |
---|
976 | default: |
---|
977 | return connectionparamtab; |
---|
978 | break; |
---|
979 | } |
---|
980 | } |
---|