source: cpp/frams/model/model.cpp @ 534

Last change on this file since 534 was 534, checked in by Maciej Komosinski, 8 years ago

Renamed: get/setGene -> get/setGenes, setGeneOnly -> setGenesAssumingSameFormat

  • Property svn:eol-style set to native
File size: 32.6 KB
Line 
1// This file is a part of Framsticks SDK.  http://www.framsticks.com/
2// Copyright (C) 1999-2015  Maciej Komosinski and Szymon Ulatowski.
3// See LICENSE.txt for details.
4
5#include <common/nonstd_math.h>
6#include "model.h"
7#include <common/log.h>
8#include <frams/util/multimap.h>
9#include <common/loggers/loggers.h>
10
11Model::Model()
12{
13        autobuildmaps = false;
14        init();
15}
16
17void Model::init()
18{
19        partmappingchanged = 0;
20        buildstatus = empty;
21        modelfromgenotype = 0;
22        startenergy = 1.0;
23        checklevel = 1;
24#ifdef MODEL_V1_COMPATIBLE
25        oldneurocount=-1; // == unknown
26        oldconnections=0;
27#endif
28        map = 0;
29        f0map = 0;
30        f0genoknown = 1;
31        shape = SHAPE_UNKNOWN;
32}
33
34void Model::moveElementsFrom(Model &source)
35{
36        int i;
37        open();
38        for (i = 0; i < source.getPartCount(); i++)
39                addPart(source.getPart(i));
40        for (i = 0; i < source.getJointCount(); i++)
41                addJoint(source.getJoint(i));
42        for (i = 0; i < source.getNeuroCount(); i++)
43                addNeuro(source.getNeuro(i));
44        source.parts.clear(); source.joints.clear(); source.neurons.clear();
45        source.clear();
46}
47
48void Model::internalCopy(const Model &mod)
49{
50        geno = mod.geno;
51        f0genoknown = 0;
52        startenergy = mod.startenergy;
53        if (mod.getStatus() == valid)
54        {
55                modelfromgenotype = mod.modelfromgenotype;
56                {for (int i = 0; i < mod.getPartCount(); i++)
57                        addPart(new Part(*mod.getPart(i))); }
58        {for (int i = 0; i < mod.getJointCount(); i++)
59        {
60                Joint *oldj = mod.getJoint(i);
61                Joint *j = new Joint(*oldj);
62                addJoint(j);
63                j->attachToParts(oldj->part1->refno, oldj->part2->refno);
64        }}
65        {for (int i = 0; i < mod.getNeuroCount(); i++)
66        {
67                Neuro *oldn = mod.getNeuro(i);
68                Neuro *n = new Neuro(*oldn);
69                addNeuro(n);
70                if (oldn->part_refno >= 0) n->attachToPart(oldn->part_refno);
71                else n->attachToJoint(oldn->joint_refno);
72        }}
73                for (int i = 0; i < mod.getNeuroCount(); i++)
74                {
75                        Neuro *oldn = mod.getNeuro(i);
76                        Neuro *n = getNeuro(i);
77                        for (int ni = 0; ni < oldn->getInputCount(); ni++)
78                        {
79                                double w;
80                                Neuro *oldinput = oldn->getInput(ni, w);
81                                SString info = n->getInputInfo(ni);
82                                n->addInput(getNeuro(oldinput->refno), w, &info);
83                        }
84                }
85        }
86}
87
88
89Model::Model(const Geno &src, bool buildmaps)
90        :autobuildmaps(buildmaps)
91{
92        init(src);
93}
94
95void Model::operator=(const Model &mod)
96{
97        clear();
98        open();
99        internalCopy(mod);
100        buildstatus = mod.buildstatus;
101}
102
103Model::Model(const Model &mod, bool buildmaps)
104        :autobuildmaps(buildmaps)
105{
106        init();
107        open();
108        internalCopy(mod);
109        buildstatus = mod.buildstatus;
110}
111
112void Model::init(const Geno &src)
113{
114        init();
115        modelfromgenotype = 1;
116        geno = src;
117        build();
118}
119
120void Model::resetAllDelta()
121{
122        for (int i = 0; i < getJointCount(); i++)
123                getJoint(i)->resetDelta();
124}
125
126void Model::useAllDelta(bool yesno)
127{
128        for (int i = 0; i < getJointCount(); i++)
129                getJoint(i)->useDelta(yesno);
130}
131
132Model::~Model()
133{
134        delmodel_list.action((intptr_t)this);
135        clear();
136}
137
138void Model::clear()
139{
140        FOREACH(Part*, p, parts)
141                delete p;
142        FOREACH(Joint*, j, joints)
143                delete j;
144        FOREACH(Neuro*, n, neurons)
145                delete n;
146        parts.clear(); joints.clear(); neurons.clear();
147        delMap();
148        delF0Map();
149        init();
150        geno = Geno();
151        f0geno = Geno();
152}
153
154Part *Model::addPart(Part *p)
155{
156        p->owner = this;
157        p->refno = parts.size();
158        parts += p;
159        return p;
160}
161
162Joint *Model::addJoint(Joint *j)
163{
164        j->owner = this;
165        j->refno = joints.size();
166        joints += j;
167        return j;
168}
169
170Neuro *Model::addNeuro(Neuro *n)
171{
172        n->owner = this;
173        n->refno = neurons.size();
174        neurons += n;
175        return n;
176}
177
178void Model::removeNeuros(SList &nlist)
179{
180        FOREACH(Neuro*, nu, nlist)
181        {
182                int i = findNeuro(nu);
183                if (i >= 0) removeNeuro(i);
184        }
185}
186
187void Model::removePart(int partindex, int removeattachedjoints, int removeattachedneurons)
188{
189        Part *p = getPart(partindex);
190        if (removeattachedjoints)
191        {
192                SList jlist;
193                findJoints(jlist, p);
194                FOREACH(Joint*, j, jlist)
195                {
196                        int i = findJoint(j);
197                        if (i >= 0) removeJoint(i, removeattachedneurons);
198                }
199        }
200        if (removeattachedneurons)
201        {
202                SList nlist;
203                findNeuros(nlist, 0, p);
204                removeNeuros(nlist);
205        }
206        parts -= partindex;
207        delete p;
208}
209
210void Model::removeJoint(int jointindex, int removeattachedneurons)
211{
212        Joint *j = getJoint(jointindex);
213        if (removeattachedneurons)
214        {
215                SList nlist;
216                findNeuros(nlist, 0, 0, j);
217                removeNeuros(nlist);
218        }
219        joints -= jointindex;
220        delete j;
221}
222
223void Model::removeNeuro(int neuroindex, bool removereferences)
224{
225        Neuro* thisN = getNeuro(neuroindex);
226
227        if (removereferences)
228        {
229                Neuro* n;
230                // remove all references to thisN
231                for (int i = 0; n = (Neuro*)neurons(i); i++)
232                {
233                        Neuro *inp;
234                        for (int j = 0; inp = n->getInput(j); j++)
235                                if (inp == thisN)
236                                {
237                                n->removeInput(j);
238                                j--;
239                                }
240                }
241        }
242
243        neurons -= neuroindex;
244        delete thisN;
245}
246
247MultiMap& Model::getMap()
248{
249        if (!map) map = new MultiMap();
250        return *map;
251}
252
253void Model::delMap()
254{
255        if (map) { delete map; map = 0; }
256}
257void Model::delF0Map()
258{
259        if (f0map) { delete f0map; f0map = 0; }
260}
261
262void Model::makeGenToGenMap(MultiMap& result, const MultiMap& gen1tomodel, const MultiMap& gen2tomodel)
263{
264        result.clear();
265        MultiMap m;
266        m.addReversed(gen2tomodel);
267        result.addCombined(gen1tomodel, m);
268}
269
270void Model::getCurrentToF0Map(MultiMap& result)
271{
272        result.clear();
273        if (!map) return;
274        const MultiMap& f0m = getF0Map();
275        makeGenToGenMap(result, *map, f0m);
276}
277
278void Model::rebuild(bool buildm)
279{
280        autobuildmaps = buildm;
281        clear();
282        build();
283}
284
285void Model::initMap()
286{
287        if (!map) map = new MultiMap();
288        else map->clear();
289}
290
291void Model::initF0Map()
292{
293        if (!f0map) f0map = new MultiMap();
294        else f0map->clear();
295}
296
297void Model::build()
298{
299        f0errorposition = -1;
300        f0warnposition = -1;
301        MultiMap *convmap = autobuildmaps ? new MultiMap() : NULL;
302        f0geno = (geno.getFormat() == '0') ? geno : geno.getConverted('0', convmap);
303        f0genoknown = 1;
304        if (f0geno.isInvalid())
305        {
306                buildstatus = invalid;
307                if (convmap) delete convmap;
308                return;
309        }
310        SString f0txt = f0geno.getGenes();
311        buildstatus = building; // was: open();
312        if (autobuildmaps)
313        {
314                partmappingchanged = 0;
315                initMap();
316                initF0Map();
317        }
318        int pos = 0, lnum = 1, lastpos = 0;
319        SString line;
320        MultiRange frommap;
321        LoggerToMemory mh(LoggerBase::Enable | LoggerBase::DontBlock);
322        for (; f0txt.getNextToken(pos, line, '\n'); lnum++)
323        {
324                if (autobuildmaps)
325                {
326                        frommap.clear();
327                        frommap.add(lastpos, pos - 1);
328                }
329                mh.reset();
330                if (singleStepBuild(line, lnum, autobuildmaps ? (&frommap) : 0) == -1)
331                {
332                        buildstatus = invalid;
333                        f0errorposition = lastpos;
334                        if (convmap) delete convmap;
335                        return;
336                }
337                if (mh.getWarningCount())
338                {
339                        if (f0warnposition < 0) f0warnposition = lastpos;
340                }
341                lastpos = pos;
342        }
343        mh.disable();
344        close();
345        if (convmap)
346        {
347                *f0map = *map;
348                if (geno.getFormat() != '0')
349                {
350                        MultiMap tmp;
351                        tmp.addCombined(*convmap, getMap());
352                        *map = tmp;
353                }
354                delete convmap;
355        }
356}
357
358const MultiMap& Model::getF0Map()
359{
360        if (!f0map)
361        {
362                f0map = new MultiMap();
363                makeGeno(f0geno, f0map);
364                f0genoknown = 1;
365        }
366        return *f0map;
367}
368
369Geno Model::rawGeno()
370{
371        Geno tmpgen;
372        makeGeno(tmpgen);
373        return tmpgen;
374}
375
376void Model::makeGeno(Geno &g, MultiMap *map, bool handle_defaults)
377{
378        if ((buildstatus != valid) && (buildstatus != building))
379        {
380                g = Geno(0, 0, 0, "invalid model");
381                return;
382        }
383
384        SString gen;
385
386        Param modelparam(f0_model_paramtab);
387        Param partparam(f0_part_paramtab);
388        Param jointparam(f0_joint_paramtab);
389        Param neuroparam(f0_neuro_paramtab);
390        Param connparam(f0_neuroconn_paramtab);
391
392        static Part defaultpart;
393        static Joint defaultjoint;
394        static Neuro defaultneuro;
395        static Model defaultmodel;
396        static NeuroConn defaultconn;
397        //static NeuroItem defaultneuroitem;
398
399        Part *p;
400        Joint *j;
401        Neuro *n;
402        int i;
403        int len;
404        int a, b;
405        //NeuroItem *ni;
406
407        SString mod_props;
408        modelparam.select(this);
409        modelparam.save2(mod_props, handle_defaults ? &defaultmodel : NULL, true, !handle_defaults);
410        if (mod_props.len() > 1) //are there any non-default values? ("\n" is empty)
411        {
412                gen += "m:";
413                gen += mod_props;
414        }
415
416        for (i = 0; p = (Part*)parts(i); i++)
417        {
418                partparam.select(p);
419                len = gen.len();
420                gen += "p:";
421                partparam.save2(gen, handle_defaults ? &defaultpart : NULL, true, !handle_defaults);
422                if (map)
423                        map->add(len, gen.len() - 1, partToMap(i));
424        }
425        for (i = 0; j = (Joint*)joints(i); i++)
426        {
427                jointparam.select(j);
428                len = gen.len();
429                jointparam.setParamTab(j->usedelta ? f0_joint_paramtab : f0_nodeltajoint_paramtab);
430                gen += "j:";
431                jointparam.save2(gen, handle_defaults ? &defaultjoint : NULL, true, !handle_defaults);
432                if (map)
433                        map->add(len, gen.len() - 1, jointToMap(i));
434        }
435        for (i = 0; n = (Neuro*)neurons(i); i++)
436        {
437                neuroparam.select(n);
438                len = gen.len();
439                gen += "n:";
440                neuroparam.save2(gen, handle_defaults ? &defaultneuro : NULL, true, !handle_defaults);
441                if (map)
442                        map->add(len, gen.len() - 1, neuroToMap(i));
443        }
444        for (a = 0; a < neurons.size(); a++)
445        { // inputs
446                n = (Neuro*)neurons(a);
447                //      if ((n->getInputCount()==1)&&(n->getInput(0).refno <= n->refno))
448                //              continue; // already done with Neuro::conn_refno
449
450                for (b = 0; b < n->getInputCount(); b++)
451                {
452                        double w;
453                        NeuroConn nc;
454                        Neuro* n2 = n->getInput(b, w);
455                        //              if (((n2.parentcount==1)&&(n2.parent)&&(n2.parent->refno < n2.refno)) ^
456                        //                  (n2.neuro_refno>=0))
457                        //                      printf("!!!! bad Neuro::neuro_refno ?!\n");
458
459                        //              if ((n2.parentcount==1)&&(n2.parent)&&(n2.parent->refno < n2.refno))
460                        //              if (n2.neuro_refno>=0)
461                        //                      continue; // already done with Neuro::neuro_refno
462
463                        nc.n1_refno = n->refno; nc.n2_refno = n2->refno;
464                        nc.weight = w;
465                        SString **s = n->inputInfo(b);
466                        if ((s) && (*s))
467                                nc.info = **s;
468                        connparam.select(&nc);
469                        len = gen.len();
470                        gen += "c:";
471                        connparam.save2(gen, handle_defaults ? &defaultconn : NULL, true, !handle_defaults);
472                        if (map)
473                                map->add(len, gen.len() - 1, neuroToMap(n->refno));
474                }
475        }
476        g = Geno(gen.c_str(), '0');
477}
478
479//////////////
480
481void Model::open()
482{
483        if (buildstatus == building) return;
484        buildstatus = building;
485        modelfromgenotype = 0;
486        partmappingchanged = 0;
487        f0genoknown = 0;
488        delMap();
489}
490
491void Model::checkpoint()
492{}
493
494void Model::setGeno(const Geno& newgeno)
495{
496        geno = newgeno;
497}
498
499void Model::clearMap()
500{
501        Part *p; Joint *j; Neuro *n;
502        int i;
503        delMap();
504        delF0Map();
505        for (i = 0; p = (Part*)parts(i); i++)
506                p->clearMapping();
507        for (i = 0; j = (Joint*)joints(i); i++)
508                j->clearMapping();
509        for (i = 0; n = (Neuro*)neurons(i); i++)
510                n->clearMapping();
511}
512
513int Model::close()
514{
515        if (buildstatus != building)
516                logPrintf("Model", "close", LOG_WARN, "Unexpected close() - no open()");
517        if (internalcheck(1) > 0)
518        {
519                buildstatus = valid;
520
521                if (partmappingchanged)
522                {
523                        getMap();
524                        Part *p; Joint *j; Neuro *n;
525                        int i;
526                        for (i = 0; p = (Part*)parts(i); i++)
527                                if (p->getMapping())
528                                        map->add(*p->getMapping(), partToMap(i));
529                        for (i = 0; j = (Joint*)joints(i); i++)
530                                if (j->getMapping())
531                                        map->add(*j->getMapping(), jointToMap(i));
532                        for (i = 0; n = (Neuro*)neurons(i); i++)
533                                if (n->getMapping())
534                                        map->add(*n->getMapping(), neuroToMap(i));
535                }
536        }
537        else
538                buildstatus = invalid;
539
540        return (buildstatus == valid);
541}
542
543int Model::validate()
544{
545        return internalcheck(0);
546}
547
548Pt3D Model::whereDelta(const Part& start, const Pt3D& rot, const Pt3D& delta)
549{
550        Orient roto;
551        roto = rot;
552        Orient o;
553        roto.transform(o, start.o);
554        //o.x=start.o/roto.x;
555        //o.y=start.o/roto.y;
556        //o.z=start.o/roto.z;
557        return o.transform(delta) + start.p;
558}
559
560int Model::singleStepBuild(const SString &singleline, const MultiRange* srcrange)
561{
562        return singleStepBuild(singleline, 0, srcrange);
563}
564
565int Model::singleStepBuild(const SString &singleline, int line_num, const MultiRange* srcrange)
566{
567        SString error_message;
568        int result = singleStepBuildNoLog(singleline, error_message, srcrange);
569        if (result < 0)
570        {
571                if (error_message.len() == 0) // generic error when no detailed message is available
572                        error_message = "Invalid f0 code";
573                if (line_num>0)
574                        error_message += SString::sprintf(", line #%d", line_num);
575                error_message += nameForErrors();
576                logPrintf("Model", "build", LOG_ERROR, "%s", error_message.c_str());
577        }
578        return result;
579}
580
581int Model::singleStepBuildNoLog(const SString &line, SString& error_message, const MultiRange* srcrange)
582{
583        error_message = SString::empty();
584        int pos = 0; const char*t = line.c_str();
585        for (; *t; t++, pos++)
586                if (!strchr(" \r\t", *t)) break;
587        if (*t == '#') return 0;
588        if (!*t) return 0;
589        if (!strncmp(t, "p:", 2))
590        {
591                Param partparam(f0_part_paramtab);
592                Part *p = new Part();
593                partparam.select(p);
594                pos += 2;
595                if (partparam.load2(line, pos) & ParamInterface::LOAD2_PARSE_FAILED) { delete p; error_message = "Invalid 'p:'"; return -1; }
596                p->o.rotate(p->rot);
597                parts += p;
598                p->owner = this;
599                if (srcrange) p->setMapping(*srcrange);
600                return getPartCount() - 1;
601        }
602        if (!strncmp(t, "m:", 2))
603        {
604                Param modelparam(f0_model_paramtab);
605                modelparam.select(this);
606                pos += 2;
607                if (modelparam.load2(line, pos) & ParamInterface::LOAD2_PARSE_FAILED) { error_message = "Invalid 'm:'"; return -1; }
608                return 0;
609        }
610        else if (!strncmp(t, "j:", 2))
611        {
612                Param jointparam(f0_joint_paramtab);
613                Joint *j = new Joint();
614                jointparam.select(j);
615                pos += 2;
616                j->owner = this;
617                if (jointparam.load2(line, pos) & ParamInterface::LOAD2_PARSE_FAILED) { delete j; error_message = "Invalid 'j:'"; return -1; }
618                bool p1_ok = false, p2_ok = false;
619                if ((p1_ok = ((j->p1_refno >= 0) && (j->p1_refno < getPartCount()))) &&
620                        (p2_ok = ((j->p2_refno >= 0) && (j->p2_refno < getPartCount()))))
621                {
622                        addJoint(j);
623                        if ((j->d.x != JOINT_DELTA_MARKER) || (j->d.y != JOINT_DELTA_MARKER) || (j->d.z != JOINT_DELTA_MARKER))
624                        {
625                                j->useDelta(1);
626                                j->resetDeltaMarkers();
627                        }
628                        j->attachToParts(j->p1_refno, j->p2_refno);
629                        if (srcrange) j->setMapping(*srcrange);
630                        return j->refno;
631                }
632                else
633                {
634                        error_message = SString::sprintf("Invalid reference to Part #%d", p1_ok ? j->p1_refno : j->p2_refno);
635                        delete j;
636                        return -1;
637                }
638        }
639        else if (!strncmp(t, "n:", 2)) // neuro (or the old neuro object as the special case)
640        {
641                Param neuroparam(f0_neuro_paramtab);
642                Neuro *nu = new Neuro();
643                neuroparam.select(nu);
644                pos += 2;
645                if (neuroparam.load2(line, pos) & ParamInterface::LOAD2_PARSE_FAILED) { delete nu; error_message = "Invalid 'n:'"; return -1; }
646#ifdef MODEL_V1_COMPATIBLE
647                if (nu->neuro_refno>=0) // parent specified...
648                {
649                        if (nu->neuro_refno >= getNeuroCount()) // and it's illegal
650                        {
651                                delete nu;
652                                return -1;
653                        }
654                        Neuro *parentNU=getNeuro(nu->neuro_refno);
655                        parentNU->addInput(nu,nu->weight);
656                        // default class for parented units: n-n link
657                        //if (nu->moredata.len()==0) nu->moredata="-";
658                }
659                else
660#endif
661                {
662                        // default class for unparented units: standard neuron
663                        if (nu->getClassName().len() == 0) nu->setClassName("N");
664                }
665                /*
666                        if (nu->conn_refno>=0) // input specified...
667                        {
668                        if (nu->conn_refno >= getNeuroCount()) // and it's illegal
669                        {
670                        delete nu;
671                        return -1;
672                        }
673                        Neuro *inputNU=getNeuro(nu->conn_refno);
674                        nu->addInput(inputNU,nu->weight);
675                        }
676                        */
677#ifdef MODEL_V1_COMPATIBLE
678                nu->weight=1.0;
679#endif
680                nu->owner = this;
681                // attach to part/joint
682                if (nu->part_refno >= 0)
683                        nu->attachToPart(nu->part_refno);
684                if (nu->joint_refno >= 0)
685                        nu->attachToJoint(nu->joint_refno);
686                if (srcrange) nu->setMapping(*srcrange);
687                // todo: check part/joint ref#
688#ifdef MODEL_V1_COMPATIBLE
689                if (hasOldNeuroLayout())
690                {
691                        int count=old_getNeuroCount();
692                        neurons.insert(count,nu);
693                        oldneurocount=count+1;
694                        return oldneurocount-1;
695                }
696                else
697#endif
698                {
699                        neurons += nu;
700                        return neurons.size() - 1;
701                }
702        }
703        else if (!strncmp(t, "c:", 2)) // add input
704        {
705                Param ncparam(f0_neuroconn_paramtab);
706                NeuroConn c;
707                ncparam.select(&c);
708                pos += 2;
709                if (ncparam.load2(line, pos) & ParamInterface::LOAD2_PARSE_FAILED) { error_message = "Invalid 'c:'"; return -1; }
710                bool n1_ok = false, n2_ok = false;
711                if ((n1_ok = ((c.n1_refno >= 0) && (c.n1_refno < getNeuroCount())))
712                        && (n2_ok = ((c.n2_refno >= 0) && (c.n2_refno < getNeuroCount()))))
713                {
714                        Neuro *na = getNeuro(c.n1_refno);
715                        Neuro *nb = getNeuro(c.n2_refno);
716                        na->addInput(nb, c.weight, &c.info);
717                        if (srcrange)
718                                na->addMapping(*srcrange);
719                        return 0;
720                }
721                error_message = SString::sprintf("Invalid reference to Neuro #%d", n1_ok ? c.n2_refno : c.n1_refno);
722                return -1;
723        }
724#ifdef MODEL_V1_COMPATIBLE
725        else if (!strncmp(t,"ni:",3)) // old neuroitem
726        {
727                // we always use old layout for "ni:"
728                Param neuroitemparam(f0_neuroitem_paramtab);
729                Neuro *nu=new Neuro();
730                neuroitemparam.select(nu);
731                pos+=3;
732                if (neuroitemparam.load2(line,pos) & ParamInterface::LOAD2_PARSE_FAILED) {delete nu; return -1;}
733                // illegal parent?
734                if ((nu->neuro_refno<0)||(nu->neuro_refno>=old_getNeuroCount()))
735                {
736                        delete nu;
737                        return -1;
738                }
739                Neuro *parentN=getNeuro(nu->neuro_refno);
740                // copy part/joint refno from parent, if possible
741                if ((nu->part_refno<0)&&(parentN->part_refno>=0))
742                        nu->part_refno=parentN->part_refno;
743                if ((nu->joint_refno<0)&&(parentN->joint_refno>=0))
744                        nu->joint_refno=parentN->joint_refno;
745                nu->owner=this;
746                // attach to part/joint
747                if (nu->part_refno>=0)
748                        nu->attachToPart(nu->part_refno);
749                if (nu->joint_refno>=0)
750                        nu->attachToJoint(nu->joint_refno);
751                if (srcrange)
752                        nu->setMapping(*srcrange);
753                // special case: old muscles
754                // PARENT neuron will be set up to be the CHILD of the current one (!)
755                if (nu->isOldEffector())
756                {
757                        nu->neuro_refno=parentN->refno;
758                        neurons+=nu;
759                        nu->owner=this;
760                        nu->addInput(parentN,nu->weight); // (!)
761                        nu->weight=1.0;
762                        parentN->invalidateOldItems();
763                        return 0; // !!! -> ...getItemCount()-1;
764                }
765                parentN->addInput(nu,nu->weight);
766                neurons+=nu;
767                parentN->invalidateOldItems();
768                if (nu->getClassName().len()==0)
769                {
770                        nu->setClassName("-");
771                        // for compatibility, "ni" can define a n-n connection
772                        // referring to non-existent neuron (which will be hopefully defined later)
773                        // internal check will add the proper input to this unit
774                        // if conn_refno >=0 and input count==0
775                        oldconnections=1;
776                        if (srcrange)
777                                parentN->addMapping(*srcrange);
778                }
779                else
780                        nu->weight=1.0;
781                return 0; // !!! -> ...getItemCount()-1;
782        }
783#endif
784        else return -1;
785}
786
787#ifdef MODEL_V1_COMPATIBLE
788int Model::addOldConnectionsInputs()
789{
790        if (!oldconnections) return 1;
791        Neuro* n;
792        for (int i=0;i<neurons.size();i++)
793        {
794                n=(Neuro*)neurons(i);
795                if (n->conn_refno>=0)
796                        if (n->isNNConnection())
797                                if (n->conn_refno < old_getNeuroCount())
798                                { // good reference
799                        Neuro* parent=n->parent; // nn connection has exactly one parent
800                        int inp=parent->findInput(n);
801                        Neuro* target=getNeuro(n->conn_refno);
802                        parent->setInput(inp,target,n->weight);
803                        removeNeuro(i,0); // no need to remove references
804                        i--;
805                                }
806                                else
807                                {
808                                        logPrintf("Model","internalCheck",LOG_ERROR,
809                                                "illegal N-N connection #%d (reference to #%d)%s",
810                                                i,n->conn_refno,nameForErrors().c_str());
811                                        return 0;
812                                }
813        }
814        oldconnections=0;
815        return 1;
816}
817#endif
818
819/////////////
820
821/** change the sequence of neuro units
822        and fix references in "-" objects (n-n connections)  */
823void Model::moveNeuro(int oldpos, int newpos)
824{
825        if (oldpos == newpos) return; // nop!
826        Neuro *n = getNeuro(oldpos);
827        neurons -= oldpos;
828        neurons.insert(newpos, n);
829        // conn_refno could be broken -> fix it
830#ifdef MODEL_V1_COMPATIBLE
831        for (int i=0;i<neurons.size();i++)
832        {
833                Neuro *n2=getNeuro(i);
834                if (n2->isNNConnection())
835                        if (n2->conn_refno == oldpos) n2->conn_refno=newpos;
836                        else
837                        { if (n2->conn_refno > oldpos) n2->conn_refno--;
838                if (n2->conn_refno >= newpos) n2->conn_refno++; }
839        }
840        invalidateOldNeuroCount();
841#endif
842}
843
844#ifdef MODEL_V1_COMPATIBLE
845/** move all old neurons (class "N") to the front of the neurons list.
846        @return number of old neurons
847        */
848int Model::reorderToOldLayout()
849{
850        Neuro *n;
851        int neurocount=0;
852        for (int i=0;i<neurons.size();i++)
853        {
854                n=(Neuro*)neurons(i);
855                if (n->isOldNeuron())
856                {
857                        moveNeuro(i,neurocount);
858                        neurocount++;
859                        i=neurocount-1;
860                }
861        }
862        return neurocount;
863}
864#endif
865////////////
866
867void Model::updateNeuroRefno()
868{
869        for (int i = 0; i < neurons.size(); i++)
870        {
871                Neuro* n = (Neuro*)neurons(i);
872                n->refno = i;
873        }
874}
875
876#define VALIDMINMAX(var,template,field) \
877if (var -> field < getMin ## template () . field) \
878        { var->field= getMin ## template () . field; \
879        logPrintf("Model","internalCheck",LOG_WARN,# field " too small in " # template " #%d (adjusted)",i);} \
880else if (var -> field > getMax ## template () . field) \
881        { var->field= getMax ## template ()  . field; \
882        logPrintf("Model","internalCheck",LOG_WARN,# field " too big in " # template " #%d (adjusted)",i);}
883
884#define LINKFLAG 0x8000000
885
886void Model::updateRefno()
887{
888        int i;
889        for (i = 0; i < getPartCount(); i++) getPart(i)->refno = i;
890        for (i = 0; i < getJointCount(); i++) getJoint(i)->refno = i;
891        for (i = 0; i < getNeuroCount(); i++) getNeuro(i)->refno = i;
892}
893
894SString Model::nameForErrors() const
895{
896        if (geno.getName().len()>0)
897                return SString::sprintf(" in '%s'", geno.getName().c_str());
898        return SString::empty();
899}
900
901int Model::internalcheck(int final)
902{
903        Part *p;
904        Joint *j;
905        Neuro *n;
906        int i, k;
907        int ret = 1;
908        shape = SHAPE_UNKNOWN;
909        if ((parts.size() == 0) && (neurons.size() == 0)) return 0;
910        if (parts.size() == 0)
911                size = Pt3D_0;
912        else
913        {
914                Pt3D bbmin = ((Part*)parts(0))->p, bbmax = bbmin;
915                for (i = 0; i < parts.size(); i++)
916                {
917                        p = (Part*)parts(i);
918                        p->owner = this;
919                        p->refno = i;
920                        if (p->shape == 0)
921                        {
922                                if (checklevel > 0)
923                                        p->mass = 0.0;
924                        }
925                        //VALIDMINMAX(p,part,mass);//mass is very special
926                        // VALIDMINMAX are managed manually when adding part properties in f0-def!
927                        // (could be made dynamic but not really worth the effort)
928                        VALIDMINMAX(p, Part, size);
929                        VALIDMINMAX(p, Part, scale.x);
930                        VALIDMINMAX(p, Part, scale.y);
931                        VALIDMINMAX(p, Part, scale.z);
932                        VALIDMINMAX(p, Part, hollow);
933                        VALIDMINMAX(p, Part, density);
934                        VALIDMINMAX(p, Part, friction);
935                        VALIDMINMAX(p, Part, ingest);
936                        VALIDMINMAX(p, Part, assim);
937                        VALIDMINMAX(p, Part, vsize);
938                        VALIDMINMAX(p, Part, vcolor.x);
939                        VALIDMINMAX(p, Part, vcolor.y);
940                        VALIDMINMAX(p, Part, vcolor.z);
941                        p->flags &= ~LINKFLAG; // for delta joint cycle detection
942                        if (p->p.x - p->size < bbmin.x) bbmin.x = p->p.x - p->size;
943                        if (p->p.y - p->size < bbmin.y) bbmin.y = p->p.y - p->size;
944                        if (p->p.z - p->size < bbmin.z) bbmin.z = p->p.z - p->size;
945                        if (p->p.x + p->size > bbmax.x) bbmax.x = p->p.x + p->size;
946                        if (p->p.y + p->size > bbmax.y) bbmax.y = p->p.y + p->size;
947                        if (p->p.z + p->size > bbmax.z) bbmax.z = p->p.z + p->size;
948                        if (shape == SHAPE_UNKNOWN)
949                                shape = (p->shape == Part::SHAPE_DEFAULT) ? SHAPE_OLD : SHAPE_NEW;
950                        else if (shape != SHAPE_ILLEGAL)
951                        {
952                                if ((p->shape == Part::SHAPE_DEFAULT) ^ (shape == SHAPE_OLD))
953                                {
954                                        shape = SHAPE_ILLEGAL;
955                                        logPrintf("Model", "internalCheck", LOG_WARN, "Inconsistent part shapes (mixed old and new shapes)%s", nameForErrors().c_str());
956                                }
957                        }
958                }
959                size = bbmax - bbmin;
960                for (i = 0; i < joints.size(); i++)
961                {
962                        j = (Joint*)joints(i);
963                        // VALIDMINMAX are managed manually when adding joint properties in f0-def!
964                        // (could be made dynamic but not really worth the effort)
965                        VALIDMINMAX(j, Joint, stamina);
966                        VALIDMINMAX(j, Joint, stif);
967                        VALIDMINMAX(j, Joint, rotstif);
968                        VALIDMINMAX(p, Part, vcolor.x);
969                        VALIDMINMAX(p, Part, vcolor.y);
970                        VALIDMINMAX(p, Part, vcolor.z);
971                        j->refno = i;
972                        j->owner = this;
973                        if (j->part1 && j->part2 && (j->part1 != j->part2))
974                        {
975                                j->p1_refno = j->part1->refno;
976                                j->p2_refno = j->part2->refno;
977                                if (checklevel > 0)
978                                {
979                                        if (j->part1->shape == 0)
980                                                j->part1->mass += 1.0;
981                                        if (j->part2->shape == 0)
982                                                j->part2->mass += 1.0;
983                                }
984                                if ((j->usedelta) && ((j->d.x != JOINT_DELTA_MARKER) || (j->d.y != JOINT_DELTA_MARKER) || (j->d.z != JOINT_DELTA_MARKER)))
985                                { // delta positioning -> calc. orient.
986                                        if (j->part2->flags & LINKFLAG)
987                                        {
988                                                ret = 0;
989                                                logPrintf("Model", "internalCheck", LOG_ERROR,
990                                                        "Delta joint cycle detected at Joint #%d%s",
991                                                        i, nameForErrors().c_str());
992                                        }
993                                        j->resetDeltaMarkers();
994                                        j->o = j->rot;
995                                        j->part1->o.transform(j->part2->o, j->o);
996                                        //                      j->part2->o.x=j->part1->o/j->o.x;
997                                        //                      j->part2->o.y=j->part1->o/j->o.y;
998                                        //                      j->part2->o.z=j->part1->o/j->o.z;
999                                        j->part2->p = j->part2->o.transform(j->d) + j->part1->p;
1000                                        j->part2->flags |= LINKFLAG; j->part1->flags |= LINKFLAG; // for delta joint cycle detection
1001                                }
1002                                else
1003                                { // abs.positioning -> calc. delta
1004                                        if (final)
1005                                        {
1006                                                // calc orient delta
1007                                                //                      Orient tmpo(j->part2->o);
1008                                                //                      tmpo*=j->part1->o;
1009                                                Orient tmpo;
1010                                                j->part1->o.revTransform(tmpo, j->part2->o);
1011                                                tmpo.getAngles(j->rot);
1012                                                j->o = j->rot;
1013                                                // calc position delta
1014                                                Pt3D tmpp(j->part2->p);
1015                                                tmpp -= j->part1->p;
1016                                                j->d = j->part2->o.revTransform(tmpp);
1017                                        }
1018                                }
1019                                if (final)
1020                                {
1021                                        if (j->shape != Joint::SHAPE_SOLID)
1022                                        {
1023                                                if (j->d() > getMaxJoint().d.x)
1024                                                {
1025                                                        ret = 0;
1026                                                        logPrintf("Model", "internalCheck", LOG_ERROR, "Joint #%d too long%s", i, nameForErrors().c_str());
1027                                                }
1028                                        }
1029                                }
1030                        }
1031                        else
1032                        {
1033                                logPrintf("Model", "internalCheck", LOG_ERROR, "Illegal part references in Joint #%d%s", i, nameForErrors().c_str());
1034                                ret = 0;
1035                        }
1036                        if (shape != SHAPE_ILLEGAL)
1037                        {
1038                                if ((j->shape == Joint::SHAPE_DEFAULT) ^ (shape == SHAPE_OLD))
1039                                {
1040                                        shape = SHAPE_ILLEGAL;
1041                                        logPrintf("Model", "internalCheck", LOG_WARN, "Inconsistent joint shapes (mixed old and new shapes)%s", nameForErrors().c_str());
1042                                }
1043                        }
1044                }
1045        }
1046#ifdef MODEL_V1_COMPATIBLE
1047        if (!addOldConnectionsInputs())
1048                return 0;
1049#endif
1050
1051        updateNeuroRefno(); // valid refno is important for n-n connections check (later)
1052
1053        for (i = 0; i < neurons.size(); i++)
1054        {
1055                n = (Neuro*)neurons(i);
1056#ifdef MODEL_V1_COMPATIBLE
1057                VALIDMINMAX(n,Neuro,inertia);
1058                VALIDMINMAX(n,Neuro,force);
1059                VALIDMINMAX(n,Neuro,sigmo);
1060                n->conn_refno=-1;
1061                n->weight=1.0;
1062                n->neuro_refno=-1;
1063#endif
1064                n->part_refno = (n->part) ? n->part->refno : -1;
1065                n->joint_refno = (n->joint) ? n->joint->refno : -1;
1066        }
1067
1068        if (parts.size() && (checklevel > 0))
1069        {
1070                for (i = 0; i < parts.size(); i++)
1071                {
1072                        p = (Part*)parts(i);
1073                        if (p->shape == 0)
1074                                if (p->mass <= 0.001)
1075                                        p->mass = 1.0;
1076                        p->flags &= ~LINKFLAG;
1077                }
1078                getPart(0)->flags |= LINKFLAG;
1079                int change = 1;
1080                while (change)
1081                {
1082                        change = 0;
1083                        for (i = 0; i < joints.size(); i++)
1084                        {
1085                                j = (Joint*)joints(i);
1086                                if (j->part1->flags&LINKFLAG)
1087                                {
1088                                        if (!(j->part2->flags&LINKFLAG))
1089                                        {
1090                                                change = 1;
1091                                                j->part2->flags |= LINKFLAG;
1092                                        }
1093                                }
1094                                else
1095                                        if (j->part2->flags&LINKFLAG)
1096                                        {
1097                                        if (!(j->part1->flags&LINKFLAG))
1098                                        {
1099                                                change = 1;
1100                                                j->part1->flags |= LINKFLAG;
1101                                        }
1102                                        }
1103                        }
1104                }
1105                for (i = 0; i < parts.size(); i++)
1106                {
1107                        p = (Part*)parts(i);
1108                        if (!(p->flags&LINKFLAG))
1109                        {
1110                                logPrintf("Model", "internalCheck", LOG_ERROR, "Not all parts connected (eg. Part #0 and Part #%d)%s", i, nameForErrors().c_str());
1111                                ret = 0;
1112                                break;
1113                        }
1114                }
1115        }
1116
1117        for (i = 0; i < joints.size(); i++)
1118        {
1119                j = (Joint*)joints(i);
1120                if (j->p1_refno == j->p2_refno)
1121                {
1122                        logPrintf("Model", "internalCheck", LOG_ERROR, "Illegal self connection, Joint #%d%s", i, nameForErrors().c_str());
1123                        ret = 0;
1124                        break;
1125                }
1126                for (k = i + 1; k < joints.size(); k++)
1127                {
1128                        Joint* j2 = (Joint*)joints(k);
1129                        if (((j->p1_refno == j2->p1_refno) && (j->p2_refno == j2->p2_refno))
1130                                || ((j->p1_refno == j2->p2_refno) && (j->p2_refno == j2->p1_refno)))
1131                        {
1132                                logPrintf("Model", "internalCheck", LOG_ERROR, "Illegal duplicate Joint #%d and Joint #%d%s", i, k, nameForErrors().c_str());
1133                                ret = 0;
1134                                break;
1135                        }
1136                }
1137        }
1138        if (shape == SHAPE_ILLEGAL)
1139                ret = 0;
1140        return ret;
1141}
1142
1143/////////////
1144
1145int Model::getErrorPosition(bool includingwarnings)
1146{
1147        return includingwarnings ?
1148                ((f0errorposition >= 0) ? f0errorposition : f0warnposition)
1149                :
1150                f0errorposition;
1151}
1152
1153const Geno& Model::getGeno() const
1154{
1155        return geno;
1156}
1157
1158const Geno Model::getF0Geno()
1159{
1160        if (buildstatus == building)
1161                logPrintf("Model", "getGeno", LOG_WARN, "Model was not completed - missing close()");
1162        if (buildstatus != valid)
1163                return Geno("", '0', "", "invalid");
1164        if (!f0genoknown)
1165        {
1166                if (autobuildmaps)
1167                {
1168                        initF0Map();
1169                        makeGeno(f0geno, f0map);
1170                }
1171                else
1172                {
1173                        delF0Map();
1174                        makeGeno(f0geno);
1175                }
1176                f0genoknown = 1;
1177        }
1178        return f0geno;
1179}
1180
1181int Model::getPartCount() const
1182{
1183        return parts.size();
1184}
1185
1186Part* Model::getPart(int i) const
1187{
1188        return ((Part*)parts(i));
1189}
1190
1191int Model::getJointCount() const
1192{
1193        return joints.size();
1194}
1195
1196Joint* Model::getJoint(int i) const
1197{
1198        return ((Joint*)joints(i));
1199}
1200
1201int Model::findJoints(SList& result, const Part* part)
1202{
1203        Joint *j;
1204        int n0 = result.size();
1205        if (part)
1206                for (int i = 0; j = (Joint*)joints(i); i++)
1207                        if ((j->part1 == part) || (j->part2 == part)) result += (void*)j;
1208        return result.size() - n0;
1209}
1210
1211int Model::findNeuro(Neuro* n)
1212{
1213        return neurons.find(n);
1214}
1215
1216int Model::findPart(Part* p)
1217{
1218        return parts.find(p);
1219}
1220
1221int Model::findJoint(Joint* j)
1222{
1223        return joints.find(j);
1224}
1225
1226int Model::findJoint(Part *p1, Part *p2)
1227{
1228        Joint* j;
1229        for (int i = 0; j = getJoint(i); i++)
1230                if ((j->part1 == p1) && (j->part2 == p2)) return i;
1231        return -1;
1232}
1233
1234
1235////////////////////
1236
1237#ifdef MODEL_V1_COMPATIBLE
1238void Model::calcOldNeuroCount()
1239{
1240        if (oldneurocount>=0) return;
1241        oldneurocount=reorderToOldLayout();
1242}
1243
1244int Model::old_getNeuroCount()
1245{ calcOldNeuroCount();
1246return oldneurocount;}
1247
1248Neuro* Model::old_getNeuro(int i)
1249{calcOldNeuroCount();
1250return (i<oldneurocount)? (Neuro*)getNeuro(i) : (Neuro*)0;
1251}
1252
1253int Model::old_findNeuro(Neuro* n)
1254{calcOldNeuroCount();
1255return findNeuro(n);}
1256
1257Neuro *Model::old_addNewNeuro()
1258{
1259        int count=old_getNeuroCount();
1260        Neuro *nu=addNewNeuro();
1261        nu->setClassName("N");
1262        moveNeuro(nu->refno,oldneurocount);
1263        oldneurocount=count+1;
1264        return (Neuro*)nu;
1265}
1266#endif
1267
1268///////////////////////
1269
1270int Model::getNeuroCount() const
1271{
1272        return neurons.size();
1273}
1274
1275Neuro* Model::getNeuro(int i) const
1276{
1277        return (Neuro*)neurons(i);
1278}
1279
1280int Model::getConnectionCount() const
1281{
1282        int n = 0;
1283        for (int i = 0; i < getNeuroCount(); i++)
1284                n += getNeuro(i)->getInputCount();
1285        return n;
1286}
1287
1288int Model::findNeuros(SList& result,
1289        const char* classname, const Part* part, const Joint* joint)
1290{
1291        Neuro *nu;
1292        SString cn(classname);
1293        int n0 = result.size();
1294        for (int i = 0; nu = (Neuro*)neurons(i); i++)
1295        {
1296                if (part)
1297                        if (nu->part != part) continue;
1298                if (joint)
1299                        if (nu->joint != joint) continue;
1300                if (classname)
1301                        if (nu->getClassName() != cn) continue;
1302                result += (void*)nu;
1303        }
1304        return result.size() - n0;
1305}
1306
1307///////////////////
1308
1309void Model::disturb(double amount)
1310{
1311        int i;
1312        if (amount <= 0) return;
1313        for (i = 0; i < parts.size(); i++)
1314        {
1315                Part *p = getPart(i);
1316                p->p.x += (rnd01 - 0.5)*amount;
1317                p->p.y += (rnd01 - 0.5)*amount;
1318                p->p.z += (rnd01 - 0.5)*amount;
1319        }
1320        for (i = 0; i < joints.size(); i++)
1321        {
1322                Joint *j = getJoint(i);
1323                Pt3D tmpp(j->part2->p);
1324                tmpp -= j->part1->p;
1325                j->d = j->part2->o.revTransform(tmpp);
1326        }
1327}
1328
1329void Model::move(const Pt3D& shift)
1330{
1331        FOREACH(Part*, p, parts)
1332                p->p += shift;
1333}
1334
1335void Model::rotate(const Orient& rotation)
1336{
1337        FOREACH(Part*, p, parts)
1338        {
1339                p->p = rotation.transform(p->p);
1340                p->setOrient(rotation.transform(p->o));
1341        }
1342}
1343
1344void Model::buildUsingNewShapes(const Model& old, Part::Shape default_shape, float thickness)
1345{
1346        for (int i = 0; i < old.getJointCount(); i++)
1347        {
1348                Joint *oj = old.getJoint(i);
1349                Part *p = addNewPart(default_shape);
1350                p->p = (oj->part1->p + oj->part2->p) / 2;
1351                Orient o;
1352                o.lookAt(oj->part1->p - oj->part2->p);
1353                p->rot = o.getAngles();
1354                p->scale.x = oj->part1->p.distanceTo(oj->part2->p) / 2;
1355                p->scale.y = thickness;
1356                p->scale.z = thickness;
1357        }
1358        for (int i = 0; i < old.getPartCount(); i++)
1359        {
1360                Part *op = old.getPart(i);
1361                for (int j = 0; j < old.getJointCount(); j++)
1362                {
1363                        Joint *oj = old.getJoint(j);
1364                        if ((oj->part1 == op) || (oj->part2 == op))
1365                        {
1366                                for (int j2 = j + 1; j2 < old.getJointCount(); j2++)
1367                                {
1368                                        Joint *oj2 = old.getJoint(j2);
1369                                        if ((oj2->part1 == op) || (oj2->part2 == op))
1370                                        {
1371                                                addNewJoint(getPart(j), getPart(j2), Joint::SHAPE_SOLID);
1372                                        }
1373                                }
1374                                break;
1375                        }
1376                }
1377        }
1378}
1379
1380//////////////////////
1381
1382class MinPart : public Part { public: MinPart() { Param par(f0_part_paramtab, this); par.setMin(); } };
1383class MaxPart : public Part { public: MaxPart() { Param par(f0_part_paramtab, this); par.setMax(); } };
1384class MinJoint : public Joint { public: MinJoint() { Param par(f0_joint_paramtab, this); par.setMin(); } };
1385class MaxJoint : public Joint { public: MaxJoint() { Param par(f0_joint_paramtab, this); par.setMax(); } };
1386class MinNeuro : public Neuro { public: MinNeuro() { Param par(f0_neuro_paramtab, this); par.setMin(); } };
1387class MaxNeuro : public Neuro { public: MaxNeuro() { Param par(f0_neuro_paramtab, this); par.setMax(); } };
1388
1389Part& Model::getMinPart() { static MinPart part; return part; }
1390Part& Model::getMaxPart() { static MaxPart part; return part; }
1391Part& Model::getDefPart() { static Part part; return part; }
1392Joint& Model::getMinJoint() { static MinJoint joint; return joint; }
1393Joint& Model::getMaxJoint() { static MaxJoint joint; return joint; }
1394Joint& Model::getDefJoint() { static Joint joint; return joint; }
1395Neuro& Model::getMinNeuro() { static MinNeuro neuro; return neuro; }
1396Neuro& Model::getMaxNeuro() { static MaxNeuro neuro; return neuro; }
1397Neuro& Model::getDefNeuro() { static Neuro neuro; return neuro; }
Note: See TracBrowser for help on using the repository browser.