source: cpp/frams/genetics/f9/f9_oper.cpp @ 1313

// This file is a part of Framsticks SDK.  http://www.framsticks.com/
// Copyright (C) 1999-2021  Maciej Komosinski and Szymon Ulatowski.
// See LICENSE.txt for details.

#include "f9_oper.h"
#include "f9_conv.h"
#include <common/nonstd.h> //rndUint, rndDouble
#include <string.h>
#include <assert.h>


#define FIELDSTRUCT GenoOper_f9
static ParamEntry geno_f9_paramtab[] =
{
        { "Genetics: f9", 1, 1, },
        { "f9_mut", 0, 0, "Mutation probability", "f 0 1 0", FIELD(mut_prob), "How many genes should be mutated during a single mutation (1=all genes, 0.1=ten percent, 0=one gene)", },
        { 0, },
};
#undef FIELDSTRUCT


GenoOper_f9::GenoOper_f9()
{
        par.setParamTab(geno_f9_paramtab);
        par.select(this);
        par.setDefault();
        supported_format = '9';
}

int GenoOper_f9::checkValidity(const char* gene, const char *genoname)
{
        if (!gene[0]) return 1; //empty is not valid
        bool ok = true;
        size_t i;
        for (i = 0; i < strlen(gene); i++) if (!strchr(turtle_commands_f9, gene[i])) { ok = false; break; }
        return ok ? GENOPER_OK : int(i) + 1;
}

///Remove all invalid letters from the genotype
int GenoOper_f9::validate(char *&gene, const char *genoname)
{
        SString validated; //new genotype (everything except turtle_commands_f9 is skipped)
        for (size_t i = 0; i < strlen(gene); i++)
                if (strchr(turtle_commands_f9, gene[i])) validated += gene[i];  //validated contains only turtle_commands_f9
        free(gene);
        gene = strdup(validated.c_str()); //reallocate
        return GENOPER_OK;
}

void GenoOper_f9::mutate_replace(char *gene, int i)
{
        char oldgene = gene[i];
        const char *pos = strchr(turtle_commands_f9, oldgene);
        if (pos == NULL) //gene not in the set of allowed commands
                gene[i] = turtle_commands_f9[rndUint(turtle_commands_f9_count)];
        else
                gene[i] = turtle_commands_f9[((pos - turtle_commands_f9) + 1 + rndUint(turtle_commands_f9_count - 1)) % turtle_commands_f9_count]; //always change to a different command
        //printf("%c %c\n", oldgene, gene[i]);
        assert(gene[i] != oldgene);
}

void GenoOper_f9::mutate_add_or_del(char *& gene, int len, int add_0_del_1)
{
        string newgeno(gene);
        if (add_0_del_1 == 0) //add
        {
                int p = rndUint(len + 1);  //random location
                //printf("before add: %s\n",(const char*)newgeno);
                newgeno = newgeno.substr(0, p) + string(turtle_commands_f9 + rndUint(turtle_commands_f9_count), 1) + newgeno.substr(p);
                //printf("after add: %s\n",(const char*)newgeno);
        }
        else //delete
        {
                int p = rndUint(len);  //random location
                //printf("before delete: %s\n",(const char*)newgeno);
                newgeno = newgeno.substr(0, p) + newgeno.substr(p + 1);
                //printf("after delete: %s\n",(const char*)newgeno);
        }
        free(gene);
        gene = strdup(newgeno.c_str()); //reallocate
}

///Simple mutation
int GenoOper_f9::mutate(char *&gene, float &chg, int &method)
{
        method = 0;
        int changes = 0, len = (int)strlen(gene);

        if (mut_prob > 0)
                for (int i = 0; i < len; i++)
                {
                        if (rndDouble(1) < mut_prob) //normalize prob with the length of the genotype
                        {
                                mutate_replace(gene, i);
                                changes++;
                        }
                }

        if (rndDouble(1) < mut_prob) //add or delete a random char
        {
                int change_add_or_del = 1 + (len > 1); //either add a new symbol (1), or delete a random symbol (+(len>1))
                int add_0_del_1 = rndUint(change_add_or_del);
                mutate_add_or_del(gene, len, add_0_del_1);
                changes++;
        }

        if (changes == 0) //if nothing changed so far, then fallback to minimal change: always changes one symbol
        {
                int change_what = len + 1 + (len > 1); //either replace an existing gene, or add a new symbol (+1), or delete a random symbol (+(len>1))
                int change_selected = rndUint(change_what);
                if (change_selected < len)
                        mutate_replace(gene, change_selected);
                else
                        mutate_add_or_del(gene, len, change_selected - len);
                changes++;
        }

        chg = (float)changes / len;
        return changes > 0 ? GENOPER_OK : GENOPER_OPFAIL; //no changes? (should never happen) => OPFAIL so that GenMan will call mutate again
}

///A simple one-point crossover
int GenoOper_f9::crossOver(char *&g1, char *&g2, float& chg1, float& chg2)
{
        int len1 = (int)strlen(g1), len2 = (int)strlen(g2);
        int p1 = rndUint(len1);  //random cut point for the first genotype
        int p2 = rndUint(len2);  //random cut point for the second genotype
        char *child1 = (char*)malloc(p1 + len2 - p2 + 1);
        char *child2 = (char*)malloc(p2 + len1 - p1 + 1);
        strncpy(child1, g1, p1);   strcpy(child1 + p1, g2 + p2);
        strncpy(child2, g2, p2);   strcpy(child2 + p2, g1 + p1);
        free(g1); g1 = child1;
        free(g2); g2 = child2;
        chg1 = (float)p1 / strlen(child1);
        chg2 = (float)p2 / strlen(child2);
        return GENOPER_OK;
}

///Applying some colors and font styles...
uint32_t GenoOper_f9::style(const char *g, int pos)
{
        char ch = g[pos];
        uint32_t style = GENSTYLE_CS(0, GENSTYLE_INVALID); //default, should be changed below
        char *ptr = strchr((char*)turtle_commands_f9, ch);
        if (ptr)
        {
                int pos = ptr - turtle_commands_f9;
                int axis = pos / 2;
                style = GENSTYLE_RGBS(axis == 0 ? 200 : 0, axis == 2 ? 200 : 0, axis == 1 ? 200 : 0, GENSTYLE_NONE);
        }
        return style;
}