// 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 <frams/genetics/geneprops.h>

/*
Testing the influence of Framsticks f1/f4 modifier sequences. See also geneprops_graph.py

Examples:

- experimental exponential friction:
$ ./geneprops_test f 3 -e | sort -n -k 2

fff	0.0158037
ff	0.0366025
Fff	0.0639754
f	0.1
fFf	0.147411
Ff	0.209808
FFf	0.291926
-	0.4
ffF	0.542233
fF	0.729423
FfF	0.975778
F	1.3
fFF	1.7267
FF	2.28827
FFF	3.02733


- legacy friction:
$ ./geneprops_test f 3 -l | sort -n -k 2

fff	0.2048
ff	0.256
f	0.32
-	0.4
Fff	0.7168
fFf	0.8448
Ff	0.896
ffF	1.0048
fF	1.056
F	1.12
FFf	1.3568
FfF	1.5168
fFF	1.6448
FF	1.696
FFF	2.1568


*/

double genePropsValueForModifier(GeneProps& gp, char modifier)
{
	switch (toupper(modifier))
	{
	case 'L': return gp.length; break;
	case 'W': return gp.weight; break;
	case 'F': return gp.friction; break;
	case 'C': return gp.curvedness; break;
	case 'Q': return gp.twist; break;
	case 'E': return gp.energy; break;

	case 'A': return gp.assimilation; break;
	case 'I': return gp.ingestion; break;
	case 'S': return gp.stamina; break;
	case 'M': return gp.muscle_power; break;

	case 'D': return gp.cred; break;
	case 'G': return gp.cgreen; break;
	case 'B': return gp.cblue;  break;
	}
	return 0;
}

int main(int argc, char *argv[])
{
	char modifier = 'L';
	int maxcount = 4;

	enum Function { Legacy, AllChange05, ExperimentalExponential };
	Function fun = AllChange05;

	int argpos = 0;
	for (int i = 1; i < argc; i++)
	{
		const char* arg = argv[i];
		if (arg[0] == '-')
			switch (arg[1])
			{
			case 'l': fun = Legacy; break;
			case 'n': fun = AllChange05; break;
			case 'e': fun = ExperimentalExponential; break;
			case 'h': printf("%s args: [modifier [max_count]] (one of " F14_MODIFIERS ")\n"
				"\t-l = legacy\n"
				"\t-n = modern, change=0.5, normalizeBiol4 disabled\n"
				"\t-e = experimental exponential\n", argv[0]);
				return 0;
			default: fprintf(stderr, "%s: invalid option: %s\n", argv[0], arg); return 2;
			}
		else
			switch (argpos++)
			{
			case 0: modifier = arg[0]; break;
			case 1: maxcount = atoi(arg); break;
			default: fprintf(stderr, "%s: too many arguments: %s\n", argv[0], arg); return 2;
			}
	}

	char modifier_u = toupper(modifier);
	char modifier_l = tolower(modifier);

	GenePropsOps_Exponential exponential_ops;
	GenePropsOps_Legacy legacy_ops;
	GenePropsOps_AllChange05 allchange05_ops;
	GenePropsOps *ops;

	switch (fun)
	{
	case Legacy: ops = &legacy_ops; break;
	case ExperimentalExponential:  ops = &exponential_ops; break;
	case AllChange05:  ops = &allchange05_ops; break;
	default: ops = NULL;
	}

	char *tmp = new char[maxcount + 1];
	for (int count = 0; count <= maxcount; count++)
	{
		int N = 1 << count;
		for (int i = 0; i < N; i++)
		{
			GeneProps props;
			for (int c = 0; c < count; c++)
			{
				char m = ((1 << c) & i) ? modifier_u : modifier_l;
				tmp[c] = m;
				props.executeModifier(m, ops);
			}
			if (count > 0)
				tmp[count] = 0;
			else
			{
				tmp[0] = '-'; tmp[1] = 0;
			}
			double value = genePropsValueForModifier(props, modifier);
			printf("%s\t%g\n", tmp, value);
		}
	}

	return 0;
}