source: cpp/frams/genetics/geneprops.cpp @ 1305

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

#include "geneprops.h"
#include <algorithm>

GeneProps GeneProps::standard_values;

GeneProps::GeneProps()
{
        Part_MinMaxDef default_part = Model::getDefPart();
        Joint default_joint = Model::getDefJoint();

        length = 1.0;
        weight = 1.0;
        friction = default_part.friction;
        curvedness = 0.0;
        twist = 0.0;
        energy = 1.0;

        // before 2023-05, the four fields below were aggregated and normalized using normalizeBiol4(), but this normalization only worked for f1 and f4 - other genetic encodings do not perform such normalization. If at all, this should be handled by the expdef depending on the goal of the experiment (e.g., to encourage specialization).
        muscle_power = 0.25; // "biological" property, same as findNeuroClass("BendMuscle or RotMuscle")->paraminterface->getDoubleById("power")
        assimilation = default_part.assim; // "biological" property
        stamina = default_joint.stamina; // "biological" property
        ingestion = default_part.ingest; // "biological" property

        muscle_bend_range = 1.0; // same as findNeuroClass("BendMuscle")->paraminterface->getDoubleById("range")
        muscle_reset_range = true;

        cred = default_part.vcolor.x;
        cgreen = default_part.vcolor.y;
        cblue = default_part.vcolor.z;
}

void GeneProps::normalizeBiol4()
{
        // make them sum to 1
        double sum = muscle_power + assimilation + stamina + ingestion;
        if (sum == 0)
        {
                muscle_power = assimilation = stamina = ingestion = 0.25;
        }
        else
        {
                muscle_power /= sum;
                assimilation /= sum;
                stamina /= sum;
                ingestion /= sum;
        }
}

int GeneProps::executeModifier(char modif, GenePropsOps* ops)
{
        if (ops == NULL)
                ops = getDefaultOps();

#define APPLY(name) ops->name->apply(name, modif)
#define APPLY_AND_MAYBE_NORMALIZEBIOL4(name) {APPLY(name); if (ops->use_normalizebiol4) normalizeBiol4();}

        switch (toupper(modif))
        {
        case 'L': APPLY(length);
                length = std::min(length, Model::getMaxJoint().d.x); break;
                break;

        case 'l': APPLY(length);
                length = std::max(length, Model::getMinJoint().d.x); break;
                break;

        case 'W': APPLY(weight); break;
        case 'F': APPLY(friction); break;
        case 'C': APPLY(curvedness); break;
        case 'Q': APPLY(twist); break;
        case 'E': APPLY(energy); break;

        case 'A': APPLY_AND_MAYBE_NORMALIZEBIOL4(assimilation); break;
        case 'I': APPLY_AND_MAYBE_NORMALIZEBIOL4(ingestion); break;
        case 'S': APPLY_AND_MAYBE_NORMALIZEBIOL4(stamina); break;
        case 'M': APPLY_AND_MAYBE_NORMALIZEBIOL4(muscle_power); break;

        case 'D': APPLY(cred); break;
        case 'G': APPLY(cgreen); break;
        case 'B': APPLY(cblue); break;

        default: return -1;
        }
        return 0;

#undef APPLY
#undef APPLY_AND_MAYBE_NORMALIZEBIOL4
}

void GeneProps::propagateAlong(bool use_f1_muscle_reset_range, GenePropsOps* ops)
{
        length = 0.5 * length + 0.5 * standard_values.length;
        weight += (standard_values.weight - weight) * 0.5;
        friction = 0.8 * friction + 0.2 * standard_values.friction;
        curvedness = 0.66 * curvedness;
        twist = 0.66 * twist;

        assimilation = 0.8 * assimilation + 0.2 * standard_values.assimilation;
        ingestion = 0.8 * ingestion + 0.2 * standard_values.ingestion;
        stamina = 0.8 * stamina + 0.2 * standard_values.stamina;
        muscle_power = 0.8 * muscle_power + 0.2 * standard_values.muscle_power;

        if (ops == NULL)
                ops = getDefaultOps();
        if (ops->use_normalizebiol4) normalizeBiol4();

        if (use_f1_muscle_reset_range)
        {
                if (muscle_reset_range) muscle_bend_range = 1.0; else muscle_reset_range = true;
        }
}

////////////////////////////////////////////////////

void GenePropsOp::apply(double &value, char modif) const
{
        if (isupper(modif))
                value = increase(value);
        else
                value = decrease(value);
}

double GenePropsOp_Legacy::increase(double value) const
{
        return value + (maxvalue - value) * change;
}

double GenePropsOp_Legacy::decrease(double value) const
{
        return value + (minvalue - value) * revchange;
}

GenePropsOp_Legacy::GenePropsOp_Legacy(double minvalue, double maxvalue, double defvalue, double change, double revchange)
{
        this->minvalue = minvalue;
        this->maxvalue = maxvalue;
        this->defvalue = defvalue;
        this->change = change;
        this->revchange = (revchange < 0) ? change : revchange;
}

GenePropsOp_Exponential::GenePropsOp_Exponential(double minvalue, double maxvalue, double defvalue, double change)
        :GenePropsOp_NormalizedAndScaled(change)
{
        double mid = (maxvalue + minvalue) / 2;
        if (fabs(mid - defvalue) < 0.01)
        {
                linear = true;
                a = (maxvalue - minvalue) / 2;
                b = defvalue;
        }
        else
        {
                linear = false;
                a = -(maxvalue - defvalue) / (minvalue - defvalue);
                b = (minvalue * minvalue - 2 * defvalue * minvalue + defvalue * defvalue) / (minvalue + maxvalue - 2 * defvalue);
                c = (maxvalue * minvalue - defvalue * defvalue) / (minvalue + maxvalue - 2 * defvalue);
                log_a = log(a);
        }
}

double GenePropsOp_Exponential::scale(double value) const
{
        if (linear)
                return a * value + b;
        else
                return pow(a, (value + 1)) * b + c;
}

double GenePropsOp_Exponential::scaleInv(double value) const
{
        if (linear)
                return (value - b) / a;
        else
                return -(log_a - log(value / b - c / b)) / log_a;
}

////////////////////////////////////////////////////////////////////


GenePropsOps::~GenePropsOps()
{
        delete length;
        delete curvedness;
        delete weight;
        delete friction;
        delete muscle_power;
        delete assimilation;
        delete stamina;
        delete ingestion;
        delete twist;
        delete energy;
        delete cred;
        delete cgreen;
        delete cblue;
}

GenePropsOps_Legacy::GenePropsOps_Legacy()
{
        use_normalizebiol4 = true;

        length = new GenePropsOp_Legacy(0.33, 2.0, GeneProps::standard_values.length, 0.3);
        weight = new GenePropsOp_Legacy(0.5, 2.0, GeneProps::standard_values.weight, 0.3);
        friction = new GenePropsOp_Legacy(0, 4.0, GeneProps::standard_values.friction, 0.2);
        curvedness = new GenePropsOp_Legacy(-2, 2, GeneProps::standard_values.curvedness, 0.25);
        twist = new GenePropsOp_Legacy(-M_PI_2, M_PI_2, GeneProps::standard_values.twist, 0.3);
        energy = new GenePropsOp_Legacy(0, 10, GeneProps::standard_values.energy, 0.1);

        assimilation = new GenePropsOp_Legacy(0, 1, GeneProps::standard_values.assimilation, 0.8, 0.4);
        ingestion = new GenePropsOp_Legacy(0, 1, GeneProps::standard_values.ingestion, 0.8, 0.4);
        stamina = new GenePropsOp_Legacy(0, 1, GeneProps::standard_values.stamina, 0.8, 0.4);
        muscle_power = new GenePropsOp_Legacy(0, 1, GeneProps::standard_values.muscle_power, 0.8, 0.4);

        cred = new GenePropsOp_Legacy(0, 1, GeneProps::standard_values.cred, 0.25);
        cgreen = new GenePropsOp_Legacy(0, 1, GeneProps::standard_values.cgreen, 0.25);
        cblue = new GenePropsOp_Legacy(0, 1, GeneProps::standard_values.cblue, 0.25);
}

GenePropsOps_AllChange05::GenePropsOps_AllChange05()
{
        use_normalizebiol4 = false;

        constexpr float CHANGE = 0.5;
        auto fields = { length,curvedness,weight,friction,muscle_power,assimilation,stamina,ingestion,twist,energy,cred,cgreen,cblue };
        for (auto f : fields)
        {
                auto f_gpo = dynamic_cast<GenePropsOp_Legacy*>(f);
                if (f_gpo)
                        f_gpo->change = f_gpo->revchange = CHANGE;
        }

        delete curvedness;
        curvedness = new GenePropsOp_Legacy(-M_PI_2, M_PI_2, GeneProps::standard_values.curvedness, CHANGE);
}

GenePropsOps_Exponential::GenePropsOps_Exponential()
{
        length = new GenePropsOp_Exponential(0.33, 2.0, GeneProps::standard_values.length);
        weight = new GenePropsOp_Exponential(0.5, 2.0, GeneProps::standard_values.weight);
        friction = new GenePropsOp_Exponential(0, 4.0, GeneProps::standard_values.friction);
        curvedness = new GenePropsOp_Exponential(-2, 2, GeneProps::standard_values.curvedness);
        twist = new GenePropsOp_Exponential(-M_PI_2, M_PI_2, GeneProps::standard_values.twist);
        energy = new GenePropsOp_Exponential(0, 10, GeneProps::standard_values.energy);

        assimilation = new GenePropsOp_Exponential(0, 1, GeneProps::standard_values.assimilation);
        ingestion = new GenePropsOp_Exponential(0, 1, GeneProps::standard_values.ingestion);
        stamina = new GenePropsOp_Exponential(0, 1, GeneProps::standard_values.stamina);
        muscle_power = new GenePropsOp_Exponential(0, 1, GeneProps::standard_values.muscle_power);

        cred = new GenePropsOp_Exponential(0, 1, GeneProps::standard_values.cred);
        cgreen = new GenePropsOp_Exponential(0, 1, GeneProps::standard_values.cgreen);
        cblue = new GenePropsOp_Exponential(0, 1, GeneProps::standard_values.cblue);
}

GenePropsOps* GeneProps::getAllChange05Ops()
{
        static GenePropsOps_AllChange05 ops;
        return &ops;
}

GenePropsOps* GeneProps::getLegacyOps()
{
        static GenePropsOps_Legacy ops;
        return &ops;
}

static GenePropsOps* default_ops = NULL;

GenePropsOps* GeneProps::getDefaultOps()
{
        if (!default_ops)
                default_ops = getAllChange05Ops();
        return default_ops;
}

void GeneProps::setDefaultOps(GenePropsOps* ops)
{
        default_ops = ops;
}