source: cpp/frams/genetics/f9/conv_f9.cpp @ 123

// This file is a part of the Framsticks GDK.
// Copyright (C) 2002-2014  Maciej Komosinski and Szymon Ulatowski.  See LICENSE.txt for details.
// Refer to http://www.framsticks.com/ for further information.

#include "conv_f9.h"
#include <frams/model/model.h>
#include <string.h>
#include <common/nonstd_stl.h> //ARRAY_LENGTH

//#define APPLY_DETERMINISTIC_BODY_NOISE

GenoConv_F90::GenoConv_F90()
{
        name = "Turtle3D-ortho encoding";
        in_format  = '9';
        out_format = '0';
        mapsupport = 0; //would be easy and nice to add!
}


const char* turtle_commands_f9 ="LRDUBF";

//const char* turtle_commandsX_f9="-+0000";
//const char* turtle_commandsY_f9="00-+00";
//const char* turtle_commandsZ_f9="0000-+";

SString GenoConv_F90::convert(SString &in, MultiMap *map)
{
        vector<XYZ_LOC> vertices;
        XYZ_LOC current;
        Model m;
        m.open();
        int recently_added=addSegment(m,vertices,current,0xDead);
        for(int i=0;i<in.len();i++)
        {
                char command=in[i];
                char *ptr=strchr((char*)turtle_commands_f9,command);
                if (ptr)
                {
                        int delta[]={0,0,0};
                        int pos=ptr-turtle_commands_f9;
                        int axis=pos/2;
                        int dir=pos%2;
                        (*(delta+axis))+=dir*2-1; //+1 or -1 in the given axis
                        current.add(delta);
                        recently_added=addSegment(m,vertices,current,recently_added);
                }
        }
#ifdef APPLY_DETERMINISTIC_BODY_NOISE
        perturbPartLocations(m);
#endif
        setColors(m);
        m.close();
        return m.getF0Geno().getGene();
}

int GenoConv_F90::addSegment(Model &m,vector<XYZ_LOC> &vertices,const XYZ_LOC &new_vertex,int recently_added)
{
        if (vertices.size()<1)
        {
                return addNewVertex(m,vertices,new_vertex);
        } else
        {
                int vertex_here=findVertexAt(vertices,new_vertex);
                if (vertex_here<0) //need to create a new Part
                {
                        vertex_here=addNewVertex(m,vertices,new_vertex);
                } //else there already exists a Part in new_vertex; new Joint may or may not be needed
                Part *p1=m.getPart(recently_added);
                Part *p2=m.getPart(vertex_here);
                if (m.findJoint(p1,p2)<0 && m.findJoint(p2,p1)<0) //new Joint needed? should always be true if we just created a new Part (vertex_here was <0)
                {
                        m.addNewJoint(p1,p2);
                }
                return vertex_here;
        }
}

int GenoConv_F90::findVertexAt(vector<XYZ_LOC> &vertices,const XYZ_LOC &vertex)
{
        for(int i=0;i<vertices.size();i++)
                if (vertices[i].same_coordinates(vertex)) return i;
        return -1;
}


int GenoConv_F90::addNewVertex(Model &m,vector<XYZ_LOC> &vertices,const XYZ_LOC &new_vertex)
{
        Part *p=new Part;
        p->p.x=new_vertex.x;
        p->p.y=new_vertex.y;
        p->p.z=new_vertex.z;
        m.addPart(p);

        vertices.push_back(new_vertex);
        return vertices.size()-1;
}

double mix(int *colortab,int maxind,double ind)
{
        int indpre=(int)ind;
        int indpost=indpre+1;
        if (indpost>maxind) indpost=maxind;
        int v1=colortab[indpre];
        int v2=colortab[indpost];
        double d1=ind-indpre;
        double d2=indpost-ind;
        double v=indpre==indpost?v1:d2*v1+d1*v2; //d1+d2==1
        return v;
}

void GenoConv_F90::setColors(Model &m) //sets fixed (independent from genes) colors and widths on a model, purely for aesthetic purposes
{
        //a rainbow on Joints: from the first one red, through middle green, to blue or violet - last
        static int r[]={1,1,0,0,0,1};
        static int g[]={0,1,1,1,0,0};
        static int b[]={0,0,0,1,1,1};
        int maxind=ARRAY_LENGTH(r)-1;

        int joints_count=m.getJointCount();
        for(int i=0;i<joints_count;i++)
        {
                Joint *j=m.getJoint(i);
                double x=joints_count<2?0:(double)i/(joints_count-1); //0..1, postion in the rainbow
                double ind=x*maxind;
                j->vcolor.x=mix(r,maxind,ind);
                j->vcolor.y=mix(g,maxind,ind);
                j->vcolor.z=mix(b,maxind,ind);
        }

        int parts_count=m.getPartCount();
        SList jlist;
        for(int i=0;i<parts_count;i++)
        {
                Part *p=m.getPart(i);
                jlist.clear();
                int count=m.findJoints(jlist,p);
                Pt3D averagecolor(0,0,0); //Parts will get averaged colors from all attached Joints
                FOREACH(Joint*,j,jlist)
                        averagecolor+=j->vcolor;
                if (count>5) count=5; //avoid too fat...
                p->vsize=0.3+count/15.0; //the more Joints is attached to a Part, the fatter it is
                p->vcolor=averagecolor/count;
        }
}

void GenoConv_F90::perturbPartLocations(Model &m) //deterministic "body noise"
{
        for(int i=0;i<m.getPartCount();i++)
        {
                Part *p=m.getPart(i);
                Pt3D noise(
                        ((i+1)  %10)-4.5  ,
                        ((3*i+5)%10)-4.5  ,
                        ((7*i+2)%10)-4.5
                        ); //-4.5 .. 4.5 in each axis
                p->p+=noise/1000;
        }
}